Add Elf 32 Support to Courgette.

This change takes advantage of recent refactoring and adds support for
Elf X86 32 executables to courgette. It should have no effect on handling
of Windows PE executables.

We have planned ahead to be able to restrict the code size of the courgette
library in different cases to reduce patcher sizes, but this change does
not yet take advantage of that (all platforms are supported everywhere).

Also, the patcher class currently contains a very small amount of Elf/PE
specific code for recreating relocation tables that cannot (currently) be
compiled out.

BUG=chromium-os:22149
TEST=Please verify that Chrome/Chromium patches can still be generated and
     work.
     Also, please see how much the updater executable which is downloaded to
     users has changed in size since R16.


Review URL: http://codereview.chromium.org/8428009

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@108929 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 17 insertions, 12 deletions

diff --git a/courgette/encode_decode_unittest.cc b/courgette/encode_decode_unittest.cc
index 90f5cd6..20f0e16 100644
--- a/courgette/encode_decode_unittest.cc
+++ b/courgette/encode_decode_unittest.cc
@@ -8,20 +8,20 @@
 
 class EncodeDecodeTest : public BaseTest {
  public:
-  void TestExe(const char *) const;
+  void TestAssembleToStreamDisassemble(std::string file,
+                                       size_t expected_encoded_lenth) const;
 };
 
-void EncodeDecodeTest::TestExe(const char* file_name) const {
-  // Test top-level Courgette API for converting an a file to a binary
-  // assembly representation and back.
-  std::string file1 = FileContents(file_name);
-
-  const void* original_buffer = file1.c_str();
-  size_t original_length = file1.size();
+void EncodeDecodeTest::TestAssembleToStreamDisassemble(
+    std::string file,
+    size_t expected_encoded_lenth) const {
+  const void* original_buffer = file.c_str();
+  size_t original_length = file.length();
 
   courgette::AssemblyProgram* program = NULL;
   const courgette::Status parse_status =
-      courgette::ParseDetectedExecutable(original_buffer, original_length,
+      courgette::ParseDetectedExecutable(original_buffer,
+                                         original_length,
                                          &program);
   EXPECT_EQ(courgette::C_OK, parse_status);
 
@@ -45,7 +45,7 @@ void EncodeDecodeTest::TestExe(const char* file_name) const {
   const void* buffer = sink.Buffer();
   size_t length = sink.Length();
 
-  EXPECT_EQ(971850U, length);
+  EXPECT_EQ(expected_encoded_lenth, length);
 
   courgette::SourceStreamSet sources;
   bool can_get_source_streams = sources.Init(buffer, length);
@@ -68,7 +68,12 @@ void EncodeDecodeTest::TestExe(const char* file_name) const {
   DeleteEncodedProgram(encoded2);
 }
 
+TEST_F(EncodeDecodeTest, PE) {
+  std::string file = FileContents("setup1.exe");
+  TestAssembleToStreamDisassemble(file, 971850);
+}
 
-TEST_F(EncodeDecodeTest, All) {
-  TestExe("setup1.exe");
+TEST_F(EncodeDecodeTest, Elf_Small) {
+  std::string file = FileContents("elf-32-1");
+  TestAssembleToStreamDisassemble(file, 135988);
 }