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