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_handshake_handler.h" | 5 #include "net/websockets/websocket_handshake_handler.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "net/socket/next_proto.h" | 9 #include "net/socket/next_proto.h" |
10 #include "net/spdy/spdy_header_block.h" | 10 #include "net/spdy/spdy_header_block.h" |
11 #include "net/spdy/spdy_websocket_test_util.h" | 11 #include "net/spdy/spdy_websocket_test_util.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 #include "url/gurl.h" | 13 #include "url/gurl.h" |
14 | 14 |
15 namespace net { | 15 namespace net { |
16 | 16 |
17 namespace { | 17 namespace { |
18 | 18 |
19 class WebSocketHandshakeHandlerSpdyTest | 19 class WebSocketHandshakeHandlerSpdyTest |
20 : public ::testing::Test, | 20 : public ::testing::Test, |
21 public ::testing::WithParamInterface<NextProto> { | 21 public ::testing::WithParamInterface<NextProto> { |
22 protected: | 22 protected: |
23 WebSocketHandshakeHandlerSpdyTest() : spdy_util_(GetParam()) {} | 23 WebSocketHandshakeHandlerSpdyTest() : spdy_util_(GetParam()) {} |
24 | 24 |
25 SpdyWebSocketTestUtil spdy_util_; | 25 SpdyWebSocketTestUtil spdy_util_; |
26 }; | 26 }; |
27 | 27 |
28 INSTANTIATE_TEST_CASE_P( | 28 INSTANTIATE_TEST_CASE_P(NextProto, |
29 NextProto, | 29 WebSocketHandshakeHandlerSpdyTest, |
30 WebSocketHandshakeHandlerSpdyTest, | 30 testing::Values(kProtoDeprecatedSPDY2, |
31 testing::Values(kProtoDeprecatedSPDY2, | 31 kProtoSPDY3, |
32 kProtoSPDY3, kProtoSPDY31, kProtoSPDY4)); | 32 kProtoSPDY31, |
| 33 kProtoSPDY4)); |
33 | 34 |
34 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponse) { | 35 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponse) { |
35 WebSocketHandshakeRequestHandler request_handler; | 36 WebSocketHandshakeRequestHandler request_handler; |
36 | 37 |
37 static const char kHandshakeRequestMessage[] = | 38 static const char kHandshakeRequestMessage[] = |
38 "GET /demo HTTP/1.1\r\n" | 39 "GET /demo HTTP/1.1\r\n" |
39 "Host: example.com\r\n" | 40 "Host: example.com\r\n" |
40 "Upgrade: websocket\r\n" | 41 "Upgrade: websocket\r\n" |
41 "Connection: Upgrade\r\n" | 42 "Connection: Upgrade\r\n" |
42 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" | 43 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" |
43 "Origin: http://example.com\r\n" | 44 "Origin: http://example.com\r\n" |
44 "Sec-WebSocket-Protocol: sample\r\n" | 45 "Sec-WebSocket-Protocol: sample\r\n" |
45 "Sec-WebSocket-Extensions: foo\r\n" | 46 "Sec-WebSocket-Extensions: foo\r\n" |
46 "Sec-WebSocket-Version: 13\r\n" | 47 "Sec-WebSocket-Version: 13\r\n" |
47 "\r\n"; | 48 "\r\n"; |
48 | 49 |
49 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | 50 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, |
50 strlen(kHandshakeRequestMessage))); | 51 strlen(kHandshakeRequestMessage))); |
51 | 52 |
52 GURL url("ws://example.com/demo"); | 53 GURL url("ws://example.com/demo"); |
53 std::string challenge; | 54 std::string challenge; |
54 SpdyHeaderBlock headers; | 55 SpdyHeaderBlock headers; |
55 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, | 56 ASSERT_TRUE(request_handler.GetRequestHeaderBlock( |
56 &headers, | 57 url, &headers, &challenge, spdy_util_.spdy_version())); |
57 &challenge, | |
58 spdy_util_.spdy_version())); | |
59 | 58 |
60 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); | 59 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); |
61 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); | 60 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); |
62 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); | 61 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); |
63 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); | 62 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); |
64 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); | 63 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); |
65 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); | 64 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); |
66 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); | 65 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); |
67 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); | 66 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); |
68 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); | 67 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 "Sec-WebSocket-Version: 13\r\n" | 116 "Sec-WebSocket-Version: 13\r\n" |
118 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n" | 117 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n" |
119 "\r\n"; | 118 "\r\n"; |
120 | 119 |
121 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | 120 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, |
122 strlen(kHandshakeRequestMessage))); | 121 strlen(kHandshakeRequestMessage))); |
123 | 122 |
124 GURL url("ws://example.com/demo"); | 123 GURL url("ws://example.com/demo"); |
125 std::string challenge; | 124 std::string challenge; |
126 SpdyHeaderBlock headers; | 125 SpdyHeaderBlock headers; |
127 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, | 126 ASSERT_TRUE(request_handler.GetRequestHeaderBlock( |
128 &headers, | 127 url, &headers, &challenge, spdy_util_.spdy_version())); |
129 &challenge, | |
130 spdy_util_.spdy_version())); | |
131 | 128 |
132 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); | 129 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); |
133 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); | 130 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); |
134 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); | 131 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); |
135 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); | 132 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); |
136 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); | 133 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); |
137 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); | 134 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); |
138 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); | 135 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); |
139 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); | 136 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); |
140 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); | 137 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); |
(...skipping 13 matching lines...) Expand all Loading... |
154 headers.clear(); | 151 headers.clear(); |
155 | 152 |
156 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers); | 153 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers); |
157 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers); | 154 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers); |
158 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers); | 155 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers); |
159 std::string cookie = "WK-websocket-test=1"; | 156 std::string cookie = "WK-websocket-test=1"; |
160 cookie.append(1, '\0'); | 157 cookie.append(1, '\0'); |
161 cookie += "WK-websocket-test-httponly=1; HttpOnly"; | 158 cookie += "WK-websocket-test-httponly=1; HttpOnly"; |
162 headers["set-cookie"] = cookie; | 159 headers["set-cookie"] = cookie; |
163 | 160 |
164 | |
165 WebSocketHandshakeResponseHandler response_handler; | 161 WebSocketHandshakeResponseHandler response_handler; |
166 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock( | 162 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock( |
167 headers, challenge, spdy_util_.spdy_version())); | 163 headers, challenge, spdy_util_.spdy_version())); |
168 EXPECT_TRUE(response_handler.HasResponse()); | 164 EXPECT_TRUE(response_handler.HasResponse()); |
169 | 165 |
170 // Note that order of sec-websocket-* is sensitive with hash_map order. | 166 // Note that order of sec-websocket-* is sensitive with hash_map order. |
171 static const char kHandshakeResponseExpectedMessage[] = | 167 static const char kHandshakeResponseExpectedMessage[] = |
172 "HTTP/1.1 101 Switching Protocols\r\n" | 168 "HTTP/1.1 101 Switching Protocols\r\n" |
173 "Upgrade: websocket\r\n" | 169 "Upgrade: websocket\r\n" |
174 "Connection: Upgrade\r\n" | 170 "Connection: Upgrade\r\n" |
175 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | 171 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" |
176 "sec-websocket-extensions: foo\r\n" | 172 "sec-websocket-extensions: foo\r\n" |
177 "sec-websocket-protocol: sample\r\n" | 173 "sec-websocket-protocol: sample\r\n" |
178 "set-cookie: WK-websocket-test=1\r\n" | 174 "set-cookie: WK-websocket-test=1\r\n" |
179 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n" | 175 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n" |
180 "\r\n"; | 176 "\r\n"; |
181 | 177 |
182 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | 178 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); |
183 } | 179 } |
184 | 180 |
185 } // namespace | 181 } // namespace |
186 | 182 |
187 } // namespace net | 183 } // namespace net |
OLD | NEW |