OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 #include "media/cast/rtp_receiver/rtp_parser/rtp_parser.h" | 5 #include "media/cast/rtp_receiver/rtp_parser/rtp_parser.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "media/cast/cast_defines.h" | 8 #include "media/cast/cast_defines.h" |
9 #include "media/cast/rtp_receiver/rtp_receiver.h" | 9 #include "media/cast/rtp_receiver/rtp_receiver.h" |
10 #include "net/base/big_endian.h" | 10 #include "net/base/big_endian.h" |
11 | 11 |
12 namespace media { | 12 namespace media { |
13 namespace cast { | 13 namespace cast { |
14 | 14 |
15 static const size_t kRtpCommonHeaderLength = 12; | 15 static const size_t kRtpCommonHeaderLength = 12; |
16 static const size_t kRtpCastHeaderLength = 7; | 16 static const size_t kRtpCastHeaderLength = 7; |
17 static const uint8 kCastKeyFrameBitMask = 0x80; | 17 static const uint8 kCastKeyFrameBitMask = 0x80; |
18 static const uint8 kCastReferenceFrameIdBitMask = 0x40; | 18 static const uint8 kCastReferenceFrameIdBitMask = 0x40; |
19 | 19 |
20 RtpParser::RtpParser(RtpData* incoming_payload_callback, | 20 RtpParser::RtpParser(RtpData* incoming_payload_callback, |
21 const RtpParserConfig parser_config) | 21 const RtpParserConfig parser_config) |
22 : data_callback_(incoming_payload_callback), | 22 : data_callback_(incoming_payload_callback), |
23 parser_config_(parser_config) {} | 23 parser_config_(parser_config) {} |
24 | 24 |
25 RtpParser::~RtpParser() {} | 25 RtpParser::~RtpParser() {} |
26 | 26 |
27 bool RtpParser::ParsePacket(const uint8* packet, size_t length, | 27 bool RtpParser::ParsePacket(const uint8* packet, |
| 28 size_t length, |
28 RtpCastHeader* rtp_header) { | 29 RtpCastHeader* rtp_header) { |
29 if (length == 0) return false; | 30 if (length == 0) |
| 31 return false; |
30 // Get RTP general header. | 32 // Get RTP general header. |
31 if (!ParseCommon(packet, length, rtp_header)) return false; | 33 if (!ParseCommon(packet, length, rtp_header)) |
| 34 return false; |
32 if (rtp_header->webrtc.header.payloadType == parser_config_.payload_type && | 35 if (rtp_header->webrtc.header.payloadType == parser_config_.payload_type && |
33 rtp_header->webrtc.header.ssrc == parser_config_.ssrc) { | 36 rtp_header->webrtc.header.ssrc == parser_config_.ssrc) { |
34 return ParseCast(packet + kRtpCommonHeaderLength, | 37 return ParseCast(packet + kRtpCommonHeaderLength, |
35 length - kRtpCommonHeaderLength, rtp_header); | 38 length - kRtpCommonHeaderLength, |
| 39 rtp_header); |
36 } | 40 } |
37 // Not a valid payload type / ssrc combination. | 41 // Not a valid payload type / ssrc combination. |
38 return false; | 42 return false; |
39 } | 43 } |
40 | 44 |
41 bool RtpParser::ParseCommon(const uint8* packet, | 45 bool RtpParser::ParseCommon(const uint8* packet, |
42 size_t length, | 46 size_t length, |
43 RtpCastHeader* rtp_header) { | 47 RtpCastHeader* rtp_header) { |
44 if (length < kRtpCommonHeaderLength) return false; | 48 if (length < kRtpCommonHeaderLength) |
| 49 return false; |
45 uint8 version = packet[0] >> 6; | 50 uint8 version = packet[0] >> 6; |
46 if (version != 2) return false; | 51 if (version != 2) |
| 52 return false; |
47 uint8 cc = packet[0] & 0x0f; | 53 uint8 cc = packet[0] & 0x0f; |
48 bool marker = ((packet[1] & 0x80) != 0); | 54 bool marker = ((packet[1] & 0x80) != 0); |
49 int payload_type = packet[1] & 0x7f; | 55 int payload_type = packet[1] & 0x7f; |
50 | 56 |
51 uint16 sequence_number; | 57 uint16 sequence_number; |
52 uint32 rtp_timestamp, ssrc; | 58 uint32 rtp_timestamp, ssrc; |
53 net::BigEndianReader big_endian_reader(packet + 2, 10); | 59 net::BigEndianReader big_endian_reader(packet + 2, 10); |
54 big_endian_reader.ReadU16(&sequence_number); | 60 big_endian_reader.ReadU16(&sequence_number); |
55 big_endian_reader.ReadU32(&rtp_timestamp); | 61 big_endian_reader.ReadU32(&rtp_timestamp); |
56 big_endian_reader.ReadU32(&ssrc); | 62 big_endian_reader.ReadU32(&ssrc); |
57 | 63 |
58 if (ssrc != parser_config_.ssrc) return false; | 64 if (ssrc != parser_config_.ssrc) |
| 65 return false; |
59 | 66 |
60 rtp_header->webrtc.header.markerBit = marker; | 67 rtp_header->webrtc.header.markerBit = marker; |
61 rtp_header->webrtc.header.payloadType = payload_type; | 68 rtp_header->webrtc.header.payloadType = payload_type; |
62 rtp_header->webrtc.header.sequenceNumber = sequence_number; | 69 rtp_header->webrtc.header.sequenceNumber = sequence_number; |
63 rtp_header->webrtc.header.timestamp = rtp_timestamp; | 70 rtp_header->webrtc.header.timestamp = rtp_timestamp; |
64 rtp_header->webrtc.header.ssrc = ssrc; | 71 rtp_header->webrtc.header.ssrc = ssrc; |
65 rtp_header->webrtc.header.numCSRCs = cc; | 72 rtp_header->webrtc.header.numCSRCs = cc; |
66 | 73 |
67 uint8 csrc_octs = cc * 4; | 74 uint8 csrc_octs = cc * 4; |
68 rtp_header->webrtc.type.Audio.numEnergy = rtp_header->webrtc.header.numCSRCs; | 75 rtp_header->webrtc.type.Audio.numEnergy = rtp_header->webrtc.header.numCSRCs; |
69 rtp_header->webrtc.header.headerLength = kRtpCommonHeaderLength + csrc_octs; | 76 rtp_header->webrtc.header.headerLength = kRtpCommonHeaderLength + csrc_octs; |
70 rtp_header->webrtc.type.Audio.isCNG = false; | 77 rtp_header->webrtc.type.Audio.isCNG = false; |
71 rtp_header->webrtc.type.Audio.channel = parser_config_.audio_channels; | 78 rtp_header->webrtc.type.Audio.channel = parser_config_.audio_channels; |
72 // TODO(pwestin): look at x bit and skip data. | 79 // TODO(pwestin): look at x bit and skip data. |
73 return true; | 80 return true; |
74 } | 81 } |
75 | 82 |
76 bool RtpParser::ParseCast(const uint8* packet, | 83 bool RtpParser::ParseCast(const uint8* packet, |
77 size_t length, | 84 size_t length, |
78 RtpCastHeader* rtp_header) { | 85 RtpCastHeader* rtp_header) { |
79 if (length < kRtpCastHeaderLength) return false; | 86 if (length < kRtpCastHeaderLength) |
| 87 return false; |
80 | 88 |
81 // Extract header. | 89 // Extract header. |
82 const uint8* data_ptr = packet; | 90 const uint8* data_ptr = packet; |
83 size_t data_length = length; | 91 size_t data_length = length; |
84 rtp_header->is_key_frame = (data_ptr[0] & kCastKeyFrameBitMask); | 92 rtp_header->is_key_frame = (data_ptr[0] & kCastKeyFrameBitMask); |
85 rtp_header->is_reference = (data_ptr[0] & kCastReferenceFrameIdBitMask); | 93 rtp_header->is_reference = (data_ptr[0] & kCastReferenceFrameIdBitMask); |
86 rtp_header->frame_id = frame_id_wrap_helper_.MapTo32bitsFrameId(data_ptr[1]); | 94 rtp_header->frame_id = frame_id_wrap_helper_.MapTo32bitsFrameId(data_ptr[1]); |
87 | 95 |
88 net::BigEndianReader big_endian_reader(data_ptr + 2, 4); | 96 net::BigEndianReader big_endian_reader(data_ptr + 2, 4); |
89 big_endian_reader.ReadU16(&rtp_header->packet_id); | 97 big_endian_reader.ReadU16(&rtp_header->packet_id); |
90 big_endian_reader.ReadU16(&rtp_header->max_packet_id); | 98 big_endian_reader.ReadU16(&rtp_header->max_packet_id); |
91 | 99 |
92 if (rtp_header->is_reference) { | 100 if (rtp_header->is_reference) { |
93 rtp_header->reference_frame_id = | 101 rtp_header->reference_frame_id = |
94 reference_frame_id_wrap_helper_.MapTo32bitsFrameId(data_ptr[6]); | 102 reference_frame_id_wrap_helper_.MapTo32bitsFrameId(data_ptr[6]); |
95 data_ptr += kRtpCastHeaderLength; | 103 data_ptr += kRtpCastHeaderLength; |
96 data_length -= kRtpCastHeaderLength; | 104 data_length -= kRtpCastHeaderLength; |
97 } else { | 105 } else { |
98 data_ptr += kRtpCastHeaderLength - 1; | 106 data_ptr += kRtpCastHeaderLength - 1; |
99 data_length -= kRtpCastHeaderLength - 1; | 107 data_length -= kRtpCastHeaderLength - 1; |
100 } | 108 } |
101 | 109 |
102 if (rtp_header->max_packet_id < rtp_header->packet_id) return false; | 110 if (rtp_header->max_packet_id < rtp_header->packet_id) |
| 111 return false; |
103 | 112 |
104 data_callback_->OnReceivedPayloadData(data_ptr, data_length, rtp_header); | 113 data_callback_->OnReceivedPayloadData(data_ptr, data_length, rtp_header); |
105 return true; | 114 return true; |
106 } | 115 } |
107 | 116 |
108 } // namespace cast | 117 } // namespace cast |
109 } // namespace media | 118 } // namespace media |
OLD | NEW |