| 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_H_ | 5 #ifndef CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_ |
| 6 #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_ | 6 #define CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_ |
| 7 | 7 |
| 8 #include "content/common/content_export.h" | 8 #include "content/common/content_export.h" |
| 9 #include "content/common/p2p_socket_type.h" | 9 #include "content/common/p2p_socket_type.h" |
| 10 #include "content/public/browser/render_process_host.h" |
| 10 #include "net/base/ip_endpoint.h" | 11 #include "net/base/ip_endpoint.h" |
| 11 #include "net/udp/datagram_socket.h" | 12 #include "net/udp/datagram_socket.h" |
| 12 | 13 |
| 13 namespace IPC { | 14 namespace IPC { |
| 14 class Sender; | 15 class Sender; |
| 15 } | 16 } |
| 16 | 17 |
| 17 namespace net { | 18 namespace net { |
| 18 class URLRequestContextGetter; | 19 class URLRequestContextGetter; |
| 19 } | 20 } |
| (...skipping 10 matching lines...) Expand all Loading... |
| 30 // This method can handle only RTP packet, otherwise this method must not be | 31 // This method can handle only RTP packet, otherwise this method must not be |
| 31 // called. It will try to do, 1. update absolute send time extension header | 32 // called. It will try to do, 1. update absolute send time extension header |
| 32 // if present with current time and 2. update HMAC in RTP packet. | 33 // if present with current time and 2. update HMAC in RTP packet. |
| 33 // If abs_send_time is 0, ApplyPacketOption will get current time from system. | 34 // If abs_send_time is 0, ApplyPacketOption will get current time from system. |
| 34 CONTENT_EXPORT bool ApplyPacketOptions(char* data, int length, | 35 CONTENT_EXPORT bool ApplyPacketOptions(char* data, int length, |
| 35 const talk_base::PacketOptions& options, | 36 const talk_base::PacketOptions& options, |
| 36 uint32 abs_send_time); | 37 uint32 abs_send_time); |
| 37 | 38 |
| 38 // Helper method which finds RTP ofset and length if the packet is encapsulated | 39 // Helper method which finds RTP ofset and length if the packet is encapsulated |
| 39 // in a TURN Channel Message or TURN Send Indication message. | 40 // in a TURN Channel Message or TURN Send Indication message. |
| 40 CONTENT_EXPORT bool GetRtpPacketStartPositionAndLength(char* data, int length, | 41 CONTENT_EXPORT bool GetRtpPacketStartPositionAndLength(const char* data, |
| 42 int length, |
| 41 int* rtp_start_pos, | 43 int* rtp_start_pos, |
| 42 int* rtp_packet_length); | 44 int* rtp_packet_length); |
| 43 // Helper method which updates absoulute send time extension if present. | 45 // Helper method which updates absoulute send time extension if present. |
| 44 CONTENT_EXPORT bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, | 46 CONTENT_EXPORT bool UpdateRtpAbsSendTimeExtn(char* rtp, int length, |
| 45 int extension_id, | 47 int extension_id, |
| 46 uint32 abs_send_time); | 48 uint32 abs_send_time); |
| 47 | 49 |
| 48 } // packet_processing_helpers | 50 } // packet_processing_helpers |
| 49 | 51 |
| 50 // Base class for P2P sockets. | 52 // Base class for P2P sockets. |
| 51 class CONTENT_EXPORT P2PSocketHost { | 53 class CONTENT_EXPORT P2PSocketHost { |
| 52 public: | 54 public: |
| 53 static const int kStunHeaderSize = 20; | 55 static const int kStunHeaderSize = 20; |
| 54 // Creates P2PSocketHost of the specific type. | 56 // Creates P2PSocketHost of the specific type. |
| 55 static P2PSocketHost* Create(IPC::Sender* message_sender, | 57 static P2PSocketHost* Create(IPC::Sender* message_sender, |
| 56 int id, P2PSocketType type, | 58 int socket_id, |
| 59 P2PSocketType type, |
| 57 net::URLRequestContextGetter* url_context, | 60 net::URLRequestContextGetter* url_context, |
| 58 P2PMessageThrottler* throttler); | 61 P2PMessageThrottler* throttler); |
| 59 | 62 |
| 60 virtual ~P2PSocketHost(); | 63 virtual ~P2PSocketHost(); |
| 61 | 64 |
| 62 // Initalizes the socket. Returns false when initiazations fails. | 65 // Initalizes the socket. Returns false when initiazations fails. |
| 63 virtual bool Init(const net::IPEndPoint& local_address, | 66 virtual bool Init(const net::IPEndPoint& local_address, |
| 64 const P2PHostAndIPEndPoint& remote_address) = 0; | 67 const P2PHostAndIPEndPoint& remote_address) = 0; |
| 65 | 68 |
| 66 // Sends |data| on the socket to |to|. | 69 // Sends |data| on the socket to |to|. |
| 67 virtual void Send(const net::IPEndPoint& to, | 70 virtual void Send(const net::IPEndPoint& to, |
| 68 const std::vector<char>& data, | 71 const std::vector<char>& data, |
| 69 const talk_base::PacketOptions& options, | 72 const talk_base::PacketOptions& options, |
| 70 uint64 packet_id) = 0; | 73 uint64 packet_id) = 0; |
| 71 | 74 |
| 72 virtual P2PSocketHost* AcceptIncomingTcpConnection( | 75 virtual P2PSocketHost* AcceptIncomingTcpConnection( |
| 73 const net::IPEndPoint& remote_address, int id) = 0; | 76 const net::IPEndPoint& remote_address, int id) = 0; |
| 74 | 77 |
| 75 virtual bool SetOption(P2PSocketOption option, int value) = 0; | 78 virtual bool SetOption(P2PSocketOption option, int value) = 0; |
| 76 | 79 |
| 80 void StartRtpDump( |
| 81 bool incoming, |
| 82 bool outgoing, |
| 83 const RenderProcessHost::WebRtcRtpPacketCallback& packet_callback); |
| 84 void StopRtpDump(bool incoming, bool outgoing); |
| 85 |
| 77 protected: | 86 protected: |
| 78 friend class P2PSocketHostTcpTestBase; | 87 friend class P2PSocketHostTcpTestBase; |
| 79 | 88 |
| 80 // TODO(mallinath) - Remove this below enum and use one defined in | 89 // TODO(mallinath) - Remove this below enum and use one defined in |
| 81 // libjingle/souce/talk/p2p/base/stun.h | 90 // libjingle/souce/talk/p2p/base/stun.h |
| 82 enum StunMessageType { | 91 enum StunMessageType { |
| 83 STUN_BINDING_REQUEST = 0x0001, | 92 STUN_BINDING_REQUEST = 0x0001, |
| 84 STUN_BINDING_RESPONSE = 0x0101, | 93 STUN_BINDING_RESPONSE = 0x0101, |
| 85 STUN_BINDING_ERROR_RESPONSE = 0x0111, | 94 STUN_BINDING_ERROR_RESPONSE = 0x0111, |
| 86 STUN_SHARED_SECRET_REQUEST = 0x0002, | 95 STUN_SHARED_SECRET_REQUEST = 0x0002, |
| (...skipping 17 matching lines...) Expand all Loading... |
| 104 }; | 113 }; |
| 105 | 114 |
| 106 enum State { | 115 enum State { |
| 107 STATE_UNINITIALIZED, | 116 STATE_UNINITIALIZED, |
| 108 STATE_CONNECTING, | 117 STATE_CONNECTING, |
| 109 STATE_TLS_CONNECTING, | 118 STATE_TLS_CONNECTING, |
| 110 STATE_OPEN, | 119 STATE_OPEN, |
| 111 STATE_ERROR, | 120 STATE_ERROR, |
| 112 }; | 121 }; |
| 113 | 122 |
| 114 P2PSocketHost(IPC::Sender* message_sender, int id); | 123 P2PSocketHost(IPC::Sender* message_sender, int socket_id); |
| 115 | 124 |
| 116 // Verifies that the packet |data| has a valid STUN header. In case | 125 // Verifies that the packet |data| has a valid STUN header. In case |
| 117 // of success stores type of the message in |type|. | 126 // of success stores type of the message in |type|. |
| 118 static bool GetStunPacketType(const char* data, int data_size, | 127 static bool GetStunPacketType(const char* data, int data_size, |
| 119 StunMessageType* type); | 128 StunMessageType* type); |
| 120 static bool IsRequestOrResponse(StunMessageType type); | 129 static bool IsRequestOrResponse(StunMessageType type); |
| 121 | 130 |
| 131 // Calls RenderProcessHostImpl::OnRtpPacket to record the RTP header. |
| 132 void DumpRtpPacket(const char* packet, size_t length, bool incoming); |
| 133 |
| 122 IPC::Sender* message_sender_; | 134 IPC::Sender* message_sender_; |
| 123 int id_; | 135 int id_; |
| 124 State state_; | 136 State state_; |
| 137 bool dump_incoming_rtp_packet_; |
| 138 bool dump_outgoing_rtp_packet_; |
| 139 RenderProcessHost::WebRtcRtpPacketCallback packet_dump_callback_; |
| 125 | 140 |
| 126 DISALLOW_COPY_AND_ASSIGN(P2PSocketHost); | 141 DISALLOW_COPY_AND_ASSIGN(P2PSocketHost); |
| 127 }; | 142 }; |
| 128 | 143 |
| 129 } // namespace content | 144 } // namespace content |
| 130 | 145 |
| 131 #endif // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_ | 146 #endif // CONTENT_BROWSER_RENDERER_HOST_P2P_SOCKET_HOST_H_ |
| OLD | NEW |