OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef NET_SOCKET_BUFFERED_WRITE_STREAM_SOCKET_H_ | |
6 #define NET_SOCKET_BUFFERED_WRITE_STREAM_SOCKET_H_ | |
7 | |
8 #include "base/basictypes.h" | |
9 #include "base/memory/scoped_ptr.h" | |
10 #include "base/memory/weak_ptr.h" | |
11 #include "net/base/net_log.h" | |
12 #include "net/socket/stream_socket.h" | |
13 | |
14 namespace base { | |
15 class TimeDelta; | |
16 } | |
17 | |
18 namespace net { | |
19 | |
20 class AddressList; | |
21 class GrowableIOBuffer; | |
22 class IPEndPoint; | |
23 | |
24 // A StreamSocket decorator. All functions are passed through to the wrapped | |
25 // socket, except for Write(). | |
26 // | |
27 // Writes are buffered locally so that multiple Write()s to this class are | |
28 // issued as only one Write() to the wrapped socket. This is useful to force | |
29 // multiple requests to be issued in a single packet, as is needed to trigger | |
30 // edge cases in HTTP pipelining. | |
31 // | |
32 // Note that the Write() always returns synchronously. It will either buffer the | |
33 // entire input or return the most recently reported error. | |
34 // | |
35 // There are no bounds on the local buffer size. Use carefully. | |
36 class NET_EXPORT_PRIVATE BufferedWriteStreamSocket : public StreamSocket { | |
37 public: | |
38 explicit BufferedWriteStreamSocket(scoped_ptr<StreamSocket> socket_to_wrap); | |
39 virtual ~BufferedWriteStreamSocket(); | |
40 | |
41 // Socket interface | |
42 virtual int Read(IOBuffer* buf, int buf_len, | |
43 const CompletionCallback& callback) OVERRIDE; | |
44 virtual int Write(IOBuffer* buf, int buf_len, | |
45 const CompletionCallback& callback) OVERRIDE; | |
46 virtual int SetReceiveBufferSize(int32 size) OVERRIDE; | |
47 virtual int SetSendBufferSize(int32 size) OVERRIDE; | |
48 | |
49 // StreamSocket interface | |
50 virtual int Connect(const CompletionCallback& callback) OVERRIDE; | |
51 virtual void Disconnect() OVERRIDE; | |
52 virtual bool IsConnected() const OVERRIDE; | |
53 virtual bool IsConnectedAndIdle() const OVERRIDE; | |
54 virtual int GetPeerAddress(IPEndPoint* address) const OVERRIDE; | |
55 virtual int GetLocalAddress(IPEndPoint* address) const OVERRIDE; | |
56 virtual const BoundNetLog& NetLog() const OVERRIDE; | |
57 virtual void SetSubresourceSpeculation() OVERRIDE; | |
58 virtual void SetOmniboxSpeculation() OVERRIDE; | |
59 virtual bool WasEverUsed() const OVERRIDE; | |
60 virtual bool UsingTCPFastOpen() const OVERRIDE; | |
61 virtual bool WasNpnNegotiated() const OVERRIDE; | |
62 virtual NextProto GetNegotiatedProtocol() const OVERRIDE; | |
63 virtual bool GetSSLInfo(SSLInfo* ssl_info) OVERRIDE; | |
64 | |
65 private: | |
66 void DoDelayedWrite(); | |
67 void OnIOComplete(int result); | |
68 | |
69 scoped_ptr<StreamSocket> wrapped_socket_; | |
70 scoped_refptr<GrowableIOBuffer> io_buffer_; | |
71 scoped_refptr<GrowableIOBuffer> backup_buffer_; | |
72 bool callback_pending_; | |
73 bool wrapped_write_in_progress_; | |
74 int error_; | |
75 | |
76 base::WeakPtrFactory<BufferedWriteStreamSocket> weak_factory_; | |
77 | |
78 DISALLOW_COPY_AND_ASSIGN(BufferedWriteStreamSocket); | |
79 }; | |
80 | |
81 } // namespace net | |
82 | |
83 #endif // NET_SOCKET_STREAM_SOCKET_H_ | |
OLD | NEW |