mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-30 17:46:06 +00:00
127 lines
3.8 KiB
C++
127 lines
3.8 KiB
C++
//===- llvm/unittest/Support/DynamicLibrary/DynamicLibraryTest.cpp --------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/DynamicLibrary.h"
|
|
#include "llvm/Config/config.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "PipSqueak.h"
|
|
|
|
// FIXME: Missing globals/DSO https://github.com/llvm/llvm-project/issues/57206.
|
|
#if !LLVM_HWADDRESS_SANITIZER_BUILD
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::sys;
|
|
|
|
std::string LibPath(const std::string Name = "PipSqueak") {
|
|
const auto &Argvs = testing::internal::GetArgvs();
|
|
const char *Argv0 =
|
|
Argvs.size() > 0 ? Argvs[0].c_str() : "DynamicLibraryTests";
|
|
void *Ptr = (void*)(intptr_t)TestA;
|
|
std::string Path = fs::getMainExecutable(Argv0, Ptr);
|
|
llvm::SmallString<256> Buf(path::parent_path(Path));
|
|
path::append(Buf, (Name + LLVM_PLUGIN_EXT).c_str());
|
|
return std::string(Buf.str());
|
|
}
|
|
|
|
#if defined(_WIN32) || (defined(HAVE_DLFCN_H) && defined(HAVE_DLOPEN))
|
|
|
|
typedef void (*SetStrings)(std::string &GStr, std::string &LStr);
|
|
typedef void (*TestOrder)(std::vector<std::string> &V);
|
|
typedef const char *(*GetString)();
|
|
|
|
template <class T> static T FuncPtr(void *Ptr) {
|
|
union {
|
|
T F;
|
|
void *P;
|
|
} Tmp;
|
|
Tmp.P = Ptr;
|
|
return Tmp.F;
|
|
}
|
|
template <class T> static void* PtrFunc(T *Func) {
|
|
union {
|
|
T *F;
|
|
void *P;
|
|
} Tmp;
|
|
Tmp.F = Func;
|
|
return Tmp.P;
|
|
}
|
|
|
|
static const char *OverloadTestA() { return "OverloadCall"; }
|
|
|
|
std::string StdString(const char *Ptr) { return Ptr ? Ptr : ""; }
|
|
|
|
TEST(DynamicLibrary, Overload) {
|
|
{
|
|
std::string Err;
|
|
DynamicLibrary DL =
|
|
DynamicLibrary::getPermanentLibrary(LibPath().c_str(), &Err);
|
|
EXPECT_TRUE(DL.isValid());
|
|
EXPECT_TRUE(Err.empty());
|
|
|
|
GetString GS = FuncPtr<GetString>(DL.getAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_NE(GS, &TestA);
|
|
EXPECT_EQ(StdString(GS()), "LibCall");
|
|
|
|
GS = FuncPtr<GetString>(DynamicLibrary::SearchForAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_NE(GS, &TestA);
|
|
EXPECT_EQ(StdString(GS()), "LibCall");
|
|
|
|
DL = DynamicLibrary::getPermanentLibrary(nullptr, &Err);
|
|
EXPECT_TRUE(DL.isValid());
|
|
EXPECT_TRUE(Err.empty());
|
|
|
|
// Test overloading local symbols does not occur by default
|
|
GS = FuncPtr<GetString>(DynamicLibrary::SearchForAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_EQ(GS, &TestA);
|
|
EXPECT_EQ(StdString(GS()), "ProcessCall");
|
|
|
|
GS = FuncPtr<GetString>(DL.getAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_EQ(GS, &TestA);
|
|
EXPECT_EQ(StdString(GS()), "ProcessCall");
|
|
|
|
// Test overloading by forcing library priority when searching for a symbol
|
|
DynamicLibrary::SearchOrder = DynamicLibrary::SO_LoadedFirst;
|
|
GS = FuncPtr<GetString>(DynamicLibrary::SearchForAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_NE(GS, &TestA);
|
|
EXPECT_EQ(StdString(GS()), "LibCall");
|
|
|
|
DynamicLibrary::AddSymbol("TestA", PtrFunc(&OverloadTestA));
|
|
GS = FuncPtr<GetString>(DL.getAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_NE(GS, &OverloadTestA);
|
|
|
|
GS = FuncPtr<GetString>(DynamicLibrary::SearchForAddressOfSymbol("TestA"));
|
|
EXPECT_NE(GS, nullptr);
|
|
EXPECT_EQ(GS, &OverloadTestA);
|
|
EXPECT_EQ(StdString(GS()), "OverloadCall");
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
TEST(DynamicLibrary, Unsupported) {
|
|
std::string Err;
|
|
DynamicLibrary DL =
|
|
DynamicLibrary::getPermanentLibrary(LibPath().c_str(), &Err);
|
|
EXPECT_FALSE(DL.isValid());
|
|
EXPECT_EQ(Err, "dlopen() not supported on this platform");
|
|
}
|
|
|
|
#endif
|
|
|
|
#endif
|