src/content/renderer/p2p/socket_client_impl.h

   1 // Copyright 2013 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_RENDERER_P2P_SOCKET_CLIENT_IMPL_H_
   6 #define CONTENT_RENDERER_P2P_SOCKET_CLIENT_IMPL_H_
   7
   8 #include <vector>
   9
  10 #include "base/memory/ref_counted.h"
  11 #include "content/common/p2p_socket_type.h"
  12 #include "content/renderer/p2p/socket_client.h"
  13 #include "net/base/ip_endpoint.h"
  14
  15 namespace base {
  16 class MessageLoopProxy;
  17 class TimeTicks;
  18 }  // namespace base
  19
  20 namespace content {
  21
  22 class P2PSocketDispatcher;
  23
  24 // P2P socket that routes all calls over IPC.
  25 //
  26 // The object runs on two threads: IPC thread and delegate thread. The
  27 // IPC thread is used to interact with P2PSocketDispatcher. All
  28 // callbacks to the user of this class are called on the delegate
  29 // thread which is specified in Init().
  30 class P2PSocketClientImpl : public P2PSocketClient {
  31  public:
  32   explicit P2PSocketClientImpl(P2PSocketDispatcher* dispatcher);
  33
  34   // Initialize socket of the specified |type| and connected to the
  35   // specified |address|. |address| matters only when |type| is set to
  36   // P2P_SOCKET_TCP_CLIENT.
  37   virtual void Init(P2PSocketType type,
  38                     const net::IPEndPoint& local_address,
  39                     const P2PHostAndIPEndPoint& remote_address,
  40                     P2PSocketClientDelegate* delegate);
  41
  42   // Send the |data| to the |address|.
  43   virtual void Send(const net::IPEndPoint& address,
  44                     const std::vector<char>& data) OVERRIDE;
  45
  46   // Send the |data| to the |address| using Differentiated Services Code Point
  47   // |dscp|.
  48   virtual void SendWithDscp(const net::IPEndPoint& address,
  49                             const std::vector<char>& data,
  50                             const talk_base::PacketOptions& options) OVERRIDE;
  51
  52   // Setting socket options.
  53   virtual void SetOption(P2PSocketOption option, int value) OVERRIDE;
  54
  55   // Must be called before the socket is destroyed. The delegate may
  56   // not be called after |closed_task| is executed.
  57   virtual void Close() OVERRIDE;
  58
  59   virtual int GetSocketID() const OVERRIDE;
  60
  61   virtual void SetDelegate(P2PSocketClientDelegate* delegate) OVERRIDE;
  62
  63  private:
  64   enum State {
  65     STATE_UNINITIALIZED,
  66     STATE_OPENING,
  67     STATE_OPEN,
  68     STATE_CLOSED,
  69     STATE_ERROR,
  70   };
  71
  72   friend class P2PSocketDispatcher;
  73
  74   virtual ~P2PSocketClientImpl();
  75
  76   // Message handlers that run on IPC thread.
  77   void OnSocketCreated(const net::IPEndPoint& address);
  78   void OnIncomingTcpConnection(const net::IPEndPoint& address);
  79   void OnSendComplete(int packet_id);
  80   void OnSendComplete();
  81   void OnError();
  82   void OnDataReceived(const net::IPEndPoint& address,
  83                       const std::vector<char>& data,
  84                       const base::TimeTicks& timestamp);
  85
  86   // Proxy methods that deliver messages to the delegate thread.
  87   void DeliverOnSocketCreated(const net::IPEndPoint& address);
  88   void DeliverOnIncomingTcpConnection(
  89       const net::IPEndPoint& address,
  90       scoped_refptr<P2PSocketClient> new_client);
  91   void DeliverOnSendComplete();
  92   void DeliverOnError();
  93   void DeliverOnDataReceived(const net::IPEndPoint& address,
  94                              const std::vector<char>& data,
  95                              const base::TimeTicks& timestamp);
  96
  97   // Scheduled on the IPC thread to finish initialization.
  98   void DoInit(P2PSocketType type,
  99               const net::IPEndPoint& local_address,
 100               const P2PHostAndIPEndPoint& remote_address);
 101
 102   // Scheduled on the IPC thread to finish closing the connection.
 103   void DoClose();
 104
 105   // Called by the dispatcher when it is destroyed.
 106   void Detach();
 107
 108   P2PSocketDispatcher* dispatcher_;
 109   scoped_refptr<base::MessageLoopProxy> ipc_message_loop_;
 110   scoped_refptr<base::MessageLoopProxy> delegate_message_loop_;
 111   int socket_id_;
 112   P2PSocketClientDelegate* delegate_;
 113   State state_;
 114
 115   // These two fields are used to identify packets for tracing.
 116   uint32 random_socket_id_;
 117   uint32 next_packet_id_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(P2PSocketClientImpl);
 120 };
 121
 122 }  // namespace content
 123
 124 #endif  // CONTENT_RENDERER_P2P_SOCKET_CLIENT_IMPL_H_