src/content/browser/renderer_host/p2p/socket_host.h

   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_H_
   6 #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_
   7
   8 #include "base/memory/weak_ptr.h"
   9 #include "content/common/content_export.h"
  10 #include "content/common/p2p_socket_type.h"
  11 #include "content/public/browser/render_process_host.h"
  12 #include "net/base/ip_endpoint.h"
  13 #include "net/udp/datagram_socket.h"
  14
  15 namespace IPC {
  16 class Sender;
  17 }
  18
  19 namespace net {
  20 class URLRequestContextGetter;
  21 }
  22
  23 namespace rtc {
  24 struct PacketOptions;
  25 }
  26
  27 namespace content {
  28 class P2PMessageThrottler;
  29
  30 namespace packet_processing_helpers {
  31
  32 // This method can handle only RTP packet, otherwise this method must not be
  33 // called. It will try to do, 1. update absolute send time extension header
  34 // if present with current time and 2. update HMAC in RTP packet.
  35 // If abs_send_time is 0, ApplyPacketOption will get current time from system.
  36 CONTENT_EXPORT bool ApplyPacketOptions(char* data,
  37                                        size_t length,
  38                                        const rtc::PacketOptions& options,
  39                                        uint32 abs_send_time);
  40
  41 // Helper method which finds RTP ofset and length if the packet is encapsulated
  42 // in a TURN Channel Message or TURN Send Indication message.
  43 CONTENT_EXPORT bool GetRtpPacketStartPositionAndLength(
  44     const char* data,
  45     size_t length,
  46     size_t* rtp_start_pos,
  47     size_t* rtp_packet_length);
  48
  49 // Helper method which updates absoulute send time extension if present.
  50 CONTENT_EXPORT bool UpdateRtpAbsSendTimeExtension(char* rtp,
  51                                                   size_t length,
  52                                                   int extension_id,
  53                                                   uint32 abs_send_time);
  54
  55 }  // packet_processing_helpers
  56
  57 // Base class for P2P sockets.
  58 class CONTENT_EXPORT P2PSocketHost {
  59  public:
  60   static const int kStunHeaderSize = 20;
  61   // Creates P2PSocketHost of the specific type.
  62   static P2PSocketHost* Create(IPC::Sender* message_sender,
  63                                int socket_id,
  64                                P2PSocketType type,
  65                                net::URLRequestContextGetter* url_context,
  66                                P2PMessageThrottler* throttler);
  67
  68   virtual ~P2PSocketHost();
  69
  70   // Initalizes the socket. Returns false when initiazations fails.
  71   virtual bool Init(const net::IPEndPoint& local_address,
  72                     const P2PHostAndIPEndPoint& remote_address) = 0;
  73
  74   // Sends |data| on the socket to |to|.
  75   virtual void Send(const net::IPEndPoint& to,
  76                     const std::vector<char>& data,
  77                     const rtc::PacketOptions& options,
  78                     uint64 packet_id) = 0;
  79
  80   virtual P2PSocketHost* AcceptIncomingTcpConnection(
  81       const net::IPEndPoint& remote_address, int id) = 0;
  82
  83   virtual bool SetOption(P2PSocketOption option, int value) = 0;
  84
  85   void StartRtpDump(
  86       bool incoming,
  87       bool outgoing,
  88       const RenderProcessHost::WebRtcRtpPacketCallback& packet_callback);
  89   void StopRtpDump(bool incoming, bool outgoing);
  90
  91  protected:
  92   friend class P2PSocketHostTcpTestBase;
  93
  94   // TODO(mallinath) - Remove this below enum and use one defined in
  95   // libjingle/souce/talk/p2p/base/stun.h
  96   enum StunMessageType {
  97     STUN_BINDING_REQUEST = 0x0001,
  98     STUN_BINDING_RESPONSE = 0x0101,
  99     STUN_BINDING_ERROR_RESPONSE = 0x0111,
 100     STUN_SHARED_SECRET_REQUEST = 0x0002,
 101     STUN_SHARED_SECRET_RESPONSE = 0x0102,
 102     STUN_SHARED_SECRET_ERROR_RESPONSE = 0x0112,
 103     STUN_ALLOCATE_REQUEST = 0x0003,
 104     STUN_ALLOCATE_RESPONSE = 0x0103,
 105     STUN_ALLOCATE_ERROR_RESPONSE = 0x0113,
 106     STUN_SEND_REQUEST = 0x0004,
 107     STUN_SEND_RESPONSE = 0x0104,
 108     STUN_SEND_ERROR_RESPONSE = 0x0114,
 109     STUN_DATA_INDICATION = 0x0115,
 110     TURN_SEND_INDICATION = 0x0016,
 111     TURN_DATA_INDICATION = 0x0017,
 112     TURN_CREATE_PERMISSION_REQUEST = 0x0008,
 113     TURN_CREATE_PERMISSION_RESPONSE = 0x0108,
 114     TURN_CREATE_PERMISSION_ERROR_RESPONSE = 0x0118,
 115     TURN_CHANNEL_BIND_REQUEST = 0x0009,
 116     TURN_CHANNEL_BIND_RESPONSE = 0x0109,
 117     TURN_CHANNEL_BIND_ERROR_RESPONSE = 0x0119,
 118   };
 119
 120   enum State {
 121     STATE_UNINITIALIZED,
 122     STATE_CONNECTING,
 123     STATE_TLS_CONNECTING,
 124     STATE_OPEN,
 125     STATE_ERROR,
 126   };
 127
 128   P2PSocketHost(IPC::Sender* message_sender, int socket_id);
 129
 130   // Verifies that the packet |data| has a valid STUN header. In case
 131   // of success stores type of the message in |type|.
 132   static bool GetStunPacketType(const char* data, int data_size,
 133                                 StunMessageType* type);
 134   static bool IsRequestOrResponse(StunMessageType type);
 135
 136   // Calls |packet_dump_callback_| to record the RTP header.
 137   void DumpRtpPacket(const char* packet, size_t length, bool incoming);
 138
 139   // A helper to dump the packet on the IO thread.
 140   void DumpRtpPacketOnIOThread(scoped_ptr<uint8[]> packet_header,
 141                                size_t header_length,
 142                                size_t packet_length,
 143                                bool incoming);
 144
 145   IPC::Sender* message_sender_;
 146   int id_;
 147   State state_;
 148   bool dump_incoming_rtp_packet_;
 149   bool dump_outgoing_rtp_packet_;
 150   RenderProcessHost::WebRtcRtpPacketCallback packet_dump_callback_;
 151
 152   base::WeakPtrFactory<P2PSocketHost> weak_ptr_factory_;
 153
 154   DISALLOW_COPY_AND_ASSIGN(P2PSocketHost);
 155 };
 156
 157 }  // namespace content
 158
 159 #endif  // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_