git.cworth.org Git - apitrace/blob - os_posix.cpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2010-2011 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26
  27 #include <assert.h>
  28 #include <string.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31
  32 #include <unistd.h>
  33 #include <sys/time.h>
  34 #include <pthread.h>
  35 #include <sys/stat.h>
  36 #include <fcntl.h>
  37 #include <signal.h>
  38
  39 #ifdef __APPLE__
  40 #include <mach-o/dyld.h>
  41 #endif
  42
  43 #include "os.hpp"
  44
  45
  46 namespace OS {
  47
  48
  49 static pthread_mutex_t
  50 mutex = PTHREAD_MUTEX_INITIALIZER;
  51
  52
  53 void
  54 AcquireMutex(void)
  55 {
  56     pthread_mutex_lock(&mutex);
  57 }
  58
  59
  60 void
  61 ReleaseMutex(void)
  62 {
  63     pthread_mutex_unlock(&mutex);
  64 }
  65
  66
  67 bool
  68 GetProcessName(char *str, size_t size)
  69 {
  70     char szProcessPath[PATH_MAX + 1];
  71     char *lpProcessName;
  72
  73     // http://stackoverflow.com/questions/1023306/finding-current-executables-path-without-proc-self-exe
  74 #ifdef __APPLE__
  75     uint32_t len = sizeof szProcessPath;
  76     if (_NSGetExecutablePath(szProcessPath, &len) != 0) {
  77         *str = 0;
  78         return false;
  79     }
  80 #else
  81     ssize_t len;
  82     len = readlink("/proc/self/exe", szProcessPath, sizeof(szProcessPath) - 1);
  83     if (len == -1) {
  84         // /proc/self/exe is not available on setuid processes, so fallback to
  85         // /proc/self/cmdline.
  86         int fd = open("/proc/self/cmdline", O_RDONLY);
  87         if (fd >= 0) {
  88             len = read(fd, szProcessPath, sizeof(szProcessPath) - 1);
  89             close(fd);
  90         }
  91     }
  92     if (len <= 0) {
  93         snprintf(str, size, "%i", (int)getpid());
  94         return true;
  95     }
  96 #endif
  97     szProcessPath[len] = 0;
  98
  99     lpProcessName = strrchr(szProcessPath, '/');
 100     lpProcessName = lpProcessName ? lpProcessName + 1 : szProcessPath;
 101
 102     strncpy(str, lpProcessName, size);
 103     if (size)
 104         str[size - 1] = 0;
 105
 106     return true;
 107 }
 108
 109 bool
 110 GetCurrentDir(char *str, size_t size)
 111 {
 112     char *ret;
 113     ret = getcwd(str, size);
 114     str[size - 1] = 0;
 115     return ret ? true : false;
 116 }
 117
 118 void
 119 DebugMessage(const char *format, ...)
 120 {
 121     va_list ap;
 122     va_start(ap, format);
 123     fflush(stdout);
 124     vfprintf(stderr, format, ap);
 125     va_end(ap);
 126 }
 127
 128 long long GetTime(void)
 129 {
 130     struct timeval tv;
 131     gettimeofday(&tv, NULL);
 132     return tv.tv_usec + tv.tv_sec*1000000LL;
 133 }
 134
 135 void
 136 Abort(void)
 137 {
 138     exit(0);
 139 }
 140
 141
 142 static void (*gCallback)(void) = NULL;
 143
 144 #define NUM_SIGNALS 16
 145
 146 struct sigaction old_actions[NUM_SIGNALS];
 147
 148 static void signal_handler(int sig, siginfo_t *info, void *context)
 149 {
 150     static int recursion_count = 0;
 151
 152     fprintf(stderr, "signal_handler: sig = %i\n", sig);
 153
 154     if (recursion_count) {
 155         fprintf(stderr, "recursion with sig %i\n", sig);
 156     } else {
 157         if (gCallback) {
 158             ++recursion_count;
 159             gCallback();
 160             --recursion_count;
 161         }
 162     }
 163
 164     struct sigaction *old_action;
 165     if (sig >= NUM_SIGNALS) {
 166         /* This should never happen */
 167         fprintf(stderr, "Unexpected signal %i\n", sig);
 168         raise(SIGKILL);
 169     }
 170     old_action = &old_actions[sig];
 171
 172     if (old_action->sa_flags & SA_SIGINFO) {
 173         // Handler is in sa_sigaction
 174         old_action->sa_sigaction(sig, info, context);
 175     } else {
 176         if (old_action->sa_handler == SIG_DFL) {
 177             fprintf(stderr, "taking default action for signal %i\n", sig);
 178
 179 #if 1
 180             struct sigaction dfl_action;
 181             dfl_action.sa_handler = SIG_DFL;
 182             sigemptyset (&dfl_action.sa_mask);
 183             dfl_action.sa_flags = 0;
 184             sigaction(sig, &dfl_action, NULL);
 185
 186             raise(sig);
 187 #else
 188             raise(SIGKILL);
 189 #endif
 190         } else if (old_action->sa_handler == SIG_IGN) {
 191             /* ignore */
 192         } else {
 193             /* dispatch to handler */
 194             old_action->sa_handler(sig);
 195         }
 196     }
 197 }
 198
 199 void
 200 SetExceptionCallback(void (*callback)(void))
 201 {
 202     assert(!gCallback);
 203     if (!gCallback) {
 204         gCallback = callback;
 205
 206         struct sigaction new_action;
 207         new_action.sa_sigaction = signal_handler;
 208         sigemptyset(&new_action.sa_mask);
 209         new_action.sa_flags = SA_SIGINFO | SA_RESTART;
 210
 211
 212         for (int sig = 1; sig < NUM_SIGNALS; ++sig) {
 213             // SIGKILL and SIGSTOP can't be handled
 214             if (sig != SIGKILL && sig != SIGSTOP) {
 215                 if (sigaction(sig,  NULL, &old_actions[sig]) >= 0) {
 216                     sigaction(sig,  &new_action, NULL);
 217                 }
 218             }
 219         }
 220     }
 221 }
 222
 223 void
 224 ResetExceptionCallback(void)
 225 {
 226     gCallback = NULL;
 227 }
 228
 229 } /* namespace OS */
 230