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 |