Index: remoting/protocol/fake_stream_socket.h |
diff --git a/remoting/protocol/fake_stream_socket.h b/remoting/protocol/fake_stream_socket.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8bbae10fa78bbb9461345b13c26d1ac4df0ba39c |
--- /dev/null |
+++ b/remoting/protocol/fake_stream_socket.h |
@@ -0,0 +1,158 @@ |
+// 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. |
+ |
+#ifndef REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |
+#define REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |
+ |
+#include <map> |
+#include <string> |
+ |
+#include "base/memory/scoped_ptr.h" |
+#include "base/memory/weak_ptr.h" |
+#include "net/base/completion_callback.h" |
+#include "net/socket/stream_socket.h" |
+#include "remoting/protocol/stream_channel_factory.h" |
+ |
+namespace base { |
+class SingleThreadTaskRunner; |
+} |
+ |
+namespace remoting { |
+namespace protocol { |
+ |
+// FakeStreamSocket implement net::StreamSocket interface. All data written to |
+// FakeStreamSocket is stored in a buffer returned by written_data(). Read() |
+// reads data from another buffer that can be set with AppendInputData(). |
+// Pending reads are supported, so if there is a pending read AppendInputData() |
+// calls the read callback. |
+// |
+// Two fake sockets can be connected to each other using the |
+// PairWith() method, e.g.: a->PairWith(b). After this all data |
+// written to |a| can be read from |b| and vice versa. Two connected |
+// sockets |a| and |b| must be created and used on the same thread. |
+class FakeStreamSocket : public net::StreamSocket { |
+ public: |
+ FakeStreamSocket(); |
+ virtual ~FakeStreamSocket(); |
+ |
+ // Returns all data written to the socket. |
+ const std::string& written_data() const { return written_data_; } |
+ |
+ // Set's maximum number of bytes written by each Write() call. |
Wez
2014/09/23 17:07:38
nit: Sets
Sergey Ulanov
2014/09/23 19:06:09
Done.
|
+ void set_write_limit(int write_limit) { write_limit_ = write_limit; } |
+ |
+ // Enables asynchronous Write(). |
+ void set_async_write(bool async_write) { async_write_ = async_write; } |
+ |
+ // Set error codes for the next Read() and Write() calls. Once returned the |
+ // values are automatically reset to net::OK . |
+ void set_next_read_error(int error) { next_read_error_ = error; } |
+ void set_next_write_error(int error) { next_write_error_ = error; } |
+ |
+ // Appends |data| to the read buffer. |
+ void AppendInputData(const std::string& data); |
+ |
+ // Pairs the socket with |peer_socket|. Deleting either of the paired sockets |
+ // unpairs them. |
+ void PairWith(FakeStreamSocket* peer_socket); |
+ |
+ // Current input position in bytes. |
+ int input_pos() const { return input_pos_; } |
+ |
+ // True if a Read() call is currently pending. |
+ bool read_pending() const { return !read_callback_.is_null(); } |
+ |
+ base::WeakPtr<FakeStreamSocket> GetWeakPtr(); |
+ |
+ // net::Socket implementation. |
+ virtual int Read(net::IOBuffer* buf, int buf_len, |
+ const net::CompletionCallback& callback) OVERRIDE; |
+ virtual int Write(net::IOBuffer* buf, int buf_len, |
+ const net::CompletionCallback& callback) OVERRIDE; |
+ virtual int SetReceiveBufferSize(int32 size) OVERRIDE; |
+ virtual int SetSendBufferSize(int32 size) OVERRIDE; |
+ |
+ // net::StreamSocket interface. |
+ virtual int Connect(const net::CompletionCallback& callback) OVERRIDE; |
+ virtual void Disconnect() OVERRIDE; |
+ virtual bool IsConnected() const OVERRIDE; |
+ virtual bool IsConnectedAndIdle() const OVERRIDE; |
+ virtual int GetPeerAddress(net::IPEndPoint* address) const OVERRIDE; |
+ virtual int GetLocalAddress(net::IPEndPoint* address) const OVERRIDE; |
+ virtual const net::BoundNetLog& NetLog() const OVERRIDE; |
+ virtual void SetSubresourceSpeculation() OVERRIDE; |
+ virtual void SetOmniboxSpeculation() OVERRIDE; |
+ virtual bool WasEverUsed() const OVERRIDE; |
+ virtual bool UsingTCPFastOpen() const OVERRIDE; |
+ virtual bool WasNpnNegotiated() const OVERRIDE; |
+ virtual net::NextProto GetNegotiatedProtocol() const OVERRIDE; |
+ virtual bool GetSSLInfo(net::SSLInfo* ssl_info) OVERRIDE; |
+ |
+ private: |
+ void DoAsyncWrite(scoped_refptr<net::IOBuffer> buf, int buf_len, |
+ const net::CompletionCallback& callback); |
+ void DoWrite(net::IOBuffer* buf, int buf_len); |
+ |
+ bool async_write_; |
+ bool write_pending_; |
+ int write_limit_; |
+ int next_write_error_; |
+ |
+ int next_read_error_; |
+ scoped_refptr<net::IOBuffer> read_buffer_; |
+ int read_buffer_size_; |
+ net::CompletionCallback read_callback_; |
+ base::WeakPtr<FakeStreamSocket> peer_socket_; |
+ |
+ std::string written_data_; |
+ std::string input_data_; |
+ int input_pos_; |
+ |
+ net::BoundNetLog net_log_; |
+ |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
+ base::WeakPtrFactory<FakeStreamSocket> weak_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeStreamSocket); |
+}; |
+ |
+// StreamChannelFactory that creates FakeStreamSocket. |
+class FakeStreamChannelFactory : public StreamChannelFactory { |
+ public: |
+ FakeStreamChannelFactory(); |
+ virtual ~FakeStreamChannelFactory(); |
+ |
+ void set_asynchronous_create(bool asynchronous_create) { |
+ asynchronous_create_ = asynchronous_create; |
+ } |
+ |
+ void set_fail_create(bool fail_create) { fail_create_ = fail_create; } |
+ |
+ FakeStreamSocket* GetFakeChannel(const std::string& name); |
+ |
+ // ChannelFactory interface. |
+ virtual void CreateChannel(const std::string& name, |
+ const ChannelCreatedCallback& callback) OVERRIDE; |
+ virtual void CancelChannelCreation(const std::string& name) OVERRIDE; |
+ |
+ private: |
+ void NotifyChannelCreated(scoped_ptr<FakeStreamSocket> owned_channel, |
+ const std::string& name, |
+ const ChannelCreatedCallback& callback); |
+ |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
+ bool asynchronous_create_; |
+ std::map<std::string, base::WeakPtr<FakeStreamSocket> > channels_; |
+ |
+ bool fail_create_; |
+ |
+ base::WeakPtrFactory<FakeStreamChannelFactory> weak_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FakeStreamChannelFactory); |
+}; |
+ |
+} // namespace protocol |
+} // namespace remoting |
+ |
+#endif // REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |