whereami/example/executable.c
2015-03-05 20:19:44 +01:00

124 lines
2.4 KiB
C

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <whereami.h>
#if defined(_WIN32)
#define WIN32_LEAN_AND_MEAN
#if defined(_MSC_VER)
#pragma warning(push, 3)
#endif
#include <windows.h>
#define RTLD_LAZY 1
#define RTLD_NOW 2
#define RTLD_GLOBAL 4
#define RTLD_LOCAL 8
static void* dlopen(const char* fileName, int mode)
{
wchar_t buffer[MAX_PATH];
if (MultiByteToWideChar(CP_UTF8, 0, fileName, -1, buffer, sizeof(buffer) / sizeof(*buffer)))
{
wchar_t buffer_[MAX_PATH];
GetFullPathNameW(buffer, sizeof(buffer_) / sizeof(*buffer_), buffer_, NULL);
return (void*)LoadLibraryW(buffer_);
}
return NULL;
}
static int dlclose(void* handle)
{
return FreeLibrary((HMODULE)handle) ? 0 : -1;
}
static const char* dlerror(void)
{
DWORD error;
error = GetLastError();
if (error)
{
static char message[1024];
DWORD length;
length = FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, error, MAKELANGID(LANG_ENGLISH, SUBLANG_DEFAULT), message, sizeof(message), NULL);
return message;
}
return "no error";
}
#if defined(_MSC_VER)
#pragma warning(pop)
#endif
#else
#include <dlfcn.h>
#endif
int main(int argc, char** argv)
{
char* path = NULL;
int length, dirname_length;
int i;
length = wai_getExecutablePath(NULL, 0, &dirname_length);
if (length > 0)
{
path = malloc(length + 1);
wai_getExecutablePath(path, length, &dirname_length);
path[length] = '\0';
printf("executable path: %s\n", path);
path[dirname_length] = '\0';
printf(" dirname: %s\n", path);
printf(" basename: %s\n", path + dirname_length + 1);
free(path);
}
length = wai_getModulePath(NULL, 0, &dirname_length);
if (length > 0)
{
path = malloc(length + 1);
wai_getModulePath(path, length, &dirname_length);
path[length] = '\0';
printf("module path: %s\n", path);
path[dirname_length] = '\0';
printf(" dirname: %s\n", path);
printf(" basename: %s\n", path + dirname_length + 1);
free(path);
}
for (i = 1; i < argc; ++i)
{
if (strncmp(argv[i], "--load-library=", 15) == 0)
{
char* name = argv[i] + 15;
void* handle;
printf("\n");
handle = dlopen(name, RTLD_NOW);
if (!handle)
printf("failed to load library: %s\n", dlerror());
if (handle)
dlclose(handle);
}
}
return 0;
}