WAI_NOINLINE WAI_FUNCSPEC int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length) { char buffer[PATH_MAX]; char* resolved = NULL; int length = -1; for(;;) { Dl_info info; if (dladdr(WAI_RETURN_ADDRESS(), &info)) { resolved = realpath(info.dli_fname, buffer); if (!resolved) break; length = (int)strlen(resolved); if (length <= capacity) { memcpy(out, resolved, length); if (dirname_length) { int i; for (i = length - 1; i >= 0; --i) { if (out[i] == '/') { *dirname_length = i; break; } } } } } break; } return length; }
WAI_NOINLINE WAI_FUNCSPEC int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length) { int length = -1; FILE* maps = NULL; for (int r = 0; r < WAI_PROC_SELF_MAPS_RETRY; ++r) { maps = fopen(WAI_PROC_SELF_MAPS, "r"); if (!maps) break; for (;;) { char buffer[PATH_MAX < 1024 ? 1024 : PATH_MAX]; uint64_t low, high; char perms[5]; uint64_t offset; uint32_t major, minor; char path[PATH_MAX]; uint32_t inode; if (!fgets(buffer, sizeof(buffer), maps)) break; if (sscanf(buffer, "%" PRIx64 "-%" PRIx64 " %s %" PRIx64 " %x:%x %u %s\n", &low, &high, perms, &offset, &major, &minor, &inode, path) == 8) { uint64_t addr = (uintptr_t)WAI_RETURN_ADDRESS(); if (low <= addr && addr <= high) { char* resolved; resolved = realpath(path, buffer); if (!resolved) break; length = (int)strlen(resolved); #if defined(__ANDROID__) || defined(ANDROID) if (length > 4 &&buffer[length - 1] == 'k' &&buffer[length - 2] == 'p' &&buffer[length - 3] == 'a' &&buffer[length - 4] == '.') { int fd = open(path, O_RDONLY); char* begin; char* p; begin = (char*)mmap(0, offset, PROT_READ, MAP_SHARED, fd, 0); p = begin + offset; while (p >= begin) // scan backwards { if (*((uint32_t*)p) == 0x04034b50UL) // local file header found { uint16_t length_ = *((uint16_t*)(p + 26)); if (length + 2 + length_ < (int)sizeof(buffer)) { memcpy(&buffer[length], "!/", 2); memcpy(&buffer[length + 2], p + 30, length_); length += 2 + length_; } break; } p -= 4; } munmap(begin, offset); close(fd); } #endif if (length <= capacity) { memcpy(out, resolved, length); if (dirname_length) { int i; for (i = length - 1; i >= 0; --i) { if (out[i] == '/') { *dirname_length = i; break; } } } } break; } } } fclose(maps); maps = NULL; if (length != -1) break; } if (maps) fclose(maps); return length; }
WAI_NOINLINE WAI_FUNCSPEC int WAI_PREFIX(getModulePath)(char* out, int capacity, int* dirname_length) { HMODULE module; int length = -1; #if defined(_MSC_VER) #pragma warning(push) #pragma warning(disable: 4054) #endif if (GetModuleHandleEx(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, (LPCTSTR)WAI_RETURN_ADDRESS(), &module)) #if defined(_MSC_VER) #pragma warning(pop) #endif { length = WAI_PREFIX(getModulePath_)(module, out, capacity, dirname_length); } return length; }