src/msw/utilsexc.cpp

   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(__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 #ifdef __WIN32__
  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     data->handler     = handler;
 264
 265     DWORD tid;
 266     HANDLE hThread = ::CreateThread(NULL,
 267                                     0,
 268                                     (LPTHREAD_START_ROUTINE)wxExecuteThread,
 269                                     (void *)data,
 270                                     0,
 271                                     &tid);
 272
 273     if ( !hThread )
 274     {
 275         wxLogLastError("CreateThread in wxExecute");
 276
 277         DestroyWindow(hwnd);
 278         delete data;
 279
 280         // the process still started up successfully...
 281         return pi.dwProcessId;
 282     }
 283
 284     if ( !sync )
 285     {
 286         // clean up will be done when the process terminates
 287         return pi.dwProcessId;
 288     }
 289
 290     // waiting until command executed
 291     while ( data->state )
 292         wxYield();
 293
 294     delete data;
 295
 296     return pi.dwProcessId;
 297 #endif // 0/1
 298 #else // Win16
 299     long instanceID = WinExec((LPCSTR) WXSTRINGCAST command, SW_SHOW);
 300     if (instanceID < 32) return(0);
 301
 302     if (sync) {
 303         int running;
 304         do {
 305             wxYield();
 306             running = GetModuleUsage((HANDLE)instanceID);
 307         } while (running);
 308     }
 309
 310     return(instanceID);
 311 #endif // Win16/32
 312 }
 313
 314 long wxExecute(char **argv, bool sync, wxProcess *handler)
 315 {
 316     wxString command;
 317
 318     while ( *argv != NULL )
 319     {
 320         command << *argv++ << ' ';
 321     }
 322
 323     command.RemoveLast();
 324
 325     return wxExecute(command, sync, handler);
 326 }