ClickHouse/programs/library-bridge/SharedLibrary.cpp

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

50 lines
1.2 KiB
C++
Raw Normal View History

#include "SharedLibrary.h"
#include <string>
2021-10-02 07:13:14 +00:00
#include <base/phdr_cache.h>
2022-09-17 01:02:34 +00:00
#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)
{
2021-03-14 11:29:02 +00:00
handle = dlopen(path.data(), flags);
if (!handle)
throw Exception(ErrorCodes::CANNOT_DLOPEN, "Cannot dlopen: ({})", dlerror()); // NOLINT(concurrency-mt-unsafe) // MT-Safe on Linux, see man dlerror
2019-07-25 19:56:51 +00:00
updatePHDRCache();
2019-07-28 14:55:02 +00:00
/// 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
2021-03-14 11:29:02 +00:00
auto * res = dlsym(handle, name.data());
if (char * error = dlerror()) // NOLINT(concurrency-mt-unsafe) // MT-Safe on Linux, see man dlerror
{
if (no_throw)
return nullptr;
2021-03-14 11:29:02 +00:00
throw Exception(ErrorCodes::CANNOT_DLSYM, "Cannot dlsym: ({})", error);
}
return res;
}
}