/**************************************************************************
*
- * Copyright 2010 VMware, Inc.
+ * Copyright 2010-2011 VMware, Inc.
* All Rights Reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
*
**************************************************************************/
+
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/time.h>
#include <pthread.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <signal.h>
+
+#ifdef __APPLE__
+#include <mach-o/dyld.h>
+#endif
#include "os.hpp"
+
namespace OS {
bool
GetProcessName(char *str, size_t size)
{
- ssize_t len;
char szProcessPath[PATH_MAX + 1];
char *lpProcessName;
// http://stackoverflow.com/questions/1023306/finding-current-executables-path-without-proc-self-exe
- len = readlink("/proc/self/exe", szProcessPath, sizeof(szProcessPath) - 1);
- if (len == -1) {
+#ifdef __APPLE__
+ uint32_t len = sizeof szProcessPath;
+ if (_NSGetExecutablePath(szProcessPath, &len) != 0) {
*str = 0;
return false;
}
+#else
+ ssize_t len;
+ len = readlink("/proc/self/exe", szProcessPath, sizeof(szProcessPath) - 1);
+ if (len == -1) {
+ // /proc/self/exe is not available on setuid processes, so fallback to
+ // /proc/self/cmdline.
+ int fd = open("/proc/self/cmdline", O_RDONLY);
+ if (fd >= 0) {
+ len = read(fd, szProcessPath, sizeof(szProcessPath) - 1);
+ close(fd);
+ }
+ }
+ if (len <= 0) {
+ snprintf(str, size, "%i", (int)getpid());
+ return true;
+ }
+#endif
szProcessPath[len] = 0;
lpProcessName = strrchr(szProcessPath, '/');
}
+void
+CatchInterrupts(void (*func)(int))
+{
+ struct sigaction new_action, old_action;
+
+ new_action.sa_handler = func;
+ sigemptyset(&new_action.sa_mask);
+ new_action.sa_flags = 0;
+
+#define SET_IF_NOT_IGNORED(sig) \
+ do { \
+ sigaction(sig, NULL, &old_action); \
+ if (old_action.sa_handler != SIG_IGN) \
+ sigaction(sig, &new_action, NULL); \
+ } while (0)
+
+ SET_IF_NOT_IGNORED(SIGINT);
+ SET_IF_NOT_IGNORED(SIGHUP);
+ SET_IF_NOT_IGNORED(SIGTERM);
+
+#undef SET_IF_NOT_IGNORED
+}
+
} /* namespace OS */