| 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 |