| 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 // Fuzz testing for EncodedProgram serialized format and assembly. | 5 // Fuzz testing for EncodedProgram serialized format and assembly. |
| 6 // | 6 // |
| 7 // We would like some assurance that if an EncodedProgram is malformed we will | 7 // We would like some assurance that if an EncodedProgram is malformed we will |
| 8 // not crash. The EncodedProgram could be malformed either due to malicious | 8 // not crash. The EncodedProgram could be malformed either due to malicious |
| 9 // attack to due to an error in patch generation. | 9 // attack to due to an error in patch generation. |
| 10 // | 10 // |
| (...skipping 28 matching lines...) Expand all Loading... |
| 39 // Loads an executable and does fuzz testing in the serialized format. | 39 // Loads an executable and does fuzz testing in the serialized format. |
| 40 void DecodeFuzzTest::FuzzExe(const char* file_name) const { | 40 void DecodeFuzzTest::FuzzExe(const char* file_name) const { |
| 41 std::string file1 = FileContents(file_name); | 41 std::string file1 = FileContents(file_name); |
| 42 | 42 |
| 43 const uint8_t* original_data = reinterpret_cast<const uint8_t*>(file1.data()); | 43 const uint8_t* original_data = reinterpret_cast<const uint8_t*>(file1.data()); |
| 44 size_t original_length = file1.length(); | 44 size_t original_length = file1.length(); |
| 45 courgette::CourgetteFlow flow; | 45 courgette::CourgetteFlow flow; |
| 46 | 46 |
| 47 courgette::RegionBuffer original_buffer( | 47 courgette::RegionBuffer original_buffer( |
| 48 courgette::Region(original_data, original_length)); | 48 courgette::Region(original_data, original_length)); |
| 49 flow.ReadAssemblyProgramFromBuffer(flow.ONLY, original_buffer, false); | 49 flow.ReadDisassemblerFromBuffer(flow.ONLY, original_buffer); |
| 50 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 51 EXPECT_TRUE(nullptr != flow.data(flow.ONLY)->disassembler.get()); |
| 52 |
| 53 flow.CreateAssemblyProgramFromDisassembler(flow.ONLY, false); |
| 50 EXPECT_EQ(courgette::C_OK, flow.status()); | 54 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 51 EXPECT_TRUE(nullptr != flow.data(flow.ONLY)->program.get()); | 55 EXPECT_TRUE(nullptr != flow.data(flow.ONLY)->program.get()); |
| 52 | 56 |
| 53 flow.CreateEncodedProgramFromAssemblyProgram(flow.ONLY); | 57 flow.CreateEncodedProgramFromDisassemblerAndAssemblyProgram(flow.ONLY); |
| 54 EXPECT_EQ(courgette::C_OK, flow.status()); | 58 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 55 EXPECT_TRUE(nullptr != flow.data(flow.ONLY)->encoded.get()); | 59 EXPECT_TRUE(nullptr != flow.data(flow.ONLY)->encoded.get()); |
| 56 | 60 |
| 57 flow.DestroyAssemblyProgram(flow.ONLY); | 61 flow.DestroyAssemblyProgram(flow.ONLY); |
| 58 EXPECT_EQ(courgette::C_OK, flow.status()); | 62 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 59 EXPECT_TRUE(nullptr == flow.data(flow.ONLY)->program.get()); | 63 EXPECT_TRUE(nullptr == flow.data(flow.ONLY)->program.get()); |
| 60 | 64 |
| 65 flow.DestroyDisassembler(flow.ONLY); |
| 66 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 67 EXPECT_TRUE(nullptr == flow.data(flow.ONLY)->disassembler.get()); |
| 68 |
| 61 flow.WriteSinkStreamSetFromEncodedProgram(flow.ONLY); | 69 flow.WriteSinkStreamSetFromEncodedProgram(flow.ONLY); |
| 62 EXPECT_EQ(courgette::C_OK, flow.status()); | 70 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 63 | 71 |
| 64 flow.DestroyEncodedProgram(flow.ONLY); | 72 flow.DestroyEncodedProgram(flow.ONLY); |
| 65 EXPECT_EQ(courgette::C_OK, flow.status()); | 73 EXPECT_EQ(courgette::C_OK, flow.status()); |
| 66 EXPECT_TRUE(nullptr == flow.data(flow.ONLY)->encoded.get()); | 74 EXPECT_TRUE(nullptr == flow.data(flow.ONLY)->encoded.get()); |
| 67 | 75 |
| 68 courgette::SinkStream sink; | 76 courgette::SinkStream sink; |
| 69 flow.WriteSinkStreamFromSinkStreamSet(flow.ONLY, &sink); | 77 flow.WriteSinkStreamFromSinkStreamSet(flow.ONLY, &sink); |
| 70 EXPECT_EQ(courgette::C_OK, flow.status()); | 78 EXPECT_EQ(courgette::C_OK, flow.status()); |
| (...skipping 129 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 200 } | 208 } |
| 201 | 209 |
| 202 TEST_F(DecodeFuzzTest, All) { | 210 TEST_F(DecodeFuzzTest, All) { |
| 203 FuzzExe("setup1.exe"); | 211 FuzzExe("setup1.exe"); |
| 204 FuzzExe("elf-32-1.exe"); | 212 FuzzExe("elf-32-1.exe"); |
| 205 } | 213 } |
| 206 | 214 |
| 207 int main(int argc, char** argv) { | 215 int main(int argc, char** argv) { |
| 208 return base::TestSuite(argc, argv).Run(); | 216 return base::TestSuite(argc, argv).Run(); |
| 209 } | 217 } |
| OLD | NEW |