1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/cast/rtp_receiver/rtp_parser/test/rtp_packet_builder.h"
#include "base/logging.h"
#include "net/base/big_endian.h"
namespace media {
namespace cast {
const uint32 kCastRtpHeaderLength = 7;
const uint32 kGenericRtpHeaderLength = 12;
RtpPacketBuilder::RtpPacketBuilder()
: is_key_(false),
frame_id_(0),
packet_id_(0),
max_packet_id_(0),
reference_frame_id_(0),
is_reference_set_(false),
timestamp_(0),
sequence_number_(0),
marker_(false),
payload_type_(0),
ssrc_(0) {}
void RtpPacketBuilder::SetKeyFrame(bool is_key) {
is_key_ = is_key;
}
void RtpPacketBuilder::SetFrameId(uint8 frame_id) {
frame_id_ = frame_id;
}
void RtpPacketBuilder::SetPacketId(uint16 packet_id) {
packet_id_ = packet_id;
}
void RtpPacketBuilder::SetMaxPacketId(uint16 max_packet_id) {
max_packet_id_ = max_packet_id;
}
void RtpPacketBuilder::SetReferenceFrameId(uint8 reference_frame_id,
bool is_set) {
is_reference_set_ = is_set;
if (is_set)
reference_frame_id_ = reference_frame_id;
}
void RtpPacketBuilder::SetTimestamp(uint32 timestamp) {
timestamp_ = timestamp;
}
void RtpPacketBuilder::SetSequenceNumber(uint16 sequence_number) {
sequence_number_ = sequence_number;
}
void RtpPacketBuilder::SetMarkerBit(bool marker) {
marker_ = marker;
}
void RtpPacketBuilder::SetPayloadType(int payload_type) {
payload_type_ = payload_type;
}
void RtpPacketBuilder::SetSsrc(uint32 ssrc) {
ssrc_ = ssrc;
}
void RtpPacketBuilder::BuildHeader(uint8* data, uint32 data_length) {
BuildCommonHeader(data, data_length);
BuildCastHeader(data + kGenericRtpHeaderLength,
data_length - kGenericRtpHeaderLength);
}
void RtpPacketBuilder::BuildCastHeader(uint8* data, uint32 data_length) {
// Build header.
DCHECK_LE(kCastRtpHeaderLength, data_length);
// Set the first 7 bytes to 0.
memset(data, 0, kCastRtpHeaderLength);
net::BigEndianWriter big_endian_writer(data, 56);
big_endian_writer.WriteU8(
(is_key_ ? 0x80 : 0) | (is_reference_set_ ? 0x40 : 0));
big_endian_writer.WriteU8(frame_id_);
big_endian_writer.WriteU16(packet_id_);
big_endian_writer.WriteU16(max_packet_id_);
if (is_reference_set_) {
big_endian_writer.WriteU8(reference_frame_id_);
}
}
void RtpPacketBuilder::BuildCommonHeader(uint8* data, uint32 data_length) {
DCHECK_LE(kGenericRtpHeaderLength, data_length);
net::BigEndianWriter big_endian_writer(data, 96);
big_endian_writer.WriteU8(0x80);
big_endian_writer.WriteU8(payload_type_ | (marker_ ? kRtpMarkerBitMask : 0));
big_endian_writer.WriteU16(sequence_number_);
big_endian_writer.WriteU32(timestamp_);
big_endian_writer.WriteU32(ssrc_);
}
} // namespace cast
} // namespace media
|
