net/websockets/websocket_handshake_handler_spdy_test.cc

   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