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 |