]> git.saurik.com Git - wxWidgets.git/blob - src/unix/utilsunx.cpp
cleaned up the mess caused by FloodFill patch
[wxWidgets.git] / src / unix / utilsunx.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: 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 #include "wx/defs.h"
19 #include "wx/string.h"
20
21 #include "wx/intl.h"
22 #include "wx/log.h"
23 #include "wx/app.h"
24
25 #include "wx/utils.h"
26 #include "wx/process.h"
27 #include "wx/thread.h"
28
29 #include "wx/stream.h"
30
31 #ifdef HAVE_STATFS
32 # ifdef __BSD__
33 # include <sys/param.h>
34 # include <sys/mount.h>
35 # else
36 # include <sys/vfs.h>
37 # endif
38 #endif // HAVE_STATFS
39
40 #ifdef HAVE_STATVFS
41 #include <sys/statvfs.h>
42
43 #define statfs statvfs
44 #endif // HAVE_STATVFS
45
46 #if wxUSE_GUI
47 #include "wx/unix/execute.h"
48 #endif
49
50 // SGI signal.h defines signal handler arguments differently depending on
51 // whether _LANGUAGE_C_PLUS_PLUS is set or not - do set it
52 #if defined(__SGI__) && !defined(_LANGUAGE_C_PLUS_PLUS)
53 #define _LANGUAGE_C_PLUS_PLUS 1
54 #endif // SGI hack
55
56 #include <stdarg.h>
57 #include <dirent.h>
58 #include <string.h>
59 #include <sys/stat.h>
60 #include <sys/types.h>
61 #include <unistd.h>
62 #include <sys/wait.h>
63 #include <pwd.h>
64 #include <errno.h>
65 #include <netdb.h>
66 #include <signal.h>
67 #include <fcntl.h> // for O_WRONLY and friends
68 #include <time.h> // nanosleep() and/or usleep()
69 #include <ctype.h> // isspace()
70 #include <sys/time.h> // needed for FD_SETSIZE
71
72 #ifdef HAVE_UNAME
73 #include <sys/utsname.h> // for uname()
74 #endif // HAVE_UNAME
75
76 // ----------------------------------------------------------------------------
77 // conditional compilation
78 // ----------------------------------------------------------------------------
79
80 // many versions of Unices have this function, but it is not defined in system
81 // headers - please add your system here if it is the case for your OS.
82 // SunOS < 5.6 (i.e. Solaris < 2.6) and DG-UX are like this.
83 #if !defined(HAVE_USLEEP) && \
84 (defined(__SUN__) && !defined(__SunOs_5_6) && \
85 !defined(__SunOs_5_7) && !defined(__SUNPRO_CC)) || \
86 defined(__osf__) || defined(__EMX__)
87 extern "C"
88 {
89 #ifdef __SUN__
90 int usleep(unsigned int usec);
91 #else // !Sun
92 #ifdef __EMX__
93 /* I copied this from the XFree86 diffs. AV. */
94 #define INCL_DOSPROCESS
95 #include <os2.h>
96 inline void usleep(unsigned long delay)
97 {
98 DosSleep(delay ? (delay/1000l) : 1l);
99 }
100 #else // !Sun && !EMX
101 void usleep(unsigned long usec);
102 #endif
103 #endif // Sun/EMX/Something else
104 };
105
106 #define HAVE_USLEEP 1
107 #endif // Unices without usleep()
108
109 // ============================================================================
110 // implementation
111 // ============================================================================
112
113 // ----------------------------------------------------------------------------
114 // sleeping
115 // ----------------------------------------------------------------------------
116
117 void wxSleep(int nSecs)
118 {
119 sleep(nSecs);
120 }
121
122 void wxUsleep(unsigned long milliseconds)
123 {
124 #if defined(HAVE_NANOSLEEP)
125 timespec tmReq;
126 tmReq.tv_sec = (time_t)(milliseconds / 1000);
127 tmReq.tv_nsec = (milliseconds % 1000) * 1000 * 1000;
128
129 // we're not interested in remaining time nor in return value
130 (void)nanosleep(&tmReq, (timespec *)NULL);
131 #elif defined(HAVE_USLEEP)
132 // uncomment this if you feel brave or if you are sure that your version
133 // of Solaris has a safe usleep() function but please notice that usleep()
134 // is known to lead to crashes in MT programs in Solaris 2.[67] and is not
135 // documented as MT-Safe
136 #if defined(__SUN__) && wxUSE_THREADS
137 #error "usleep() cannot be used in MT programs under Solaris."
138 #endif // Sun
139
140 usleep(milliseconds * 1000); // usleep(3) wants microseconds
141 #elif defined(HAVE_SLEEP)
142 // under BeOS sleep() takes seconds (what about other platforms, if any?)
143 sleep(milliseconds * 1000);
144 #else // !sleep function
145 #error "usleep() or nanosleep() function required for wxUsleep"
146 #endif // sleep function
147 }
148
149 // ----------------------------------------------------------------------------
150 // process management
151 // ----------------------------------------------------------------------------
152
153 int wxKill(long pid, wxSignal sig, wxKillError *rc)
154 {
155 int err = kill((pid_t)pid, (int)sig);
156 if ( rc )
157 {
158 switch ( errno )
159 {
160 case 0:
161 *rc = wxKILL_OK;
162 break;
163
164 case EINVAL:
165 *rc = wxKILL_BAD_SIGNAL;
166 break;
167
168 case EPERM:
169 *rc = wxKILL_ACCESS_DENIED;
170 break;
171
172 case ESRCH:
173 *rc = wxKILL_NO_PROCESS;
174 break;
175
176 default:
177 // this goes against Unix98 docs so log it
178 wxLogDebug(_T("unexpected kill(2) return value %d"), err);
179
180 // something else...
181 *rc = wxKILL_ERROR;
182 }
183 }
184
185 return err;
186 }
187
188 #define WXEXECUTE_NARGS 127
189
190 long wxExecute( const wxString& command, int flags, wxProcess *process )
191 {
192 wxCHECK_MSG( !command.IsEmpty(), 0, wxT("can't exec empty command") );
193
194 int argc = 0;
195 wxChar *argv[WXEXECUTE_NARGS];
196 wxString argument;
197 const wxChar *cptr = command.c_str();
198 wxChar quotechar = wxT('\0'); // is arg quoted?
199 bool escaped = FALSE;
200
201 // split the command line in arguments
202 do
203 {
204 argument=wxT("");
205 quotechar = wxT('\0');
206
207 // eat leading whitespace:
208 while ( wxIsspace(*cptr) )
209 cptr++;
210
211 if ( *cptr == wxT('\'') || *cptr == wxT('"') )
212 quotechar = *cptr++;
213
214 do
215 {
216 if ( *cptr == wxT('\\') && ! escaped )
217 {
218 escaped = TRUE;
219 cptr++;
220 continue;
221 }
222
223 // all other characters:
224 argument += *cptr++;
225 escaped = FALSE;
226
227 // have we reached the end of the argument?
228 if ( (*cptr == quotechar && ! escaped)
229 || (quotechar == wxT('\0') && wxIsspace(*cptr))
230 || *cptr == wxT('\0') )
231 {
232 wxASSERT_MSG( argc < WXEXECUTE_NARGS,
233 wxT("too many arguments in wxExecute") );
234
235 argv[argc] = new wxChar[argument.length() + 1];
236 wxStrcpy(argv[argc], argument.c_str());
237 argc++;
238
239 // if not at end of buffer, swallow last character:
240 if(*cptr)
241 cptr++;
242
243 break; // done with this one, start over
244 }
245 } while(*cptr);
246 } while(*cptr);
247 argv[argc] = NULL;
248
249 // do execute the command
250 long lRc = wxExecute(argv, flags, process);
251
252 // clean up
253 argc = 0;
254 while( argv[argc] )
255 delete [] argv[argc++];
256
257 return lRc;
258 }
259
260 // ----------------------------------------------------------------------------
261 // wxShell
262 // ----------------------------------------------------------------------------
263
264 static wxString wxMakeShellCommand(const wxString& command)
265 {
266 wxString cmd;
267 if ( !command )
268 {
269 // just an interactive shell
270 cmd = _T("xterm");
271 }
272 else
273 {
274 // execute command in a shell
275 cmd << _T("/bin/sh -c '") << command << _T('\'');
276 }
277
278 return cmd;
279 }
280
281 bool wxShell(const wxString& command)
282 {
283 return wxExecute(wxMakeShellCommand(command), wxEXEC_SYNC) == 0;
284 }
285
286 bool wxShell(const wxString& command, wxArrayString& output)
287 {
288 wxCHECK_MSG( !!command, FALSE, _T("can't exec shell non interactively") );
289
290 return wxExecute(wxMakeShellCommand(command), output);
291 }
292
293 #if wxUSE_GUI
294
295 void wxHandleProcessTermination(wxEndProcessData *proc_data)
296 {
297 // notify user about termination if required
298 if ( proc_data->process )
299 {
300 proc_data->process->OnTerminate(proc_data->pid, proc_data->exitcode);
301 }
302
303 // clean up
304 if ( proc_data->pid > 0 )
305 {
306 delete proc_data;
307 }
308 else
309 {
310 // let wxExecute() know that the process has terminated
311 proc_data->pid = 0;
312 }
313 }
314
315 #endif // wxUSE_GUI
316
317 // ----------------------------------------------------------------------------
318 // wxStream classes to support IO redirection in wxExecute
319 // ----------------------------------------------------------------------------
320
321 #if wxUSE_STREAMS
322
323 class wxProcessFileInputStream : public wxInputStream
324 {
325 public:
326 wxProcessFileInputStream(int fd) { m_fd = fd; }
327 ~wxProcessFileInputStream() { close(m_fd); }
328
329 virtual bool Eof() const;
330
331 protected:
332 size_t OnSysRead(void *buffer, size_t bufsize);
333
334 protected:
335 int m_fd;
336 };
337
338 class wxProcessFileOutputStream : public wxOutputStream
339 {
340 public:
341 wxProcessFileOutputStream(int fd) { m_fd = fd; }
342 ~wxProcessFileOutputStream() { close(m_fd); }
343
344 protected:
345 size_t OnSysWrite(const void *buffer, size_t bufsize);
346
347 protected:
348 int m_fd;
349 };
350
351 bool wxProcessFileInputStream::Eof() const
352 {
353 if ( m_lasterror == wxSTREAM_EOF )
354 return TRUE;
355
356 // check if there is any input available
357 struct timeval tv;
358 tv.tv_sec = 0;
359 tv.tv_usec = 0;
360
361 fd_set readfds;
362 FD_ZERO(&readfds);
363 FD_SET(m_fd, &readfds);
364 switch ( select(m_fd + 1, &readfds, NULL, NULL, &tv) )
365 {
366 case -1:
367 wxLogSysError(_("Impossible to get child process input"));
368 // fall through
369
370 case 0:
371 return TRUE;
372
373 default:
374 wxFAIL_MSG(_T("unexpected select() return value"));
375 // still fall through
376
377 case 1:
378 // input available: check if there is any
379 return wxInputStream::Eof();
380 }
381 }
382
383 size_t wxProcessFileInputStream::OnSysRead(void *buffer, size_t bufsize)
384 {
385 int ret = read(m_fd, buffer, bufsize);
386 if ( ret == 0 )
387 {
388 m_lasterror = wxSTREAM_EOF;
389 }
390 else if ( ret == -1 )
391 {
392 m_lasterror = wxSTREAM_READ_ERROR;
393 ret = 0;
394 }
395 else
396 {
397 m_lasterror = wxSTREAM_NOERROR;
398 }
399
400 return ret;
401 }
402
403 size_t wxProcessFileOutputStream::OnSysWrite(const void *buffer, size_t bufsize)
404 {
405 int ret = write(m_fd, buffer, bufsize);
406 if ( ret == -1 )
407 {
408 m_lasterror = wxSTREAM_WRITE_ERROR;
409 ret = 0;
410 }
411 else
412 {
413 m_lasterror = wxSTREAM_NOERROR;
414 }
415
416 return ret;
417 }
418
419 // ----------------------------------------------------------------------------
420 // wxStreamTempBuffer
421 // ----------------------------------------------------------------------------
422
423 /*
424 Extract of a mail to wx-users to give the context of the problem we are
425 trying to solve here:
426
427 MC> If I run the command:
428 MC> find . -name "*.h" -exec grep linux {} \;
429 MC> in the exec sample synchronously from the 'Capture command output'
430 MC> menu, wxExecute never returns. I have to xkill it. Has anyone
431 MC> else encountered this?
432
433 Yes, I can reproduce it too.
434
435 I even think I understand why it happens: before launching the external
436 command we set up a pipe with a valid file descriptor on the reading side
437 when the output is redirected. So the subprocess happily writes to it ...
438 until the pipe buffer (which is usually quite big on Unix, I think the
439 default is 4Mb) is full. Then the writing process stops and waits until we
440 read some data from the pipe to be able to continue writing to it but we
441 never do it because we wait until it terminates to start reading and so we
442 have a classical deadlock.
443
444 Here is the fix: we now read the output as soon as it appears into a temp
445 buffer (wxStreamTempBuffer object) and later just stuff it back into the
446 stream when the process terminates. See supporting code in wxExecute()
447 itself as well.
448 */
449
450 class wxStreamTempBuffer
451 {
452 public:
453 wxStreamTempBuffer();
454
455 // call to associate a stream with this buffer, otherwise nothing happens
456 // at all
457 void Init(wxInputStream *stream);
458
459 // check for input on our stream and cache it in our buffer if any
460 void Update();
461
462 ~wxStreamTempBuffer();
463
464 private:
465 // the stream we're buffering, if NULL we don't do anything at all
466 wxInputStream *m_stream;
467
468 // the buffer of size m_size (NULL if m_size == 0)
469 void *m_buffer;
470
471 // the size of the buffer
472 size_t m_size;
473 };
474
475 wxStreamTempBuffer::wxStreamTempBuffer()
476 {
477 m_stream = NULL;
478 m_buffer = NULL;
479 m_size = 0;
480 }
481
482 void wxStreamTempBuffer::Init(wxInputStream *stream)
483 {
484 m_stream = stream;
485 }
486
487 void wxStreamTempBuffer::Update()
488 {
489 if ( m_stream && !m_stream->Eof() )
490 {
491 // realloc in blocks of 1Kb - surely not the best strategy but which
492 // one is?
493 static const size_t incSize = 1024;
494
495 void *buf = realloc(m_buffer, m_size + incSize);
496 if ( !buf )
497 {
498 // don't read any more, we don't have enough memory to do it
499 m_stream = NULL;
500 }
501 else // got memory for the buffer
502 {
503 m_buffer = buf;
504 m_stream->Read((char *)m_buffer + m_size, incSize);
505 m_size += incSize;
506 }
507 }
508 }
509
510 wxStreamTempBuffer::~wxStreamTempBuffer()
511 {
512 if ( m_buffer )
513 {
514 m_stream->Ungetch(m_buffer, m_size);
515 free(m_buffer);
516 }
517 }
518
519 #endif // wxUSE_STREAMS
520
521 long wxExecute(wxChar **argv,
522 int flags,
523 wxProcess *process)
524 {
525 // for the sync execution, we return -1 to indicate failure, but for async
526 // case we return 0 which is never a valid PID
527 //
528 // we define this as a macro, not a variable, to avoid compiler warnings
529 // about "ERROR_RETURN_CODE value may be clobbered by fork()"
530 #define ERROR_RETURN_CODE ((flags & wxEXEC_SYNC) ? -1 : 0)
531
532 wxCHECK_MSG( *argv, ERROR_RETURN_CODE, wxT("can't exec empty command") );
533
534 #if wxUSE_UNICODE
535 int mb_argc = 0;
536 char *mb_argv[WXEXECUTE_NARGS];
537
538 while (argv[mb_argc])
539 {
540 wxWX2MBbuf mb_arg = wxConvertWX2MB(argv[mb_argc]);
541 mb_argv[mb_argc] = strdup(mb_arg);
542 mb_argc++;
543 }
544 mb_argv[mb_argc] = (char *) NULL;
545
546 // this macro will free memory we used above
547 #define ARGS_CLEANUP \
548 for ( mb_argc = 0; mb_argv[mb_argc]; mb_argc++ ) \
549 free(mb_argv[mb_argc])
550 #else // ANSI
551 // no need for cleanup
552 #define ARGS_CLEANUP
553
554 wxChar **mb_argv = argv;
555 #endif // Unicode/ANSI
556
557 #if wxUSE_GUI
558 // create pipes
559 int end_proc_detect[2];
560 if ( pipe(end_proc_detect) == -1 )
561 {
562 wxLogSysError( _("Pipe creation failed") );
563 wxLogError( _("Failed to execute '%s'\n"), *argv );
564
565 ARGS_CLEANUP;
566
567 return ERROR_RETURN_CODE;
568 }
569 #endif // wxUSE_GUI
570
571 // pipes for inter process communication
572 int pipeIn[2], // stdin
573 pipeOut[2], // stdout
574 pipeErr[2]; // stderr
575
576 pipeIn[0] = pipeIn[1] =
577 pipeOut[0] = pipeOut[1] =
578 pipeErr[0] = pipeErr[1] = -1;
579
580 if ( process && process->IsRedirected() )
581 {
582 if ( pipe(pipeIn) == -1 || pipe(pipeOut) == -1 || pipe(pipeErr) == -1 )
583 {
584 #if wxUSE_GUI
585 // free previously allocated resources
586 close(end_proc_detect[0]);
587 close(end_proc_detect[1]);
588 #endif // wxUSE_GUI
589
590 wxLogSysError( _("Pipe creation failed") );
591 wxLogError( _("Failed to execute '%s'\n"), *argv );
592
593 ARGS_CLEANUP;
594
595 return ERROR_RETURN_CODE;
596 }
597 }
598
599 // fork the process
600 #ifdef HAVE_VFORK
601 pid_t pid = vfork();
602 #else
603 pid_t pid = fork();
604 #endif
605
606 if ( pid == -1 ) // error?
607 {
608 #if wxUSE_GUI
609 close(end_proc_detect[0]);
610 close(end_proc_detect[1]);
611 close(pipeIn[0]);
612 close(pipeIn[1]);
613 close(pipeOut[0]);
614 close(pipeOut[1]);
615 close(pipeErr[0]);
616 close(pipeErr[1]);
617 #endif // wxUSE_GUI
618
619 wxLogSysError( _("Fork failed") );
620
621 ARGS_CLEANUP;
622
623 return ERROR_RETURN_CODE;
624 }
625 else if ( pid == 0 ) // we're in child
626 {
627 #if wxUSE_GUI
628 close(end_proc_detect[0]); // close reading side
629 #endif // wxUSE_GUI
630
631 // These lines close the open file descriptors to to avoid any
632 // input/output which might block the process or irritate the user. If
633 // one wants proper IO for the subprocess, the right thing to do is to
634 // start an xterm executing it.
635 if ( !(flags & wxEXEC_SYNC) )
636 {
637 for ( int fd = 0; fd < FD_SETSIZE; fd++ )
638 {
639 if ( fd == pipeIn[0] || fd == pipeOut[1] || fd == pipeErr[1]
640 #if wxUSE_GUI
641 || fd == end_proc_detect[1]
642 #endif // wxUSE_GUI
643 )
644 {
645 // don't close this one, we still need it
646 continue;
647 }
648
649 // leave stderr opened too, it won't do any hurm
650 if ( fd != STDERR_FILENO )
651 close(fd);
652 }
653
654 #ifndef __VMS
655 if ( flags & wxEXEC_MAKE_GROUP_LEADER )
656 {
657 // Set process group to child process' pid. Then killing -pid
658 // of the parent will kill the process and all of its children.
659 setsid();
660 }
661 #endif
662 }
663
664 // redirect stdio, stdout and stderr
665 if ( pipeIn[0] != -1 )
666 {
667 if ( dup2(pipeIn[0], STDIN_FILENO) == -1 ||
668 dup2(pipeOut[1], STDOUT_FILENO) == -1 ||
669 dup2(pipeErr[1], STDERR_FILENO) == -1 )
670 {
671 wxLogSysError(_("Failed to redirect child process input/output"));
672 }
673
674 close(pipeIn[0]);
675 close(pipeOut[1]);
676 close(pipeErr[1]);
677 }
678
679 execvp (*mb_argv, mb_argv);
680
681 // there is no return after successful exec()
682 _exit(-1);
683
684 // some compilers complain about missing return - of course, they
685 // should know that exit() doesn't return but what else can we do if
686 // they don't?
687 #if defined(__VMS) || defined(__INTEL_COMPILER)
688 return 0;
689 #endif
690 }
691 else // we're in parent
692 {
693 ARGS_CLEANUP;
694
695 // pipe initialization: construction of the wxStreams
696 #if wxUSE_STREAMS
697 wxStreamTempBuffer bufIn, bufErr;
698 #endif // wxUSE_STREAMS
699
700 if ( process && process->IsRedirected() )
701 {
702 #if wxUSE_STREAMS
703 // in/out for subprocess correspond to our out/in
704 wxOutputStream *outStream = new wxProcessFileOutputStream(pipeIn[1]);
705 wxInputStream *inStream = new wxProcessFileInputStream(pipeOut[0]);
706 wxInputStream *errStream = new wxProcessFileInputStream(pipeErr[0]);
707
708 process->SetPipeStreams(inStream, outStream, errStream);
709
710 bufIn.Init(inStream);
711 bufErr.Init(inStream);
712 #endif // wxUSE_STREAMS
713
714 close(pipeIn[0]); // close reading side
715 close(pipeOut[1]); // close writing side
716 close(pipeErr[1]); // close writing side
717 }
718
719 #if wxUSE_GUI && !defined(__WXMICROWIN__)
720 wxEndProcessData *data = new wxEndProcessData;
721
722 if ( flags & wxEXEC_SYNC )
723 {
724 // we may have process for capturing the program output, but it's
725 // not used in wxEndProcessData in the case of sync execution
726 data->process = NULL;
727
728 // sync execution: indicate it by negating the pid
729 data->pid = -pid;
730 data->tag = wxAddProcessCallback(data, end_proc_detect[0]);
731
732 close(end_proc_detect[1]); // close writing side
733
734 wxBusyCursor bc;
735 wxWindowDisabler wd;
736
737 // data->pid will be set to 0 from GTK_EndProcessDetector when the
738 // process terminates
739 while ( data->pid != 0 )
740 {
741 #if wxUSE_STREAMS
742 bufIn.Update();
743 bufErr.Update();
744 #endif // wxUSE_STREAMS
745
746 // give GTK+ a chance to call GTK_EndProcessDetector here and
747 // also repaint the GUI
748 wxYield();
749 }
750
751 int exitcode = data->exitcode;
752
753 delete data;
754
755 return exitcode;
756 }
757 else // async execution
758 {
759 // async execution, nothing special to do - caller will be
760 // notified about the process termination if process != NULL, data
761 // will be deleted in GTK_EndProcessDetector
762 data->process = process;
763 data->pid = pid;
764 data->tag = wxAddProcessCallback(data, end_proc_detect[0]);
765
766 close(end_proc_detect[1]); // close writing side
767
768 return pid;
769 }
770 #else // !wxUSE_GUI
771
772 wxASSERT_MSG( flags & wxEXEC_SYNC,
773 wxT("async execution not supported yet") );
774
775 int exitcode = 0;
776 if ( waitpid(pid, &exitcode, 0) == -1 || !WIFEXITED(exitcode) )
777 {
778 wxLogSysError(_("Waiting for subprocess termination failed"));
779 }
780
781 return exitcode;
782 #endif // wxUSE_GUI
783 }
784 }
785
786 #undef ERROR_RETURN_CODE
787 #undef ARGS_CLEANUP
788
789 // ----------------------------------------------------------------------------
790 // file and directory functions
791 // ----------------------------------------------------------------------------
792
793 const wxChar* wxGetHomeDir( wxString *home )
794 {
795 *home = wxGetUserHome( wxString() );
796 wxString tmp;
797 if ( home->IsEmpty() )
798 *home = wxT("/");
799 #ifdef __VMS
800 tmp = *home;
801 if ( tmp.Last() != wxT(']'))
802 if ( tmp.Last() != wxT('/')) *home << wxT('/');
803 #endif
804 return home->c_str();
805 }
806
807 #if wxUSE_UNICODE
808 const wxMB2WXbuf wxGetUserHome( const wxString &user )
809 #else // just for binary compatibility -- there is no 'const' here
810 char *wxGetUserHome( const wxString &user )
811 #endif
812 {
813 struct passwd *who = (struct passwd *) NULL;
814
815 if ( !user )
816 {
817 wxChar *ptr;
818
819 if ((ptr = wxGetenv(wxT("HOME"))) != NULL)
820 {
821 return ptr;
822 }
823 if ((ptr = wxGetenv(wxT("USER"))) != NULL || (ptr = wxGetenv(wxT("LOGNAME"))) != NULL)
824 {
825 who = getpwnam(wxConvertWX2MB(ptr));
826 }
827
828 // We now make sure the the user exists!
829 if (who == NULL)
830 {
831 who = getpwuid(getuid());
832 }
833 }
834 else
835 {
836 who = getpwnam (user.mb_str());
837 }
838
839 return wxConvertMB2WX(who ? who->pw_dir : 0);
840 }
841
842 // ----------------------------------------------------------------------------
843 // network and user id routines
844 // ----------------------------------------------------------------------------
845
846 // retrieve either the hostname or FQDN depending on platform (caller must
847 // check whether it's one or the other, this is why this function is for
848 // private use only)
849 static bool wxGetHostNameInternal(wxChar *buf, int sz)
850 {
851 wxCHECK_MSG( buf, FALSE, wxT("NULL pointer in wxGetHostNameInternal") );
852
853 *buf = wxT('\0');
854
855 // we're using uname() which is POSIX instead of less standard sysinfo()
856 #if defined(HAVE_UNAME)
857 struct utsname uts;
858 bool ok = uname(&uts) != -1;
859 if ( ok )
860 {
861 wxStrncpy(buf, wxConvertMB2WX(uts.nodename), sz - 1);
862 buf[sz] = wxT('\0');
863 }
864 #elif defined(HAVE_GETHOSTNAME)
865 bool ok = gethostname(buf, sz) != -1;
866 #else // no uname, no gethostname
867 wxFAIL_MSG(wxT("don't know host name for this machine"));
868
869 bool ok = FALSE;
870 #endif // uname/gethostname
871
872 if ( !ok )
873 {
874 wxLogSysError(_("Cannot get the hostname"));
875 }
876
877 return ok;
878 }
879
880 bool wxGetHostName(wxChar *buf, int sz)
881 {
882 bool ok = wxGetHostNameInternal(buf, sz);
883
884 if ( ok )
885 {
886 // BSD systems return the FQDN, we only want the hostname, so extract
887 // it (we consider that dots are domain separators)
888 wxChar *dot = wxStrchr(buf, wxT('.'));
889 if ( dot )
890 {
891 // nuke it
892 *dot = wxT('\0');
893 }
894 }
895
896 return ok;
897 }
898
899 bool wxGetFullHostName(wxChar *buf, int sz)
900 {
901 bool ok = wxGetHostNameInternal(buf, sz);
902
903 if ( ok )
904 {
905 if ( !wxStrchr(buf, wxT('.')) )
906 {
907 struct hostent *host = gethostbyname(wxConvertWX2MB(buf));
908 if ( !host )
909 {
910 wxLogSysError(_("Cannot get the official hostname"));
911
912 ok = FALSE;
913 }
914 else
915 {
916 // the canonical name
917 wxStrncpy(buf, wxConvertMB2WX(host->h_name), sz);
918 }
919 }
920 //else: it's already a FQDN (BSD behaves this way)
921 }
922
923 return ok;
924 }
925
926 bool wxGetUserId(wxChar *buf, int sz)
927 {
928 struct passwd *who;
929
930 *buf = wxT('\0');
931 if ((who = getpwuid(getuid ())) != NULL)
932 {
933 wxStrncpy (buf, wxConvertMB2WX(who->pw_name), sz - 1);
934 return TRUE;
935 }
936
937 return FALSE;
938 }
939
940 bool wxGetUserName(wxChar *buf, int sz)
941 {
942 struct passwd *who;
943
944 *buf = wxT('\0');
945 if ((who = getpwuid (getuid ())) != NULL)
946 {
947 // pw_gecos field in struct passwd is not standard
948 #ifdef HAVE_PW_GECOS
949 char *comma = strchr(who->pw_gecos, ',');
950 if (comma)
951 *comma = '\0'; // cut off non-name comment fields
952 wxStrncpy (buf, wxConvertMB2WX(who->pw_gecos), sz - 1);
953 #else // !HAVE_PW_GECOS
954 wxStrncpy (buf, wxConvertMB2WX(who->pw_name), sz - 1);
955 #endif // HAVE_PW_GECOS/!HAVE_PW_GECOS
956 return TRUE;
957 }
958
959 return FALSE;
960 }
961
962 #ifndef __WXMAC__
963 wxString wxGetOsDescription()
964 {
965 #ifndef WXWIN_OS_DESCRIPTION
966 #error WXWIN_OS_DESCRIPTION should be defined in config.h by configure
967 #else
968 return WXWIN_OS_DESCRIPTION;
969 #endif
970 }
971 #endif
972
973 // this function returns the GUI toolkit version in GUI programs, but OS
974 // version in non-GUI ones
975 #if !wxUSE_GUI
976
977 int wxGetOsVersion(int *majorVsn, int *minorVsn)
978 {
979 int major, minor;
980 char name[256];
981
982 if ( sscanf(WXWIN_OS_DESCRIPTION, "%s %d.%d", name, &major, &minor) != 3 )
983 {
984 // unreckognized uname string format
985 major = minor = -1;
986 }
987
988 if ( majorVsn )
989 *majorVsn = major;
990 if ( minorVsn )
991 *minorVsn = minor;
992
993 return wxUNIX;
994 }
995
996 #endif // !wxUSE_GUI
997
998 long wxGetFreeMemory()
999 {
1000 #if defined(__LINUX__)
1001 // get it from /proc/meminfo
1002 FILE *fp = fopen("/proc/meminfo", "r");
1003 if ( fp )
1004 {
1005 long memFree = -1;
1006
1007 char buf[1024];
1008 if ( fgets(buf, WXSIZEOF(buf), fp) && fgets(buf, WXSIZEOF(buf), fp) )
1009 {
1010 long memTotal, memUsed;
1011 sscanf(buf, "Mem: %ld %ld %ld", &memTotal, &memUsed, &memFree);
1012 }
1013
1014 fclose(fp);
1015
1016 return memFree;
1017 }
1018 #elif defined(__SUN__) && defined(_SC_AVPHYS_PAGES)
1019 return sysconf(_SC_AVPHYS_PAGES)*sysconf(_SC_PAGESIZE);
1020 //#elif defined(__FREEBSD__) -- might use sysctl() to find it out, probably
1021 #endif
1022
1023 // can't find it out
1024 return -1;
1025 }
1026
1027 bool wxGetDiskSpace(const wxString& path, wxLongLong *pTotal, wxLongLong *pFree)
1028 {
1029 #if defined(HAVE_STATFS) || defined(HAVE_STATVFS)
1030 // the case to "char *" is needed for AIX 4.3
1031 struct statfs fs;
1032 if ( statfs((char *)path.fn_str(), &fs) != 0 )
1033 {
1034 wxLogSysError("Failed to get file system statistics");
1035
1036 return FALSE;
1037 }
1038
1039 // under Solaris we might have to use fs.f_frsize instead as I think it
1040 // may be a multiple of the block size in general (TODO)
1041
1042 if ( pTotal )
1043 {
1044 *pTotal = wxLongLong(fs.f_blocks) * fs.f_bsize;
1045 }
1046
1047 if ( pFree )
1048 {
1049 *pFree = wxLongLong(fs.f_bavail) * fs.f_bsize;
1050 }
1051
1052 return TRUE;
1053 #endif // HAVE_STATFS
1054
1055 return FALSE;
1056 }
1057
1058 // ----------------------------------------------------------------------------
1059 // env vars
1060 // ----------------------------------------------------------------------------
1061
1062 bool wxGetEnv(const wxString& var, wxString *value)
1063 {
1064 // wxGetenv is defined as getenv()
1065 wxChar *p = wxGetenv(var);
1066 if ( !p )
1067 return FALSE;
1068
1069 if ( value )
1070 {
1071 *value = p;
1072 }
1073
1074 return TRUE;
1075 }
1076
1077 bool wxSetEnv(const wxString& variable, const wxChar *value)
1078 {
1079 #if defined(HAVE_SETENV)
1080 return setenv(variable.mb_str(),
1081 value ? (const char *)wxString(value).mb_str()
1082 : NULL,
1083 1 /* overwrite */) == 0;
1084 #elif defined(HAVE_PUTENV)
1085 wxString s = variable;
1086 if ( value )
1087 s << _T('=') << value;
1088
1089 // transform to ANSI
1090 const char *p = s.mb_str();
1091
1092 // the string will be free()d by libc
1093 char *buf = (char *)malloc(strlen(p) + 1);
1094 strcpy(buf, p);
1095
1096 return putenv(buf) == 0;
1097 #else // no way to set an env var
1098 return FALSE;
1099 #endif
1100 }
1101
1102 // ----------------------------------------------------------------------------
1103 // signal handling
1104 // ----------------------------------------------------------------------------
1105
1106 #if wxUSE_ON_FATAL_EXCEPTION
1107
1108 #include <signal.h>
1109
1110 extern "C" void wxFatalSignalHandler(wxTYPE_SA_HANDLER)
1111 {
1112 if ( wxTheApp )
1113 {
1114 // give the user a chance to do something special about this
1115 wxTheApp->OnFatalException();
1116 }
1117
1118 abort();
1119 }
1120
1121 bool wxHandleFatalExceptions(bool doit)
1122 {
1123 // old sig handlers
1124 static bool s_savedHandlers = FALSE;
1125 static struct sigaction s_handlerFPE,
1126 s_handlerILL,
1127 s_handlerBUS,
1128 s_handlerSEGV;
1129
1130 bool ok = TRUE;
1131 if ( doit && !s_savedHandlers )
1132 {
1133 // install the signal handler
1134 struct sigaction act;
1135
1136 // some systems extend it with non std fields, so zero everything
1137 memset(&act, 0, sizeof(act));
1138
1139 act.sa_handler = wxFatalSignalHandler;
1140 sigemptyset(&act.sa_mask);
1141 act.sa_flags = 0;
1142
1143 ok &= sigaction(SIGFPE, &act, &s_handlerFPE) == 0;
1144 ok &= sigaction(SIGILL, &act, &s_handlerILL) == 0;
1145 ok &= sigaction(SIGBUS, &act, &s_handlerBUS) == 0;
1146 ok &= sigaction(SIGSEGV, &act, &s_handlerSEGV) == 0;
1147 if ( !ok )
1148 {
1149 wxLogDebug(_T("Failed to install our signal handler."));
1150 }
1151
1152 s_savedHandlers = TRUE;
1153 }
1154 else if ( s_savedHandlers )
1155 {
1156 // uninstall the signal handler
1157 ok &= sigaction(SIGFPE, &s_handlerFPE, NULL) == 0;
1158 ok &= sigaction(SIGILL, &s_handlerILL, NULL) == 0;
1159 ok &= sigaction(SIGBUS, &s_handlerBUS, NULL) == 0;
1160 ok &= sigaction(SIGSEGV, &s_handlerSEGV, NULL) == 0;
1161 if ( !ok )
1162 {
1163 wxLogDebug(_T("Failed to uninstall our signal handler."));
1164 }
1165
1166 s_savedHandlers = FALSE;
1167 }
1168 //else: nothing to do
1169
1170 return ok;
1171 }
1172
1173 #endif // wxUSE_ON_FATAL_EXCEPTION
1174
1175 // ----------------------------------------------------------------------------
1176 // error and debug output routines (deprecated, use wxLog)
1177 // ----------------------------------------------------------------------------
1178
1179 #if WXWIN_COMPATIBILITY_2_2
1180
1181 void wxDebugMsg( const char *format, ... )
1182 {
1183 va_list ap;
1184 va_start( ap, format );
1185 vfprintf( stderr, format, ap );
1186 fflush( stderr );
1187 va_end(ap);
1188 }
1189
1190 void wxError( const wxString &msg, const wxString &title )
1191 {
1192 wxFprintf( stderr, _("Error ") );
1193 if (!title.IsNull()) wxFprintf( stderr, wxT("%s "), WXSTRINGCAST(title) );
1194 if (!msg.IsNull()) wxFprintf( stderr, wxT(": %s"), WXSTRINGCAST(msg) );
1195 wxFprintf( stderr, wxT(".\n") );
1196 }
1197
1198 void wxFatalError( const wxString &msg, const wxString &title )
1199 {
1200 wxFprintf( stderr, _("Error ") );
1201 if (!title.IsNull()) wxFprintf( stderr, wxT("%s "), WXSTRINGCAST(title) );
1202 if (!msg.IsNull()) wxFprintf( stderr, wxT(": %s"), WXSTRINGCAST(msg) );
1203 wxFprintf( stderr, wxT(".\n") );
1204 exit(3); // the same exit code as for abort()
1205 }
1206
1207 #endif // WXWIN_COMPATIBILITY_2_2
1208