1 | //===-- TestMCDisasmInstanceRISCV.cpp -------------------------------------===// |
2 | |
3 | // |
4 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
5 | // See https://llvm.org/LICENSE.txt for license information. |
6 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
7 | // |
8 | //===----------------------------------------------------------------------===// |
9 | |
10 | #include "llvm/Support/TargetSelect.h" |
11 | #include "gtest/gtest.h" |
12 | |
13 | #include "lldb/Core/Address.h" |
14 | #include "lldb/Core/Disassembler.h" |
15 | #include "lldb/Target/ExecutionContext.h" |
16 | #include "lldb/Utility/ArchSpec.h" |
17 | |
18 | #include "Plugins/Disassembler/LLVMC/DisassemblerLLVMC.h" |
19 | |
20 | using namespace lldb; |
21 | using namespace lldb_private; |
22 | |
23 | namespace { |
24 | class TestMCDisasmInstanceRISCV : public testing::Test { |
25 | public: |
26 | static void SetUpTestCase(); |
27 | static void TearDownTestCase(); |
28 | |
29 | protected: |
30 | }; |
31 | |
32 | void TestMCDisasmInstanceRISCV::SetUpTestCase() { |
33 | llvm::InitializeAllTargets(); |
34 | llvm::InitializeAllAsmPrinters(); |
35 | llvm::InitializeAllTargetMCs(); |
36 | llvm::InitializeAllDisassemblers(); |
37 | DisassemblerLLVMC::Initialize(); |
38 | } |
39 | |
40 | void TestMCDisasmInstanceRISCV::TearDownTestCase() { |
41 | DisassemblerLLVMC::Terminate(); |
42 | } |
43 | } // namespace |
44 | |
45 | TEST_F(TestMCDisasmInstanceRISCV, TestRISCV32Instruction) { |
46 | ArchSpec arch("riscv32-*-linux" ); |
47 | |
48 | const unsigned num_of_instructions = 5; |
49 | uint8_t data[] = { |
50 | 0xef, 0x00, 0x00, 0x00, // call -- jal x1, 0 |
51 | 0xe7, 0x00, 0x00, 0x00, // call -- jalr x1, x0, 0 |
52 | 0x6f, 0x00, 0x00, 0x00, // jump -- jal x0, 0 |
53 | 0x67, 0x00, 0x00, 0x00, // jump -- jalr x0, x0, 0 |
54 | 0x67, 0x80, 0x00, 0x00 // ret -- jalr x0, x1, 0 |
55 | }; |
56 | |
57 | DisassemblerSP disass_sp; |
58 | Address start_addr(0x100); |
59 | disass_sp = |
60 | Disassembler::DisassembleBytes(arch, plugin_name: nullptr, flavor: nullptr, start: start_addr, bytes: &data, |
61 | length: sizeof (data), max_num_instructions: num_of_instructions, data_from_file: false); |
62 | |
63 | // If we failed to get a disassembler, we can assume it is because |
64 | // the llvm we linked against was not built with the riscv target, |
65 | // and we should skip these tests without marking anything as failing. |
66 | if (!disass_sp) |
67 | return; |
68 | |
69 | const InstructionList inst_list(disass_sp->GetInstructionList()); |
70 | EXPECT_EQ(num_of_instructions, inst_list.GetSize()); |
71 | |
72 | InstructionSP inst_sp; |
73 | inst_sp = inst_list.GetInstructionAtIndex(idx: 0); |
74 | EXPECT_TRUE(inst_sp->IsCall()); |
75 | EXPECT_TRUE(inst_sp->DoesBranch()); |
76 | |
77 | inst_sp = inst_list.GetInstructionAtIndex(idx: 1); |
78 | EXPECT_TRUE(inst_sp->IsCall()); |
79 | EXPECT_TRUE(inst_sp->DoesBranch()); |
80 | |
81 | inst_sp = inst_list.GetInstructionAtIndex(idx: 2); |
82 | EXPECT_FALSE(inst_sp->IsCall()); |
83 | EXPECT_TRUE(inst_sp->DoesBranch()); |
84 | |
85 | inst_sp = inst_list.GetInstructionAtIndex(idx: 3); |
86 | EXPECT_FALSE(inst_sp->IsCall()); |
87 | EXPECT_TRUE(inst_sp->DoesBranch()); |
88 | |
89 | inst_sp = inst_list.GetInstructionAtIndex(idx: 4); |
90 | EXPECT_FALSE(inst_sp->IsCall()); |
91 | EXPECT_TRUE(inst_sp->DoesBranch()); |
92 | } |
93 | |