1 // Copyright (c) 2012 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.
5 #include "content/browser/renderer_host/p2p/socket_host.h"
7 #include "base/sys_byteorder.h"
8 #include "content/browser/renderer_host/p2p/socket_host_tcp.h"
9 #include "content/browser/renderer_host/p2p/socket_host_tcp_server.h"
10 #include "content/browser/renderer_host/p2p/socket_host_udp.h"
11 #include "content/browser/renderer_host/render_process_host_impl.h"
12 #include "content/public/browser/browser_thread.h"
13 #include "crypto/hmac.h"
14 #include "third_party/libjingle/source/talk/p2p/base/stun.h"
15 #include "third_party/webrtc/base/asyncpacketsocket.h"
16 #include "third_party/webrtc/base/byteorder.h"
17 #include "third_party/webrtc/base/messagedigest.h"
21 const uint32 kStunMagicCookie
= 0x2112A442;
22 const int kMinRtpHdrLen
= 12;
23 const int kRtpExtnHdrLen
= 4;
24 const int kDtlsRecordHeaderLen
= 13;
25 const int kTurnChannelHdrLen
= 4;
26 const int kAbsSendTimeExtnLen
= 3;
27 const int kOneByteHdrLen
= 1;
29 // Fake auth tag written by the render process if external authentication is
30 // enabled. HMAC in packet will be compared against this value before updating
31 // packet with actual HMAC value.
32 static const unsigned char kFakeAuthTag
[10] = {
33 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd, 0xba, 0xdd
36 bool IsTurnChannelData(const char* data
) {
37 return ((*data
& 0xC0) == 0x40);
40 bool IsDtlsPacket(const char* data
, int len
) {
41 const uint8
* u
= reinterpret_cast<const uint8
*>(data
);
42 return (len
>= kDtlsRecordHeaderLen
&& (u
[0] > 19 && u
[0] < 64));
45 bool IsRtcpPacket(const char* data
) {
46 int type
= (static_cast<uint8
>(data
[1]) & 0x7F);
47 return (type
>= 64 && type
< 96);
50 bool IsTurnSendIndicationPacket(const char* data
) {
51 uint16 type
= rtc::GetBE16(data
);
52 return (type
== cricket::TURN_SEND_INDICATION
);
55 bool IsRtpPacket(const char* data
, int len
) {
56 return ((*data
& 0xC0) == 0x80);
59 // Verifies rtp header and message length.
60 bool ValidateRtpHeader(const char* rtp
, int length
, size_t* header_length
) {
64 int cc_count
= rtp
[0] & 0x0F;
65 int rtp_hdr_len_without_extn
= kMinRtpHdrLen
+ 4 * cc_count
;
66 if (rtp_hdr_len_without_extn
> length
) {
70 // If extension bit is not set, we are done with header processing, as input
71 // length is verified above.
72 if (!(rtp
[0] & 0x10)) {
74 *header_length
= rtp_hdr_len_without_extn
;
79 rtp
+= rtp_hdr_len_without_extn
;
81 // Getting extension profile length.
82 // Length is in 32 bit words.
83 uint16 extn_length
= rtc::GetBE16(rtp
+ 2) * 4;
85 // Verify input length against total header size.
86 if (rtp_hdr_len_without_extn
+ kRtpExtnHdrLen
+ extn_length
> length
) {
91 *header_length
= rtp_hdr_len_without_extn
+ kRtpExtnHdrLen
+ extn_length
;
95 void UpdateAbsSendTimeExtnValue(char* extn_data
, int len
,
96 uint32 abs_send_time
) {
97 // Absolute send time in RTP streams.
99 // The absolute send time is signaled to the receiver in-band using the
100 // general mechanism for RTP header extensions [RFC5285]. The payload
101 // of this extension (the transmitted value) is a 24-bit unsigned integer
102 // containing the sender's current time in seconds as a fixed point number
103 // with 18 bits fractional part.
105 // The form of the absolute send time extension block:
108 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
109 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
110 // | ID | len=2 | absolute send time |
111 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
112 DCHECK_EQ(len
, kAbsSendTimeExtnLen
);
113 // Now() has resolution ~1-15ms, using HighResNow(). But it is warned not to
114 // use it unless necessary, as it is expensive than Now().
115 uint32 now_second
= abs_send_time
;
118 (base::TimeTicks::HighResNow() - base::TimeTicks()).InMicroseconds();
119 // Convert second to 24-bit unsigned with 18 bit fractional part
121 ((now_us
<< 18) / base::Time::kMicrosecondsPerSecond
) & 0x00FFFFFF;
123 // TODO(mallinath) - Add SetBE24 to byteorder.h in libjingle.
124 extn_data
[0] = static_cast<uint8
>(now_second
>> 16);
125 extn_data
[1] = static_cast<uint8
>(now_second
>> 8);
126 extn_data
[2] = static_cast<uint8
>(now_second
);
129 // Assumes |len| is actual packet length + tag length. Updates HMAC at end of
131 void UpdateRtpAuthTag(char* rtp
, int len
,
132 const rtc::PacketOptions
& options
) {
133 // If there is no key, return.
134 if (options
.packet_time_params
.srtp_auth_key
.empty())
137 size_t tag_length
= options
.packet_time_params
.srtp_auth_tag_len
;
138 char* auth_tag
= rtp
+ (len
- tag_length
);
140 // We should have a fake HMAC value @ auth_tag.
141 DCHECK_EQ(0, memcmp(auth_tag
, kFakeAuthTag
, tag_length
));
143 crypto::HMAC
hmac(crypto::HMAC::SHA1
);
144 if (!hmac
.Init(reinterpret_cast<const unsigned char*>(
145 &options
.packet_time_params
.srtp_auth_key
[0]),
146 options
.packet_time_params
.srtp_auth_key
.size())) {
151 if (hmac
.DigestLength() < tag_length
) {
156 // Copy ROC after end of rtp packet.
157 memcpy(auth_tag
, &options
.packet_time_params
.srtp_packet_index
, 4);
158 // Authentication of a RTP packet will have RTP packet + ROC size.
159 int auth_required_length
= len
- tag_length
+ 4;
161 unsigned char output
[64];
162 if (!hmac
.Sign(base::StringPiece(rtp
, auth_required_length
),
163 output
, sizeof(output
))) {
167 // Copy HMAC from output to packet. This is required as auth tag length
168 // may not be equal to the actual HMAC length.
169 memcpy(auth_tag
, output
, tag_length
);
176 namespace packet_processing_helpers
{
178 bool ApplyPacketOptions(char* data
, int length
,
179 const rtc::PacketOptions
& options
,
180 uint32 abs_send_time
) {
181 DCHECK(data
!= NULL
);
183 // if there is no valid |rtp_sendtime_extension_id| and |srtp_auth_key| in
184 // PacketOptions, nothing to be updated in this packet.
185 if (options
.packet_time_params
.rtp_sendtime_extension_id
== -1 &&
186 options
.packet_time_params
.srtp_auth_key
.empty()) {
190 DCHECK(!IsDtlsPacket(data
, length
));
191 DCHECK(!IsRtcpPacket(data
));
193 // If there is a srtp auth key present then packet must be a RTP packet.
194 // RTP packet may have been wrapped in a TURN Channel Data or
195 // TURN send indication.
198 if (!GetRtpPacketStartPositionAndLength(
199 data
, length
, &rtp_start_pos
, &rtp_length
)) {
200 // This method should never return false.
205 // Skip to rtp packet.
206 char* start
= data
+ rtp_start_pos
;
207 // If packet option has non default value (-1) for sendtime extension id,
208 // then we should parse the rtp packet to update the timestamp. Otherwise
209 // just calculate HMAC and update packet with it.
210 if (options
.packet_time_params
.rtp_sendtime_extension_id
!= -1) {
211 UpdateRtpAbsSendTimeExtn(
213 options
.packet_time_params
.rtp_sendtime_extension_id
, abs_send_time
);
216 UpdateRtpAuthTag(start
, rtp_length
, options
);
220 bool GetRtpPacketStartPositionAndLength(const char* packet
,
223 int* rtp_packet_length
) {
226 if (IsTurnChannelData(packet
)) {
227 // Turn Channel Message header format.
229 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
230 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
231 // | Channel Number | Length |
232 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
234 // / Application Data /
236 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
237 if (length
< kTurnChannelHdrLen
) {
241 rtp_begin
= kTurnChannelHdrLen
;
242 rtp_length
= rtc::GetBE16(&packet
[2]);
243 if (length
< rtp_length
+ kTurnChannelHdrLen
) {
246 } else if (IsTurnSendIndicationPacket(packet
)) {
247 if (length
<= P2PSocketHost::kStunHeaderSize
) {
248 // Message must be greater than 20 bytes, if it's carrying any payload.
251 // Validate STUN message length.
252 int stun_msg_len
= rtc::GetBE16(&packet
[2]);
253 if (stun_msg_len
+ P2PSocketHost::kStunHeaderSize
!= length
) {
257 // First skip mandatory stun header which is of 20 bytes.
258 rtp_begin
= P2PSocketHost::kStunHeaderSize
;
259 // Loop through STUN attributes until we find STUN DATA attribute.
260 const char* start
= packet
+ rtp_begin
;
261 bool data_attr_present
= false;
262 while ((packet
+ rtp_begin
) - start
< stun_msg_len
) {
263 // Keep reading STUN attributes until we hit DATA attribute.
264 // Attribute will be a TLV structure.
265 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
267 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
268 // | Value (variable) ....
269 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
270 // The value in the length field MUST contain the length of the Value
271 // part of the attribute, prior to padding, measured in bytes. Since
272 // STUN aligns attributes on 32-bit boundaries, attributes whose content
273 // is not a multiple of 4 bytes are padded with 1, 2, or 3 bytes of
274 // padding so that its value contains a multiple of 4 bytes. The
275 // padding bits are ignored, and may be any value.
276 uint16 attr_type
, attr_length
;
277 // Getting attribute type and length.
278 attr_type
= rtc::GetBE16(&packet
[rtp_begin
]);
279 attr_length
= rtc::GetBE16(
280 &packet
[rtp_begin
+ sizeof(attr_type
)]);
281 // Checking for bogus attribute length.
282 if (length
< attr_length
+ rtp_begin
) {
286 if (attr_type
!= cricket::STUN_ATTR_DATA
) {
287 rtp_begin
+= sizeof(attr_type
) + sizeof(attr_length
) + attr_length
;
288 if ((attr_length
% 4) != 0) {
289 rtp_begin
+= (4 - (attr_length
% 4));
294 data_attr_present
= true;
295 rtp_begin
+= 4; // Skip STUN_DATA_ATTR header.
296 rtp_length
= attr_length
;
297 // One final check of length before exiting.
298 if (length
< rtp_length
+ rtp_begin
) {
301 // We found STUN_DATA_ATTR. We can skip parsing rest of the packet.
305 if (!data_attr_present
) {
306 // There is no data attribute present in the message. We can't do anything
312 // This is a raw RTP packet.
317 // Making sure we have a valid RTP packet at the end.
318 if ((rtp_length
>= kMinRtpHdrLen
) &&
319 IsRtpPacket(packet
+ rtp_begin
, rtp_length
) &&
320 ValidateRtpHeader(packet
+ rtp_begin
, rtp_length
, NULL
)) {
321 *rtp_start_pos
= rtp_begin
;
322 *rtp_packet_length
= rtp_length
;
328 // ValidateRtpHeader must be called before this method to make sure, we have
329 // a sane rtp packet.
330 bool UpdateRtpAbsSendTimeExtn(char* rtp
, int length
,
331 int extension_id
, uint32 abs_send_time
) {
333 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
334 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
335 // |V=2|P|X| CC |M| PT | sequence number |
336 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
338 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
339 // | synchronization source (SSRC) identifier |
340 // +=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+
341 // | contributing source (CSRC) identifiers |
343 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
345 // Return if extension bit is not set.
346 if (!(rtp
[0] & 0x10)) {
350 int cc_count
= rtp
[0] & 0x0F;
351 int rtp_hdr_len_without_extn
= kMinRtpHdrLen
+ 4 * cc_count
;
353 rtp
+= rtp_hdr_len_without_extn
;
355 // Getting extension profile ID and length.
356 uint16 profile_id
= rtc::GetBE16(rtp
);
357 // Length is in 32 bit words.
358 uint16 extn_length
= rtc::GetBE16(rtp
+ 2) * 4;
360 rtp
+= kRtpExtnHdrLen
; // Moving past extn header.
363 // WebRTC is using one byte header extension.
364 // TODO(mallinath) - Handle two byte header extension.
365 if (profile_id
== 0xBEDE) { // OneByte extension header
373 // 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
374 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
375 // | 0xBE | 0xDE | length=3 |
376 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
377 // | ID | L=0 | data | ID | L=1 | data...
378 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
379 // ...data | 0 (pad) | 0 (pad) | ID | L=3 |
380 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
382 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
383 char* extn_start
= rtp
;
384 while (rtp
- extn_start
< extn_length
) {
385 const int id
= (*rtp
& 0xF0) >> 4;
386 const int len
= (*rtp
& 0x0F) + 1;
387 // The 4-bit length is the number minus one of data bytes of this header
388 // extension element following the one-byte header.
389 if (id
== extension_id
) {
390 UpdateAbsSendTimeExtnValue(rtp
+ kOneByteHdrLen
, len
, abs_send_time
);
394 rtp
+= kOneByteHdrLen
+ len
;
395 // Counting padding bytes.
396 while ((*rtp
== 0) && (rtp
- extn_start
< extn_length
)) {
404 } // packet_processing_helpers
406 P2PSocketHost::P2PSocketHost(IPC::Sender
* message_sender
, int socket_id
)
407 : message_sender_(message_sender
),
409 state_(STATE_UNINITIALIZED
),
410 dump_incoming_rtp_packet_(false),
411 dump_outgoing_rtp_packet_(false),
412 weak_ptr_factory_(this) {
415 P2PSocketHost::~P2PSocketHost() { }
417 // Verifies that the packet |data| has a valid STUN header.
419 bool P2PSocketHost::GetStunPacketType(
420 const char* data
, int data_size
, StunMessageType
* type
) {
422 if (data_size
< kStunHeaderSize
)
425 uint32 cookie
= base::NetToHost32(*reinterpret_cast<const uint32
*>(data
+ 4));
426 if (cookie
!= kStunMagicCookie
)
429 uint16 length
= base::NetToHost16(*reinterpret_cast<const uint16
*>(data
+ 2));
430 if (length
!= data_size
- kStunHeaderSize
)
433 int message_type
= base::NetToHost16(*reinterpret_cast<const uint16
*>(data
));
435 // Verify that the type is known:
436 switch (message_type
) {
437 case STUN_BINDING_REQUEST
:
438 case STUN_BINDING_RESPONSE
:
439 case STUN_BINDING_ERROR_RESPONSE
:
440 case STUN_SHARED_SECRET_REQUEST
:
441 case STUN_SHARED_SECRET_RESPONSE
:
442 case STUN_SHARED_SECRET_ERROR_RESPONSE
:
443 case STUN_ALLOCATE_REQUEST
:
444 case STUN_ALLOCATE_RESPONSE
:
445 case STUN_ALLOCATE_ERROR_RESPONSE
:
446 case STUN_SEND_REQUEST
:
447 case STUN_SEND_RESPONSE
:
448 case STUN_SEND_ERROR_RESPONSE
:
449 case STUN_DATA_INDICATION
:
450 *type
= static_cast<StunMessageType
>(message_type
);
459 bool P2PSocketHost::IsRequestOrResponse(StunMessageType type
) {
460 return type
== STUN_BINDING_REQUEST
|| type
== STUN_BINDING_RESPONSE
||
461 type
== STUN_ALLOCATE_REQUEST
|| type
== STUN_ALLOCATE_RESPONSE
;
465 P2PSocketHost
* P2PSocketHost::Create(IPC::Sender
* message_sender
,
468 net::URLRequestContextGetter
* url_context
,
469 P2PMessageThrottler
* throttler
) {
472 return new P2PSocketHostUdp(message_sender
, socket_id
, throttler
);
473 case P2P_SOCKET_TCP_SERVER
:
474 return new P2PSocketHostTcpServer(
475 message_sender
, socket_id
, P2P_SOCKET_TCP_CLIENT
);
477 case P2P_SOCKET_STUN_TCP_SERVER
:
478 return new P2PSocketHostTcpServer(
479 message_sender
, socket_id
, P2P_SOCKET_STUN_TCP_CLIENT
);
481 case P2P_SOCKET_TCP_CLIENT
:
482 case P2P_SOCKET_SSLTCP_CLIENT
:
483 case P2P_SOCKET_TLS_CLIENT
:
484 return new P2PSocketHostTcp(message_sender
, socket_id
, type
, url_context
);
486 case P2P_SOCKET_STUN_TCP_CLIENT
:
487 case P2P_SOCKET_STUN_SSLTCP_CLIENT
:
488 case P2P_SOCKET_STUN_TLS_CLIENT
:
489 return new P2PSocketHostStunTcp(
490 message_sender
, socket_id
, type
, url_context
);
497 void P2PSocketHost::StartRtpDump(
500 const RenderProcessHost::WebRtcRtpPacketCallback
& packet_callback
) {
501 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO
));
502 DCHECK(!packet_callback
.is_null());
503 DCHECK(incoming
|| outgoing
);
506 dump_incoming_rtp_packet_
= true;
509 dump_outgoing_rtp_packet_
= true;
511 packet_dump_callback_
= packet_callback
;
514 void P2PSocketHost::StopRtpDump(bool incoming
, bool outgoing
) {
515 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO
));
516 DCHECK(incoming
|| outgoing
);
519 dump_incoming_rtp_packet_
= false;
522 dump_outgoing_rtp_packet_
= false;
524 if (!dump_incoming_rtp_packet_
&& !dump_outgoing_rtp_packet_
)
525 packet_dump_callback_
.Reset();
528 void P2PSocketHost::DumpRtpPacket(const char* packet
,
531 if (IsDtlsPacket(packet
, length
) || IsRtcpPacket(packet
))
534 int rtp_packet_pos
= 0;
535 int rtp_packet_length
= length
;
536 if (!packet_processing_helpers::GetRtpPacketStartPositionAndLength(
537 packet
, length
, &rtp_packet_pos
, &rtp_packet_length
))
540 packet
+= rtp_packet_pos
;
542 size_t header_length
= 0;
543 bool valid
= ValidateRtpHeader(packet
, rtp_packet_length
, &header_length
);
549 scoped_ptr
<uint8
[]> header_buffer(new uint8
[header_length
]);
550 memcpy(header_buffer
.get(), packet
, header_length
);
552 // Posts to the IO thread as the data members should be accessed on the IO
554 BrowserThread::PostTask(BrowserThread::IO
,
556 base::Bind(&P2PSocketHost::DumpRtpPacketOnIOThread
,
557 weak_ptr_factory_
.GetWeakPtr(),
558 Passed(&header_buffer
),
564 void P2PSocketHost::DumpRtpPacketOnIOThread(scoped_ptr
<uint8
[]> packet_header
,
565 size_t header_length
,
566 size_t packet_length
,
568 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO
));
570 if ((incoming
&& !dump_incoming_rtp_packet_
) ||
571 (!incoming
&& !dump_outgoing_rtp_packet_
) ||
572 packet_dump_callback_
.is_null()) {
576 // |packet_dump_callback_| must be called on the UI thread.
577 BrowserThread::PostTask(BrowserThread::UI
,
579 base::Bind(packet_dump_callback_
,
580 Passed(&packet_header
),
586 } // namespace content