src/library/running-executable.cpp

   1 #include "running-executable.hpp"
   2 #include "directory.hpp"
   3 #include <stdexcept>
   4
   5 namespace
   6 {
   7 #if __unix__
   8 #include <unistd.h>
   9         std::string readlink_generic(std::string path)
  10         {
  11                 char buf[8192] = {0};
  12                 if(readlink(path.c_str(), buf, sizeof(buf)) > 0)
  13                         return buf;
  14                 throw std::runtime_error("Can't determine executable path");
  15         }
  16 #endif
  17
  18 #if __linux__
  19 #include <unistd.h>
  20 std::string _running_executable()
  21 {
  22         return readlink_generic("/proc/self/exe");
  23 }
  24 #elif __APPLE__
  25 #include <mach-o/dyld.h>
  26 std::string _running_executable()
  27 {
  28         char buf[8192];
  29         uint32_t size = 8192;
  30         _NSGetExecutablePath(buf, &size);
  31         return buf;
  32 }
  33 #elif __WIN32__ || __WIN64__
  34 #include "windows.h"
  35 std::string _running_executable()
  36 {
  37         char buf[8192];
  38         GetModuleFileName(NULL, buf, 8191);
  39         return buf;
  40 }
  41 #elif __OpenBSD__
  42 std::string _running_executable()
  43 {
  44         //TODO: This isn't doable?
  45         throw std::runtime_error("running_executable() unsupported");
  46 }
  47 #elif __FreeBSD__
  48 #include <sys/param.h>
  49 #include <sys/sysctl.h>
  50 std::string _running_executable()
  51 {
  52         char buf[8192];
  53         int args[] = {CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1};
  54         size_t len = 8192;
  55         if(sysctl(args, 4, buf, &len, NULL, 0) < 0)
  56                 throw std::runtime_error("Can't determine executable path");
  57         return buf;
  58 }
  59 #elif __NetBSD__
  60 std::string _running_executable()
  61 {
  62         return readlink_generic("/proc/curproc/exe");
  63 }
  64 #elif __sun__
  65 #include <stdlib.h>
  66 std::string _running_executable()
  67 {
  68         return getexecname();
  69 }
  70 #else
  71 //Unsupported
  72 std::string _running_executable()
  73 {
  74         throw std::runtime_error("running_executable() unsupported");
  75 }
  76 #endif
  77 }
  78
  79 std::string running_executable()
  80 {
  81         return get_absolute_path(_running_executable());
  82 }