| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/websockets/websocket_extension_parser.h" | 5 #include "net/websockets/websocket_extension_parser.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "net/websockets/websocket_extension.h" | 9 #include "net/websockets/websocket_extension.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 namespace net { | 12 namespace net { |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 TEST(WebSocketExtensionParserTest, ParseEmpty) { | 16 TEST(WebSocketExtensionParserTest, ParseEmpty) { |
| 17 WebSocketExtensionParser parser; | 17 WebSocketExtensionParser parser; |
| 18 parser.Parse("", 0); | 18 EXPECT_FALSE(parser.Parse("", 0)); |
| 19 | 19 |
| 20 EXPECT_TRUE(parser.has_error()); | |
| 21 EXPECT_EQ(0U, parser.extensions().size()); | 20 EXPECT_EQ(0U, parser.extensions().size()); |
| 22 } | 21 } |
| 23 | 22 |
| 24 TEST(WebSocketExtensionParserTest, ParseSimple) { | 23 TEST(WebSocketExtensionParserTest, ParseSimple) { |
| 25 WebSocketExtensionParser parser; | 24 WebSocketExtensionParser parser; |
| 26 WebSocketExtension expected("foo"); | 25 WebSocketExtension expected("foo"); |
| 27 | 26 |
| 28 parser.Parse("foo"); | 27 EXPECT_TRUE(parser.Parse("foo")); |
| 29 | 28 |
| 30 ASSERT_FALSE(parser.has_error()); | |
| 31 ASSERT_EQ(1U, parser.extensions().size()); | 29 ASSERT_EQ(1U, parser.extensions().size()); |
| 32 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 30 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 33 } | 31 } |
| 34 | 32 |
| 35 TEST(WebSocketExtensionParserTest, ParseMoreThanOnce) { | 33 TEST(WebSocketExtensionParserTest, ParseMoreThanOnce) { |
| 36 WebSocketExtensionParser parser; | 34 WebSocketExtensionParser parser; |
| 37 WebSocketExtension expected("foo"); | 35 WebSocketExtension expected("foo"); |
| 38 | 36 |
| 39 parser.Parse("foo"); | 37 EXPECT_TRUE(parser.Parse("foo")); |
| 40 ASSERT_FALSE(parser.has_error()); | |
| 41 ASSERT_EQ(1U, parser.extensions().size()); | 38 ASSERT_EQ(1U, parser.extensions().size()); |
| 42 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 39 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 43 | 40 |
| 44 parser.Parse(""); | 41 EXPECT_FALSE(parser.Parse("")); |
| 45 EXPECT_TRUE(parser.has_error()); | |
| 46 EXPECT_EQ(0U, parser.extensions().size()); | 42 EXPECT_EQ(0U, parser.extensions().size()); |
| 47 | 43 |
| 48 parser.Parse("foo"); | 44 EXPECT_TRUE(parser.Parse("foo")); |
| 49 ASSERT_FALSE(parser.has_error()); | |
| 50 ASSERT_EQ(1U, parser.extensions().size()); | 45 ASSERT_EQ(1U, parser.extensions().size()); |
| 51 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 46 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 52 } | 47 } |
| 53 | 48 |
| 54 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) { | 49 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) { |
| 55 WebSocketExtensionParser parser; | 50 WebSocketExtensionParser parser; |
| 56 WebSocketExtension expected("foo"); | 51 WebSocketExtension expected("foo"); |
| 57 expected.Add(WebSocketExtension::Parameter("bar")); | 52 expected.Add(WebSocketExtension::Parameter("bar")); |
| 58 | 53 |
| 59 parser.Parse("\tfoo ; bar"); | 54 EXPECT_TRUE(parser.Parse("\tfoo ; bar")); |
| 60 | 55 |
| 61 ASSERT_FALSE(parser.has_error()); | |
| 62 ASSERT_EQ(1U, parser.extensions().size()); | 56 ASSERT_EQ(1U, parser.extensions().size()); |
| 63 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 57 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 64 } | 58 } |
| 65 | 59 |
| 66 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) { | 60 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) { |
| 67 WebSocketExtensionParser parser; | 61 WebSocketExtensionParser parser; |
| 68 WebSocketExtension expected("foo"); | 62 WebSocketExtension expected("foo"); |
| 69 expected.Add(WebSocketExtension::Parameter("bar", "baz")); | 63 expected.Add(WebSocketExtension::Parameter("bar", "baz")); |
| 70 | 64 |
| 71 parser.Parse("foo ; bar= baz\t"); | 65 EXPECT_TRUE(parser.Parse("foo ; bar= baz\t")); |
| 72 | 66 |
| 73 ASSERT_FALSE(parser.has_error()); | |
| 74 ASSERT_EQ(1U, parser.extensions().size()); | 67 ASSERT_EQ(1U, parser.extensions().size()); |
| 75 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 68 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 76 } | 69 } |
| 77 | 70 |
| 78 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) { | 71 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) { |
| 79 WebSocketExtensionParser parser; | 72 WebSocketExtensionParser parser; |
| 80 WebSocketExtension expected("foo"); | 73 WebSocketExtension expected("foo"); |
| 81 expected.Add(WebSocketExtension::Parameter("bar", "baz")); | 74 expected.Add(WebSocketExtension::Parameter("bar", "baz")); |
| 82 expected.Add(WebSocketExtension::Parameter("hoge", "fuga")); | 75 expected.Add(WebSocketExtension::Parameter("hoge", "fuga")); |
| 83 | 76 |
| 84 parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga"); | 77 EXPECT_TRUE(parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga")); |
| 85 | 78 |
| 86 ASSERT_FALSE(parser.has_error()); | |
| 87 ASSERT_EQ(1U, parser.extensions().size()); | 79 ASSERT_EQ(1U, parser.extensions().size()); |
| 88 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 80 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 89 } | 81 } |
| 90 | 82 |
| 91 TEST(WebSocketExtensionParserTest, ParseTwoExtensions) { | 83 TEST(WebSocketExtensionParserTest, ParseTwoExtensions) { |
| 92 WebSocketExtensionParser parser; | 84 WebSocketExtensionParser parser; |
| 93 | 85 |
| 94 WebSocketExtension expected0("foo"); | 86 WebSocketExtension expected0("foo"); |
| 95 expected0.Add(WebSocketExtension::Parameter("alpha", "x")); | 87 expected0.Add(WebSocketExtension::Parameter("alpha", "x")); |
| 96 | 88 |
| 97 WebSocketExtension expected1("bar"); | 89 WebSocketExtension expected1("bar"); |
| 98 expected1.Add(WebSocketExtension::Parameter("beta", "y")); | 90 expected1.Add(WebSocketExtension::Parameter("beta", "y")); |
| 99 | 91 |
| 100 parser.Parse(" foo ; alpha = x , bar ; beta = y "); | 92 EXPECT_TRUE(parser.Parse(" foo ; alpha = x , bar ; beta = y ")); |
| 101 | 93 |
| 102 ASSERT_FALSE(parser.has_error()); | |
| 103 ASSERT_EQ(2U, parser.extensions().size()); | 94 ASSERT_EQ(2U, parser.extensions().size()); |
| 104 | 95 |
| 105 EXPECT_TRUE(expected0.Equals(parser.extensions()[0])); | 96 EXPECT_TRUE(expected0.Equals(parser.extensions()[0])); |
| 106 EXPECT_TRUE(expected1.Equals(parser.extensions()[1])); | 97 EXPECT_TRUE(expected1.Equals(parser.extensions()[1])); |
| 107 } | 98 } |
| 108 | 99 |
| 109 TEST(WebSocketExtensionParserTest, InvalidPatterns) { | 100 TEST(WebSocketExtensionParserTest, InvalidPatterns) { |
| 110 const char* const patterns[] = { | 101 const char* const patterns[] = { |
| 111 ",", // just a comma | 102 ",", // just a comma |
| 112 " , ", // just a comma with surrounding spaces | 103 " , ", // just a comma with surrounding spaces |
| 113 "foo,", // second extension is incomplete (empty) | 104 "foo,", // second extension is incomplete (empty) |
| 114 "foo , ", // second extension is incomplete (space) | 105 "foo , ", // second extension is incomplete (space) |
| 115 "foo,;", // second extension is incomplete (semicolon) | 106 "foo,;", // second extension is incomplete (semicolon) |
| 107 "foo;, bar", // first extension is incomplete |
| 116 "fo\ao", // control in extension name | 108 "fo\ao", // control in extension name |
| 117 "fo\x01o", // control in extension name | 109 "fo\x01o", // control in extension name |
| 118 "fo<o", // separator in extension name | 110 "fo<o", // separator in extension name |
| 119 "foo/", // separator in extension name | 111 "foo/", // separator in extension name |
| 120 ";bar", // empty extension name | 112 ";bar", // empty extension name |
| 121 "foo bar", // missing ';' | 113 "foo bar", // missing ';' |
| 122 "foo;", // extension parameter without name and value | 114 "foo;", // extension parameter without name and value |
| 123 "foo; b\ar", // control in parameter name | 115 "foo; b\ar", // control in parameter name |
| 124 "foo; b\x7fr", // control in parameter name | 116 "foo; b\x7fr", // control in parameter name |
| 125 "foo; b[r", // separator in parameter name | 117 "foo; b[r", // separator in parameter name |
| (...skipping 14 matching lines...) Expand all Loading... |
| 140 "foo; bar=\"baz", // unterminated quoted string | 132 "foo; bar=\"baz", // unterminated quoted string |
| 141 "foo; bar=\"baz \"", // space in quoted string | 133 "foo; bar=\"baz \"", // space in quoted string |
| 142 "foo; bar baz", // mising '=' | 134 "foo; bar baz", // mising '=' |
| 143 "foo; bar - baz", // '-' instead of '=' (note: "foo; bar-baz" is valid). | 135 "foo; bar - baz", // '-' instead of '=' (note: "foo; bar-baz" is valid). |
| 144 "foo; bar=\r\nbaz", // CRNL not followed by a space | 136 "foo; bar=\r\nbaz", // CRNL not followed by a space |
| 145 "foo; bar=\r\n baz", // CRNL followed by a space | 137 "foo; bar=\r\n baz", // CRNL followed by a space |
| 146 }; | 138 }; |
| 147 | 139 |
| 148 for (size_t i = 0; i < arraysize(patterns); ++i) { | 140 for (size_t i = 0; i < arraysize(patterns); ++i) { |
| 149 WebSocketExtensionParser parser; | 141 WebSocketExtensionParser parser; |
| 150 parser.Parse(patterns[i]); | 142 EXPECT_FALSE(parser.Parse(patterns[i])); |
| 151 EXPECT_TRUE(parser.has_error()); | |
| 152 EXPECT_EQ(0U, parser.extensions().size()); | 143 EXPECT_EQ(0U, parser.extensions().size()); |
| 153 } | 144 } |
| 154 } | 145 } |
| 155 | 146 |
| 156 TEST(WebSocketExtensionParserTest, QuotedParameterValue) { | 147 TEST(WebSocketExtensionParserTest, QuotedParameterValue) { |
| 157 WebSocketExtensionParser parser; | 148 WebSocketExtensionParser parser; |
| 158 WebSocketExtension expected("foo"); | 149 WebSocketExtension expected("foo"); |
| 159 expected.Add(WebSocketExtension::Parameter("bar", "baz")); | 150 expected.Add(WebSocketExtension::Parameter("bar", "baz")); |
| 160 | 151 |
| 161 parser.Parse("foo; bar = \"ba\\z\" "); | 152 EXPECT_TRUE(parser.Parse("foo; bar = \"ba\\z\" ")); |
| 162 | 153 |
| 163 ASSERT_FALSE(parser.has_error()); | |
| 164 ASSERT_EQ(1U, parser.extensions().size()); | 154 ASSERT_EQ(1U, parser.extensions().size()); |
| 165 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); | 155 EXPECT_TRUE(expected.Equals(parser.extensions()[0])); |
| 166 } | 156 } |
| 167 | 157 |
| 168 } // namespace | 158 } // namespace |
| 169 | 159 |
| 170 } // namespace net | 160 } // namespace net |
| OLD | NEW |