fixed IO redirection to work in wxExecute() in console applications (including the...
[wxWidgets.git] / src / unix / utilsunx.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/unix/utilsunx.cpp
3 // Purpose: generic Unix implementation of many wx functions
4 // Author: Vadim Zeitlin
5 // Id: $Id$
6 // Copyright: (c) 1998 Robert Roebling, 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/thread.h"
41
42 #include "wx/wfstream.h"
43
44 #include "wx/private/selectdispatcher.h"
45 #include "wx/private/fdiodispatcher.h"
46 #include "wx/unix/execute.h"
47 #include "wx/unix/private.h"
48
49 #ifdef wxHAS_GENERIC_PROCESS_CALLBACK
50 #include "wx/private/fdiodispatcher.h"
51 #endif
52
53 #include <pwd.h>
54 #include <sys/wait.h> // waitpid()
55
56 #ifdef HAVE_SYS_SELECT_H
57 # include <sys/select.h>
58 #endif
59
60 #define HAS_PIPE_INPUT_STREAM (wxUSE_STREAMS && wxUSE_FILE)
61
62 #if HAS_PIPE_INPUT_STREAM
63
64 // define this to let wxexec.cpp know that we know what we're doing
65 #define _WX_USED_BY_WXEXECUTE_
66 #include "../common/execcmn.cpp"
67
68 #endif // HAS_PIPE_INPUT_STREAM
69
70 #if defined(__MWERKS__) && defined(__MACH__)
71 #ifndef WXWIN_OS_DESCRIPTION
72 #define WXWIN_OS_DESCRIPTION "MacOS X"
73 #endif
74 #ifndef HAVE_NANOSLEEP
75 #define HAVE_NANOSLEEP
76 #endif
77 #ifndef HAVE_UNAME
78 #define HAVE_UNAME
79 #endif
80
81 // our configure test believes we can use sigaction() if the function is
82 // available but Metrowekrs with MSL run-time does have the function but
83 // doesn't have sigaction struct so finally we can't use it...
84 #ifdef __MSL__
85 #undef wxUSE_ON_FATAL_EXCEPTION
86 #define wxUSE_ON_FATAL_EXCEPTION 0
87 #endif
88 #endif
89
90 // not only the statfs syscall is called differently depending on platform, but
91 // one of its incarnations, statvfs(), takes different arguments under
92 // different platforms and even different versions of the same system (Solaris
93 // 7 and 8): if you want to test for this, don't forget that the problems only
94 // appear if the large files support is enabled
95 #ifdef HAVE_STATFS
96 #ifdef __BSD__
97 #include <sys/param.h>
98 #include <sys/mount.h>
99 #else // !__BSD__
100 #include <sys/vfs.h>
101 #endif // __BSD__/!__BSD__
102
103 #define wxStatfs statfs
104
105 #ifndef HAVE_STATFS_DECL
106 // some systems lack statfs() prototype in the system headers (AIX 4)
107 extern "C" int statfs(const char *path, struct statfs *buf);
108 #endif
109 #endif // HAVE_STATFS
110
111 #ifdef HAVE_STATVFS
112 #include <sys/statvfs.h>
113
114 #define wxStatfs statvfs
115 #endif // HAVE_STATVFS
116
117 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
118 // WX_STATFS_T is detected by configure
119 #define wxStatfs_t WX_STATFS_T
120 #endif
121
122 // SGI signal.h defines signal handler arguments differently depending on
123 // whether _LANGUAGE_C_PLUS_PLUS is set or not - do set it
124 #if defined(__SGI__) && !defined(_LANGUAGE_C_PLUS_PLUS)
125 #define _LANGUAGE_C_PLUS_PLUS 1
126 #endif // SGI hack
127
128 #include <stdarg.h>
129 #include <dirent.h>
130 #include <string.h>
131 #include <sys/stat.h>
132 #include <sys/types.h>
133 #include <sys/wait.h>
134 #include <unistd.h>
135 #include <errno.h>
136 #include <netdb.h>
137 #include <signal.h>
138 #include <fcntl.h> // for O_WRONLY and friends
139 #include <time.h> // nanosleep() and/or usleep()
140 #include <ctype.h> // isspace()
141 #include <sys/time.h> // needed for FD_SETSIZE
142
143 #ifdef HAVE_UNAME
144 #include <sys/utsname.h> // for uname()
145 #endif // HAVE_UNAME
146
147 // Used by wxGetFreeMemory().
148 #ifdef __SGI__
149 #include <sys/sysmp.h>
150 #include <sys/sysinfo.h> // for SAGET and MINFO structures
151 #endif
152
153 // ----------------------------------------------------------------------------
154 // conditional compilation
155 // ----------------------------------------------------------------------------
156
157 // many versions of Unices have this function, but it is not defined in system
158 // headers - please add your system here if it is the case for your OS.
159 // SunOS < 5.6 (i.e. Solaris < 2.6) and DG-UX are like this.
160 #if !defined(HAVE_USLEEP) && \
161 ((defined(__SUN__) && !defined(__SunOs_5_6) && \
162 !defined(__SunOs_5_7) && !defined(__SUNPRO_CC)) || \
163 defined(__osf__) || defined(__EMX__))
164 extern "C"
165 {
166 #ifdef __EMX__
167 /* I copied this from the XFree86 diffs. AV. */
168 #define INCL_DOSPROCESS
169 #include <os2.h>
170 inline void usleep(unsigned long delay)
171 {
172 DosSleep(delay ? (delay/1000l) : 1l);
173 }
174 #else // Unix
175 int usleep(unsigned int usec);
176 #endif // __EMX__/Unix
177 };
178
179 #define HAVE_USLEEP 1
180 #endif // Unices without usleep()
181
182 // ============================================================================
183 // implementation
184 // ============================================================================
185
186 // ----------------------------------------------------------------------------
187 // sleeping
188 // ----------------------------------------------------------------------------
189
190 void wxSleep(int nSecs)
191 {
192 sleep(nSecs);
193 }
194
195 void wxMicroSleep(unsigned long microseconds)
196 {
197 #if defined(HAVE_NANOSLEEP)
198 timespec tmReq;
199 tmReq.tv_sec = (time_t)(microseconds / 1000000);
200 tmReq.tv_nsec = (microseconds % 1000000) * 1000;
201
202 // we're not interested in remaining time nor in return value
203 (void)nanosleep(&tmReq, (timespec *)NULL);
204 #elif defined(HAVE_USLEEP)
205 // uncomment this if you feel brave or if you are sure that your version
206 // of Solaris has a safe usleep() function but please notice that usleep()
207 // is known to lead to crashes in MT programs in Solaris 2.[67] and is not
208 // documented as MT-Safe
209 #if defined(__SUN__) && wxUSE_THREADS
210 #error "usleep() cannot be used in MT programs under Solaris."
211 #endif // Sun
212
213 usleep(microseconds);
214 #elif defined(HAVE_SLEEP)
215 // under BeOS sleep() takes seconds (what about other platforms, if any?)
216 sleep(microseconds * 1000000);
217 #else // !sleep function
218 #error "usleep() or nanosleep() function required for wxMicroSleep"
219 #endif // sleep function
220 }
221
222 void wxMilliSleep(unsigned long milliseconds)
223 {
224 wxMicroSleep(milliseconds*1000);
225 }
226
227 // ----------------------------------------------------------------------------
228 // process management
229 // ----------------------------------------------------------------------------
230
231 int wxKill(long pid, wxSignal sig, wxKillError *rc, int flags)
232 {
233 int err = kill((pid_t) (flags & wxKILL_CHILDREN) ? -pid : pid, (int)sig);
234 if ( rc )
235 {
236 switch ( err ? errno : 0 )
237 {
238 case 0:
239 *rc = wxKILL_OK;
240 break;
241
242 case EINVAL:
243 *rc = wxKILL_BAD_SIGNAL;
244 break;
245
246 case EPERM:
247 *rc = wxKILL_ACCESS_DENIED;
248 break;
249
250 case ESRCH:
251 *rc = wxKILL_NO_PROCESS;
252 break;
253
254 default:
255 // this goes against Unix98 docs so log it
256 wxLogDebug(_T("unexpected kill(2) return value %d"), err);
257
258 // something else...
259 *rc = wxKILL_ERROR;
260 }
261 }
262
263 return err;
264 }
265
266 #define WXEXECUTE_NARGS 127
267
268 #if defined(__DARWIN__)
269 long wxMacExecute(wxChar **argv,
270 int flags,
271 wxProcess *process);
272 #endif
273
274 long wxExecute( const wxString& command, int flags, wxProcess *process )
275 {
276 wxCHECK_MSG( !command.empty(), 0, wxT("can't exec empty command") );
277
278 wxLogTrace(wxT("exec"), wxT("Executing \"%s\""), command.c_str());
279
280 #if wxUSE_THREADS
281 // fork() doesn't mix well with POSIX threads: on many systems the program
282 // deadlocks or crashes for some reason. Probably our code is buggy and
283 // doesn't do something which must be done to allow this to work, but I
284 // don't know what yet, so for now just warn the user (this is the least we
285 // can do) about it
286 wxASSERT_MSG( wxThread::IsMain(),
287 _T("wxExecute() can be called only from the main thread") );
288 #endif // wxUSE_THREADS
289
290 int argc = 0;
291 wxChar *argv[WXEXECUTE_NARGS];
292 wxString argument;
293 const wxChar *cptr = command.c_str();
294 wxChar quotechar = wxT('\0'); // is arg quoted?
295 bool escaped = false;
296
297 // split the command line in arguments
298 do
299 {
300 argument = wxEmptyString;
301 quotechar = wxT('\0');
302
303 // eat leading whitespace:
304 while ( wxIsspace(*cptr) )
305 cptr++;
306
307 if ( *cptr == wxT('\'') || *cptr == wxT('"') )
308 quotechar = *cptr++;
309
310 do
311 {
312 if ( *cptr == wxT('\\') && ! escaped )
313 {
314 escaped = true;
315 cptr++;
316 continue;
317 }
318
319 // all other characters:
320 argument += *cptr++;
321 escaped = false;
322
323 // have we reached the end of the argument?
324 if ( (*cptr == quotechar && ! escaped)
325 || (quotechar == wxT('\0') && wxIsspace(*cptr))
326 || *cptr == wxT('\0') )
327 {
328 wxASSERT_MSG( argc < WXEXECUTE_NARGS,
329 wxT("too many arguments in wxExecute") );
330
331 argv[argc] = new wxChar[argument.length() + 1];
332 wxStrcpy(argv[argc], argument.c_str());
333 argc++;
334
335 // if not at end of buffer, swallow last character:
336 if(*cptr)
337 cptr++;
338
339 break; // done with this one, start over
340 }
341 } while(*cptr);
342 } while(*cptr);
343 argv[argc] = NULL;
344
345 long lRc;
346 #if defined(__DARWIN__)
347 // wxMacExecute only executes app bundles.
348 // It returns an error code if the target is not an app bundle, thus falling
349 // through to the regular wxExecute for non app bundles.
350 lRc = wxMacExecute(argv, flags, process);
351 if( lRc != ((flags & wxEXEC_SYNC) ? -1 : 0))
352 return lRc;
353 #endif
354
355 // do execute the command
356 lRc = wxExecute(argv, flags, process);
357
358 // clean up
359 argc = 0;
360 while( argv[argc] )
361 delete [] argv[argc++];
362
363 return lRc;
364 }
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 = _T("xterm");
377 }
378 else
379 {
380 // execute command in a shell
381 cmd << _T("/bin/sh -c '") << command << _T('\'');
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, _T("can't exec shell non interactively") );
395
396 return wxExecute(wxMakeShellCommand(command), output);
397 }
398
399 // Shutdown or reboot the PC
400 bool wxShutdown(wxShutdownFlags wFlags)
401 {
402 wxChar level;
403 switch ( wFlags )
404 {
405 case wxSHUTDOWN_POWEROFF:
406 level = _T('0');
407 break;
408
409 case wxSHUTDOWN_REBOOT:
410 level = _T('6');
411 break;
412
413 default:
414 wxFAIL_MSG( _T("unknown wxShutdown() flag") );
415 return false;
416 }
417
418 return system(wxString::Format(_T("init %c"), level).mb_str()) == 0;
419 }
420
421 // ----------------------------------------------------------------------------
422 // wxStream classes to support IO redirection in wxExecute
423 // ----------------------------------------------------------------------------
424
425 #if HAS_PIPE_INPUT_STREAM
426
427 bool wxPipeInputStream::CanRead() const
428 {
429 if ( m_lasterror == wxSTREAM_EOF )
430 return false;
431
432 // check if there is any input available
433 struct timeval tv;
434 tv.tv_sec = 0;
435 tv.tv_usec = 0;
436
437 const int fd = m_file->fd();
438
439 fd_set readfds;
440
441 wxFD_ZERO(&readfds);
442 wxFD_SET(fd, &readfds);
443
444 switch ( select(fd + 1, &readfds, NULL, NULL, &tv) )
445 {
446 case -1:
447 wxLogSysError(_("Impossible to get child process input"));
448 // fall through
449
450 case 0:
451 return false;
452
453 default:
454 wxFAIL_MSG(_T("unexpected select() return value"));
455 // still fall through
456
457 case 1:
458 // input available -- or maybe not, as select() returns 1 when a
459 // read() will complete without delay, but it could still not read
460 // anything
461 return !Eof();
462 }
463 }
464
465 #endif // HAS_PIPE_INPUT_STREAM
466
467 // ----------------------------------------------------------------------------
468 // wxExecute: the real worker function
469 // ----------------------------------------------------------------------------
470
471 long wxExecute(wxChar **argv, int flags, wxProcess *process)
472 {
473 // for the sync execution, we return -1 to indicate failure, but for async
474 // case we return 0 which is never a valid PID
475 //
476 // we define this as a macro, not a variable, to avoid compiler warnings
477 // about "ERROR_RETURN_CODE value may be clobbered by fork()"
478 #define ERROR_RETURN_CODE ((flags & wxEXEC_SYNC) ? -1 : 0)
479
480 wxCHECK_MSG( *argv, ERROR_RETURN_CODE, wxT("can't exec empty command") );
481
482 #if wxUSE_UNICODE
483 int mb_argc = 0;
484 char *mb_argv[WXEXECUTE_NARGS];
485
486 while (argv[mb_argc])
487 {
488 wxWX2MBbuf mb_arg = wxSafeConvertWX2MB(argv[mb_argc]);
489 mb_argv[mb_argc] = strdup(mb_arg);
490 mb_argc++;
491 }
492 mb_argv[mb_argc] = (char *) NULL;
493
494 // this macro will free memory we used above
495 #define ARGS_CLEANUP \
496 for ( mb_argc = 0; mb_argv[mb_argc]; mb_argc++ ) \
497 free(mb_argv[mb_argc])
498 #else // ANSI
499 // no need for cleanup
500 #define ARGS_CLEANUP
501
502 wxChar **mb_argv = argv;
503 #endif // Unicode/ANSI
504
505 // we want this function to work even if there is no wxApp so ensure that
506 // we have a valid traits pointer
507 wxConsoleAppTraits traitsConsole;
508 wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
509 if ( !traits )
510 traits = &traitsConsole;
511
512 // this struct contains all information which we pass to and from
513 // wxAppTraits methods
514 wxExecuteData execData;
515 execData.flags = flags;
516 execData.process = process;
517
518 // create pipes
519 if ( !traits->CreateEndProcessPipe(execData) )
520 {
521 wxLogError( _("Failed to execute '%s'\n"), *argv );
522
523 ARGS_CLEANUP;
524
525 return ERROR_RETURN_CODE;
526 }
527
528 // pipes for inter process communication
529 wxPipe pipeIn, // stdin
530 pipeOut, // stdout
531 pipeErr; // stderr
532
533 if ( process && process->IsRedirected() )
534 {
535 if ( !pipeIn.Create() || !pipeOut.Create() || !pipeErr.Create() )
536 {
537 wxLogError( _("Failed to execute '%s'\n"), *argv );
538
539 ARGS_CLEANUP;
540
541 return ERROR_RETURN_CODE;
542 }
543 }
544
545 // fork the process
546 //
547 // NB: do *not* use vfork() here, it completely breaks this code for some
548 // reason under Solaris (and maybe others, although not under Linux)
549 // But on OpenVMS we do not have fork so we have to use vfork and
550 // cross our fingers that it works.
551 #ifdef __VMS
552 pid_t pid = vfork();
553 #else
554 pid_t pid = fork();
555 #endif
556 if ( pid == -1 ) // error?
557 {
558 wxLogSysError( _("Fork failed") );
559
560 ARGS_CLEANUP;
561
562 return ERROR_RETURN_CODE;
563 }
564 else if ( pid == 0 ) // we're in child
565 {
566 // These lines close the open file descriptors to to avoid any
567 // input/output which might block the process or irritate the user. If
568 // one wants proper IO for the subprocess, the right thing to do is to
569 // start an xterm executing it.
570 if ( !(flags & wxEXEC_SYNC) )
571 {
572 // FD_SETSIZE is unsigned under BSD, signed under other platforms
573 // so we need a cast to avoid warnings on all platforms
574 for ( int fd = 0; fd < (int)FD_SETSIZE; fd++ )
575 {
576 if ( fd == pipeIn[wxPipe::Read]
577 || fd == pipeOut[wxPipe::Write]
578 || fd == pipeErr[wxPipe::Write]
579 || traits->IsWriteFDOfEndProcessPipe(execData, fd) )
580 {
581 // don't close this one, we still need it
582 continue;
583 }
584
585 // leave stderr opened too, it won't do any harm
586 if ( fd != STDERR_FILENO )
587 close(fd);
588 }
589 }
590
591 #if !defined(__VMS) && !defined(__EMX__)
592 if ( flags & wxEXEC_MAKE_GROUP_LEADER )
593 {
594 // Set process group to child process' pid. Then killing -pid
595 // of the parent will kill the process and all of its children.
596 setsid();
597 }
598 #endif // !__VMS
599
600 // reading side can be safely closed but we should keep the write one
601 // opened
602 traits->DetachWriteFDOfEndProcessPipe(execData);
603
604 // redirect stdin, stdout and stderr
605 if ( pipeIn.IsOk() )
606 {
607 if ( dup2(pipeIn[wxPipe::Read], STDIN_FILENO) == -1 ||
608 dup2(pipeOut[wxPipe::Write], STDOUT_FILENO) == -1 ||
609 dup2(pipeErr[wxPipe::Write], STDERR_FILENO) == -1 )
610 {
611 wxLogSysError(_("Failed to redirect child process input/output"));
612 }
613
614 pipeIn.Close();
615 pipeOut.Close();
616 pipeErr.Close();
617 }
618
619 execvp (*mb_argv, mb_argv);
620
621 fprintf(stderr, "execvp(");
622 // CS changed ppc to ppc_ as ppc is not available under mac os CW Mach-O
623 for ( char **ppc_ = mb_argv; *ppc_; ppc_++ )
624 fprintf(stderr, "%s%s", ppc_ == mb_argv ? "" : ", ", *ppc_);
625 fprintf(stderr, ") failed with error %d!\n", errno);
626
627 // there is no return after successful exec()
628 _exit(-1);
629
630 // some compilers complain about missing return - of course, they
631 // should know that exit() doesn't return but what else can we do if
632 // they don't?
633 //
634 // and, sure enough, other compilers complain about unreachable code
635 // after exit() call, so we can just always have return here...
636 #if defined(__VMS) || defined(__INTEL_COMPILER)
637 return 0;
638 #endif
639 }
640 else // we're in parent
641 {
642 ARGS_CLEANUP;
643
644 // save it for WaitForChild() use
645 execData.pid = pid;
646
647 // prepare for IO redirection
648
649 #if HAS_PIPE_INPUT_STREAM
650 // the input buffer bufOut is connected to stdout, this is why it is
651 // called bufOut and not bufIn
652 wxStreamTempInputBuffer bufOut,
653 bufErr;
654
655 if ( process && process->IsRedirected() )
656 {
657 wxOutputStream *inStream =
658 new wxFileOutputStream(pipeIn.Detach(wxPipe::Write));
659
660 const int fdOut = pipeOut.Detach(wxPipe::Read);
661 wxPipeInputStream *outStream = new wxPipeInputStream(fdOut);
662
663 const int fdErr = pipeErr.Detach(wxPipe::Read);
664 wxPipeInputStream *errStream = new wxPipeInputStream(fdErr);
665
666 process->SetPipeStreams(outStream, inStream, errStream);
667
668 bufOut.Init(outStream);
669 bufErr.Init(errStream);
670
671 execData.bufOut = &bufOut;
672 execData.bufErr = &bufErr;
673
674 execData.fdOut = fdOut;
675 execData.fdErr = fdErr;
676 }
677 #endif // HAS_PIPE_INPUT_STREAM
678
679 if ( pipeIn.IsOk() )
680 {
681 pipeIn.Close();
682 pipeOut.Close();
683 pipeErr.Close();
684 }
685
686 return traits->WaitForChild(execData);
687 }
688
689 #if !defined(__VMS) && !defined(__INTEL_COMPILER)
690 return ERROR_RETURN_CODE;
691 #endif
692 }
693
694 #undef ERROR_RETURN_CODE
695 #undef ARGS_CLEANUP
696
697 // ----------------------------------------------------------------------------
698 // file and directory functions
699 // ----------------------------------------------------------------------------
700
701 const wxChar* wxGetHomeDir( wxString *home )
702 {
703 *home = wxGetUserHome();
704 wxString tmp;
705 if ( home->empty() )
706 *home = wxT("/");
707 #ifdef __VMS
708 tmp = *home;
709 if ( tmp.Last() != wxT(']'))
710 if ( tmp.Last() != wxT('/')) *home << wxT('/');
711 #endif
712 return home->c_str();
713 }
714
715 wxString wxGetUserHome( const wxString &user )
716 {
717 struct passwd *who = (struct passwd *) NULL;
718
719 if ( !user )
720 {
721 wxChar *ptr;
722
723 if ((ptr = wxGetenv(wxT("HOME"))) != NULL)
724 {
725 return ptr;
726 }
727
728 if ((ptr = wxGetenv(wxT("USER"))) != NULL ||
729 (ptr = wxGetenv(wxT("LOGNAME"))) != NULL)
730 {
731 who = getpwnam(wxSafeConvertWX2MB(ptr));
732 }
733
734 // make sure the user exists!
735 if ( !who )
736 {
737 who = getpwuid(getuid());
738 }
739 }
740 else
741 {
742 who = getpwnam (user.mb_str());
743 }
744
745 return wxSafeConvertMB2WX(who ? who->pw_dir : 0);
746 }
747
748 // ----------------------------------------------------------------------------
749 // network and user id routines
750 // ----------------------------------------------------------------------------
751
752 // private utility function which returns output of the given command, removing
753 // the trailing newline
754 static wxString wxGetCommandOutput(const wxString &cmd)
755 {
756 FILE *f = popen(cmd.ToAscii(), "r");
757 if ( !f )
758 {
759 wxLogSysError(_T("Executing \"%s\" failed"), cmd.c_str());
760 return wxEmptyString;
761 }
762
763 wxString s;
764 char buf[256];
765 while ( !feof(f) )
766 {
767 if ( !fgets(buf, sizeof(buf), f) )
768 break;
769
770 s += wxString::FromAscii(buf);
771 }
772
773 pclose(f);
774
775 if ( !s.empty() && s.Last() == _T('\n') )
776 s.RemoveLast();
777
778 return s;
779 }
780
781 // retrieve either the hostname or FQDN depending on platform (caller must
782 // check whether it's one or the other, this is why this function is for
783 // private use only)
784 static bool wxGetHostNameInternal(wxChar *buf, int sz)
785 {
786 wxCHECK_MSG( buf, false, wxT("NULL pointer in wxGetHostNameInternal") );
787
788 *buf = wxT('\0');
789
790 // we're using uname() which is POSIX instead of less standard sysinfo()
791 #if defined(HAVE_UNAME)
792 struct utsname uts;
793 bool ok = uname(&uts) != -1;
794 if ( ok )
795 {
796 wxStrncpy(buf, wxSafeConvertMB2WX(uts.nodename), sz - 1);
797 buf[sz] = wxT('\0');
798 }
799 #elif defined(HAVE_GETHOSTNAME)
800 char cbuf[sz];
801 bool ok = gethostname(cbuf, sz) != -1;
802 if ( ok )
803 {
804 wxStrncpy(buf, wxSafeConvertMB2WX(cbuf), sz - 1);
805 buf[sz] = wxT('\0');
806 }
807 #else // no uname, no gethostname
808 wxFAIL_MSG(wxT("don't know host name for this machine"));
809
810 bool ok = false;
811 #endif // uname/gethostname
812
813 if ( !ok )
814 {
815 wxLogSysError(_("Cannot get the hostname"));
816 }
817
818 return ok;
819 }
820
821 bool wxGetHostName(wxChar *buf, int sz)
822 {
823 bool ok = wxGetHostNameInternal(buf, sz);
824
825 if ( ok )
826 {
827 // BSD systems return the FQDN, we only want the hostname, so extract
828 // it (we consider that dots are domain separators)
829 wxChar *dot = wxStrchr(buf, wxT('.'));
830 if ( dot )
831 {
832 // nuke it
833 *dot = wxT('\0');
834 }
835 }
836
837 return ok;
838 }
839
840 bool wxGetFullHostName(wxChar *buf, int sz)
841 {
842 bool ok = wxGetHostNameInternal(buf, sz);
843
844 if ( ok )
845 {
846 if ( !wxStrchr(buf, wxT('.')) )
847 {
848 struct hostent *host = gethostbyname(wxSafeConvertWX2MB(buf));
849 if ( !host )
850 {
851 wxLogSysError(_("Cannot get the official hostname"));
852
853 ok = false;
854 }
855 else
856 {
857 // the canonical name
858 wxStrncpy(buf, wxSafeConvertMB2WX(host->h_name), sz);
859 }
860 }
861 //else: it's already a FQDN (BSD behaves this way)
862 }
863
864 return ok;
865 }
866
867 bool wxGetUserId(wxChar *buf, int sz)
868 {
869 struct passwd *who;
870
871 *buf = wxT('\0');
872 if ((who = getpwuid(getuid ())) != NULL)
873 {
874 wxStrncpy (buf, wxSafeConvertMB2WX(who->pw_name), sz - 1);
875 return true;
876 }
877
878 return false;
879 }
880
881 bool wxGetUserName(wxChar *buf, int sz)
882 {
883 #ifdef HAVE_PW_GECOS
884 struct passwd *who;
885
886 *buf = wxT('\0');
887 if ((who = getpwuid (getuid ())) != NULL)
888 {
889 char *comma = strchr(who->pw_gecos, ',');
890 if (comma)
891 *comma = '\0'; // cut off non-name comment fields
892 wxStrncpy (buf, wxSafeConvertMB2WX(who->pw_gecos), sz - 1);
893 return true;
894 }
895
896 return false;
897 #else // !HAVE_PW_GECOS
898 return wxGetUserId(buf, sz);
899 #endif // HAVE_PW_GECOS/!HAVE_PW_GECOS
900 }
901
902 bool wxIsPlatform64Bit()
903 {
904 const wxString machine = wxGetCommandOutput(wxT("uname -m"));
905
906 // the test for "64" is obviously not 100% reliable but seems to work fine
907 // in practice
908 return machine.Contains(wxT("64")) ||
909 machine.Contains(wxT("alpha"));
910 }
911
912 // these functions are in mac/utils.cpp for wxMac
913 #ifndef __WXMAC__
914
915 wxOperatingSystemId wxGetOsVersion(int *verMaj, int *verMin)
916 {
917 // get OS version
918 int major, minor;
919 wxString release = wxGetCommandOutput(wxT("uname -r"));
920 if ( release.empty() ||
921 wxSscanf(release.c_str(), wxT("%d.%d"), &major, &minor) != 2 )
922 {
923 // failed to get version string or unrecognized format
924 major =
925 minor = -1;
926 }
927
928 if ( verMaj )
929 *verMaj = major;
930 if ( verMin )
931 *verMin = minor;
932
933 // try to understand which OS are we running
934 wxString kernel = wxGetCommandOutput(wxT("uname -s"));
935 if ( kernel.empty() )
936 kernel = wxGetCommandOutput(wxT("uname -o"));
937
938 if ( kernel.empty() )
939 return wxOS_UNKNOWN;
940
941 return wxPlatformInfo::GetOperatingSystemId(kernel);
942 }
943
944 wxString wxGetOsDescription()
945 {
946 return wxGetCommandOutput(wxT("uname -s -r -m"));
947 }
948
949 #endif // !__WXMAC__
950
951 unsigned long wxGetProcessId()
952 {
953 return (unsigned long)getpid();
954 }
955
956 wxMemorySize wxGetFreeMemory()
957 {
958 #if defined(__LINUX__)
959 // get it from /proc/meminfo
960 FILE *fp = fopen("/proc/meminfo", "r");
961 if ( fp )
962 {
963 long memFree = -1;
964
965 char buf[1024];
966 if ( fgets(buf, WXSIZEOF(buf), fp) && fgets(buf, WXSIZEOF(buf), fp) )
967 {
968 // /proc/meminfo changed its format in kernel 2.6
969 if ( wxPlatformInfo().CheckOSVersion(2, 6) )
970 {
971 unsigned long cached, buffers;
972 sscanf(buf, "MemFree: %ld", &memFree);
973
974 fgets(buf, WXSIZEOF(buf), fp);
975 sscanf(buf, "Buffers: %lu", &buffers);
976
977 fgets(buf, WXSIZEOF(buf), fp);
978 sscanf(buf, "Cached: %lu", &cached);
979
980 // add to "MemFree" also the "Buffers" and "Cached" values as
981 // free(1) does as otherwise the value never makes sense: for
982 // kernel 2.6 it's always almost 0
983 memFree += buffers + cached;
984
985 // values here are always expressed in kB and we want bytes
986 memFree *= 1024;
987 }
988 else // Linux 2.4 (or < 2.6, anyhow)
989 {
990 long memTotal, memUsed;
991 sscanf(buf, "Mem: %ld %ld %ld", &memTotal, &memUsed, &memFree);
992 }
993 }
994
995 fclose(fp);
996
997 return (wxMemorySize)memFree;
998 }
999 #elif defined(__SGI__)
1000 struct rminfo realmem;
1001 if ( sysmp(MP_SAGET, MPSA_RMINFO, &realmem, sizeof realmem) == 0 )
1002 return ((wxMemorySize)realmem.physmem * sysconf(_SC_PAGESIZE));
1003 #elif defined(_SC_AVPHYS_PAGES)
1004 return ((wxMemorySize)sysconf(_SC_AVPHYS_PAGES))*sysconf(_SC_PAGESIZE);
1005 //#elif defined(__FREEBSD__) -- might use sysctl() to find it out, probably
1006 #endif
1007
1008 // can't find it out
1009 return -1;
1010 }
1011
1012 bool wxGetDiskSpace(const wxString& path, wxDiskspaceSize_t *pTotal, wxDiskspaceSize_t *pFree)
1013 {
1014 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
1015 // the case to "char *" is needed for AIX 4.3
1016 wxStatfs_t fs;
1017 if ( wxStatfs((char *)(const char*)path.fn_str(), &fs) != 0 )
1018 {
1019 wxLogSysError( wxT("Failed to get file system statistics") );
1020
1021 return false;
1022 }
1023
1024 // under Solaris we also have to use f_frsize field instead of f_bsize
1025 // which is in general a multiple of f_frsize
1026 #ifdef HAVE_STATVFS
1027 wxDiskspaceSize_t blockSize = fs.f_frsize;
1028 #else // HAVE_STATFS
1029 wxDiskspaceSize_t blockSize = fs.f_bsize;
1030 #endif // HAVE_STATVFS/HAVE_STATFS
1031
1032 if ( pTotal )
1033 {
1034 *pTotal = wxDiskspaceSize_t(fs.f_blocks) * blockSize;
1035 }
1036
1037 if ( pFree )
1038 {
1039 *pFree = wxDiskspaceSize_t(fs.f_bavail) * blockSize;
1040 }
1041
1042 return true;
1043 #else // !HAVE_STATFS && !HAVE_STATVFS
1044 return false;
1045 #endif // HAVE_STATFS
1046 }
1047
1048 // ----------------------------------------------------------------------------
1049 // env vars
1050 // ----------------------------------------------------------------------------
1051
1052 #if USE_PUTENV
1053
1054 WX_DECLARE_STRING_HASH_MAP(char *, wxEnvVars);
1055
1056 static wxEnvVars gs_envVars;
1057
1058 class wxSetEnvModule : public wxModule
1059 {
1060 public:
1061 virtual bool OnInit() { return true; }
1062 virtual void OnExit()
1063 {
1064 for ( wxEnvVars::const_iterator i = gs_envVars.begin();
1065 i != gs_envVars.end();
1066 ++i )
1067 {
1068 free(i->second);
1069 }
1070
1071 gs_envVars.clear();
1072 }
1073
1074 DECLARE_DYNAMIC_CLASS(wxSetEnvModule)
1075 };
1076
1077 IMPLEMENT_DYNAMIC_CLASS(wxSetEnvModule, wxModule)
1078
1079 #endif // USE_PUTENV
1080
1081 bool wxGetEnv(const wxString& var, wxString *value)
1082 {
1083 // wxGetenv is defined as getenv()
1084 char *p = wxGetenv(var);
1085 if ( !p )
1086 return false;
1087
1088 if ( value )
1089 {
1090 *value = p;
1091 }
1092
1093 return true;
1094 }
1095
1096 static bool wxDoSetEnv(const wxString& variable, const char *value)
1097 {
1098 #if defined(HAVE_SETENV)
1099 if ( !value )
1100 {
1101 #ifdef HAVE_UNSETENV
1102 // don't test unsetenv() return value: it's void on some systems (at
1103 // least Darwin)
1104 unsetenv(variable.mb_str());
1105 return true;
1106 #else
1107 value = ""; // we can't pass NULL to setenv()
1108 #endif
1109 }
1110
1111 return setenv(variable.mb_str(), value, 1 /* overwrite */) == 0;
1112 #elif defined(HAVE_PUTENV)
1113 wxString s = variable;
1114 if ( value )
1115 s << _T('=') << value;
1116
1117 // transform to ANSI
1118 const wxWX2MBbuf p = s.mb_str();
1119
1120 char *buf = (char *)malloc(strlen(p) + 1);
1121 strcpy(buf, p);
1122
1123 // store the string to free() it later
1124 wxEnvVars::iterator i = gs_envVars.find(variable);
1125 if ( i != gs_envVars.end() )
1126 {
1127 free(i->second);
1128 i->second = buf;
1129 }
1130 else // this variable hadn't been set before
1131 {
1132 gs_envVars[variable] = buf;
1133 }
1134
1135 return putenv(buf) == 0;
1136 #else // no way to set an env var
1137 return false;
1138 #endif
1139 }
1140
1141 bool wxSetEnv(const wxString& variable, const wxString& value)
1142 {
1143 return wxDoSetEnv(variable, value.mb_str());
1144 }
1145
1146 bool wxUnsetEnv(const wxString& variable)
1147 {
1148 return wxDoSetEnv(variable, NULL);
1149 }
1150
1151 // ----------------------------------------------------------------------------
1152 // signal handling
1153 // ----------------------------------------------------------------------------
1154
1155 #if wxUSE_ON_FATAL_EXCEPTION
1156
1157 #include <signal.h>
1158
1159 extern "C" void wxFatalSignalHandler(wxTYPE_SA_HANDLER)
1160 {
1161 if ( wxTheApp )
1162 {
1163 // give the user a chance to do something special about this
1164 wxTheApp->OnFatalException();
1165 }
1166
1167 abort();
1168 }
1169
1170 bool wxHandleFatalExceptions(bool doit)
1171 {
1172 // old sig handlers
1173 static bool s_savedHandlers = false;
1174 static struct sigaction s_handlerFPE,
1175 s_handlerILL,
1176 s_handlerBUS,
1177 s_handlerSEGV;
1178
1179 bool ok = true;
1180 if ( doit && !s_savedHandlers )
1181 {
1182 // install the signal handler
1183 struct sigaction act;
1184
1185 // some systems extend it with non std fields, so zero everything
1186 memset(&act, 0, sizeof(act));
1187
1188 act.sa_handler = wxFatalSignalHandler;
1189 sigemptyset(&act.sa_mask);
1190 act.sa_flags = 0;
1191
1192 ok &= sigaction(SIGFPE, &act, &s_handlerFPE) == 0;
1193 ok &= sigaction(SIGILL, &act, &s_handlerILL) == 0;
1194 ok &= sigaction(SIGBUS, &act, &s_handlerBUS) == 0;
1195 ok &= sigaction(SIGSEGV, &act, &s_handlerSEGV) == 0;
1196 if ( !ok )
1197 {
1198 wxLogDebug(_T("Failed to install our signal handler."));
1199 }
1200
1201 s_savedHandlers = true;
1202 }
1203 else if ( s_savedHandlers )
1204 {
1205 // uninstall the signal handler
1206 ok &= sigaction(SIGFPE, &s_handlerFPE, NULL) == 0;
1207 ok &= sigaction(SIGILL, &s_handlerILL, NULL) == 0;
1208 ok &= sigaction(SIGBUS, &s_handlerBUS, NULL) == 0;
1209 ok &= sigaction(SIGSEGV, &s_handlerSEGV, NULL) == 0;
1210 if ( !ok )
1211 {
1212 wxLogDebug(_T("Failed to uninstall our signal handler."));
1213 }
1214
1215 s_savedHandlers = false;
1216 }
1217 //else: nothing to do
1218
1219 return ok;
1220 }
1221
1222 #endif // wxUSE_ON_FATAL_EXCEPTION
1223
1224 // ----------------------------------------------------------------------------
1225 // wxExecute support
1226 // ----------------------------------------------------------------------------
1227
1228 bool wxAppTraits::CreateEndProcessPipe(wxExecuteData& execData)
1229 {
1230 return execData.pipeEndProcDetect.Create();
1231 }
1232
1233 bool wxAppTraits::IsWriteFDOfEndProcessPipe(wxExecuteData& execData, int fd)
1234 {
1235 return fd == (execData.pipeEndProcDetect)[wxPipe::Write];
1236 }
1237
1238 void wxAppTraits::DetachWriteFDOfEndProcessPipe(wxExecuteData& execData)
1239 {
1240 execData.pipeEndProcDetect.Detach(wxPipe::Write);
1241 execData.pipeEndProcDetect.Close();
1242 }
1243
1244 int wxAppTraits::AddProcessCallback(wxEndProcessData *data, int fd)
1245 {
1246 // define a custom handler processing only the closure of the descriptor
1247 struct wxEndProcessFDIOHandler : public wxFDIOHandler
1248 {
1249 wxEndProcessFDIOHandler(wxEndProcessData *data, int fd)
1250 : m_data(data), m_fd(fd)
1251 {}
1252
1253 virtual void OnReadWaiting()
1254 { wxFAIL_MSG("this isn't supposed to happen"); }
1255 virtual void OnWriteWaiting()
1256 { wxFAIL_MSG("this isn't supposed to happen"); }
1257
1258 virtual void OnExceptionWaiting()
1259 {
1260 const int pid = m_data->pid > 0 ? m_data->pid : -(m_data->pid);
1261 int status = 0;
1262
1263 // has the process really terminated?
1264 int rc = waitpid(pid, &status, WNOHANG);
1265 if ( rc == 0 )
1266 {
1267 // This can only happen if the child application closes our
1268 // dummy pipe that is used to monitor its lifetime; in that
1269 // case, our best bet is to pretend the process did terminate,
1270 // because otherwise wxExecute() would hang indefinitely
1271 // (OnExceptionWaiting() won't be called again, the descriptor
1272 // is closed now).
1273 wxLogDebug("Child process (PID %d) still alive, "
1274 "even though pipe was closed.", pid);
1275 }
1276 else if ( rc == -1 )
1277 {
1278 // As above, if waitpid() fails, the best we can do is to log the
1279 // error and pretend the child terminated:
1280 wxLogSysError(_("Failed to check child process' status"));
1281 }
1282
1283 // set exit code to -1 if something bad happened
1284 m_data->exitcode = rc > 0 && WIFEXITED(status) ? WEXITSTATUS(status)
1285 : -1;
1286
1287 wxLogTrace("exec",
1288 "Child process (PID %d) terminated with exit code %d",
1289 pid, m_data->exitcode);
1290
1291 // child exited, end waiting
1292 wxFDIODispatcher::Get()->UnregisterFD(m_fd);
1293 close(m_fd);
1294
1295 wxHandleProcessTermination(m_data);
1296
1297 delete this;
1298 }
1299
1300 wxEndProcessData * const m_data;
1301 const int m_fd;
1302 };
1303
1304 wxFDIODispatcher::Get()->RegisterFD
1305 (
1306 fd,
1307 new wxEndProcessFDIOHandler(data, fd),
1308 wxFDIO_EXCEPTION
1309 );
1310 return fd; // unused, but return something unique for the tag
1311 }
1312
1313 bool wxAppTraits::CheckForRedirectedIO(wxExecuteData& execData)
1314 {
1315 #if HAS_PIPE_INPUT_STREAM
1316 bool hasIO = false;
1317
1318 if ( execData.bufOut && execData.bufOut->Update() )
1319 hasIO = true;
1320
1321 if ( execData.bufErr && execData.bufErr->Update() )
1322 hasIO = true;
1323
1324 return hasIO;
1325 #else // !HAS_PIPE_INPUT_STREAM
1326 return false;
1327 #endif // HAS_PIPE_INPUT_STREAM/!HAS_PIPE_INPUT_STREAM
1328 }
1329
1330 class wxReadFDIOHandler : public wxFDIOHandler
1331 {
1332 public:
1333 wxReadFDIOHandler(wxFDIODispatcher& disp, int fd) : m_fd(fd)
1334 {
1335 if ( fd )
1336 disp.RegisterFD(fd, this, wxFDIO_INPUT);
1337 }
1338
1339 int GetFD() const { return m_fd; }
1340
1341 virtual void OnWriteWaiting() { wxFAIL_MSG("unreachable"); }
1342 virtual void OnExceptionWaiting() { wxFAIL_MSG("unreachable"); }
1343
1344 private:
1345 const int m_fd;
1346
1347 DECLARE_NO_COPY_CLASS(wxReadFDIOHandler)
1348 };
1349
1350 class wxEndHandler : public wxReadFDIOHandler
1351 {
1352 public:
1353 wxEndHandler(wxFDIODispatcher& disp, int fd)
1354 : wxReadFDIOHandler(disp, fd)
1355 {
1356 m_terminated = false;
1357 }
1358
1359 bool Terminated() const { return m_terminated; }
1360
1361 virtual void OnReadWaiting() { m_terminated = true; }
1362
1363 private:
1364 bool m_terminated;
1365
1366 DECLARE_NO_COPY_CLASS(wxEndHandler)
1367 };
1368
1369 #if wxUSE_STREAMS
1370
1371 class wxRedirectedIOHandler : public wxReadFDIOHandler
1372 {
1373 public:
1374 wxRedirectedIOHandler(wxFDIODispatcher& disp,
1375 int fd,
1376 wxStreamTempInputBuffer *buf)
1377 : wxReadFDIOHandler(disp, fd),
1378 m_buf(buf)
1379 {
1380 }
1381
1382 virtual void OnReadWaiting()
1383 {
1384 m_buf->Update();
1385 }
1386
1387 private:
1388 wxStreamTempInputBuffer * const m_buf;
1389
1390 DECLARE_NO_COPY_CLASS(wxRedirectedIOHandler)
1391 };
1392
1393 #endif // wxUSE_STREAMS
1394
1395 int wxAppTraits::WaitForChild(wxExecuteData& execData)
1396 {
1397 if ( !(execData.flags & wxEXEC_SYNC) )
1398 {
1399 // asynchronous execution: just launch the process and return
1400 wxEndProcessData *endProcData = new wxEndProcessData;
1401 endProcData->process = execData.process;
1402 endProcData->pid = execData.pid;
1403 endProcData->tag = AddProcessCallback
1404 (
1405 endProcData,
1406 execData.GetEndProcReadFD()
1407 );
1408
1409 return execData.pid;
1410 }
1411
1412 #if wxUSE_STREAMS
1413 wxProcess * const process = execData.process;
1414 if ( process && process->IsRedirected() )
1415 {
1416 // we can't simply block waiting for the child to terminate as we would
1417 // dead lock if it writes more than the pipe buffer size (typically
1418 // 4KB) bytes of output -- it would then block waiting for us to read
1419 // the data while we'd block waiting for it to terminate
1420 //
1421 // so multiplex here waiting for any input from the child or closure of
1422 // the pipe used to indicate its termination
1423 wxSelectDispatcher disp;
1424
1425 wxEndHandler endHandler(disp, execData.GetEndProcReadFD());
1426
1427 wxRedirectedIOHandler outHandler(disp, execData.fdOut, execData.bufOut),
1428 errHandler(disp, execData.fdErr, execData.bufErr);
1429
1430 while ( !endHandler.Terminated() )
1431 {
1432 disp.Dispatch();
1433 }
1434 }
1435 //else: no IO redirection, just block waiting for the child to exit
1436 #endif // wxUSE_STREAMS
1437
1438 int status = 0;
1439
1440 int result = waitpid(execData.pid, &status, 0);
1441 #ifdef __DARWIN__
1442 /* DE: waitpid manpage states that waitpid can fail with EINTR
1443 if the call is interrupted by a caught signal. I suppose
1444 that means that this ought to be a while loop.
1445
1446 The odd thing is that it seems to fail EVERY time. It fails
1447 with a quickly exiting process (e.g. echo), and fails with a
1448 slowly exiting process (e.g. sleep 2) but clearly after
1449 having waited for the child to exit. Maybe it's a bug in
1450 my particular version.
1451
1452 It works, however, from the CFSocket callback without this
1453 trick but in that case it's used only after CFSocket calls
1454 the callback and with the WNOHANG flag which would seem to
1455 preclude it from being interrupted or at least make it much
1456 less likely since it would not then be waiting.
1457
1458 If Darwin's man page is to be believed then this is definitely
1459 necessary. It's just weird that I've never seen it before
1460 and apparently no one else has either or you'd think they'd
1461 have reported it by now. Perhaps blocking the GUI while
1462 waiting for a child process to exit is simply not that common.
1463 */
1464 if ( result == -1 && errno == EINTR )
1465 {
1466 result = waitpid(execData.pid, &status, 0);
1467 }
1468 #endif // __DARWIN__
1469
1470 if ( result == -1 )
1471 {
1472 wxLogLastError("waitpid");
1473 }
1474 else // child terminated
1475 {
1476 wxASSERT_MSG( result == execData.pid,
1477 "unexpected waitpid() return value" );
1478
1479 if ( WIFEXITED(status) )
1480 {
1481 return WEXITSTATUS(status);
1482 }
1483 else // abnormal termination?
1484 {
1485 wxASSERT_MSG( WIFSIGNALED(status),
1486 "unexpected child wait status" );
1487 }
1488 }
1489
1490 wxLogSysError(_("Waiting for subprocess termination failed"));
1491
1492 return -1;
1493 }
1494
1495 void wxHandleProcessTermination(wxEndProcessData *proc_data)
1496 {
1497 // notify user about termination if required
1498 if ( proc_data->process )
1499 {
1500 proc_data->process->OnTerminate(proc_data->pid, proc_data->exitcode);
1501 }
1502
1503 // clean up
1504 if ( proc_data->pid > 0 )
1505 {
1506 // async execution
1507 delete proc_data;
1508 }
1509 else // sync execution
1510 {
1511 // let wxExecute() know that the process has terminated
1512 proc_data->pid = 0;
1513 }
1514 }
1515