| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 63 | 63 |
| 64 // Loads an executable and does fuzz testing in the serialized format. | 64 // Loads an executable and does fuzz testing in the serialized format. |
| 65 void DecodeFuzzTest::FuzzExe(const char* file_name) const { | 65 void DecodeFuzzTest::FuzzExe(const char* file_name) const { |
| 66 std::string file1 = FileContents(file_name); | 66 std::string file1 = FileContents(file_name); |
| 67 | 67 |
| 68 const void* original_buffer = file1.c_str(); | 68 const void* original_buffer = file1.c_str(); |
| 69 size_t original_length = file1.size(); | 69 size_t original_length = file1.size(); |
| 70 | 70 |
| 71 courgette::AssemblyProgram* program = NULL; | 71 courgette::AssemblyProgram* program = NULL; |
| 72 const courgette::Status parse_status = | 72 const courgette::Status parse_status = |
| 73 courgette::ParseWin32X86PE(original_buffer, original_length, &program); | 73 courgette::ParseDetectedExecutable(original_buffer, original_length, |
| 74 &program); |
| 74 EXPECT_EQ(courgette::C_OK, parse_status); | 75 EXPECT_EQ(courgette::C_OK, parse_status); |
| 75 | 76 |
| 76 courgette::EncodedProgram* encoded = NULL; | 77 courgette::EncodedProgram* encoded = NULL; |
| 77 | 78 |
| 78 const courgette::Status encode_status = Encode(program, &encoded); | 79 const courgette::Status encode_status = Encode(program, &encoded); |
| 79 EXPECT_EQ(courgette::C_OK, encode_status); | 80 EXPECT_EQ(courgette::C_OK, encode_status); |
| 80 | 81 |
| 81 DeleteAssemblyProgram(program); | 82 DeleteAssemblyProgram(program); |
| 82 | 83 |
| 83 courgette::SinkStreamSet sinks; | 84 courgette::SinkStreamSet sinks; |
| (...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 226 return result; | 227 return result; |
| 227 } | 228 } |
| 228 | 229 |
| 229 TEST_F(DecodeFuzzTest, All) { | 230 TEST_F(DecodeFuzzTest, All) { |
| 230 FuzzExe("setup1.exe"); | 231 FuzzExe("setup1.exe"); |
| 231 } | 232 } |
| 232 | 233 |
| 233 int main(int argc, char** argv) { | 234 int main(int argc, char** argv) { |
| 234 return base::TestSuite(argc, argv).Run(); | 235 return base::TestSuite(argc, argv).Run(); |
| 235 } | 236 } |
| OLD | NEW |