src/mac/corefoundation/utilsexc_cf.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/mac/corefoundation/utilsexec_cf.cpp
   3 // Purpose:     Execution-related utilities for Darwin
   4 // Author:      David Elliott, Ryan Norton (wxMacExecute)
   5 // Modified by: Stefan Csomor (added necessary wxT for unicode builds)
   6 // Created:     2004-11-04
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) David Elliott, Ryan Norton
   9 // Licence:     wxWindows licence
  10 // Notes:       This code comes from src/mac/carbon/utilsexc.cpp,1.11
  11 /////////////////////////////////////////////////////////////////////////////
  12
  13 #include "wx/wxprec.h"
  14 #ifndef WX_PRECOMP
  15     #include "wx/log.h"
  16     #include "wx/utils.h"
  17 #endif //ndef WX_PRECOMP
  18 #include "wx/unix/execute.h"
  19 #include "wx/stdpaths.h"
  20 #include "wx/apptrait.h"
  21 #include "wx/thread.h"
  22 #include "wx/process.h"
  23
  24 #include <sys/wait.h>
  25
  26 // Use polling instead of Mach ports, which doesn't work on Intel
  27 // due to task_for_pid security issues.
  28
  29 // What's a better test for Intel vs PPC?
  30 #ifdef WORDS_BIGENDIAN
  31 #define USE_POLLING 0
  32 #else
  33 #define USE_POLLING 1
  34 #endif
  35
  36 #if USE_POLLING
  37
  38 #if wxUSE_THREADS
  39 class wxProcessTerminationEventHandler: public wxEvtHandler
  40 {
  41   public:
  42     wxProcessTerminationEventHandler(wxEndProcessData* data)
  43     {
  44         m_data = data;
  45         Connect(-1, wxEVT_END_PROCESS, wxProcessEventHandler(wxProcessTerminationEventHandler::OnTerminate));
  46     }
  47
  48     void OnTerminate(wxProcessEvent& event)
  49     {
  50         Disconnect(-1, wxEVT_END_PROCESS, wxProcessEventHandler(wxProcessTerminationEventHandler::OnTerminate));
  51         wxHandleProcessTermination(m_data);
  52         delete this;
  53     }
  54
  55     wxEndProcessData* m_data;
  56 };
  57
  58 class wxProcessTerminationThread: public wxThread
  59 {
  60   public:
  61     wxProcessTerminationThread(wxEndProcessData* data, wxProcessTerminationEventHandler* handler): wxThread(wxTHREAD_DETACHED)
  62     {
  63         m_data = data;
  64         m_handler = handler;
  65     }
  66
  67     virtual void* Entry();
  68
  69     wxProcessTerminationEventHandler* m_handler;
  70     wxEndProcessData* m_data;
  71 };
  72
  73 // The problem with this is that we may be examining the
  74 // process e.g. in OnIdle at the point this cleans up the process,
  75 // so we need to delay until it's safe.
  76
  77 void* wxProcessTerminationThread::Entry()
  78 {
  79     while (true)
  80     {
  81         usleep(100);
  82         int status = 0;
  83         int rc = waitpid(abs(m_data->pid), & status, 0);
  84         if (rc != 0)
  85         {
  86             if ((rc != -1) && WIFEXITED(status))
  87                 m_data->exitcode = WEXITSTATUS(status);
  88             else
  89                 m_data->exitcode = -1;
  90
  91             wxProcessEvent event;
  92             wxPostEvent(m_handler, event);
  93
  94             break;
  95         }
  96     }
  97
  98     return NULL;
  99 }
 100
 101 int wxAddProcessCallbackForPid(wxEndProcessData *proc_data, int pid)
 102 {
 103     if (pid < 1)
 104         return -1;
 105
 106     wxProcessTerminationEventHandler* handler = new wxProcessTerminationEventHandler(proc_data);
 107     wxProcessTerminationThread* thread = new wxProcessTerminationThread(proc_data, handler);
 108
 109     if (thread->Create() != wxTHREAD_NO_ERROR)
 110     {
 111         wxLogDebug(wxT("Could not create termination detection thread."));
 112         delete thread;
 113         delete handler;
 114         return -1;
 115     }
 116
 117     thread->Run();
 118
 119     return 0;
 120 }
 121 #else // !wxUSE_THREADS
 122 int wxAddProcessCallbackForPid(wxEndProcessData*, int)
 123 {
 124     wxLogDebug(wxT("Could not create termination detection thread."));
 125     return -1;
 126 }
 127 #endif // wxUSE_THREADS/!wxUSE_THREADS
 128
 129 #else // !USE_POLLING
 130
 131 #include <CoreFoundation/CFMachPort.h>
 132 extern "C" {
 133 #include <mach/mach.h>
 134 }
 135
 136 void wxMAC_MachPortEndProcessDetect(CFMachPortRef port, void *data)
 137 {
 138     wxEndProcessData *proc_data = (wxEndProcessData*)data;
 139     wxLogDebug(wxT("Process ended"));
 140     int status = 0;
 141     int rc = waitpid(abs(proc_data->pid), &status, WNOHANG);
 142     if(!rc)
 143     {
 144         wxLogDebug(wxT("Mach port was invalidated, but process hasn't terminated!"));
 145         return;
 146     }
 147     if((rc != -1) && WIFEXITED(status))
 148         proc_data->exitcode = WEXITSTATUS(status);
 149     else
 150         proc_data->exitcode = -1;
 151     wxHandleProcessTermination(proc_data);
 152 }
 153
 154 int wxAddProcessCallbackForPid(wxEndProcessData *proc_data, int pid)
 155 {
 156     if(pid < 1)
 157         return -1;
 158     kern_return_t    kernResult;
 159     mach_port_t    taskOfOurProcess;
 160     mach_port_t    machPortForProcess;
 161     taskOfOurProcess = mach_task_self();
 162     if(taskOfOurProcess == MACH_PORT_NULL)
 163     {
 164         wxLogDebug(wxT("No mach_task_self()"));
 165         return -1;
 166     }
 167     wxLogDebug(wxT("pid=%d"),pid);
 168     kernResult = task_for_pid(taskOfOurProcess,pid, &machPortForProcess);
 169     if(kernResult != KERN_SUCCESS)
 170     {
 171         wxLogDebug(wxT("no task_for_pid()"));
 172         // try seeing if it is already dead or something
 173         // FIXME: a better method would be to call the callback function
 174         // from idle time until the process terminates. Of course, how
 175         // likely is it that it will take more than 0.1 seconds for the
 176         // mach terminate event to make its way to the BSD subsystem?
 177         usleep(100); // sleep for 0.1 seconds
 178         wxMAC_MachPortEndProcessDetect(NULL, (void*)proc_data);
 179         return -1;
 180     }
 181     CFMachPortContext termcb_contextinfo;
 182     termcb_contextinfo.version = 0;
 183     termcb_contextinfo.info = (void*)proc_data;
 184     termcb_contextinfo.retain = NULL;
 185     termcb_contextinfo.release = NULL;
 186     termcb_contextinfo.copyDescription = NULL;
 187     CFMachPortRef    CFMachPortForProcess;
 188     Boolean        ShouldFreePort;
 189     CFMachPortForProcess = CFMachPortCreateWithPort(NULL, machPortForProcess, NULL, &termcb_contextinfo, &ShouldFreePort);
 190     if(!CFMachPortForProcess)
 191     {
 192         wxLogDebug(wxT("No CFMachPortForProcess"));
 193         mach_port_deallocate(taskOfOurProcess, machPortForProcess);
 194         return -1;
 195     }
 196     if(ShouldFreePort)
 197     {
 198         kernResult = mach_port_deallocate(taskOfOurProcess, machPortForProcess);
 199         if(kernResult!=KERN_SUCCESS)
 200         {
 201             wxLogDebug(wxT("Couldn't deallocate mach port"));
 202             return -1;
 203         }
 204     }
 205     CFMachPortSetInvalidationCallBack(CFMachPortForProcess, &wxMAC_MachPortEndProcessDetect);
 206     CFRunLoopSourceRef    runloopsource;
 207     runloopsource = CFMachPortCreateRunLoopSource(NULL,CFMachPortForProcess, (CFIndex)0);
 208     if(!runloopsource)
 209     {
 210         wxLogDebug(wxT("Couldn't create runloopsource"));
 211         return -1;
 212     }
 213
 214     CFRelease(CFMachPortForProcess);
 215
 216     CFRunLoopAddSource(CFRunLoopGetCurrent(),runloopsource,kCFRunLoopDefaultMode);
 217     CFRelease(runloopsource);
 218     wxLogDebug(wxT("Successfully added notification to the runloop"));
 219     return 0;
 220 }
 221
 222 #endif // USE_POLLING/!USE_POLLING
 223
 224 // NOTE: This doesn't really belong here but this was a handy file to
 225 // put it in because it's already compiled for wxCocoa and wxMac GUI lib.
 226 #if wxUSE_GUI
 227
 228 static wxStandardPathsCF gs_stdPaths;
 229 wxStandardPathsBase& wxGUIAppTraits::GetStandardPaths()
 230 {
 231     return gs_stdPaths;
 232 }
 233
 234 #endif // wxUSE_GUI
 235