dl.cpp

Go to the documentation of this file.

#include "thorin/util/dl.h"
 
#include <cstdlib>
 
#include <sstream>
 
#include "thorin/util/print.h"
 
#ifdef _WIN32
#    include <windows.h>
#else
#    include <dlfcn.h>
#endif
 
namespace thorin::dl {
 
void* open(const char* file) {
#ifdef _WIN32
    if (HMODULE handle = LoadLibraryA(file)) {
        return static_cast<void*>(handle);
    } else {
        error("could not load plugin '{}' due to error '{}'\n"
              "see https://docs.microsoft.com/en-us/windows/win32/debug/system-error-codes\n",
              file, GetLastError());
    }
#else
    if (void* handle = dlopen(file, RTLD_NOW))
        return handle;
    else if (auto err = dlerror())
        error("could not load plugin '{}' due to error '{}'\n", file, err);
    else
        error("could not load plugin '{}'\n", file);
#endif
}
 
void* get(void* handle, const char* symbol) {
#ifdef _WIN32
    if (auto addr = GetProcAddress(static_cast<HMODULE>(handle), symbol)) {
        return reinterpret_cast<void*>(addr);
    } else {
        error("could not find symbol '{}' in plugin due to error '{}'\n"
              "see (https://docs.microsoft.com/en-us/windows/win32/debug/system-error-codes)\n",
              symbol, GetLastError());
    }
#else
    dlerror(); // clear error state
    void* addr = dlsym(handle, symbol);
    if (auto err = dlerror())
        error("could not find symbol '{}' in plugin due to error '{}' \n", symbol, err);
    else
        return addr;
#endif
}
 
void close(void* handle) {
#ifdef _WIN32
    if (!FreeLibrary(static_cast<HMODULE>(handle))) error("FreeLibrary() failed\n");
#else
    if (auto err = dlclose(handle)) error("dlclose() failed with error code '{}'\n", err);
#endif
}
 
} // namespace thorin::dl