samples/exec/exec.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        exec.cpp
   3 // Purpose:     exec sample demonstrates wxExecute and related functions
   4 // Author:      Vadim Zeitlin
   5 // Modified by:
   6 // Created:     15.01.00
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Vadim Zeitlin
   9 // Licence:     wxWindows licence
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 // For compilers that support precompilation, includes "wx/wx.h".
  21 #include "wx/wxprec.h"
  22
  23 #ifdef __BORLANDC__
  24     #pragma hdrstop
  25 #endif
  26
  27 // for all others, include the necessary headers (this file is usually all you
  28 // need because it includes almost all "standard" wxWidgets headers
  29 #ifndef WX_PRECOMP
  30     #include "wx/app.h"
  31     #include "wx/log.h"
  32     #include "wx/frame.h"
  33     #include "wx/panel.h"
  34
  35     #include "wx/timer.h"
  36
  37     #include "wx/utils.h"
  38     #include "wx/menu.h"
  39
  40     #include "wx/msgdlg.h"
  41     #include "wx/textdlg.h"
  42     #include "wx/filedlg.h"
  43     #include "wx/choicdlg.h"
  44
  45     #include "wx/button.h"
  46     #include "wx/textctrl.h"
  47     #include "wx/listbox.h"
  48
  49     #include "wx/sizer.h"
  50 #endif
  51
  52 #include "wx/txtstrm.h"
  53 #include "wx/numdlg.h"
  54 #include "wx/textdlg.h"
  55 #include "wx/ffile.h"
  56
  57 #include "wx/process.h"
  58
  59 #include "wx/mimetype.h"
  60
  61 #ifdef __WINDOWS__
  62     #include "wx/dde.h"
  63 #endif // __WINDOWS__
  64
  65 // ----------------------------------------------------------------------------
  66 // the usual application and main frame classes
  67 // ----------------------------------------------------------------------------
  68
  69 // Define a new application type, each program should derive a class from wxApp
  70 class MyApp : public wxApp
  71 {
  72 public:
  73     // override base class virtuals
  74     // ----------------------------
  75
  76     // this one is called on application startup and is a good place for the app
  77     // initialization (doing it here and not in the ctor allows to have an error
  78     // return: if OnInit() returns false, the application terminates)
  79     virtual bool OnInit();
  80 };
  81
  82 // Define an array of process pointers used by MyFrame
  83 class MyPipedProcess;
  84 WX_DEFINE_ARRAY_PTR(MyPipedProcess *, MyPipedProcessesArray);
  85
  86 class MyProcess;
  87 WX_DEFINE_ARRAY_PTR(MyProcess *, MyProcessesArray);
  88
  89 // Define a new frame type: this is going to be our main frame
  90 class MyFrame : public wxFrame
  91 {
  92 public:
  93     // ctor and dtor
  94     MyFrame(const wxString& title, const wxPoint& pos, const wxSize& size);
  95     virtual ~MyFrame();
  96
  97     // event handlers (these functions should _not_ be virtual)
  98     void OnQuit(wxCommandEvent& event);
  99
 100     void OnKill(wxCommandEvent& event);
 101
 102     void OnClear(wxCommandEvent& event);
 103
 104     void OnBeginBusyCursor(wxCommandEvent& event);
 105     void OnEndBusyCursor(wxCommandEvent& event);
 106
 107     void OnSyncExec(wxCommandEvent& event);
 108     void OnSyncNoEventsExec(wxCommandEvent& event);
 109     void OnAsyncExec(wxCommandEvent& event);
 110     void OnShell(wxCommandEvent& event);
 111     void OnExecWithRedirect(wxCommandEvent& event);
 112     void OnExecWithPipe(wxCommandEvent& event);
 113
 114     void OnPOpen(wxCommandEvent& event);
 115
 116     void OnFileExec(wxCommandEvent& event);
 117     void OnOpenURL(wxCommandEvent& event);
 118
 119     void OnAbout(wxCommandEvent& event);
 120
 121     // polling output of async processes
 122     void OnTimer(wxTimerEvent& event);
 123     void OnIdle(wxIdleEvent& event);
 124
 125     // for MyPipedProcess
 126     void OnProcessTerminated(MyPipedProcess *process);
 127     wxListBox *GetLogListBox() const { return m_lbox; }
 128
 129     // for MyProcess
 130     void OnAsyncTermination(MyProcess *process);
 131
 132 private:
 133     void ShowOutput(const wxString& cmd,
 134                     const wxArrayString& output,
 135                     const wxString& title);
 136
 137     void DoAsyncExec(const wxString& cmd);
 138
 139     void AddAsyncProcess(MyProcess *process) { m_allAsync.push_back(process); }
 140
 141     void AddPipedProcess(MyPipedProcess *process);
 142     void RemovePipedProcess(MyPipedProcess *process);
 143
 144
 145     // the PID of the last process we launched asynchronously
 146     long m_pidLast;
 147
 148     // last command we executed
 149     wxString m_cmdLast;
 150
 151 #ifdef __WINDOWS__
 152     void OnDDEExec(wxCommandEvent& event);
 153     void OnDDERequest(wxCommandEvent& event);
 154
 155     bool GetDDEServer();
 156
 157     // last params of a DDE transaction
 158     wxString m_server,
 159              m_topic,
 160              m_cmdDde;
 161 #endif // __WINDOWS__
 162
 163     wxListBox *m_lbox;
 164
 165     // array of running processes with redirected IO
 166     MyPipedProcessesArray m_running;
 167
 168     // array of all asynchrously running processes
 169     MyProcessesArray m_allAsync;
 170
 171     // the idle event wake up timer
 172     wxTimer m_timerIdleWakeUp;
 173
 174     // any class wishing to process wxWidgets events must use this macro
 175     DECLARE_EVENT_TABLE()
 176 };
 177
 178 // ----------------------------------------------------------------------------
 179 // MyPipeFrame: allows the user to communicate with the child process
 180 // ----------------------------------------------------------------------------
 181
 182 class MyPipeFrame : public wxFrame
 183 {
 184 public:
 185     MyPipeFrame(wxFrame *parent,
 186                 const wxString& cmd,
 187                 wxProcess *process);
 188
 189 protected:
 190     void OnTextEnter(wxCommandEvent& WXUNUSED(event)) { DoSend(); }
 191     void OnBtnSend(wxCommandEvent& WXUNUSED(event)) { DoSend(); }
 192     void OnBtnSendFile(wxCommandEvent& WXUNUSED(event));
 193     void OnBtnGet(wxCommandEvent& WXUNUSED(event)) { DoGet(); }
 194     void OnBtnClose(wxCommandEvent& WXUNUSED(event)) { DoClose(); }
 195
 196     void OnClose(wxCloseEvent& event);
 197
 198     void OnProcessTerm(wxProcessEvent& event);
 199
 200     void DoSend()
 201     {
 202         wxString s(m_textOut->GetValue());
 203         s += _T('\n');
 204         m_out.Write(s.c_str(), s.length());
 205         m_textOut->Clear();
 206
 207         DoGet();
 208     }
 209
 210     void DoGet();
 211     void DoClose();
 212
 213 private:
 214     void DoGetFromStream(wxTextCtrl *text, wxInputStream& in);
 215     void DisableInput();
 216     void DisableOutput();
 217
 218
 219     wxProcess *m_process;
 220
 221     wxOutputStream &m_out;
 222     wxInputStream &m_in,
 223                   &m_err;
 224
 225     wxTextCtrl *m_textOut,
 226                *m_textIn,
 227                *m_textErr;
 228
 229     DECLARE_EVENT_TABLE()
 230 };
 231
 232 // ----------------------------------------------------------------------------
 233 // wxProcess-derived classes
 234 // ----------------------------------------------------------------------------
 235
 236 // This is the handler for process termination events
 237 class MyProcess : public wxProcess
 238 {
 239 public:
 240     MyProcess(MyFrame *parent, const wxString& cmd)
 241         : wxProcess(parent), m_cmd(cmd)
 242     {
 243         m_parent = parent;
 244     }
 245
 246     // instead of overriding this virtual function we might as well process the
 247     // event from it in the frame class - this might be more convenient in some
 248     // cases
 249     virtual void OnTerminate(int pid, int status);
 250
 251 protected:
 252     MyFrame *m_parent;
 253     wxString m_cmd;
 254 };
 255
 256 // A specialization of MyProcess for redirecting the output
 257 class MyPipedProcess : public MyProcess
 258 {
 259 public:
 260     MyPipedProcess(MyFrame *parent, const wxString& cmd)
 261         : MyProcess(parent, cmd)
 262         {
 263             Redirect();
 264         }
 265
 266     virtual void OnTerminate(int pid, int status);
 267
 268     virtual bool HasInput();
 269 };
 270
 271 // A version of MyPipedProcess which also sends input to the stdin of the
 272 // child process
 273 class MyPipedProcess2 : public MyPipedProcess
 274 {
 275 public:
 276     MyPipedProcess2(MyFrame *parent, const wxString& cmd, const wxString& input)
 277         : MyPipedProcess(parent, cmd), m_input(input)
 278         {
 279         }
 280
 281     virtual bool HasInput();
 282
 283 private:
 284     wxString m_input;
 285 };
 286
 287 // ----------------------------------------------------------------------------
 288 // constants
 289 // ----------------------------------------------------------------------------
 290
 291 // IDs for the controls and the menu commands
 292 enum
 293 {
 294     // menu items
 295     Exec_Quit = 100,
 296     Exec_Kill,
 297     Exec_ClearLog,
 298     Exec_BeginBusyCursor,
 299     Exec_EndBusyCursor,
 300     Exec_SyncExec = 200,
 301     Exec_SyncNoEventsExec,
 302     Exec_AsyncExec,
 303     Exec_Shell,
 304     Exec_POpen,
 305     Exec_OpenFile,
 306     Exec_OpenURL,
 307     Exec_DDEExec,
 308     Exec_DDERequest,
 309     Exec_Redirect,
 310     Exec_Pipe,
 311     Exec_About = 300,
 312
 313     // control ids
 314     Exec_Btn_Send = 1000,
 315     Exec_Btn_SendFile,
 316     Exec_Btn_Get,
 317     Exec_Btn_Close
 318 };
 319
 320 static const wxChar *DIALOG_TITLE = _T("Exec sample");
 321
 322 // ----------------------------------------------------------------------------
 323 // event tables and other macros for wxWidgets
 324 // ----------------------------------------------------------------------------
 325
 326 // the event tables connect the wxWidgets events with the functions (event
 327 // handlers) which process them. It can be also done at run-time, but for the
 328 // simple menu events like this the static method is much simpler.
 329 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
 330     EVT_MENU(Exec_Quit,  MyFrame::OnQuit)
 331     EVT_MENU(Exec_Kill,  MyFrame::OnKill)
 332     EVT_MENU(Exec_ClearLog,  MyFrame::OnClear)
 333     EVT_MENU(Exec_BeginBusyCursor,  MyFrame::OnBeginBusyCursor)
 334     EVT_MENU(Exec_EndBusyCursor,  MyFrame::OnEndBusyCursor)
 335
 336     EVT_MENU(Exec_SyncExec, MyFrame::OnSyncExec)
 337     EVT_MENU(Exec_SyncNoEventsExec, MyFrame::OnSyncNoEventsExec)
 338     EVT_MENU(Exec_AsyncExec, MyFrame::OnAsyncExec)
 339     EVT_MENU(Exec_Shell, MyFrame::OnShell)
 340     EVT_MENU(Exec_Redirect, MyFrame::OnExecWithRedirect)
 341     EVT_MENU(Exec_Pipe, MyFrame::OnExecWithPipe)
 342
 343     EVT_MENU(Exec_POpen, MyFrame::OnPOpen)
 344
 345     EVT_MENU(Exec_OpenFile, MyFrame::OnFileExec)
 346     EVT_MENU(Exec_OpenURL, MyFrame::OnOpenURL)
 347
 348 #ifdef __WINDOWS__
 349     EVT_MENU(Exec_DDEExec, MyFrame::OnDDEExec)
 350     EVT_MENU(Exec_DDERequest, MyFrame::OnDDERequest)
 351 #endif // __WINDOWS__
 352
 353     EVT_MENU(Exec_About, MyFrame::OnAbout)
 354
 355     EVT_IDLE(MyFrame::OnIdle)
 356
 357     EVT_TIMER(wxID_ANY, MyFrame::OnTimer)
 358 END_EVENT_TABLE()
 359
 360 BEGIN_EVENT_TABLE(MyPipeFrame, wxFrame)
 361     EVT_BUTTON(Exec_Btn_Send, MyPipeFrame::OnBtnSend)
 362     EVT_BUTTON(Exec_Btn_SendFile, MyPipeFrame::OnBtnSendFile)
 363     EVT_BUTTON(Exec_Btn_Get, MyPipeFrame::OnBtnGet)
 364     EVT_BUTTON(Exec_Btn_Close, MyPipeFrame::OnBtnClose)
 365
 366     EVT_TEXT_ENTER(wxID_ANY, MyPipeFrame::OnTextEnter)
 367
 368     EVT_CLOSE(MyPipeFrame::OnClose)
 369
 370     EVT_END_PROCESS(wxID_ANY, MyPipeFrame::OnProcessTerm)
 371 END_EVENT_TABLE()
 372
 373 // Create a new application object: this macro will allow wxWidgets to create
 374 // the application object during program execution (it's better than using a
 375 // static object for many reasons) and also declares the accessor function
 376 // wxGetApp() which will return the reference of the right type (i.e. MyApp and
 377 // not wxApp)
 378 IMPLEMENT_APP(MyApp)
 379
 380 // ============================================================================
 381 // implementation
 382 // ============================================================================
 383
 384 // ----------------------------------------------------------------------------
 385 // the application class
 386 // ----------------------------------------------------------------------------
 387
 388 // `Main program' equivalent: the program execution "starts" here
 389 bool MyApp::OnInit()
 390 {
 391     if ( !wxApp::OnInit() )
 392         return false;
 393
 394     // Create the main application window
 395     MyFrame *frame = new MyFrame(_T("Exec wxWidgets sample"),
 396                                  wxDefaultPosition, wxSize(500, 140));
 397
 398     // Show it and tell the application that it's our main window
 399     frame->Show(true);
 400     SetTopWindow(frame);
 401
 402     // success: wxApp::OnRun() will be called which will enter the main message
 403     // loop and the application will run. If we returned false here, the
 404     // application would exit immediately.
 405     return true;
 406 }
 407
 408 // ----------------------------------------------------------------------------
 409 // main frame
 410 // ----------------------------------------------------------------------------
 411
 412 #ifdef __VISUALC__
 413 #pragma warning(disable: 4355) // this used in base member initializer list
 414 #endif
 415
 416 // frame constructor
 417 MyFrame::MyFrame(const wxString& title, const wxPoint& pos, const wxSize& size)
 418        : wxFrame((wxFrame *)NULL, wxID_ANY, title, pos, size),
 419          m_timerIdleWakeUp(this)
 420 {
 421     m_pidLast = 0;
 422
 423 #ifdef __WXMAC__
 424     // we need this in order to allow the about menu relocation, since ABOUT is
 425     // not the default id of the about menu
 426     wxApp::s_macAboutMenuItemId = Exec_About;
 427 #endif
 428
 429     // create a menu bar
 430     wxMenu *menuFile = new wxMenu(wxEmptyString, wxMENU_TEAROFF);
 431     menuFile->Append(Exec_Kill, _T("&Kill process...\tCtrl-K"),
 432                      _T("Kill a process by PID"));
 433     menuFile->AppendSeparator();
 434     menuFile->Append(Exec_ClearLog, _T("&Clear log\tCtrl-L"),
 435                      _T("Clear the log window"));
 436     menuFile->AppendSeparator();
 437     menuFile->Append(Exec_BeginBusyCursor, _T("Show &busy cursor\tCtrl-C"));
 438     menuFile->Append(Exec_EndBusyCursor, _T("Show &normal cursor\tShift-Ctrl-C"));
 439     menuFile->AppendSeparator();
 440     menuFile->Append(Exec_Quit, _T("E&xit\tAlt-X"), _T("Quit this program"));
 441
 442     wxMenu *execMenu = new wxMenu;
 443     execMenu->Append(Exec_SyncExec, _T("Sync &execution...\tCtrl-E"),
 444                      _T("Launch a program and return when it terminates"));
 445     execMenu->Append(Exec_SyncNoEventsExec, _T("Sync execution and &block...\tCtrl-B"),
 446                      _T("Launch a program and block until it terminates"));
 447     execMenu->Append(Exec_AsyncExec, _T("&Async execution...\tCtrl-A"),
 448                      _T("Launch a program and return immediately"));
 449     execMenu->Append(Exec_Shell, _T("Execute &shell command...\tCtrl-S"),
 450                      _T("Launch a shell and execute a command in it"));
 451     execMenu->AppendSeparator();
 452     execMenu->Append(Exec_Redirect, _T("Capture command &output...\tCtrl-O"),
 453                      _T("Launch a program and capture its output"));
 454     execMenu->Append(Exec_Pipe, _T("&Pipe through command..."),
 455                      _T("Pipe a string through a filter"));
 456     execMenu->Append(Exec_POpen, _T("&Open a pipe to a command...\tCtrl-P"),
 457                      _T("Open a pipe to and from another program"));
 458
 459     execMenu->AppendSeparator();
 460     execMenu->Append(Exec_OpenFile, _T("Open &file...\tCtrl-F"),
 461                      _T("Launch the command to open this kind of files"));
 462     execMenu->Append(Exec_OpenURL, _T("Open &URL...\tCtrl-U"),
 463                      _T("Launch the default browser with the given URL"));
 464 #ifdef __WINDOWS__
 465     execMenu->AppendSeparator();
 466     execMenu->Append(Exec_DDEExec, _T("Execute command via &DDE...\tCtrl-D"));
 467     execMenu->Append(Exec_DDERequest, _T("Send DDE &request...\tCtrl-R"));
 468 #endif
 469
 470     wxMenu *helpMenu = new wxMenu(wxEmptyString, wxMENU_TEAROFF);
 471     helpMenu->Append(Exec_About, _T("&About...\tF1"), _T("Show about dialog"));
 472
 473     // now append the freshly created menu to the menu bar...
 474     wxMenuBar *menuBar = new wxMenuBar();
 475     menuBar->Append(menuFile, _T("&File"));
 476     menuBar->Append(execMenu, _T("&Exec"));
 477     menuBar->Append(helpMenu, _T("&Help"));
 478
 479     // ... and attach this menu bar to the frame
 480     SetMenuBar(menuBar);
 481
 482     // create the listbox in which we will show misc messages as they come
 483     m_lbox = new wxListBox(this, wxID_ANY);
 484     wxFont font(12, wxFONTFAMILY_TELETYPE, wxFONTSTYLE_NORMAL,
 485                 wxFONTWEIGHT_NORMAL);
 486     if ( font.Ok() )
 487         m_lbox->SetFont(font);
 488
 489 #if wxUSE_STATUSBAR
 490     // create a status bar just for fun (by default with 1 pane only)
 491     CreateStatusBar();
 492     SetStatusText(_T("Welcome to wxWidgets exec sample!"));
 493 #endif // wxUSE_STATUSBAR
 494 }
 495
 496 MyFrame::~MyFrame()
 497 {
 498     // any processes left until now must be deleted manually: normally this is
 499     // done when the associated process terminates but it must be still running
 500     // if this didn't happen until now
 501     for ( size_t n = 0; n < m_allAsync.size(); n++ )
 502     {
 503         delete m_allAsync[n];
 504     }
 505 }
 506
 507 // ----------------------------------------------------------------------------
 508 // event handlers: file and help menu
 509 // ----------------------------------------------------------------------------
 510
 511 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
 512 {
 513     // true is to force the frame to close
 514     Close(true);
 515 }
 516
 517 void MyFrame::OnClear(wxCommandEvent& WXUNUSED(event))
 518 {
 519     m_lbox->Clear();
 520 }
 521
 522 void MyFrame::OnBeginBusyCursor(wxCommandEvent& WXUNUSED(event))
 523 {
 524     wxBeginBusyCursor();
 525 }
 526
 527 void MyFrame::OnEndBusyCursor(wxCommandEvent& WXUNUSED(event))
 528 {
 529     wxEndBusyCursor();
 530 }
 531
 532 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
 533 {
 534     wxMessageBox(_T("Exec wxWidgets Sample\n(c) 2000-2002 Vadim Zeitlin"),
 535                  _T("About Exec"), wxOK | wxICON_INFORMATION, this);
 536 }
 537
 538 void MyFrame::OnKill(wxCommandEvent& WXUNUSED(event))
 539 {
 540     long pid = wxGetNumberFromUser(_T("Please specify the process to kill"),
 541                                    _T("Enter PID:"),
 542                                    _T("Exec question"),
 543                                    m_pidLast,
 544                                    // we need the full unsigned int range
 545                                    -INT_MAX, INT_MAX,
 546                                    this);
 547     if ( pid == -1 )
 548     {
 549         // cancelled
 550         return;
 551     }
 552
 553     static const wxString signalNames[] =
 554     {
 555         _T("Just test (SIGNONE)"),
 556         _T("Hangup (SIGHUP)"),
 557         _T("Interrupt (SIGINT)"),
 558         _T("Quit (SIGQUIT)"),
 559         _T("Illegal instruction (SIGILL)"),
 560         _T("Trap (SIGTRAP)"),
 561         _T("Abort (SIGABRT)"),
 562         _T("Emulated trap (SIGEMT)"),
 563         _T("FP exception (SIGFPE)"),
 564         _T("Kill (SIGKILL)"),
 565         _T("Bus (SIGBUS)"),
 566         _T("Segment violation (SIGSEGV)"),
 567         _T("System (SIGSYS)"),
 568         _T("Broken pipe (SIGPIPE)"),
 569         _T("Alarm (SIGALRM)"),
 570         _T("Terminate (SIGTERM)"),
 571     };
 572
 573     int sig = wxGetSingleChoiceIndex(_T("How to kill the process?"),
 574                                      _T("Exec question"),
 575                                      WXSIZEOF(signalNames), signalNames,
 576                                      this);
 577     switch ( sig )
 578     {
 579         default:
 580             wxFAIL_MSG( _T("unexpected return value") );
 581             // fall through
 582
 583         case -1:
 584             // cancelled
 585             return;
 586
 587         case wxSIGNONE:
 588         case wxSIGHUP:
 589         case wxSIGINT:
 590         case wxSIGQUIT:
 591         case wxSIGILL:
 592         case wxSIGTRAP:
 593         case wxSIGABRT:
 594         case wxSIGEMT:
 595         case wxSIGFPE:
 596         case wxSIGKILL:
 597         case wxSIGBUS:
 598         case wxSIGSEGV:
 599         case wxSIGSYS:
 600         case wxSIGPIPE:
 601         case wxSIGALRM:
 602         case wxSIGTERM:
 603             break;
 604     }
 605
 606     if ( sig == 0 )
 607     {
 608         if ( wxProcess::Exists(pid) )
 609             wxLogStatus(_T("Process %ld is running."), pid);
 610         else
 611             wxLogStatus(_T("No process with pid = %ld."), pid);
 612     }
 613     else // not SIGNONE
 614     {
 615         wxKillError rc = wxProcess::Kill(pid, (wxSignal)sig);
 616         if ( rc == wxKILL_OK )
 617         {
 618             wxLogStatus(_T("Process %ld killed with signal %d."), pid, sig);
 619         }
 620         else
 621         {
 622             static const wxChar *errorText[] =
 623             {
 624                 _T(""), // no error
 625                 _T("signal not supported"),
 626                 _T("permission denied"),
 627                 _T("no such process"),
 628                 _T("unspecified error"),
 629             };
 630
 631             wxLogStatus(_T("Failed to kill process %ld with signal %d: %s"),
 632                         pid, sig, errorText[rc]);
 633         }
 634     }
 635 }
 636
 637 // ----------------------------------------------------------------------------
 638 // event handlers: exec menu
 639 // ----------------------------------------------------------------------------
 640
 641 void MyFrame::DoAsyncExec(const wxString& cmd)
 642 {
 643     MyProcess * const process = new MyProcess(this, cmd);
 644     m_pidLast = wxExecute(cmd, wxEXEC_ASYNC, process);
 645     if ( !m_pidLast )
 646     {
 647         wxLogError(_T("Execution of '%s' failed."), cmd.c_str());
 648
 649         delete process;
 650     }
 651     else
 652     {
 653         wxLogStatus(_T("Process %ld (%s) launched."), m_pidLast, cmd.c_str());
 654
 655         m_cmdLast = cmd;
 656
 657         // the parent frame keeps track of all async processes as it needs to
 658         // free them if we exit before the child process terminates
 659         AddAsyncProcess(process);
 660     }
 661 }
 662
 663 void MyFrame::OnSyncExec(wxCommandEvent& WXUNUSED(event))
 664 {
 665     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 666                                      DIALOG_TITLE,
 667                                      m_cmdLast);
 668
 669     if ( !cmd )
 670         return;
 671
 672     wxLogStatus( _T("'%s' is running please wait..."), cmd.c_str() );
 673
 674     int code = wxExecute(cmd, wxEXEC_SYNC);
 675
 676     wxLogStatus(_T("Process '%s' terminated with exit code %d."),
 677         cmd.c_str(), code);
 678
 679     m_cmdLast = cmd;
 680 }
 681
 682 void MyFrame::OnSyncNoEventsExec(wxCommandEvent& WXUNUSED(event))
 683 {
 684     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 685                                      DIALOG_TITLE,
 686                                      m_cmdLast);
 687
 688     if ( !cmd )
 689         return;
 690
 691     wxLogStatus( _T("'%s' is running please wait..."), cmd.c_str() );
 692
 693     int code = wxExecute(cmd, wxEXEC_BLOCK);
 694
 695     wxLogStatus(_T("Process '%s' terminated with exit code %d."),
 696         cmd.c_str(), code);
 697
 698     m_cmdLast = cmd;
 699 }
 700
 701 void MyFrame::OnAsyncExec(wxCommandEvent& WXUNUSED(event))
 702 {
 703     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 704                                      DIALOG_TITLE,
 705                                      m_cmdLast);
 706
 707     if ( !cmd )
 708         return;
 709
 710     DoAsyncExec(cmd);
 711 }
 712
 713 void MyFrame::OnShell(wxCommandEvent& WXUNUSED(event))
 714 {
 715     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 716                                      DIALOG_TITLE,
 717                                      m_cmdLast);
 718
 719     if ( !cmd )
 720         return;
 721
 722     int code = wxShell(cmd);
 723     wxLogStatus(_T("Shell command '%s' terminated with exit code %d."),
 724                 cmd.c_str(), code);
 725     m_cmdLast = cmd;
 726 }
 727
 728 void MyFrame::OnExecWithRedirect(wxCommandEvent& WXUNUSED(event))
 729 {
 730     if ( !m_cmdLast )
 731     {
 732 #ifdef __WXMSW__
 733         m_cmdLast = "type Makefile.in";
 734 #else
 735         m_cmdLast = "cat -n Makefile";
 736 #endif
 737     }
 738
 739     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 740                                      DIALOG_TITLE,
 741                                      m_cmdLast);
 742
 743     if ( !cmd )
 744         return;
 745
 746     bool sync;
 747     switch ( wxMessageBox(_T("Execute it synchronously?"),
 748                           _T("Exec question"),
 749                           wxYES_NO | wxCANCEL | wxICON_QUESTION, this) )
 750     {
 751         case wxYES:
 752             sync = true;
 753             break;
 754
 755         case wxNO:
 756             sync = false;
 757             break;
 758
 759         default:
 760             return;
 761     }
 762
 763     if ( sync )
 764     {
 765         wxLogStatus( _T("'%s' is running please wait..."), cmd.c_str() );
 766
 767         wxArrayString output, errors;
 768         int code = wxExecute(cmd, output, errors);
 769
 770         wxLogStatus(_T("Command '%s' terminated with exit code %d."),
 771                     cmd.c_str(), code);
 772
 773         if ( code != -1 )
 774         {
 775             ShowOutput(cmd, output, _T("Output"));
 776             ShowOutput(cmd, errors, _T("Errors"));
 777         }
 778     }
 779     else // async exec
 780     {
 781         MyPipedProcess *process = new MyPipedProcess(this, cmd);
 782         if ( !wxExecute(cmd, wxEXEC_ASYNC, process) )
 783         {
 784             wxLogError(_T("Execution of '%s' failed."), cmd.c_str());
 785
 786             delete process;
 787         }
 788         else
 789         {
 790             AddPipedProcess(process);
 791         }
 792     }
 793
 794     m_cmdLast = cmd;
 795 }
 796
 797 void MyFrame::OnExecWithPipe(wxCommandEvent& WXUNUSED(event))
 798 {
 799     if ( !m_cmdLast )
 800         m_cmdLast = _T("tr [a-z] [A-Z]");
 801
 802     wxString cmd = wxGetTextFromUser(_T("Enter the command: "),
 803                                      DIALOG_TITLE,
 804                                      m_cmdLast);
 805
 806     if ( !cmd )
 807         return;
 808
 809     wxString input = wxGetTextFromUser(_T("Enter the string to send to it: "),
 810                                        DIALOG_TITLE);
 811     if ( !input )
 812         return;
 813
 814     // always execute the filter asynchronously
 815     MyPipedProcess2 *process = new MyPipedProcess2(this, cmd, input);
 816     long pid = wxExecute(cmd, wxEXEC_ASYNC, process);
 817     if ( pid )
 818     {
 819         wxLogStatus(_T("Process %ld (%s) launched."), pid, cmd.c_str());
 820
 821         AddPipedProcess(process);
 822     }
 823     else
 824     {
 825         wxLogError(_T("Execution of '%s' failed."), cmd.c_str());
 826
 827         delete process;
 828     }
 829
 830     m_cmdLast = cmd;
 831 }
 832
 833 void MyFrame::OnPOpen(wxCommandEvent& WXUNUSED(event))
 834 {
 835     wxString cmd = wxGetTextFromUser(_T("Enter the command to launch: "),
 836                                      DIALOG_TITLE,
 837                                      m_cmdLast);
 838     if ( cmd.empty() )
 839         return;
 840
 841     wxProcess *process = wxProcess::Open(cmd);
 842     if ( !process )
 843     {
 844         wxLogError(_T("Failed to launch the command."));
 845         return;
 846     }
 847
 848     wxLogVerbose(_T("PID of the new process: %ld"), process->GetPid());
 849
 850     wxOutputStream *out = process->GetOutputStream();
 851     if ( !out )
 852     {
 853         wxLogError(_T("Failed to connect to child stdin"));
 854         return;
 855     }
 856
 857     wxInputStream *in = process->GetInputStream();
 858     if ( !in )
 859     {
 860         wxLogError(_T("Failed to connect to child stdout"));
 861         return;
 862     }
 863
 864     new MyPipeFrame(this, cmd, process);
 865 }
 866
 867 void MyFrame::OnFileExec(wxCommandEvent& WXUNUSED(event))
 868 {
 869     static wxString s_filename;
 870
 871     wxString filename;
 872
 873 #if wxUSE_FILEDLG
 874     filename = wxLoadFileSelector(_T("any file"), wxEmptyString, s_filename, this);
 875 #else // !wxUSE_FILEDLG
 876     filename = wxGetTextFromUser(_T("Enter the file name"), _T("exec sample"),
 877                                  s_filename, this);
 878 #endif // wxUSE_FILEDLG/!wxUSE_FILEDLG
 879
 880     if ( filename.empty() )
 881         return;
 882
 883     s_filename = filename;
 884
 885     wxString ext = filename.AfterFirst(_T('.'));
 886     wxFileType *ft = wxTheMimeTypesManager->GetFileTypeFromExtension(ext);
 887     if ( !ft )
 888     {
 889         wxLogError(_T("Impossible to determine the file type for extension '%s'"),
 890                    ext.c_str());
 891         return;
 892     }
 893
 894     wxString cmd;
 895     bool ok = ft->GetOpenCommand(&cmd,
 896                                  wxFileType::MessageParameters(filename));
 897     delete ft;
 898     if ( !ok )
 899     {
 900         wxLogError(_T("Impossible to find out how to open files of extension '%s'"),
 901                    ext.c_str());
 902         return;
 903     }
 904
 905     DoAsyncExec(cmd);
 906 }
 907
 908 void MyFrame::OnOpenURL(wxCommandEvent& WXUNUSED(event))
 909 {
 910     static wxString s_filename(_T("http://www.wxwidgets.org/"));
 911
 912     wxString filename = wxGetTextFromUser
 913                         (
 914                             _T("Enter the URL"),
 915                             _T("exec sample"),
 916                             s_filename,
 917                             this
 918                         );
 919
 920     if ( filename.empty() )
 921         return;
 922
 923     s_filename = filename;
 924
 925     if ( !wxLaunchDefaultBrowser(s_filename) )
 926         wxLogError(_T("Failed to open URL \"%s\""), s_filename.c_str());
 927 }
 928
 929 // ----------------------------------------------------------------------------
 930 // DDE stuff
 931 // ----------------------------------------------------------------------------
 932
 933 #ifdef __WINDOWS__
 934
 935 bool MyFrame::GetDDEServer()
 936 {
 937     wxString server = wxGetTextFromUser(_T("Server to connect to:"),
 938                                         DIALOG_TITLE, m_server);
 939     if ( !server )
 940         return false;
 941
 942     m_server = server;
 943
 944     wxString topic = wxGetTextFromUser(_T("DDE topic:"), DIALOG_TITLE, m_topic);
 945     if ( !topic )
 946         return false;
 947
 948     m_topic = topic;
 949
 950     wxString cmd = wxGetTextFromUser(_T("DDE command:"), DIALOG_TITLE, m_cmdDde);
 951     if ( !cmd )
 952         return false;
 953
 954     m_cmdDde = cmd;
 955
 956     return true;
 957 }
 958
 959 void MyFrame::OnDDEExec(wxCommandEvent& WXUNUSED(event))
 960 {
 961     if ( !GetDDEServer() )
 962         return;
 963
 964     wxDDEClient client;
 965     wxConnectionBase *conn = client.MakeConnection(wxEmptyString, m_server, m_topic);
 966     if ( !conn )
 967     {
 968         wxLogError(_T("Failed to connect to the DDE server '%s'."),
 969                    m_server.c_str());
 970     }
 971     else
 972     {
 973         if ( !conn->Execute(m_cmdDde) )
 974         {
 975             wxLogError(_T("Failed to execute command '%s' via DDE."),
 976                        m_cmdDde.c_str());
 977         }
 978         else
 979         {
 980             wxLogStatus(_T("Successfully executed DDE command"));
 981         }
 982     }
 983 }
 984
 985 void MyFrame::OnDDERequest(wxCommandEvent& WXUNUSED(event))
 986 {
 987     if ( !GetDDEServer() )
 988         return;
 989
 990     wxDDEClient client;
 991     wxConnectionBase *conn = client.MakeConnection(wxEmptyString, m_server, m_topic);
 992     if ( !conn )
 993     {
 994         wxLogError(_T("Failed to connect to the DDE server '%s'."),
 995                    m_server.c_str());
 996     }
 997     else
 998     {
 999         if ( !conn->Request(m_cmdDde) )
1000         {
1001             wxLogError(_T("Failed to  send request '%s' via DDE."),
1002                        m_cmdDde.c_str());
1003         }
1004         else
1005         {
1006             wxLogStatus(_T("Successfully sent DDE request."));
1007         }
1008     }
1009 }
1010
1011 #endif // __WINDOWS__
1012
1013 // ----------------------------------------------------------------------------
1014 // various helpers
1015 // ----------------------------------------------------------------------------
1016
1017 // input polling
1018 void MyFrame::OnIdle(wxIdleEvent& event)
1019 {
1020     size_t count = m_running.GetCount();
1021     for ( size_t n = 0; n < count; n++ )
1022     {
1023         if ( m_running[n]->HasInput() )
1024         {
1025             event.RequestMore();
1026         }
1027     }
1028 }
1029
1030 void MyFrame::OnTimer(wxTimerEvent& WXUNUSED(event))
1031 {
1032     wxWakeUpIdle();
1033 }
1034
1035 void MyFrame::OnProcessTerminated(MyPipedProcess *process)
1036 {
1037     RemovePipedProcess(process);
1038 }
1039
1040 void MyFrame::OnAsyncTermination(MyProcess *process)
1041 {
1042     m_allAsync.Remove(process);
1043
1044     delete process;
1045 }
1046
1047 void MyFrame::AddPipedProcess(MyPipedProcess *process)
1048 {
1049     if ( m_running.IsEmpty() )
1050     {
1051         // we want to start getting the timer events to ensure that a
1052         // steady stream of idle events comes in -- otherwise we
1053         // wouldn't be able to poll the child process input
1054         m_timerIdleWakeUp.Start(100);
1055     }
1056     //else: the timer is already running
1057
1058     m_running.Add(process);
1059     m_allAsync.Add(process);
1060 }
1061
1062 void MyFrame::RemovePipedProcess(MyPipedProcess *process)
1063 {
1064     m_running.Remove(process);
1065
1066     if ( m_running.IsEmpty() )
1067     {
1068         // we don't need to get idle events all the time any more
1069         m_timerIdleWakeUp.Stop();
1070     }
1071 }
1072
1073 void MyFrame::ShowOutput(const wxString& cmd,
1074                          const wxArrayString& output,
1075                          const wxString& title)
1076 {
1077     size_t count = output.GetCount();
1078     if ( !count )
1079         return;
1080
1081     m_lbox->Append(wxString::Format(_T("--- %s of '%s' ---"),
1082                                     title.c_str(), cmd.c_str()));
1083
1084     for ( size_t n = 0; n < count; n++ )
1085     {
1086         m_lbox->Append(output[n]);
1087     }
1088
1089     m_lbox->Append(wxString::Format(_T("--- End of %s ---"),
1090                                     title.Lower().c_str()));
1091 }
1092
1093 // ----------------------------------------------------------------------------
1094 // MyProcess
1095 // ----------------------------------------------------------------------------
1096
1097 void MyProcess::OnTerminate(int pid, int status)
1098 {
1099     wxLogStatus(m_parent, _T("Process %u ('%s') terminated with exit code %d."),
1100                 pid, m_cmd.c_str(), status);
1101
1102     m_parent->OnAsyncTermination(this);
1103 }
1104
1105 // ----------------------------------------------------------------------------
1106 // MyPipedProcess
1107 // ----------------------------------------------------------------------------
1108
1109 bool MyPipedProcess::HasInput()
1110 {
1111     bool hasInput = false;
1112
1113     if ( IsInputAvailable() )
1114     {
1115         wxTextInputStream tis(*GetInputStream());
1116
1117         // this assumes that the output is always line buffered
1118         wxString msg;
1119         msg << m_cmd << _T(" (stdout): ") << tis.ReadLine();
1120
1121         m_parent->GetLogListBox()->Append(msg);
1122
1123         hasInput = true;
1124     }
1125
1126     if ( IsErrorAvailable() )
1127     {
1128         wxTextInputStream tis(*GetErrorStream());
1129
1130         // this assumes that the output is always line buffered
1131         wxString msg;
1132         msg << m_cmd << _T(" (stderr): ") << tis.ReadLine();
1133
1134         m_parent->GetLogListBox()->Append(msg);
1135
1136         hasInput = true;
1137     }
1138
1139     return hasInput;
1140 }
1141
1142 void MyPipedProcess::OnTerminate(int pid, int status)
1143 {
1144     // show the rest of the output
1145     while ( HasInput() )
1146         ;
1147
1148     m_parent->OnProcessTerminated(this);
1149
1150     MyProcess::OnTerminate(pid, status);
1151 }
1152
1153 // ----------------------------------------------------------------------------
1154 // MyPipedProcess2
1155 // ----------------------------------------------------------------------------
1156
1157 bool MyPipedProcess2::HasInput()
1158 {
1159     if ( !m_input.empty() )
1160     {
1161         wxTextOutputStream os(*GetOutputStream());
1162         os.WriteString(m_input);
1163
1164         CloseOutput();
1165         m_input.clear();
1166
1167         // call us once again - may be we'll have output
1168         return true;
1169     }
1170
1171     return MyPipedProcess::HasInput();
1172 }
1173
1174 // ============================================================================
1175 // MyPipeFrame implementation
1176 // ============================================================================
1177
1178 MyPipeFrame::MyPipeFrame(wxFrame *parent,
1179                          const wxString& cmd,
1180                          wxProcess *process)
1181            : wxFrame(parent, wxID_ANY, cmd),
1182              m_process(process),
1183              // in a real program we'd check that the streams are !NULL here
1184              m_out(*process->GetOutputStream()),
1185              m_in(*process->GetInputStream()),
1186              m_err(*process->GetErrorStream())
1187 {
1188     m_process->SetNextHandler(this);
1189
1190     wxPanel *panel = new wxPanel(this, wxID_ANY);
1191
1192     m_textOut = new wxTextCtrl(panel, wxID_ANY, wxEmptyString,
1193                               wxDefaultPosition, wxDefaultSize,
1194                               wxTE_PROCESS_ENTER);
1195     m_textIn = new wxTextCtrl(panel, wxID_ANY, wxEmptyString,
1196                                wxDefaultPosition, wxDefaultSize,
1197                                wxTE_MULTILINE | wxTE_RICH);
1198     m_textIn->SetEditable(false);
1199     m_textErr = new wxTextCtrl(panel, wxID_ANY, wxEmptyString,
1200                                wxDefaultPosition, wxDefaultSize,
1201                                wxTE_MULTILINE | wxTE_RICH);
1202     m_textErr->SetEditable(false);
1203
1204     wxSizer *sizerTop = new wxBoxSizer(wxVERTICAL);
1205     sizerTop->Add(m_textOut, 0, wxGROW | wxALL, 5);
1206
1207     wxSizer *sizerBtns = new wxBoxSizer(wxHORIZONTAL);
1208     sizerBtns->
1209         Add(new wxButton(panel, Exec_Btn_Send, _T("&Send")), 0, wxALL, 5);
1210     sizerBtns->
1211         Add(new wxButton(panel, Exec_Btn_SendFile, _T("&File...")), 0, wxALL, 5);
1212     sizerBtns->
1213         Add(new wxButton(panel, Exec_Btn_Get, _T("&Get")), 0, wxALL, 5);
1214     sizerBtns->
1215         Add(new wxButton(panel, Exec_Btn_Close, _T("&Close")), 0, wxALL, 5);
1216
1217     sizerTop->Add(sizerBtns, 0, wxCENTRE | wxALL, 5);
1218     sizerTop->Add(m_textIn, 1, wxGROW | wxALL, 5);
1219     sizerTop->Add(m_textErr, 1, wxGROW | wxALL, 5);
1220
1221     panel->SetSizer(sizerTop);
1222     sizerTop->Fit(this);
1223
1224     Show();
1225 }
1226
1227 void MyPipeFrame::OnBtnSendFile(wxCommandEvent& WXUNUSED(event))
1228 {
1229 #if wxUSE_FILEDLG
1230     wxFileDialog filedlg(this, _T("Select file to send"));
1231     if ( filedlg.ShowModal() != wxID_OK )
1232         return;
1233
1234     wxFFile file(filedlg.GetFilename(), _T("r"));
1235     wxString data;
1236     if ( !file.IsOpened() || !file.ReadAll(&data) )
1237         return;
1238
1239     // can't write the entire string at once, this risk overflowing the pipe
1240     // and we would dead lock
1241     size_t len = data.length();
1242     const wxChar *pc = data.c_str();
1243     while ( len )
1244     {
1245         const size_t CHUNK_SIZE = 4096;
1246         m_out.Write(pc, len > CHUNK_SIZE ? CHUNK_SIZE : len);
1247
1248         // note that not all data could have been written as we don't block on
1249         // the write end of the pipe
1250         const size_t lenChunk = m_out.LastWrite();
1251
1252         pc += lenChunk;
1253         len -= lenChunk;
1254
1255         DoGet();
1256     }
1257 #endif // wxUSE_FILEDLG
1258 }
1259
1260 void MyPipeFrame::DoGet()
1261 {
1262     // we don't have any way to be notified when any input appears on the
1263     // stream so we have to poll it :-(
1264     DoGetFromStream(m_textIn, m_in);
1265     DoGetFromStream(m_textErr, m_err);
1266 }
1267
1268 void MyPipeFrame::DoGetFromStream(wxTextCtrl *text, wxInputStream& in)
1269 {
1270     while ( in.CanRead() )
1271     {
1272         wxChar buffer[4096];
1273         buffer[in.Read(buffer, WXSIZEOF(buffer) - 1).LastRead()] = _T('\0');
1274
1275         text->AppendText(buffer);
1276     }
1277 }
1278
1279 void MyPipeFrame::DoClose()
1280 {
1281     m_process->CloseOutput();
1282
1283     DisableInput();
1284 }
1285
1286 void MyPipeFrame::DisableInput()
1287 {
1288     m_textOut->SetEditable(false);
1289     FindWindow(Exec_Btn_Send)->Disable();
1290     FindWindow(Exec_Btn_SendFile)->Disable();
1291     FindWindow(Exec_Btn_Close)->Disable();
1292 }
1293
1294 void MyPipeFrame::DisableOutput()
1295 {
1296     FindWindow(Exec_Btn_Get)->Disable();
1297 }
1298
1299 void MyPipeFrame::OnClose(wxCloseEvent& event)
1300 {
1301     if ( m_process )
1302     {
1303         // we're not interested in getting the process termination notification
1304         // if we are closing it ourselves
1305         wxProcess *process = m_process;
1306         m_process = NULL;
1307         process->SetNextHandler(NULL);
1308
1309         process->CloseOutput();
1310     }
1311
1312     event.Skip();
1313 }
1314
1315 void MyPipeFrame::OnProcessTerm(wxProcessEvent& WXUNUSED(event))
1316 {
1317     DoGet();
1318
1319     delete m_process;
1320     m_process = NULL;
1321
1322     wxLogWarning(_T("The other process has terminated, closing"));
1323
1324     DisableInput();
1325     DisableOutput();
1326 }