text: // Copyright (c) 2011 The Chromium Authors. All rights reserved.
  // Use of this source code is governed by a BSD-style license that can be
  // found in the LICENSE file.
  
  #include "courgette/assembly_program.h"
  #include "courgette/base_test_unittest.h"
  #include "courgette/disassembler_elf_32_x86.h"
  
  class DisassemblerElf32X86Test : public BaseTest {
   public:
  
    void TestExe(const char* file_name,
                 size_t expected_abs_count,
                 size_t expected_rel_count) const;
  };
  
  void DisassemblerElf32X86Test::TestExe(const char* file_name,
                                         size_t expected_abs_count,
                                         size_t expected_rel_count) const {
    std::string file1 = FileContents(file_name);
  
    scoped_ptr<courgette::DisassemblerElf32X86> disassembler(
        new courgette::DisassemblerElf32X86(file1.c_str(), file1.length()));
  
    bool can_parse_header = disassembler->ParseHeader();
    EXPECT_TRUE(can_parse_header);
    EXPECT_TRUE(disassembler->ok());
  
    // The length of the disassembled value will be slightly smaller than the
    // real file, since trailing debug info is not included
    EXPECT_EQ(file1.length(), disassembler->length());
  
    const uint8* offset_p = disassembler->OffsetToPointer(0);
    EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
              reinterpret_cast<const void*>(offset_p));
    EXPECT_EQ(0x7F, offset_p[0]);
    EXPECT_EQ('E', offset_p[1]);
    EXPECT_EQ('L', offset_p[2]);
    EXPECT_EQ('F', offset_p[3]);
  
    courgette::AssemblyProgram* program =
      new courgette::AssemblyProgram(courgette::EXE_ELF_32_X86);
  
    EXPECT_TRUE(disassembler->Disassemble(program));
  
    EXPECT_EQ(disassembler->Abs32Locations().size(), expected_abs_count);
    EXPECT_EQ(disassembler->Rel32Locations().size(), expected_rel_count);
  
    // Prove that none of the rel32 RVAs overlap with abs32 RVAs
    std::set<courgette::RVA> abs(disassembler->Abs32Locations().begin(),
                                 disassembler->Abs32Locations().end());
    std::set<courgette::DisassemblerElf32::TypedRVA*>
      rel(disassembler->Rel32Locations().begin(),
          disassembler->Rel32Locations().end());
    for (std::vector<courgette::DisassemblerElf32::TypedRVA*>::iterator
           rel32 = disassembler->Rel32Locations().begin();
         rel32 !=  disassembler->Rel32Locations().end();
         rel32++) {
      EXPECT_TRUE(abs.find((*rel32)->rva()) == abs.end());
    }
  
    for (std::vector<courgette::RVA>::iterator abs32 =
          disassembler->Abs32Locations().begin();
         abs32 !=  disassembler->Abs32Locations().end();
         abs32++) {
      bool found = false;
      for (std::vector<courgette::DisassemblerElf32::TypedRVA*>::iterator
             rel32 = disassembler->Rel32Locations().begin();
           rel32 !=  disassembler->Rel32Locations().end();
           rel32++) {
        if (*abs32 == (*rel32)->rva()) {
          found = true;
          break;
        }
      }
      EXPECT_TRUE(!found);
    }
    delete program;
  }
  
  TEST_F(DisassemblerElf32X86Test, All) {
    TestExe("elf-32-1", 200, 3442);
  }
  