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_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ | 5 #ifndef CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ |
6 #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ | 6 #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/scoped_ptr.h" |
13 #include "base/message_loop.h" | 13 #include "base/message_loop.h" |
14 #include "content/browser/renderer_host/p2p/socket_host.h" | 14 #include "content/browser/renderer_host/p2p/socket_host.h" |
15 #include "content/common/content_export.h" | 15 #include "content/common/content_export.h" |
16 #include "content/common/p2p_sockets.h" | 16 #include "content/common/p2p_sockets.h" |
17 #include "ipc/ipc_sender.h" | 17 #include "ipc/ipc_sender.h" |
18 #include "net/base/completion_callback.h" | 18 #include "net/base/completion_callback.h" |
19 #include "net/socket/tcp_server_socket.h" | 19 #include "net/socket/tcp_server_socket.h" |
20 | 20 |
21 namespace net { | 21 namespace net { |
22 class StreamSocket; | 22 class StreamSocket; |
| 23 class URLRequestContextGetter; |
23 } // namespace net | 24 } // namespace net |
24 | 25 |
25 namespace content { | 26 namespace content { |
26 | 27 |
27 class CONTENT_EXPORT P2PSocketHostTcpServer : public P2PSocketHost { | 28 class CONTENT_EXPORT P2PSocketHostTcpServer : public P2PSocketHost { |
28 public: | 29 public: |
29 typedef std::map<net::IPEndPoint, net::StreamSocket*> AcceptedSocketsMap; | 30 typedef std::map<net::IPEndPoint, net::StreamSocket*> AcceptedSocketsMap; |
30 | 31 |
31 P2PSocketHostTcpServer(IPC::Sender* message_sender, int id, | 32 P2PSocketHostTcpServer(IPC::Sender* message_sender, int id, |
32 P2PSocketType client_type); | 33 P2PSocketType client_type, |
| 34 net::URLRequestContextGetter* getter); |
33 virtual ~P2PSocketHostTcpServer(); | 35 virtual ~P2PSocketHostTcpServer(); |
34 | 36 |
35 // P2PSocketHost overrides. | 37 // P2PSocketHost overrides. |
36 virtual bool Init(const net::IPEndPoint& local_address, | 38 virtual bool Init(const net::IPEndPoint& local_address, |
37 const net::IPEndPoint& remote_address) OVERRIDE; | 39 const net::IPEndPoint& remote_address) OVERRIDE; |
38 virtual void Send(const net::IPEndPoint& to, | 40 virtual void Send(const net::IPEndPoint& to, |
39 const std::vector<char>& data) OVERRIDE; | 41 const std::vector<char>& data) OVERRIDE; |
40 virtual P2PSocketHost* AcceptIncomingTcpConnection( | 42 virtual P2PSocketHost* AcceptIncomingTcpConnection( |
41 const net::IPEndPoint& remote_address, int id) OVERRIDE; | 43 const net::IPEndPoint& remote_address, int id) OVERRIDE; |
42 | 44 |
43 private: | 45 private: |
44 friend class P2PSocketHostTcpServerTest; | 46 friend class P2PSocketHostTcpServerTest; |
45 | 47 |
46 void OnError(); | 48 void OnError(); |
47 | 49 |
48 void DoAccept(); | 50 void DoAccept(); |
49 void HandleAcceptResult(int result); | 51 void HandleAcceptResult(int result); |
50 | 52 |
51 // Callback for Accept(). | 53 // Callback for Accept(). |
52 void OnAccepted(int result); | 54 void OnAccepted(int result); |
53 | 55 |
54 const P2PSocketType client_type_; | 56 const P2PSocketType client_type_; |
| 57 scoped_refptr<net::URLRequestContextGetter> context_getter_; |
55 scoped_ptr<net::ServerSocket> socket_; | 58 scoped_ptr<net::ServerSocket> socket_; |
56 net::IPEndPoint local_address_; | 59 net::IPEndPoint local_address_; |
57 | 60 |
58 scoped_ptr<net::StreamSocket> accept_socket_; | 61 scoped_ptr<net::StreamSocket> accept_socket_; |
59 AcceptedSocketsMap accepted_sockets_; | 62 AcceptedSocketsMap accepted_sockets_; |
60 | 63 |
61 net::CompletionCallback accept_callback_; | 64 net::CompletionCallback accept_callback_; |
62 | 65 |
63 DISALLOW_COPY_AND_ASSIGN(P2PSocketHostTcpServer); | 66 DISALLOW_COPY_AND_ASSIGN(P2PSocketHostTcpServer); |
64 }; | 67 }; |
65 | 68 |
66 } // namespace content | 69 } // namespace content |
67 | 70 |
68 #endif // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ | 71 #endif // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_TCP_SERVER_H_ |
OLD | NEW |