1 // (‑●‑●)> dual licensed under the WTFPL v2 and MIT licenses
2 // without any warranty.
3 // by Gregory Pakosz (@gpakosz)
4 // https://github.com/gpakosz/whereami
10 // in case you want to #include "whereami.cc" in a larger compilation unit
11 #if !defined(WHEREAMI_H)
19 #if !defined(WAI_MALLOC) || !defined(WAI_FREE) || !defined(WAI_REALLOC)
23 #if !defined(WAI_MALLOC)
24 #define WAI_MALLOC(size) malloc(size)
27 #if !defined(WAI_FREE)
28 #define WAI_FREE(p) free(p)
31 #if !defined(WAI_REALLOC)
32 #define WAI_REALLOC(p, size) realloc(p, size)
37 #define WAI_NOINLINE __declspec(noinline)
38 #elif defined(__GNUC__)
39 #define WAI_NOINLINE __attribute__((noinline))
41 #error unsupported compiler
46 #define WAI_RETURN_ADDRESS() _ReturnAddress()
47 #elif defined(__GNUC__)
48 #define WAI_RETURN_ADDRESS() __builtin_extract_return_addr(__builtin_return_address(0))
50 #error unsupported compiler
55 #ifndef WIN32_LEAN_AND_MEAN
56 #define WIN32_LEAN_AND_MEAN
59 #pragma warning(push, 3)
67 static int WAI_PREFIX(getModulePath_)(HMODULE module, char* out, int capacity, int* dirname_length)
69 wchar_t buffer1[MAX_PATH];
70 wchar_t buffer2[MAX_PATH];
77 int length_, length__;
79 size = GetModuleFileNameW(module, buffer1, sizeof(buffer1) / sizeof(buffer1[0]));
83 else if (size == (DWORD)(sizeof(buffer1) / sizeof(buffer1[0])))
90 path_ = (wchar_t*)WAI_REALLOC(path, sizeof(wchar_t) * size_ * 2);
95 size = GetModuleFileNameW(module, path, size_);
97 while (size == size_);
105 if (!_wfullpath(buffer2, path, MAX_PATH))
107 length_ = (int)wcslen(buffer2);
108 length__ = WideCharToMultiByte(CP_UTF8, 0, buffer2, length_ , out, capacity, NULL, NULL);
111 length__ = WideCharToMultiByte(CP_UTF8, 0, buffer2, length_, NULL, 0, NULL, NULL);
115 if (length__ <= capacity && dirname_length)
119 for (i = length__ - 1; i >= 0; --i)
140 WAI_NOINLINE WAI_FUNCSPEC
141 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
143 return WAI_PREFIX(getModulePath_)(NULL, out, capacity, dirname_length);
146 WAI_NOINLINE WAI_FUNCSPEC
147 int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length)
152 #if defined(_MSC_VER)
153 #pragma warning(push)
154 #pragma warning(disable: 4054)
156 if (GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, (LPCTSTR)WAI_RETURN_ADDRESS(), &module))
157 #if defined(_MSC_VER)
161 length = WAI_PREFIX(getModulePath_)(module, out, capacity, dirname_length);
167 #elif defined(__linux__) || defined(__CYGWIN__) || defined(__sun) || defined(__GNU__) || defined(WAI_USE_PROC_SELF_EXE)
172 #if defined(__linux__)
173 #include <linux/limits.h>
177 #ifndef __STDC_FORMAT_MACROS
178 #define __STDC_FORMAT_MACROS
180 #include <inttypes.h>
182 #if !defined(WAI_PROC_SELF_EXE)
184 #define WAI_PROC_SELF_EXE "/proc/self/path/a.out"
186 #define WAI_PROC_SELF_EXE "/proc/self/exe"
191 #define PATH_MAX 4096
195 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
197 char buffer[PATH_MAX];
198 char* resolved = nullptr;
203 resolved = realpath(WAI_PROC_SELF_EXE, buffer);
207 length = (int)strlen(resolved);
208 if (length <= capacity)
210 memcpy(out, resolved, length);
216 for (i = length - 1; i >= 0; --i)
233 #if !defined(WAI_PROC_SELF_MAPS_RETRY)
234 #define WAI_PROC_SELF_MAPS_RETRY 5
237 #if !defined(WAI_PROC_SELF_MAPS)
239 #define WAI_PROC_SELF_MAPS "/proc/self/map"
241 #define WAI_PROC_SELF_MAPS "/proc/self/maps"
245 #if defined(__ANDROID__) || defined(ANDROID)
247 #include <sys/mman.h>
251 WAI_NOINLINE WAI_FUNCSPEC
252 int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length)
255 FILE* maps = nullptr;
257 for (int r = 0; r < WAI_PROC_SELF_MAPS_RETRY; ++r)
259 maps = fopen(WAI_PROC_SELF_MAPS, "r");
265 char buffer[PATH_MAX < 1024 ? 1024 : PATH_MAX];
269 uint32_t major, minor;
273 if (!fgets(buffer, sizeof(buffer), maps))
276 if (sscanf(buffer, "%" PRIx64 "-%" PRIx64 " %s %" PRIx64 " %x:%x %u %s\n", &low, &high, perms, &offset, &major, &minor, &inode, path) == 8)
278 auto addr = reinterpret_cast<uintptr_t>(WAI_RETURN_ADDRESS());
279 if (low <= addr && addr <= high)
283 resolved = realpath(path, buffer);
287 length = (int)strlen(resolved);
288 #if defined(__ANDROID__) || defined(ANDROID)
290 &&buffer[length - 1] == 'k'
291 &&buffer[length - 2] == 'p'
292 &&buffer[length - 3] == 'a'
293 &&buffer[length - 4] == '.')
295 int fd = open(path, O_RDONLY);
298 length = -1; // retry
302 char* begin = (char*)mmap(0, offset, PROT_READ, MAP_SHARED, fd, 0);
303 if (begin == MAP_FAILED)
306 length = -1; // retry
310 char* p = begin + offset - 30; // minimum size of local file header
311 while (p >= begin) // scan backwards
313 if (*((uint32_t*)p) == 0x04034b50UL) // local file header signature found
315 uint16_t length_ = *((uint16_t*)(p + 26));
317 if (length + 2 + length_ < (int)sizeof(buffer))
319 memcpy(&buffer[length], "!/", 2);
320 memcpy(&buffer[length + 2], p + 30, length_);
321 length += 2 + length_;
330 munmap(begin, offset);
334 if (length <= capacity)
336 memcpy(out, resolved, length);
342 for (i = length - 1; i >= 0; --i)
371 #elif defined(__APPLE__)
373 #define _DARWIN_BETTER_REALPATH
374 #include <mach-o/dyld.h>
381 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
383 char buffer1[PATH_MAX];
384 char buffer2[PATH_MAX];
385 char* path = buffer1;
386 char* resolved = NULL;
391 uint32_t size = (uint32_t)sizeof(buffer1);
392 if (_NSGetExecutablePath(path, &size) == -1)
394 path = (char*)WAI_MALLOC(size);
395 if (!_NSGetExecutablePath(path, &size))
399 resolved = realpath(path, buffer2);
403 length = (int)strlen(resolved);
404 if (length <= capacity)
406 memcpy(out, resolved, length);
412 for (i = length - 1; i >= 0; --i)
432 WAI_NOINLINE WAI_FUNCSPEC
433 int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length)
435 char buffer[PATH_MAX];
436 char* resolved = NULL;
443 if (dladdr(WAI_RETURN_ADDRESS(), &info))
445 resolved = realpath(info.dli_fname, buffer);
449 length = (int)strlen(resolved);
450 if (length <= capacity)
452 memcpy(out, resolved, length);
458 for (i = length - 1; i >= 0; --i)
476 #elif defined(__QNXNTO__)
484 #if !defined(WAI_PROC_SELF_EXE)
485 #define WAI_PROC_SELF_EXE "/proc/self/exefile"
489 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
491 char buffer1[PATH_MAX];
492 char buffer2[PATH_MAX];
493 char* resolved = NULL;
494 FILE* self_exe = NULL;
499 self_exe = fopen(WAI_PROC_SELF_EXE, "r");
503 if (!fgets(buffer1, sizeof(buffer1), self_exe))
506 resolved = realpath(buffer1, buffer2);
510 length = (int)strlen(resolved);
511 if (length <= capacity)
513 memcpy(out, resolved, length);
519 for (i = length - 1; i >= 0; --i)
539 int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length)
541 char buffer[PATH_MAX];
542 char* resolved = NULL;
549 if (dladdr(WAI_RETURN_ADDRESS(), &info))
551 resolved = realpath(info.dli_fname, buffer);
555 length = (int)strlen(resolved);
556 if (length <= capacity)
558 memcpy(out, resolved, length);
564 for (i = length - 1; i >= 0; --i)
582 #elif defined(__DragonFly__) || defined(__FreeBSD__) || \
583 defined(__FreeBSD_kernel__) || defined(__NetBSD__) || defined(__OpenBSD__)
588 #include <sys/types.h>
589 #include <sys/sysctl.h>
592 #if defined(__OpenBSD__)
597 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
600 char buffer2[PATH_MAX];
601 char buffer3[PATH_MAX];
602 char** argv = (char**)buffer1;
603 char* resolved = NULL;
608 int mib[4] = { CTL_KERN, KERN_PROC_ARGS, getpid(), KERN_PROC_ARGV };
611 if (sysctl(mib, 4, NULL, &size, NULL, 0) != 0)
614 if (size > sizeof(buffer1))
616 argv = (char**)WAI_MALLOC(size);
621 if (sysctl(mib, 4, argv, &size, NULL, 0) != 0)
624 if (strchr(argv[0], '/'))
626 resolved = realpath(argv[0], buffer2);
632 const char* PATH = getenv("PATH");
636 size_t argv0_length = strlen(argv[0]);
638 const char* begin = PATH;
641 const char* separator = strchr(begin, ':');
642 const char* end = separator ? separator : begin + strlen(begin);
646 if (*(end -1) == '/')
649 if (((end - begin) + 1 + argv0_length + 1) <= sizeof(buffer2))
651 memcpy(buffer2, begin, end - begin);
652 buffer2[end - begin] = '/';
653 memcpy(buffer2 + (end - begin) + 1, argv[0], argv0_length + 1);
655 resolved = realpath(buffer2, buffer3);
671 length = (int)strlen(resolved);
672 if (length <= capacity)
674 memcpy(out, resolved, length);
680 for (i = length - 1; i >= 0; --i)
694 if (argv != (char**)buffer1)
703 int WAI_PREFIX(getExecutablePath)(char* out, int capacity, int* dirname_length)
705 char buffer1[PATH_MAX];
706 char buffer2[PATH_MAX];
707 char* path = buffer1;
708 char* resolved = NULL;
713 #if defined(__NetBSD__)
714 int mib[4] = { CTL_KERN, KERN_PROC_ARGS, -1, KERN_PROC_PATHNAME };
716 int mib[4] = { CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, -1 };
718 size_t size = sizeof(buffer1);
720 if (sysctl(mib, 4, path, &size, NULL, 0) != 0)
723 resolved = realpath(path, buffer2);
727 length = (int)strlen(resolved);
728 if (length <= capacity)
730 memcpy(out, resolved, length);
736 for (i = length - 1; i >= 0; --i)
755 WAI_NOINLINE WAI_FUNCSPEC
756 int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length)
758 char buffer[PATH_MAX];
759 char* resolved = NULL;
766 if (dladdr(WAI_RETURN_ADDRESS(), &info))
768 resolved = realpath(info.dli_fname, buffer);
772 length = (int)strlen(resolved);
773 if (length <= capacity)
775 memcpy(out, resolved, length);
781 for (i = length - 1; i >= 0; --i)
801 #error unsupported platform