]> git.saurik.com Git - wxWidgets.git/blob - src/unix/utilsunx.cpp
Handle non-ASCII output from lsb_release in wxGetLinuxDistributionInfo().
[wxWidgets.git] / src / unix / utilsunx.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/unix/utilsunx.cpp
3 // Purpose: generic Unix implementation of many wx functions (for wxBase)
4 // Author: Vadim Zeitlin
5 // Copyright: (c) 1998 Robert Roebling, Vadim Zeitlin
6 // (c) 2013 Rob Bresalier, Vadim Zeitlin
7 // Licence: wxWindows licence
8 /////////////////////////////////////////////////////////////////////////////
9
10 // ============================================================================
11 // declarations
12 // ============================================================================
13
14 // ----------------------------------------------------------------------------
15 // headers
16 // ----------------------------------------------------------------------------
17
18 // for compilers that support precompilation, includes "wx.h".
19 #include "wx/wxprec.h"
20
21 #include "wx/utils.h"
22
23 #define USE_PUTENV (!defined(HAVE_SETENV) && defined(HAVE_PUTENV))
24
25 #ifndef WX_PRECOMP
26 #include "wx/string.h"
27 #include "wx/intl.h"
28 #include "wx/log.h"
29 #include "wx/app.h"
30 #include "wx/wxcrtvararg.h"
31 #if USE_PUTENV
32 #include "wx/module.h"
33 #include "wx/hashmap.h"
34 #endif
35 #endif
36
37 #include "wx/apptrait.h"
38
39 #include "wx/process.h"
40 #include "wx/scopedptr.h"
41 #include "wx/thread.h"
42
43 #include "wx/cmdline.h"
44
45 #include "wx/wfstream.h"
46
47 #include "wx/private/selectdispatcher.h"
48 #include "wx/private/fdiodispatcher.h"
49 #include "wx/unix/execute.h"
50 #include "wx/unix/pipe.h"
51 #include "wx/unix/private.h"
52
53 #include "wx/evtloop.h"
54 #include "wx/mstream.h"
55 #include "wx/private/fdioeventloopsourcehandler.h"
56
57 #include <pwd.h>
58 #include <sys/wait.h> // waitpid()
59
60 #ifdef HAVE_SYS_SELECT_H
61 # include <sys/select.h>
62 #endif
63
64 #define HAS_PIPE_STREAMS (wxUSE_STREAMS && wxUSE_FILE)
65
66 #if HAS_PIPE_STREAMS
67
68 #include "wx/private/pipestream.h"
69 #include "wx/private/streamtempinput.h"
70 #include "wx/unix/private/executeiohandler.h"
71
72 #endif // HAS_PIPE_STREAMS
73
74 // not only the statfs syscall is called differently depending on platform, but
75 // one of its incarnations, statvfs(), takes different arguments under
76 // different platforms and even different versions of the same system (Solaris
77 // 7 and 8): if you want to test for this, don't forget that the problems only
78 // appear if the large files support is enabled
79 #ifdef HAVE_STATFS
80 #ifdef __BSD__
81 #include <sys/param.h>
82 #include <sys/mount.h>
83 #else // !__BSD__
84 #include <sys/vfs.h>
85 #endif // __BSD__/!__BSD__
86
87 #define wxStatfs statfs
88
89 #ifndef HAVE_STATFS_DECL
90 // some systems lack statfs() prototype in the system headers (AIX 4)
91 extern "C" int statfs(const char *path, struct statfs *buf);
92 #endif
93 #endif // HAVE_STATFS
94
95 #ifdef HAVE_STATVFS
96 #include <sys/statvfs.h>
97
98 #define wxStatfs statvfs
99 #endif // HAVE_STATVFS
100
101 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
102 // WX_STATFS_T is detected by configure
103 #define wxStatfs_t WX_STATFS_T
104 #endif
105
106 // SGI signal.h defines signal handler arguments differently depending on
107 // whether _LANGUAGE_C_PLUS_PLUS is set or not - do set it
108 #if defined(__SGI__) && !defined(_LANGUAGE_C_PLUS_PLUS)
109 #define _LANGUAGE_C_PLUS_PLUS 1
110 #endif // SGI hack
111
112 #include <stdarg.h>
113 #include <dirent.h>
114 #include <string.h>
115 #include <sys/stat.h>
116 #include <sys/types.h>
117 #include <sys/wait.h>
118 #include <unistd.h>
119 #include <errno.h>
120 #include <netdb.h>
121 #include <signal.h>
122 #include <time.h> // nanosleep() and/or usleep()
123 #include <ctype.h> // isspace()
124 #include <sys/time.h> // needed for FD_SETSIZE
125
126 #ifdef HAVE_UNAME
127 #include <sys/utsname.h> // for uname()
128 #endif // HAVE_UNAME
129
130 // Used by wxGetFreeMemory().
131 #ifdef __SGI__
132 #include <sys/sysmp.h>
133 #include <sys/sysinfo.h> // for SAGET and MINFO structures
134 #endif
135
136 #ifdef HAVE_SETPRIORITY
137 #include <sys/resource.h> // for setpriority()
138 #endif
139
140 // ----------------------------------------------------------------------------
141 // conditional compilation
142 // ----------------------------------------------------------------------------
143
144 // many versions of Unices have this function, but it is not defined in system
145 // headers - please add your system here if it is the case for your OS.
146 // SunOS < 5.6 (i.e. Solaris < 2.6) and DG-UX are like this.
147 #if !defined(HAVE_USLEEP) && \
148 ((defined(__SUN__) && !defined(__SunOs_5_6) && \
149 !defined(__SunOs_5_7) && !defined(__SUNPRO_CC)) || \
150 defined(__osf__) || defined(__EMX__))
151 extern "C"
152 {
153 #ifdef __EMX__
154 /* I copied this from the XFree86 diffs. AV. */
155 #define INCL_DOSPROCESS
156 #include <os2.h>
157 inline void usleep(unsigned long delay)
158 {
159 DosSleep(delay ? (delay/1000l) : 1l);
160 }
161 #else // Unix
162 int usleep(unsigned int usec);
163 #endif // __EMX__/Unix
164 };
165
166 #define HAVE_USLEEP 1
167 #endif // Unices without usleep()
168
169 // ============================================================================
170 // implementation
171 // ============================================================================
172
173 // ----------------------------------------------------------------------------
174 // sleeping
175 // ----------------------------------------------------------------------------
176
177 void wxSleep(int nSecs)
178 {
179 sleep(nSecs);
180 }
181
182 void wxMicroSleep(unsigned long microseconds)
183 {
184 #if defined(HAVE_NANOSLEEP)
185 timespec tmReq;
186 tmReq.tv_sec = (time_t)(microseconds / 1000000);
187 tmReq.tv_nsec = (microseconds % 1000000) * 1000;
188
189 // we're not interested in remaining time nor in return value
190 (void)nanosleep(&tmReq, NULL);
191 #elif defined(HAVE_USLEEP)
192 // uncomment this if you feel brave or if you are sure that your version
193 // of Solaris has a safe usleep() function but please notice that usleep()
194 // is known to lead to crashes in MT programs in Solaris 2.[67] and is not
195 // documented as MT-Safe
196 #if defined(__SUN__) && wxUSE_THREADS
197 #error "usleep() cannot be used in MT programs under Solaris."
198 #endif // Sun
199
200 usleep(microseconds);
201 #elif defined(HAVE_SLEEP)
202 // under BeOS sleep() takes seconds (what about other platforms, if any?)
203 sleep(microseconds * 1000000);
204 #else // !sleep function
205 #error "usleep() or nanosleep() function required for wxMicroSleep"
206 #endif // sleep function
207 }
208
209 void wxMilliSleep(unsigned long milliseconds)
210 {
211 wxMicroSleep(milliseconds*1000);
212 }
213
214 // ----------------------------------------------------------------------------
215 // process management
216 // ----------------------------------------------------------------------------
217
218 int wxKill(long pid, wxSignal sig, wxKillError *rc, int flags)
219 {
220 int err = kill((pid_t) (flags & wxKILL_CHILDREN) ? -pid : pid, (int)sig);
221 if ( rc )
222 {
223 switch ( err ? errno : 0 )
224 {
225 case 0:
226 *rc = wxKILL_OK;
227 break;
228
229 case EINVAL:
230 *rc = wxKILL_BAD_SIGNAL;
231 break;
232
233 case EPERM:
234 *rc = wxKILL_ACCESS_DENIED;
235 break;
236
237 case ESRCH:
238 *rc = wxKILL_NO_PROCESS;
239 break;
240
241 default:
242 // this goes against Unix98 docs so log it
243 wxLogDebug(wxT("unexpected kill(2) return value %d"), err);
244
245 // something else...
246 *rc = wxKILL_ERROR;
247 }
248 }
249
250 return err;
251 }
252
253 // Shutdown or reboot the PC
254 bool wxShutdown(int flags)
255 {
256 flags &= ~wxSHUTDOWN_FORCE;
257
258 wxChar level;
259 switch ( flags )
260 {
261 case wxSHUTDOWN_POWEROFF:
262 level = wxT('0');
263 break;
264
265 case wxSHUTDOWN_REBOOT:
266 level = wxT('6');
267 break;
268
269 case wxSHUTDOWN_LOGOFF:
270 // TODO: use dcop to log off?
271 return false;
272
273 default:
274 wxFAIL_MSG( wxT("unknown wxShutdown() flag") );
275 return false;
276 }
277
278 return system(wxString::Format("init %c", level).mb_str()) == 0;
279 }
280
281 // ----------------------------------------------------------------------------
282 // wxStream classes to support IO redirection in wxExecute
283 // ----------------------------------------------------------------------------
284
285 #if HAS_PIPE_STREAMS
286
287 bool wxPipeInputStream::CanRead() const
288 {
289 if ( m_lasterror == wxSTREAM_EOF )
290 return false;
291
292 // check if there is any input available
293 struct timeval tv;
294 tv.tv_sec = 0;
295 tv.tv_usec = 0;
296
297 const int fd = m_file->fd();
298
299 fd_set readfds;
300
301 wxFD_ZERO(&readfds);
302 wxFD_SET(fd, &readfds);
303
304 switch ( select(fd + 1, &readfds, NULL, NULL, &tv) )
305 {
306 case -1:
307 wxLogSysError(_("Impossible to get child process input"));
308 // fall through
309
310 case 0:
311 return false;
312
313 default:
314 wxFAIL_MSG(wxT("unexpected select() return value"));
315 // still fall through
316
317 case 1:
318 // input available -- or maybe not, as select() returns 1 when a
319 // read() will complete without delay, but it could still not read
320 // anything
321 return !Eof();
322 }
323 }
324
325 size_t wxPipeOutputStream::OnSysWrite(const void *buffer, size_t size)
326 {
327 // We need to suppress error logging here, because on writing to a pipe
328 // which is full, wxFile::Write reports a system error. However, this is
329 // not an extraordinary situation, and it should not be reported to the
330 // user (but if really needed, the program can recognize it by checking
331 // whether LastRead() == 0.) Other errors will be reported below.
332 size_t ret;
333 {
334 wxLogNull logNo;
335 ret = m_file->Write(buffer, size);
336 }
337
338 switch ( m_file->GetLastError() )
339 {
340 // pipe is full
341 #ifdef EAGAIN
342 case EAGAIN:
343 #endif
344 #if defined(EWOULDBLOCK) && (EWOULDBLOCK != EAGAIN)
345 case EWOULDBLOCK:
346 #endif
347 // do not treat it as an error
348 m_file->ClearLastError();
349 // fall through
350
351 // no error
352 case 0:
353 break;
354
355 // some real error
356 default:
357 wxLogSysError(_("Can't write to child process's stdin"));
358 m_lasterror = wxSTREAM_WRITE_ERROR;
359 }
360
361 return ret;
362 }
363
364 #endif // HAS_PIPE_STREAMS
365
366 // ----------------------------------------------------------------------------
367 // wxShell
368 // ----------------------------------------------------------------------------
369
370 static wxString wxMakeShellCommand(const wxString& command)
371 {
372 wxString cmd;
373 if ( !command )
374 {
375 // just an interactive shell
376 cmd = wxT("xterm");
377 }
378 else
379 {
380 // execute command in a shell
381 cmd << wxT("/bin/sh -c '") << command << wxT('\'');
382 }
383
384 return cmd;
385 }
386
387 bool wxShell(const wxString& command)
388 {
389 return wxExecute(wxMakeShellCommand(command), wxEXEC_SYNC) == 0;
390 }
391
392 bool wxShell(const wxString& command, wxArrayString& output)
393 {
394 wxCHECK_MSG( !command.empty(), false, wxT("can't exec shell non interactively") );
395
396 return wxExecute(wxMakeShellCommand(command), output);
397 }
398
399 namespace
400 {
401
402 // helper class for storing arguments as char** array suitable for passing to
403 // execvp(), whatever form they were passed to us
404 class ArgsArray
405 {
406 public:
407 ArgsArray(const wxArrayString& args)
408 {
409 Init(args.size());
410
411 for ( int i = 0; i < m_argc; i++ )
412 {
413 m_argv[i] = wxStrdup(args[i]);
414 }
415 }
416
417 #if wxUSE_UNICODE
418 ArgsArray(wchar_t **wargv)
419 {
420 int argc = 0;
421 while ( wargv[argc] )
422 argc++;
423
424 Init(argc);
425
426 for ( int i = 0; i < m_argc; i++ )
427 {
428 m_argv[i] = wxSafeConvertWX2MB(wargv[i]).release();
429 }
430 }
431 #endif // wxUSE_UNICODE
432
433 ~ArgsArray()
434 {
435 for ( int i = 0; i < m_argc; i++ )
436 {
437 free(m_argv[i]);
438 }
439
440 delete [] m_argv;
441 }
442
443 operator char**() const { return m_argv; }
444
445 private:
446 void Init(int argc)
447 {
448 m_argc = argc;
449 m_argv = new char *[m_argc + 1];
450 m_argv[m_argc] = NULL;
451 }
452
453 int m_argc;
454 char **m_argv;
455
456 wxDECLARE_NO_COPY_CLASS(ArgsArray);
457 };
458
459 } // anonymous namespace
460
461 // ----------------------------------------------------------------------------
462 // wxExecute implementations
463 // ----------------------------------------------------------------------------
464
465 #if defined(__DARWIN__)
466 bool wxMacLaunch(char **argv);
467 #endif
468
469 long wxExecute(const wxString& command, int flags, wxProcess *process,
470 const wxExecuteEnv *env)
471 {
472 ArgsArray argv(wxCmdLineParser::ConvertStringToArgs(command,
473 wxCMD_LINE_SPLIT_UNIX));
474
475 return wxExecute(argv, flags, process, env);
476 }
477
478 #if wxUSE_UNICODE
479
480 long wxExecute(wchar_t **wargv, int flags, wxProcess *process,
481 const wxExecuteEnv *env)
482 {
483 ArgsArray argv(wargv);
484
485 return wxExecute(argv, flags, process, env);
486 }
487
488 #endif // wxUSE_UNICODE
489
490 namespace
491 {
492
493 // Helper function of wxExecute(): wait for the process termination without
494 // dispatching any events.
495 //
496 // This is used in wxEXEC_NOEVENTS case.
497 int BlockUntilChildExit(wxExecuteData& execData)
498 {
499 wxCHECK_MSG( wxTheApp, -1,
500 wxS("Can't block until child exit without wxTheApp") );
501
502 #if wxUSE_SELECT_DISPATCHER
503
504 // Even if we don't want to dispatch events, we still need to handle
505 // child IO notifications and process termination concurrently, i.e.
506 // we can't simply block waiting for the child to terminate as we would
507 // dead lock if it writes more than the pipe buffer size (typically
508 // 4KB) bytes of output -- it would then block waiting for us to read
509 // the data while we'd block waiting for it to terminate.
510 //
511 // So while we don't use the full blown event loop, we still do use a
512 // dispatcher with which we register just the 3 FDs we're interested
513 // in: the child stdout and stderr and the pipe written to by the
514 // signal handler so that we could react to the child process
515 // termination too.
516
517 // Notice that we must create a new dispatcher object here instead of
518 // reusing the global wxFDIODispatcher::Get() because we want to
519 // monitor only the events on the FDs explicitly registered with this
520 // one and not all the other ones that could be registered with the
521 // global dispatcher (think about the case of nested wxExecute() calls).
522 wxSelectDispatcher dispatcher;
523
524 // Do register all the FDs we want to monitor here: first, the one used to
525 // handle the signals asynchronously.
526 wxScopedPtr<wxFDIOHandler>
527 signalHandler(wxTheApp->RegisterSignalWakeUpPipe(dispatcher));
528
529 #if wxUSE_STREAMS
530 // And then the two for the child output and error streams if necessary.
531 wxScopedPtr<wxFDIOHandler>
532 stdoutHandler,
533 stderrHandler;
534 if ( execData.IsRedirected() )
535 {
536 stdoutHandler.reset(new wxExecuteFDIOHandler
537 (
538 dispatcher,
539 execData.fdOut,
540 execData.bufOut
541 ));
542 stderrHandler.reset(new wxExecuteFDIOHandler
543 (
544 dispatcher,
545 execData.fdErr,
546 execData.bufErr
547 ));
548 }
549 #endif // wxUSE_STREAMS
550
551 // And dispatch until the PID is reset from wxExecuteData::OnExit().
552 while ( execData.pid )
553 {
554 dispatcher.Dispatch();
555 }
556
557 return execData.exitcode;
558 #else // !wxUSE_SELECT_DISPATCHER
559 wxFAIL_MSG( wxS("Can't block until child exit without wxSelectDispatcher") );
560
561 return -1;
562 #endif // wxUSE_SELECT_DISPATCHER/!wxUSE_SELECT_DISPATCHER
563 }
564
565 } // anonymous namespace
566
567 // wxExecute: the real worker function
568 long wxExecute(char **argv, int flags, wxProcess *process,
569 const wxExecuteEnv *env)
570 {
571 // for the sync execution, we return -1 to indicate failure, but for async
572 // case we return 0 which is never a valid PID
573 //
574 // we define this as a macro, not a variable, to avoid compiler warnings
575 // about "ERROR_RETURN_CODE value may be clobbered by fork()"
576 #define ERROR_RETURN_CODE ((flags & wxEXEC_SYNC) ? -1 : 0)
577
578 wxCHECK_MSG( *argv, ERROR_RETURN_CODE, wxT("can't exec empty command") );
579
580 #if wxUSE_THREADS
581 // fork() doesn't mix well with POSIX threads: on many systems the program
582 // deadlocks or crashes for some reason. Probably our code is buggy and
583 // doesn't do something which must be done to allow this to work, but I
584 // don't know what yet, so for now just warn the user (this is the least we
585 // can do) about it
586 wxASSERT_MSG( wxThread::IsMain(),
587 wxT("wxExecute() can be called only from the main thread") );
588 #endif // wxUSE_THREADS
589
590 #if defined(__WXCOCOA__) || ( defined(__WXOSX_MAC__) && wxOSX_USE_COCOA_OR_CARBON )
591 // wxMacLaunch() only executes app bundles and only does it asynchronously.
592 // It returns false if the target is not an app bundle, thus falling
593 // through to the regular code for non app bundles.
594 if ( !(flags & wxEXEC_SYNC) && wxMacLaunch(argv) )
595 {
596 // we don't have any PID to return so just make up something non null
597 return -1;
598 }
599 #endif // __DARWIN__
600
601 // this struct contains all information which we use for housekeeping
602 wxScopedPtr<wxExecuteData> execDataPtr(new wxExecuteData);
603 wxExecuteData& execData = *execDataPtr;
604
605 execData.flags = flags;
606 execData.process = process;
607
608 // create pipes for inter process communication
609 wxPipe pipeIn, // stdin
610 pipeOut, // stdout
611 pipeErr; // stderr
612
613 if ( process && process->IsRedirected() )
614 {
615 if ( !pipeIn.Create() || !pipeOut.Create() || !pipeErr.Create() )
616 {
617 wxLogError( _("Failed to execute '%s'\n"), *argv );
618
619 return ERROR_RETURN_CODE;
620 }
621 }
622
623 // priority: we need to map wxWidgets priority which is in the range 0..100
624 // to Unix nice value which is in the range -20..19. As there is an odd
625 // number of elements in our range and an even number in the Unix one, we
626 // have to do it in this rather ugly way to guarantee that:
627 // 1. wxPRIORITY_{MIN,DEFAULT,MAX} map to -20, 0 and 19 respectively.
628 // 2. The mapping is monotonously increasing.
629 // 3. The mapping is onto the target range.
630 int prio = process ? process->GetPriority() : 0;
631 if ( prio <= 50 )
632 prio = (2*prio)/5 - 20;
633 else if ( prio < 55 )
634 prio = 1;
635 else
636 prio = (2*prio)/5 - 21;
637
638 // fork the process
639 //
640 // NB: do *not* use vfork() here, it completely breaks this code for some
641 // reason under Solaris (and maybe others, although not under Linux)
642 // But on OpenVMS we do not have fork so we have to use vfork and
643 // cross our fingers that it works.
644 #ifdef __VMS
645 pid_t pid = vfork();
646 #else
647 pid_t pid = fork();
648 #endif
649 if ( pid == -1 ) // error?
650 {
651 wxLogSysError( _("Fork failed") );
652
653 return ERROR_RETURN_CODE;
654 }
655 else if ( pid == 0 ) // we're in child
656 {
657 // NB: we used to close all the unused descriptors of the child here
658 // but this broke some programs which relied on e.g. FD 1 being
659 // always opened so don't do it any more, after all there doesn't
660 // seem to be any real problem with keeping them opened
661
662 #if !defined(__VMS) && !defined(__EMX__)
663 if ( flags & wxEXEC_MAKE_GROUP_LEADER )
664 {
665 // Set process group to child process' pid. Then killing -pid
666 // of the parent will kill the process and all of its children.
667 setsid();
668 }
669 #endif // !__VMS
670
671 #if defined(HAVE_SETPRIORITY)
672 if ( prio && setpriority(PRIO_PROCESS, 0, prio) != 0 )
673 {
674 wxLogSysError(_("Failed to set process priority"));
675 }
676 #endif // HAVE_SETPRIORITY
677
678 // redirect stdin, stdout and stderr
679 if ( pipeIn.IsOk() )
680 {
681 if ( dup2(pipeIn[wxPipe::Read], STDIN_FILENO) == -1 ||
682 dup2(pipeOut[wxPipe::Write], STDOUT_FILENO) == -1 ||
683 dup2(pipeErr[wxPipe::Write], STDERR_FILENO) == -1 )
684 {
685 wxLogSysError(_("Failed to redirect child process input/output"));
686 }
687
688 pipeIn.Close();
689 pipeOut.Close();
690 pipeErr.Close();
691 }
692
693 // Close all (presumably accidentally) inherited file descriptors to
694 // avoid descriptor leaks. This means that we don't allow inheriting
695 // them purposefully but this seems like a lesser evil in wx code.
696 // Ideally we'd provide some flag to indicate that none (or some?) of
697 // the descriptors do not need to be closed but for now this is better
698 // than never closing them at all as wx code never used FD_CLOEXEC.
699
700 // TODO: Iterating up to FD_SETSIZE is both inefficient (because it may
701 // be quite big) and incorrect (because in principle we could
702 // have more opened descriptions than this number). Unfortunately
703 // there is no good portable solution for closing all descriptors
704 // above a certain threshold but non-portable solutions exist for
705 // most platforms, see [http://stackoverflow.com/questions/899038/
706 // getting-the-highest-allocated-file-descriptor]
707 for ( int fd = 0; fd < (int)FD_SETSIZE; ++fd )
708 {
709 if ( fd != STDIN_FILENO &&
710 fd != STDOUT_FILENO &&
711 fd != STDERR_FILENO )
712 {
713 close(fd);
714 }
715 }
716
717
718 // Process additional options if we have any
719 if ( env )
720 {
721 // Change working directory if it is specified
722 if ( !env->cwd.empty() )
723 wxSetWorkingDirectory(env->cwd);
724
725 // Change environment if needed.
726 //
727 // NB: We can't use execve() currently because we allow using
728 // non full paths to wxExecute(), i.e. we want to search for
729 // the program in PATH. However it just might be simpler/better
730 // to do the search manually and use execve() envp parameter to
731 // set up the environment of the child process explicitly
732 // instead of doing what we do below.
733 if ( !env->env.empty() )
734 {
735 wxEnvVariableHashMap oldenv;
736 wxGetEnvMap(&oldenv);
737
738 // Remove unwanted variables
739 wxEnvVariableHashMap::const_iterator it;
740 for ( it = oldenv.begin(); it != oldenv.end(); ++it )
741 {
742 if ( env->env.find(it->first) == env->env.end() )
743 wxUnsetEnv(it->first);
744 }
745
746 // And add the new ones (possibly replacing the old values)
747 for ( it = env->env.begin(); it != env->env.end(); ++it )
748 wxSetEnv(it->first, it->second);
749 }
750 }
751
752 execvp(*argv, argv);
753
754 fprintf(stderr, "execvp(");
755 for ( char **a = argv; *a; a++ )
756 fprintf(stderr, "%s%s", a == argv ? "" : ", ", *a);
757 fprintf(stderr, ") failed with error %d!\n", errno);
758
759 // there is no return after successful exec()
760 _exit(-1);
761
762 // some compilers complain about missing return - of course, they
763 // should know that exit() doesn't return but what else can we do if
764 // they don't?
765 //
766 // and, sure enough, other compilers complain about unreachable code
767 // after exit() call, so we can just always have return here...
768 #if defined(__VMS) || defined(__INTEL_COMPILER)
769 return 0;
770 #endif
771 }
772 else // we're in parent
773 {
774 execData.OnStart(pid);
775
776 // prepare for IO redirection
777
778 #if HAS_PIPE_STREAMS
779
780 if ( process && process->IsRedirected() )
781 {
782 // Avoid deadlocks which could result from trying to write to the
783 // child input pipe end while the child itself is writing to its
784 // output end and waiting for us to read from it.
785 if ( !pipeIn.MakeNonBlocking(wxPipe::Write) )
786 {
787 // This message is not terrible useful for the user but what
788 // else can we do? Also, should we fail here or take the risk
789 // to continue and deadlock? Currently we choose the latter but
790 // it might not be the best idea.
791 wxLogSysError(_("Failed to set up non-blocking pipe, "
792 "the program might hang."));
793 #if wxUSE_LOG
794 wxLog::FlushActive();
795 #endif
796 }
797
798 wxOutputStream *inStream =
799 new wxPipeOutputStream(pipeIn.Detach(wxPipe::Write));
800
801 const int fdOut = pipeOut.Detach(wxPipe::Read);
802 wxPipeInputStream *outStream = new wxPipeInputStream(fdOut);
803
804 const int fdErr = pipeErr.Detach(wxPipe::Read);
805 wxPipeInputStream *errStream = new wxPipeInputStream(fdErr);
806
807 process->SetPipeStreams(outStream, inStream, errStream);
808
809 if ( flags & wxEXEC_SYNC )
810 {
811 execData.bufOut.Init(outStream);
812 execData.bufErr.Init(errStream);
813
814 execData.fdOut = fdOut;
815 execData.fdErr = fdErr;
816 }
817 }
818 #endif // HAS_PIPE_STREAMS
819
820 if ( pipeIn.IsOk() )
821 {
822 pipeIn.Close();
823 pipeOut.Close();
824 pipeErr.Close();
825 }
826
827 // For the asynchronous case we don't have to do anything else, just
828 // let the process run.
829 if ( !(flags & wxEXEC_SYNC) )
830 {
831 // Ensure that the housekeeping data is kept alive, it will be
832 // destroyed only when the child terminates.
833 execDataPtr.release();
834
835 return execData.pid;
836 }
837
838
839 // If we don't need to dispatch any events, things are relatively
840 // simple and we don't need to delegate to wxAppTraits.
841 if ( flags & wxEXEC_NOEVENTS )
842 {
843 return BlockUntilChildExit(execData);
844 }
845
846
847 // If we do need to dispatch events, enter a local event loop waiting
848 // until the child exits. As the exact kind of event loop depends on
849 // the sort of application we're in (console or GUI), we delegate this
850 // to wxAppTraits which virtualizes all the differences between the
851 // console and the GUI programs.
852 return wxApp::GetValidTraits().WaitForChild(execData);
853 }
854
855 #if !defined(__VMS) && !defined(__INTEL_COMPILER)
856 return ERROR_RETURN_CODE;
857 #endif
858 }
859
860 #undef ERROR_RETURN_CODE
861
862 // ----------------------------------------------------------------------------
863 // file and directory functions
864 // ----------------------------------------------------------------------------
865
866 const wxChar* wxGetHomeDir( wxString *home )
867 {
868 *home = wxGetUserHome();
869 wxString tmp;
870 if ( home->empty() )
871 *home = wxT("/");
872 #ifdef __VMS
873 tmp = *home;
874 if ( tmp.Last() != wxT(']'))
875 if ( tmp.Last() != wxT('/')) *home << wxT('/');
876 #endif
877 return home->c_str();
878 }
879
880 wxString wxGetUserHome( const wxString &user )
881 {
882 struct passwd *who = (struct passwd *) NULL;
883
884 if ( !user )
885 {
886 wxChar *ptr;
887
888 if ((ptr = wxGetenv(wxT("HOME"))) != NULL)
889 {
890 return ptr;
891 }
892
893 if ((ptr = wxGetenv(wxT("USER"))) != NULL ||
894 (ptr = wxGetenv(wxT("LOGNAME"))) != NULL)
895 {
896 who = getpwnam(wxSafeConvertWX2MB(ptr));
897 }
898
899 // make sure the user exists!
900 if ( !who )
901 {
902 who = getpwuid(getuid());
903 }
904 }
905 else
906 {
907 who = getpwnam (user.mb_str());
908 }
909
910 return wxSafeConvertMB2WX(who ? who->pw_dir : 0);
911 }
912
913 // ----------------------------------------------------------------------------
914 // network and user id routines
915 // ----------------------------------------------------------------------------
916
917 // Private utility function which returns output of the given command, removing
918 // the trailing newline.
919 //
920 // Note that by default use Latin-1 just to ensure that we never fail, but if
921 // the encoding is known (e.g. UTF-8 for lsb_release), it should be explicitly
922 // used instead.
923 static wxString
924 wxGetCommandOutput(const wxString &cmd, wxMBConv& conv = wxConvISO8859_1)
925 {
926 // Suppress stderr from the shell to avoid outputting errors if the command
927 // doesn't exist.
928 FILE *f = popen((cmd + " 2>/dev/null").ToAscii(), "r");
929 if ( !f )
930 {
931 // Notice that this doesn't happen simply if the command doesn't exist,
932 // but only in case of some really catastrophic failure inside popen()
933 // so we should really notify the user about this as this is not normal.
934 wxLogSysError(wxT("Executing \"%s\" failed"), cmd);
935 return wxString();
936 }
937
938 wxString s;
939 char buf[256];
940 while ( !feof(f) )
941 {
942 if ( !fgets(buf, sizeof(buf), f) )
943 break;
944
945 s += wxString(buf, conv);
946 }
947
948 pclose(f);
949
950 if ( !s.empty() && s.Last() == wxT('\n') )
951 s.RemoveLast();
952
953 return s;
954 }
955
956 // retrieve either the hostname or FQDN depending on platform (caller must
957 // check whether it's one or the other, this is why this function is for
958 // private use only)
959 static bool wxGetHostNameInternal(wxChar *buf, int sz)
960 {
961 wxCHECK_MSG( buf, false, wxT("NULL pointer in wxGetHostNameInternal") );
962
963 *buf = wxT('\0');
964
965 // we're using uname() which is POSIX instead of less standard sysinfo()
966 #if defined(HAVE_UNAME)
967 struct utsname uts;
968 bool ok = uname(&uts) != -1;
969 if ( ok )
970 {
971 wxStrlcpy(buf, wxSafeConvertMB2WX(uts.nodename), sz);
972 }
973 #elif defined(HAVE_GETHOSTNAME)
974 char cbuf[sz];
975 bool ok = gethostname(cbuf, sz) != -1;
976 if ( ok )
977 {
978 wxStrlcpy(buf, wxSafeConvertMB2WX(cbuf), sz);
979 }
980 #else // no uname, no gethostname
981 wxFAIL_MSG(wxT("don't know host name for this machine"));
982
983 bool ok = false;
984 #endif // uname/gethostname
985
986 if ( !ok )
987 {
988 wxLogSysError(_("Cannot get the hostname"));
989 }
990
991 return ok;
992 }
993
994 bool wxGetHostName(wxChar *buf, int sz)
995 {
996 bool ok = wxGetHostNameInternal(buf, sz);
997
998 if ( ok )
999 {
1000 // BSD systems return the FQDN, we only want the hostname, so extract
1001 // it (we consider that dots are domain separators)
1002 wxChar *dot = wxStrchr(buf, wxT('.'));
1003 if ( dot )
1004 {
1005 // nuke it
1006 *dot = wxT('\0');
1007 }
1008 }
1009
1010 return ok;
1011 }
1012
1013 bool wxGetFullHostName(wxChar *buf, int sz)
1014 {
1015 bool ok = wxGetHostNameInternal(buf, sz);
1016
1017 if ( ok )
1018 {
1019 if ( !wxStrchr(buf, wxT('.')) )
1020 {
1021 struct hostent *host = gethostbyname(wxSafeConvertWX2MB(buf));
1022 if ( !host )
1023 {
1024 wxLogSysError(_("Cannot get the official hostname"));
1025
1026 ok = false;
1027 }
1028 else
1029 {
1030 // the canonical name
1031 wxStrlcpy(buf, wxSafeConvertMB2WX(host->h_name), sz);
1032 }
1033 }
1034 //else: it's already a FQDN (BSD behaves this way)
1035 }
1036
1037 return ok;
1038 }
1039
1040 bool wxGetUserId(wxChar *buf, int sz)
1041 {
1042 struct passwd *who;
1043
1044 *buf = wxT('\0');
1045 if ((who = getpwuid(getuid ())) != NULL)
1046 {
1047 wxStrlcpy (buf, wxSafeConvertMB2WX(who->pw_name), sz);
1048 return true;
1049 }
1050
1051 return false;
1052 }
1053
1054 bool wxGetUserName(wxChar *buf, int sz)
1055 {
1056 #ifdef HAVE_PW_GECOS
1057 struct passwd *who;
1058
1059 *buf = wxT('\0');
1060 if ((who = getpwuid (getuid ())) != NULL)
1061 {
1062 char *comma = strchr(who->pw_gecos, ',');
1063 if (comma)
1064 *comma = '\0'; // cut off non-name comment fields
1065 wxStrlcpy(buf, wxSafeConvertMB2WX(who->pw_gecos), sz);
1066 return true;
1067 }
1068
1069 return false;
1070 #else // !HAVE_PW_GECOS
1071 return wxGetUserId(buf, sz);
1072 #endif // HAVE_PW_GECOS/!HAVE_PW_GECOS
1073 }
1074
1075 bool wxIsPlatform64Bit()
1076 {
1077 const wxString machine = wxGetCommandOutput(wxT("uname -m"));
1078
1079 // the test for "64" is obviously not 100% reliable but seems to work fine
1080 // in practice
1081 return machine.Contains(wxT("64")) ||
1082 machine.Contains(wxT("alpha"));
1083 }
1084
1085 #ifdef __LINUX__
1086
1087 static bool
1088 wxGetValueFromLSBRelease(wxString arg, const wxString& lhs, wxString* rhs)
1089 {
1090 // lsb_release seems to just read a global file which is always in UTF-8
1091 // and hence its output is always in UTF-8 as well, regardless of the
1092 // locale currently configured by our environment.
1093 return wxGetCommandOutput(wxS("lsb_release ") + arg, wxConvUTF8)
1094 .StartsWith(lhs, rhs);
1095 }
1096
1097 wxLinuxDistributionInfo wxGetLinuxDistributionInfo()
1098 {
1099 wxLinuxDistributionInfo ret;
1100
1101 if ( !wxGetValueFromLSBRelease(wxS("--id"), wxS("Distributor ID:\t"),
1102 &ret.Id) )
1103 {
1104 // Don't bother to continue, lsb_release is probably not available.
1105 return ret;
1106 }
1107
1108 wxGetValueFromLSBRelease(wxS("--description"), wxS("Description:\t"),
1109 &ret.Description);
1110 wxGetValueFromLSBRelease(wxS("--release"), wxS("Release:\t"),
1111 &ret.Release);
1112 wxGetValueFromLSBRelease(wxS("--codename"), wxS("Codename:\t"),
1113 &ret.CodeName);
1114
1115 return ret;
1116 }
1117 #endif // __LINUX__
1118
1119 // these functions are in src/osx/utilsexc_base.cpp for wxMac
1120 #ifndef __DARWIN__
1121
1122 wxOperatingSystemId wxGetOsVersion(int *verMaj, int *verMin)
1123 {
1124 // get OS version
1125 int major, minor;
1126 wxString release = wxGetCommandOutput(wxT("uname -r"));
1127 if ( release.empty() ||
1128 wxSscanf(release.c_str(), wxT("%d.%d"), &major, &minor) != 2 )
1129 {
1130 // failed to get version string or unrecognized format
1131 major =
1132 minor = -1;
1133 }
1134
1135 if ( verMaj )
1136 *verMaj = major;
1137 if ( verMin )
1138 *verMin = minor;
1139
1140 // try to understand which OS are we running
1141 wxString kernel = wxGetCommandOutput(wxT("uname -s"));
1142 if ( kernel.empty() )
1143 kernel = wxGetCommandOutput(wxT("uname -o"));
1144
1145 if ( kernel.empty() )
1146 return wxOS_UNKNOWN;
1147
1148 return wxPlatformInfo::GetOperatingSystemId(kernel);
1149 }
1150
1151 wxString wxGetOsDescription()
1152 {
1153 return wxGetCommandOutput(wxT("uname -s -r -m"));
1154 }
1155
1156 #endif // !__DARWIN__
1157
1158 unsigned long wxGetProcessId()
1159 {
1160 return (unsigned long)getpid();
1161 }
1162
1163 wxMemorySize wxGetFreeMemory()
1164 {
1165 #if defined(__LINUX__)
1166 // get it from /proc/meminfo
1167 FILE *fp = fopen("/proc/meminfo", "r");
1168 if ( fp )
1169 {
1170 long memFree = -1;
1171
1172 char buf[1024];
1173 if ( fgets(buf, WXSIZEOF(buf), fp) && fgets(buf, WXSIZEOF(buf), fp) )
1174 {
1175 // /proc/meminfo changed its format in kernel 2.6
1176 if ( wxPlatformInfo().CheckOSVersion(2, 6) )
1177 {
1178 unsigned long cached, buffers;
1179 sscanf(buf, "MemFree: %ld", &memFree);
1180
1181 fgets(buf, WXSIZEOF(buf), fp);
1182 sscanf(buf, "Buffers: %lu", &buffers);
1183
1184 fgets(buf, WXSIZEOF(buf), fp);
1185 sscanf(buf, "Cached: %lu", &cached);
1186
1187 // add to "MemFree" also the "Buffers" and "Cached" values as
1188 // free(1) does as otherwise the value never makes sense: for
1189 // kernel 2.6 it's always almost 0
1190 memFree += buffers + cached;
1191
1192 // values here are always expressed in kB and we want bytes
1193 memFree *= 1024;
1194 }
1195 else // Linux 2.4 (or < 2.6, anyhow)
1196 {
1197 long memTotal, memUsed;
1198 sscanf(buf, "Mem: %ld %ld %ld", &memTotal, &memUsed, &memFree);
1199 }
1200 }
1201
1202 fclose(fp);
1203
1204 return (wxMemorySize)memFree;
1205 }
1206 #elif defined(__SGI__)
1207 struct rminfo realmem;
1208 if ( sysmp(MP_SAGET, MPSA_RMINFO, &realmem, sizeof realmem) == 0 )
1209 return ((wxMemorySize)realmem.physmem * sysconf(_SC_PAGESIZE));
1210 #elif defined(_SC_AVPHYS_PAGES)
1211 return ((wxMemorySize)sysconf(_SC_AVPHYS_PAGES))*sysconf(_SC_PAGESIZE);
1212 //#elif defined(__FREEBSD__) -- might use sysctl() to find it out, probably
1213 #endif
1214
1215 // can't find it out
1216 return -1;
1217 }
1218
1219 bool wxGetDiskSpace(const wxString& path, wxDiskspaceSize_t *pTotal, wxDiskspaceSize_t *pFree)
1220 {
1221 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
1222 // the case to "char *" is needed for AIX 4.3
1223 wxStatfs_t fs;
1224 if ( wxStatfs((char *)(const char*)path.fn_str(), &fs) != 0 )
1225 {
1226 wxLogSysError( wxT("Failed to get file system statistics") );
1227
1228 return false;
1229 }
1230
1231 // under Solaris we also have to use f_frsize field instead of f_bsize
1232 // which is in general a multiple of f_frsize
1233 #ifdef HAVE_STATVFS
1234 wxDiskspaceSize_t blockSize = fs.f_frsize;
1235 #else // HAVE_STATFS
1236 wxDiskspaceSize_t blockSize = fs.f_bsize;
1237 #endif // HAVE_STATVFS/HAVE_STATFS
1238
1239 if ( pTotal )
1240 {
1241 *pTotal = wxDiskspaceSize_t(fs.f_blocks) * blockSize;
1242 }
1243
1244 if ( pFree )
1245 {
1246 *pFree = wxDiskspaceSize_t(fs.f_bavail) * blockSize;
1247 }
1248
1249 return true;
1250 #else // !HAVE_STATFS && !HAVE_STATVFS
1251 return false;
1252 #endif // HAVE_STATFS
1253 }
1254
1255 // ----------------------------------------------------------------------------
1256 // env vars
1257 // ----------------------------------------------------------------------------
1258
1259 #if USE_PUTENV
1260
1261 WX_DECLARE_STRING_HASH_MAP(char *, wxEnvVars);
1262
1263 static wxEnvVars gs_envVars;
1264
1265 class wxSetEnvModule : public wxModule
1266 {
1267 public:
1268 virtual bool OnInit() { return true; }
1269 virtual void OnExit()
1270 {
1271 for ( wxEnvVars::const_iterator i = gs_envVars.begin();
1272 i != gs_envVars.end();
1273 ++i )
1274 {
1275 free(i->second);
1276 }
1277
1278 gs_envVars.clear();
1279 }
1280
1281 DECLARE_DYNAMIC_CLASS(wxSetEnvModule)
1282 };
1283
1284 IMPLEMENT_DYNAMIC_CLASS(wxSetEnvModule, wxModule)
1285
1286 #endif // USE_PUTENV
1287
1288 bool wxGetEnv(const wxString& var, wxString *value)
1289 {
1290 // wxGetenv is defined as getenv()
1291 char *p = wxGetenv(var);
1292 if ( !p )
1293 return false;
1294
1295 if ( value )
1296 {
1297 *value = p;
1298 }
1299
1300 return true;
1301 }
1302
1303 static bool wxDoSetEnv(const wxString& variable, const char *value)
1304 {
1305 #if defined(HAVE_SETENV)
1306 if ( !value )
1307 {
1308 #ifdef HAVE_UNSETENV
1309 // don't test unsetenv() return value: it's void on some systems (at
1310 // least Darwin)
1311 unsetenv(variable.mb_str());
1312 return true;
1313 #else
1314 value = ""; // we can't pass NULL to setenv()
1315 #endif
1316 }
1317
1318 return setenv(variable.mb_str(), value, 1 /* overwrite */) == 0;
1319 #elif defined(HAVE_PUTENV)
1320 wxString s = variable;
1321 if ( value )
1322 s << wxT('=') << value;
1323
1324 // transform to ANSI
1325 const wxWX2MBbuf p = s.mb_str();
1326
1327 char *buf = (char *)malloc(strlen(p) + 1);
1328 strcpy(buf, p);
1329
1330 // store the string to free() it later
1331 wxEnvVars::iterator i = gs_envVars.find(variable);
1332 if ( i != gs_envVars.end() )
1333 {
1334 free(i->second);
1335 i->second = buf;
1336 }
1337 else // this variable hadn't been set before
1338 {
1339 gs_envVars[variable] = buf;
1340 }
1341
1342 return putenv(buf) == 0;
1343 #else // no way to set an env var
1344 return false;
1345 #endif
1346 }
1347
1348 bool wxSetEnv(const wxString& variable, const wxString& value)
1349 {
1350 return wxDoSetEnv(variable, value.mb_str());
1351 }
1352
1353 bool wxUnsetEnv(const wxString& variable)
1354 {
1355 return wxDoSetEnv(variable, NULL);
1356 }
1357
1358 // ----------------------------------------------------------------------------
1359 // signal handling
1360 // ----------------------------------------------------------------------------
1361
1362 #if wxUSE_ON_FATAL_EXCEPTION
1363
1364 #include <signal.h>
1365
1366 extern "C" void wxFatalSignalHandler(wxTYPE_SA_HANDLER)
1367 {
1368 if ( wxTheApp )
1369 {
1370 // give the user a chance to do something special about this
1371 wxTheApp->OnFatalException();
1372 }
1373
1374 abort();
1375 }
1376
1377 bool wxHandleFatalExceptions(bool doit)
1378 {
1379 // old sig handlers
1380 static bool s_savedHandlers = false;
1381 static struct sigaction s_handlerFPE,
1382 s_handlerILL,
1383 s_handlerBUS,
1384 s_handlerSEGV;
1385
1386 bool ok = true;
1387 if ( doit && !s_savedHandlers )
1388 {
1389 // install the signal handler
1390 struct sigaction act;
1391
1392 // some systems extend it with non std fields, so zero everything
1393 memset(&act, 0, sizeof(act));
1394
1395 act.sa_handler = wxFatalSignalHandler;
1396 sigemptyset(&act.sa_mask);
1397 act.sa_flags = 0;
1398
1399 ok &= sigaction(SIGFPE, &act, &s_handlerFPE) == 0;
1400 ok &= sigaction(SIGILL, &act, &s_handlerILL) == 0;
1401 ok &= sigaction(SIGBUS, &act, &s_handlerBUS) == 0;
1402 ok &= sigaction(SIGSEGV, &act, &s_handlerSEGV) == 0;
1403 if ( !ok )
1404 {
1405 wxLogDebug(wxT("Failed to install our signal handler."));
1406 }
1407
1408 s_savedHandlers = true;
1409 }
1410 else if ( s_savedHandlers )
1411 {
1412 // uninstall the signal handler
1413 ok &= sigaction(SIGFPE, &s_handlerFPE, NULL) == 0;
1414 ok &= sigaction(SIGILL, &s_handlerILL, NULL) == 0;
1415 ok &= sigaction(SIGBUS, &s_handlerBUS, NULL) == 0;
1416 ok &= sigaction(SIGSEGV, &s_handlerSEGV, NULL) == 0;
1417 if ( !ok )
1418 {
1419 wxLogDebug(wxT("Failed to uninstall our signal handler."));
1420 }
1421
1422 s_savedHandlers = false;
1423 }
1424 //else: nothing to do
1425
1426 return ok;
1427 }
1428
1429 #endif // wxUSE_ON_FATAL_EXCEPTION
1430
1431 // ----------------------------------------------------------------------------
1432 // wxExecute support
1433 // ----------------------------------------------------------------------------
1434
1435 int wxAppTraits::WaitForChild(wxExecuteData& execData)
1436 {
1437 #if wxUSE_CONSOLE_EVENTLOOP
1438 wxConsoleEventLoop loop;
1439 return RunLoopUntilChildExit(execData, loop);
1440 #else // !wxUSE_CONSOLE_EVENTLOOP
1441 wxFAIL_MSG( wxS("Can't wait for child process without wxConsoleEventLoop") );
1442
1443 return -1;
1444 #endif // wxUSE_CONSOLE_EVENTLOOP/!wxUSE_CONSOLE_EVENTLOOP
1445 }
1446
1447 // This function is common code for both console and GUI applications and used
1448 // by wxExecute() to wait for the child exit while dispatching events.
1449 //
1450 // Notice that it should not be used for all the other cases, e.g. when we
1451 // don't need to wait for the child (wxEXEC_ASYNC) or when the events must not
1452 // dispatched (wxEXEC_NOEVENTS).
1453 int
1454 wxAppTraits::RunLoopUntilChildExit(wxExecuteData& execData,
1455 wxEventLoopBase& loop)
1456 {
1457 // It is possible that wxExecuteData::OnExit() had already been called
1458 // and reset the PID to 0, in which case we don't need to do anything
1459 // at all.
1460 if ( !execData.pid )
1461 return execData.exitcode;
1462
1463 #if wxUSE_STREAMS
1464 // Monitor the child streams if necessary.
1465 wxScopedPtr<wxEventLoopSourceHandler>
1466 stdoutHandler,
1467 stderrHandler;
1468 if ( execData.IsRedirected() )
1469 {
1470 stdoutHandler.reset(new wxExecuteEventLoopSourceHandler
1471 (
1472 execData.fdOut, execData.bufOut
1473 ));
1474 stderrHandler.reset(new wxExecuteEventLoopSourceHandler
1475 (
1476 execData.fdErr, execData.bufErr
1477 ));
1478 }
1479 #endif // wxUSE_STREAMS
1480
1481 // Store the event loop in the data associated with the child
1482 // process so that it could exit the loop when the child exits.
1483 execData.syncEventLoop = &loop;
1484
1485 // And run it.
1486 loop.Run();
1487
1488 // The exit code will have been set when the child termination was detected.
1489 return execData.exitcode;
1490 }
1491
1492 // ----------------------------------------------------------------------------
1493 // wxExecuteData
1494 // ----------------------------------------------------------------------------
1495
1496 namespace
1497 {
1498
1499 // Helper function that checks whether the child with the given PID has exited
1500 // and fills the provided parameter with its return code if it did.
1501 bool CheckForChildExit(int pid, int* exitcodeOut)
1502 {
1503 wxASSERT_MSG( pid > 0, "invalid PID" );
1504
1505 int status, rc;
1506
1507 // loop while we're getting EINTR
1508 for ( ;; )
1509 {
1510 rc = waitpid(pid, &status, WNOHANG);
1511
1512 if ( rc != -1 || errno != EINTR )
1513 break;
1514 }
1515
1516 switch ( rc )
1517 {
1518 case 0:
1519 // No error but the child is still running.
1520 return false;
1521
1522 case -1:
1523 // Checking child status failed. Invalid PID?
1524 wxLogLastError(wxString::Format("waitpid(%d)", pid));
1525 return false;
1526
1527 default:
1528 // Child did terminate.
1529 wxASSERT_MSG( rc == pid, "unexpected waitpid() return value" );
1530
1531 // notice that the caller expects the exit code to be signed, e.g. -1
1532 // instead of 255 so don't assign WEXITSTATUS() to an int
1533 signed char exitcode;
1534 if ( WIFEXITED(status) )
1535 exitcode = WEXITSTATUS(status);
1536 else if ( WIFSIGNALED(status) )
1537 exitcode = -WTERMSIG(status);
1538 else
1539 {
1540 wxLogError("Child process (PID %d) exited for unknown reason, "
1541 "status = %d", pid, status);
1542 exitcode = -1;
1543 }
1544
1545 if ( exitcodeOut )
1546 *exitcodeOut = exitcode;
1547
1548 return true;
1549 }
1550 }
1551
1552 } // anonymous namespace
1553
1554 wxExecuteData::ChildProcessesData wxExecuteData::ms_childProcesses;
1555
1556 /* static */
1557 void wxExecuteData::OnSomeChildExited(int WXUNUSED(sig))
1558 {
1559 // We know that some child process has terminated, but we don't know which
1560 // one, so check all of them (notice that more than one could have exited).
1561 //
1562 // An alternative approach would be to call waitpid(-1, &status, WNOHANG)
1563 // (in a loop to take care of the multiple children exiting case) and
1564 // perhaps this would be more efficient. But for now this seems to work.
1565
1566
1567 // Make a copy of the list before iterating over it to avoid problems due
1568 // to deleting entries from it in the process.
1569 const ChildProcessesData allChildProcesses = ms_childProcesses;
1570 for ( ChildProcessesData::const_iterator it = allChildProcesses.begin();
1571 it != allChildProcesses.end();
1572 ++it )
1573 {
1574 const int pid = it->first;
1575
1576 // Check whether this child exited.
1577 int exitcode;
1578 if ( !CheckForChildExit(pid, &exitcode) )
1579 continue;
1580
1581 // And handle its termination if it did.
1582 //
1583 // Notice that this will implicitly remove it from ms_childProcesses.
1584 it->second->OnExit(exitcode);
1585 }
1586 }
1587
1588 void wxExecuteData::OnStart(int pid_)
1589 {
1590 wxCHECK_RET( wxTheApp,
1591 wxS("Ensure wxTheApp is set before calling wxExecute()") );
1592
1593 // Setup the signal handler for SIGCHLD to be able to detect the child
1594 // termination.
1595 //
1596 // Notice that SetSignalHandler() is idempotent, so it's fine to call
1597 // it more than once with the same handler.
1598 wxTheApp->SetSignalHandler(SIGCHLD, OnSomeChildExited);
1599
1600
1601 // Remember the child PID to be able to wait for it later.
1602 pid = pid_;
1603
1604 // Also save it in wxProcess where it will be accessible to the user code.
1605 if ( process )
1606 process->SetPid(pid);
1607
1608 // Finally, add this object itself to the list of child processes so that
1609 // we can check for its termination the next time we get SIGCHLD.
1610 ms_childProcesses[pid] = this;
1611 }
1612
1613 void wxExecuteData::OnExit(int exitcode_)
1614 {
1615 // Remove this process from the hash list of child processes that are
1616 // still open as soon as possible to ensure we don't process it again even
1617 // if another SIGCHLD happens.
1618 if ( !ms_childProcesses.erase(pid) )
1619 {
1620 wxFAIL_MSG(wxString::Format(wxS("Data for PID %d not in the list?"), pid));
1621 }
1622
1623
1624 exitcode = exitcode_;
1625
1626 #if wxUSE_STREAMS
1627 if ( IsRedirected() )
1628 {
1629 // Read the remaining data in a blocking way: this is fine because the
1630 // child has already exited and hence all the data must be already
1631 // available in the streams buffers.
1632 bufOut.ReadAll();
1633 bufErr.ReadAll();
1634 }
1635 #endif // wxUSE_STREAMS
1636
1637 // Notify user about termination if required
1638 if ( !(flags & wxEXEC_SYNC) )
1639 {
1640 if ( process )
1641 process->OnTerminate(pid, exitcode);
1642
1643 // in case of asynchronous execution we don't need this object any more
1644 // after the child terminates
1645 delete this;
1646 }
1647 else // sync execution
1648 {
1649 // let wxExecute() know that the process has terminated
1650 pid = 0;
1651
1652 // Stop the event loop for synchronous wxExecute() if we're running one.
1653 if ( syncEventLoop )
1654 syncEventLoop->ScheduleExit();
1655 }
1656 }