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