OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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/transport/rtcp/rtcp_builder.h" | 5 #include "media/cast/transport/rtcp/rtcp_builder.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <string> | 8 #include <string> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
48 packet_type_flags & kRtcpRpsi || | 48 packet_type_flags & kRtcpRpsi || |
49 packet_type_flags & kRtcpRemb || | 49 packet_type_flags & kRtcpRemb || |
50 packet_type_flags & kRtcpNack) { | 50 packet_type_flags & kRtcpNack) { |
51 NOTREACHED() << "Invalid argument"; | 51 NOTREACHED() << "Invalid argument"; |
52 } | 52 } |
53 ssrc_ = sending_ssrc; | 53 ssrc_ = sending_ssrc; |
54 c_name_ = c_name; | 54 c_name_ = c_name; |
55 Packet packet; | 55 Packet packet; |
56 packet.reserve(kMaxIpPacketSize); | 56 packet.reserve(kMaxIpPacketSize); |
57 if (packet_type_flags & kRtcpSr) { | 57 if (packet_type_flags & kRtcpSr) { |
58 BuildSR(sender_info, &packet); | 58 if (!BuildSR(sender_info, &packet)) return; |
59 BuildSdec(&packet); | 59 if (!BuildSdec(&packet)) return; |
60 } | 60 } |
61 if (packet_type_flags & kRtcpBye) { | 61 if (packet_type_flags & kRtcpBye) { |
62 BuildBye(&packet); | 62 if (!BuildBye(&packet)) return; |
63 } | 63 } |
64 if (packet_type_flags & kRtcpDlrr) { | 64 if (packet_type_flags & kRtcpDlrr) { |
65 BuildDlrrRb(dlrr, &packet); | 65 if (!BuildDlrrRb(dlrr, &packet)) return; |
66 } | 66 } |
67 if (packet_type_flags & kRtcpSenderLog) { | 67 if (packet_type_flags & kRtcpSenderLog) { |
68 BuildSenderLog(sender_log, &packet); | 68 if (!BuildSenderLog(sender_log, &packet)) return; |
69 } | 69 } |
70 if (packet.empty()) | 70 if (packet.empty()) |
71 return; // Sanity - don't send empty packets. | 71 return; // Sanity - don't send empty packets. |
72 | 72 |
73 transport_->SendRtcpPacket(packet); | 73 transport_->SendRtcpPacket(packet); |
74 } | 74 } |
75 | 75 |
76 void RtcpBuilder::BuildSR(const RtcpSenderInfo& sender_info, | 76 bool RtcpBuilder::BuildSR(const RtcpSenderInfo& sender_info, |
77 Packet* packet) const { | 77 Packet* packet) const { |
78 // Sender report. | 78 // Sender report. |
79 size_t start_size = packet->size(); | 79 size_t start_size = packet->size(); |
80 DCHECK_LT(start_size + 52, kMaxIpPacketSize) << "Not enough buffer space"; | 80 if (start_size + 52 > kMaxIpPacketSize) { |
81 if (start_size + 52 > kMaxIpPacketSize) return; | 81 DLOG(FATAL) << "Not enough buffer space"; |
| 82 return false; |
| 83 } |
82 | 84 |
83 uint16 number_of_rows = 6; | 85 uint16 number_of_rows = 6; |
84 packet->resize(start_size + 28); | 86 packet->resize(start_size + 28); |
85 | 87 |
86 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 28); | 88 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 28); |
87 big_endian_writer.WriteU8(0x80); | 89 big_endian_writer.WriteU8(0x80); |
88 big_endian_writer.WriteU8(kPacketTypeSenderReport); | 90 big_endian_writer.WriteU8(kPacketTypeSenderReport); |
89 big_endian_writer.WriteU16(number_of_rows); | 91 big_endian_writer.WriteU16(number_of_rows); |
90 big_endian_writer.WriteU32(ssrc_); | 92 big_endian_writer.WriteU32(ssrc_); |
91 big_endian_writer.WriteU32(sender_info.ntp_seconds); | 93 big_endian_writer.WriteU32(sender_info.ntp_seconds); |
92 big_endian_writer.WriteU32(sender_info.ntp_fraction); | 94 big_endian_writer.WriteU32(sender_info.ntp_fraction); |
93 big_endian_writer.WriteU32(sender_info.rtp_timestamp); | 95 big_endian_writer.WriteU32(sender_info.rtp_timestamp); |
94 big_endian_writer.WriteU32(sender_info.send_packet_count); | 96 big_endian_writer.WriteU32(sender_info.send_packet_count); |
95 big_endian_writer.WriteU32(static_cast<uint32>(sender_info.send_octet_count)); | 97 big_endian_writer.WriteU32(static_cast<uint32>(sender_info.send_octet_count)); |
| 98 return true; |
96 } | 99 } |
97 | 100 |
98 void RtcpBuilder::BuildSdec(Packet* packet) const { | 101 bool RtcpBuilder::BuildSdec(Packet* packet) const { |
99 size_t start_size = packet->size(); | 102 size_t start_size = packet->size(); |
100 DCHECK_LT(start_size + 12 + c_name_.length(), kMaxIpPacketSize) | 103 if (start_size + 12 + c_name_.length() > kMaxIpPacketSize) { |
101 << "Not enough buffer space"; | 104 DLOG(FATAL) << "Not enough buffer space"; |
102 if (start_size + 12 > kMaxIpPacketSize) return; | 105 return false; |
| 106 } |
103 | 107 |
104 // SDES Source Description. | 108 // SDES Source Description. |
105 packet->resize(start_size + 10); | 109 packet->resize(start_size + 10); |
106 | 110 |
107 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 10); | 111 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 10); |
108 // We always need to add one SDES CNAME. | 112 // We always need to add one SDES CNAME. |
109 big_endian_writer.WriteU8(0x80 + 1); | 113 big_endian_writer.WriteU8(0x80 + 1); |
110 big_endian_writer.WriteU8(kPacketTypeSdes); | 114 big_endian_writer.WriteU8(kPacketTypeSdes); |
111 | 115 |
112 // Handle SDES length later on. | 116 // Handle SDES length later on. |
(...skipping 16 matching lines...) Expand all Loading... |
129 } | 133 } |
130 while ((packet->size() % 4) != 0) { | 134 while ((packet->size() % 4) != 0) { |
131 padding++; | 135 padding++; |
132 packet->push_back(0); | 136 packet->push_back(0); |
133 } | 137 } |
134 sdes_length += padding; | 138 sdes_length += padding; |
135 | 139 |
136 // In 32-bit words minus one and we don't count the header. | 140 // In 32-bit words minus one and we don't count the header. |
137 uint8 buffer_length = static_cast<uint8>((sdes_length / 4) - 1); | 141 uint8 buffer_length = static_cast<uint8>((sdes_length / 4) - 1); |
138 (*packet)[sdes_length_position] = buffer_length; | 142 (*packet)[sdes_length_position] = buffer_length; |
| 143 return true; |
139 } | 144 } |
140 | 145 |
141 void RtcpBuilder::BuildBye(Packet* packet) const { | 146 bool RtcpBuilder::BuildBye(Packet* packet) const { |
142 size_t start_size = packet->size(); | 147 size_t start_size = packet->size(); |
143 DCHECK_LT(start_size + 8, kMaxIpPacketSize) << "Not enough buffer space"; | 148 if (start_size + 8 > kMaxIpPacketSize) { |
144 if (start_size + 8 > kMaxIpPacketSize) return; | 149 DLOG(FATAL) << "Not enough buffer space"; |
| 150 return false; |
| 151 } |
145 | 152 |
146 packet->resize(start_size + 8); | 153 packet->resize(start_size + 8); |
147 | 154 |
148 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 8); | 155 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 8); |
149 big_endian_writer.WriteU8(0x80 + 1); | 156 big_endian_writer.WriteU8(0x80 + 1); |
150 big_endian_writer.WriteU8(kPacketTypeBye); | 157 big_endian_writer.WriteU8(kPacketTypeBye); |
151 big_endian_writer.WriteU16(1); // Length. | 158 big_endian_writer.WriteU16(1); // Length. |
152 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. | 159 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. |
| 160 return true; |
153 } | 161 } |
154 | 162 |
155 /* | 163 /* |
156 0 1 2 3 | 164 0 1 2 3 |
157 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 | 165 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 |
158 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | 166 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
159 |V=2|P|reserved | PT=XR=207 | length | | 167 |V=2|P|reserved | PT=XR=207 | length | |
160 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | 168 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
161 | SSRC | | 169 | SSRC | |
162 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | 170 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
163 | BT=5 | reserved | block length | | 171 | BT=5 | reserved | block length | |
164 +=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+ | 172 +=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+ |
165 | SSRC_1 (SSRC of first receiver) | sub- | 173 | SSRC_1 (SSRC of first receiver) | sub- |
166 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ block | 174 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ block |
167 | last RR (LRR) | 1 | 175 | last RR (LRR) | 1 |
168 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | 176 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
169 | delay since last RR (DLRR) | | 177 | delay since last RR (DLRR) | |
170 +=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+ | 178 +=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+=+ |
171 */ | 179 */ |
172 void RtcpBuilder::BuildDlrrRb(const RtcpDlrrReportBlock& dlrr, | 180 bool RtcpBuilder::BuildDlrrRb(const RtcpDlrrReportBlock& dlrr, |
173 Packet* packet) const { | 181 Packet* packet) const { |
174 size_t start_size = packet->size(); | 182 size_t start_size = packet->size(); |
175 DCHECK_LT(start_size + 24, kMaxIpPacketSize) << "Not enough buffer space"; | 183 if (start_size + 24 > kMaxIpPacketSize) { |
176 if (start_size + 24 > kMaxIpPacketSize) return; | 184 DLOG(FATAL) << "Not enough buffer space"; |
| 185 return false; |
| 186 } |
177 | 187 |
178 packet->resize(start_size + 24); | 188 packet->resize(start_size + 24); |
179 | 189 |
180 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 24); | 190 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), 24); |
181 big_endian_writer.WriteU8(0x80); | 191 big_endian_writer.WriteU8(0x80); |
182 big_endian_writer.WriteU8(kPacketTypeXr); | 192 big_endian_writer.WriteU8(kPacketTypeXr); |
183 big_endian_writer.WriteU16(5); // Length. | 193 big_endian_writer.WriteU16(5); // Length. |
184 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. | 194 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. |
185 big_endian_writer.WriteU8(5); // Add block type. | 195 big_endian_writer.WriteU8(5); // Add block type. |
186 big_endian_writer.WriteU8(0); // Add reserved. | 196 big_endian_writer.WriteU8(0); // Add reserved. |
187 big_endian_writer.WriteU16(3); // Block length. | 197 big_endian_writer.WriteU16(3); // Block length. |
188 big_endian_writer.WriteU32(ssrc_); // Add the media (received RTP) SSRC. | 198 big_endian_writer.WriteU32(ssrc_); // Add the media (received RTP) SSRC. |
189 big_endian_writer.WriteU32(dlrr.last_rr); | 199 big_endian_writer.WriteU32(dlrr.last_rr); |
190 big_endian_writer.WriteU32(dlrr.delay_since_last_rr); | 200 big_endian_writer.WriteU32(dlrr.delay_since_last_rr); |
| 201 return true; |
191 } | 202 } |
192 | 203 |
193 void RtcpBuilder::BuildSenderLog(const RtcpSenderLogMessage& sender_log_message, | 204 bool RtcpBuilder::BuildSenderLog(const RtcpSenderLogMessage& sender_log_message, |
194 Packet* packet) const { | 205 Packet* packet) const { |
195 DCHECK(packet); | 206 DCHECK(packet); |
196 size_t start_size = packet->size(); | 207 size_t start_size = packet->size(); |
197 size_t remaining_space = kMaxIpPacketSize - start_size; | 208 size_t remaining_space = kMaxIpPacketSize - start_size; |
198 DCHECK_GE(remaining_space, kRtcpCastLogHeaderSize + kRtcpSenderFrameLogSize) | 209 if (remaining_space < kRtcpCastLogHeaderSize + kRtcpSenderFrameLogSize) { |
199 << "Not enough buffer space"; | 210 DLOG(FATAL) << "Not enough buffer space"; |
200 if (remaining_space < kRtcpCastLogHeaderSize + kRtcpSenderFrameLogSize) | 211 return false; |
201 return; | 212 } |
202 | 213 |
203 size_t space_for_x_messages = | 214 size_t space_for_x_messages = |
204 (remaining_space - kRtcpCastLogHeaderSize) / kRtcpSenderFrameLogSize; | 215 (remaining_space - kRtcpCastLogHeaderSize) / kRtcpSenderFrameLogSize; |
205 size_t number_of_messages = std::min(space_for_x_messages, | 216 size_t number_of_messages = std::min(space_for_x_messages, |
206 sender_log_message.size()); | 217 sender_log_message.size()); |
207 | 218 |
208 size_t log_size = kRtcpCastLogHeaderSize + | 219 size_t log_size = kRtcpCastLogHeaderSize + |
209 number_of_messages * kRtcpSenderFrameLogSize; | 220 number_of_messages * kRtcpSenderFrameLogSize; |
210 packet->resize(start_size + log_size); | 221 packet->resize(start_size + log_size); |
211 | 222 |
212 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), log_size); | 223 net::BigEndianWriter big_endian_writer(&((*packet)[start_size]), log_size); |
213 big_endian_writer.WriteU8(0x80 + kSenderLogSubtype); | 224 big_endian_writer.WriteU8(0x80 + kSenderLogSubtype); |
214 big_endian_writer.WriteU8(kPacketTypeApplicationDefined); | 225 big_endian_writer.WriteU8(kPacketTypeApplicationDefined); |
215 big_endian_writer.WriteU16(static_cast<uint16>(2 + number_of_messages)); | 226 big_endian_writer.WriteU16(static_cast<uint16>(2 + number_of_messages)); |
216 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. | 227 big_endian_writer.WriteU32(ssrc_); // Add our own SSRC. |
217 big_endian_writer.WriteU32(kCast); | 228 big_endian_writer.WriteU32(kCast); |
218 | 229 |
219 std::list<RtcpSenderFrameLogMessage>::const_iterator it = | 230 std::vector<RtcpSenderFrameLogMessage>::const_iterator it = |
220 sender_log_message.begin(); | 231 sender_log_message.begin(); |
221 for (; number_of_messages > 0; --number_of_messages) { | 232 for (; number_of_messages > 0; --number_of_messages) { |
222 DCHECK(!sender_log_message.empty()); | 233 DCHECK(!sender_log_message.empty()); |
223 const RtcpSenderFrameLogMessage& message = *it; | 234 const RtcpSenderFrameLogMessage& message = *it; |
224 big_endian_writer.WriteU8(static_cast<uint8>(message.frame_status)); | 235 big_endian_writer.WriteU8(static_cast<uint8>(message.frame_status)); |
225 // We send the 24 east significant bits of the RTP timestamp. | 236 // We send the 24 east significant bits of the RTP timestamp. |
226 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp >> 16)); | 237 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp >> 16)); |
227 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp >> 8)); | 238 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp >> 8)); |
228 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp)); | 239 big_endian_writer.WriteU8(static_cast<uint8>(message.rtp_timestamp)); |
229 ++it; | 240 ++it; |
230 } | 241 } |
| 242 return true; |
231 } | 243 } |
232 | 244 |
233 } // namespace transport | 245 } // namespace transport |
234 } // namespace cast | 246 } // namespace cast |
235 } // namespace media | 247 } // namespace media |
OLD | NEW |