| 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 |