mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-10 01:10:35 +00:00
70 lines
1.6 KiB
C++
70 lines
1.6 KiB
C++
#include "dlhandle.h"
|
|
|
|
#include <string>
|
|
|
|
#ifndef _WIN32
|
|
# include <dlfcn.h>
|
|
#else
|
|
# include <cassert>
|
|
# include <sstream>
|
|
# define WIN32_LEAN_AND_MEAN
|
|
# ifndef NOMINMAX
|
|
# define NOMINMAX
|
|
# endif
|
|
# include <windows.h>
|
|
#endif
|
|
|
|
using namespace std::string_literals;
|
|
namespace fs = std::filesystem;
|
|
|
|
|
|
#ifndef _WIN32
|
|
|
|
Dlhandle::Dlhandle(const fs::path &fpath) {
|
|
chandle = dlopen(fpath.c_str(), RTLD_LAZY | RTLD_LOCAL);
|
|
if (!chandle) {
|
|
throw Exception("dlopen: "s + dlerror());
|
|
}
|
|
}
|
|
|
|
Dlhandle::~Dlhandle() {
|
|
if (chandle) dlclose(chandle);
|
|
}
|
|
|
|
void *Dlhandle::get_internal(const char *symbol) const {
|
|
return dlsym(chandle, symbol);
|
|
}
|
|
|
|
#else // defined(_WIN32)
|
|
|
|
Dlhandle::Dlhandle(const fs::path &fpath) {
|
|
fs::path afpath = fs::absolute(fpath);
|
|
|
|
// Suppress the "Entry Point Not Found" dialog, caused by outdated nvcuda.dll from the GPU driver
|
|
UINT lastErrorMode = GetErrorMode();
|
|
UINT success = SetErrorMode(lastErrorMode | SEM_FAILCRITICALERRORS);
|
|
assert(success);
|
|
|
|
chandle = LoadLibraryExW(afpath.c_str(), NULL, LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR);
|
|
|
|
success = SetErrorMode(lastErrorMode);
|
|
assert(success);
|
|
|
|
if (!chandle) {
|
|
DWORD err = GetLastError();
|
|
std::ostringstream ss;
|
|
ss << "LoadLibraryExW failed with error 0x" << std::hex << err;
|
|
throw Exception(ss.str());
|
|
}
|
|
}
|
|
|
|
Dlhandle::~Dlhandle() {
|
|
if (chandle) FreeLibrary(HMODULE(chandle));
|
|
}
|
|
|
|
void *Dlhandle::get_internal(const char *symbol) const {
|
|
return GetProcAddress(HMODULE(chandle), symbol);
|
|
}
|
|
|
|
#endif // defined(_WIN32)
|