| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "courgette/base_test_unittest.h" | 5 #include "courgette/base_test_unittest.h" |
| 6 #include "courgette/courgette.h" | 6 #include "courgette/courgette.h" |
| 7 #include "courgette/streams.h" | 7 #include "courgette/streams.h" |
| 8 | 8 |
| 9 class EncodeDecodeTest : public BaseTest { | 9 class EncodeDecodeTest : public BaseTest { |
| 10 public: | 10 public: |
| 11 void TestExe(const char *) const; | 11 void TestExe(const char *) const; |
| 12 }; | 12 }; |
| 13 | 13 |
| 14 void EncodeDecodeTest::TestExe(const char* file_name) const { | 14 void EncodeDecodeTest::TestExe(const char* file_name) const { |
| 15 // Test top-level Courgette API for converting an a file to a binary | 15 // Test top-level Courgette API for converting an a file to a binary |
| 16 // assembly representation and back. | 16 // assembly representation and back. |
| 17 std::string file1 = FileContents(file_name); | 17 std::string file1 = FileContents(file_name); |
| 18 | 18 |
| 19 const void* original_buffer = file1.c_str(); | 19 const void* original_buffer = file1.c_str(); |
| 20 size_t original_length = file1.size(); | 20 size_t original_length = file1.size(); |
| 21 | 21 |
| 22 courgette::AssemblyProgram* program = NULL; | 22 courgette::AssemblyProgram* program = NULL; |
| 23 const courgette::Status parse_status = | 23 const courgette::Status parse_status = |
| 24 courgette::ParseWin32X86PE(original_buffer, original_length, &program); | 24 courgette::ParseDetectedExecutable(original_buffer, original_length, |
| 25 &program); |
| 25 EXPECT_EQ(courgette::C_OK, parse_status); | 26 EXPECT_EQ(courgette::C_OK, parse_status); |
| 26 | 27 |
| 27 courgette::EncodedProgram* encoded = NULL; | 28 courgette::EncodedProgram* encoded = NULL; |
| 28 | 29 |
| 29 const courgette::Status encode_status = Encode(program, &encoded); | 30 const courgette::Status encode_status = Encode(program, &encoded); |
| 30 EXPECT_EQ(courgette::C_OK, encode_status); | 31 EXPECT_EQ(courgette::C_OK, encode_status); |
| 31 | 32 |
| 32 DeleteAssemblyProgram(program); | 33 DeleteAssemblyProgram(program); |
| 33 | 34 |
| 34 courgette::SinkStreamSet sinks; | 35 courgette::SinkStreamSet sinks; |
| (...skipping 29 matching lines...) Expand all Loading... |
| 64 EXPECT_EQ(original_length, assembled_length); | 65 EXPECT_EQ(original_length, assembled_length); |
| 65 EXPECT_EQ(0, memcmp(original_buffer, assembled_buffer, original_length)); | 66 EXPECT_EQ(0, memcmp(original_buffer, assembled_buffer, original_length)); |
| 66 | 67 |
| 67 DeleteEncodedProgram(encoded2); | 68 DeleteEncodedProgram(encoded2); |
| 68 } | 69 } |
| 69 | 70 |
| 70 | 71 |
| 71 TEST_F(EncodeDecodeTest, All) { | 72 TEST_F(EncodeDecodeTest, All) { |
| 72 TestExe("setup1.exe"); | 73 TestExe("setup1.exe"); |
| 73 } | 74 } |
| OLD | NEW |