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