X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2Fllvm%2Ftests%2Famd64LinuxDisassemblerTest.cxx;fp=src%2Fdisassembler%2Fllvm%2Ftests%2Famd64LinuxDisassemblerTest.cxx;h=932f79a96216599c81db5820f3ee351ca12922d5;hp=0000000000000000000000000000000000000000;hb=e232dc612584f55a6426051c8f59b6d9a00c3029;hpb=85728a79afa06bc7d44a931225a9231efb55df74

diff --git a/src/disassembler/llvm/tests/amd64LinuxDisassemblerTest.cxx b/src/disassembler/llvm/tests/amd64LinuxDisassemblerTest.cxx
new file mode 100644
index 0000000..932f79a
--- /dev/null
+++ b/src/disassembler/llvm/tests/amd64LinuxDisassemblerTest.cxx
@@ -0,0 +1,72 @@
+#include <gtest/gtest.h>
+#include <string>
+
+#include "core/InformationManager.hxx"
+#include "core/Function.hxx"
+#include "core/BasicBlock.hxx"
+#include "disassembler/llvm/LLVMDisassembler.hxx"
+
+extern char * TEST_DATA_DIRECTORY;
+
+static void testMainFunction(Function* main) {
+	ASSERT_NE(nullptr, main);
+	EXPECT_STREQ("main", main->getName().c_str());
+
+	auto blocks = main->blocks();
+	auto block = blocks.begin()->second;
+	EXPECT_EQ(4, blocks.size());
+	EXPECT_EQ(17, block->getInstructions().size());
+	EXPECT_EQ(0x4007ea, block->getNextBlock(0));
+	EXPECT_EQ(0x4007ce, block->getNextBlock(1));
+
+	EXPECT_TRUE(blocks.find(0x4007ea) != blocks.end());
+	EXPECT_TRUE(blocks.find(0x4007ce) != blocks.end());
+}
+
+static void testCheckFunction(Function* check) {
+	ASSERT_NE(nullptr, check);
+	EXPECT_STREQ("check", check->getName().c_str());
+
+	auto blocks = check->blocks();
+	auto block = blocks.begin()->second;
+	EXPECT_EQ(1, blocks.size());
+	EXPECT_EQ(29, block->getInstructions().size());
+	EXPECT_EQ(0x0, block->getNextBlock(0));
+	EXPECT_EQ(0x0, block->getNextBlock(1));
+}
+
+TEST(llvmDisassemblerTest, amd64LinuxOpenBinaryBasicStructure) {
+	SCOPED_TRACE("opening cryptsample.clang.amd64.elf");
+	InformationManager manager;
+
+	llvm::InitializeAllTargetInfos();
+	llvm::InitializeAllTargetMCs();
+	llvm::InitializeAllAsmParsers();
+	llvm::InitializeAllDisassemblers();
+
+	manager.reset(std::string(TEST_DATA_DIRECTORY)
+	              + "/disassembler/binaries/crypt.clang.O2.amd64.elf");
+
+	ASSERT_NE(nullptr, manager.getDisassembler());
+	EXPECT_EQ(0x4005F0, manager.getDisassembler()->entryAddress());
+	testCheckFunction(manager.getFunction(0x4006f0));
+	testMainFunction(manager.getFunction(0x400770));
+}
+
+TEST(llvmDisassemblerTest, amd64LinuxLoadBinaryBasicStructure) {
+	SCOPED_TRACE("loading cryptsample.clang.amd64.elf");
+	InformationManager manager;
+
+	llvm::InitializeAllTargetInfos();
+	llvm::InitializeAllTargetMCs();
+	llvm::InitializeAllAsmParsers();
+	llvm::InitializeAllDisassemblers();
+
+	manager.load(std::string(TEST_DATA_DIRECTORY)
+	              + "/disassembler/binaries/crypt.clang.O2.amd64.elf.frida");
+
+	ASSERT_NE(nullptr, manager.getDisassembler());
+	EXPECT_EQ(0x4005F0, manager.getDisassembler()->entryAddress());
+	testCheckFunction(manager.getFunction(0x4006f0));
+	testMainFunction(manager.getFunction(0x400770));
+}