]>
Commit | Line | Data |
---|---|---|
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 += wxT('\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 = wxT("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(wxT("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, wxT("&Kill process...\tCtrl-K"), | |
449 | wxT("Kill a process by PID")); | |
450 | menuFile->AppendSeparator(); | |
451 | menuFile->Append(Exec_ClearLog, wxT("&Clear log\tCtrl-L"), | |
452 | wxT("Clear the log window")); | |
453 | menuFile->AppendSeparator(); | |
454 | menuFile->Append(Exec_BeginBusyCursor, wxT("Show &busy cursor\tCtrl-C")); | |
455 | menuFile->Append(Exec_EndBusyCursor, wxT("Show &normal cursor\tShift-Ctrl-C")); | |
456 | menuFile->AppendSeparator(); | |
457 | menuFile->Append(Exec_Quit, wxT("E&xit\tAlt-X"), wxT("Quit this program")); | |
458 | ||
459 | wxMenu *execMenu = new wxMenu; | |
460 | execMenu->Append(Exec_SyncExec, wxT("Sync &execution...\tCtrl-E"), | |
461 | wxT("Launch a program and return when it terminates")); | |
462 | execMenu->Append(Exec_SyncNoEventsExec, wxT("Sync execution and &block...\tCtrl-B"), | |
463 | wxT("Launch a program and block until it terminates")); | |
464 | execMenu->Append(Exec_AsyncExec, wxT("&Async execution...\tCtrl-A"), | |
465 | wxT("Launch a program and return immediately")); | |
466 | execMenu->Append(Exec_Shell, wxT("Execute &shell command...\tCtrl-S"), | |
467 | wxT("Launch a shell and execute a command in it")); | |
468 | execMenu->AppendSeparator(); | |
469 | execMenu->Append(Exec_Redirect, wxT("Capture command &output...\tCtrl-O"), | |
470 | wxT("Launch a program and capture its output")); | |
471 | execMenu->Append(Exec_Pipe, wxT("&Pipe through command..."), | |
472 | wxT("Pipe a string through a filter")); | |
473 | execMenu->Append(Exec_POpen, wxT("&Open a pipe to a command...\tCtrl-P"), | |
474 | wxT("Open a pipe to and from another program")); | |
475 | ||
476 | execMenu->AppendSeparator(); | |
477 | execMenu->Append(Exec_OpenFile, wxT("Open &file...\tCtrl-F"), | |
478 | wxT("Launch the command to open this kind of files")); | |
479 | execMenu->Append(Exec_LaunchFile, wxT("La&unch file...\tShift-Ctrl-F"), | |
480 | wxT("Launch the default application associated with the file")); | |
481 | execMenu->Append(Exec_OpenURL, wxT("Open &URL...\tCtrl-U"), | |
482 | wxT("Launch the default browser with the given URL")); | |
483 | #ifdef __WINDOWS__ | |
484 | execMenu->AppendSeparator(); | |
485 | execMenu->Append(Exec_DDEExec, wxT("Execute command via &DDE...\tCtrl-D")); | |
486 | execMenu->Append(Exec_DDERequest, wxT("Send DDE &request...\tCtrl-R")); | |
487 | #endif | |
488 | ||
489 | wxMenu *helpMenu = new wxMenu(wxEmptyString, wxMENU_TEAROFF); | |
490 | helpMenu->Append(Exec_About, wxT("&About...\tF1"), wxT("Show about dialog")); | |
491 | ||
492 | // now append the freshly created menu to the menu bar... | |
493 | wxMenuBar *menuBar = new wxMenuBar(); | |
494 | menuBar->Append(menuFile, wxT("&File")); | |
495 | menuBar->Append(execMenu, wxT("&Exec")); | |
496 | menuBar->Append(helpMenu, wxT("&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(wxT("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(wxT("Exec wxWidgets Sample\n(c) 2000-2002 Vadim Zeitlin"), | |
556 | wxT("About Exec"), wxOK | wxICON_INFORMATION, this); | |
557 | } | |
558 | ||
559 | void MyFrame::OnKill(wxCommandEvent& WXUNUSED(event)) | |
560 | { | |
561 | long pid = wxGetNumberFromUser(wxT("Please specify the process to kill"), | |
562 | wxT("Enter PID:"), | |
563 | wxT("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 | wxT("Just test (SIGNONE)"), | |
577 | wxT("Hangup (SIGHUP)"), | |
578 | wxT("Interrupt (SIGINT)"), | |
579 | wxT("Quit (SIGQUIT)"), | |
580 | wxT("Illegal instruction (SIGILL)"), | |
581 | wxT("Trap (SIGTRAP)"), | |
582 | wxT("Abort (SIGABRT)"), | |
583 | wxT("Emulated trap (SIGEMT)"), | |
584 | wxT("FP exception (SIGFPE)"), | |
585 | wxT("Kill (SIGKILL)"), | |
586 | wxT("Bus (SIGBUS)"), | |
587 | wxT("Segment violation (SIGSEGV)"), | |
588 | wxT("System (SIGSYS)"), | |
589 | wxT("Broken pipe (SIGPIPE)"), | |
590 | wxT("Alarm (SIGALRM)"), | |
591 | wxT("Terminate (SIGTERM)"), | |
592 | }; | |
593 | ||
594 | int sig = wxGetSingleChoiceIndex(wxT("How to kill the process?"), | |
595 | wxT("Exec question"), | |
596 | WXSIZEOF(signalNames), signalNames, | |
597 | this); | |
598 | switch ( sig ) | |
599 | { | |
600 | default: | |
601 | wxFAIL_MSG( wxT("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 | { | |
631 | wxLogStatus(wxT("Process %ld is running."), pid); | |
632 | } | |
633 | else | |
634 | { | |
635 | wxLogStatus(wxT("No process with pid = %ld."), pid); | |
636 | } | |
637 | } | |
638 | else // not SIGNONE | |
639 | { | |
640 | wxKillError rc = wxProcess::Kill(pid, (wxSignal)sig); | |
641 | if ( rc == wxKILL_OK ) | |
642 | { | |
643 | wxLogStatus(wxT("Process %ld killed with signal %d."), pid, sig); | |
644 | } | |
645 | else | |
646 | { | |
647 | static const wxChar *errorText[] = | |
648 | { | |
649 | wxT(""), // no error | |
650 | wxT("signal not supported"), | |
651 | wxT("permission denied"), | |
652 | wxT("no such process"), | |
653 | wxT("unspecified error"), | |
654 | }; | |
655 | ||
656 | wxLogStatus(wxT("Failed to kill process %ld with signal %d: %s"), | |
657 | pid, sig, errorText[rc]); | |
658 | } | |
659 | } | |
660 | } | |
661 | ||
662 | // ---------------------------------------------------------------------------- | |
663 | // event handlers: exec menu | |
664 | // ---------------------------------------------------------------------------- | |
665 | ||
666 | void MyFrame::DoAsyncExec(const wxString& cmd) | |
667 | { | |
668 | MyProcess * const process = new MyProcess(this, cmd); | |
669 | m_pidLast = wxExecute(cmd, wxEXEC_ASYNC, process); | |
670 | if ( !m_pidLast ) | |
671 | { | |
672 | wxLogError(wxT("Execution of '%s' failed."), cmd.c_str()); | |
673 | ||
674 | delete process; | |
675 | } | |
676 | else | |
677 | { | |
678 | wxLogStatus(wxT("Process %ld (%s) launched."), m_pidLast, cmd.c_str()); | |
679 | ||
680 | m_cmdLast = cmd; | |
681 | ||
682 | // the parent frame keeps track of all async processes as it needs to | |
683 | // free them if we exit before the child process terminates | |
684 | AddAsyncProcess(process); | |
685 | } | |
686 | } | |
687 | ||
688 | void MyFrame::OnSyncExec(wxCommandEvent& WXUNUSED(event)) | |
689 | { | |
690 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
691 | DIALOG_TITLE, | |
692 | m_cmdLast); | |
693 | ||
694 | if ( !cmd ) | |
695 | return; | |
696 | ||
697 | wxLogStatus( wxT("'%s' is running please wait..."), cmd.c_str() ); | |
698 | ||
699 | int code = wxExecute(cmd, wxEXEC_SYNC); | |
700 | ||
701 | wxLogStatus(wxT("Process '%s' terminated with exit code %d."), | |
702 | cmd.c_str(), code); | |
703 | ||
704 | m_cmdLast = cmd; | |
705 | } | |
706 | ||
707 | void MyFrame::OnSyncNoEventsExec(wxCommandEvent& WXUNUSED(event)) | |
708 | { | |
709 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
710 | DIALOG_TITLE, | |
711 | m_cmdLast); | |
712 | ||
713 | if ( !cmd ) | |
714 | return; | |
715 | ||
716 | wxLogStatus( wxT("'%s' is running please wait..."), cmd.c_str() ); | |
717 | ||
718 | int code = wxExecute(cmd, wxEXEC_BLOCK); | |
719 | ||
720 | wxLogStatus(wxT("Process '%s' terminated with exit code %d."), | |
721 | cmd.c_str(), code); | |
722 | ||
723 | m_cmdLast = cmd; | |
724 | } | |
725 | ||
726 | void MyFrame::OnAsyncExec(wxCommandEvent& WXUNUSED(event)) | |
727 | { | |
728 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
729 | DIALOG_TITLE, | |
730 | m_cmdLast); | |
731 | ||
732 | if ( !cmd ) | |
733 | return; | |
734 | ||
735 | DoAsyncExec(cmd); | |
736 | } | |
737 | ||
738 | void MyFrame::OnShell(wxCommandEvent& WXUNUSED(event)) | |
739 | { | |
740 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
741 | DIALOG_TITLE, | |
742 | m_cmdLast); | |
743 | ||
744 | if ( !cmd ) | |
745 | return; | |
746 | ||
747 | int code = wxShell(cmd); | |
748 | wxLogStatus(wxT("Shell command '%s' terminated with exit code %d."), | |
749 | cmd.c_str(), code); | |
750 | m_cmdLast = cmd; | |
751 | } | |
752 | ||
753 | void MyFrame::OnExecWithRedirect(wxCommandEvent& WXUNUSED(event)) | |
754 | { | |
755 | if ( !m_cmdLast ) | |
756 | { | |
757 | #ifdef __WXMSW__ | |
758 | m_cmdLast = "type Makefile.in"; | |
759 | #else | |
760 | m_cmdLast = "cat -n Makefile"; | |
761 | #endif | |
762 | } | |
763 | ||
764 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
765 | DIALOG_TITLE, | |
766 | m_cmdLast); | |
767 | ||
768 | if ( !cmd ) | |
769 | return; | |
770 | ||
771 | bool sync; | |
772 | switch ( wxMessageBox(wxT("Execute it synchronously?"), | |
773 | wxT("Exec question"), | |
774 | wxYES_NO | wxCANCEL | wxICON_QUESTION, this) ) | |
775 | { | |
776 | case wxYES: | |
777 | sync = true; | |
778 | break; | |
779 | ||
780 | case wxNO: | |
781 | sync = false; | |
782 | break; | |
783 | ||
784 | default: | |
785 | return; | |
786 | } | |
787 | ||
788 | if ( sync ) | |
789 | { | |
790 | wxLogStatus("\"%s\" is running please wait...", cmd); | |
791 | ||
792 | wxStopWatch sw; | |
793 | ||
794 | wxArrayString output, errors; | |
795 | int code = wxExecute(cmd, output, errors); | |
796 | ||
797 | wxLogStatus("Command \"%s\" terminated after %ldms; exit code %d.", | |
798 | cmd, sw.Time(), code); | |
799 | ||
800 | if ( code != -1 ) | |
801 | { | |
802 | ShowOutput(cmd, output, wxT("Output")); | |
803 | ShowOutput(cmd, errors, wxT("Errors")); | |
804 | } | |
805 | } | |
806 | else // async exec | |
807 | { | |
808 | MyPipedProcess *process = new MyPipedProcess(this, cmd); | |
809 | if ( !wxExecute(cmd, wxEXEC_ASYNC, process) ) | |
810 | { | |
811 | wxLogError(wxT("Execution of '%s' failed."), cmd.c_str()); | |
812 | ||
813 | delete process; | |
814 | } | |
815 | else | |
816 | { | |
817 | AddPipedProcess(process); | |
818 | } | |
819 | } | |
820 | ||
821 | m_cmdLast = cmd; | |
822 | } | |
823 | ||
824 | void MyFrame::OnExecWithPipe(wxCommandEvent& WXUNUSED(event)) | |
825 | { | |
826 | if ( !m_cmdLast ) | |
827 | m_cmdLast = wxT("tr [a-z] [A-Z]"); | |
828 | ||
829 | wxString cmd = wxGetTextFromUser(wxT("Enter the command: "), | |
830 | DIALOG_TITLE, | |
831 | m_cmdLast); | |
832 | ||
833 | if ( !cmd ) | |
834 | return; | |
835 | ||
836 | wxString input = wxGetTextFromUser(wxT("Enter the string to send to it: "), | |
837 | DIALOG_TITLE); | |
838 | if ( !input ) | |
839 | return; | |
840 | ||
841 | // always execute the filter asynchronously | |
842 | MyPipedProcess2 *process = new MyPipedProcess2(this, cmd, input); | |
843 | long pid = wxExecute(cmd, wxEXEC_ASYNC, process); | |
844 | if ( pid ) | |
845 | { | |
846 | wxLogStatus(wxT("Process %ld (%s) launched."), pid, cmd.c_str()); | |
847 | ||
848 | AddPipedProcess(process); | |
849 | } | |
850 | else | |
851 | { | |
852 | wxLogError(wxT("Execution of '%s' failed."), cmd.c_str()); | |
853 | ||
854 | delete process; | |
855 | } | |
856 | ||
857 | m_cmdLast = cmd; | |
858 | } | |
859 | ||
860 | void MyFrame::OnPOpen(wxCommandEvent& WXUNUSED(event)) | |
861 | { | |
862 | wxString cmd = wxGetTextFromUser(wxT("Enter the command to launch: "), | |
863 | DIALOG_TITLE, | |
864 | m_cmdLast); | |
865 | if ( cmd.empty() ) | |
866 | return; | |
867 | ||
868 | wxProcess *process = wxProcess::Open(cmd); | |
869 | if ( !process ) | |
870 | { | |
871 | wxLogError(wxT("Failed to launch the command.")); | |
872 | return; | |
873 | } | |
874 | ||
875 | wxLogVerbose(wxT("PID of the new process: %ld"), process->GetPid()); | |
876 | ||
877 | wxOutputStream *out = process->GetOutputStream(); | |
878 | if ( !out ) | |
879 | { | |
880 | wxLogError(wxT("Failed to connect to child stdin")); | |
881 | return; | |
882 | } | |
883 | ||
884 | wxInputStream *in = process->GetInputStream(); | |
885 | if ( !in ) | |
886 | { | |
887 | wxLogError(wxT("Failed to connect to child stdout")); | |
888 | return; | |
889 | } | |
890 | ||
891 | new MyPipeFrame(this, cmd, process); | |
892 | } | |
893 | ||
894 | static wxString gs_lastFile; | |
895 | ||
896 | static bool AskUserForFileName() | |
897 | { | |
898 | wxString filename; | |
899 | ||
900 | #if wxUSE_FILEDLG | |
901 | filename = wxLoadFileSelector(wxT("any"), wxEmptyString, gs_lastFile); | |
902 | #else // !wxUSE_FILEDLG | |
903 | filename = wxGetTextFromUser(wxT("Enter the file name"), wxT("exec sample"), | |
904 | gs_lastFile); | |
905 | #endif // wxUSE_FILEDLG/!wxUSE_FILEDLG | |
906 | ||
907 | if ( filename.empty() ) | |
908 | return false; | |
909 | ||
910 | gs_lastFile = filename; | |
911 | ||
912 | return true; | |
913 | } | |
914 | ||
915 | void MyFrame::OnFileExec(wxCommandEvent& WXUNUSED(event)) | |
916 | { | |
917 | if ( !AskUserForFileName() ) | |
918 | return; | |
919 | ||
920 | wxString ext = gs_lastFile.AfterLast(wxT('.')); | |
921 | wxFileType *ft = wxTheMimeTypesManager->GetFileTypeFromExtension(ext); | |
922 | if ( !ft ) | |
923 | { | |
924 | wxLogError(wxT("Impossible to determine the file type for extension '%s'"), | |
925 | ext.c_str()); | |
926 | return; | |
927 | } | |
928 | ||
929 | wxString cmd; | |
930 | bool ok = ft->GetOpenCommand(&cmd, | |
931 | wxFileType::MessageParameters(gs_lastFile)); | |
932 | delete ft; | |
933 | if ( !ok ) | |
934 | { | |
935 | wxLogError(wxT("Impossible to find out how to open files of extension '%s'"), | |
936 | ext.c_str()); | |
937 | return; | |
938 | } | |
939 | ||
940 | DoAsyncExec(cmd); | |
941 | } | |
942 | ||
943 | void MyFrame::OnFileLaunch(wxCommandEvent& WXUNUSED(event)) | |
944 | { | |
945 | if ( !AskUserForFileName() ) | |
946 | return; | |
947 | ||
948 | if ( !wxLaunchDefaultApplication(gs_lastFile) ) | |
949 | { | |
950 | wxLogError("Opening \"%s\" in default application failed.", gs_lastFile); | |
951 | } | |
952 | } | |
953 | ||
954 | void MyFrame::OnOpenURL(wxCommandEvent& WXUNUSED(event)) | |
955 | { | |
956 | static wxString s_url(wxT("http://www.wxwidgets.org/")); | |
957 | ||
958 | wxString filename = wxGetTextFromUser | |
959 | ( | |
960 | wxT("Enter the URL"), | |
961 | wxT("exec sample"), | |
962 | s_url, | |
963 | this | |
964 | ); | |
965 | ||
966 | if ( filename.empty() ) | |
967 | return; | |
968 | ||
969 | s_url = filename; | |
970 | ||
971 | if ( !wxLaunchDefaultBrowser(s_url) ) | |
972 | { | |
973 | wxLogError(wxT("Failed to open URL \"%s\""), s_url.c_str()); | |
974 | } | |
975 | } | |
976 | ||
977 | // ---------------------------------------------------------------------------- | |
978 | // DDE stuff | |
979 | // ---------------------------------------------------------------------------- | |
980 | ||
981 | #ifdef __WINDOWS__ | |
982 | ||
983 | bool MyFrame::GetDDEServer() | |
984 | { | |
985 | wxString server = wxGetTextFromUser(wxT("Server to connect to:"), | |
986 | DIALOG_TITLE, m_server); | |
987 | if ( !server ) | |
988 | return false; | |
989 | ||
990 | m_server = server; | |
991 | ||
992 | wxString topic = wxGetTextFromUser(wxT("DDE topic:"), DIALOG_TITLE, m_topic); | |
993 | if ( !topic ) | |
994 | return false; | |
995 | ||
996 | m_topic = topic; | |
997 | ||
998 | wxString cmd = wxGetTextFromUser(wxT("DDE command:"), DIALOG_TITLE, m_cmdDde); | |
999 | if ( !cmd ) | |
1000 | return false; | |
1001 | ||
1002 | m_cmdDde = cmd; | |
1003 | ||
1004 | return true; | |
1005 | } | |
1006 | ||
1007 | void MyFrame::OnDDEExec(wxCommandEvent& WXUNUSED(event)) | |
1008 | { | |
1009 | if ( !GetDDEServer() ) | |
1010 | return; | |
1011 | ||
1012 | wxDDEClient client; | |
1013 | wxConnectionBase *conn = client.MakeConnection(wxEmptyString, m_server, m_topic); | |
1014 | if ( !conn ) | |
1015 | { | |
1016 | wxLogError(wxT("Failed to connect to the DDE server '%s'."), | |
1017 | m_server.c_str()); | |
1018 | } | |
1019 | else | |
1020 | { | |
1021 | if ( !conn->Execute(m_cmdDde) ) | |
1022 | { | |
1023 | wxLogError(wxT("Failed to execute command '%s' via DDE."), | |
1024 | m_cmdDde.c_str()); | |
1025 | } | |
1026 | else | |
1027 | { | |
1028 | wxLogStatus(wxT("Successfully executed DDE command")); | |
1029 | } | |
1030 | } | |
1031 | } | |
1032 | ||
1033 | void MyFrame::OnDDERequest(wxCommandEvent& WXUNUSED(event)) | |
1034 | { | |
1035 | if ( !GetDDEServer() ) | |
1036 | return; | |
1037 | ||
1038 | wxDDEClient client; | |
1039 | wxConnectionBase *conn = client.MakeConnection(wxEmptyString, m_server, m_topic); | |
1040 | if ( !conn ) | |
1041 | { | |
1042 | wxLogError(wxT("Failed to connect to the DDE server '%s'."), | |
1043 | m_server.c_str()); | |
1044 | } | |
1045 | else | |
1046 | { | |
1047 | if ( !conn->Request(m_cmdDde) ) | |
1048 | { | |
1049 | wxLogError(wxT("Failed to send request '%s' via DDE."), | |
1050 | m_cmdDde.c_str()); | |
1051 | } | |
1052 | else | |
1053 | { | |
1054 | wxLogStatus(wxT("Successfully sent DDE request.")); | |
1055 | } | |
1056 | } | |
1057 | } | |
1058 | ||
1059 | #endif // __WINDOWS__ | |
1060 | ||
1061 | // ---------------------------------------------------------------------------- | |
1062 | // various helpers | |
1063 | // ---------------------------------------------------------------------------- | |
1064 | ||
1065 | // input polling | |
1066 | void MyFrame::OnIdle(wxIdleEvent& event) | |
1067 | { | |
1068 | size_t count = m_running.GetCount(); | |
1069 | for ( size_t n = 0; n < count; n++ ) | |
1070 | { | |
1071 | if ( m_running[n]->HasInput() ) | |
1072 | { | |
1073 | event.RequestMore(); | |
1074 | } | |
1075 | } | |
1076 | } | |
1077 | ||
1078 | void MyFrame::OnIdleTimer(wxTimerEvent& WXUNUSED(event)) | |
1079 | { | |
1080 | wxWakeUpIdle(); | |
1081 | } | |
1082 | ||
1083 | void MyFrame::OnBgTimer(wxTimerEvent& WXUNUSED(event)) | |
1084 | { | |
1085 | static unsigned long s_ticks = 0; | |
1086 | SetStatusText(wxString::Format("%lu ticks", s_ticks++), 1); | |
1087 | } | |
1088 | ||
1089 | void MyFrame::OnProcessTerminated(MyPipedProcess *process) | |
1090 | { | |
1091 | RemovePipedProcess(process); | |
1092 | } | |
1093 | ||
1094 | void MyFrame::OnAsyncTermination(MyProcess *process) | |
1095 | { | |
1096 | m_allAsync.Remove(process); | |
1097 | ||
1098 | delete process; | |
1099 | } | |
1100 | ||
1101 | void MyFrame::AddPipedProcess(MyPipedProcess *process) | |
1102 | { | |
1103 | if ( m_running.IsEmpty() ) | |
1104 | { | |
1105 | // we want to start getting the timer events to ensure that a | |
1106 | // steady stream of idle events comes in -- otherwise we | |
1107 | // wouldn't be able to poll the child process input | |
1108 | m_timerIdleWakeUp.Start(100); | |
1109 | } | |
1110 | //else: the timer is already running | |
1111 | ||
1112 | m_running.Add(process); | |
1113 | m_allAsync.Add(process); | |
1114 | } | |
1115 | ||
1116 | void MyFrame::RemovePipedProcess(MyPipedProcess *process) | |
1117 | { | |
1118 | m_running.Remove(process); | |
1119 | ||
1120 | if ( m_running.IsEmpty() ) | |
1121 | { | |
1122 | // we don't need to get idle events all the time any more | |
1123 | m_timerIdleWakeUp.Stop(); | |
1124 | } | |
1125 | } | |
1126 | ||
1127 | void MyFrame::ShowOutput(const wxString& cmd, | |
1128 | const wxArrayString& output, | |
1129 | const wxString& title) | |
1130 | { | |
1131 | size_t count = output.GetCount(); | |
1132 | if ( !count ) | |
1133 | return; | |
1134 | ||
1135 | m_lbox->Append(wxString::Format(wxT("--- %s of '%s' ---"), | |
1136 | title.c_str(), cmd.c_str())); | |
1137 | ||
1138 | for ( size_t n = 0; n < count; n++ ) | |
1139 | { | |
1140 | m_lbox->Append(output[n]); | |
1141 | } | |
1142 | ||
1143 | m_lbox->Append(wxString::Format(wxT("--- End of %s ---"), | |
1144 | title.Lower().c_str())); | |
1145 | } | |
1146 | ||
1147 | // ---------------------------------------------------------------------------- | |
1148 | // MyProcess | |
1149 | // ---------------------------------------------------------------------------- | |
1150 | ||
1151 | void MyProcess::OnTerminate(int pid, int status) | |
1152 | { | |
1153 | wxLogStatus(m_parent, wxT("Process %u ('%s') terminated with exit code %d."), | |
1154 | pid, m_cmd.c_str(), status); | |
1155 | ||
1156 | m_parent->OnAsyncTermination(this); | |
1157 | } | |
1158 | ||
1159 | // ---------------------------------------------------------------------------- | |
1160 | // MyPipedProcess | |
1161 | // ---------------------------------------------------------------------------- | |
1162 | ||
1163 | bool MyPipedProcess::HasInput() | |
1164 | { | |
1165 | bool hasInput = false; | |
1166 | ||
1167 | if ( IsInputAvailable() ) | |
1168 | { | |
1169 | wxTextInputStream tis(*GetInputStream()); | |
1170 | ||
1171 | // this assumes that the output is always line buffered | |
1172 | wxString msg; | |
1173 | msg << m_cmd << wxT(" (stdout): ") << tis.ReadLine(); | |
1174 | ||
1175 | m_parent->GetLogListBox()->Append(msg); | |
1176 | ||
1177 | hasInput = true; | |
1178 | } | |
1179 | ||
1180 | if ( IsErrorAvailable() ) | |
1181 | { | |
1182 | wxTextInputStream tis(*GetErrorStream()); | |
1183 | ||
1184 | // this assumes that the output is always line buffered | |
1185 | wxString msg; | |
1186 | msg << m_cmd << wxT(" (stderr): ") << tis.ReadLine(); | |
1187 | ||
1188 | m_parent->GetLogListBox()->Append(msg); | |
1189 | ||
1190 | hasInput = true; | |
1191 | } | |
1192 | ||
1193 | return hasInput; | |
1194 | } | |
1195 | ||
1196 | void MyPipedProcess::OnTerminate(int pid, int status) | |
1197 | { | |
1198 | // show the rest of the output | |
1199 | while ( HasInput() ) | |
1200 | ; | |
1201 | ||
1202 | m_parent->OnProcessTerminated(this); | |
1203 | ||
1204 | MyProcess::OnTerminate(pid, status); | |
1205 | } | |
1206 | ||
1207 | // ---------------------------------------------------------------------------- | |
1208 | // MyPipedProcess2 | |
1209 | // ---------------------------------------------------------------------------- | |
1210 | ||
1211 | bool MyPipedProcess2::HasInput() | |
1212 | { | |
1213 | if ( !m_input.empty() ) | |
1214 | { | |
1215 | wxTextOutputStream os(*GetOutputStream()); | |
1216 | os.WriteString(m_input); | |
1217 | ||
1218 | CloseOutput(); | |
1219 | m_input.clear(); | |
1220 | ||
1221 | // call us once again - may be we'll have output | |
1222 | return true; | |
1223 | } | |
1224 | ||
1225 | return MyPipedProcess::HasInput(); | |
1226 | } | |
1227 | ||
1228 | // ============================================================================ | |
1229 | // MyPipeFrame implementation | |
1230 | // ============================================================================ | |
1231 | ||
1232 | MyPipeFrame::MyPipeFrame(wxFrame *parent, | |
1233 | const wxString& cmd, | |
1234 | wxProcess *process) | |
1235 | : wxFrame(parent, wxID_ANY, cmd), | |
1236 | m_process(process), | |
1237 | // in a real program we'd check that the streams are !NULL here | |
1238 | m_out(*process->GetOutputStream()), | |
1239 | m_in(*process->GetInputStream()), | |
1240 | m_err(*process->GetErrorStream()) | |
1241 | { | |
1242 | m_process->SetNextHandler(this); | |
1243 | ||
1244 | wxPanel *panel = new wxPanel(this, wxID_ANY); | |
1245 | ||
1246 | m_textOut = new wxTextCtrl(panel, wxID_ANY, wxEmptyString, | |
1247 | wxDefaultPosition, wxDefaultSize, | |
1248 | wxTE_PROCESS_ENTER); | |
1249 | m_textIn = new wxTextCtrl(panel, wxID_ANY, wxEmptyString, | |
1250 | wxDefaultPosition, wxDefaultSize, | |
1251 | wxTE_MULTILINE | wxTE_RICH); | |
1252 | m_textIn->SetEditable(false); | |
1253 | m_textErr = new wxTextCtrl(panel, wxID_ANY, wxEmptyString, | |
1254 | wxDefaultPosition, wxDefaultSize, | |
1255 | wxTE_MULTILINE | wxTE_RICH); | |
1256 | m_textErr->SetEditable(false); | |
1257 | ||
1258 | wxSizer *sizerTop = new wxBoxSizer(wxVERTICAL); | |
1259 | sizerTop->Add(m_textOut, 0, wxGROW | wxALL, 5); | |
1260 | ||
1261 | wxSizer *sizerBtns = new wxBoxSizer(wxHORIZONTAL); | |
1262 | sizerBtns-> | |
1263 | Add(new wxButton(panel, Exec_Btn_Send, wxT("&Send")), 0, wxALL, 5); | |
1264 | sizerBtns-> | |
1265 | Add(new wxButton(panel, Exec_Btn_SendFile, wxT("&File...")), 0, wxALL, 5); | |
1266 | sizerBtns-> | |
1267 | Add(new wxButton(panel, Exec_Btn_Get, wxT("&Get")), 0, wxALL, 5); | |
1268 | sizerBtns-> | |
1269 | Add(new wxButton(panel, Exec_Btn_Close, wxT("&Close")), 0, wxALL, 5); | |
1270 | ||
1271 | sizerTop->Add(sizerBtns, 0, wxCENTRE | wxALL, 5); | |
1272 | sizerTop->Add(m_textIn, 1, wxGROW | wxALL, 5); | |
1273 | sizerTop->Add(m_textErr, 1, wxGROW | wxALL, 5); | |
1274 | ||
1275 | panel->SetSizer(sizerTop); | |
1276 | sizerTop->Fit(this); | |
1277 | ||
1278 | Show(); | |
1279 | } | |
1280 | ||
1281 | void MyPipeFrame::OnBtnSendFile(wxCommandEvent& WXUNUSED(event)) | |
1282 | { | |
1283 | #if wxUSE_FILEDLG | |
1284 | wxFileDialog filedlg(this, wxT("Select file to send")); | |
1285 | if ( filedlg.ShowModal() != wxID_OK ) | |
1286 | return; | |
1287 | ||
1288 | wxFFile file(filedlg.GetFilename(), wxT("r")); | |
1289 | wxString data; | |
1290 | if ( !file.IsOpened() || !file.ReadAll(&data) ) | |
1291 | return; | |
1292 | ||
1293 | // can't write the entire string at once, this risk overflowing the pipe | |
1294 | // and we would dead lock | |
1295 | size_t len = data.length(); | |
1296 | const wxChar *pc = data.c_str(); | |
1297 | while ( len ) | |
1298 | { | |
1299 | const size_t CHUNK_SIZE = 4096; | |
1300 | m_out.Write(pc, len > CHUNK_SIZE ? CHUNK_SIZE : len); | |
1301 | ||
1302 | // note that not all data could have been written as we don't block on | |
1303 | // the write end of the pipe | |
1304 | const size_t lenChunk = m_out.LastWrite(); | |
1305 | ||
1306 | pc += lenChunk; | |
1307 | len -= lenChunk; | |
1308 | ||
1309 | DoGet(); | |
1310 | } | |
1311 | #endif // wxUSE_FILEDLG | |
1312 | } | |
1313 | ||
1314 | void MyPipeFrame::DoGet() | |
1315 | { | |
1316 | // we don't have any way to be notified when any input appears on the | |
1317 | // stream so we have to poll it :-( | |
1318 | DoGetFromStream(m_textIn, m_in); | |
1319 | DoGetFromStream(m_textErr, m_err); | |
1320 | } | |
1321 | ||
1322 | void MyPipeFrame::DoGetFromStream(wxTextCtrl *text, wxInputStream& in) | |
1323 | { | |
1324 | while ( in.CanRead() ) | |
1325 | { | |
1326 | wxChar buffer[4096]; | |
1327 | buffer[in.Read(buffer, WXSIZEOF(buffer) - 1).LastRead()] = wxT('\0'); | |
1328 | ||
1329 | text->AppendText(buffer); | |
1330 | } | |
1331 | } | |
1332 | ||
1333 | void MyPipeFrame::DoClose() | |
1334 | { | |
1335 | m_process->CloseOutput(); | |
1336 | ||
1337 | DisableInput(); | |
1338 | } | |
1339 | ||
1340 | void MyPipeFrame::DisableInput() | |
1341 | { | |
1342 | m_textOut->SetEditable(false); | |
1343 | FindWindow(Exec_Btn_Send)->Disable(); | |
1344 | FindWindow(Exec_Btn_SendFile)->Disable(); | |
1345 | FindWindow(Exec_Btn_Close)->Disable(); | |
1346 | } | |
1347 | ||
1348 | void MyPipeFrame::DisableOutput() | |
1349 | { | |
1350 | FindWindow(Exec_Btn_Get)->Disable(); | |
1351 | } | |
1352 | ||
1353 | void MyPipeFrame::OnClose(wxCloseEvent& event) | |
1354 | { | |
1355 | if ( m_process ) | |
1356 | { | |
1357 | // we're not interested in getting the process termination notification | |
1358 | // if we are closing it ourselves | |
1359 | wxProcess *process = m_process; | |
1360 | m_process = NULL; | |
1361 | process->SetNextHandler(NULL); | |
1362 | ||
1363 | process->CloseOutput(); | |
1364 | } | |
1365 | ||
1366 | event.Skip(); | |
1367 | } | |
1368 | ||
1369 | void MyPipeFrame::OnProcessTerm(wxProcessEvent& WXUNUSED(event)) | |
1370 | { | |
1371 | DoGet(); | |
1372 | ||
1373 | delete m_process; | |
1374 | m_process = NULL; | |
1375 | ||
1376 | wxLogWarning(wxT("The other process has terminated, closing")); | |
1377 | ||
1378 | DisableInput(); | |
1379 | DisableOutput(); | |
1380 | } |