| Index: net/socket/buffered_write_stream_socket.cc
|
| diff --git a/net/socket/buffered_write_stream_socket.cc b/net/socket/buffered_write_stream_socket.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..94c6d573d93c01c7f148b8e436e15f76abb58fab
|
| --- /dev/null
|
| +++ b/net/socket/buffered_write_stream_socket.cc
|
| @@ -0,0 +1,156 @@
|
| +// 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/socket/buffered_write_stream_socket.h"
|
| +
|
| +#include "base/bind.h"
|
| +#include "base/location.h"
|
| +#include "base/message_loop.h"
|
| +#include "net/base/io_buffer.h"
|
| +#include "net/base/net_errors.h"
|
| +
|
| +namespace net {
|
| +
|
| +namespace {
|
| +
|
| +void AppendBuffer(GrowableIOBuffer* dst, IOBuffer* src, int src_len) {
|
| + int old_capacity = dst->capacity();
|
| + dst->SetCapacity(old_capacity + src_len);
|
| + memcpy(dst->StartOfBuffer() + old_capacity, src->data(), src_len);
|
| +}
|
| +
|
| +} // anonymous namespace
|
| +
|
| +BufferedWriteStreamSocket::BufferedWriteStreamSocket(
|
| + StreamSocket* socket_to_wrap)
|
| + : wrapped_socket_(socket_to_wrap),
|
| + io_buffer_(new GrowableIOBuffer()),
|
| + backup_buffer_(new GrowableIOBuffer()),
|
| + ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)),
|
| + callback_pending_(false),
|
| + wrapped_write_in_progress_(false),
|
| + error_(0) {
|
| +}
|
| +
|
| +BufferedWriteStreamSocket::~BufferedWriteStreamSocket() {
|
| +}
|
| +
|
| +int BufferedWriteStreamSocket::Read(IOBuffer* buf, int buf_len,
|
| + const CompletionCallback& callback) {
|
| + return wrapped_socket_->Read(buf, buf_len, callback);
|
| +}
|
| +
|
| +int BufferedWriteStreamSocket::Write(IOBuffer* buf, int buf_len,
|
| + const CompletionCallback& callback) {
|
| + if (error_) {
|
| + return error_;
|
| + }
|
| + GrowableIOBuffer* idle_buffer =
|
| + wrapped_write_in_progress_ ? backup_buffer_.get() : io_buffer_.get();
|
| + AppendBuffer(idle_buffer, buf, buf_len);
|
| + if (!callback_pending_) {
|
| + MessageLoop::current()->PostTask(
|
| + FROM_HERE,
|
| + base::Bind(&BufferedWriteStreamSocket::DoDelayedWrite,
|
| + weak_factory_.GetWeakPtr()));
|
| + callback_pending_ = true;
|
| + }
|
| + return buf_len;
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::SetReceiveBufferSize(int32 size) {
|
| + return wrapped_socket_->SetReceiveBufferSize(size);
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::SetSendBufferSize(int32 size) {
|
| + return wrapped_socket_->SetSendBufferSize(size);
|
| +}
|
| +
|
| +int BufferedWriteStreamSocket::Connect(const CompletionCallback& callback) {
|
| + return wrapped_socket_->Connect(callback);
|
| +}
|
| +
|
| +void BufferedWriteStreamSocket::Disconnect() {
|
| + wrapped_socket_->Disconnect();
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::IsConnected() const {
|
| + return wrapped_socket_->IsConnected();
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::IsConnectedAndIdle() const {
|
| + return wrapped_socket_->IsConnectedAndIdle();
|
| +}
|
| +
|
| +int BufferedWriteStreamSocket::GetPeerAddress(AddressList* address) const {
|
| + return wrapped_socket_->GetPeerAddress(address);
|
| +}
|
| +
|
| +int BufferedWriteStreamSocket::GetLocalAddress(IPEndPoint* address) const {
|
| + return wrapped_socket_->GetLocalAddress(address);
|
| +}
|
| +
|
| +const BoundNetLog& BufferedWriteStreamSocket::NetLog() const {
|
| + return wrapped_socket_->NetLog();
|
| +}
|
| +
|
| +void BufferedWriteStreamSocket::SetSubresourceSpeculation() {
|
| + wrapped_socket_->SetSubresourceSpeculation();
|
| +}
|
| +
|
| +void BufferedWriteStreamSocket::SetOmniboxSpeculation() {
|
| + wrapped_socket_->SetOmniboxSpeculation();
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::WasEverUsed() const {
|
| + return wrapped_socket_->WasEverUsed();
|
| +}
|
| +
|
| +bool BufferedWriteStreamSocket::UsingTCPFastOpen() const {
|
| + return wrapped_socket_->UsingTCPFastOpen();
|
| +}
|
| +
|
| +int64 BufferedWriteStreamSocket::NumBytesRead() const {
|
| + return wrapped_socket_->NumBytesRead();
|
| +}
|
| +
|
| +base::TimeDelta BufferedWriteStreamSocket::GetConnectTimeMicros() const {
|
| + return wrapped_socket_->GetConnectTimeMicros();
|
| +}
|
| +
|
| +void BufferedWriteStreamSocket::DoDelayedWrite() {
|
| + int result = wrapped_socket_->Write(
|
| + io_buffer_, io_buffer_->RemainingCapacity(),
|
| + base::Bind(&BufferedWriteStreamSocket::OnIOComplete,
|
| + base::Unretained(this)));
|
| + if (result == ERR_IO_PENDING) {
|
| + callback_pending_ = true;
|
| + wrapped_write_in_progress_ = true;
|
| + } else {
|
| + OnIOComplete(result);
|
| + }
|
| +}
|
| +
|
| +void BufferedWriteStreamSocket::OnIOComplete(int result) {
|
| + callback_pending_ = false;
|
| + wrapped_write_in_progress_ = false;
|
| + if (backup_buffer_->RemainingCapacity()) {
|
| + AppendBuffer(io_buffer_.get(), backup_buffer_.get(),
|
| + backup_buffer_->RemainingCapacity());
|
| + backup_buffer_->SetCapacity(0);
|
| + }
|
| + if (result < 0) {
|
| + error_ = result;
|
| + io_buffer_->SetCapacity(0);
|
| + } else {
|
| + io_buffer_->set_offset(io_buffer_->offset() + result);
|
| + if (io_buffer_->RemainingCapacity()) {
|
| + DoDelayedWrite();
|
| + } else {
|
| + io_buffer_->SetCapacity(0);
|
| + }
|
| + }
|
| +}
|
| +
|
| +} // namespace net
|
|
|