]>
Commit | Line | Data |
---|---|---|
1 | ///////////////////////////////////////////////////////////////////////////// | |
2 | // Name: utilsexec.cpp | |
3 | // Purpose: Various utilities | |
4 | // Author: Julian Smart | |
5 | // Modified by: | |
6 | // Created: 04/01/98 | |
7 | // RCS-ID: $Id$ | |
8 | // Copyright: (c) Julian Smart and Markus Holzem | |
9 | // Licence: wxWindows license | |
10 | ///////////////////////////////////////////////////////////////////////////// | |
11 | ||
12 | #ifdef __GNUG__ | |
13 | #pragma implementation | |
14 | #endif | |
15 | ||
16 | // For compilers that support precompilation, includes "wx.h". | |
17 | #include "wx/wxprec.h" | |
18 | ||
19 | #ifdef __BORLANDC__ | |
20 | #pragma hdrstop | |
21 | #endif | |
22 | ||
23 | #ifndef WX_PRECOMP | |
24 | #include "wx/setup.h" | |
25 | #include "wx/utils.h" | |
26 | #include "wx/app.h" | |
27 | #include "wx/intl.h" | |
28 | #endif | |
29 | ||
30 | #include "wx/log.h" | |
31 | ||
32 | #ifdef __WIN32__ | |
33 | #include "wx/process.h" | |
34 | #endif | |
35 | ||
36 | #include "wx/msw/private.h" | |
37 | ||
38 | #include <windows.h> | |
39 | ||
40 | #include <ctype.h> | |
41 | ||
42 | #if !defined(__GNUWIN32__) && !defined(__WXWINE__) && !defined(__SALFORDC__) | |
43 | #include <direct.h> | |
44 | #ifndef __MWERKS__ | |
45 | #include <dos.h> | |
46 | #endif | |
47 | #endif | |
48 | ||
49 | #ifdef __GNUWIN32__ | |
50 | #ifndef __TWIN32__ | |
51 | #include <sys/unistd.h> | |
52 | #include <sys/stat.h> | |
53 | #endif | |
54 | #endif | |
55 | ||
56 | #if defined(__WIN32__) && !defined(__WXWINE__) | |
57 | #include <io.h> | |
58 | ||
59 | #ifndef __GNUWIN32__ | |
60 | #include <shellapi.h> | |
61 | #endif | |
62 | #endif | |
63 | ||
64 | #include <stdio.h> | |
65 | #include <stdlib.h> | |
66 | #include <string.h> | |
67 | #ifndef __WATCOMC__ | |
68 | #if !(defined(_MSC_VER) && (_MSC_VER > 800)) | |
69 | #include <errno.h> | |
70 | #endif | |
71 | #endif | |
72 | #include <stdarg.h> | |
73 | ||
74 | // this message is sent when the process we're waiting for terminates | |
75 | #define wxWM_PROC_TERMINATED (WM_USER + 10000) | |
76 | ||
77 | // structure describing the process we're being waiting for | |
78 | struct wxExecuteData | |
79 | { | |
80 | public: | |
81 | ~wxExecuteData() | |
82 | { | |
83 | #ifndef __WIN16__ | |
84 | if ( !::CloseHandle(hProcess) ) | |
85 | { | |
86 | wxLogLastError("CloseHandle(hProcess)"); | |
87 | } | |
88 | #endif | |
89 | } | |
90 | ||
91 | HWND hWnd; // window to send wxWM_PROC_TERMINATED to | |
92 | HANDLE hProcess; // handle of the process | |
93 | DWORD dwProcessId; // pid of the process | |
94 | wxProcess *handler; | |
95 | DWORD dwExitCode; // the exit code of the process | |
96 | bool state; // set to FALSE when the process finishes | |
97 | }; | |
98 | ||
99 | ||
100 | #ifdef __WIN32__ | |
101 | static DWORD wxExecuteThread(wxExecuteData *data) | |
102 | { | |
103 | WaitForSingleObject(data->hProcess, INFINITE); | |
104 | ||
105 | // get the exit code | |
106 | if ( !GetExitCodeProcess(data->hProcess, &data->dwExitCode) ) | |
107 | { | |
108 | wxLogLastError("GetExitCodeProcess"); | |
109 | } | |
110 | ||
111 | wxASSERT_MSG( data->dwExitCode != STILL_ACTIVE, | |
112 | wxT("process should have terminated") ); | |
113 | ||
114 | // send a message indicating process termination to the window | |
115 | SendMessage(data->hWnd, wxWM_PROC_TERMINATED, 0, (LPARAM)data); | |
116 | ||
117 | return 0; | |
118 | } | |
119 | ||
120 | // window procedure of a hidden window which is created just to receive | |
121 | // the notification message when a process exits | |
122 | LRESULT APIENTRY _EXPORT wxExecuteWindowCbk(HWND hWnd, UINT message, | |
123 | WPARAM wParam, LPARAM lParam) | |
124 | { | |
125 | if ( message == wxWM_PROC_TERMINATED ) | |
126 | { | |
127 | DestroyWindow(hWnd); // we don't need it any more | |
128 | ||
129 | wxExecuteData *data = (wxExecuteData *)lParam; | |
130 | if ( data->handler ) | |
131 | { | |
132 | data->handler->OnTerminate((int)data->dwProcessId, | |
133 | (int)data->dwExitCode); | |
134 | } | |
135 | ||
136 | if ( data->state ) | |
137 | { | |
138 | // we're executing synchronously, tell the waiting thread | |
139 | // that the process finished | |
140 | data->state = 0; | |
141 | } | |
142 | else | |
143 | { | |
144 | // asynchronous execution - we should do the clean up | |
145 | delete data; | |
146 | } | |
147 | } | |
148 | ||
149 | return 0; | |
150 | } | |
151 | #endif | |
152 | ||
153 | extern wxChar wxPanelClassName[]; | |
154 | ||
155 | long wxExecute(const wxString& command, bool sync, wxProcess *handler) | |
156 | { | |
157 | wxCHECK_MSG( !!command, 0, wxT("empty command in wxExecute") ); | |
158 | ||
159 | #if defined(__WIN32__) && !defined(__TWIN32__) | |
160 | // the old code is disabled because we really need a process handle | |
161 | // if we want to execute it asynchronously or even just get its | |
162 | // return code and for this we must use CreateProcess() and not | |
163 | // ShellExecute() | |
164 | #if 0 | |
165 | // isolate command and arguments | |
166 | wxString commandName; | |
167 | bool insideQuotes = FALSE; | |
168 | const char *pc; | |
169 | for ( pc = command.c_str(); *pc != '\0'; pc++ ) | |
170 | { | |
171 | switch ( *pc ) | |
172 | { | |
173 | case ' ': | |
174 | case '\t': | |
175 | if ( !insideQuotes ) | |
176 | break; | |
177 | // fall through | |
178 | ||
179 | case '"': | |
180 | insideQuotes = !insideQuotes; | |
181 | // fall through | |
182 | ||
183 | default: | |
184 | commandName += *pc; | |
185 | continue; // skip the next break | |
186 | } | |
187 | ||
188 | // only reached for space not inside quotes | |
189 | break; | |
190 | } | |
191 | ||
192 | wxString commandArgs = pc; | |
193 | ||
194 | wxWindow *winTop = wxTheApp->GetTopWindow(); | |
195 | HWND hwndTop = (HWND)(winTop ? winTop->GetHWND() : 0); | |
196 | ||
197 | HANDLE result; | |
198 | #ifdef __GNUWIN32__ | |
199 | result = ShellExecute(hwndTop, | |
200 | (const wchar_t)"open", | |
201 | (const wchar_t)commandName, | |
202 | (const wchar_t)commandArgs, | |
203 | (const wchar_t)NULL, | |
204 | SW_SHOWNORMAL); | |
205 | #else // !GNUWIN32 | |
206 | result = ShellExecute(hwndTop, "open", commandName, | |
207 | commandArgs, NULL, SW_SHOWNORMAL); | |
208 | #endif // GNUWIN32 | |
209 | ||
210 | if ( ((long)result) <= 32 ) | |
211 | wxLogSysError(_("Can't execute command '%s'"), command.c_str()); | |
212 | ||
213 | return result; | |
214 | #else // 1 | |
215 | // create the process | |
216 | STARTUPINFO si; | |
217 | wxZeroMemory(si); | |
218 | ||
219 | si.cb = sizeof(si); | |
220 | ||
221 | PROCESS_INFORMATION pi; | |
222 | ||
223 | if ( ::CreateProcess( | |
224 | NULL, // application name (use only cmd line) | |
225 | (wxChar *)command.c_str(), // full command line | |
226 | NULL, // security attributes: defaults for both | |
227 | NULL, // the process and its main thread | |
228 | FALSE, // don't inherit handles | |
229 | CREATE_DEFAULT_ERROR_MODE, // flags | |
230 | NULL, // environment (use the same) | |
231 | NULL, // current directory (use the same) | |
232 | &si, // startup info (unused here) | |
233 | &pi // process info | |
234 | ) == 0 ) | |
235 | { | |
236 | wxLogSysError(_("Execution of command '%s' failed"), command.c_str()); | |
237 | ||
238 | return 0; | |
239 | } | |
240 | ||
241 | // close unneeded handle | |
242 | if ( !::CloseHandle(pi.hThread) ) | |
243 | wxLogLastError("CloseHandle(hThread)"); | |
244 | ||
245 | // create a hidden window to receive notification about process | |
246 | // termination | |
247 | HWND hwnd = ::CreateWindow(wxPanelClassName, NULL, 0, 0, 0, 0, 0, NULL, | |
248 | (HMENU)NULL, wxGetInstance(), 0); | |
249 | wxASSERT_MSG( hwnd, wxT("can't create a hidden window for wxExecute") ); | |
250 | ||
251 | FARPROC ExecuteWindowInstance = MakeProcInstance((FARPROC)wxExecuteWindowCbk, | |
252 | wxGetInstance()); | |
253 | ||
254 | ::SetWindowLong(hwnd, GWL_WNDPROC, (LONG) ExecuteWindowInstance); | |
255 | ||
256 | // Alloc data | |
257 | wxExecuteData *data = new wxExecuteData; | |
258 | data->hProcess = pi.hProcess; | |
259 | data->dwProcessId = pi.dwProcessId; | |
260 | data->hWnd = hwnd; | |
261 | data->state = sync; | |
262 | if ( sync ) | |
263 | { | |
264 | wxASSERT_MSG( !handler, wxT("wxProcess param ignored for sync execution") ); | |
265 | ||
266 | data->handler = NULL; | |
267 | } | |
268 | else | |
269 | { | |
270 | // may be NULL or not | |
271 | data->handler = handler; | |
272 | } | |
273 | ||
274 | DWORD tid; | |
275 | HANDLE hThread = ::CreateThread(NULL, | |
276 | 0, | |
277 | (LPTHREAD_START_ROUTINE)wxExecuteThread, | |
278 | (void *)data, | |
279 | 0, | |
280 | &tid); | |
281 | ||
282 | if ( !hThread ) | |
283 | { | |
284 | wxLogLastError("CreateThread in wxExecute"); | |
285 | ||
286 | DestroyWindow(hwnd); | |
287 | delete data; | |
288 | ||
289 | // the process still started up successfully... | |
290 | return pi.dwProcessId; | |
291 | } | |
292 | ||
293 | if ( !sync ) | |
294 | { | |
295 | // clean up will be done when the process terminates | |
296 | ||
297 | // return the pid | |
298 | return pi.dwProcessId; | |
299 | } | |
300 | ||
301 | // waiting until command executed | |
302 | while ( data->state ) | |
303 | wxYield(); | |
304 | ||
305 | DWORD dwExitCode = data->dwExitCode; | |
306 | delete data; | |
307 | ||
308 | // return the exit code | |
309 | return dwExitCode; | |
310 | #endif // 0/1 | |
311 | #else // Win16 | |
312 | long instanceID = WinExec((LPCSTR) WXSTRINGCAST command, SW_SHOW); | |
313 | if (instanceID < 32) return(0); | |
314 | ||
315 | if (sync) { | |
316 | int running; | |
317 | do { | |
318 | wxYield(); | |
319 | running = GetModuleUsage((HINSTANCE)instanceID); | |
320 | } while (running); | |
321 | } | |
322 | ||
323 | return(instanceID); | |
324 | #endif // Win16/32 | |
325 | } | |
326 | ||
327 | long wxExecute(char **argv, bool sync, wxProcess *handler) | |
328 | { | |
329 | wxString command; | |
330 | ||
331 | while ( *argv != NULL ) | |
332 | { | |
333 | command << *argv++ << ' '; | |
334 | } | |
335 | ||
336 | command.RemoveLast(); | |
337 | ||
338 | return wxExecute(command, sync, handler); | |
339 | } | |
340 | ||
341 | bool wxGetFullHostName(wxChar *buf, int maxSize) | |
342 | { | |
343 | #if defined(__WIN32__) && !defined(__TWIN32__) | |
344 | DWORD nSize = maxSize ; | |
345 | if ( !::GetComputerName(buf, &nSize) ) | |
346 | { | |
347 | wxLogLastError("GetComputerName"); | |
348 | ||
349 | return FALSE; | |
350 | } | |
351 | #else | |
352 | char *sysname; | |
353 | const char *default_host = "noname"; | |
354 | static const char WX_SECTION[] = "wxWindows"; | |
355 | static const char eHOSTNAME[] = "HostName"; | |
356 | ||
357 | if ((sysname = getenv("SYSTEM_NAME")) == NULL) { | |
358 | GetProfileString(WX_SECTION, eHOSTNAME, default_host, buf, maxSize - 1); | |
359 | } else | |
360 | strncpy(buf, sysname, maxSize - 1); | |
361 | buf[maxSize] = '\0'; | |
362 | if ( *buf == '\0' ) | |
363 | { | |
364 | wxLogLastError("GetComputerName"); | |
365 | ||
366 | return FALSE; | |
367 | } | |
368 | #endif | |
369 | return TRUE; | |
370 | } |