Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(39)

Side by Side Diff: courgette/encode_decode_unittest.cc

Issue 1928683002: [Courgette] ELF: Fix abs32 / rel32 ordering in ParseFile() and restrict rel32 parsing to .text. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 <stddef.h> 5 #include <stddef.h>
6 6
7 #include <memory> 7 #include <memory>
8 8
9 #include "courgette/assembly_program.h" 9 #include "courgette/assembly_program.h"
10 #include "courgette/base_test_unittest.h" 10 #include "courgette/base_test_unittest.h"
11 #include "courgette/courgette.h" 11 #include "courgette/courgette.h"
12 #include "courgette/encoded_program.h" 12 #include "courgette/encoded_program.h"
13 #include "courgette/program_detector.h" 13 #include "courgette/program_detector.h"
14 #include "courgette/streams.h" 14 #include "courgette/streams.h"
15 15
16 class EncodeDecodeTest : public BaseTest { 16 class EncodeDecodeTest : public BaseTest {
17 public: 17 public:
18 void TestAssembleToStreamDisassemble(std::string file, 18 void TestAssembleToStreamDisassemble(std::string file,
19 size_t expected_encoded_lenth) const; 19 size_t expected_encoded_length) const;
20 }; 20 };
21 21
22 void EncodeDecodeTest::TestAssembleToStreamDisassemble( 22 void EncodeDecodeTest::TestAssembleToStreamDisassemble(
23 std::string file, 23 std::string file,
24 size_t expected_encoded_lenth) const { 24 size_t expected_encoded_length) const {
25 const void* original_buffer = file.c_str(); 25 const void* original_buffer = file.c_str();
26 size_t original_length = file.length(); 26 size_t original_length = file.length();
27 27
28 std::unique_ptr<courgette::AssemblyProgram> program; 28 std::unique_ptr<courgette::AssemblyProgram> program;
29 const courgette::Status parse_status = 29 const courgette::Status parse_status =
30 courgette::ParseDetectedExecutable(original_buffer, 30 courgette::ParseDetectedExecutable(original_buffer,
31 original_length, 31 original_length,
32 &program); 32 &program);
33 EXPECT_EQ(courgette::C_OK, parse_status); 33 EXPECT_EQ(courgette::C_OK, parse_status);
34 34
(...skipping 10 matching lines...) Expand all
45 45
46 encoded.reset(); 46 encoded.reset();
47 47
48 courgette::SinkStream sink; 48 courgette::SinkStream sink;
49 bool can_collect = sinks.CopyTo(&sink); 49 bool can_collect = sinks.CopyTo(&sink);
50 EXPECT_TRUE(can_collect); 50 EXPECT_TRUE(can_collect);
51 51
52 const void* buffer = sink.Buffer(); 52 const void* buffer = sink.Buffer();
53 size_t length = sink.Length(); 53 size_t length = sink.Length();
54 54
55 EXPECT_EQ(expected_encoded_lenth, length); 55 EXPECT_EQ(expected_encoded_length, length);
56 56
57 courgette::SourceStreamSet sources; 57 courgette::SourceStreamSet sources;
58 bool can_get_source_streams = sources.Init(buffer, length); 58 bool can_get_source_streams = sources.Init(buffer, length);
59 EXPECT_TRUE(can_get_source_streams); 59 EXPECT_TRUE(can_get_source_streams);
60 60
61 std::unique_ptr<courgette::EncodedProgram> encoded2; 61 std::unique_ptr<courgette::EncodedProgram> encoded2;
62 const courgette::Status read_status = ReadEncodedProgram(&sources, &encoded2); 62 const courgette::Status read_status = ReadEncodedProgram(&sources, &encoded2);
63 EXPECT_EQ(courgette::C_OK, read_status); 63 EXPECT_EQ(courgette::C_OK, read_status);
64 64
65 courgette::SinkStream assembled; 65 courgette::SinkStream assembled;
(...skipping 15 matching lines...) Expand all
81 TestAssembleToStreamDisassemble(file, 972845); 81 TestAssembleToStreamDisassemble(file, 972845);
82 } 82 }
83 83
84 TEST_F(EncodeDecodeTest, PE64) { 84 TEST_F(EncodeDecodeTest, PE64) {
85 std::string file = FileContents("chrome64_1.exe"); 85 std::string file = FileContents("chrome64_1.exe");
86 TestAssembleToStreamDisassemble(file, 809635); 86 TestAssembleToStreamDisassemble(file, 809635);
87 } 87 }
88 88
89 TEST_F(EncodeDecodeTest, Elf_Small) { 89 TEST_F(EncodeDecodeTest, Elf_Small) {
90 std::string file = FileContents("elf-32-1"); 90 std::string file = FileContents("elf-32-1");
91 TestAssembleToStreamDisassemble(file, 136218); 91 TestAssembleToStreamDisassemble(file, 136201);
92 } 92 }
93 93
94 TEST_F(EncodeDecodeTest, Elf_HighBSS) { 94 TEST_F(EncodeDecodeTest, Elf_HighBSS) {
95 std::string file = FileContents("elf-32-high-bss"); 95 std::string file = FileContents("elf-32-high-bss");
96 TestAssembleToStreamDisassemble(file, 7312); 96 TestAssembleToStreamDisassemble(file, 7308);
97 } 97 }
98 98
99 TEST_F(EncodeDecodeTest, Elf_Arm) { 99 TEST_F(EncodeDecodeTest, Elf_Arm) {
100 std::string file = FileContents("elf-armv7"); 100 std::string file = FileContents("elf-armv7");
101 TestAssembleToStreamDisassemble(file, 8531); 101 TestAssembleToStreamDisassemble(file, 8531);
102 } 102 }
OLDNEW
« courgette/disassembler_elf_32.cc ('K') | « courgette/disassembler_elf_32_x86_unittest.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698