| 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/file_util.h" | 6 #include "base/file_util.h" |
| 7 #include "base/files/file.h" | 7 #include "base/files/file.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 11 matching lines...) Expand all Loading... |
| 22 } // namespace | 22 } // namespace |
| 23 | 23 |
| 24 using base::StringPiece; | 24 using base::StringPiece; |
| 25 using net::HpackFuzzUtil; | 25 using net::HpackFuzzUtil; |
| 26 using std::string; | 26 using std::string; |
| 27 | 27 |
| 28 // Reads length-prefixed input blocks, applies a bit-flipping mutation to each | 28 // Reads length-prefixed input blocks, applies a bit-flipping mutation to each |
| 29 // block, and writes length-prefixed blocks to the output file. While blocks | 29 // block, and writes length-prefixed blocks to the output file. While blocks |
| 30 // themselves are mutated, the length-prefixes of written blocks are not. | 30 // themselves are mutated, the length-prefixes of written blocks are not. |
| 31 int main(int argc, char** argv) { | 31 int main(int argc, char** argv) { |
| 32 CommandLine::Init(argc, argv); | 32 base::CommandLine::Init(argc, argv); |
| 33 const CommandLine& command_line = *CommandLine::ForCurrentProcess(); | 33 const base::CommandLine& command_line = |
| 34 *base::CommandLine::ForCurrentProcess(); |
| 34 | 35 |
| 35 if (!command_line.HasSwitch(kFileToParse) || | 36 if (!command_line.HasSwitch(kFileToParse) || |
| 36 !command_line.HasSwitch(kFileToWrite) || | 37 !command_line.HasSwitch(kFileToWrite) || |
| 37 !command_line.HasSwitch(kFlipsPerThousand)) { | 38 !command_line.HasSwitch(kFlipsPerThousand)) { |
| 38 LOG(ERROR) << "Usage: " << argv[0] | 39 LOG(ERROR) << "Usage: " << argv[0] |
| 39 << " --" << kFileToParse << "=/path/to/file.in" | 40 << " --" << kFileToParse << "=/path/to/file.in" |
| 40 << " --" << kFileToWrite << "=/path/to/file.out" | 41 << " --" << kFileToWrite << "=/path/to/file.out" |
| 41 << " --" << kFlipsPerThousand << "=10"; | 42 << " --" << kFlipsPerThousand << "=10"; |
| 42 return -1; | 43 return -1; |
| 43 } | 44 } |
| (...skipping 27 matching lines...) Expand all Loading... |
| 71 string prefix = HpackFuzzUtil::HeaderBlockPrefix(block.size()); | 72 string prefix = HpackFuzzUtil::HeaderBlockPrefix(block.size()); |
| 72 | 73 |
| 73 CHECK_LT(0, file_out.WriteAtCurrentPos(prefix.data(), prefix.size())); | 74 CHECK_LT(0, file_out.WriteAtCurrentPos(prefix.data(), prefix.size())); |
| 74 CHECK_LT(0, file_out.WriteAtCurrentPos(block.data(), block.size())); | 75 CHECK_LT(0, file_out.WriteAtCurrentPos(block.data(), block.size())); |
| 75 ++block_count; | 76 ++block_count; |
| 76 } | 77 } |
| 77 CHECK(file_out.Flush()); | 78 CHECK(file_out.Flush()); |
| 78 DVLOG(1) << "Mutated " << block_count << " blocks."; | 79 DVLOG(1) << "Mutated " << block_count << " blocks."; |
| 79 return 0; | 80 return 0; |
| 80 } | 81 } |
| OLD | NEW |