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