OLD | NEW |
---|---|
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/cast/net/rtp/rtp_packetizer.h" | 5 #include "media/cast/net/rtp/rtp_packetizer.h" |
6 | 6 |
7 #include "base/big_endian.h" | 7 #include "base/big_endian.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "media/cast/net/pacing/paced_sender.h" | 9 #include "media/cast/net/pacing/paced_sender.h" |
10 #include "media/cast/net/rtp/rtp_defines.h" | |
10 | 11 |
11 namespace media { | 12 namespace media { |
12 namespace cast { | 13 namespace cast { |
13 | 14 |
14 static const uint16 kCommonRtpHeaderLength = 12; | |
15 static const uint16 kCastRtpHeaderLength = 7; | |
16 static const uint8 kCastKeyFrameBitMask = 0x80; | |
17 static const uint8 kCastReferenceFrameIdBitMask = 0x40; | |
18 static const uint8 kRtpMarkerBitMask = 0x80; | |
19 | |
20 RtpPacketizerConfig::RtpPacketizerConfig() | 15 RtpPacketizerConfig::RtpPacketizerConfig() |
21 : payload_type(-1), | 16 : payload_type(-1), |
22 max_payload_length(kMaxIpPacketSize - 28), // Default is IP-v4/UDP. | 17 max_payload_length(kMaxIpPacketSize - 28), // Default is IP-v4/UDP. |
23 sequence_number(0), | 18 sequence_number(0), |
24 ssrc(0) {} | 19 ssrc(0) {} |
25 | 20 |
26 RtpPacketizerConfig::~RtpPacketizerConfig() {} | 21 RtpPacketizerConfig::~RtpPacketizerConfig() {} |
27 | 22 |
28 RtpPacketizer::RtpPacketizer(PacedSender* const transport, | 23 RtpPacketizer::RtpPacketizer(PacedSender* const transport, |
29 PacketStorage* packet_storage, | 24 PacketStorage* packet_storage, |
(...skipping 10 matching lines...) Expand all Loading... | |
40 } | 35 } |
41 | 36 |
42 RtpPacketizer::~RtpPacketizer() {} | 37 RtpPacketizer::~RtpPacketizer() {} |
43 | 38 |
44 uint16 RtpPacketizer::NextSequenceNumber() { | 39 uint16 RtpPacketizer::NextSequenceNumber() { |
45 ++sequence_number_; | 40 ++sequence_number_; |
46 return sequence_number_ - 1; | 41 return sequence_number_ - 1; |
47 } | 42 } |
48 | 43 |
49 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { | 44 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { |
50 uint16 rtp_header_length = kCommonRtpHeaderLength + kCastRtpHeaderLength; | 45 uint16 rtp_header_length = kRtpHeaderLength + kCastHeaderLength; |
51 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; | 46 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; |
52 rtp_timestamp_ = frame.rtp_timestamp; | 47 rtp_timestamp_ = frame.rtp_timestamp; |
53 | 48 |
54 // Split the payload evenly (round number up). | 49 // Split the payload evenly (round number up). |
55 size_t num_packets = (frame.data.size() + max_length) / max_length; | 50 size_t num_packets = (frame.data.size() + max_length) / max_length; |
56 size_t payload_length = (frame.data.size() + num_packets) / num_packets; | 51 size_t payload_length = (frame.data.size() + num_packets) / num_packets; |
57 DCHECK_LE(payload_length, max_length) << "Invalid argument"; | 52 DCHECK_LE(payload_length, max_length) << "Invalid argument"; |
58 | 53 |
59 SendPacketVector packets; | 54 SendPacketVector packets; |
60 | 55 |
61 size_t remaining_size = frame.data.size(); | 56 size_t remaining_size = frame.data.size(); |
62 std::string::const_iterator data_iter = frame.data.begin(); | 57 std::string::const_iterator data_iter = frame.data.begin(); |
63 while (remaining_size > 0) { | 58 while (remaining_size > 0) { |
64 PacketRef packet(new base::RefCountedData<Packet>); | 59 PacketRef packet(new base::RefCountedData<Packet>); |
65 | 60 |
66 if (remaining_size < payload_length) { | 61 if (remaining_size < payload_length) { |
67 payload_length = remaining_size; | 62 payload_length = remaining_size; |
68 } | 63 } |
69 remaining_size -= payload_length; | 64 remaining_size -= payload_length; |
70 BuildCommonRTPheader( | 65 BuildCommonRTPheader( |
71 &packet->data, remaining_size == 0, frame.rtp_timestamp); | 66 &packet->data, remaining_size == 0, frame.rtp_timestamp); |
72 | 67 |
73 // Build Cast header. | 68 // Build Cast header. |
74 // TODO(miu): Should we always set the ref frame bit and the ref_frame_id? | 69 // TODO(miu): Should we always set the ref frame bit and the ref_frame_id? |
75 DCHECK_NE(frame.dependency, EncodedFrame::UNKNOWN_DEPENDENCY); | 70 DCHECK_NE(frame.dependency, EncodedFrame::UNKNOWN_DEPENDENCY); |
76 packet->data.push_back( | 71 uint8 byte0 = kCastReferenceFrameIdBitMask; |
77 ((frame.dependency == EncodedFrame::KEY) ? kCastKeyFrameBitMask : 0) | | 72 if (frame.dependency == EncodedFrame::KEY) |
78 kCastReferenceFrameIdBitMask); | 73 byte0 |= kCastKeyFrameBitMask; |
74 if (frame.new_playout_delay_ms) { | |
75 // One extension block. | |
76 byte0 |= 1; | |
Alpha Left Google
2014/08/20 20:17:06
I suggest having an additional counter for the num
hubbe
2014/08/20 21:42:02
Total overkill, but ok.
| |
77 } | |
78 packet->data.push_back(byte0); | |
79 packet->data.push_back(static_cast<uint8>(frame.frame_id)); | 79 packet->data.push_back(static_cast<uint8>(frame.frame_id)); |
80 size_t start_size = packet->data.size(); | 80 size_t start_size = packet->data.size(); |
81 packet->data.resize(start_size + 4); | 81 packet->data.resize(start_size + 4); |
82 base::BigEndianWriter big_endian_writer( | 82 base::BigEndianWriter big_endian_writer( |
83 reinterpret_cast<char*>(&(packet->data[start_size])), 4); | 83 reinterpret_cast<char*>(&(packet->data[start_size])), 4); |
84 big_endian_writer.WriteU16(packet_id_); | 84 big_endian_writer.WriteU16(packet_id_); |
85 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); | 85 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); |
86 packet->data.push_back(static_cast<uint8>(frame.referenced_frame_id)); | 86 packet->data.push_back(static_cast<uint8>(frame.referenced_frame_id)); |
87 if (frame.new_playout_delay_ms) { | |
88 packet->data.push_back(kCastRTPExtensionAdaptiveLatency << 2); | |
89 packet->data.push_back(2); // 2 bytes | |
90 packet->data.push_back(frame.new_playout_delay_ms >> 8); | |
91 packet->data.push_back(frame.new_playout_delay_ms); | |
92 } | |
87 | 93 |
88 // Copy payload data. | 94 // Copy payload data. |
89 packet->data.insert(packet->data.end(), | 95 packet->data.insert(packet->data.end(), |
90 data_iter, | 96 data_iter, |
91 data_iter + payload_length); | 97 data_iter + payload_length); |
92 data_iter += payload_length; | 98 data_iter += payload_length; |
93 | 99 |
94 const PacketKey key = | 100 const PacketKey key = |
95 PacedPacketSender::MakePacketKey(frame.reference_time, | 101 PacedPacketSender::MakePacketKey(frame.reference_time, |
96 config_.ssrc, | 102 config_.ssrc, |
(...skipping 26 matching lines...) Expand all Loading... | |
123 base::BigEndianWriter big_endian_writer( | 129 base::BigEndianWriter big_endian_writer( |
124 reinterpret_cast<char*>(&((*packet)[start_size])), 10); | 130 reinterpret_cast<char*>(&((*packet)[start_size])), 10); |
125 big_endian_writer.WriteU16(sequence_number_); | 131 big_endian_writer.WriteU16(sequence_number_); |
126 big_endian_writer.WriteU32(time_stamp); | 132 big_endian_writer.WriteU32(time_stamp); |
127 big_endian_writer.WriteU32(config_.ssrc); | 133 big_endian_writer.WriteU32(config_.ssrc); |
128 ++sequence_number_; | 134 ++sequence_number_; |
129 } | 135 } |
130 | 136 |
131 } // namespace cast | 137 } // namespace cast |
132 } // namespace media | 138 } // namespace media |
OLD | NEW |