1// Copyright (c) 2011 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "courgette/assembly_program.h"
6#include "courgette/base_test_unittest.h"
7#include "courgette/disassembler_elf_32_x86.h"
8
9class DisassemblerElf32X86Test : public BaseTest {
10 public:
11
12  void TestExe(const char* file_name,
13               size_t expected_abs_count,
14               size_t expected_rel_count) const;
15};
16
17void DisassemblerElf32X86Test::TestExe(const char* file_name,
18                                       size_t expected_abs_count,
19                                       size_t expected_rel_count) const {
20  std::string file1 = FileContents(file_name);
21
22  scoped_ptr<courgette::DisassemblerElf32X86> disassembler(
23      new courgette::DisassemblerElf32X86(file1.c_str(), file1.length()));
24
25  bool can_parse_header = disassembler->ParseHeader();
26  EXPECT_TRUE(can_parse_header);
27  EXPECT_TRUE(disassembler->ok());
28
29  // The length of the disassembled value will be slightly smaller than the
30  // real file, since trailing debug info is not included
31  EXPECT_EQ(file1.length(), disassembler->length());
32
33  const uint8* offset_p = disassembler->OffsetToPointer(0);
34  EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
35            reinterpret_cast<const void*>(offset_p));
36  EXPECT_EQ(0x7F, offset_p[0]);
37  EXPECT_EQ('E', offset_p[1]);
38  EXPECT_EQ('L', offset_p[2]);
39  EXPECT_EQ('F', offset_p[3]);
40
41  courgette::AssemblyProgram* program =
42    new courgette::AssemblyProgram(courgette::EXE_ELF_32_X86);
43
44  EXPECT_TRUE(disassembler->Disassemble(program));
45
46  EXPECT_EQ(disassembler->Abs32Locations().size(), expected_abs_count);
47  EXPECT_EQ(disassembler->Rel32Locations().size(), expected_rel_count);
48
49  // Prove that none of the rel32 RVAs overlap with abs32 RVAs
50  std::set<courgette::RVA> abs(disassembler->Abs32Locations().begin(),
51                               disassembler->Abs32Locations().end());
52  std::set<courgette::DisassemblerElf32::TypedRVA*>
53    rel(disassembler->Rel32Locations().begin(),
54        disassembler->Rel32Locations().end());
55  for (std::vector<courgette::DisassemblerElf32::TypedRVA*>::iterator
56         rel32 = disassembler->Rel32Locations().begin();
57       rel32 !=  disassembler->Rel32Locations().end();
58       rel32++) {
59    EXPECT_TRUE(abs.find((*rel32)->rva()) == abs.end());
60  }
61
62  for (std::vector<courgette::RVA>::iterator abs32 =
63        disassembler->Abs32Locations().begin();
64       abs32 !=  disassembler->Abs32Locations().end();
65       abs32++) {
66    bool found = false;
67    for (std::vector<courgette::DisassemblerElf32::TypedRVA*>::iterator
68           rel32 = disassembler->Rel32Locations().begin();
69         rel32 !=  disassembler->Rel32Locations().end();
70         rel32++) {
71      if (*abs32 == (*rel32)->rva()) {
72        found = true;
73        break;
74      }
75    }
76    EXPECT_TRUE(!found);
77  }
78  delete program;
79}
80
81TEST_F(DisassemblerElf32X86Test, All) {
82  TestExe("elf-32-1", 200, 3442);
83}
84