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