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