Index: net/server/web_socket_encoder.cc |
diff --git a/net/server/web_socket_encoder.cc b/net/server/web_socket_encoder.cc |
deleted file mode 100644 |
index cb23326c28a61610ddd239a5c1a9db67f987cf84..0000000000000000000000000000000000000000 |
--- a/net/server/web_socket_encoder.cc |
+++ /dev/null |
@@ -1,364 +0,0 @@ |
-// Copyright 2014 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_encoder.h" |
- |
-#include "base/logging.h" |
-#include "base/strings/string_number_conversions.h" |
-#include "base/strings/stringprintf.h" |
-#include "net/base/io_buffer.h" |
-#include "net/websockets/websocket_extension_parser.h" |
- |
-namespace net { |
- |
-const char WebSocketEncoder::kClientExtensions[] = |
- "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits"; |
- |
-namespace { |
- |
-const int kInflaterChunkSize = 16 * 1024; |
- |
-// Constants for hybi-10 frame format. |
- |
-typedef int OpCode; |
- |
-const OpCode kOpCodeContinuation = 0x0; |
-const OpCode kOpCodeText = 0x1; |
-const OpCode kOpCodeBinary = 0x2; |
-const OpCode kOpCodeClose = 0x8; |
-const OpCode kOpCodePing = 0x9; |
-const OpCode kOpCodePong = 0xA; |
- |
-const unsigned char kFinalBit = 0x80; |
-const unsigned char kReserved1Bit = 0x40; |
-const unsigned char kReserved2Bit = 0x20; |
-const unsigned char kReserved3Bit = 0x10; |
-const unsigned char kOpCodeMask = 0xF; |
-const unsigned char kMaskBit = 0x80; |
-const unsigned char kPayloadLengthMask = 0x7F; |
- |
-const size_t kMaxSingleBytePayloadLength = 125; |
-const size_t kTwoBytePayloadLengthField = 126; |
-const size_t kEightBytePayloadLengthField = 127; |
-const size_t kMaskingKeyWidthInBytes = 4; |
- |
-WebSocket::ParseResult DecodeFrameHybi17(const base::StringPiece& frame, |
- bool client_frame, |
- int* bytes_consumed, |
- std::string* output, |
- bool* compressed) { |
- size_t data_length = frame.length(); |
- if (data_length < 2) |
- return WebSocket::FRAME_INCOMPLETE; |
- |
- const char* buffer_begin = const_cast<char*>(frame.data()); |
- const char* p = buffer_begin; |
- const char* buffer_end = p + data_length; |
- |
- unsigned char first_byte = *p++; |
- unsigned char second_byte = *p++; |
- |
- bool final = (first_byte & kFinalBit) != 0; |
- bool reserved1 = (first_byte & kReserved1Bit) != 0; |
- bool reserved2 = (first_byte & kReserved2Bit) != 0; |
- bool reserved3 = (first_byte & kReserved3Bit) != 0; |
- int op_code = first_byte & kOpCodeMask; |
- bool masked = (second_byte & kMaskBit) != 0; |
- *compressed = reserved1; |
- if (!final || reserved2 || reserved3) |
- return WebSocket::FRAME_ERROR; // Only compression extension is supported. |
- |
- bool closed = false; |
- switch (op_code) { |
- case kOpCodeClose: |
- closed = true; |
- break; |
- case kOpCodeText: |
- break; |
- case kOpCodeBinary: // We don't support binary frames yet. |
- case kOpCodeContinuation: // We don't support binary frames yet. |
- case kOpCodePing: // We don't support binary frames yet. |
- case kOpCodePong: // We don't support binary frames yet. |
- default: |
- return WebSocket::FRAME_ERROR; |
- } |
- |
- if (client_frame && !masked) // In Hybi-17 spec client MUST mask his frame. |
- return WebSocket::FRAME_ERROR; |
- |
- uint64 payload_length64 = second_byte & kPayloadLengthMask; |
- if (payload_length64 > kMaxSingleBytePayloadLength) { |
- int extended_payload_length_size; |
- if (payload_length64 == kTwoBytePayloadLengthField) |
- extended_payload_length_size = 2; |
- else { |
- DCHECK(payload_length64 == kEightBytePayloadLengthField); |
- extended_payload_length_size = 8; |
- } |
- if (buffer_end - p < extended_payload_length_size) |
- return WebSocket::FRAME_INCOMPLETE; |
- payload_length64 = 0; |
- for (int i = 0; i < extended_payload_length_size; ++i) { |
- payload_length64 <<= 8; |
- payload_length64 |= static_cast<unsigned char>(*p++); |
- } |
- } |
- |
- size_t actual_masking_key_length = masked ? kMaskingKeyWidthInBytes : 0; |
- static const uint64 max_payload_length = 0x7FFFFFFFFFFFFFFFull; |
- static size_t max_length = std::numeric_limits<size_t>::max(); |
- if (payload_length64 > max_payload_length || |
- payload_length64 + actual_masking_key_length > max_length) { |
- // WebSocket frame length too large. |
- return WebSocket::FRAME_ERROR; |
- } |
- size_t payload_length = static_cast<size_t>(payload_length64); |
- |
- size_t total_length = actual_masking_key_length + payload_length; |
- if (static_cast<size_t>(buffer_end - p) < total_length) |
- return WebSocket::FRAME_INCOMPLETE; |
- |
- if (masked) { |
- output->resize(payload_length); |
- const char* masking_key = p; |
- char* payload = const_cast<char*>(p + kMaskingKeyWidthInBytes); |
- for (size_t i = 0; i < payload_length; ++i) // Unmask the payload. |
- (*output)[i] = payload[i] ^ masking_key[i % kMaskingKeyWidthInBytes]; |
- } else { |
- output->assign(p, p + payload_length); |
- } |
- |
- size_t pos = p + actual_masking_key_length + payload_length - buffer_begin; |
- *bytes_consumed = pos; |
- return closed ? WebSocket::FRAME_CLOSE : WebSocket::FRAME_OK; |
-} |
- |
-void EncodeFrameHybi17(const std::string& message, |
- int masking_key, |
- bool compressed, |
- std::string* output) { |
- std::vector<char> frame; |
- OpCode op_code = kOpCodeText; |
- size_t data_length = message.length(); |
- |
- int reserved1 = compressed ? kReserved1Bit : 0; |
- frame.push_back(kFinalBit | op_code | reserved1); |
- char mask_key_bit = masking_key != 0 ? kMaskBit : 0; |
- if (data_length <= kMaxSingleBytePayloadLength) |
- frame.push_back(data_length | mask_key_bit); |
- else if (data_length <= 0xFFFF) { |
- frame.push_back(kTwoBytePayloadLengthField | mask_key_bit); |
- frame.push_back((data_length & 0xFF00) >> 8); |
- frame.push_back(data_length & 0xFF); |
- } else { |
- frame.push_back(kEightBytePayloadLengthField | mask_key_bit); |
- char extended_payload_length[8]; |
- size_t remaining = data_length; |
- // Fill the length into extended_payload_length in the network byte order. |
- for (int i = 0; i < 8; ++i) { |
- extended_payload_length[7 - i] = remaining & 0xFF; |
- remaining >>= 8; |
- } |
- frame.insert(frame.end(), extended_payload_length, |
- extended_payload_length + 8); |
- DCHECK(!remaining); |
- } |
- |
- const char* data = const_cast<char*>(message.data()); |
- if (masking_key != 0) { |
- const char* mask_bytes = reinterpret_cast<char*>(&masking_key); |
- frame.insert(frame.end(), mask_bytes, mask_bytes + 4); |
- for (size_t i = 0; i < data_length; ++i) // Mask the payload. |
- frame.push_back(data[i] ^ mask_bytes[i % kMaskingKeyWidthInBytes]); |
- } else { |
- frame.insert(frame.end(), data, data + data_length); |
- } |
- *output = std::string(&frame[0], frame.size()); |
-} |
- |
-} // anonymous namespace |
- |
-// static |
-WebSocketEncoder* WebSocketEncoder::CreateServer( |
- const std::string& request_extensions, |
- std::string* response_extensions) { |
- bool deflate; |
- int client_window_bits; |
- int server_window_bits; |
- bool client_no_context_takeover; |
- bool server_no_context_takeover; |
- ParseExtensions(request_extensions, &deflate, &client_window_bits, |
- &server_window_bits, &client_no_context_takeover, |
- &server_no_context_takeover); |
- |
- if (deflate) { |
- *response_extensions = base::StringPrintf( |
- "permessage-deflate; server_max_window_bits=%d; " |
- "client_max_window_bits=%d%s", |
- server_window_bits, client_window_bits, |
- server_no_context_takeover ? "; server_no_context_takeover" : ""); |
- return new WebSocketEncoder(true /* is_server */, server_window_bits, |
- client_window_bits, server_no_context_takeover); |
- } else { |
- *response_extensions = std::string(); |
- return new WebSocketEncoder(true /* is_server */); |
- } |
-} |
- |
-// static |
-WebSocketEncoder* WebSocketEncoder::CreateClient( |
- const std::string& response_extensions) { |
- bool deflate; |
- int client_window_bits; |
- int server_window_bits; |
- bool client_no_context_takeover; |
- bool server_no_context_takeover; |
- ParseExtensions(response_extensions, &deflate, &client_window_bits, |
- &server_window_bits, &client_no_context_takeover, |
- &server_no_context_takeover); |
- |
- if (deflate) { |
- return new WebSocketEncoder(false /* is_server */, client_window_bits, |
- server_window_bits, client_no_context_takeover); |
- } else { |
- return new WebSocketEncoder(false /* is_server */); |
- } |
-} |
- |
-// static |
-void WebSocketEncoder::ParseExtensions(const std::string& extensions, |
- bool* deflate, |
- int* client_window_bits, |
- int* server_window_bits, |
- bool* client_no_context_takeover, |
- bool* server_no_context_takeover) { |
- *deflate = false; |
- *client_window_bits = 15; |
- *server_window_bits = 15; |
- *client_no_context_takeover = false; |
- *server_no_context_takeover = false; |
- |
- if (extensions.empty()) |
- return; |
- |
- // TODO(dgozman): split extensions header if another extension is introduced. |
- WebSocketExtensionParser parser; |
- parser.Parse(extensions); |
- if (parser.has_error()) |
- return; |
- if (parser.extension().name() != "permessage-deflate") |
- return; |
- |
- const std::vector<WebSocketExtension::Parameter>& parameters = |
- parser.extension().parameters(); |
- for (const auto& param : parameters) { |
- const std::string& name = param.name(); |
- if (name == "client_max_window_bits" && param.HasValue()) { |
- int bits = 0; |
- if (base::StringToInt(param.value(), &bits) && bits >= 8 && bits <= 15) |
- *client_window_bits = bits; |
- } |
- if (name == "server_max_window_bits" && param.HasValue()) { |
- int bits = 0; |
- if (base::StringToInt(param.value(), &bits) && bits >= 8 && bits <= 15) |
- *server_window_bits = bits; |
- } |
- if (name == "client_no_context_takeover") |
- *client_no_context_takeover = true; |
- if (name == "server_no_context_takeover") |
- *server_no_context_takeover = true; |
- } |
- *deflate = true; |
-} |
- |
-WebSocketEncoder::WebSocketEncoder(bool is_server) : is_server_(is_server) { |
-} |
- |
-WebSocketEncoder::WebSocketEncoder(bool is_server, |
- int deflate_bits, |
- int inflate_bits, |
- bool no_context_takeover) |
- : is_server_(is_server) { |
- deflater_.reset(new WebSocketDeflater( |
- no_context_takeover ? WebSocketDeflater::DO_NOT_TAKE_OVER_CONTEXT |
- : WebSocketDeflater::TAKE_OVER_CONTEXT)); |
- inflater_.reset( |
- new WebSocketInflater(kInflaterChunkSize, kInflaterChunkSize)); |
- |
- if (!deflater_->Initialize(deflate_bits) || |
- !inflater_->Initialize(inflate_bits)) { |
- // Disable deflate support. |
- deflater_.reset(); |
- inflater_.reset(); |
- } |
-} |
- |
-WebSocketEncoder::~WebSocketEncoder() { |
-} |
- |
-WebSocket::ParseResult WebSocketEncoder::DecodeFrame( |
- const base::StringPiece& frame, |
- int* bytes_consumed, |
- std::string* output) { |
- bool compressed; |
- WebSocket::ParseResult result = |
- DecodeFrameHybi17(frame, is_server_, bytes_consumed, output, &compressed); |
- if (result == WebSocket::FRAME_OK && compressed) { |
- if (!Inflate(output)) |
- result = WebSocket::FRAME_ERROR; |
- } |
- return result; |
-} |
- |
-void WebSocketEncoder::EncodeFrame(const std::string& frame, |
- int masking_key, |
- std::string* output) { |
- std::string compressed; |
- if (Deflate(frame, &compressed)) |
- EncodeFrameHybi17(compressed, masking_key, true, output); |
- else |
- EncodeFrameHybi17(frame, masking_key, false, output); |
-} |
- |
-bool WebSocketEncoder::Inflate(std::string* message) { |
- if (!inflater_) |
- return false; |
- if (!inflater_->AddBytes(message->data(), message->length())) |
- return false; |
- if (!inflater_->Finish()) |
- return false; |
- |
- std::vector<char> output; |
- while (inflater_->CurrentOutputSize() > 0) { |
- scoped_refptr<IOBufferWithSize> chunk = |
- inflater_->GetOutput(inflater_->CurrentOutputSize()); |
- if (!chunk.get()) |
- return false; |
- output.insert(output.end(), chunk->data(), chunk->data() + chunk->size()); |
- } |
- |
- *message = |
- output.size() ? std::string(&output[0], output.size()) : std::string(); |
- return true; |
-} |
- |
-bool WebSocketEncoder::Deflate(const std::string& message, |
- std::string* output) { |
- if (!deflater_) |
- return false; |
- if (!deflater_->AddBytes(message.data(), message.length())) { |
- deflater_->Finish(); |
- return false; |
- } |
- if (!deflater_->Finish()) |
- return false; |
- scoped_refptr<IOBufferWithSize> buffer = |
- deflater_->GetOutput(deflater_->CurrentOutputSize()); |
- if (!buffer.get()) |
- return false; |
- *output = std::string(buffer->data(), buffer->size()); |
- return true; |
-} |
- |
-} // namespace net |