| 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 "net/spdy/fuzzing/hpack_fuzz_util.h" | 5 #include "net/spdy/fuzzing/hpack_fuzz_util.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 | 8 |
| 9 #include "base/base_paths.h" | 9 #include "base/base_paths.h" |
| 10 #include "base/files/file.h" | 10 #include "base/files/file.h" |
| 11 #include "base/files/file_util.h" | 11 #include "base/files/file_util.h" |
| 12 #include "base/path_service.h" | 12 #include "base/path_service.h" |
| 13 #include "net/spdy/spdy_test_utils.h" | 13 #include "net/spdy/spdy_test_utils.h" |
| 14 #include "testing/gmock/include/gmock/gmock.h" | 14 #include "testing/gmock/include/gmock/gmock.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 16 |
| 17 namespace net { | 17 namespace net { |
| 18 namespace test { | 18 namespace test { |
| 19 | 19 |
| 20 using std::map; | 20 using std::map; |
| 21 using std::string; | |
| 22 using test::a2b_hex; | 21 using test::a2b_hex; |
| 23 | 22 |
| 24 TEST(HpackFuzzUtilTest, GeneratorContextInitialization) { | 23 TEST(HpackFuzzUtilTest, GeneratorContextInitialization) { |
| 25 HpackFuzzUtil::GeneratorContext context; | 24 HpackFuzzUtil::GeneratorContext context; |
| 26 HpackFuzzUtil::InitializeGeneratorContext(&context); | 25 HpackFuzzUtil::InitializeGeneratorContext(&context); |
| 27 | 26 |
| 28 // Context was seeded with initial name & value fixtures. | 27 // Context was seeded with initial name & value fixtures. |
| 29 EXPECT_LT(0u, context.names.size()); | 28 EXPECT_LT(0u, context.names.size()); |
| 30 EXPECT_LT(0u, context.values.size()); | 29 EXPECT_LT(0u, context.values.size()); |
| 31 } | 30 } |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 78 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); | 77 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); |
| 79 EXPECT_EQ("e", block); | 78 EXPECT_EQ("e", block); |
| 80 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); | 79 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); |
| 81 EXPECT_EQ("", block); | 80 EXPECT_EQ("", block); |
| 82 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); | 81 EXPECT_TRUE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); |
| 83 EXPECT_EQ("fin", block); | 82 EXPECT_EQ("fin", block); |
| 84 EXPECT_FALSE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); | 83 EXPECT_FALSE(HpackFuzzUtil::NextHeaderBlock(&input, &block)); |
| 85 } | 84 } |
| 86 | 85 |
| 87 TEST(HpackFuzzUtilTest, SerializedHeaderBlockPrefixes) { | 86 TEST(HpackFuzzUtilTest, SerializedHeaderBlockPrefixes) { |
| 88 EXPECT_EQ(string("\x00\x00\x00\x00", 4), HpackFuzzUtil::HeaderBlockPrefix(0)); | 87 EXPECT_EQ(SpdyString("\x00\x00\x00\x00", 4), |
| 89 EXPECT_EQ(string("\x00\x00\x00\x05", 4), HpackFuzzUtil::HeaderBlockPrefix(5)); | 88 HpackFuzzUtil::HeaderBlockPrefix(0)); |
| 89 EXPECT_EQ(SpdyString("\x00\x00\x00\x05", 4), |
| 90 HpackFuzzUtil::HeaderBlockPrefix(5)); |
| 90 EXPECT_EQ("\x4f\xb3\x0a\x91", HpackFuzzUtil::HeaderBlockPrefix(1337133713)); | 91 EXPECT_EQ("\x4f\xb3\x0a\x91", HpackFuzzUtil::HeaderBlockPrefix(1337133713)); |
| 91 } | 92 } |
| 92 | 93 |
| 93 TEST(HpackFuzzUtilTest, PassValidInputThroughAllStages) { | 94 TEST(HpackFuzzUtilTest, PassValidInputThroughAllStages) { |
| 94 // Example lifted from HpackDecoderTest.SectionD4RequestHuffmanExamples. | 95 // Example lifted from HpackDecoderTest.SectionD4RequestHuffmanExamples. |
| 95 string input = a2b_hex("828684418cf1e3c2e5f23a6ba0ab90f4" | 96 SpdyString input = a2b_hex("828684418cf1e3c2e5f23a6ba0ab90f4ff"); |
| 96 "ff"); | |
| 97 | 97 |
| 98 HpackFuzzUtil::FuzzerContext context; | 98 HpackFuzzUtil::FuzzerContext context; |
| 99 HpackFuzzUtil::InitializeFuzzerContext(&context); | 99 HpackFuzzUtil::InitializeFuzzerContext(&context); |
| 100 | 100 |
| 101 EXPECT_TRUE( | 101 EXPECT_TRUE( |
| 102 HpackFuzzUtil::RunHeaderBlockThroughFuzzerStages(&context, input)); | 102 HpackFuzzUtil::RunHeaderBlockThroughFuzzerStages(&context, input)); |
| 103 | 103 |
| 104 SpdyHeaderBlock expect; | 104 SpdyHeaderBlock expect; |
| 105 expect[":method"] = "GET"; | 105 expect[":method"] = "GET"; |
| 106 expect[":scheme"] = "http"; | 106 expect[":scheme"] = "http"; |
| (...skipping 21 matching lines...) Expand all Loading... |
| 128 SpdyStringPiece block; | 128 SpdyStringPiece block; |
| 129 while (HpackFuzzUtil::NextHeaderBlock(&input, &block)) { | 129 while (HpackFuzzUtil::NextHeaderBlock(&input, &block)) { |
| 130 // As these are valid examples, all fuzz stages should succeed. | 130 // As these are valid examples, all fuzz stages should succeed. |
| 131 EXPECT_TRUE(HpackFuzzUtil::RunHeaderBlockThroughFuzzerStages( | 131 EXPECT_TRUE(HpackFuzzUtil::RunHeaderBlockThroughFuzzerStages( |
| 132 &context, block)); | 132 &context, block)); |
| 133 } | 133 } |
| 134 } | 134 } |
| 135 | 135 |
| 136 TEST(HpackFuzzUtilTest, FlipBitsMutatesBuffer) { | 136 TEST(HpackFuzzUtilTest, FlipBitsMutatesBuffer) { |
| 137 char buffer[] = "testbuffer1234567890"; | 137 char buffer[] = "testbuffer1234567890"; |
| 138 string unmodified(buffer, arraysize(buffer) - 1); | 138 SpdyString unmodified(buffer, arraysize(buffer) - 1); |
| 139 | 139 |
| 140 EXPECT_EQ(unmodified, buffer); | 140 EXPECT_EQ(unmodified, buffer); |
| 141 HpackFuzzUtil::FlipBits(reinterpret_cast<uint8_t*>(buffer), | 141 HpackFuzzUtil::FlipBits(reinterpret_cast<uint8_t*>(buffer), |
| 142 arraysize(buffer) - 1, 1); | 142 arraysize(buffer) - 1, 1); |
| 143 EXPECT_NE(unmodified, buffer); | 143 EXPECT_NE(unmodified, buffer); |
| 144 } | 144 } |
| 145 | 145 |
| 146 } // namespace test | 146 } // namespace test |
| 147 } // namespace net | 147 } // namespace net |
| OLD | NEW |