src/media/cast/rtp_receiver/rtp_parser/test/rtp_packet_builder.cc

   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/logging.h"
   8 #include "net/base/big_endian.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       is_reference_set_(false),
  24       timestamp_(0),
  25       sequence_number_(0),
  26       marker_(false),
  27       payload_type_(0),
  28       ssrc_(0) {}
  29
  30 void RtpPacketBuilder::SetKeyFrame(bool is_key) { is_key_ = is_key; }
  31
  32 void RtpPacketBuilder::SetFrameId(uint32 frame_id) { frame_id_ = frame_id; }
  33
  34 void RtpPacketBuilder::SetPacketId(uint16 packet_id) { packet_id_ = packet_id; }
  35
  36 void RtpPacketBuilder::SetMaxPacketId(uint16 max_packet_id) {
  37   max_packet_id_ = max_packet_id;
  38 }
  39
  40 void RtpPacketBuilder::SetReferenceFrameId(uint32 reference_frame_id,
  41                                            bool is_set) {
  42   is_reference_set_ = is_set;
  43   if (is_set)
  44     reference_frame_id_ = reference_frame_id;
  45 }
  46 void RtpPacketBuilder::SetTimestamp(uint32 timestamp) {
  47   timestamp_ = timestamp;
  48 }
  49
  50 void RtpPacketBuilder::SetSequenceNumber(uint16 sequence_number) {
  51   sequence_number_ = sequence_number;
  52 }
  53
  54 void RtpPacketBuilder::SetMarkerBit(bool marker) { marker_ = marker; }
  55
  56 void RtpPacketBuilder::SetPayloadType(int payload_type) {
  57   payload_type_ = payload_type;
  58 }
  59
  60 void RtpPacketBuilder::SetSsrc(uint32 ssrc) { ssrc_ = ssrc; }
  61
  62 void RtpPacketBuilder::BuildHeader(uint8* data, uint32 data_length) {
  63   BuildCommonHeader(data, data_length);
  64   BuildCastHeader(data + kGenericRtpHeaderLength,
  65                   data_length - kGenericRtpHeaderLength);
  66 }
  67
  68 void RtpPacketBuilder::BuildCastHeader(uint8* data, uint32 data_length) {
  69   // Build header.
  70   DCHECK_LE(kCastRtpHeaderLength, data_length);
  71   // Set the first 7 bytes to 0.
  72   memset(data, 0, kCastRtpHeaderLength);
  73   net::BigEndianWriter big_endian_writer(data, 56);
  74   big_endian_writer.WriteU8((is_key_ ? 0x80 : 0) |
  75                             (is_reference_set_ ? 0x40 : 0));
  76   big_endian_writer.WriteU8(frame_id_);
  77   big_endian_writer.WriteU16(packet_id_);
  78   big_endian_writer.WriteU16(max_packet_id_);
  79   if (is_reference_set_) {
  80     big_endian_writer.WriteU8(reference_frame_id_);
  81   }
  82 }
  83
  84 void RtpPacketBuilder::BuildCommonHeader(uint8* data, uint32 data_length) {
  85   DCHECK_LE(kGenericRtpHeaderLength, data_length);
  86   net::BigEndianWriter big_endian_writer(data, 96);
  87   big_endian_writer.WriteU8(0x80);
  88   big_endian_writer.WriteU8(payload_type_ | (marker_ ? kRtpMarkerBitMask : 0));
  89   big_endian_writer.WriteU16(sequence_number_);
  90   big_endian_writer.WriteU32(timestamp_);
  91   big_endian_writer.WriteU32(ssrc_);
  92 }
  93
  94 }  // namespace cast
  95 }  // namespace media