1 // Copyright 2011 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "courgette/disassembler_win32_x86.h"
13 #include "base/ranges/algorithm.h"
14 #include "courgette/base_test_unittest.h"
16 class DisassemblerWin32X86Test : public BaseTest {
19 void TestExe64ShouldFail() const;
20 void TestResourceDll() const;
23 void DisassemblerWin32X86Test::TestExe() const {
24 std::string file1 = FileContents("setup1.exe");
26 std::unique_ptr<courgette::DisassemblerWin32X86> disassembler(
27 new courgette::DisassemblerWin32X86(
28 reinterpret_cast<const uint8_t*>(file1.c_str()), file1.length()));
30 bool can_parse_header = disassembler->ParseHeader();
31 EXPECT_TRUE(can_parse_header);
33 // The executable is the whole file, not 'embedded' with the file
34 EXPECT_EQ(file1.length(), disassembler->length());
36 EXPECT_TRUE(disassembler->ok());
37 EXPECT_TRUE(disassembler->has_text_section());
38 EXPECT_EQ(449536U, disassembler->size_of_code());
39 EXPECT_EQ(courgette::DisassemblerWin32X86::SectionName(
40 disassembler->RVAToSection(0x00401234 - 0x00400000)),
41 std::string(".text"));
43 EXPECT_EQ(0U, disassembler->RVAToFileOffset(0));
44 EXPECT_EQ(1024U, disassembler->RVAToFileOffset(4096));
45 EXPECT_EQ(46928U, disassembler->RVAToFileOffset(50000));
47 std::vector<courgette::RVA> relocs;
48 bool can_parse_relocs = disassembler->ParseRelocs(&relocs);
49 EXPECT_TRUE(can_parse_relocs);
50 EXPECT_TRUE(base::ranges::is_sorted(relocs));
52 const uint8_t* offset_p = disassembler->FileOffsetToPointer(0);
53 EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
54 reinterpret_cast<const void*>(offset_p));
55 EXPECT_EQ('M', offset_p[0]);
56 EXPECT_EQ('Z', offset_p[1]);
58 const uint8_t* rva_p = disassembler->RVAToPointer(0);
59 EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
60 reinterpret_cast<const void*>(rva_p));
61 EXPECT_EQ('M', rva_p[0]);
62 EXPECT_EQ('Z', rva_p[1]);
65 void DisassemblerWin32X86Test::TestExe64ShouldFail() const {
66 std::string file1 = FileContents("pe-64.exe");
68 std::unique_ptr<courgette::DisassemblerWin32X86> disassembler(
69 new courgette::DisassemblerWin32X86(
70 reinterpret_cast<const uint8_t*>(file1.c_str()), file1.length()));
72 bool can_parse_header = disassembler->ParseHeader();
73 EXPECT_FALSE(can_parse_header);
75 // The executable is the whole file, not 'embedded' with the file
76 EXPECT_EQ(file1.length(), disassembler->length());
78 EXPECT_FALSE(disassembler->ok());
81 void DisassemblerWin32X86Test::TestResourceDll() const {
82 std::string file1 = FileContents("en-US.dll");
84 std::unique_ptr<courgette::DisassemblerWin32X86> disassembler(
85 new courgette::DisassemblerWin32X86(
86 reinterpret_cast<const uint8_t*>(file1.c_str()), file1.length()));
88 bool can_parse_header = disassembler->ParseHeader();
89 EXPECT_FALSE(can_parse_header);
91 // The executable is the whole file, not 'embedded' with the file
92 EXPECT_EQ(file1.length(), disassembler->length());
94 EXPECT_FALSE(disassembler->ok());
97 TEST_F(DisassemblerWin32X86Test, All) {
99 TestExe64ShouldFail();