Skip to content

Commit

Permalink
Add vtable unittest back
Browse files Browse the repository at this point in the history
  • Loading branch information
@fabianbs96
fabianbs96 committed May 1, 2024
1 parent b64b926 commit 68d8f62
Show file tree
Hide file tree
Showing 2 changed files with 173 additions and 0 deletions.
1 change: 1 addition & 0 deletions unittests/PhasarLLVM/ControlFlow/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ set(ControlFlowSources
LLVMBasedICFGExportTest.cpp
LLVMBasedICFGGlobCtorDtorTest.cpp
LLVMBasedICFGSerializationTest.cpp
LLVMVFTableProviderTest.cpp
)

set(LLVM_LINK_COMPONENTS Linker) # The CtorDtorTest needs the linker
Expand Down
172 changes: 172 additions & 0 deletions unittests/PhasarLLVM/ControlFlow/LLVMVFTableProviderTest.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,172 @@
#include "phasar/PhasarLLVM/ControlFlow/LLVMVFTableProvider.h"

#include "phasar/PhasarLLVM/DB/LLVMProjectIRDB.h"
#include "phasar/PhasarLLVM/TypeHierarchy/LLVMVFTable.h"

#include "llvm/ADT/StringRef.h"
#include "llvm/Demangle/Demangle.h"
#include "llvm/IR/DerivedTypes.h"
#include "llvm/IR/Module.h"

#include "TestConfig.h"
#include "gtest/gtest.h"

using namespace psr;

namespace {

using llvm::demangle;

static const llvm::StructType *getType(const LLVMProjectIRDB &IRDB,
llvm::StringRef Name) {
// TODO: Optimize
for (const auto *Ty : IRDB.getModule()->getIdentifiedStructTypes()) {
if (Ty->getName() == Name) {
return Ty;
}
}
return nullptr;
}

// check if the vtables are constructed correctly in more complex scenarios

TEST(VTableTest, VTableConstruction_01) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_1_cpp.ll"});
LLVMVFTableProvider TH(IRDB);
// TODO
}

TEST(VTableTest, VTableConstruction_02) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_7_cpp.ll"});
LLVMVFTableProvider TH(IRDB);
// TODO
}

TEST(VTableTest, VTableConstruction_03) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_8_cpp.ll"});
LLVMVFTableProvider TH(IRDB);

ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Base")));
ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Child")));

EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(0)
->getName()
.str()),
"Base::foo()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))->size(), 2U);
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(0)
->getName()
.str()),
"Child::foo()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(2)
->getName()
.str()),
"Child::baz()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))->size(), 3U);
}

TEST(VTableTest, VTableConstruction_04) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_9_cpp.ll"});
LLVMVFTableProvider TH(IRDB);

ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Base")));
ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Child")));

EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(0)
->getName()
.str()),
"Base::foo()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))->size(), 2U);
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(0)
->getName()
.str()),
"Child::foo()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(2)
->getName()
.str()),
"Child::baz()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))->size(), 3U);
}

TEST(VTableTest, VTableConstruction_05) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_10_cpp.ll"});
LLVMVFTableProvider TH(IRDB);

ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Base")));
ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "struct.Child")));

EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(0)
->getName()
.str()),
"__cxa_pure_virtual");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Base"))->size(), 2U);
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(0)
->getName()
.str()),
"Child::foo()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(1)
->getName()
.str()),
"Base::bar()");
EXPECT_EQ(demangle(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))
->getFunction(2)
->getName()
.str()),
"Child::baz()");
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "struct.Child"))->size(), 3U);
}

TEST(VTableTest, VTableConstruction_6) {
LLVMProjectIRDB IRDB({unittest::PathToLLTestFiles +
"type_hierarchies/type_hierarchy_14_cpp.ll"});

LLVMVFTableProvider TH(IRDB);

ASSERT_TRUE(TH.hasVFTable(getType(IRDB, "class.Base")));
EXPECT_EQ(TH.getVFTableOrNull(getType(IRDB, "class.Base"))->size(), 3U);
}
} // namespace

int main(int Argc, char **Argv) {
::testing::InitGoogleTest(&Argc, Argv);
return RUN_ALL_TESTS();
}

0 comments on commit 68d8f62

Please sign in to comment.