| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "base/command_line.h" | 5 #include "base/command_line.h" |
| 6 #include "base/files/file.h" | 6 #include "base/files/file.h" |
| 7 #include "base/files/file_util.h" | 7 #include "base/files/file_util.h" |
| 8 #include "base/strings/string_number_conversions.h" | 8 #include "base/strings/string_number_conversions.h" |
| 9 #include "net/spdy/fuzzing/hpack_fuzz_util.h" | 9 #include "net/spdy/fuzzing/hpack_fuzz_util.h" |
| 10 | 10 |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 50 &flip_ratio)); | 50 &flip_ratio)); |
| 51 | 51 |
| 52 DVLOG(1) << "Reading input from " << file_to_parse; | 52 DVLOG(1) << "Reading input from " << file_to_parse; |
| 53 HpackFuzzUtil::Input input; | 53 HpackFuzzUtil::Input input; |
| 54 CHECK(base::ReadFileToString(base::FilePath::FromUTF8Unsafe(file_to_parse), | 54 CHECK(base::ReadFileToString(base::FilePath::FromUTF8Unsafe(file_to_parse), |
| 55 &input.input)); | 55 &input.input)); |
| 56 | 56 |
| 57 DVLOG(1) << "Writing output to " << file_to_write; | 57 DVLOG(1) << "Writing output to " << file_to_write; |
| 58 base::File file_out(base::FilePath::FromUTF8Unsafe(file_to_write), | 58 base::File file_out(base::FilePath::FromUTF8Unsafe(file_to_write), |
| 59 base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE); | 59 base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE); |
| 60 CHECK(file_out.IsValid()) << file_out.error_details(); | 60 CHECK(file_out.IsValid()); |
| 61 | 61 |
| 62 DVLOG(1) << "Flipping " << flip_ratio << " bits per 1024 input bytes"; | 62 DVLOG(1) << "Flipping " << flip_ratio << " bits per 1024 input bytes"; |
| 63 | 63 |
| 64 size_t block_count = 0; | 64 size_t block_count = 0; |
| 65 StringPiece block; | 65 StringPiece block; |
| 66 while (HpackFuzzUtil::NextHeaderBlock(&input, &block)) { | 66 while (HpackFuzzUtil::NextHeaderBlock(&input, &block)) { |
| 67 HpackFuzzUtil::FlipBits( | 67 HpackFuzzUtil::FlipBits( |
| 68 reinterpret_cast<uint8_t*>(const_cast<char*>(block.data())), | 68 reinterpret_cast<uint8_t*>(const_cast<char*>(block.data())), |
| 69 block.size(), flip_ratio); | 69 block.size(), flip_ratio); |
| 70 | 70 |
| 71 string prefix = HpackFuzzUtil::HeaderBlockPrefix(block.size()); | 71 string prefix = HpackFuzzUtil::HeaderBlockPrefix(block.size()); |
| 72 | 72 |
| 73 CHECK_LT(0, file_out.WriteAtCurrentPos(prefix.data(), prefix.size())); | 73 CHECK_LT(0, file_out.WriteAtCurrentPos(prefix.data(), prefix.size())); |
| 74 CHECK_LT(0, file_out.WriteAtCurrentPos(block.data(), block.size())); | 74 CHECK_LT(0, file_out.WriteAtCurrentPos(block.data(), block.size())); |
| 75 ++block_count; | 75 ++block_count; |
| 76 } | 76 } |
| 77 CHECK(file_out.Flush()); | 77 CHECK(file_out.Flush()); |
| 78 DVLOG(1) << "Mutated " << block_count << " blocks."; | 78 DVLOG(1) << "Mutated " << block_count << " blocks."; |
| 79 return 0; | 79 return 0; |
| 80 } | 80 } |
| OLD | NEW |