Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(143)

Unified Diff: net/server/web_socket_encoder.cc

Issue 992733002: Remove //net (except for Android test stuff) and sdch (Closed) Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « net/server/web_socket_encoder.h ('k') | net/server/web_socket_encoder_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « net/server/web_socket_encoder.h ('k') | net/server/web_socket_encoder_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698