| 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 |