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