| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ | 5 #ifndef CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ |
| 6 #define CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ | 6 #define CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ |
| 7 | 7 |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| 11 #include "content/common/p2p_sockets.h" | 11 #include "content/common/p2p_sockets.h" |
| 12 #include "net/base/ip_endpoint.h" | 12 #include "net/base/ip_endpoint.h" |
| 13 | 13 |
| 14 namespace base { | 14 namespace base { |
| 15 class MessageLoopProxy; | 15 class MessageLoopProxy; |
| 16 } // namespace base | 16 } // namespace base |
| 17 | 17 |
| 18 namespace content { | 18 namespace content { |
| 19 | 19 |
| 20 class P2PSocketDispatcher; | 20 class P2PSocketDispatcher; |
| 21 | 21 |
| 22 // P2P socket that rountes all calls over IPC. | 22 // P2P socket that rountes all calls over IPC. |
| 23 // | 23 // |
| 24 // The object runs on two threads: IPC thread and delegate thread. The | 24 // The object runs on two threads: IPC thread and delegate thread. The |
| 25 // IPC thread is used to interact with P2PSocketDispatcher. All | 25 // IPC thread is used to interact with P2PSocketDispatcher. All |
| 26 // callbacks to the user of this class are called on the delegate | 26 // callbacks to the user of this class are called on the delegate |
| 27 // thread which is specified in Init(). | 27 // thread which is specified in Init(). |
| 28 class P2PSocketClient : public base::RefCountedThreadSafe<P2PSocketClient> { | 28 class P2PSocketClient : public base::RefCountedThreadSafe<P2PSocketClient> { |
| 29 public: | 29 public: |
| 30 // Delegate is called on the the same thread on the delegate thread. | 30 // Delegate is called on the the same thread on which P2PSocketCLient is |
| 31 // created. |
| 31 class Delegate { | 32 class Delegate { |
| 32 public: | 33 public: |
| 33 virtual ~Delegate() { } | 34 virtual ~Delegate() { } |
| 34 | 35 |
| 35 virtual void OnOpen(const net::IPEndPoint& address) = 0; | 36 virtual void OnOpen(const net::IPEndPoint& address) = 0; |
| 36 virtual void OnIncomingTcpConnection(const net::IPEndPoint& address, | 37 virtual void OnIncomingTcpConnection(const net::IPEndPoint& address, |
| 37 P2PSocketClient* client) = 0; | 38 P2PSocketClient* client) = 0; |
| 39 virtual void OnSendComplete() = 0; |
| 38 virtual void OnError() = 0; | 40 virtual void OnError() = 0; |
| 39 virtual void OnDataReceived(const net::IPEndPoint& address, | 41 virtual void OnDataReceived(const net::IPEndPoint& address, |
| 40 const std::vector<char>& data) = 0; | 42 const std::vector<char>& data) = 0; |
| 41 }; | 43 }; |
| 42 | 44 |
| 43 explicit P2PSocketClient(P2PSocketDispatcher* dispatcher); | 45 explicit P2PSocketClient(P2PSocketDispatcher* dispatcher); |
| 44 | 46 |
| 45 // Initialize socket of the specified |type| and connected to the | 47 // Initialize socket of the specified |type| and connected to the |
| 46 // specified |address|. |address| matters only when |type| is set to | 48 // specified |address|. |address| matters only when |type| is set to |
| 47 // P2P_SOCKET_TCP_CLIENT. | 49 // P2P_SOCKET_TCP_CLIENT. |
| (...skipping 25 matching lines...) Expand all Loading... |
| 73 friend class P2PSocketDispatcher; | 75 friend class P2PSocketDispatcher; |
| 74 | 76 |
| 75 // Calls destructor. | 77 // Calls destructor. |
| 76 friend class base::RefCountedThreadSafe<P2PSocketClient>; | 78 friend class base::RefCountedThreadSafe<P2PSocketClient>; |
| 77 | 79 |
| 78 virtual ~P2PSocketClient(); | 80 virtual ~P2PSocketClient(); |
| 79 | 81 |
| 80 // Message handlers that run on IPC thread. | 82 // Message handlers that run on IPC thread. |
| 81 void OnSocketCreated(const net::IPEndPoint& address); | 83 void OnSocketCreated(const net::IPEndPoint& address); |
| 82 void OnIncomingTcpConnection(const net::IPEndPoint& address); | 84 void OnIncomingTcpConnection(const net::IPEndPoint& address); |
| 85 void OnSendComplete(int packet_id); |
| 86 void OnSendComplete(); |
| 83 void OnError(); | 87 void OnError(); |
| 84 void OnDataReceived(const net::IPEndPoint& address, | 88 void OnDataReceived(const net::IPEndPoint& address, |
| 85 const std::vector<char>& data); | 89 const std::vector<char>& data); |
| 86 | 90 |
| 87 // Proxy methods that deliver messages to the delegate thread. | 91 // Proxy methods that deliver messages to the delegate thread. |
| 88 void DeliverOnSocketCreated(const net::IPEndPoint& address); | 92 void DeliverOnSocketCreated(const net::IPEndPoint& address); |
| 89 void DeliverOnIncomingTcpConnection( | 93 void DeliverOnIncomingTcpConnection( |
| 90 const net::IPEndPoint& address, | 94 const net::IPEndPoint& address, |
| 91 scoped_refptr<P2PSocketClient> new_client); | 95 scoped_refptr<P2PSocketClient> new_client); |
| 96 void DeliverOnSendComplete(); |
| 92 void DeliverOnError(); | 97 void DeliverOnError(); |
| 93 void DeliverOnDataReceived(const net::IPEndPoint& address, | 98 void DeliverOnDataReceived(const net::IPEndPoint& address, |
| 94 const std::vector<char>& data); | 99 const std::vector<char>& data); |
| 95 | 100 |
| 96 // Scheduled on the IPC thread to finish initialization. | 101 // Scheduled on the IPC thread to finish initialization. |
| 97 void DoInit(P2PSocketType type, | 102 void DoInit(P2PSocketType type, |
| 98 const net::IPEndPoint& local_address, | 103 const net::IPEndPoint& local_address, |
| 99 const net::IPEndPoint& remote_address); | 104 const net::IPEndPoint& remote_address); |
| 100 | 105 |
| 101 // Scheduled on the IPC thread to finish closing the connection. | 106 // Scheduled on the IPC thread to finish closing the connection. |
| 102 void DoClose(); | 107 void DoClose(); |
| 103 | 108 |
| 104 // Called by the dispatcher when it is destroyed. | 109 // Called by the dispatcher when it is destroyed. |
| 105 void Detach(); | 110 void Detach(); |
| 106 | 111 |
| 107 P2PSocketDispatcher* dispatcher_; | 112 P2PSocketDispatcher* dispatcher_; |
| 108 scoped_refptr<base::MessageLoopProxy> ipc_message_loop_; | 113 scoped_refptr<base::MessageLoopProxy> ipc_message_loop_; |
| 109 scoped_refptr<base::MessageLoopProxy> delegate_message_loop_; | 114 scoped_refptr<base::MessageLoopProxy> delegate_message_loop_; |
| 110 int socket_id_; | 115 int socket_id_; |
| 111 Delegate* delegate_; | 116 Delegate* delegate_; |
| 112 State state_; | 117 State state_; |
| 113 | 118 |
| 114 DISALLOW_COPY_AND_ASSIGN(P2PSocketClient); | 119 DISALLOW_COPY_AND_ASSIGN(P2PSocketClient); |
| 115 }; | 120 }; |
| 116 | 121 |
| 117 } // namespace content | 122 } // namespace content |
| 118 | 123 |
| 119 #endif // CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ | 124 #endif // CONTENT_RENDERER_P2P_SOCKET_CLIENT_H_ |
| OLD | NEW |