Jip J. Dekker
fad1b07018
git-subtree-dir: software/minizinc git-subtree-split: 4f10c82056ffcb1041d7ffef29d77a7eef92cf76
138 lines
3.3 KiB
C++
138 lines
3.3 KiB
C++
/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
|
|
|
|
/*
|
|
* Main authors:
|
|
* Jason Nguyen <jason.nguyen@monash.edu>
|
|
*/
|
|
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#define NOMINMAX // Ensure the words min/max remain available
|
|
#include <Windows.h>
|
|
#undef ERROR
|
|
#else
|
|
#include <dlfcn.h>
|
|
#endif
|
|
|
|
#include <minizinc/exception.hh>
|
|
#include <minizinc/file_utils.hh>
|
|
|
|
#include <cstdlib>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
/// Convenience macro for loading symbols
|
|
#define load_symbol(name) *(void**)(&(name)) = symbol(#name)
|
|
|
|
namespace MiniZinc {
|
|
/// Base class for plugins loaded from DLLs
|
|
class Plugin {
|
|
public:
|
|
class PluginError : public Exception {
|
|
public:
|
|
/// Construct with message \a msg
|
|
PluginError(const std::string& msg) : Exception(msg) {}
|
|
/// Destructor
|
|
~PluginError() throw() override {}
|
|
/// Return description
|
|
const char* what() const throw() override { return "MiniZinc: plugin loading error"; }
|
|
};
|
|
|
|
/// Load a plugin with given DLL path
|
|
Plugin(const std::string& file) {
|
|
if (!open(file)) {
|
|
throw PluginError("Failed to load plugin " + file);
|
|
}
|
|
}
|
|
/// Load a plugin by trying the given DLL file paths
|
|
Plugin(const std::vector<std::string>& files) {
|
|
for (const auto& file : files) {
|
|
if (open(file)) {
|
|
return;
|
|
}
|
|
}
|
|
bool first = true;
|
|
std::stringstream ss;
|
|
ss << "Failed to load plugin. Tried ";
|
|
for (const auto& file : files) {
|
|
if (first) {
|
|
first = false;
|
|
} else {
|
|
ss << ", ";
|
|
}
|
|
ss << file;
|
|
}
|
|
throw PluginError(ss.str());
|
|
}
|
|
|
|
~Plugin() { close(); }
|
|
|
|
/// Get the path to the loaded DLL
|
|
const std::string& path() { return _loaded; }
|
|
|
|
protected:
|
|
/// Load a symbol from this DLL
|
|
void* symbol(const char* name) {
|
|
void* ret;
|
|
#ifdef _WIN32
|
|
ret = (void*)GetProcAddress((HMODULE)_dll, (LPCSTR)name);
|
|
#else
|
|
ret = dlsym(_dll, name);
|
|
#endif
|
|
if (ret == nullptr) {
|
|
throw PluginError(std::string("Failed to load symbol ") + name);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
private:
|
|
void* _dll;
|
|
std::string _loaded;
|
|
bool open(const std::string& file) {
|
|
#ifdef _WIN32
|
|
const std::string ext = ".dll";
|
|
#elif __APPLE__
|
|
const std::string ext = ".dylib";
|
|
#else
|
|
const std::string ext = ".so";
|
|
#endif
|
|
bool hasExt =
|
|
file.size() >= ext.size() && file.compare(file.size() - ext.size(), ext.size(), ext) == 0;
|
|
auto path = (hasExt || MiniZinc::FileUtils::is_absolute(file)) ? file : (file + ext);
|
|
#ifdef _WIN32
|
|
auto dir = MiniZinc::FileUtils::dir_name(path);
|
|
if (!dir.empty()) {
|
|
// Add the path with the DLL to the search path for dependency loading
|
|
SetDllDirectoryW(MiniZinc::FileUtils::utf8_to_wide(dir).c_str());
|
|
}
|
|
_dll = (void*)LoadLibrary((LPCSTR)path.c_str());
|
|
if (!dir.empty()) {
|
|
SetDllDirectoryW(nullptr);
|
|
}
|
|
#else
|
|
_dll = dlopen(path.c_str(), RTLD_NOW);
|
|
#endif
|
|
if (_dll != nullptr) {
|
|
_loaded = path;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
void close() {
|
|
#ifdef _WIN32
|
|
FreeLibrary((HMODULE)_dll);
|
|
#else
|
|
dlclose(_dll);
|
|
#endif
|
|
_dll = nullptr;
|
|
}
|
|
};
|
|
} // namespace MiniZinc
|