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 <string> | 7 #include <string> |
8 | 8 |
9 #include "base/big_endian.h" | 9 #include "base/big_endian.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
(...skipping 11 matching lines...) Expand all Loading... |
22 | 22 |
23 RtpPacketizerConfig::~RtpPacketizerConfig() {} | 23 RtpPacketizerConfig::~RtpPacketizerConfig() {} |
24 | 24 |
25 RtpPacketizer::RtpPacketizer(PacedSender* const transport, | 25 RtpPacketizer::RtpPacketizer(PacedSender* const transport, |
26 PacketStorage* packet_storage, | 26 PacketStorage* packet_storage, |
27 RtpPacketizerConfig rtp_packetizer_config) | 27 RtpPacketizerConfig rtp_packetizer_config) |
28 : config_(rtp_packetizer_config), | 28 : config_(rtp_packetizer_config), |
29 transport_(transport), | 29 transport_(transport), |
30 packet_storage_(packet_storage), | 30 packet_storage_(packet_storage), |
31 sequence_number_(config_.sequence_number), | 31 sequence_number_(config_.sequence_number), |
32 rtp_timestamp_(0), | |
33 packet_id_(0), | |
34 send_packet_count_(0), | 32 send_packet_count_(0), |
35 send_octet_count_(0) { | 33 send_octet_count_(0) { |
36 DCHECK(transport) << "Invalid argument"; | 34 DCHECK(transport) << "Invalid argument"; |
37 } | 35 } |
38 | 36 |
39 RtpPacketizer::~RtpPacketizer() {} | 37 RtpPacketizer::~RtpPacketizer() {} |
40 | 38 |
41 uint16_t RtpPacketizer::NextSequenceNumber() { | 39 uint16_t RtpPacketizer::NextSequenceNumber() { |
42 ++sequence_number_; | 40 ++sequence_number_; |
43 return sequence_number_ - 1; | 41 return sequence_number_ - 1; |
44 } | 42 } |
45 | 43 |
46 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { | 44 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { |
47 uint16_t rtp_header_length = kRtpHeaderLength + kCastHeaderLength; | 45 uint16_t rtp_header_length = kRtpHeaderLength + kCastHeaderLength; |
48 uint16_t max_length = config_.max_payload_length - rtp_header_length - 1; | 46 uint16_t max_length = config_.max_payload_length - rtp_header_length - 1; |
49 rtp_timestamp_ = frame.rtp_timestamp; | |
50 | 47 |
51 // Split the payload evenly (round number up). | 48 // Split the payload evenly (round number up). |
52 size_t num_packets = (frame.data.size() + max_length) / max_length; | 49 size_t num_packets = (frame.data.size() + max_length) / max_length; |
53 size_t payload_length = (frame.data.size() + num_packets) / num_packets; | 50 size_t payload_length = (frame.data.size() + num_packets) / num_packets; |
54 DCHECK_LE(payload_length, max_length) << "Invalid argument"; | 51 DCHECK_LE(payload_length, max_length) << "Invalid argument"; |
55 | 52 |
56 SendPacketVector packets; | 53 SendPacketVector packets; |
57 | 54 |
58 size_t remaining_size = frame.data.size(); | 55 size_t remaining_size = frame.data.size(); |
59 std::string::const_iterator data_iter = frame.data.begin(); | 56 std::string::const_iterator data_iter = frame.data.begin(); |
(...skipping 13 matching lines...) Expand all Loading... |
73 BuildCommonRTPheader( | 70 BuildCommonRTPheader( |
74 &packet->data, remaining_size == 0, frame.rtp_timestamp); | 71 &packet->data, remaining_size == 0, frame.rtp_timestamp); |
75 | 72 |
76 // Build Cast header. | 73 // Build Cast header. |
77 // TODO(miu): Should we always set the ref frame bit and the ref_frame_id? | 74 // TODO(miu): Should we always set the ref frame bit and the ref_frame_id? |
78 DCHECK_NE(frame.dependency, EncodedFrame::UNKNOWN_DEPENDENCY); | 75 DCHECK_NE(frame.dependency, EncodedFrame::UNKNOWN_DEPENDENCY); |
79 uint8_t byte0 = kCastReferenceFrameIdBitMask; | 76 uint8_t byte0 = kCastReferenceFrameIdBitMask; |
80 if (frame.dependency == EncodedFrame::KEY) | 77 if (frame.dependency == EncodedFrame::KEY) |
81 byte0 |= kCastKeyFrameBitMask; | 78 byte0 |= kCastKeyFrameBitMask; |
82 // Extensions only go on the first packet of the frame | 79 // Extensions only go on the first packet of the frame |
83 if (packet_id_ == 0) | 80 const uint16_t packet_id = static_cast<uint16_t>(packets.size()); |
| 81 if (packet_id == 0) |
84 byte0 |= num_extensions; | 82 byte0 |= num_extensions; |
85 packet->data.push_back(byte0); | 83 packet->data.push_back(byte0); |
86 packet->data.push_back(static_cast<uint8_t>(frame.frame_id)); | 84 packet->data.push_back(static_cast<uint8_t>(frame.frame_id)); |
87 size_t start_size = packet->data.size(); | 85 size_t start_size = packet->data.size(); |
88 packet->data.resize(start_size + 4); | 86 packet->data.resize(start_size + 4); |
89 base::BigEndianWriter big_endian_writer( | 87 base::BigEndianWriter big_endian_writer( |
90 reinterpret_cast<char*>(&(packet->data[start_size])), 4); | 88 reinterpret_cast<char*>(&(packet->data[start_size])), 4); |
91 big_endian_writer.WriteU16(packet_id_); | 89 big_endian_writer.WriteU16(packet_id); |
92 big_endian_writer.WriteU16(static_cast<uint16_t>(num_packets - 1)); | 90 big_endian_writer.WriteU16(static_cast<uint16_t>(num_packets - 1)); |
93 packet->data.push_back(static_cast<uint8_t>(frame.referenced_frame_id)); | 91 packet->data.push_back(static_cast<uint8_t>(frame.referenced_frame_id)); |
94 // Add extension details only on the first packet of the frame | 92 // Add extension details only on the first packet of the frame |
95 if (packet_id_ == 0 && frame.new_playout_delay_ms) { | 93 if (packet_id == 0 && frame.new_playout_delay_ms) { |
96 packet->data.push_back(kCastRtpExtensionAdaptiveLatency << 2); | 94 packet->data.push_back(kCastRtpExtensionAdaptiveLatency << 2); |
97 packet->data.push_back(2); // 2 bytes | 95 packet->data.push_back(2); // 2 bytes |
98 packet->data.push_back( | 96 packet->data.push_back( |
99 static_cast<uint8_t>(frame.new_playout_delay_ms >> 8)); | 97 static_cast<uint8_t>(frame.new_playout_delay_ms >> 8)); |
100 packet->data.push_back(static_cast<uint8_t>(frame.new_playout_delay_ms)); | 98 packet->data.push_back(static_cast<uint8_t>(frame.new_playout_delay_ms)); |
101 } | 99 } |
102 | 100 |
103 // Copy payload data. | 101 // Copy payload data. |
104 packet->data.insert(packet->data.end(), | 102 packet->data.insert(packet->data.end(), |
105 data_iter, | 103 data_iter, |
106 data_iter + payload_length); | 104 data_iter + payload_length); |
107 data_iter += payload_length; | 105 data_iter += payload_length; |
108 | 106 |
109 const PacketKey key = PacedPacketSender::MakePacketKey( | 107 const PacketKey key = PacedPacketSender::MakePacketKey( |
110 PacketKey::RTP, frame.frame_id, config_.ssrc, packet_id_++); | 108 PacketKey::RTP, frame.frame_id, config_.ssrc, packet_id); |
111 packets.push_back(make_pair(key, packet)); | 109 packets.push_back(make_pair(key, packet)); |
112 | 110 |
113 // Update stats. | 111 // Update stats. |
114 ++send_packet_count_; | 112 ++send_packet_count_; |
115 send_octet_count_ += payload_length; | 113 send_octet_count_ += payload_length; |
116 } | 114 } |
117 DCHECK(packet_id_ == num_packets) << "Invalid state"; | 115 DCHECK_EQ(num_packets, packets.size()) << "Invalid state"; |
118 | 116 |
119 packet_storage_->StoreFrame(frame.frame_id, packets); | 117 packet_storage_->StoreFrame(frame.frame_id, packets); |
120 | 118 |
121 // Send to network. | 119 // Send to network. |
122 transport_->SendPackets(packets); | 120 transport_->SendPackets(packets); |
123 | |
124 // Prepare for next frame. | |
125 packet_id_ = 0; | |
126 } | 121 } |
127 | 122 |
128 void RtpPacketizer::BuildCommonRTPheader(Packet* packet, | 123 void RtpPacketizer::BuildCommonRTPheader(Packet* packet, |
129 bool marker_bit, | 124 bool marker_bit, |
130 uint32_t time_stamp) { | 125 RtpTimeTicks rtp_timestamp) { |
131 packet->push_back(0x80); | 126 packet->push_back(0x80); |
132 packet->push_back(static_cast<uint8_t>(config_.payload_type) | | 127 packet->push_back(static_cast<uint8_t>(config_.payload_type) | |
133 (marker_bit ? kRtpMarkerBitMask : 0)); | 128 (marker_bit ? kRtpMarkerBitMask : 0)); |
134 size_t start_size = packet->size(); | 129 size_t start_size = packet->size(); |
135 packet->resize(start_size + 10); | 130 packet->resize(start_size + 10); |
136 base::BigEndianWriter big_endian_writer( | 131 base::BigEndianWriter big_endian_writer( |
137 reinterpret_cast<char*>(&((*packet)[start_size])), 10); | 132 reinterpret_cast<char*>(&((*packet)[start_size])), 10); |
138 big_endian_writer.WriteU16(sequence_number_); | 133 big_endian_writer.WriteU16(sequence_number_); |
139 big_endian_writer.WriteU32(time_stamp); | 134 big_endian_writer.WriteU32(rtp_timestamp.lower_32_bits()); |
140 big_endian_writer.WriteU32(config_.ssrc); | 135 big_endian_writer.WriteU32(config_.ssrc); |
141 ++sequence_number_; | 136 ++sequence_number_; |
142 } | 137 } |
143 | 138 |
144 } // namespace cast | 139 } // namespace cast |
145 } // namespace media | 140 } // namespace media |
OLD | NEW |