| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ | 5 #ifndef REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |
| 6 #define REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ | 6 #define REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |
| 7 | 7 |
| 8 #include <map> | 8 #include <map> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 // Pending reads are supported, so if there is a pending read AppendInputData() | 27 // Pending reads are supported, so if there is a pending read AppendInputData() |
| 28 // calls the read callback. | 28 // calls the read callback. |
| 29 // | 29 // |
| 30 // Two fake sockets can be connected to each other using the | 30 // Two fake sockets can be connected to each other using the |
| 31 // PairWith() method, e.g.: a->PairWith(b). After this all data | 31 // PairWith() method, e.g.: a->PairWith(b). After this all data |
| 32 // written to |a| can be read from |b| and vice versa. Two connected | 32 // written to |a| can be read from |b| and vice versa. Two connected |
| 33 // sockets |a| and |b| must be created and used on the same thread. | 33 // sockets |a| and |b| must be created and used on the same thread. |
| 34 class FakeStreamSocket : public net::StreamSocket { | 34 class FakeStreamSocket : public net::StreamSocket { |
| 35 public: | 35 public: |
| 36 FakeStreamSocket(); | 36 FakeStreamSocket(); |
| 37 virtual ~FakeStreamSocket(); | 37 ~FakeStreamSocket() override; |
| 38 | 38 |
| 39 // Returns all data written to the socket. | 39 // Returns all data written to the socket. |
| 40 const std::string& written_data() const { return written_data_; } | 40 const std::string& written_data() const { return written_data_; } |
| 41 | 41 |
| 42 // Sets maximum number of bytes written by each Write() call. | 42 // Sets maximum number of bytes written by each Write() call. |
| 43 void set_write_limit(int write_limit) { write_limit_ = write_limit; } | 43 void set_write_limit(int write_limit) { write_limit_ = write_limit; } |
| 44 | 44 |
| 45 // Enables asynchronous Write(). | 45 // Enables asynchronous Write(). |
| 46 void set_async_write(bool async_write) { async_write_ = async_write; } | 46 void set_async_write(bool async_write) { async_write_ = async_write; } |
| 47 | 47 |
| (...skipping 11 matching lines...) Expand all Loading... |
| 59 | 59 |
| 60 // Current input position in bytes. | 60 // Current input position in bytes. |
| 61 int input_pos() const { return input_pos_; } | 61 int input_pos() const { return input_pos_; } |
| 62 | 62 |
| 63 // True if a Read() call is currently pending. | 63 // True if a Read() call is currently pending. |
| 64 bool read_pending() const { return !read_callback_.is_null(); } | 64 bool read_pending() const { return !read_callback_.is_null(); } |
| 65 | 65 |
| 66 base::WeakPtr<FakeStreamSocket> GetWeakPtr(); | 66 base::WeakPtr<FakeStreamSocket> GetWeakPtr(); |
| 67 | 67 |
| 68 // net::Socket implementation. | 68 // net::Socket implementation. |
| 69 virtual int Read(net::IOBuffer* buf, int buf_len, | 69 int Read(net::IOBuffer* buf, |
| 70 const net::CompletionCallback& callback) override; | 70 int buf_len, |
| 71 virtual int Write(net::IOBuffer* buf, int buf_len, | 71 const net::CompletionCallback& callback) override; |
| 72 const net::CompletionCallback& callback) override; | 72 int Write(net::IOBuffer* buf, |
| 73 virtual int SetReceiveBufferSize(int32 size) override; | 73 int buf_len, |
| 74 virtual int SetSendBufferSize(int32 size) override; | 74 const net::CompletionCallback& callback) override; |
| 75 int SetReceiveBufferSize(int32 size) override; |
| 76 int SetSendBufferSize(int32 size) override; |
| 75 | 77 |
| 76 // net::StreamSocket interface. | 78 // net::StreamSocket interface. |
| 77 virtual int Connect(const net::CompletionCallback& callback) override; | 79 int Connect(const net::CompletionCallback& callback) override; |
| 78 virtual void Disconnect() override; | 80 void Disconnect() override; |
| 79 virtual bool IsConnected() const override; | 81 bool IsConnected() const override; |
| 80 virtual bool IsConnectedAndIdle() const override; | 82 bool IsConnectedAndIdle() const override; |
| 81 virtual int GetPeerAddress(net::IPEndPoint* address) const override; | 83 int GetPeerAddress(net::IPEndPoint* address) const override; |
| 82 virtual int GetLocalAddress(net::IPEndPoint* address) const override; | 84 int GetLocalAddress(net::IPEndPoint* address) const override; |
| 83 virtual const net::BoundNetLog& NetLog() const override; | 85 const net::BoundNetLog& NetLog() const override; |
| 84 virtual void SetSubresourceSpeculation() override; | 86 void SetSubresourceSpeculation() override; |
| 85 virtual void SetOmniboxSpeculation() override; | 87 void SetOmniboxSpeculation() override; |
| 86 virtual bool WasEverUsed() const override; | 88 bool WasEverUsed() const override; |
| 87 virtual bool UsingTCPFastOpen() const override; | 89 bool UsingTCPFastOpen() const override; |
| 88 virtual bool WasNpnNegotiated() const override; | 90 bool WasNpnNegotiated() const override; |
| 89 virtual net::NextProto GetNegotiatedProtocol() const override; | 91 net::NextProto GetNegotiatedProtocol() const override; |
| 90 virtual bool GetSSLInfo(net::SSLInfo* ssl_info) override; | 92 bool GetSSLInfo(net::SSLInfo* ssl_info) override; |
| 91 | 93 |
| 92 private: | 94 private: |
| 93 void DoAsyncWrite(scoped_refptr<net::IOBuffer> buf, int buf_len, | 95 void DoAsyncWrite(scoped_refptr<net::IOBuffer> buf, int buf_len, |
| 94 const net::CompletionCallback& callback); | 96 const net::CompletionCallback& callback); |
| 95 void DoWrite(net::IOBuffer* buf, int buf_len); | 97 void DoWrite(net::IOBuffer* buf, int buf_len); |
| 96 | 98 |
| 97 bool async_write_; | 99 bool async_write_; |
| 98 bool write_pending_; | 100 bool write_pending_; |
| 99 int write_limit_; | 101 int write_limit_; |
| 100 int next_write_error_; | 102 int next_write_error_; |
| (...skipping 13 matching lines...) Expand all Loading... |
| 114 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 116 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 115 base::WeakPtrFactory<FakeStreamSocket> weak_factory_; | 117 base::WeakPtrFactory<FakeStreamSocket> weak_factory_; |
| 116 | 118 |
| 117 DISALLOW_COPY_AND_ASSIGN(FakeStreamSocket); | 119 DISALLOW_COPY_AND_ASSIGN(FakeStreamSocket); |
| 118 }; | 120 }; |
| 119 | 121 |
| 120 // StreamChannelFactory that creates FakeStreamSocket. | 122 // StreamChannelFactory that creates FakeStreamSocket. |
| 121 class FakeStreamChannelFactory : public StreamChannelFactory { | 123 class FakeStreamChannelFactory : public StreamChannelFactory { |
| 122 public: | 124 public: |
| 123 FakeStreamChannelFactory(); | 125 FakeStreamChannelFactory(); |
| 124 virtual ~FakeStreamChannelFactory(); | 126 ~FakeStreamChannelFactory() override; |
| 125 | 127 |
| 126 void set_asynchronous_create(bool asynchronous_create) { | 128 void set_asynchronous_create(bool asynchronous_create) { |
| 127 asynchronous_create_ = asynchronous_create; | 129 asynchronous_create_ = asynchronous_create; |
| 128 } | 130 } |
| 129 | 131 |
| 130 void set_fail_create(bool fail_create) { fail_create_ = fail_create; } | 132 void set_fail_create(bool fail_create) { fail_create_ = fail_create; } |
| 131 | 133 |
| 132 FakeStreamSocket* GetFakeChannel(const std::string& name); | 134 FakeStreamSocket* GetFakeChannel(const std::string& name); |
| 133 | 135 |
| 134 // ChannelFactory interface. | 136 // ChannelFactory interface. |
| 135 virtual void CreateChannel(const std::string& name, | 137 void CreateChannel(const std::string& name, |
| 136 const ChannelCreatedCallback& callback) override; | 138 const ChannelCreatedCallback& callback) override; |
| 137 virtual void CancelChannelCreation(const std::string& name) override; | 139 void CancelChannelCreation(const std::string& name) override; |
| 138 | 140 |
| 139 private: | 141 private: |
| 140 void NotifyChannelCreated(scoped_ptr<FakeStreamSocket> owned_channel, | 142 void NotifyChannelCreated(scoped_ptr<FakeStreamSocket> owned_channel, |
| 141 const std::string& name, | 143 const std::string& name, |
| 142 const ChannelCreatedCallback& callback); | 144 const ChannelCreatedCallback& callback); |
| 143 | 145 |
| 144 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 146 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 145 bool asynchronous_create_; | 147 bool asynchronous_create_; |
| 146 std::map<std::string, base::WeakPtr<FakeStreamSocket> > channels_; | 148 std::map<std::string, base::WeakPtr<FakeStreamSocket> > channels_; |
| 147 | 149 |
| 148 bool fail_create_; | 150 bool fail_create_; |
| 149 | 151 |
| 150 base::WeakPtrFactory<FakeStreamChannelFactory> weak_factory_; | 152 base::WeakPtrFactory<FakeStreamChannelFactory> weak_factory_; |
| 151 | 153 |
| 152 DISALLOW_COPY_AND_ASSIGN(FakeStreamChannelFactory); | 154 DISALLOW_COPY_AND_ASSIGN(FakeStreamChannelFactory); |
| 153 }; | 155 }; |
| 154 | 156 |
| 155 } // namespace protocol | 157 } // namespace protocol |
| 156 } // namespace remoting | 158 } // namespace remoting |
| 157 | 159 |
| 158 #endif // REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ | 160 #endif // REMOTING_PROTOCOL_FAKE_STREAM_SOCKET_H_ |
| OLD | NEW |