net/websockets/websocket_handshake_handler_spdy3_unittest.cc

   1 // Copyright (c) 2012 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 "googleurl/src/gurl.h"
  10 #include "net/http/http_util.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace net {
  14
  15 namespace {
  16
  17 TEST(WebSocketHandshakeHandlerSpdy3Test, RequestResponse) {
  18   WebSocketHandshakeRequestHandler request_handler;
  19
  20   static const char kHandshakeRequestMessage[] =
  21       "GET /demo HTTP/1.1\r\n"
  22       "Host: example.com\r\n"
  23       "Upgrade: websocket\r\n"
  24       "Connection: Upgrade\r\n"
  25       "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
  26       "Origin: http://example.com\r\n"
  27       "Sec-WebSocket-Protocol: sample\r\n"
  28       "Sec-WebSocket-Extensions: foo\r\n"
  29       "Sec-WebSocket-Version: 13\r\n"
  30       "X-Foo: foo\r\n"
  31       "\r\n";
  32
  33   EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
  34                                            strlen(kHandshakeRequestMessage)));
  35   EXPECT_EQ(13, request_handler.protocol_version());
  36
  37   GURL url("ws://example.com/demo");
  38   std::string challenge;
  39   SpdyHeaderBlock headers;
  40   ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
  41                                                     &headers,
  42                                                     &challenge,
  43                                                     3));
  44
  45   EXPECT_EQ(url.path(), headers[":path"]);
  46   EXPECT_TRUE(headers.find(":upgrade") == headers.end());
  47   EXPECT_TRUE(headers.find(":Upgrade") == headers.end());
  48   EXPECT_TRUE(headers.find(":connection") == headers.end());
  49   EXPECT_TRUE(headers.find(":Connection") == headers.end());
  50   EXPECT_TRUE(headers.find(":Sec-WebSocket-Key") == headers.end());
  51   EXPECT_TRUE(headers.find(":sec-websocket-key") == headers.end());
  52   EXPECT_TRUE(headers.find(":Sec-WebSocket-Version") == headers.end());
  53   EXPECT_TRUE(headers.find(":sec-webSocket-version") == headers.end());
  54   EXPECT_TRUE(headers.find(":x-foo") == headers.end());
  55   EXPECT_EQ("example.com", headers[":host"]);
  56   EXPECT_EQ("http://example.com", headers[":origin"]);
  57   EXPECT_EQ("sample", headers[":sec-websocket-protocol"]);
  58   EXPECT_EQ("foo", headers[":sec-websocket-extensions"]);
  59   EXPECT_EQ("ws", headers[":scheme"]);
  60   EXPECT_EQ("WebSocket/13", headers[":version"]);
  61   EXPECT_EQ("foo", headers["x-foo"]);
  62
  63   static const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
  64
  65   EXPECT_EQ(expected_challenge, challenge);
  66
  67   headers.clear();
  68
  69   headers[":status"] = "101 Switching Protocols";
  70   headers[":sec-websocket-protocol"] = "sample";
  71   headers[":sec-websocket-extensions"] = "foo";
  72   headers["x-bar"] = "bar";
  73
  74   WebSocketHandshakeResponseHandler response_handler;
  75   response_handler.set_protocol_version(13);
  76   EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
  77                                                         challenge,
  78                                                         3));
  79   EXPECT_TRUE(response_handler.HasResponse());
  80
  81   // Note that order of sec-websocket-* is sensitive with hash_map order.
  82   static const char kHandshakeResponseExpectedMessage[] =
  83       "HTTP/1.1 101 Switching Protocols\r\n"
  84       "Upgrade: websocket\r\n"
  85       "Connection: Upgrade\r\n"
  86       "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
  87       "sec-websocket-extensions: foo\r\n"
  88       "sec-websocket-protocol: sample\r\n"
  89       "x-bar: bar\r\n"
  90       "\r\n";
  91
  92   EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
  93 }
  94
  95 TEST(WebSocketHandshakeHandlerSpdy3Test, RequestResponseWithCookies) {
  96   WebSocketHandshakeRequestHandler request_handler;
  97
  98   // Note that websocket won't use multiple headers in request now.
  99   static const char kHandshakeRequestMessage[] =
 100       "GET /demo HTTP/1.1\r\n"
 101       "Host: example.com\r\n"
 102       "Upgrade: websocket\r\n"
 103       "Connection: Upgrade\r\n"
 104       "Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
 105       "Origin: http://example.com\r\n"
 106       "Sec-WebSocket-Protocol: sample\r\n"
 107       "Sec-WebSocket-Extensions: foo\r\n"
 108       "Sec-WebSocket-Version: 13\r\n"
 109       "Cookie: WK-websocket-test=1; WK-websocket-test-httponly=1\r\n"
 110       "\r\n";
 111
 112   EXPECT_TRUE(request_handler.ParseRequest(kHandshakeRequestMessage,
 113                                            strlen(kHandshakeRequestMessage)));
 114   EXPECT_EQ(13, request_handler.protocol_version());
 115
 116   GURL url("ws://example.com/demo");
 117   std::string challenge;
 118   SpdyHeaderBlock headers;
 119   ASSERT_TRUE(request_handler.GetRequestHeaderBlock(url,
 120                                                     &headers,
 121                                                     &challenge,
 122                                                     3));
 123
 124   EXPECT_EQ(url.path(), headers[":path"]);
 125   EXPECT_TRUE(headers.find(":upgrade") == headers.end());
 126   EXPECT_TRUE(headers.find(":Upgrade") == headers.end());
 127   EXPECT_TRUE(headers.find(":connection") == headers.end());
 128   EXPECT_TRUE(headers.find(":Connection") == headers.end());
 129   EXPECT_TRUE(headers.find(":Sec-WebSocket-Key") == headers.end());
 130   EXPECT_TRUE(headers.find(":sec-websocket-key") == headers.end());
 131   EXPECT_TRUE(headers.find(":Sec-WebSocket-Version") == headers.end());
 132   EXPECT_TRUE(headers.find(":sec-webSocket-version") == headers.end());
 133   EXPECT_TRUE(headers.find(":Cookie") == headers.end());
 134   EXPECT_TRUE(headers.find(":cookie") == headers.end());
 135   EXPECT_EQ("example.com", headers[":host"]);
 136   EXPECT_EQ("http://example.com", headers[":origin"]);
 137   EXPECT_EQ("sample", headers[":sec-websocket-protocol"]);
 138   EXPECT_EQ("foo", headers[":sec-websocket-extensions"]);
 139   EXPECT_EQ("ws", headers[":scheme"]);
 140   EXPECT_EQ("WebSocket/13", headers[":version"]);
 141   EXPECT_EQ("WK-websocket-test=1; WK-websocket-test-httponly=1",
 142             headers["cookie"]);
 143
 144   const char expected_challenge[] = "dGhlIHNhbXBsZSBub25jZQ==";
 145
 146   EXPECT_EQ(expected_challenge, challenge);
 147
 148   headers.clear();
 149
 150   headers[":status"] = "101 Switching Protocols";
 151   headers[":sec-websocket-protocol"] = "sample";
 152   headers[":sec-websocket-extensions"] = "foo";
 153   std::string cookie = "WK-websocket-test=1";
 154   cookie.append(1, '\0');
 155   cookie += "WK-websocket-test-httponly=1; HttpOnly";
 156   headers["set-cookie"] = cookie;
 157
 158
 159   WebSocketHandshakeResponseHandler response_handler;
 160   response_handler.set_protocol_version(13);
 161   EXPECT_TRUE(response_handler.ParseResponseHeaderBlock(headers,
 162                                                         challenge,
 163                                                         3));
 164   EXPECT_TRUE(response_handler.HasResponse());
 165
 166   // Note that order of sec-websocket-* is sensitive with hash_map order.
 167   static const char kHandshakeResponseExpectedMessage[] =
 168       "HTTP/1.1 101 Switching Protocols\r\n"
 169       "Upgrade: websocket\r\n"
 170       "Connection: Upgrade\r\n"
 171       "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
 172       "sec-websocket-extensions: foo\r\n"
 173       "sec-websocket-protocol: sample\r\n"
 174       "set-cookie: WK-websocket-test=1\r\n"
 175       "set-cookie: WK-websocket-test-httponly=1; HttpOnly\r\n"
 176       "\r\n";
 177
 178   EXPECT_EQ(kHandshakeResponseExpectedMessage, response_handler.GetResponse());
 179 }
 180
 181 }  // namespace
 182
 183 }  // namespace net