objc/dyld.c

   1 #import <objc/dyld.h>
   2
   3 #ifndef MAXPATHLEN
   4 #define MAXPATHLEN 8192
   5 #endif
   6
   7 #if defined(WIN32)
   8
   9 #import <windows.h>
  10
  11
  12 int _NSGetExecutablePath(char *path, uint32_t *capacity) {
  13     int bufferCapacity = MAXPATHLEN;
  14     uint16_t buffer[bufferCapacity + 1];
  15     DWORD i, bufferSize = GetModuleFileNameW(GetModuleHandle(NULL), buffer, bufferCapacity);
  16
  17     for(i = 0; i < bufferSize; i++)
  18         if(buffer[i] == '\\')
  19             buffer[i] = '/';
  20
  21     int size = WideCharToMultiByte(CP_UTF8, 0, buffer, bufferSize, NULL, 0, NULL, NULL);
  22
  23     if(size + 1 > *capacity) {
  24         *capacity = size + 1;
  25         return -1;
  26     }
  27
  28     size = WideCharToMultiByte(CP_UTF8, 0, buffer, bufferSize, path, *capacity, NULL, NULL);
  29     path[size] = '\0';
  30     size++;
  31
  32     *capacity = size;
  33
  34     return 0;
  35 }
  36
  37 // FIXME: these implementations do not return the size needed
  38
  39 #elif defined(LINUX)
  40
  41 int _NSGetExecutablePath(char *path, uint32_t *capacity) {
  42     if(*capacity < MAXPATHLEN)
  43         return MAXPATHLEN;
  44
  45     if((*capacity = readlink("/proc/self/exe", path, *capacity)) < 0) {
  46         *capacity = 0;
  47         return -1;
  48     }
  49
  50     return 0;
  51 }
  52
  53 #elif defined(__APPLE__)
  54
  55 extern int _NSGetExecutablePath(char *path, uint32_t *capacity);
  56
  57 #elif defined(BSD)
  58
  59 int _NSGetExecutablePath(char *path, uint32_t *capacity) {
  60 #if defined(FREEBSD)
  61     if(*capacity < MAXPATHLEN)
  62         return MAXPATHLEN;
  63
  64     int mib[4];
  65
  66     mib[0] = CTL_KERN;
  67     mib[1] = KERN_PROC;
  68     mib[2] = KERN_PROC_PATHNAME;
  69     mib[3] = -1;
  70
  71     size_t cb = *capacity;
  72
  73     if(sysctl(mib, 4, path, &cb, NULL, 0) < 0) {
  74         *capacity = 0;
  75         return -1;
  76     }
  77     *capacity = strlen(path);
  78
  79     return 0;
  80 #else
  81     if(*capacity < MAXPATHLEN)
  82         return MAXPATHLEN;
  83
  84     int length;
  85
  86     if((length = readlink("/proc/curproc/file", path, 1024)) < 0) {
  87         *capacity = 0;
  88         return -1;
  89     }
  90     *capacity = length;
  91
  92 #endif
  93     return 0;
  94 }
  95
  96 #elif defined(SOLARIS)
  97
  98 int _NSGetExecutablePath(char *path, uint32_t *capacity) {
  99     if(*capacity < MAXPATHLEN)
 100         return MAXPATHLEN;
 101
 102     char probe[MAXPATHLEN + 1];
 103
 104     sprintf(probe, "/proc/%ld/path/a.out", (long)getpid());
 105
 106     if((*capacity = readlink(probe, path, *capacity)) < 0) {
 107         *capacity = 0;
 108         return -1;
 109     }
 110
 111     return 0;
 112 }
 113 #endif