OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet.h" | 11 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet.h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 | 14 |
15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
16 #include "webrtc/base/logging.h" | 16 #include "webrtc/base/logging.h" |
17 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" | 17 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
18 | 18 |
19 using webrtc::RTCPUtility::PT_APP; | 19 using webrtc::RTCPUtility::PT_APP; |
20 using webrtc::RTCPUtility::PT_IJ; | 20 using webrtc::RTCPUtility::PT_IJ; |
21 using webrtc::RTCPUtility::PT_PSFB; | |
22 using webrtc::RTCPUtility::PT_RTPFB; | 21 using webrtc::RTCPUtility::PT_RTPFB; |
23 using webrtc::RTCPUtility::PT_SDES; | 22 using webrtc::RTCPUtility::PT_SDES; |
24 using webrtc::RTCPUtility::PT_SR; | 23 using webrtc::RTCPUtility::PT_SR; |
25 | 24 |
26 using webrtc::RTCPUtility::RTCPPacketAPP; | 25 using webrtc::RTCPUtility::RTCPPacketAPP; |
27 using webrtc::RTCPUtility::RTCPPacketPSFBRPSI; | |
28 using webrtc::RTCPUtility::RTCPPacketReportBlockItem; | 26 using webrtc::RTCPUtility::RTCPPacketReportBlockItem; |
29 using webrtc::RTCPUtility::RTCPPacketRTPFBNACK; | 27 using webrtc::RTCPUtility::RTCPPacketRTPFBNACK; |
30 using webrtc::RTCPUtility::RTCPPacketRTPFBNACKItem; | 28 using webrtc::RTCPUtility::RTCPPacketRTPFBNACKItem; |
31 using webrtc::RTCPUtility::RTCPPacketSR; | 29 using webrtc::RTCPUtility::RTCPPacketSR; |
32 | 30 |
33 namespace webrtc { | 31 namespace webrtc { |
34 namespace rtcp { | 32 namespace rtcp { |
35 namespace { | 33 namespace { |
36 void AssignUWord8(uint8_t* buffer, size_t* offset, uint8_t value) { | 34 void AssignUWord8(uint8_t* buffer, size_t* offset, uint8_t value) { |
37 buffer[(*offset)++] = value; | 35 buffer[(*offset)++] = value; |
(...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
135 it != chunks.end(); ++it) { | 133 it != chunks.end(); ++it) { |
136 AssignUWord32(buffer, pos, (*it).ssrc); | 134 AssignUWord32(buffer, pos, (*it).ssrc); |
137 AssignUWord8(buffer, pos, kSdesItemType); | 135 AssignUWord8(buffer, pos, kSdesItemType); |
138 AssignUWord8(buffer, pos, (*it).name.length()); | 136 AssignUWord8(buffer, pos, (*it).name.length()); |
139 memcpy(buffer + *pos, (*it).name.data(), (*it).name.length()); | 137 memcpy(buffer + *pos, (*it).name.data(), (*it).name.length()); |
140 *pos += (*it).name.length(); | 138 *pos += (*it).name.length(); |
141 memset(buffer + *pos, 0, (*it).null_octets); | 139 memset(buffer + *pos, 0, (*it).null_octets); |
142 *pos += (*it).null_octets; | 140 *pos += (*it).null_octets; |
143 } | 141 } |
144 } | 142 } |
145 | |
146 // Reference picture selection indication (RPSI) (RFC 4585). | |
147 // | |
148 // FCI: | |
149 // | |
150 // 0 1 2 3 | |
151 // 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 | |
152 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | |
153 // | PB |0| Payload Type| Native RPSI bit string | | |
154 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | |
155 // | defined per codec ... | Padding (0) | | |
156 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | |
157 | |
158 void CreateRpsi(const RTCPPacketPSFBRPSI& rpsi, | |
159 uint8_t padding_bytes, | |
160 uint8_t* buffer, | |
161 size_t* pos) { | |
162 // Native bit string should be a multiple of 8 bits. | |
163 assert(rpsi.NumberOfValidBits % 8 == 0); | |
164 AssignUWord32(buffer, pos, rpsi.SenderSSRC); | |
165 AssignUWord32(buffer, pos, rpsi.MediaSSRC); | |
166 AssignUWord8(buffer, pos, padding_bytes * 8); | |
167 AssignUWord8(buffer, pos, rpsi.PayloadType); | |
168 memcpy(buffer + *pos, rpsi.NativeBitString, rpsi.NumberOfValidBits / 8); | |
169 *pos += rpsi.NumberOfValidBits / 8; | |
170 memset(buffer + *pos, 0, padding_bytes); | |
171 *pos += padding_bytes; | |
172 } | |
173 } // namespace | 143 } // namespace |
174 | 144 |
175 void RtcpPacket::Append(RtcpPacket* packet) { | 145 void RtcpPacket::Append(RtcpPacket* packet) { |
176 assert(packet); | 146 assert(packet); |
177 appended_packets_.push_back(packet); | 147 appended_packets_.push_back(packet); |
178 } | 148 } |
179 | 149 |
180 rtc::scoped_ptr<RawPacket> RtcpPacket::Build() const { | 150 rtc::scoped_ptr<RawPacket> RtcpPacket::Build() const { |
181 size_t length = 0; | 151 size_t length = 0; |
182 rtc::scoped_ptr<RawPacket> packet(new RawPacket(IP_PACKET_SIZE)); | 152 rtc::scoped_ptr<RawPacket> packet(new RawPacket(IP_PACKET_SIZE)); |
(...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
328 // Header (4 bytes). | 298 // Header (4 bytes). |
329 // Chunk: | 299 // Chunk: |
330 // SSRC/CSRC (4 bytes) | CNAME (1 byte) | length (1 byte) | name | padding. | 300 // SSRC/CSRC (4 bytes) | CNAME (1 byte) | length (1 byte) | name | padding. |
331 size_t length = kHeaderLength; | 301 size_t length = kHeaderLength; |
332 for (const Chunk& chunk : chunks_) | 302 for (const Chunk& chunk : chunks_) |
333 length += 6 + chunk.name.length() + chunk.null_octets; | 303 length += 6 + chunk.name.length() + chunk.null_octets; |
334 assert(length % 4 == 0); | 304 assert(length % 4 == 0); |
335 return length; | 305 return length; |
336 } | 306 } |
337 | 307 |
338 bool Rpsi::Create(uint8_t* packet, | |
339 size_t* index, | |
340 size_t max_length, | |
341 RtcpPacket::PacketReadyCallback* callback) const { | |
342 assert(rpsi_.NumberOfValidBits > 0); | |
343 while (*index + BlockLength() > max_length) { | |
344 if (!OnBufferFull(packet, index, callback)) | |
345 return false; | |
346 } | |
347 const uint8_t kFmt = 3; | |
348 CreateHeader(kFmt, PT_PSFB, HeaderLength(), packet, index); | |
349 CreateRpsi(rpsi_, padding_bytes_, packet, index); | |
350 return true; | |
351 } | |
352 | |
353 void Rpsi::WithPictureId(uint64_t picture_id) { | |
354 const uint32_t kPidBits = 7; | |
355 const uint64_t k7MsbZeroMask = 0x1ffffffffffffffULL; | |
356 uint8_t required_bytes = 0; | |
357 uint64_t shifted_pid = picture_id; | |
358 do { | |
359 ++required_bytes; | |
360 shifted_pid = (shifted_pid >> kPidBits) & k7MsbZeroMask; | |
361 } while (shifted_pid > 0); | |
362 | |
363 // Convert picture id to native bit string (natively defined by the video | |
364 // codec). | |
365 int pos = 0; | |
366 for (int i = required_bytes - 1; i > 0; i--) { | |
367 rpsi_.NativeBitString[pos++] = | |
368 0x80 | static_cast<uint8_t>(picture_id >> (i * kPidBits)); | |
369 } | |
370 rpsi_.NativeBitString[pos++] = static_cast<uint8_t>(picture_id & 0x7f); | |
371 rpsi_.NumberOfValidBits = pos * 8; | |
372 | |
373 // Calculate padding bytes (to reach next 32-bit boundary, 1, 2 or 3 bytes). | |
374 padding_bytes_ = 4 - ((2 + required_bytes) % 4); | |
375 if (padding_bytes_ == 4) { | |
376 padding_bytes_ = 0; | |
377 } | |
378 } | |
379 | |
380 RawPacket::RawPacket(size_t buffer_length) | 308 RawPacket::RawPacket(size_t buffer_length) |
381 : buffer_length_(buffer_length), length_(0) { | 309 : buffer_length_(buffer_length), length_(0) { |
382 buffer_.reset(new uint8_t[buffer_length]); | 310 buffer_.reset(new uint8_t[buffer_length]); |
383 } | 311 } |
384 | 312 |
385 RawPacket::RawPacket(const uint8_t* packet, size_t packet_length) | 313 RawPacket::RawPacket(const uint8_t* packet, size_t packet_length) |
386 : buffer_length_(packet_length), length_(packet_length) { | 314 : buffer_length_(packet_length), length_(packet_length) { |
387 buffer_.reset(new uint8_t[packet_length]); | 315 buffer_.reset(new uint8_t[packet_length]); |
388 memcpy(buffer_.get(), packet, packet_length); | 316 memcpy(buffer_.get(), packet, packet_length); |
389 } | 317 } |
(...skipping 14 matching lines...) Expand all Loading... |
404 return length_; | 332 return length_; |
405 } | 333 } |
406 | 334 |
407 void RawPacket::SetLength(size_t length) { | 335 void RawPacket::SetLength(size_t length) { |
408 assert(length <= buffer_length_); | 336 assert(length <= buffer_length_); |
409 length_ = length; | 337 length_ = length; |
410 } | 338 } |
411 | 339 |
412 } // namespace rtcp | 340 } // namespace rtcp |
413 } // namespace webrtc | 341 } // namespace webrtc |
OLD | NEW |