| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/websockets/websocket_handshake_handler.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "net/socket/next_proto.h" | |
| 10 #include "net/spdy/spdy_header_block.h" | |
| 11 #include "net/spdy/spdy_websocket_test_util.h" | |
| 12 #include "testing/gtest/include/gtest/gtest.h" | |
| 13 #include "url/gurl.h" | |
| 14 | |
| 15 namespace net { | |
| 16 | |
| 17 namespace { | |
| 18 | |
| 19 class WebSocketHandshakeHandlerSpdyTest | |
| 20 : public ::testing::Test, | |
| 21 public ::testing::WithParamInterface<NextProto> { | |
| 22 protected: | |
| 23 WebSocketHandshakeHandlerSpdyTest() : spdy_util_(GetParam()) {} | |
| 24 | |
| 25 SpdyWebSocketTestUtil spdy_util_; | |
| 26 }; | |
| 27 | |
| 28 INSTANTIATE_TEST_CASE_P( | |
| 29 NextProto, | |
| 30 WebSocketHandshakeHandlerSpdyTest, | |
| 31 testing::Values(kProtoDeprecatedSPDY2, | |
| 32 kProtoSPDY3, kProtoSPDY31, kProtoSPDY4)); | |
| 33 | |
| 34 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponse) { | |
| 35 WebSocketHandshakeRequestHandler request_handler; | |
| 36 | |
| 37 static const char kHandshakeRequestMessage[] = | |
| 38 "GET /demo HTTP/1.1\r\n" | |
| 39 "Host: example.com\r\n" | |
| 40 "Upgrade: websocket\r\n" | |
| 41 "Connection: Upgrade\r\n" | |
| 42 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" | |
| 43 "Origin: http://example.com\r\n" | |
| 44 "Sec-WebSocket-Protocol: sample\r\n" | |
| 45 "Sec-WebSocket-Extensions: foo\r\n" | |
| 46 "Sec-WebSocket-Version: 13\r\n" | |
| 47 "\r\n"; | |
| 48 | |
| 49 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | |
| 50 strlen(kHandshakeRequestMessage))); | |
| 51 | |
| 52 GURL url("ws://example.com/demo"); | |
| 53 std::string challenge; | |
| 54 SpdyHeaderBlock headers; | |
| 55 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, | |
| 56 &headers, | |
| 57 &challenge, | |
| 58 spdy_util_.spdy_version())); | |
| 59 | |
| 60 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); | |
| 61 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); | |
| 62 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); | |
| 63 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); | |
| 64 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); | |
| 65 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); | |
| 66 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); | |
| 67 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); | |
| 68 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); | |
| 69 EXPECT_EQ("example.com", spdy_util_.GetHeader(headers, "host")); | |
| 70 EXPECT_EQ("http://example.com", spdy_util_.GetHeader(headers, "origin")); | |
| 71 EXPECT_EQ("sample", spdy_util_.GetHeader(headers, "sec-websocket-protocol")); | |
| 72 EXPECT_EQ("foo", spdy_util_.GetHeader(headers, "sec-websocket-extensions")); | |
| 73 EXPECT_EQ("ws", spdy_util_.GetHeader(headers, "scheme")); | |
| 74 EXPECT_EQ("WebSocket/13", spdy_util_.GetHeader(headers, "version")); | |
| 75 | |
| 76 static const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ=="; | |
| 77 | |
| 78 EXPECT_EQ(expected_challenge, challenge); | |
| 79 | |
| 80 headers.clear(); | |
| 81 | |
| 82 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers); | |
| 83 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers); | |
| 84 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers); | |
| 85 | |
| 86 WebSocketHandshakeResponseHandler response_handler; | |
| 87 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock( | |
| 88 headers, challenge, spdy_util_.spdy_version())); | |
| 89 EXPECT_TRUE(response_handler.HasResponse()); | |
| 90 | |
| 91 // Note that order of sec-websocket-* is sensitive with hash_map order. | |
| 92 static const char kHandshakeResponseExpectedMessage[] = | |
| 93 "HTTP/1.1 101 Switching Protocols\r\n" | |
| 94 "Upgrade: websocket\r\n" | |
| 95 "Connection: Upgrade\r\n" | |
| 96 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | |
| 97 "sec-websocket-extensions: foo\r\n" | |
| 98 "sec-websocket-protocol: sample\r\n" | |
| 99 "\r\n"; | |
| 100 | |
| 101 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | |
| 102 } | |
| 103 | |
| 104 TEST_P(WebSocketHandshakeHandlerSpdyTest, RequestResponseWithCookies) { | |
| 105 WebSocketHandshakeRequestHandler request_handler; | |
| 106 | |
| 107 // Note that websocket won't use multiple headers in request now. | |
| 108 static const char kHandshakeRequestMessage[] = | |
| 109 "GET /demo HTTP/1.1\r\n" | |
| 110 "Host: example.com\r\n" | |
| 111 "Upgrade: websocket\r\n" | |
| 112 "Connection: Upgrade\r\n" | |
| 113 "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n" | |
| 114 "Origin: http://example.com\r\n" | |
| 115 "Sec-WebSocket-Protocol: sample\r\n" | |
| 116 "Sec-WebSocket-Extensions: foo\r\n" | |
| 117 "Sec-WebSocket-Version: 13\r\n" | |
| 118 "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n" | |
| 119 "\r\n"; | |
| 120 | |
| 121 EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage, | |
| 122 strlen(kHandshakeRequestMessage))); | |
| 123 | |
| 124 GURL url("ws://example.com/demo"); | |
| 125 std::string challenge; | |
| 126 SpdyHeaderBlock headers; | |
| 127 ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url, | |
| 128 &headers, | |
| 129 &challenge, | |
| 130 spdy_util_.spdy_version())); | |
| 131 | |
| 132 EXPECT_EQ(url.path(), spdy_util_.GetHeader(headers, "path")); | |
| 133 EXPECT_TRUE(spdy_util_.GetHeader(headers, "upgrade").empty()); | |
| 134 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Upgrade").empty()); | |
| 135 EXPECT_TRUE(spdy_util_.GetHeader(headers, "connection").empty()); | |
| 136 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Connection").empty()); | |
| 137 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Key").empty()); | |
| 138 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-websocket-key").empty()); | |
| 139 EXPECT_TRUE(spdy_util_.GetHeader(headers, "Sec-WebSocket-Version").empty()); | |
| 140 EXPECT_TRUE(spdy_util_.GetHeader(headers, "sec-webSocket-version").empty()); | |
| 141 EXPECT_EQ("example.com", spdy_util_.GetHeader(headers, "host")); | |
| 142 EXPECT_EQ("http://example.com", spdy_util_.GetHeader(headers, "origin")); | |
| 143 EXPECT_EQ("sample", spdy_util_.GetHeader(headers, "sec-websocket-protocol")); | |
| 144 EXPECT_EQ("foo", spdy_util_.GetHeader(headers, "sec-websocket-extensions")); | |
| 145 EXPECT_EQ("ws", spdy_util_.GetHeader(headers, "scheme")); | |
| 146 EXPECT_EQ("WebSocket/13", spdy_util_.GetHeader(headers, "version")); | |
| 147 EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1", | |
| 148 headers["cookie"]); | |
| 149 | |
| 150 const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ=="; | |
| 151 | |
| 152 EXPECT_EQ(expected_challenge, challenge); | |
| 153 | |
| 154 headers.clear(); | |
| 155 | |
| 156 spdy_util_.SetHeader("status", "101 Switching Protocols", &headers); | |
| 157 spdy_util_.SetHeader("sec-websocket-protocol", "sample", &headers); | |
| 158 spdy_util_.SetHeader("sec-websocket-extensions", "foo", &headers); | |
| 159 std::string cookie = "WK-websocket-test=1"; | |
| 160 cookie.append(1, '\0'); | |
| 161 cookie += "WK-websocket-test-httponly=1; HttpOnly"; | |
| 162 headers["set-cookie"] = cookie; | |
| 163 | |
| 164 | |
| 165 WebSocketHandshakeResponseHandler response_handler; | |
| 166 EXPECT_TRUE(response_handler.ParseResponseHeaderBlock( | |
| 167 headers, challenge, spdy_util_.spdy_version())); | |
| 168 EXPECT_TRUE(response_handler.HasResponse()); | |
| 169 | |
| 170 // Note that order of sec-websocket-* is sensitive with hash_map order. | |
| 171 static const char kHandshakeResponseExpectedMessage[] = | |
| 172 "HTTP/1.1 101 Switching Protocols\r\n" | |
| 173 "Upgrade: websocket\r\n" | |
| 174 "Connection: Upgrade\r\n" | |
| 175 "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n" | |
| 176 "sec-websocket-extensions: foo\r\n" | |
| 177 "sec-websocket-protocol: sample\r\n" | |
| 178 "set-cookie: WK-websocket-test=1\r\n" | |
| 179 "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n" | |
| 180 "\r\n"; | |
| 181 | |
| 182 EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse()); | |
| 183 } | |
| 184 | |
| 185 } // namespace | |
| 186 | |
| 187 } // namespace net | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 723343002:
Update from https://crrev.com/304121 (Closed)
Base URL: git@github.com:domokit/mojo.git@master