| // Copyright (c) 2009 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 <string> |
| |
| #include "base/path_service.h" |
| #include "base/file_util.h" |
| #include "base/string_util.h" |
| |
| #include "courgette/courgette.h" |
| #include "courgette/streams.h" |
| |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| class EncodeDecodeTest : public testing::Test { |
| public: |
| void TestExe(const char *) const; |
| |
| private: |
| void SetUp() { |
| PathService::Get(base::DIR_SOURCE_ROOT, &testdata_dir_); |
| file_util::AppendToPath(&testdata_dir_, L"courgette"); |
| file_util::AppendToPath(&testdata_dir_, L"testdata"); |
| } |
| |
| void TearDown() { } |
| |
| // Returns contents of |file_name| as uninterprested bytes stored in a string. |
| std::string FileContents(const char* file_name) const; |
| |
| std::wstring testdata_dir_; // Full path name of testdata directory |
| }; |
| |
| // Reads a test file into a string. |
| std::string EncodeDecodeTest::FileContents(const char* file_name) const { |
| std::wstring file_path = testdata_dir_; |
| file_util::AppendToPath(&file_path, UTF8ToWide(file_name)); |
| std::string file_contents; |
| if (!file_util::ReadFileToString(file_path, &file_contents)) { |
| EXPECT_TRUE(!"Could not read test data"); |
| } |
| return file_contents; |
| } |
| |
| 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(); |
| |
| courgette::AssemblyProgram* program = NULL; |
| const courgette::Status parse_status = |
| courgette::ParseWin32X86PE(original_buffer, original_length, &program); |
| EXPECT_EQ(courgette::C_OK, parse_status); |
| |
| courgette::EncodedProgram* encoded = NULL; |
| |
| const courgette::Status encode_status = Encode(program, &encoded); |
| EXPECT_EQ(courgette::C_OK, encode_status); |
| |
| DeleteAssemblyProgram(program); |
| |
| courgette::SinkStreamSet sinks; |
| const courgette::Status write_status = WriteEncodedProgram(encoded, &sinks); |
| EXPECT_EQ(courgette::C_OK, write_status); |
| |
| DeleteEncodedProgram(encoded); |
| |
| courgette::SinkStream sink; |
| bool can_collect = sinks.CopyTo(&sink); |
| EXPECT_TRUE(can_collect); |
| |
| const void* buffer = sink.Buffer(); |
| size_t length = sink.Length(); |
| |
| EXPECT_EQ(971850, length); |
| |
| courgette::SourceStreamSet sources; |
| bool can_get_source_streams = sources.Init(buffer, length); |
| EXPECT_TRUE(can_get_source_streams); |
| |
| courgette::EncodedProgram *encoded2 = NULL; |
| const courgette::Status read_status = ReadEncodedProgram(&sources, &encoded2); |
| EXPECT_EQ(courgette::C_OK, read_status); |
| |
| courgette::SinkStream assembled; |
| const courgette::Status assemble_status = Assemble(encoded2, &assembled); |
| EXPECT_EQ(courgette::C_OK, assemble_status); |
| |
| const void* assembled_buffer = assembled.Buffer(); |
| size_t assembled_length = assembled.Length(); |
| |
| EXPECT_EQ(original_length, assembled_length); |
| EXPECT_EQ(0, memcmp(original_buffer, assembled_buffer, original_length)); |
| |
| DeleteEncodedProgram(encoded2); |
| } |
| |
| |
| TEST_F(EncodeDecodeTest, All) { |
| TestExe("setup1.exe"); |
| } |