mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 10:52:30 +00:00
50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
#include "SharedLibrary.h"
|
|
#include <string>
|
|
#include <base/phdr_cache.h>
|
|
#include <Common/Exception.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int CANNOT_DLOPEN;
|
|
extern const int CANNOT_DLSYM;
|
|
}
|
|
|
|
SharedLibrary::SharedLibrary(std::string_view path, int flags)
|
|
{
|
|
handle = dlopen(path.data(), flags); // NOLINT
|
|
if (!handle)
|
|
throw Exception(ErrorCodes::CANNOT_DLOPEN, "Cannot dlopen: ({})", dlerror()); // NOLINT(concurrency-mt-unsafe) // MT-Safe on Linux, see man dlerror
|
|
|
|
updatePHDRCache();
|
|
|
|
/// NOTE: race condition exists when loading multiple shared libraries concurrently.
|
|
/// We don't care (or add global mutex for this method).
|
|
}
|
|
|
|
SharedLibrary::~SharedLibrary()
|
|
{
|
|
if (handle && dlclose(handle))
|
|
std::terminate();
|
|
}
|
|
|
|
void * SharedLibrary::getImpl(std::string_view name, bool no_throw)
|
|
{
|
|
dlerror(); // NOLINT(concurrency-mt-unsafe) // MT-Safe on Linux, see man dlerror
|
|
|
|
auto * res = dlsym(handle, name.data()); // NOLINT
|
|
|
|
if (char * error = dlerror()) // NOLINT(concurrency-mt-unsafe) // MT-Safe on Linux, see man dlerror
|
|
{
|
|
if (no_throw)
|
|
return nullptr;
|
|
|
|
throw Exception(ErrorCodes::CANNOT_DLSYM, "Cannot dlsym: ({})", error);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
}
|