Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(465)

Side by Side Diff: net/spdy/fuzzing/hpack_example_generator.cc

Issue 2832973003: Split net/spdy into core and chromium subdirectories. (Closed)
Patch Set: Fix some more build rules. Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/spdy/core/zero_copy_output_buffer.h ('k') | net/spdy/fuzzing/hpack_fuzz_util.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "base/at_exit.h"
6 #include "base/command_line.h"
7 #include "base/files/file.h"
8 #include "base/files/file_util.h"
9 #include "base/strings/string_number_conversions.h"
10 #include "net/spdy/fuzzing/hpack_fuzz_util.h"
11 #include "net/spdy/hpack/hpack_constants.h"
12 #include "net/spdy/hpack/hpack_encoder.h"
13 #include "net/spdy/platform/api/spdy_string.h"
14 #include "net/spdy/spdy_protocol.h"
15
16 namespace {
17
18 // Target file for generated HPACK header sets.
19 const char kFileToWrite[] = "file-to-write";
20
21 // Number of header sets to generate.
22 const char kExampleCount[] = "example-count";
23
24 } // namespace
25
26 using net::HpackFuzzUtil;
27 using net::SpdyString;
28 using std::map;
29
30 // Generates a configurable number of header sets (using HpackFuzzUtil), and
31 // sequentially encodes each header set with an HpackEncoder. Encoded header
32 // sets are written to the output file in length-prefixed blocks.
33 int main(int argc, char** argv) {
34 base::AtExitManager exit_manager;
35
36 base::CommandLine::Init(argc, argv);
37 const base::CommandLine& command_line =
38 *base::CommandLine::ForCurrentProcess();
39
40 if (!command_line.HasSwitch(kFileToWrite) ||
41 !command_line.HasSwitch(kExampleCount)) {
42 LOG(ERROR) << "Usage: " << argv[0]
43 << " --" << kFileToWrite << "=/path/to/file.out"
44 << " --" << kExampleCount << "=1000";
45 return -1;
46 }
47 SpdyString file_to_write = command_line.GetSwitchValueASCII(kFileToWrite);
48
49 int example_count = 0;
50 base::StringToInt(command_line.GetSwitchValueASCII(kExampleCount),
51 &example_count);
52
53 DVLOG(1) << "Writing output to " << file_to_write;
54 base::File file_out(base::FilePath::FromUTF8Unsafe(file_to_write),
55 base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE);
56 CHECK(file_out.IsValid()) << file_out.error_details();
57
58 HpackFuzzUtil::GeneratorContext context;
59 HpackFuzzUtil::InitializeGeneratorContext(&context);
60 net::HpackEncoder encoder(net::ObtainHpackHuffmanTable());
61
62 for (int i = 0; i != example_count; ++i) {
63 net::SpdyHeaderBlock headers =
64 HpackFuzzUtil::NextGeneratedHeaderSet(&context);
65
66 SpdyString buffer;
67 CHECK(encoder.EncodeHeaderSet(headers, &buffer));
68
69 SpdyString prefix = HpackFuzzUtil::HeaderBlockPrefix(buffer.size());
70
71 CHECK_LT(0, file_out.WriteAtCurrentPos(prefix.data(), prefix.size()));
72 CHECK_LT(0, file_out.WriteAtCurrentPos(buffer.data(), buffer.size()));
73 }
74 CHECK(file_out.Flush());
75 DVLOG(1) << "Generated " << example_count << " blocks.";
76 return 0;
77 }
OLDNEW
« no previous file with comments | « net/spdy/core/zero_copy_output_buffer.h ('k') | net/spdy/fuzzing/hpack_fuzz_util.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698