| Index: net/server/web_socket.cc
|
| diff --git a/net/server/web_socket.cc b/net/server/web_socket.cc
|
| deleted file mode 100644
|
| index e5d1cb7b88ee5316d1bcf7f142acee839a9e8352..0000000000000000000000000000000000000000
|
| --- a/net/server/web_socket.cc
|
| +++ /dev/null
|
| @@ -1,259 +0,0 @@
|
| -// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include "net/server/web_socket.h"
|
| -
|
| -#include "base/base64.h"
|
| -#include "base/logging.h"
|
| -#include "base/md5.h"
|
| -#include "base/sha1.h"
|
| -#include "base/strings/string_number_conversions.h"
|
| -#include "base/strings/stringprintf.h"
|
| -#include "base/sys_byteorder.h"
|
| -#include "net/server/http_connection.h"
|
| -#include "net/server/http_server.h"
|
| -#include "net/server/http_server_request_info.h"
|
| -#include "net/server/http_server_response_info.h"
|
| -#include "net/server/web_socket_encoder.h"
|
| -
|
| -namespace net {
|
| -
|
| -namespace {
|
| -
|
| -static uint32 WebSocketKeyFingerprint(const std::string& str) {
|
| - std::string result;
|
| - const char* p_char = str.c_str();
|
| - int length = str.length();
|
| - int spaces = 0;
|
| - for (int i = 0; i < length; ++i) {
|
| - if (p_char[i] >= '0' && p_char[i] <= '9')
|
| - result.append(&p_char[i], 1);
|
| - else if (p_char[i] == ' ')
|
| - spaces++;
|
| - }
|
| - if (spaces == 0)
|
| - return 0;
|
| - int64 number = 0;
|
| - if (!base::StringToInt64(result, &number))
|
| - return 0;
|
| - return base::HostToNet32(static_cast<uint32>(number / spaces));
|
| -}
|
| -
|
| -class WebSocketHixie76 : public net::WebSocket {
|
| - public:
|
| - static net::WebSocket* Create(HttpServer* server,
|
| - HttpConnection* connection,
|
| - const HttpServerRequestInfo& request,
|
| - size_t* pos) {
|
| - if (connection->read_buf()->GetSize() <
|
| - static_cast<int>(*pos + kWebSocketHandshakeBodyLen))
|
| - return NULL;
|
| - return new WebSocketHixie76(server, connection, request, pos);
|
| - }
|
| -
|
| - void Accept(const HttpServerRequestInfo& request) override {
|
| - std::string key1 = request.GetHeaderValue("sec-websocket-key1");
|
| - std::string key2 = request.GetHeaderValue("sec-websocket-key2");
|
| -
|
| - uint32 fp1 = WebSocketKeyFingerprint(key1);
|
| - uint32 fp2 = WebSocketKeyFingerprint(key2);
|
| -
|
| - char data[16];
|
| - memcpy(data, &fp1, 4);
|
| - memcpy(data + 4, &fp2, 4);
|
| - memcpy(data + 8, &key3_[0], 8);
|
| -
|
| - base::MD5Digest digest;
|
| - base::MD5Sum(data, 16, &digest);
|
| -
|
| - std::string origin = request.GetHeaderValue("origin");
|
| - std::string host = request.GetHeaderValue("host");
|
| - std::string location = "ws://" + host + request.path;
|
| - server_->SendRaw(
|
| - connection_->id(),
|
| - base::StringPrintf("HTTP/1.1 101 WebSocket Protocol Handshake\r\n"
|
| - "Upgrade: WebSocket\r\n"
|
| - "Connection: Upgrade\r\n"
|
| - "Sec-WebSocket-Origin: %s\r\n"
|
| - "Sec-WebSocket-Location: %s\r\n"
|
| - "\r\n",
|
| - origin.c_str(),
|
| - location.c_str()));
|
| - server_->SendRaw(connection_->id(),
|
| - std::string(reinterpret_cast<char*>(digest.a), 16));
|
| - }
|
| -
|
| - ParseResult Read(std::string* message) override {
|
| - DCHECK(message);
|
| - HttpConnection::ReadIOBuffer* read_buf = connection_->read_buf();
|
| - if (read_buf->StartOfBuffer()[0])
|
| - return FRAME_ERROR;
|
| -
|
| - base::StringPiece data(read_buf->StartOfBuffer(), read_buf->GetSize());
|
| - size_t pos = data.find('\377', 1);
|
| - if (pos == base::StringPiece::npos)
|
| - return FRAME_INCOMPLETE;
|
| -
|
| - message->assign(data.data() + 1, pos - 1);
|
| - read_buf->DidConsume(pos + 1);
|
| -
|
| - return FRAME_OK;
|
| - }
|
| -
|
| - void Send(const std::string& message) override {
|
| - char message_start = 0;
|
| - char message_end = -1;
|
| - server_->SendRaw(connection_->id(), std::string(1, message_start));
|
| - server_->SendRaw(connection_->id(), message);
|
| - server_->SendRaw(connection_->id(), std::string(1, message_end));
|
| - }
|
| -
|
| - private:
|
| - static const int kWebSocketHandshakeBodyLen;
|
| -
|
| - WebSocketHixie76(HttpServer* server,
|
| - HttpConnection* connection,
|
| - const HttpServerRequestInfo& request,
|
| - size_t* pos)
|
| - : WebSocket(server, connection) {
|
| - std::string key1 = request.GetHeaderValue("sec-websocket-key1");
|
| - std::string key2 = request.GetHeaderValue("sec-websocket-key2");
|
| -
|
| - if (key1.empty()) {
|
| - server->SendResponse(
|
| - connection->id(),
|
| - HttpServerResponseInfo::CreateFor500(
|
| - "Invalid request format. Sec-WebSocket-Key1 is empty or isn't "
|
| - "specified."));
|
| - return;
|
| - }
|
| -
|
| - if (key2.empty()) {
|
| - server->SendResponse(
|
| - connection->id(),
|
| - HttpServerResponseInfo::CreateFor500(
|
| - "Invalid request format. Sec-WebSocket-Key2 is empty or isn't "
|
| - "specified."));
|
| - return;
|
| - }
|
| -
|
| - key3_.assign(connection->read_buf()->StartOfBuffer() + *pos,
|
| - kWebSocketHandshakeBodyLen);
|
| - *pos += kWebSocketHandshakeBodyLen;
|
| - }
|
| -
|
| - std::string key3_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(WebSocketHixie76);
|
| -};
|
| -
|
| -const int WebSocketHixie76::kWebSocketHandshakeBodyLen = 8;
|
| -
|
| -class WebSocketHybi17 : public WebSocket {
|
| - public:
|
| - static WebSocket* Create(HttpServer* server,
|
| - HttpConnection* connection,
|
| - const HttpServerRequestInfo& request,
|
| - size_t* pos) {
|
| - std::string version = request.GetHeaderValue("sec-websocket-version");
|
| - if (version != "8" && version != "13")
|
| - return NULL;
|
| -
|
| - std::string key = request.GetHeaderValue("sec-websocket-key");
|
| - if (key.empty()) {
|
| - server->SendResponse(
|
| - connection->id(),
|
| - HttpServerResponseInfo::CreateFor500(
|
| - "Invalid request format. Sec-WebSocket-Key is empty or isn't "
|
| - "specified."));
|
| - return NULL;
|
| - }
|
| - return new WebSocketHybi17(server, connection, request, pos);
|
| - }
|
| -
|
| - void Accept(const HttpServerRequestInfo& request) override {
|
| - static const char* const kWebSocketGuid =
|
| - "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
| - std::string key = request.GetHeaderValue("sec-websocket-key");
|
| - std::string data = base::StringPrintf("%s%s", key.c_str(), kWebSocketGuid);
|
| - std::string encoded_hash;
|
| - base::Base64Encode(base::SHA1HashString(data), &encoded_hash);
|
| -
|
| - server_->SendRaw(connection_->id(),
|
| - base::StringPrintf(
|
| - "HTTP/1.1 101 WebSocket Protocol Handshake\r\n"
|
| - "Upgrade: WebSocket\r\n"
|
| - "Connection: Upgrade\r\n"
|
| - "Sec-WebSocket-Accept: %s\r\n"
|
| - "%s"
|
| - "\r\n",
|
| - encoded_hash.c_str(), response_extensions_.c_str()));
|
| - }
|
| -
|
| - ParseResult Read(std::string* message) override {
|
| - HttpConnection::ReadIOBuffer* read_buf = connection_->read_buf();
|
| - base::StringPiece frame(read_buf->StartOfBuffer(), read_buf->GetSize());
|
| - int bytes_consumed = 0;
|
| - ParseResult result = encoder_->DecodeFrame(frame, &bytes_consumed, message);
|
| - if (result == FRAME_OK)
|
| - read_buf->DidConsume(bytes_consumed);
|
| - if (result == FRAME_CLOSE)
|
| - closed_ = true;
|
| - return result;
|
| - }
|
| -
|
| - void Send(const std::string& message) override {
|
| - if (closed_)
|
| - return;
|
| - std::string encoded;
|
| - encoder_->EncodeFrame(message, 0, &encoded);
|
| - server_->SendRaw(connection_->id(), encoded);
|
| - }
|
| -
|
| - private:
|
| - WebSocketHybi17(HttpServer* server,
|
| - HttpConnection* connection,
|
| - const HttpServerRequestInfo& request,
|
| - size_t* pos)
|
| - : WebSocket(server, connection),
|
| - closed_(false) {
|
| - std::string request_extensions =
|
| - request.GetHeaderValue("sec-websocket-extensions");
|
| - encoder_.reset(WebSocketEncoder::CreateServer(request_extensions,
|
| - &response_extensions_));
|
| - if (!response_extensions_.empty()) {
|
| - response_extensions_ =
|
| - "Sec-WebSocket-Extensions: " + response_extensions_ + "\r\n";
|
| - }
|
| - }
|
| -
|
| - scoped_ptr<WebSocketEncoder> encoder_;
|
| - std::string response_extensions_;
|
| - bool closed_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(WebSocketHybi17);
|
| -};
|
| -
|
| -} // anonymous namespace
|
| -
|
| -WebSocket* WebSocket::CreateWebSocket(HttpServer* server,
|
| - HttpConnection* connection,
|
| - const HttpServerRequestInfo& request,
|
| - size_t* pos) {
|
| - WebSocket* socket = WebSocketHybi17::Create(server, connection, request, pos);
|
| - if (socket)
|
| - return socket;
|
| -
|
| - return WebSocketHixie76::Create(server, connection, request, pos);
|
| -}
|
| -
|
| -WebSocket::WebSocket(HttpServer* server, HttpConnection* connection)
|
| - : server_(server),
|
| - connection_(connection) {
|
| -}
|
| -
|
| -WebSocket::~WebSocket() {
|
| -}
|
| -
|
| -} // namespace net
|
|
|