blob: 6c2716405aad6181d3869d05941b4325fd6e16b9 [file] [log] [blame]
// 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 "base/memory/weak_ptr.h"
#include "net/base/net_log.h"
#include "net/socket/stream_socket.h"
namespace base {
class TimeDelta;
namespace net {
class AddressList;
class GrowableIOBuffer;
class IPEndPoint;
// A StreamSocket decorator. All functions are passed through to the wrapped
// socket, except for Write().
// Writes are buffered locally so that multiple Write()s to this class are
// issued as only one Write() to the wrapped socket. This is useful to force
// multiple requests to be issued in a single packet, as is needed to trigger
// edge cases in HTTP pipelining.
// Note that the Write() always returns synchronously. It will either buffer the
// entire input or return the most recently reported error.
// There are no bounds on the local buffer size. Use carefully.
class NET_EXPORT_PRIVATE BufferedWriteStreamSocket : public StreamSocket {
BufferedWriteStreamSocket(StreamSocket* socket_to_wrap);
virtual ~BufferedWriteStreamSocket();
// Socket interface
virtual int Read(IOBuffer* buf, int buf_len,
const CompletionCallback& callback) override;
virtual int Write(IOBuffer* buf, int buf_len,
const CompletionCallback& callback) override;
virtual bool SetReceiveBufferSize(int32 size) override;
virtual bool SetSendBufferSize(int32 size) override;
// StreamSocket interface
virtual int Connect(const CompletionCallback& callback) override;
virtual void Disconnect() override;
virtual bool IsConnected() const override;
virtual bool IsConnectedAndIdle() const override;
virtual int GetPeerAddress(IPEndPoint* address) const override;
virtual int GetLocalAddress(IPEndPoint* address) const override;
virtual const BoundNetLog& NetLog() const override;
virtual void SetSubresourceSpeculation() override;
virtual void SetOmniboxSpeculation() override;
virtual bool WasEverUsed() const override;
virtual bool UsingTCPFastOpen() const override;
virtual int64 NumBytesRead() const override;
virtual base::TimeDelta GetConnectTimeMicros() const override;
virtual bool WasNpnNegotiated() const override;
virtual NextProto GetNegotiatedProtocol() const override;
virtual bool GetSSLInfo(SSLInfo* ssl_info) override;
void DoDelayedWrite();
void OnIOComplete(int result);
scoped_ptr<StreamSocket> wrapped_socket_;
scoped_refptr<GrowableIOBuffer> io_buffer_;
scoped_refptr<GrowableIOBuffer> backup_buffer_;
base::WeakPtrFactory<BufferedWriteStreamSocket> weak_factory_;
bool callback_pending_;
bool wrapped_write_in_progress_;
int error_;
} // namespace net