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