Mac/Tools/pythonw.c

   1 /*
   2  * This wrapper program executes a python executable hidden inside an
   3  * application bundle inside the Python framework. This is needed to run
   4  * GUI code: some GUI API's don't work unless the program is inside an
   5  * application bundle.
   6  *
   7  * This program uses posix_spawn rather than plain execv because we need
   8  * slightly more control over how the "real" interpreter is executed.
   9  *
  10  * On OSX 10.4 (and earlier) this falls back to using exec because the
  11  * posix_spawnv functions aren't available there.
  12  */
  13 #pragma weak_import posix_spawnattr_init
  14 #pragma weak_import posix_spawnattr_setbinpref_np
  15 #pragma weak_import posix_spawnattr_setflags
  16 #pragma weak_import posix_spawn
  17
  18 #include <Python.h>
  19 #include <unistd.h>
  20 #ifdef HAVE_SPAWN_H
  21 #include <spawn.h>
  22 #endif
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include <err.h>
  27 #include <dlfcn.h>
  28 #include <stdlib.h>
  29
  30
  31 extern char** environ;
  32
  33 /*
  34  * Locate the python framework by looking for the
  35  * library that contains Py_Initialize.
  36  *
  37  * In a regular framework the structure is:
  38  *
  39  *    Python.framework/Versions/2.7
  40  *              /Python
  41  *              /Resources/Python.app/Contents/MacOS/Python
  42  *
  43  * In a virtualenv style structure the expected
  44  * structure is:
  45  *
  46  *    ROOT
  47  *       /bin/pythonw
  48  *       /.Python   <- the dylib
  49  *       /.Resources/Python.app/Contents/MacOS/Python
  50  *
  51  * NOTE: virtualenv's are not an officially supported
  52  * feature, support for that structure is provided as
  53  * a convenience.
  54  */
  55 static char* get_python_path(void)
  56 {
  57     size_t len;
  58     Dl_info info;
  59     char* end;
  60     char* g_path;
  61
  62     if (dladdr(Py_Initialize, &info) == 0) {
  63         return NULL;
  64     }
  65
  66     len = strlen(info.dli_fname);
  67
  68     g_path = malloc(len+60);
  69     if (g_path == NULL) {
  70         return NULL;
  71     }
  72
  73     strcpy(g_path, info.dli_fname);
  74     end = g_path + len - 1;
  75     while (end != g_path && *end != '/') {
  76         end --;
  77     }
  78     end++;
  79     if (*end == '.') {
  80         end++;
  81     }
  82     strcpy(end, "Resources/Python.app/Contents/MacOS/" PYTHONFRAMEWORK);
  83
  84     return g_path;
  85 }
  86
  87 #ifdef HAVE_SPAWN_H
  88 static void
  89 setup_spawnattr(posix_spawnattr_t* spawnattr)
  90 {
  91     size_t ocount;
  92     size_t count;
  93     cpu_type_t cpu_types[1];
  94     short flags = 0;
  95 #ifdef __LP64__
  96     int   ch;
  97 #endif
  98
  99     if ((errno = posix_spawnattr_init(spawnattr)) != 0) {
 100         err(2, "posix_spawnattr_int");
 101         /* NOTREACHTED */
 102     }
 103
 104     count = 1;
 105
 106     /* Run the real python executable using the same architecture as this
 107      * executable, this allows users to control the architecture using
 108      * "arch -ppc python"
 109      */
 110
 111 #if defined(__ppc64__)
 112     cpu_types[0] = CPU_TYPE_POWERPC64;
 113
 114 #elif defined(__x86_64__)
 115     cpu_types[0] = CPU_TYPE_X86_64;
 116
 117 #elif defined(__ppc__)
 118     cpu_types[0] = CPU_TYPE_POWERPC;
 119 #elif defined(__i386__)
 120     cpu_types[0] = CPU_TYPE_X86;
 121 #else
 122 #       error "Unknown CPU"
 123 #endif
 124
 125     if (posix_spawnattr_setbinpref_np(spawnattr, count,
 126                             cpu_types, &ocount) == -1) {
 127         err(1, "posix_spawnattr_setbinpref");
 128         /* NOTREACHTED */
 129     }
 130     if (count != ocount) {
 131         fprintf(stderr, "posix_spawnattr_setbinpref failed to copy\n");
 132         exit(1);
 133         /* NOTREACHTED */
 134     }
 135
 136
 137     /*
 138      * Set flag that causes posix_spawn to behave like execv
 139      */
 140     flags |= POSIX_SPAWN_SETEXEC;
 141     if ((errno = posix_spawnattr_setflags(spawnattr, flags)) != 0) {
 142         err(1, "posix_spawnattr_setflags");
 143         /* NOTREACHTED */
 144     }
 145 }
 146 #endif
 147
 148 int
 149 main(int argc, char **argv) {
 150     char* exec_path = get_python_path();
 151
 152     /*
 153      * Let argv[0] refer to the new interpreter. This is needed to
 154      * get the effect we want on OSX 10.5 or earlier. That is, without
 155      * changing argv[0] the real interpreter won't have access to
 156      * the Window Server.
 157      */
 158     argv[0] = exec_path;
 159
 160 #ifdef HAVE_SPAWN_H
 161
 162     /* We're weak-linking to posix-spawnv to ensure that
 163      * an executable build on 10.5 can work on 10.4.
 164      */
 165     if (posix_spawn != NULL) {
 166         posix_spawnattr_t spawnattr = NULL;
 167
 168
 169         setup_spawnattr(&spawnattr);
 170         posix_spawn(NULL, exec_path, NULL,
 171             &spawnattr, argv, environ);
 172         err(1, "posix_spawn: %s", exec_path);
 173     }
 174 #endif
 175     execve(exec_path, argv, environ);
 176     err(1, "execve: %s", argv[0]);
 177     /* NOTREACHED */
 178 }