128 lines
3.0 KiB
C++
128 lines
3.0 KiB
C++
// Copyright 2018 Slightech Co., Ltd. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
#include "internal/dl.h"
|
|
|
|
#include <glog/logging.h>
|
|
|
|
MYNTEYE_BEGIN_NAMESPACE
|
|
|
|
#if defined(OS_WIN) && !defined(OS_MINGW) && !defined(OS_CYGWIN)
|
|
|
|
namespace {
|
|
|
|
// How to get the error message from the error code returned by GetLastError()?
|
|
// https://stackoverflow.com/questions/9272415/how-to-convert-dword-to-char
|
|
std::string GetLastErrorAsString() {
|
|
DWORD dw = ::GetLastError();
|
|
if (dw == 0)
|
|
return std::string();
|
|
LPSTR lpMsgBuf;
|
|
size_t size = FormatMessageA(
|
|
FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
|
|
FORMAT_MESSAGE_IGNORE_INSERTS,
|
|
NULL, dw, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPSTR)&lpMsgBuf, 0,
|
|
NULL);
|
|
std::string message(lpMsgBuf, size);
|
|
LocalFree(lpMsgBuf);
|
|
return message;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
#endif
|
|
|
|
DL::DL() : handle(nullptr) {
|
|
VLOG(2) << __func__;
|
|
}
|
|
|
|
DL::DL(const char *filename) : handle(nullptr) {
|
|
VLOG(2) << __func__;
|
|
Open(filename);
|
|
}
|
|
|
|
DL::~DL() {
|
|
VLOG(2) << __func__;
|
|
Close();
|
|
}
|
|
|
|
bool DL::Open(const char *filename) {
|
|
if (handle != nullptr) {
|
|
VLOG(2) << "Already opened, do nothing";
|
|
// Close();
|
|
return false;
|
|
}
|
|
#if defined(OS_WIN) && !defined(OS_MINGW) && !defined(OS_CYGWIN)
|
|
handle = LoadLibraryEx(filename, nullptr, 0);
|
|
#else
|
|
handle = dlopen(filename, RTLD_LAZY);
|
|
#endif
|
|
if (handle == nullptr) {
|
|
VLOG(2) << "Open library failed: " << filename;
|
|
return false;
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
bool DL::IsOpened() {
|
|
return handle != nullptr;
|
|
}
|
|
|
|
void *DL::Sym(const char *symbol) {
|
|
if (handle == nullptr) {
|
|
VLOG(2) << "Not opened, do nothing";
|
|
return nullptr;
|
|
}
|
|
#if defined(OS_WIN) && !defined(OS_MINGW) && !defined(OS_CYGWIN)
|
|
void *f = GetProcAddress(handle, symbol);
|
|
if (f == nullptr) {
|
|
VLOG(2) << "Load symbol failed: " << symbol;
|
|
}
|
|
#else
|
|
dlerror(); // reset errors
|
|
void *f = dlsym(handle, symbol);
|
|
const char *error = dlerror();
|
|
if (error != nullptr) {
|
|
VLOG(2) << "Load symbol failed: " << symbol;
|
|
f = nullptr;
|
|
}
|
|
#endif
|
|
return f;
|
|
}
|
|
|
|
int DL::Close() {
|
|
int ret = 0;
|
|
if (handle == nullptr) {
|
|
VLOG(2) << "Not opened, do nothing";
|
|
} else {
|
|
#if defined(OS_WIN) && !defined(OS_MINGW) && !defined(OS_CYGWIN)
|
|
ret = FreeLibrary(handle) ? 0 : 1;
|
|
#else
|
|
ret = dlclose(handle);
|
|
#endif
|
|
handle = nullptr;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
const char *DL::Error() {
|
|
#if defined(OS_WIN) && !defined(OS_MINGW) && !defined(OS_CYGWIN)
|
|
return GetLastErrorAsString().c_str();
|
|
#else
|
|
return dlerror();
|
|
#endif
|
|
}
|
|
|
|
MYNTEYE_END_NAMESPACE
|