| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "media/cast/rtp_receiver/rtp_parser/test/rtp_packet_builder.h" | |
| 6 | |
| 7 #include "base/big_endian.h" | |
| 8 #include "base/logging.h" | |
| 9 | |
| 10 namespace media { | |
| 11 namespace cast { | |
| 12 | |
| 13 const uint32 kCastRtpHeaderLength = 7; | |
| 14 const uint32 kGenericRtpHeaderLength = 12; | |
| 15 static const uint8 kRtpMarkerBitMask = 0x80; | |
| 16 | |
| 17 RtpPacketBuilder::RtpPacketBuilder() | |
| 18 : is_key_(false), | |
| 19 frame_id_(0), | |
| 20 packet_id_(0), | |
| 21 max_packet_id_(0), | |
| 22 reference_frame_id_(0), | |
| 23 timestamp_(0), | |
| 24 sequence_number_(0), | |
| 25 marker_(false), | |
| 26 payload_type_(0), | |
| 27 ssrc_(0) {} | |
| 28 | |
| 29 void RtpPacketBuilder::SetKeyFrame(bool is_key) { is_key_ = is_key; } | |
| 30 | |
| 31 void RtpPacketBuilder::SetFrameIds(uint32 frame_id, uint32 reference_frame_id) { | |
| 32 frame_id_ = frame_id; | |
| 33 reference_frame_id_ = reference_frame_id; | |
| 34 } | |
| 35 | |
| 36 void RtpPacketBuilder::SetPacketId(uint16 packet_id) { packet_id_ = packet_id; } | |
| 37 | |
| 38 void RtpPacketBuilder::SetMaxPacketId(uint16 max_packet_id) { | |
| 39 max_packet_id_ = max_packet_id; | |
| 40 } | |
| 41 | |
| 42 void RtpPacketBuilder::SetTimestamp(uint32 timestamp) { | |
| 43 timestamp_ = timestamp; | |
| 44 } | |
| 45 | |
| 46 void RtpPacketBuilder::SetSequenceNumber(uint16 sequence_number) { | |
| 47 sequence_number_ = sequence_number; | |
| 48 } | |
| 49 | |
| 50 void RtpPacketBuilder::SetMarkerBit(bool marker) { marker_ = marker; } | |
| 51 | |
| 52 void RtpPacketBuilder::SetPayloadType(int payload_type) { | |
| 53 payload_type_ = payload_type; | |
| 54 } | |
| 55 | |
| 56 void RtpPacketBuilder::SetSsrc(uint32 ssrc) { ssrc_ = ssrc; } | |
| 57 | |
| 58 void RtpPacketBuilder::BuildHeader(uint8* data, uint32 data_length) { | |
| 59 BuildCommonHeader(data, data_length); | |
| 60 BuildCastHeader(data + kGenericRtpHeaderLength, | |
| 61 data_length - kGenericRtpHeaderLength); | |
| 62 } | |
| 63 | |
| 64 void RtpPacketBuilder::BuildCastHeader(uint8* data, uint32 data_length) { | |
| 65 // Build header. | |
| 66 DCHECK_LE(kCastRtpHeaderLength, data_length); | |
| 67 // Set the first 7 bytes to 0. | |
| 68 memset(data, 0, kCastRtpHeaderLength); | |
| 69 base::BigEndianWriter big_endian_writer(reinterpret_cast<char*>(data), 56); | |
| 70 const bool includes_specific_frame_reference = | |
| 71 (is_key_ && (reference_frame_id_ != frame_id_)) || | |
| 72 (!is_key_ && (reference_frame_id_ != (frame_id_ - 1))); | |
| 73 big_endian_writer.WriteU8((is_key_ ? 0x80 : 0) | | |
| 74 (includes_specific_frame_reference ? 0x40 : 0)); | |
| 75 big_endian_writer.WriteU8(frame_id_); | |
| 76 big_endian_writer.WriteU16(packet_id_); | |
| 77 big_endian_writer.WriteU16(max_packet_id_); | |
| 78 if (includes_specific_frame_reference) { | |
| 79 big_endian_writer.WriteU8(reference_frame_id_); | |
| 80 } | |
| 81 } | |
| 82 | |
| 83 void RtpPacketBuilder::BuildCommonHeader(uint8* data, uint32 data_length) { | |
| 84 DCHECK_LE(kGenericRtpHeaderLength, data_length); | |
| 85 base::BigEndianWriter big_endian_writer(reinterpret_cast<char*>(data), 96); | |
| 86 big_endian_writer.WriteU8(0x80); | |
| 87 big_endian_writer.WriteU8(payload_type_ | (marker_ ? kRtpMarkerBitMask : 0)); | |
| 88 big_endian_writer.WriteU16(sequence_number_); | |
| 89 big_endian_writer.WriteU32(timestamp_); | |
| 90 big_endian_writer.WriteU32(ssrc_); | |
| 91 } | |
| 92 | |
| 93 } // namespace cast | |
| 94 } // namespace media | |
| OLD | NEW |