src/msw/utilsexc.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/msw/utilsexc.cpp
   3 // Purpose:     wxExecute implementation for MSW
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1998-2002 wxWidgets dev team
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 // For compilers that support precompilation, includes "wx.h".
  21 #include "wx/wxprec.h"
  22
  23 #ifdef __BORLANDC__
  24     #pragma hdrstop
  25 #endif
  26
  27 #ifndef WX_PRECOMP
  28     #include "wx/utils.h"
  29     #include "wx/app.h"
  30     #include "wx/intl.h"
  31     #include "wx/log.h"
  32     #if wxUSE_STREAMS
  33         #include "wx/stream.h"
  34     #endif
  35     #include "wx/module.h"
  36 #endif
  37
  38 #include "wx/process.h"
  39 #include "wx/thread.h"
  40 #include "wx/apptrait.h"
  41 #include "wx/vector.h"
  42
  43
  44 #include "wx/msw/private.h"
  45
  46 #include <ctype.h>
  47
  48 #if !defined(__GNUWIN32__) && !defined(__WXMICROWIN__) && !defined(__WXWINCE__)
  49     #include <direct.h>
  50 #ifndef __MWERKS__
  51     #include <dos.h>
  52 #endif
  53 #endif
  54
  55 #if defined(__GNUWIN32__)
  56     #include <sys/unistd.h>
  57     #include <sys/stat.h>
  58 #endif
  59
  60 #if !defined(__WXMICROWIN__) && !defined(__WXWINCE__)
  61     #ifndef __UNIX__
  62         #include <io.h>
  63     #endif
  64
  65     #ifndef __GNUWIN32__
  66         #include <shellapi.h>
  67     #endif
  68 #endif
  69
  70 #include <stdio.h>
  71 #include <stdlib.h>
  72 #include <string.h>
  73 #ifndef __WATCOMC__
  74     #if !(defined(_MSC_VER) && (_MSC_VER > 800))
  75         #include <errno.h>
  76     #endif
  77 #endif
  78 #include <stdarg.h>
  79
  80 #if wxUSE_IPC
  81     #include "wx/dde.h"         // for WX_DDE hack in wxExecute
  82 #endif // wxUSE_IPC
  83
  84 // implemented in utils.cpp
  85 extern "C" WXDLLIMPEXP_BASE HWND
  86 wxCreateHiddenWindow(LPCTSTR *pclassname, LPCTSTR classname, WNDPROC wndproc);
  87
  88 // ----------------------------------------------------------------------------
  89 // constants
  90 // ----------------------------------------------------------------------------
  91
  92 // this message is sent when the process we're waiting for terminates
  93 #define wxWM_PROC_TERMINATED (WM_USER + 10000)
  94
  95 // ----------------------------------------------------------------------------
  96 // this module globals
  97 // ----------------------------------------------------------------------------
  98
  99 // we need to create a hidden window to receive the process termination
 100 // notifications and for this we need a (Win) class name for it which we will
 101 // register the first time it's needed
 102 static const wxChar *wxMSWEXEC_WNDCLASSNAME = wxT("_wxExecute_Internal_Class");
 103 static const wxChar *gs_classForHiddenWindow = NULL;
 104
 105 // event used to wake up threads waiting in wxExecuteThread
 106 static HANDLE gs_heventShutdown = NULL;
 107
 108 // handles of all threads monitoring the execution of asynchronously running
 109 // processes
 110 static wxVector<HANDLE> gs_asyncThreads;
 111
 112 // ----------------------------------------------------------------------------
 113 // private types
 114 // ----------------------------------------------------------------------------
 115
 116 // structure describing the process we're being waiting for
 117 struct wxExecuteData
 118 {
 119 public:
 120     ~wxExecuteData()
 121     {
 122         if ( !::CloseHandle(hProcess) )
 123         {
 124             wxLogLastError(wxT("CloseHandle(hProcess)"));
 125         }
 126     }
 127
 128     HWND       hWnd;          // window to send wxWM_PROC_TERMINATED to
 129     HANDLE     hProcess;      // handle of the process
 130     DWORD      dwProcessId;   // pid of the process
 131     wxProcess *handler;
 132     DWORD      dwExitCode;    // the exit code of the process
 133     bool       state;         // set to false when the process finishes
 134 };
 135
 136 class wxExecuteModule : public wxModule
 137 {
 138 public:
 139     virtual bool OnInit() { return true; }
 140     virtual void OnExit()
 141     {
 142         if ( gs_heventShutdown )
 143         {
 144             // stop any threads waiting for the termination of asynchronously
 145             // running processes
 146             if ( !::SetEvent(gs_heventShutdown) )
 147             {
 148                 wxLogDebug(_T("Failed to set shutdown event in wxExecuteModule"));
 149             }
 150
 151             ::CloseHandle(gs_heventShutdown);
 152             gs_heventShutdown = NULL;
 153
 154             // now wait until they terminate
 155             if ( !gs_asyncThreads.empty() )
 156             {
 157                 const size_t numThreads = gs_asyncThreads.size();
 158
 159                 if ( ::WaitForMultipleObjects
 160                        (
 161                         numThreads,
 162                         &gs_asyncThreads[0],
 163                         TRUE,   // wait for all of them to become signalled
 164                         3000    // long but finite value
 165                        ) == WAIT_TIMEOUT )
 166                 {
 167                     wxLogDebug(_T("Failed to stop all wxExecute monitor threads"));
 168                 }
 169
 170                 for ( size_t n = 0; n < numThreads; n++ )
 171                 {
 172                     ::CloseHandle(gs_asyncThreads[n]);
 173                 }
 174
 175                 gs_asyncThreads.clear();
 176             }
 177         }
 178
 179         if ( gs_classForHiddenWindow )
 180         {
 181             if ( !::UnregisterClass(wxMSWEXEC_WNDCLASSNAME, wxGetInstance()) )
 182             {
 183                 wxLogLastError(_T("UnregisterClass(wxExecClass)"));
 184             }
 185
 186             gs_classForHiddenWindow = NULL;
 187         }
 188     }
 189
 190 private:
 191     DECLARE_DYNAMIC_CLASS(wxExecuteModule)
 192 };
 193
 194 IMPLEMENT_DYNAMIC_CLASS(wxExecuteModule, wxModule)
 195
 196 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 197
 198 // ----------------------------------------------------------------------------
 199 // wxPipeStreams
 200 // ----------------------------------------------------------------------------
 201
 202 class wxPipeInputStream: public wxInputStream
 203 {
 204 public:
 205     wxPipeInputStream(HANDLE hInput);
 206     virtual ~wxPipeInputStream();
 207
 208     // returns true if the pipe is still opened
 209     bool IsOpened() const { return m_hInput != INVALID_HANDLE_VALUE; }
 210
 211     // returns true if there is any data to be read from the pipe
 212     virtual bool CanRead() const;
 213
 214 protected:
 215     size_t OnSysRead(void *buffer, size_t len);
 216
 217 protected:
 218     HANDLE m_hInput;
 219
 220     DECLARE_NO_COPY_CLASS(wxPipeInputStream)
 221 };
 222
 223 class wxPipeOutputStream: public wxOutputStream
 224 {
 225 public:
 226     wxPipeOutputStream(HANDLE hOutput);
 227     virtual ~wxPipeOutputStream() { Close(); }
 228     bool Close();
 229
 230 protected:
 231     size_t OnSysWrite(const void *buffer, size_t len);
 232
 233 protected:
 234     HANDLE m_hOutput;
 235
 236     DECLARE_NO_COPY_CLASS(wxPipeOutputStream)
 237 };
 238
 239 // define this to let wxexec.cpp know that we know what we're doing
 240 #define _WX_USED_BY_WXEXECUTE_
 241 #include "../common/execcmn.cpp"
 242
 243 // ----------------------------------------------------------------------------
 244 // wxPipe represents a Win32 anonymous pipe
 245 // ----------------------------------------------------------------------------
 246
 247 class wxPipe
 248 {
 249 public:
 250     // the symbolic names for the pipe ends
 251     enum Direction
 252     {
 253         Read,
 254         Write
 255     };
 256
 257     // default ctor doesn't do anything
 258     wxPipe() { m_handles[Read] = m_handles[Write] = INVALID_HANDLE_VALUE; }
 259
 260     // create the pipe, return true if ok, false on error
 261     bool Create()
 262     {
 263         // default secutiry attributes
 264         SECURITY_ATTRIBUTES security;
 265
 266         security.nLength              = sizeof(security);
 267         security.lpSecurityDescriptor = NULL;
 268         security.bInheritHandle       = TRUE; // to pass it to the child
 269
 270         if ( !::CreatePipe(&m_handles[0], &m_handles[1], &security, 0) )
 271         {
 272             wxLogSysError(_("Failed to create an anonymous pipe"));
 273
 274             return false;
 275         }
 276
 277         return true;
 278     }
 279
 280     // return true if we were created successfully
 281     bool IsOk() const { return m_handles[Read] != INVALID_HANDLE_VALUE; }
 282
 283     // return the descriptor for one of the pipe ends
 284     HANDLE operator[](Direction which) const { return m_handles[which]; }
 285
 286     // detach a descriptor, meaning that the pipe dtor won't close it, and
 287     // return it
 288     HANDLE Detach(Direction which)
 289     {
 290         HANDLE handle = m_handles[which];
 291         m_handles[which] = INVALID_HANDLE_VALUE;
 292
 293         return handle;
 294     }
 295
 296     // close the pipe descriptors
 297     void Close()
 298     {
 299         for ( size_t n = 0; n < WXSIZEOF(m_handles); n++ )
 300         {
 301             if ( m_handles[n] != INVALID_HANDLE_VALUE )
 302             {
 303                 ::CloseHandle(m_handles[n]);
 304                 m_handles[n] = INVALID_HANDLE_VALUE;
 305             }
 306         }
 307     }
 308
 309     // dtor closes the pipe descriptors
 310     ~wxPipe() { Close(); }
 311
 312 private:
 313     HANDLE m_handles[2];
 314 };
 315
 316 #endif // wxUSE_STREAMS
 317
 318 // ============================================================================
 319 // implementation
 320 // ============================================================================
 321
 322 // ----------------------------------------------------------------------------
 323 // process termination detecting support
 324 // ----------------------------------------------------------------------------
 325
 326 // thread function for the thread monitoring the process termination
 327 static DWORD __stdcall wxExecuteThread(void *arg)
 328 {
 329     wxExecuteData * const data = (wxExecuteData *)arg;
 330
 331     // create the shutdown event if we're the first thread starting to wait
 332     if ( !gs_heventShutdown )
 333     {
 334         // create a manual initially non-signalled event object
 335         gs_heventShutdown = ::CreateEvent(NULL, TRUE, FALSE, NULL);
 336         if ( !gs_heventShutdown )
 337         {
 338             wxLogDebug(_T("CreateEvent() in wxExecuteThread failed"));
 339         }
 340     }
 341
 342     HANDLE handles[2] = { data->hProcess, gs_heventShutdown };
 343     switch ( ::WaitForMultipleObjects(2, handles, FALSE, INFINITE) )
 344     {
 345         case WAIT_OBJECT_0:
 346             // process terminated, get its exit code
 347             if ( !::GetExitCodeProcess(data->hProcess, &data->dwExitCode) )
 348             {
 349                 wxLogLastError(wxT("GetExitCodeProcess"));
 350             }
 351
 352             wxASSERT_MSG( data->dwExitCode != STILL_ACTIVE,
 353                           wxT("process should have terminated") );
 354
 355             // send a message indicating process termination to the window
 356             ::SendMessage(data->hWnd, wxWM_PROC_TERMINATED, 0, (LPARAM)data);
 357             break;
 358
 359         case WAIT_OBJECT_0 + 1:
 360             // we're shutting down but the process is still running -- leave it
 361             // run but clean up the associated data
 362             if ( !data->state )
 363             {
 364                 delete data;
 365             }
 366             //else: exiting while synchronously executing process is still
 367             //      running? this shouldn't happen...
 368             break;
 369
 370         default:
 371             wxLogDebug(_T("Waiting for the process termination failed!"));
 372     }
 373
 374     return 0;
 375 }
 376
 377 // window procedure of a hidden window which is created just to receive
 378 // the notification message when a process exits
 379 LRESULT APIENTRY _EXPORT wxExecuteWindowCbk(HWND hWnd, UINT message,
 380                                             WPARAM wParam, LPARAM lParam)
 381 {
 382     if ( message == wxWM_PROC_TERMINATED )
 383     {
 384         DestroyWindow(hWnd);    // we don't need it any more
 385
 386         wxExecuteData * const data = (wxExecuteData *)lParam;
 387         if ( data->handler )
 388         {
 389             data->handler->OnTerminate((int)data->dwProcessId,
 390                                        (int)data->dwExitCode);
 391         }
 392
 393         if ( data->state )
 394         {
 395             // we're executing synchronously, tell the waiting thread
 396             // that the process finished
 397             data->state = false;
 398         }
 399         else
 400         {
 401             // asynchronous execution - we should do the clean up
 402             delete data;
 403         }
 404
 405         return 0;
 406     }
 407     else
 408     {
 409         return ::DefWindowProc(hWnd, message, wParam, lParam);
 410     }
 411 }
 412
 413 // ============================================================================
 414 // implementation of IO redirection support classes
 415 // ============================================================================
 416
 417 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 418
 419 // ----------------------------------------------------------------------------
 420 // wxPipeInputStreams
 421 // ----------------------------------------------------------------------------
 422
 423 wxPipeInputStream::wxPipeInputStream(HANDLE hInput)
 424 {
 425     m_hInput = hInput;
 426 }
 427
 428 wxPipeInputStream::~wxPipeInputStream()
 429 {
 430     if ( m_hInput != INVALID_HANDLE_VALUE )
 431         ::CloseHandle(m_hInput);
 432 }
 433
 434 bool wxPipeInputStream::CanRead() const
 435 {
 436     // we can read if there's something in the put back buffer
 437     // even pipe is closed
 438     if ( m_wbacksize > m_wbackcur )
 439         return true;
 440
 441     wxPipeInputStream * const self = wxConstCast(this, wxPipeInputStream);
 442
 443     if ( !IsOpened() )
 444     {
 445         // set back to mark Eof as it may have been unset by Ungetch()
 446         self->m_lasterror = wxSTREAM_EOF;
 447         return false;
 448     }
 449
 450     DWORD nAvailable;
 451
 452     // function name is misleading, it works with anon pipes as well
 453     DWORD rc = ::PeekNamedPipe
 454                     (
 455                       m_hInput,     // handle
 456                       NULL, 0,      // ptr to buffer and its size
 457                       NULL,         // [out] bytes read
 458                       &nAvailable,  // [out] bytes available
 459                       NULL          // [out] bytes left
 460                     );
 461
 462     if ( !rc )
 463     {
 464         if ( ::GetLastError() != ERROR_BROKEN_PIPE )
 465         {
 466             // unexpected error
 467             wxLogLastError(_T("PeekNamedPipe"));
 468         }
 469
 470         // don't try to continue reading from a pipe if an error occurred or if
 471         // it had been closed
 472         ::CloseHandle(m_hInput);
 473
 474         self->m_hInput = INVALID_HANDLE_VALUE;
 475         self->m_lasterror = wxSTREAM_EOF;
 476
 477         nAvailable = 0;
 478     }
 479
 480     return nAvailable != 0;
 481 }
 482
 483 size_t wxPipeInputStream::OnSysRead(void *buffer, size_t len)
 484 {
 485     if ( !IsOpened() )
 486     {
 487         m_lasterror = wxSTREAM_EOF;
 488
 489         return 0;
 490     }
 491
 492     DWORD bytesRead;
 493     if ( !::ReadFile(m_hInput, buffer, len, &bytesRead, NULL) )
 494     {
 495         m_lasterror = ::GetLastError() == ERROR_BROKEN_PIPE
 496                         ? wxSTREAM_EOF
 497                         : wxSTREAM_READ_ERROR;
 498     }
 499
 500     // bytesRead is set to 0, as desired, if an error occurred
 501     return bytesRead;
 502 }
 503
 504 // ----------------------------------------------------------------------------
 505 // wxPipeOutputStream
 506 // ----------------------------------------------------------------------------
 507
 508 wxPipeOutputStream::wxPipeOutputStream(HANDLE hOutput)
 509 {
 510     m_hOutput = hOutput;
 511
 512     // unblock the pipe to prevent deadlocks when we're writing to the pipe
 513     // from which the child process can't read because it is writing in its own
 514     // end of it
 515     DWORD mode = PIPE_READMODE_BYTE | PIPE_NOWAIT;
 516     if ( !::SetNamedPipeHandleState
 517             (
 518                 m_hOutput,
 519                 &mode,
 520                 NULL,       // collection count (we don't set it)
 521                 NULL        // timeout (we don't set it neither)
 522             ) )
 523     {
 524         wxLogLastError(_T("SetNamedPipeHandleState(PIPE_NOWAIT)"));
 525     }
 526 }
 527
 528 bool wxPipeOutputStream::Close()
 529 {
 530    return ::CloseHandle(m_hOutput) != 0;
 531 }
 532
 533
 534 size_t wxPipeOutputStream::OnSysWrite(const void *buffer, size_t len)
 535 {
 536     m_lasterror = wxSTREAM_NO_ERROR;
 537
 538     DWORD totalWritten = 0;
 539     while ( len > 0 )
 540     {
 541         DWORD chunkWritten;
 542         if ( !::WriteFile(m_hOutput, buffer, len, &chunkWritten, NULL) )
 543         {
 544             m_lasterror = ::GetLastError() == ERROR_BROKEN_PIPE
 545                                 ? wxSTREAM_EOF
 546                                 : wxSTREAM_WRITE_ERROR;
 547             break;
 548         }
 549
 550         if ( !chunkWritten )
 551             break;
 552
 553         buffer = (char *)buffer + chunkWritten;
 554         totalWritten += chunkWritten;
 555         len -= chunkWritten;
 556     }
 557
 558     return totalWritten;
 559 }
 560
 561 #endif // wxUSE_STREAMS
 562
 563 // ============================================================================
 564 // wxExecute functions family
 565 // ============================================================================
 566
 567 #if wxUSE_IPC
 568
 569 // connect to the given server via DDE and ask it to execute the command
 570 bool
 571 wxExecuteDDE(const wxString& ddeServer,
 572              const wxString& ddeTopic,
 573              const wxString& ddeCommand)
 574 {
 575     bool ok wxDUMMY_INITIALIZE(false);
 576
 577     wxDDEClient client;
 578     wxConnectionBase *
 579         conn = client.MakeConnection(wxEmptyString, ddeServer, ddeTopic);
 580     if ( !conn )
 581     {
 582         ok = false;
 583     }
 584     else // connected to DDE server
 585     {
 586         // the added complication here is that although most programs use
 587         // XTYP_EXECUTE for their DDE API, some important ones -- like Word
 588         // and other MS stuff - use XTYP_REQUEST!
 589         //
 590         // moreover, anotheri mportant program (IE) understands both but
 591         // returns an error from Execute() so we must try Request() first
 592         // to avoid doing it twice
 593         {
 594             // we're prepared for this one to fail, so don't show errors
 595             wxLogNull noErrors;
 596
 597             ok = conn->Request(ddeCommand) != NULL;
 598         }
 599
 600         if ( !ok )
 601         {
 602             // now try execute -- but show the errors
 603             ok = conn->Execute(ddeCommand);
 604         }
 605     }
 606
 607     return ok;
 608 }
 609
 610 #endif // wxUSE_IPC
 611
 612 long wxExecute(const wxString& cmd, int flags, wxProcess *handler)
 613 {
 614     wxCHECK_MSG( !cmd.empty(), 0, wxT("empty command in wxExecute") );
 615
 616 #if wxUSE_THREADS
 617     // for many reasons, the code below breaks down if it's called from another
 618     // thread -- this could be fixed, but as Unix versions don't support this
 619     // neither I don't want to waste time on this now
 620     wxASSERT_MSG( wxThread::IsMain(),
 621                     _T("wxExecute() can be called only from the main thread") );
 622 #endif // wxUSE_THREADS
 623
 624     wxString command;
 625
 626 #if wxUSE_IPC
 627     // DDE hack: this is really not pretty, but we need to allow this for
 628     // transparent handling of DDE servers in wxMimeTypesManager. Usually it
 629     // returns the command which should be run to view/open/... a file of the
 630     // given type. Sometimes, however, this command just launches the server
 631     // and an additional DDE request must be made to really open the file. To
 632     // keep all this well hidden from the application, we allow a special form
 633     // of command: WX_DDE#<command>#DDE_SERVER#DDE_TOPIC#DDE_COMMAND in which
 634     // case we execute just <command> and process the rest below
 635     wxString ddeServer, ddeTopic, ddeCommand;
 636     static const size_t lenDdePrefix = 7;   // strlen("WX_DDE:")
 637     if ( cmd.Left(lenDdePrefix) == _T("WX_DDE#") )
 638     {
 639         // speed up the concatenations below
 640         ddeServer.reserve(256);
 641         ddeTopic.reserve(256);
 642         ddeCommand.reserve(256);
 643
 644         const wxChar *p = cmd.c_str() + 7;
 645         while ( *p && *p != _T('#') )
 646         {
 647             command += *p++;
 648         }
 649
 650         if ( *p )
 651         {
 652             // skip '#'
 653             p++;
 654         }
 655         else
 656         {
 657             wxFAIL_MSG(_T("invalid WX_DDE command in wxExecute"));
 658         }
 659
 660         while ( *p && *p != _T('#') )
 661         {
 662             ddeServer += *p++;
 663         }
 664
 665         if ( *p )
 666         {
 667             // skip '#'
 668             p++;
 669         }
 670         else
 671         {
 672             wxFAIL_MSG(_T("invalid WX_DDE command in wxExecute"));
 673         }
 674
 675         while ( *p && *p != _T('#') )
 676         {
 677             ddeTopic += *p++;
 678         }
 679
 680         if ( *p )
 681         {
 682             // skip '#'
 683             p++;
 684         }
 685         else
 686         {
 687             wxFAIL_MSG(_T("invalid WX_DDE command in wxExecute"));
 688         }
 689
 690         while ( *p )
 691         {
 692             ddeCommand += *p++;
 693         }
 694
 695         // if we want to just launch the program and not wait for its
 696         // termination, try to execute DDE command right now, it can succeed if
 697         // the process is already running - but as it fails if it's not
 698         // running, suppress any errors it might generate
 699         if ( !(flags & wxEXEC_SYNC) )
 700         {
 701             wxLogNull noErrors;
 702             if ( wxExecuteDDE(ddeServer, ddeTopic, ddeCommand) )
 703             {
 704                 // a dummy PID - this is a hack, of course, but it's well worth
 705                 // it as we don't open a new server each time we're called
 706                 // which would be quite bad
 707                 return -1;
 708             }
 709         }
 710     }
 711     else
 712 #endif // wxUSE_IPC
 713     {
 714         // no DDE
 715         command = cmd;
 716     }
 717
 718     // the IO redirection is only supported with wxUSE_STREAMS
 719     BOOL redirect = FALSE;
 720
 721 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 722     wxPipe pipeIn, pipeOut, pipeErr;
 723
 724     // we'll save here the copy of pipeIn[Write]
 725     HANDLE hpipeStdinWrite = INVALID_HANDLE_VALUE;
 726
 727     // open the pipes to which child process IO will be redirected if needed
 728     if ( handler && handler->IsRedirected() )
 729     {
 730         // create pipes for redirecting stdin, stdout and stderr
 731         if ( !pipeIn.Create() || !pipeOut.Create() || !pipeErr.Create() )
 732         {
 733             wxLogSysError(_("Failed to redirect the child process IO"));
 734
 735             // indicate failure: we need to return different error code
 736             // depending on the sync flag
 737             return flags & wxEXEC_SYNC ? -1 : 0;
 738         }
 739
 740         redirect = TRUE;
 741     }
 742 #endif // wxUSE_STREAMS
 743
 744     // create the process
 745     STARTUPINFO si;
 746     wxZeroMemory(si);
 747     si.cb = sizeof(si);
 748
 749 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 750     if ( redirect )
 751     {
 752         si.dwFlags = STARTF_USESTDHANDLES;
 753
 754         si.hStdInput = pipeIn[wxPipe::Read];
 755         si.hStdOutput = pipeOut[wxPipe::Write];
 756         si.hStdError = pipeErr[wxPipe::Write];
 757
 758         // when the std IO is redirected, we don't show the (console) process
 759         // window by default, but this can be overridden by the caller by
 760         // specifying wxEXEC_NOHIDE flag
 761         if ( !(flags & wxEXEC_NOHIDE) )
 762         {
 763             si.dwFlags |= STARTF_USESHOWWINDOW;
 764             si.wShowWindow = SW_HIDE;
 765         }
 766
 767         // we must duplicate the handle to the write side of stdin pipe to make
 768         // it non inheritable: indeed, we must close the writing end of pipeIn
 769         // before launching the child process as otherwise this handle will be
 770         // inherited by the child which will never close it and so the pipe
 771         // will never be closed and the child will be left stuck in ReadFile()
 772         HANDLE pipeInWrite = pipeIn.Detach(wxPipe::Write);
 773         if ( !::DuplicateHandle
 774                 (
 775                     ::GetCurrentProcess(),
 776                     pipeInWrite,
 777                     ::GetCurrentProcess(),
 778                     &hpipeStdinWrite,
 779                     0,                      // desired access: unused here
 780                     FALSE,                  // not inherited
 781                     DUPLICATE_SAME_ACCESS   // same access as for src handle
 782                 ) )
 783         {
 784             wxLogLastError(_T("DuplicateHandle"));
 785         }
 786
 787         ::CloseHandle(pipeInWrite);
 788     }
 789 #endif // wxUSE_STREAMS
 790
 791     PROCESS_INFORMATION pi;
 792     DWORD dwFlags = CREATE_SUSPENDED;
 793
 794 #ifndef __WXWINCE__
 795     dwFlags |= CREATE_DEFAULT_ERROR_MODE ;
 796 #else
 797     // we are assuming commands without spaces for now
 798     wxString moduleName = command.BeforeFirst(wxT(' '));
 799     wxString arguments = command.AfterFirst(wxT(' '));
 800 #endif
 801
 802     bool ok = ::CreateProcess
 803                 (
 804                     // WinCE requires appname to be non null
 805                     // Win32 allows for null
 806 #ifdef __WXWINCE__
 807                  (wxChar *)
 808                  moduleName.wx_str(),// application name
 809                  (wxChar *)
 810                  arguments.wx_str(), // arguments
 811 #else
 812                  NULL,               // application name (use only cmd line)
 813                  (wxChar *)
 814                  command.wx_str(),   // full command line
 815 #endif
 816                  NULL,               // security attributes: defaults for both
 817                  NULL,               //   the process and its main thread
 818                  redirect,           // inherit handles if we use pipes
 819                  dwFlags,            // process creation flags
 820                  NULL,               // environment (use the same)
 821                  NULL,               // current directory (use the same)
 822                  &si,                // startup info (unused here)
 823                  &pi                 // process info
 824                 ) != 0;
 825
 826 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 827     // we can close the pipe ends used by child anyhow
 828     if ( redirect )
 829     {
 830         ::CloseHandle(pipeIn.Detach(wxPipe::Read));
 831         ::CloseHandle(pipeOut.Detach(wxPipe::Write));
 832         ::CloseHandle(pipeErr.Detach(wxPipe::Write));
 833     }
 834 #endif // wxUSE_STREAMS
 835
 836     if ( !ok )
 837     {
 838 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 839         // close the other handles too
 840         if ( redirect )
 841         {
 842             ::CloseHandle(pipeOut.Detach(wxPipe::Read));
 843             ::CloseHandle(pipeErr.Detach(wxPipe::Read));
 844         }
 845 #endif // wxUSE_STREAMS
 846
 847         wxLogSysError(_("Execution of command '%s' failed"), command.c_str());
 848
 849         return flags & wxEXEC_SYNC ? -1 : 0;
 850     }
 851
 852 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 853     // the input buffer bufOut is connected to stdout, this is why it is
 854     // called bufOut and not bufIn
 855     wxStreamTempInputBuffer bufOut,
 856                             bufErr;
 857
 858     if ( redirect )
 859     {
 860         // We can now initialize the wxStreams
 861         wxPipeInputStream *
 862             outStream = new wxPipeInputStream(pipeOut.Detach(wxPipe::Read));
 863         wxPipeInputStream *
 864             errStream = new wxPipeInputStream(pipeErr.Detach(wxPipe::Read));
 865         wxPipeOutputStream *
 866             inStream = new wxPipeOutputStream(hpipeStdinWrite);
 867
 868         handler->SetPipeStreams(outStream, inStream, errStream);
 869
 870         bufOut.Init(outStream);
 871         bufErr.Init(errStream);
 872     }
 873 #endif // wxUSE_STREAMS
 874
 875     // create a hidden window to receive notification about process
 876     // termination
 877     HWND hwnd = wxCreateHiddenWindow
 878                 (
 879                     &gs_classForHiddenWindow,
 880                     wxMSWEXEC_WNDCLASSNAME,
 881                     (WNDPROC)wxExecuteWindowCbk
 882                 );
 883
 884     wxASSERT_MSG( hwnd, wxT("can't create a hidden window for wxExecute") );
 885
 886     // Alloc data
 887     wxExecuteData *data = new wxExecuteData;
 888     data->hProcess    = pi.hProcess;
 889     data->dwProcessId = pi.dwProcessId;
 890     data->hWnd        = hwnd;
 891     data->state       = (flags & wxEXEC_SYNC) != 0;
 892     if ( flags & wxEXEC_SYNC )
 893     {
 894         // handler may be !NULL for capturing program output, but we don't use
 895         // it wxExecuteData struct in this case
 896         data->handler = NULL;
 897     }
 898     else
 899     {
 900         // may be NULL or not
 901         data->handler = handler;
 902     }
 903
 904     DWORD tid;
 905     HANDLE hThread = ::CreateThread(NULL,
 906                                     0,
 907                                     wxExecuteThread,
 908                                     (void *)data,
 909                                     0,
 910                                     &tid);
 911
 912     // resume process we created now - whether the thread creation succeeded or
 913     // not
 914     if ( ::ResumeThread(pi.hThread) == (DWORD)-1 )
 915     {
 916         // ignore it - what can we do?
 917         wxLogLastError(wxT("ResumeThread in wxExecute"));
 918     }
 919
 920     // close unneeded handle
 921     if ( !::CloseHandle(pi.hThread) )
 922         wxLogLastError(wxT("CloseHandle(hThread)"));
 923
 924     if ( !hThread )
 925     {
 926         wxLogLastError(wxT("CreateThread in wxExecute"));
 927
 928         DestroyWindow(hwnd);
 929         delete data;
 930
 931         // the process still started up successfully...
 932         return pi.dwProcessId;
 933     }
 934
 935     gs_asyncThreads.push_back(hThread);
 936
 937 #if wxUSE_IPC && !defined(__WXWINCE__)
 938     // second part of DDE hack: now establish the DDE conversation with the
 939     // just launched process
 940     if ( !ddeServer.empty() )
 941     {
 942         bool ok;
 943
 944         // give the process the time to init itself
 945         //
 946         // we use a very big timeout hoping that WaitForInputIdle() will return
 947         // much sooner, but not INFINITE just in case the process hangs
 948         // completely - like this we will regain control sooner or later
 949         switch ( ::WaitForInputIdle(pi.hProcess, 10000 /* 10 seconds */) )
 950         {
 951             default:
 952                 wxFAIL_MSG( _T("unexpected WaitForInputIdle() return code") );
 953                 // fall through
 954
 955             case -1:
 956                 wxLogLastError(_T("WaitForInputIdle() in wxExecute"));
 957
 958             case WAIT_TIMEOUT:
 959                 wxLogDebug(_T("Timeout too small in WaitForInputIdle"));
 960
 961                 ok = false;
 962                 break;
 963
 964             case 0:
 965                 // ok, process ready to accept DDE requests
 966                 ok = wxExecuteDDE(ddeServer, ddeTopic, ddeCommand);
 967         }
 968
 969         if ( !ok )
 970         {
 971             wxLogDebug(_T("Failed to send DDE request to the process \"%s\"."),
 972                        cmd.c_str());
 973         }
 974     }
 975 #endif // wxUSE_IPC
 976
 977     if ( !(flags & wxEXEC_SYNC) )
 978     {
 979         // clean up will be done when the process terminates
 980
 981         // return the pid
 982         return pi.dwProcessId;
 983     }
 984
 985     wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
 986     wxCHECK_MSG( traits, -1, _T("no wxAppTraits in wxExecute()?") );
 987
 988     void *cookie = NULL;
 989     if ( !(flags & wxEXEC_NODISABLE) )
 990     {
 991         // disable all app windows while waiting for the child process to finish
 992         cookie = traits->BeforeChildWaitLoop();
 993     }
 994
 995     // wait until the child process terminates
 996     while ( data->state )
 997     {
 998 #if wxUSE_STREAMS && !defined(__WXWINCE__)
 999         if ( !bufOut.Update() && !bufErr.Update() )
1000 #endif // wxUSE_STREAMS
1001         {
1002             // don't eat 100% of the CPU -- ugly but anything else requires
1003             // real async IO which we don't have for the moment
1004             ::Sleep(50);
1005         }
1006
1007         // we must process messages or we'd never get wxWM_PROC_TERMINATED
1008         traits->AlwaysYield();
1009     }
1010
1011     if ( !(flags & wxEXEC_NODISABLE) )
1012     {
1013         // reenable disabled windows back
1014         traits->AfterChildWaitLoop(cookie);
1015     }
1016
1017     DWORD dwExitCode = data->dwExitCode;
1018     delete data;
1019
1020     // return the exit code
1021     return dwExitCode;
1022 }
1023
1024 template <typename CharType>
1025 long wxExecuteImpl(CharType **argv, int flags, wxProcess *handler)
1026 {
1027     wxString command;
1028     command.reserve(1024);
1029
1030     wxString arg;
1031     for ( ;; )
1032     {
1033         arg = *argv++;
1034
1035         bool quote;
1036         if ( arg.empty() )
1037         {
1038             // we need to quote empty arguments, otherwise they'd just
1039             // disappear
1040             quote = true;
1041         }
1042         else // non-empty
1043         {
1044             // escape any quotes present in the string to avoid interfering
1045             // with the command line parsing in the child process
1046             arg.Replace("\"", "\\\"", true /* replace all */);
1047
1048             // and quote any arguments containing the spaces to prevent them from
1049             // being broken down
1050             quote = arg.find_first_of(" \t") != wxString::npos;
1051         }
1052
1053         if ( quote )
1054             command += '\"' + arg + '\"';
1055         else
1056             command += arg;
1057
1058         if ( !*argv )
1059             break;
1060
1061         command += ' ';
1062     }
1063
1064     return wxExecute(command, flags, handler);
1065 }
1066
1067 long wxExecute(char **argv, int flags, wxProcess *handler)
1068 {
1069     return wxExecuteImpl(argv, flags, handler);
1070 }
1071
1072 #if wxUSE_UNICODE
1073
1074 long wxExecute(wchar_t **argv, int flags, wxProcess *handler)
1075 {
1076     return wxExecuteImpl(argv, flags, handler);
1077 }
1078
1079 #endif // wxUSE_UNICODE