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 // 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, 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         // NB: we used to close all the unused descriptors of the child here
 546         //     but this broke some programs which relied on e.g. FD 1 being
 547         //     always opened so don't do it any more, after all there doesn't
 548         //     seem to be any real problem with keeping them opened
 549
 550 #if !defined(__VMS) && !defined(__EMX__)
 551         if ( flags & wxEXEC_MAKE_GROUP_LEADER )
 552         {
 553             // Set process group to child process' pid.  Then killing -pid
 554             // of the parent will kill the process and all of its children.
 555             setsid();
 556         }
 557 #endif // !__VMS
 558
 559         // reading side can be safely closed but we should keep the write one
 560         // opened, it will be only closed when the process terminates resulting
 561         // in a read notification to the parent
 562         execData.pipeEndProcDetect.Detach(wxPipe::Write);
 563         execData.pipeEndProcDetect.Close();
 564
 565         // redirect stdin, stdout and stderr
 566         if ( pipeIn.IsOk() )
 567         {
 568             if ( dup2(pipeIn[wxPipe::Read], STDIN_FILENO) == -1 ||
 569                  dup2(pipeOut[wxPipe::Write], STDOUT_FILENO) == -1 ||
 570                  dup2(pipeErr[wxPipe::Write], STDERR_FILENO) == -1 )
 571             {
 572                 wxLogSysError(_("Failed to redirect child process input/output"));
 573             }
 574
 575             pipeIn.Close();
 576             pipeOut.Close();
 577             pipeErr.Close();
 578         }
 579
 580         execvp(*argv, argv);
 581
 582         fprintf(stderr, "execvp(");
 583         for ( char **a = argv; *a; a++ )
 584             fprintf(stderr, "%s%s", a == argv ? "" : ", ", *a);
 585         fprintf(stderr, ") failed with error %d!\n", errno);
 586
 587         // there is no return after successful exec()
 588         _exit(-1);
 589
 590         // some compilers complain about missing return - of course, they
 591         // should know that exit() doesn't return but what else can we do if
 592         // they don't?
 593         //
 594         // and, sure enough, other compilers complain about unreachable code
 595         // after exit() call, so we can just always have return here...
 596 #if defined(__VMS) || defined(__INTEL_COMPILER)
 597         return 0;
 598 #endif
 599     }
 600     else // we're in parent
 601     {
 602         // save it for WaitForChild() use
 603         execData.pid = pid;
 604
 605         // prepare for IO redirection
 606
 607 #if HAS_PIPE_INPUT_STREAM
 608         // the input buffer bufOut is connected to stdout, this is why it is
 609         // called bufOut and not bufIn
 610         wxStreamTempInputBuffer bufOut,
 611                                 bufErr;
 612
 613         if ( process && process->IsRedirected() )
 614         {
 615             wxOutputStream *inStream =
 616                 new wxFileOutputStream(pipeIn.Detach(wxPipe::Write));
 617
 618             const int fdOut = pipeOut.Detach(wxPipe::Read);
 619             wxPipeInputStream *outStream = new wxPipeInputStream(fdOut);
 620
 621             const int fdErr = pipeErr.Detach(wxPipe::Read);
 622             wxPipeInputStream *errStream = new wxPipeInputStream(fdErr);
 623
 624             process->SetPipeStreams(outStream, inStream, errStream);
 625
 626             bufOut.Init(outStream);
 627             bufErr.Init(errStream);
 628
 629             execData.bufOut = &bufOut;
 630             execData.bufErr = &bufErr;
 631
 632             execData.fdOut = fdOut;
 633             execData.fdErr = fdErr;
 634         }
 635 #endif // HAS_PIPE_INPUT_STREAM
 636
 637         if ( pipeIn.IsOk() )
 638         {
 639             pipeIn.Close();
 640             pipeOut.Close();
 641             pipeErr.Close();
 642         }
 643
 644         // we want this function to work even if there is no wxApp so ensure
 645         // that we have a valid traits pointer
 646         wxConsoleAppTraits traitsConsole;
 647         wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
 648         if ( !traits )
 649             traits = &traitsConsole;
 650
 651         return traits->WaitForChild(execData);
 652     }
 653
 654 #if !defined(__VMS) && !defined(__INTEL_COMPILER)
 655     return ERROR_RETURN_CODE;
 656 #endif
 657 }
 658
 659 #undef ERROR_RETURN_CODE
 660
 661 // ----------------------------------------------------------------------------
 662 // file and directory functions
 663 // ----------------------------------------------------------------------------
 664
 665 const wxChar* wxGetHomeDir( wxString *home  )
 666 {
 667     *home = wxGetUserHome();
 668     wxString tmp;
 669     if ( home->empty() )
 670         *home = wxT("/");
 671 #ifdef __VMS
 672     tmp = *home;
 673     if ( tmp.Last() != wxT(']'))
 674         if ( tmp.Last() != wxT('/')) *home << wxT('/');
 675 #endif
 676     return home->c_str();
 677 }
 678
 679 wxString wxGetUserHome( const wxString &user )
 680 {
 681     struct passwd *who = (struct passwd *) NULL;
 682
 683     if ( !user )
 684     {
 685         wxChar *ptr;
 686
 687         if ((ptr = wxGetenv(wxT("HOME"))) != NULL)
 688         {
 689             return ptr;
 690         }
 691
 692         if ((ptr = wxGetenv(wxT("USER"))) != NULL ||
 693              (ptr = wxGetenv(wxT("LOGNAME"))) != NULL)
 694         {
 695             who = getpwnam(wxSafeConvertWX2MB(ptr));
 696         }
 697
 698         // make sure the user exists!
 699         if ( !who )
 700         {
 701             who = getpwuid(getuid());
 702         }
 703     }
 704     else
 705     {
 706       who = getpwnam (user.mb_str());
 707     }
 708
 709     return wxSafeConvertMB2WX(who ? who->pw_dir : 0);
 710 }
 711
 712 // ----------------------------------------------------------------------------
 713 // network and user id routines
 714 // ----------------------------------------------------------------------------
 715
 716 // private utility function which returns output of the given command, removing
 717 // the trailing newline
 718 static wxString wxGetCommandOutput(const wxString &cmd)
 719 {
 720     FILE *f = popen(cmd.ToAscii(), "r");
 721     if ( !f )
 722     {
 723         wxLogSysError(_T("Executing \"%s\" failed"), cmd.c_str());
 724         return wxEmptyString;
 725     }
 726
 727     wxString s;
 728     char buf[256];
 729     while ( !feof(f) )
 730     {
 731         if ( !fgets(buf, sizeof(buf), f) )
 732             break;
 733
 734         s += wxString::FromAscii(buf);
 735     }
 736
 737     pclose(f);
 738
 739     if ( !s.empty() && s.Last() == _T('\n') )
 740         s.RemoveLast();
 741
 742     return s;
 743 }
 744
 745 // retrieve either the hostname or FQDN depending on platform (caller must
 746 // check whether it's one or the other, this is why this function is for
 747 // private use only)
 748 static bool wxGetHostNameInternal(wxChar *buf, int sz)
 749 {
 750     wxCHECK_MSG( buf, false, wxT("NULL pointer in wxGetHostNameInternal") );
 751
 752     *buf = wxT('\0');
 753
 754     // we're using uname() which is POSIX instead of less standard sysinfo()
 755 #if defined(HAVE_UNAME)
 756     struct utsname uts;
 757     bool ok = uname(&uts) != -1;
 758     if ( ok )
 759     {
 760         wxStrlcpy(buf, wxSafeConvertMB2WX(uts.nodename), sz);
 761     }
 762 #elif defined(HAVE_GETHOSTNAME)
 763     char cbuf[sz];
 764     bool ok = gethostname(cbuf, sz) != -1;
 765     if ( ok )
 766     {
 767         wxStrlcpy(buf, wxSafeConvertMB2WX(cbuf), sz);
 768     }
 769 #else // no uname, no gethostname
 770     wxFAIL_MSG(wxT("don't know host name for this machine"));
 771
 772     bool ok = false;
 773 #endif // uname/gethostname
 774
 775     if ( !ok )
 776     {
 777         wxLogSysError(_("Cannot get the hostname"));
 778     }
 779
 780     return ok;
 781 }
 782
 783 bool wxGetHostName(wxChar *buf, int sz)
 784 {
 785     bool ok = wxGetHostNameInternal(buf, sz);
 786
 787     if ( ok )
 788     {
 789         // BSD systems return the FQDN, we only want the hostname, so extract
 790         // it (we consider that dots are domain separators)
 791         wxChar *dot = wxStrchr(buf, wxT('.'));
 792         if ( dot )
 793         {
 794             // nuke it
 795             *dot = wxT('\0');
 796         }
 797     }
 798
 799     return ok;
 800 }
 801
 802 bool wxGetFullHostName(wxChar *buf, int sz)
 803 {
 804     bool ok = wxGetHostNameInternal(buf, sz);
 805
 806     if ( ok )
 807     {
 808         if ( !wxStrchr(buf, wxT('.')) )
 809         {
 810             struct hostent *host = gethostbyname(wxSafeConvertWX2MB(buf));
 811             if ( !host )
 812             {
 813                 wxLogSysError(_("Cannot get the official hostname"));
 814
 815                 ok = false;
 816             }
 817             else
 818             {
 819                 // the canonical name
 820                 wxStrlcpy(buf, wxSafeConvertMB2WX(host->h_name), sz);
 821             }
 822         }
 823         //else: it's already a FQDN (BSD behaves this way)
 824     }
 825
 826     return ok;
 827 }
 828
 829 bool wxGetUserId(wxChar *buf, int sz)
 830 {
 831     struct passwd *who;
 832
 833     *buf = wxT('\0');
 834     if ((who = getpwuid(getuid ())) != NULL)
 835     {
 836         wxStrlcpy (buf, wxSafeConvertMB2WX(who->pw_name), sz);
 837         return true;
 838     }
 839
 840     return false;
 841 }
 842
 843 bool wxGetUserName(wxChar *buf, int sz)
 844 {
 845 #ifdef HAVE_PW_GECOS
 846     struct passwd *who;
 847
 848     *buf = wxT('\0');
 849     if ((who = getpwuid (getuid ())) != NULL)
 850     {
 851        char *comma = strchr(who->pw_gecos, ',');
 852        if (comma)
 853            *comma = '\0'; // cut off non-name comment fields
 854        wxStrlcpy(buf, wxSafeConvertMB2WX(who->pw_gecos), sz);
 855        return true;
 856     }
 857
 858     return false;
 859 #else // !HAVE_PW_GECOS
 860     return wxGetUserId(buf, sz);
 861 #endif // HAVE_PW_GECOS/!HAVE_PW_GECOS
 862 }
 863
 864 bool wxIsPlatform64Bit()
 865 {
 866     const wxString machine = wxGetCommandOutput(wxT("uname -m"));
 867
 868     // the test for "64" is obviously not 100% reliable but seems to work fine
 869     // in practice
 870     return machine.Contains(wxT("64")) ||
 871                 machine.Contains(wxT("alpha"));
 872 }
 873
 874 // these functions are in src/osx/utilsexc_base.cpp for wxMac
 875 #ifndef __WXMAC__
 876
 877 wxOperatingSystemId wxGetOsVersion(int *verMaj, int *verMin)
 878 {
 879     // get OS version
 880     int major, minor;
 881     wxString release = wxGetCommandOutput(wxT("uname -r"));
 882     if ( release.empty() ||
 883          wxSscanf(release.c_str(), wxT("%d.%d"), &major, &minor) != 2 )
 884     {
 885         // failed to get version string or unrecognized format
 886         major =
 887         minor = -1;
 888     }
 889
 890     if ( verMaj )
 891         *verMaj = major;
 892     if ( verMin )
 893         *verMin = minor;
 894
 895     // try to understand which OS are we running
 896     wxString kernel = wxGetCommandOutput(wxT("uname -s"));
 897     if ( kernel.empty() )
 898         kernel = wxGetCommandOutput(wxT("uname -o"));
 899
 900     if ( kernel.empty() )
 901         return wxOS_UNKNOWN;
 902
 903     return wxPlatformInfo::GetOperatingSystemId(kernel);
 904 }
 905
 906 wxString wxGetOsDescription()
 907 {
 908     return wxGetCommandOutput(wxT("uname -s -r -m"));
 909 }
 910
 911 #endif // !__WXMAC__
 912
 913 unsigned long wxGetProcessId()
 914 {
 915     return (unsigned long)getpid();
 916 }
 917
 918 wxMemorySize wxGetFreeMemory()
 919 {
 920 #if defined(__LINUX__)
 921     // get it from /proc/meminfo
 922     FILE *fp = fopen("/proc/meminfo", "r");
 923     if ( fp )
 924     {
 925         long memFree = -1;
 926
 927         char buf[1024];
 928         if ( fgets(buf, WXSIZEOF(buf), fp) && fgets(buf, WXSIZEOF(buf), fp) )
 929         {
 930             // /proc/meminfo changed its format in kernel 2.6
 931             if ( wxPlatformInfo().CheckOSVersion(2, 6) )
 932             {
 933                 unsigned long cached, buffers;
 934                 sscanf(buf, "MemFree: %ld", &memFree);
 935
 936                 fgets(buf, WXSIZEOF(buf), fp);
 937                 sscanf(buf, "Buffers: %lu", &buffers);
 938
 939                 fgets(buf, WXSIZEOF(buf), fp);
 940                 sscanf(buf, "Cached: %lu", &cached);
 941
 942                 // add to "MemFree" also the "Buffers" and "Cached" values as
 943                 // free(1) does as otherwise the value never makes sense: for
 944                 // kernel 2.6 it's always almost 0
 945                 memFree += buffers + cached;
 946
 947                 // values here are always expressed in kB and we want bytes
 948                 memFree *= 1024;
 949             }
 950             else // Linux 2.4 (or < 2.6, anyhow)
 951             {
 952                 long memTotal, memUsed;
 953                 sscanf(buf, "Mem: %ld %ld %ld", &memTotal, &memUsed, &memFree);
 954             }
 955         }
 956
 957         fclose(fp);
 958
 959         return (wxMemorySize)memFree;
 960     }
 961 #elif defined(__SGI__)
 962     struct rminfo realmem;
 963     if ( sysmp(MP_SAGET, MPSA_RMINFO, &realmem, sizeof realmem) == 0 )
 964         return ((wxMemorySize)realmem.physmem * sysconf(_SC_PAGESIZE));
 965 #elif defined(_SC_AVPHYS_PAGES)
 966     return ((wxMemorySize)sysconf(_SC_AVPHYS_PAGES))*sysconf(_SC_PAGESIZE);
 967 //#elif defined(__FREEBSD__) -- might use sysctl() to find it out, probably
 968 #endif
 969
 970     // can't find it out
 971     return -1;
 972 }
 973
 974 bool wxGetDiskSpace(const wxString& path, wxDiskspaceSize_t *pTotal, wxDiskspaceSize_t *pFree)
 975 {
 976 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
 977     // the case to "char *" is needed for AIX 4.3
 978     wxStatfs_t fs;
 979     if ( wxStatfs((char *)(const char*)path.fn_str(), &fs) != 0 )
 980     {
 981         wxLogSysError( wxT("Failed to get file system statistics") );
 982
 983         return false;
 984     }
 985
 986     // under Solaris we also have to use f_frsize field instead of f_bsize
 987     // which is in general a multiple of f_frsize
 988 #ifdef HAVE_STATVFS
 989     wxDiskspaceSize_t blockSize = fs.f_frsize;
 990 #else // HAVE_STATFS
 991     wxDiskspaceSize_t blockSize = fs.f_bsize;
 992 #endif // HAVE_STATVFS/HAVE_STATFS
 993
 994     if ( pTotal )
 995     {
 996         *pTotal = wxDiskspaceSize_t(fs.f_blocks) * blockSize;
 997     }
 998
 999     if ( pFree )
1000     {
1001         *pFree = wxDiskspaceSize_t(fs.f_bavail) * blockSize;
1002     }
1003
1004     return true;
1005 #else // !HAVE_STATFS && !HAVE_STATVFS
1006     return false;
1007 #endif // HAVE_STATFS
1008 }
1009
1010 // ----------------------------------------------------------------------------
1011 // env vars
1012 // ----------------------------------------------------------------------------
1013
1014 #if USE_PUTENV
1015
1016 WX_DECLARE_STRING_HASH_MAP(char *, wxEnvVars);
1017
1018 static wxEnvVars gs_envVars;
1019
1020 class wxSetEnvModule : public wxModule
1021 {
1022 public:
1023     virtual bool OnInit() { return true; }
1024     virtual void OnExit()
1025     {
1026         for ( wxEnvVars::const_iterator i = gs_envVars.begin();
1027               i != gs_envVars.end();
1028               ++i )
1029         {
1030             free(i->second);
1031         }
1032
1033         gs_envVars.clear();
1034     }
1035
1036     DECLARE_DYNAMIC_CLASS(wxSetEnvModule)
1037 };
1038
1039 IMPLEMENT_DYNAMIC_CLASS(wxSetEnvModule, wxModule)
1040
1041 #endif // USE_PUTENV
1042
1043 bool wxGetEnv(const wxString& var, wxString *value)
1044 {
1045     // wxGetenv is defined as getenv()
1046     char *p = wxGetenv(var);
1047     if ( !p )
1048         return false;
1049
1050     if ( value )
1051     {
1052         *value = p;
1053     }
1054
1055     return true;
1056 }
1057
1058 static bool wxDoSetEnv(const wxString& variable, const char *value)
1059 {
1060 #if defined(HAVE_SETENV)
1061     if ( !value )
1062     {
1063 #ifdef HAVE_UNSETENV
1064         // don't test unsetenv() return value: it's void on some systems (at
1065         // least Darwin)
1066         unsetenv(variable.mb_str());
1067         return true;
1068 #else
1069         value = ""; // we can't pass NULL to setenv()
1070 #endif
1071     }
1072
1073     return setenv(variable.mb_str(), value, 1 /* overwrite */) == 0;
1074 #elif defined(HAVE_PUTENV)
1075     wxString s = variable;
1076     if ( value )
1077         s << _T('=') << value;
1078
1079     // transform to ANSI
1080     const wxWX2MBbuf p = s.mb_str();
1081
1082     char *buf = (char *)malloc(strlen(p) + 1);
1083     strcpy(buf, p);
1084
1085     // store the string to free() it later
1086     wxEnvVars::iterator i = gs_envVars.find(variable);
1087     if ( i != gs_envVars.end() )
1088     {
1089         free(i->second);
1090         i->second = buf;
1091     }
1092     else // this variable hadn't been set before
1093     {
1094         gs_envVars[variable] = buf;
1095     }
1096
1097     return putenv(buf) == 0;
1098 #else // no way to set an env var
1099     return false;
1100 #endif
1101 }
1102
1103 bool wxSetEnv(const wxString& variable, const wxString& value)
1104 {
1105     return wxDoSetEnv(variable, value.mb_str());
1106 }
1107
1108 bool wxUnsetEnv(const wxString& variable)
1109 {
1110     return wxDoSetEnv(variable, NULL);
1111 }
1112
1113 // ----------------------------------------------------------------------------
1114 // signal handling
1115 // ----------------------------------------------------------------------------
1116
1117 #if wxUSE_ON_FATAL_EXCEPTION
1118
1119 #include <signal.h>
1120
1121 extern "C" void wxFatalSignalHandler(wxTYPE_SA_HANDLER)
1122 {
1123     if ( wxTheApp )
1124     {
1125         // give the user a chance to do something special about this
1126         wxTheApp->OnFatalException();
1127     }
1128
1129     abort();
1130 }
1131
1132 bool wxHandleFatalExceptions(bool doit)
1133 {
1134     // old sig handlers
1135     static bool s_savedHandlers = false;
1136     static struct sigaction s_handlerFPE,
1137                             s_handlerILL,
1138                             s_handlerBUS,
1139                             s_handlerSEGV;
1140
1141     bool ok = true;
1142     if ( doit && !s_savedHandlers )
1143     {
1144         // install the signal handler
1145         struct sigaction act;
1146
1147         // some systems extend it with non std fields, so zero everything
1148         memset(&act, 0, sizeof(act));
1149
1150         act.sa_handler = wxFatalSignalHandler;
1151         sigemptyset(&act.sa_mask);
1152         act.sa_flags = 0;
1153
1154         ok &= sigaction(SIGFPE, &act, &s_handlerFPE) == 0;
1155         ok &= sigaction(SIGILL, &act, &s_handlerILL) == 0;
1156         ok &= sigaction(SIGBUS, &act, &s_handlerBUS) == 0;
1157         ok &= sigaction(SIGSEGV, &act, &s_handlerSEGV) == 0;
1158         if ( !ok )
1159         {
1160             wxLogDebug(_T("Failed to install our signal handler."));
1161         }
1162
1163         s_savedHandlers = true;
1164     }
1165     else if ( s_savedHandlers )
1166     {
1167         // uninstall the signal handler
1168         ok &= sigaction(SIGFPE, &s_handlerFPE, NULL) == 0;
1169         ok &= sigaction(SIGILL, &s_handlerILL, NULL) == 0;
1170         ok &= sigaction(SIGBUS, &s_handlerBUS, NULL) == 0;
1171         ok &= sigaction(SIGSEGV, &s_handlerSEGV, NULL) == 0;
1172         if ( !ok )
1173         {
1174             wxLogDebug(_T("Failed to uninstall our signal handler."));
1175         }
1176
1177         s_savedHandlers = false;
1178     }
1179     //else: nothing to do
1180
1181     return ok;
1182 }
1183
1184 #endif // wxUSE_ON_FATAL_EXCEPTION
1185
1186 // ----------------------------------------------------------------------------
1187 // wxExecute support
1188 // ----------------------------------------------------------------------------
1189
1190 int wxAppTraits::AddProcessCallback(wxEndProcessData *data, int fd)
1191 {
1192     // define a custom handler processing only the closure of the descriptor
1193     struct wxEndProcessFDIOHandler : public wxFDIOHandler
1194     {
1195         wxEndProcessFDIOHandler(wxEndProcessData *data, int fd)
1196             : m_data(data), m_fd(fd)
1197         {
1198         }
1199
1200         virtual void OnReadWaiting()
1201         {
1202             wxFDIODispatcher::Get()->UnregisterFD(m_fd);
1203             close(m_fd);
1204
1205             wxHandleProcessTermination(m_data);
1206
1207             delete this;
1208         }
1209
1210         virtual void OnWriteWaiting() { wxFAIL_MSG("unreachable"); }
1211         virtual void OnExceptionWaiting() { wxFAIL_MSG("unreachable"); }
1212
1213         wxEndProcessData * const m_data;
1214         const int m_fd;
1215     };
1216
1217     wxFDIODispatcher::Get()->RegisterFD
1218                              (
1219                                  fd,
1220                                  new wxEndProcessFDIOHandler(data, fd),
1221                                  wxFDIO_INPUT
1222                              );
1223     return fd; // unused, but return something unique for the tag
1224 }
1225
1226 bool wxAppTraits::CheckForRedirectedIO(wxExecuteData& execData)
1227 {
1228 #if HAS_PIPE_INPUT_STREAM
1229     bool hasIO = false;
1230
1231     if ( execData.bufOut && execData.bufOut->Update() )
1232         hasIO = true;
1233
1234     if ( execData.bufErr && execData.bufErr->Update() )
1235         hasIO = true;
1236
1237     return hasIO;
1238 #else // !HAS_PIPE_INPUT_STREAM
1239     return false;
1240 #endif // HAS_PIPE_INPUT_STREAM/!HAS_PIPE_INPUT_STREAM
1241 }
1242
1243 // helper classes/functions used by WaitForChild()
1244 namespace
1245 {
1246
1247 // convenient base class for IO handlers which are registered for read
1248 // notifications only and which also stores the FD we're reading from
1249 //
1250 // the derived classes still have to implement OnReadWaiting()
1251 class wxReadFDIOHandler : public wxFDIOHandler
1252 {
1253 public:
1254     wxReadFDIOHandler(wxFDIODispatcher& disp, int fd) : m_fd(fd)
1255     {
1256         if ( fd )
1257             disp.RegisterFD(fd, this, wxFDIO_INPUT);
1258     }
1259
1260     virtual void OnWriteWaiting() { wxFAIL_MSG("unreachable"); }
1261     virtual void OnExceptionWaiting() { wxFAIL_MSG("unreachable"); }
1262
1263 protected:
1264     const int m_fd;
1265
1266     DECLARE_NO_COPY_CLASS(wxReadFDIOHandler)
1267 };
1268
1269 // class for monitoring our end of the process detection pipe, simply sets a
1270 // flag when input on the pipe (which must be due to EOF) is detected
1271 class wxEndHandler : public wxReadFDIOHandler
1272 {
1273 public:
1274     wxEndHandler(wxFDIODispatcher& disp, int fd)
1275         : wxReadFDIOHandler(disp, fd)
1276     {
1277         m_terminated = false;
1278     }
1279
1280     bool Terminated() const { return m_terminated; }
1281
1282     virtual void OnReadWaiting() { m_terminated = true; }
1283
1284 private:
1285     bool m_terminated;
1286
1287     DECLARE_NO_COPY_CLASS(wxEndHandler)
1288 };
1289
1290 #if HAS_PIPE_INPUT_STREAM
1291
1292 // class for monitoring our ends of child stdout/err, should be constructed
1293 // with the FD and stream from wxExecuteData and will do nothing if they're
1294 // invalid
1295 //
1296 // unlike wxEndHandler this class registers itself with the provided dispatcher
1297 class wxRedirectedIOHandler : public wxReadFDIOHandler
1298 {
1299 public:
1300     wxRedirectedIOHandler(wxFDIODispatcher& disp,
1301                           int fd,
1302                           wxStreamTempInputBuffer *buf)
1303         : wxReadFDIOHandler(disp, fd),
1304           m_buf(buf)
1305     {
1306     }
1307
1308     virtual void OnReadWaiting()
1309     {
1310         m_buf->Update();
1311     }
1312
1313 private:
1314     wxStreamTempInputBuffer * const m_buf;
1315
1316     DECLARE_NO_COPY_CLASS(wxRedirectedIOHandler)
1317 };
1318
1319 #endif // HAS_PIPE_INPUT_STREAM
1320
1321 // helper function which calls waitpid() and analyzes the result
1322 int DoWaitForChild(int pid, int flags = 0)
1323 {
1324     wxASSERT_MSG( pid > 0, "invalid PID" );
1325
1326     int status, rc;
1327
1328     // loop while we're getting EINTR
1329     for ( ;; )
1330     {
1331         rc = waitpid(pid, &status, flags);
1332
1333         if ( rc != -1 || errno != EINTR )
1334             break;
1335     }
1336
1337     if ( rc == 0 )
1338     {
1339         // This can only happen if the child application closes our dummy pipe
1340         // that is used to monitor its lifetime; in that case, our best bet is
1341         // to pretend the process did terminate, because otherwise wxExecute()
1342         // would hang indefinitely (OnReadWaiting() won't be called again, the
1343         // descriptor is closed now).
1344         wxLogDebug("Child process (PID %d) still alive but pipe closed so "
1345                    "generating a close notification", pid);
1346     }
1347     else if ( rc == -1 )
1348     {
1349         wxLogLastError(wxString::Format("waitpid(%d)", pid));
1350     }
1351     else // child did terminate
1352     {
1353         wxASSERT_MSG( rc == pid, "unexpected waitpid() return value" );
1354
1355         if ( WIFEXITED(status) )
1356             return WEXITSTATUS(status);
1357         else if ( WIFSIGNALED(status) )
1358             return -WTERMSIG(status);
1359         else
1360         {
1361             wxLogError("Child process (PID %d) exited for unknown reason, "
1362                        "status = %d", pid, status);
1363         }
1364     }
1365
1366     return -1;
1367 }
1368
1369 } // anonymous namespace
1370
1371 int wxAppTraits::WaitForChild(wxExecuteData& execData)
1372 {
1373     if ( !(execData.flags & wxEXEC_SYNC) )
1374     {
1375         // asynchronous execution: just launch the process and return,
1376         // endProcData will be destroyed when it terminates (currently we leak
1377         // it if the process doesn't terminate before we do and this should be
1378         // fixed but it's not a real leak so it's not really very high
1379         // priority)
1380         wxEndProcessData *endProcData = new wxEndProcessData;
1381         endProcData->process = execData.process;
1382         endProcData->pid = execData.pid;
1383         endProcData->tag = AddProcessCallback
1384                            (
1385                              endProcData,
1386                              execData.GetEndProcReadFD()
1387                            );
1388         endProcData->async = true;
1389
1390         return execData.pid;
1391     }
1392     //else: synchronous execution case
1393
1394 #if HAS_PIPE_INPUT_STREAM
1395     wxProcess * const process = execData.process;
1396     if ( process && process->IsRedirected() )
1397     {
1398         // we can't simply block waiting for the child to terminate as we would
1399         // dead lock if it writes more than the pipe buffer size (typically
1400         // 4KB) bytes of output -- it would then block waiting for us to read
1401         // the data while we'd block waiting for it to terminate
1402         //
1403         // so multiplex here waiting for any input from the child or closure of
1404         // the pipe used to indicate its termination
1405         wxSelectDispatcher disp;
1406
1407         wxEndHandler endHandler(disp, execData.GetEndProcReadFD());
1408
1409         wxRedirectedIOHandler outHandler(disp, execData.fdOut, execData.bufOut),
1410                               errHandler(disp, execData.fdErr, execData.bufErr);
1411
1412         while ( !endHandler.Terminated() )
1413         {
1414             disp.Dispatch();
1415         }
1416     }
1417     //else: no IO redirection, just block waiting for the child to exit
1418 #endif // HAS_PIPE_INPUT_STREAM
1419
1420     return DoWaitForChild(execData.pid);
1421 }
1422
1423 void wxHandleProcessTermination(wxEndProcessData *data)
1424 {
1425     data->exitcode = DoWaitForChild(data->pid, WNOHANG);
1426
1427     // notify user about termination if required
1428     if ( data->process )
1429     {
1430         data->process->OnTerminate(data->pid, data->exitcode);
1431     }
1432
1433     if ( data->async )
1434     {
1435         // in case of asynchronous execution we don't need this data any more
1436         // after the child terminates
1437         delete data;
1438     }
1439     else // sync execution
1440     {
1441         // let wxExecute() know that the process has terminated
1442         data->pid = 0;
1443     }
1444 }
1445