| 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_sender/rtp_packetizer/test/rtp_header_parser.h" | 5 #include "media/cast/net/rtp_sender/rtp_packetizer/test/rtp_header_parser.h" |
| 6 | 6 |
| 7 #include <cstddef> | 7 #include <cstddef> |
| 8 | 8 |
| 9 #include "net/base/big_endian.h" | 9 #include "net/base/big_endian.h" |
| 10 | 10 |
| 11 namespace media { | 11 namespace media { |
| 12 namespace cast { | 12 namespace cast { |
| 13 | 13 |
| 14 static const uint8 kCastKeyFrameBitMask = 0x80; | 14 static const uint8 kCastKeyFrameBitMask = 0x80; |
| 15 static const uint8 kCastReferenceFrameIdBitMask = 0x40; | 15 static const uint8 kCastReferenceFrameIdBitMask = 0x40; |
| 16 static const size_t kRtpCommonHeaderLength = 12; | 16 static const size_t kRtpCommonHeaderLength = 12; |
| 17 static const size_t kRtpCastHeaderLength = 12; | 17 static const size_t kRtpCastHeaderLength = 12; |
| 18 | 18 |
| 19 RtpCastTestHeader::RtpCastTestHeader() |
| 20 : is_key_frame(false), |
| 21 frame_id(0), |
| 22 packet_id(0), |
| 23 max_packet_id(0), |
| 24 is_reference(false), |
| 25 reference_frame_id(0), |
| 26 marker(false), |
| 27 sequence_number(0), |
| 28 rtp_timestamp(0), |
| 29 ssrc(0), |
| 30 payload_type(0), |
| 31 num_csrcs(0), |
| 32 audio_num_energy(0), |
| 33 header_length(0) {} |
| 34 |
| 35 RtpCastTestHeader::~RtpCastTestHeader() {} |
| 19 | 36 |
| 20 RtpHeaderParser::RtpHeaderParser(const uint8* rtp_data, | 37 RtpHeaderParser::RtpHeaderParser(const uint8* rtp_data, |
| 21 size_t rtp_data_length) | 38 size_t rtp_data_length) |
| 22 : rtp_data_begin_(rtp_data), | 39 : rtp_data_begin_(rtp_data), |
| 23 length_(rtp_data_length) {} | 40 length_(rtp_data_length) {} |
| 24 | 41 |
| 25 RtpHeaderParser::~RtpHeaderParser() {} | 42 RtpHeaderParser::~RtpHeaderParser() {} |
| 26 | 43 |
| 27 bool RtpHeaderParser::Parse(RtpCastHeader* parsed_packet) const { | 44 bool RtpHeaderParser::Parse(RtpCastTestHeader* parsed_packet) const { |
| 28 if (length_ < kRtpCommonHeaderLength + kRtpCastHeaderLength) | 45 if (length_ < kRtpCommonHeaderLength + kRtpCastHeaderLength) |
| 29 return false; | 46 return false; |
| 30 if (!ParseCommon(parsed_packet)) return false; | 47 if (!ParseCommon(parsed_packet)) return false; |
| 31 return ParseCast(parsed_packet); | 48 return ParseCast(parsed_packet); |
| 32 } | 49 } |
| 33 | 50 |
| 34 bool RtpHeaderParser::ParseCommon(RtpCastHeader* parsed_packet) const { | 51 bool RtpHeaderParser::ParseCommon(RtpCastTestHeader* parsed_packet) const { |
| 35 const uint8 version = rtp_data_begin_[0] >> 6; | 52 const uint8 version = rtp_data_begin_[0] >> 6; |
| 36 if (version != 2) { | 53 if (version != 2) { |
| 37 return false; | 54 return false; |
| 38 } | 55 } |
| 39 | 56 |
| 40 const uint8 num_csrcs = rtp_data_begin_[0] & 0x0f; | 57 const uint8 num_csrcs = rtp_data_begin_[0] & 0x0f; |
| 41 const bool marker = ((rtp_data_begin_[1] & 0x80) == 0) ? false : true; | 58 const bool marker = ((rtp_data_begin_[1] & 0x80) == 0) ? false : true; |
| 42 const uint8 payload_type = rtp_data_begin_[1] & 0x7f; | 59 const uint8 payload_type = rtp_data_begin_[1] & 0x7f; |
| 43 const uint16 sequence_number = (rtp_data_begin_[2] << 8) + | 60 const uint16 sequence_number = (rtp_data_begin_[2] << 8) + |
| 44 rtp_data_begin_[3]; | 61 rtp_data_begin_[3]; |
| 45 | 62 |
| 46 const uint8* ptr = &rtp_data_begin_[4]; | 63 const uint8* ptr = &rtp_data_begin_[4]; |
| 47 | 64 |
| 48 net::BigEndianReader big_endian_reader(ptr, 8); | 65 net::BigEndianReader big_endian_reader(ptr, 8); |
| 49 uint32 rtp_timestamp, ssrc; | 66 uint32 rtp_timestamp, ssrc; |
| 50 big_endian_reader.ReadU32(&rtp_timestamp); | 67 big_endian_reader.ReadU32(&rtp_timestamp); |
| 51 big_endian_reader.ReadU32(&ssrc); | 68 big_endian_reader.ReadU32(&ssrc); |
| 52 | 69 |
| 53 const uint8 csrc_octs = num_csrcs * 4; | 70 const uint8 csrc_octs = num_csrcs * 4; |
| 54 | 71 |
| 55 parsed_packet->webrtc.header.markerBit = marker; | 72 parsed_packet->marker = marker; |
| 56 parsed_packet->webrtc.header.payloadType = payload_type; | 73 parsed_packet->payload_type = payload_type; |
| 57 parsed_packet->webrtc.header.sequenceNumber = sequence_number; | 74 parsed_packet->sequence_number = sequence_number; |
| 58 parsed_packet->webrtc.header.timestamp = rtp_timestamp; | 75 parsed_packet->rtp_timestamp = rtp_timestamp; |
| 59 parsed_packet->webrtc.header.ssrc = ssrc; | 76 parsed_packet->ssrc = ssrc; |
| 60 parsed_packet->webrtc.header.numCSRCs = num_csrcs; | 77 parsed_packet->num_csrcs = num_csrcs; |
| 61 | 78 |
| 62 parsed_packet->webrtc.type.Audio.numEnergy = | 79 parsed_packet->audio_num_energy = parsed_packet->num_csrcs; |
| 63 parsed_packet->webrtc.header.numCSRCs; | |
| 64 | 80 |
| 65 parsed_packet->webrtc.header.headerLength = 12 + csrc_octs; | 81 parsed_packet->header_length = 12 + csrc_octs; |
| 66 return true; | 82 return true; |
| 67 } | 83 } |
| 68 | 84 |
| 69 bool RtpHeaderParser::ParseCast(RtpCastHeader* parsed_packet) const { | 85 bool RtpHeaderParser::ParseCast(RtpCastTestHeader* parsed_packet) const { |
| 70 const uint8* data = rtp_data_begin_ + kRtpCommonHeaderLength; | 86 const uint8* data = rtp_data_begin_ + kRtpCommonHeaderLength; |
| 71 parsed_packet->is_key_frame = (data[0] & kCastKeyFrameBitMask); | 87 parsed_packet->is_key_frame = (data[0] & kCastKeyFrameBitMask); |
| 72 parsed_packet->is_reference = (data[0] & kCastReferenceFrameIdBitMask); | 88 parsed_packet->is_reference = (data[0] & kCastReferenceFrameIdBitMask); |
| 73 parsed_packet->frame_id = frame_id_wrap_helper_.MapTo32bitsFrameId(data[1]); | 89 parsed_packet->frame_id = frame_id_wrap_helper_.MapTo32bitsFrameId(data[1]); |
| 74 | 90 |
| 75 net::BigEndianReader big_endian_reader(data + 2, 8); | 91 net::BigEndianReader big_endian_reader(data + 2, 8); |
| 76 big_endian_reader.ReadU16(&parsed_packet->packet_id); | 92 big_endian_reader.ReadU16(&parsed_packet->packet_id); |
| 77 big_endian_reader.ReadU16(&parsed_packet->max_packet_id); | 93 big_endian_reader.ReadU16(&parsed_packet->max_packet_id); |
| 78 | 94 |
| 79 if (parsed_packet->is_reference) { | 95 if (parsed_packet->is_reference) { |
| 80 parsed_packet->reference_frame_id = | 96 parsed_packet->reference_frame_id = |
| 81 reference_frame_id_wrap_helper_.MapTo32bitsFrameId(data[6]); | 97 reference_frame_id_wrap_helper_.MapTo32bitsFrameId(data[6]); |
| 82 } | 98 } |
| 83 return true; | 99 return true; |
| 84 } | 100 } |
| 85 | 101 |
| 86 } // namespace cast | 102 } // namespace cast |
| 87 } // namespace media | 103 } // namespace media |
| OLD | NEW |