| // Copyright (c) 2011 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 CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_H_ |
| #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_H_ |
| |
| #include <queue> |
| #include <vector> |
| |
| #include "base/compiler_specific.h" |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/message_loop/message_loop.h" |
| #include "content/browser/renderer_host/p2p/socket_host.h" |
| #include "content/common/p2p_socket_type.h" |
| #include "net/base/completion_callback.h" |
| #include "net/base/ip_endpoint.h" |
| |
| namespace net { |
| class DrainableIOBuffer; |
| class GrowableIOBuffer; |
| class StreamSocket; |
| class URLRequestContextGetter; |
| } // namespace net |
| |
| namespace content { |
| |
| class CONTENT_EXPORT P2PSocketHostTcpBase : public P2PSocketHost { |
| public: |
| P2PSocketHostTcpBase(IPC::Sender* message_sender, |
| int socket_id, |
| P2PSocketType type, |
| net::URLRequestContextGetter* url_context); |
| ~P2PSocketHostTcpBase() override; |
| |
| bool InitAccepted(const net::IPEndPoint& remote_address, |
| net::StreamSocket* socket); |
| |
| // P2PSocketHost overrides. |
| bool Init(const net::IPEndPoint& local_address, |
| const P2PHostAndIPEndPoint& remote_address) override; |
| void Send(const net::IPEndPoint& to, |
| const std::vector<char>& data, |
| const rtc::PacketOptions& options, |
| uint64 packet_id) override; |
| P2PSocketHost* AcceptIncomingTcpConnection( |
| const net::IPEndPoint& remote_address, |
| int id) override; |
| bool SetOption(P2PSocketOption option, int value) override; |
| |
| protected: |
| // Derived classes will provide the implementation. |
| virtual int ProcessInput(char* input, int input_len) = 0; |
| virtual void DoSend(const net::IPEndPoint& to, |
| const std::vector<char>& data, |
| const rtc::PacketOptions& options) = 0; |
| |
| void WriteOrQueue(scoped_refptr<net::DrainableIOBuffer>& buffer); |
| void OnPacket(const std::vector<char>& data); |
| void OnError(); |
| |
| private: |
| friend class P2PSocketHostTcpTestBase; |
| friend class P2PSocketHostTcpServerTest; |
| |
| // SSL/TLS connection functions. |
| void StartTls(); |
| void ProcessTlsSslConnectDone(int status); |
| |
| void DidCompleteRead(int result); |
| void DoRead(); |
| |
| void DoWrite(); |
| void HandleWriteResult(int result); |
| |
| // Callbacks for Connect(), Read() and Write(). |
| void OnConnected(int result); |
| void OnRead(int result); |
| void OnWritten(int result); |
| |
| // Helper method to send socket create message and start read. |
| void OnOpen(); |
| void DoSendSocketCreateMsg(); |
| |
| P2PHostAndIPEndPoint remote_address_; |
| |
| scoped_ptr<net::StreamSocket> socket_; |
| scoped_refptr<net::GrowableIOBuffer> read_buffer_; |
| std::queue<scoped_refptr<net::DrainableIOBuffer> > write_queue_; |
| scoped_refptr<net::DrainableIOBuffer> write_buffer_; |
| |
| bool write_pending_; |
| |
| bool connected_; |
| P2PSocketType type_; |
| scoped_refptr<net::URLRequestContextGetter> url_context_; |
| |
| DISALLOW_COPY_AND_ASSIGN(P2PSocketHostTcpBase); |
| }; |
| |
| class CONTENT_EXPORT P2PSocketHostTcp : public P2PSocketHostTcpBase { |
| public: |
| P2PSocketHostTcp(IPC::Sender* message_sender, |
| int socket_id, |
| P2PSocketType type, |
| net::URLRequestContextGetter* url_context); |
| ~P2PSocketHostTcp() override; |
| |
| protected: |
| int ProcessInput(char* input, int input_len) override; |
| void DoSend(const net::IPEndPoint& to, |
| const std::vector<char>& data, |
| const rtc::PacketOptions& options) override; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(P2PSocketHostTcp); |
| }; |
| |
| // P2PSocketHostStunTcp class provides the framing of STUN messages when used |
| // with TURN. These messages will not have length at front of the packet and |
| // are padded to multiple of 4 bytes. |
| // Formatting of messages is defined in RFC5766. |
| class CONTENT_EXPORT P2PSocketHostStunTcp : public P2PSocketHostTcpBase { |
| public: |
| P2PSocketHostStunTcp(IPC::Sender* message_sender, |
| int socket_id, |
| P2PSocketType type, |
| net::URLRequestContextGetter* url_context); |
| |
| ~P2PSocketHostStunTcp() override; |
| |
| protected: |
| int ProcessInput(char* input, int input_len) override; |
| void DoSend(const net::IPEndPoint& to, |
| const std::vector<char>& data, |
| const rtc::PacketOptions& options) override; |
| |
| private: |
| int GetExpectedPacketSize(const char* data, int len, int* pad_bytes); |
| |
| DISALLOW_COPY_AND_ASSIGN(P2PSocketHostStunTcp); |
| }; |
| |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_H_ |