| Index: net/websockets/websocket_inflater.cc
|
| diff --git a/net/websockets/websocket_inflater.cc b/net/websockets/websocket_inflater.cc
|
| deleted file mode 100644
|
| index 1d6122bf9531ec0a6455d3fc0b2b0fdf435be54f..0000000000000000000000000000000000000000
|
| --- a/net/websockets/websocket_inflater.cc
|
| +++ /dev/null
|
| @@ -1,283 +0,0 @@
|
| -// Copyright 2013 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/websockets/websocket_inflater.h"
|
| -
|
| -#include <algorithm>
|
| -#include <deque>
|
| -#include <vector>
|
| -
|
| -#include "base/logging.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "third_party/zlib/zlib.h"
|
| -
|
| -namespace net {
|
| -
|
| -namespace {
|
| -
|
| -class ShrinkableIOBufferWithSize : public IOBufferWithSize {
|
| - public:
|
| - explicit ShrinkableIOBufferWithSize(int size)
|
| - : IOBufferWithSize(size) {}
|
| -
|
| - void Shrink(int new_size) {
|
| - DCHECK_LE(new_size, size_);
|
| - size_ = new_size;
|
| - }
|
| -
|
| - private:
|
| - ~ShrinkableIOBufferWithSize() override {}
|
| -};
|
| -
|
| -} // namespace
|
| -
|
| -WebSocketInflater::WebSocketInflater()
|
| - : input_queue_(kDefaultInputIOBufferCapacity),
|
| - output_buffer_(kDefaultBufferCapacity) {}
|
| -
|
| -WebSocketInflater::WebSocketInflater(size_t input_queue_capacity,
|
| - size_t output_buffer_capacity)
|
| - : input_queue_(input_queue_capacity),
|
| - output_buffer_(output_buffer_capacity) {
|
| - DCHECK_GT(input_queue_capacity, 0u);
|
| - DCHECK_GT(output_buffer_capacity, 0u);
|
| -}
|
| -
|
| -bool WebSocketInflater::Initialize(int window_bits) {
|
| - DCHECK_LE(8, window_bits);
|
| - DCHECK_GE(15, window_bits);
|
| - stream_.reset(new z_stream);
|
| - memset(stream_.get(), 0, sizeof(*stream_));
|
| - int result = inflateInit2(stream_.get(), -window_bits);
|
| - if (result != Z_OK) {
|
| - inflateEnd(stream_.get());
|
| - stream_.reset();
|
| - return false;
|
| - }
|
| - return true;
|
| -}
|
| -
|
| -WebSocketInflater::~WebSocketInflater() {
|
| - if (stream_) {
|
| - inflateEnd(stream_.get());
|
| - stream_.reset();
|
| - }
|
| -}
|
| -
|
| -bool WebSocketInflater::AddBytes(const char* data, size_t size) {
|
| - if (!size)
|
| - return true;
|
| -
|
| - if (!input_queue_.IsEmpty()) {
|
| - // choked
|
| - input_queue_.Push(data, size);
|
| - return true;
|
| - }
|
| -
|
| - int result = InflateWithFlush(data, size);
|
| - if (stream_->avail_in > 0)
|
| - input_queue_.Push(&data[size - stream_->avail_in], stream_->avail_in);
|
| -
|
| - return result == Z_OK || result == Z_BUF_ERROR;
|
| -}
|
| -
|
| -bool WebSocketInflater::Finish() {
|
| - return AddBytes("\x00\x00\xff\xff", 4);
|
| -}
|
| -
|
| -scoped_refptr<IOBufferWithSize> WebSocketInflater::GetOutput(size_t size) {
|
| - scoped_refptr<ShrinkableIOBufferWithSize> buffer =
|
| - new ShrinkableIOBufferWithSize(size);
|
| - size_t num_bytes_copied = 0;
|
| -
|
| - while (num_bytes_copied < size && output_buffer_.Size() > 0) {
|
| - size_t num_bytes_to_copy =
|
| - std::min(output_buffer_.Size(), size - num_bytes_copied);
|
| - output_buffer_.Read(&buffer->data()[num_bytes_copied], num_bytes_to_copy);
|
| - num_bytes_copied += num_bytes_to_copy;
|
| - int result = InflateChokedInput();
|
| - if (result != Z_OK && result != Z_BUF_ERROR)
|
| - return NULL;
|
| - }
|
| - buffer->Shrink(num_bytes_copied);
|
| - return buffer;
|
| -}
|
| -
|
| -int WebSocketInflater::InflateWithFlush(const char* next_in, size_t avail_in) {
|
| - int result = Inflate(next_in, avail_in, Z_NO_FLUSH);
|
| - if (result != Z_OK && result != Z_BUF_ERROR)
|
| - return result;
|
| -
|
| - if (CurrentOutputSize() > 0)
|
| - return result;
|
| - // CurrentOutputSize() == 0 means there is no data to be output,
|
| - // so we should make sure it by using Z_SYNC_FLUSH.
|
| - return Inflate(reinterpret_cast<const char*>(stream_->next_in),
|
| - stream_->avail_in,
|
| - Z_SYNC_FLUSH);
|
| -}
|
| -
|
| -int WebSocketInflater::Inflate(const char* next_in,
|
| - size_t avail_in,
|
| - int flush) {
|
| - stream_->next_in = reinterpret_cast<Bytef*>(const_cast<char*>(next_in));
|
| - stream_->avail_in = avail_in;
|
| -
|
| - int result = Z_BUF_ERROR;
|
| - do {
|
| - std::pair<char*, size_t> tail = output_buffer_.GetTail();
|
| - if (!tail.second)
|
| - break;
|
| -
|
| - stream_->next_out = reinterpret_cast<Bytef*>(tail.first);
|
| - stream_->avail_out = tail.second;
|
| - result = inflate(stream_.get(), flush);
|
| - output_buffer_.AdvanceTail(tail.second - stream_->avail_out);
|
| - if (result == Z_STREAM_END) {
|
| - // Received a block with BFINAL set to 1. Reset the decompression state.
|
| - result = inflateReset(stream_.get());
|
| - } else if (tail.second == stream_->avail_out) {
|
| - break;
|
| - }
|
| - } while (result == Z_OK || result == Z_BUF_ERROR);
|
| - return result;
|
| -}
|
| -
|
| -int WebSocketInflater::InflateChokedInput() {
|
| - if (input_queue_.IsEmpty())
|
| - return InflateWithFlush(NULL, 0);
|
| -
|
| - int result = Z_BUF_ERROR;
|
| - while (!input_queue_.IsEmpty()) {
|
| - std::pair<char*, size_t> top = input_queue_.Top();
|
| -
|
| - result = InflateWithFlush(top.first, top.second);
|
| - input_queue_.Consume(top.second - stream_->avail_in);
|
| -
|
| - if (result != Z_OK && result != Z_BUF_ERROR)
|
| - return result;
|
| -
|
| - if (stream_->avail_in > 0) {
|
| - // There are some data which are not consumed.
|
| - break;
|
| - }
|
| - }
|
| - return result;
|
| -}
|
| -
|
| -WebSocketInflater::OutputBuffer::OutputBuffer(size_t capacity)
|
| - : capacity_(capacity),
|
| - buffer_(capacity_ + 1), // 1 for sentinel
|
| - head_(0),
|
| - tail_(0) {}
|
| -
|
| -WebSocketInflater::OutputBuffer::~OutputBuffer() {}
|
| -
|
| -size_t WebSocketInflater::OutputBuffer::Size() const {
|
| - return (tail_ + buffer_.size() - head_) % buffer_.size();
|
| -}
|
| -
|
| -std::pair<char*, size_t> WebSocketInflater::OutputBuffer::GetTail() {
|
| - DCHECK_LT(tail_, buffer_.size());
|
| - return std::make_pair(&buffer_[tail_],
|
| - std::min(capacity_ - Size(), buffer_.size() - tail_));
|
| -}
|
| -
|
| -void WebSocketInflater::OutputBuffer::Read(char* dest, size_t size) {
|
| - DCHECK_LE(size, Size());
|
| -
|
| - size_t num_bytes_copied = 0;
|
| - if (tail_ < head_) {
|
| - size_t num_bytes_to_copy = std::min(size, buffer_.size() - head_);
|
| - DCHECK_LT(head_, buffer_.size());
|
| - memcpy(&dest[num_bytes_copied], &buffer_[head_], num_bytes_to_copy);
|
| - AdvanceHead(num_bytes_to_copy);
|
| - num_bytes_copied += num_bytes_to_copy;
|
| - }
|
| -
|
| - if (num_bytes_copied == size)
|
| - return;
|
| - DCHECK_LE(head_, tail_);
|
| - size_t num_bytes_to_copy = size - num_bytes_copied;
|
| - DCHECK_LE(num_bytes_to_copy, tail_ - head_);
|
| - DCHECK_LT(head_, buffer_.size());
|
| - memcpy(&dest[num_bytes_copied], &buffer_[head_], num_bytes_to_copy);
|
| - AdvanceHead(num_bytes_to_copy);
|
| - num_bytes_copied += num_bytes_to_copy;
|
| - DCHECK_EQ(size, num_bytes_copied);
|
| - return;
|
| -}
|
| -
|
| -void WebSocketInflater::OutputBuffer::AdvanceHead(size_t advance) {
|
| - DCHECK_LE(advance, Size());
|
| - head_ = (head_ + advance) % buffer_.size();
|
| -}
|
| -
|
| -void WebSocketInflater::OutputBuffer::AdvanceTail(size_t advance) {
|
| - DCHECK_LE(advance + Size(), capacity_);
|
| - tail_ = (tail_ + advance) % buffer_.size();
|
| -}
|
| -
|
| -WebSocketInflater::InputQueue::InputQueue(size_t capacity)
|
| - : capacity_(capacity), head_of_first_buffer_(0), tail_of_last_buffer_(0) {}
|
| -
|
| -WebSocketInflater::InputQueue::~InputQueue() {}
|
| -
|
| -std::pair<char*, size_t> WebSocketInflater::InputQueue::Top() {
|
| - DCHECK(!IsEmpty());
|
| - if (buffers_.size() == 1) {
|
| - return std::make_pair(&buffers_.front()->data()[head_of_first_buffer_],
|
| - tail_of_last_buffer_ - head_of_first_buffer_);
|
| - }
|
| - return std::make_pair(&buffers_.front()->data()[head_of_first_buffer_],
|
| - capacity_ - head_of_first_buffer_);
|
| -}
|
| -
|
| -void WebSocketInflater::InputQueue::Push(const char* data, size_t size) {
|
| - if (!size)
|
| - return;
|
| -
|
| - size_t num_copied_bytes = 0;
|
| - if (!IsEmpty())
|
| - num_copied_bytes += PushToLastBuffer(data, size);
|
| -
|
| - while (num_copied_bytes < size) {
|
| - DCHECK(IsEmpty() || tail_of_last_buffer_ == capacity_);
|
| -
|
| - buffers_.push_back(new IOBufferWithSize(capacity_));
|
| - tail_of_last_buffer_ = 0;
|
| - num_copied_bytes +=
|
| - PushToLastBuffer(&data[num_copied_bytes], size - num_copied_bytes);
|
| - }
|
| -}
|
| -
|
| -void WebSocketInflater::InputQueue::Consume(size_t size) {
|
| - DCHECK(!IsEmpty());
|
| - DCHECK_LE(size + head_of_first_buffer_, capacity_);
|
| -
|
| - head_of_first_buffer_ += size;
|
| - if (head_of_first_buffer_ == capacity_) {
|
| - buffers_.pop_front();
|
| - head_of_first_buffer_ = 0;
|
| - }
|
| - if (buffers_.size() == 1 && head_of_first_buffer_ == tail_of_last_buffer_) {
|
| - buffers_.pop_front();
|
| - head_of_first_buffer_ = 0;
|
| - tail_of_last_buffer_ = 0;
|
| - }
|
| -}
|
| -
|
| -size_t WebSocketInflater::InputQueue::PushToLastBuffer(const char* data,
|
| - size_t size) {
|
| - DCHECK(!IsEmpty());
|
| - size_t num_bytes_to_copy = std::min(size, capacity_ - tail_of_last_buffer_);
|
| - if (!num_bytes_to_copy)
|
| - return 0;
|
| - IOBufferWithSize* buffer = buffers_.back().get();
|
| - memcpy(&buffer->data()[tail_of_last_buffer_], data, num_bytes_to_copy);
|
| - tail_of_last_buffer_ += num_bytes_to_copy;
|
| - return num_bytes_to_copy;
|
| -}
|
| -
|
| -} // namespace net
|
|
|