courgette/disassembler_win32_x86_unittest.cc

   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/disassembler_win32_x86.h"
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/stl_util.h"
   9 #include "courgette/base_test_unittest.h"
  10
  11 class DisassemblerWin32X86Test : public BaseTest {
  12  public:
  13   void TestExe() const;
  14   void TestExe64() const;
  15   void TestResourceDll() const;
  16 };
  17
  18 void DisassemblerWin32X86Test::TestExe() const {
  19   std::string file1 = FileContents("setup1.exe");
  20
  21   scoped_ptr<courgette::DisassemblerWin32X86> disassembler(
  22       new courgette::DisassemblerWin32X86(file1.c_str(), file1.length()));
  23
  24   bool can_parse_header = disassembler->ParseHeader();
  25   EXPECT_TRUE(can_parse_header);
  26
  27   // The executable is the whole file, not 'embedded' with the file
  28   EXPECT_EQ(file1.length(), disassembler->length());
  29
  30   EXPECT_TRUE(disassembler->ok());
  31   EXPECT_TRUE(disassembler->has_text_section());
  32   EXPECT_EQ(449536U, disassembler->size_of_code());
  33   EXPECT_TRUE(disassembler->is_32bit());
  34   EXPECT_EQ(courgette::DisassemblerWin32X86::SectionName(
  35       disassembler->RVAToSection(0x00401234 - 0x00400000)),
  36       std::string(".text"));
  37
  38   EXPECT_EQ(0, disassembler->RVAToFileOffset(0));
  39   EXPECT_EQ(1024, disassembler->RVAToFileOffset(4096));
  40   EXPECT_EQ(46928, disassembler->RVAToFileOffset(50000));
  41
  42   std::vector<courgette::RVA> relocs;
  43   bool can_parse_relocs = disassembler->ParseRelocs(&relocs);
  44   EXPECT_TRUE(can_parse_relocs);
  45   EXPECT_TRUE(base::STLIsSorted(relocs));
  46
  47   const uint8* offset_p = disassembler->OffsetToPointer(0);
  48   EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
  49             reinterpret_cast<const void*>(offset_p));
  50   EXPECT_EQ('M', offset_p[0]);
  51   EXPECT_EQ('Z', offset_p[1]);
  52
  53   const uint8* rva_p = disassembler->RVAToPointer(0);
  54   EXPECT_EQ(reinterpret_cast<const void*>(file1.c_str()),
  55             reinterpret_cast<const void*>(rva_p));
  56   EXPECT_EQ('M', rva_p[0]);
  57   EXPECT_EQ('Z', rva_p[1]);
  58 }
  59
  60 void DisassemblerWin32X86Test::TestExe64() const {
  61   std::string file1 = FileContents("pe-64.exe");
  62
  63   scoped_ptr<courgette::DisassemblerWin32X86> disassembler(
  64       new courgette::DisassemblerWin32X86(file1.c_str(), file1.length()));
  65
  66   bool can_parse_header = disassembler->ParseHeader();
  67   EXPECT_FALSE(can_parse_header);
  68
  69   // The executable is the whole file, not 'embedded' with the file
  70   EXPECT_EQ(file1.length(), disassembler->length());
  71
  72   EXPECT_FALSE(disassembler->ok());
  73   EXPECT_TRUE(disassembler->has_text_section());
  74   EXPECT_EQ(43008U, disassembler->size_of_code());
  75   EXPECT_FALSE(disassembler->is_32bit());
  76 }
  77
  78 void DisassemblerWin32X86Test::TestResourceDll() const {
  79   std::string file1 = FileContents("en-US.dll");
  80
  81   scoped_ptr<courgette::DisassemblerWin32X86> disassembler(
  82       new courgette::DisassemblerWin32X86(file1.c_str(), file1.length()));
  83
  84   bool can_parse_header = disassembler->ParseHeader();
  85   EXPECT_FALSE(can_parse_header);
  86
  87   // The executable is the whole file, not 'embedded' with the file
  88   EXPECT_EQ(file1.length(), disassembler->length());
  89
  90   EXPECT_FALSE(disassembler->ok());
  91   EXPECT_FALSE(disassembler->has_text_section());
  92   EXPECT_EQ(0U, disassembler->size_of_code());
  93   EXPECT_TRUE(disassembler->is_32bit());
  94 }
  95
  96 TEST_F(DisassemblerWin32X86Test, All) {
  97   TestExe();
  98   TestExe64();
  99   TestResourceDll();
 100 }