Add armel testcase

[frida/frida.git] / src / disassembler / llvm / tests / armelLinuxDisassemblerTest.cxx

diff --git a/src/disassembler/llvm/tests/armelLinuxDisassemblerTest.cxx b/src/disassembler/llvm/tests/armelLinuxDisassemblerTest.cxx
new file mode 100644 (file)
index 0000000..4e79972
--- /dev/null
+++ b/src/disassembler/llvm/tests/armelLinuxDisassemblerTest.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(22, block->getInstructions().size());
+       EXPECT_EQ(0x1068c, block->getNextBlock(0));
+       EXPECT_EQ(0x10684, block->getNextBlock(1));
+
+       EXPECT_TRUE(blocks.find(0x1068c) != blocks.end());
+       EXPECT_TRUE(blocks.find(0x10684) != 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(18, block->getInstructions().size());
+       EXPECT_EQ(0x0, block->getNextBlock(0));
+       EXPECT_EQ(0x0, block->getNextBlock(1));
+}
+
+TEST(llvmDisassemblerTest, armelLinuxOpenBinaryBasicStructure) {
+       SCOPED_TRACE("opening cryptsample.clang.armel.elf");
+       InformationManager manager;
+
+       llvm::InitializeAllTargetInfos();
+       llvm::InitializeAllTargetMCs();
+       llvm::InitializeAllAsmParsers();
+       llvm::InitializeAllDisassemblers();
+
+       manager.reset(std::string(TEST_DATA_DIRECTORY)
+                     + "/disassembler/binaries/crypt.clang.O2.armel.elf");
+
+       ASSERT_NE(nullptr, manager.getDisassembler());
+       EXPECT_EQ(0x104a8, manager.getDisassembler()->entryAddress());
+       testCheckFunction(manager.getFunction(0x105d8));
+       testMainFunction(manager.getFunction(0x1062c));
+}
+
+TEST(llvmDisassemblerTest, armelLinuxLoadBinaryBasicStructure) {
+       SCOPED_TRACE("loading cryptsample.clang.armel.elf");
+       InformationManager manager;
+
+       llvm::InitializeAllTargetInfos();
+       llvm::InitializeAllTargetMCs();
+       llvm::InitializeAllAsmParsers();
+       llvm::InitializeAllDisassemblers();
+
+       manager.load(std::string(TEST_DATA_DIRECTORY)
+                     + "/disassembler/binaries/crypt.clang.O2.armel.elf.frida");
+
+       ASSERT_NE(nullptr, manager.getDisassembler());
+       EXPECT_EQ(0x104a8, manager.getDisassembler()->entryAddress());
+       testCheckFunction(manager.getFunction(0x105d8));
+       testMainFunction(manager.getFunction(0x1062c));
+}