| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 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 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ | 10 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ |
| 11 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ | 11 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ |
| 12 | 12 |
| 13 #include <vector> | 13 #include <vector> |
| 14 | 14 |
| 15 #include "webrtc/base/basictypes.h" | 15 #include "webrtc/base/basictypes.h" |
| 16 #include "webrtc/base/buffer.h" | 16 #include "webrtc/base/copyonwritebuffer.h" |
| 17 #include "webrtc/base/constructormagic.h" | |
| 18 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 17 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
| 19 | 18 |
| 20 namespace webrtc { | 19 namespace webrtc { |
| 21 struct RTPHeader; | 20 struct RTPHeader; |
| 22 class RtpHeaderExtensionMap; | 21 class RtpHeaderExtensionMap; |
| 23 class Random; | 22 class Random; |
| 24 | 23 |
| 25 namespace rtp { | 24 namespace rtp { |
| 26 class Packet { | 25 class Packet { |
| 27 public: | 26 public: |
| 28 using ExtensionType = RTPExtensionType; | 27 using ExtensionType = RTPExtensionType; |
| 29 using ExtensionManager = RtpHeaderExtensionMap; | 28 using ExtensionManager = RtpHeaderExtensionMap; |
| 30 static constexpr size_t kMaxExtensionHeaders = 14; | 29 static constexpr size_t kMaxExtensionHeaders = 14; |
| 31 | 30 |
| 32 // Parse and copy given buffer into Packet. | 31 // Parse and copy given buffer into Packet. |
| 33 bool Parse(const uint8_t* buffer, size_t size); | 32 bool Parse(const uint8_t* buffer, size_t size); |
| 34 | 33 |
| 35 // Parse and move given buffer into Packet. | 34 // Parse and move given buffer into Packet. |
| 36 bool Parse(rtc::Buffer packet); | 35 bool Parse(rtc::CopyOnWriteBuffer packet); |
| 37 | 36 |
| 38 // Maps parsed extensions to their types to allow use of GetExtension. | 37 // Maps parsed extensions to their types to allow use of GetExtension. |
| 39 // Used after parsing when |extensions| can't be provided until base rtp | 38 // Used after parsing when |extensions| can't be provided until base rtp |
| 40 // header is parsed. | 39 // header is parsed. |
| 41 void IdentifyExtensions(const ExtensionManager* extensions); | 40 void IdentifyExtensions(const ExtensionManager* extensions); |
| 42 | 41 |
| 43 // Header. | 42 // Header. |
| 44 bool Marker() const; | 43 bool Marker() const; |
| 45 uint8_t PayloadType() const; | 44 uint8_t PayloadType() const; |
| 46 uint16_t SequenceNumber() const; | 45 uint16_t SequenceNumber() const; |
| 47 uint32_t Timestamp() const; | 46 uint32_t Timestamp() const; |
| 48 uint32_t Ssrc() const; | 47 uint32_t Ssrc() const; |
| 49 std::vector<uint32_t> Csrcs() const; | 48 std::vector<uint32_t> Csrcs() const; |
| 50 | 49 |
| 51 // TODO(danilchap): Remove this function when all code update to use RtpPacket | 50 // TODO(danilchap): Remove this function when all code update to use RtpPacket |
| 52 // directly. Function is there just for easier backward compatibilty. | 51 // directly. Function is there just for easier backward compatibilty. |
| 53 void GetHeader(RTPHeader* header) const; | 52 void GetHeader(RTPHeader* header) const; |
| 54 | 53 |
| 55 size_t headers_size() const; | 54 size_t headers_size() const; |
| 56 | 55 |
| 57 // Payload. | 56 // Payload. |
| 58 size_t payload_size() const; | 57 size_t payload_size() const; |
| 59 size_t padding_size() const; | 58 size_t padding_size() const; |
| 60 const uint8_t* payload() const; | 59 const uint8_t* payload() const; |
| 61 | 60 |
| 62 // Buffer. | 61 // Buffer. |
| 62 rtc::CopyOnWriteBuffer Buffer() const; |
| 63 size_t capacity() const; | 63 size_t capacity() const; |
| 64 size_t size() const; | 64 size_t size() const; |
| 65 const uint8_t* data() const; | 65 const uint8_t* data() const; |
| 66 size_t FreeCapacity() const; | 66 size_t FreeCapacity() const; |
| 67 size_t MaxPayloadSize() const; | 67 size_t MaxPayloadSize() const; |
| 68 | 68 |
| 69 // Reset fields and buffer. | 69 // Reset fields and buffer. |
| 70 void Clear(); | 70 void Clear(); |
| 71 | 71 |
| 72 // Header setters. | 72 // Header setters. |
| 73 void CopyHeader(const Packet& packet); | 73 void CopyHeaderFrom(const Packet& packet); |
| 74 void SetMarker(bool marker_bit); | 74 void SetMarker(bool marker_bit); |
| 75 void SetPayloadType(uint8_t payload_type); | 75 void SetPayloadType(uint8_t payload_type); |
| 76 void SetSequenceNumber(uint16_t seq_no); | 76 void SetSequenceNumber(uint16_t seq_no); |
| 77 void SetTimestamp(uint32_t timestamp); | 77 void SetTimestamp(uint32_t timestamp); |
| 78 void SetSsrc(uint32_t ssrc); | 78 void SetSsrc(uint32_t ssrc); |
| 79 | 79 |
| 80 // Writes csrc list. Assumes: | 80 // Writes csrc list. Assumes: |
| 81 // a) There is enough room left in buffer. | 81 // a) There is enough room left in buffer. |
| 82 // b) Extension headers, payload or padding data has not already been added. | 82 // b) Extension headers, payload or padding data has not already been added. |
| 83 void SetCsrcs(const std::vector<uint32_t>& csrcs); | 83 void SetCsrcs(const std::vector<uint32_t>& csrcs); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 96 uint8_t* AllocatePayload(size_t size_bytes); | 96 uint8_t* AllocatePayload(size_t size_bytes); |
| 97 void SetPayloadSize(size_t size_bytes); | 97 void SetPayloadSize(size_t size_bytes); |
| 98 bool SetPadding(uint8_t size_bytes, Random* random); | 98 bool SetPadding(uint8_t size_bytes, Random* random); |
| 99 | 99 |
| 100 protected: | 100 protected: |
| 101 // |extensions| required for SetExtension/ReserveExtension functions during | 101 // |extensions| required for SetExtension/ReserveExtension functions during |
| 102 // packet creating and used if available in Parse function. | 102 // packet creating and used if available in Parse function. |
| 103 // Adding and getting extensions will fail until |extensions| is | 103 // Adding and getting extensions will fail until |extensions| is |
| 104 // provided via constructor or IdentifyExtensions function. | 104 // provided via constructor or IdentifyExtensions function. |
| 105 explicit Packet(const ExtensionManager* extensions); | 105 explicit Packet(const ExtensionManager* extensions); |
| 106 Packet(const Packet&) = default; |
| 106 Packet(const ExtensionManager* extensions, size_t capacity); | 107 Packet(const ExtensionManager* extensions, size_t capacity); |
| 107 virtual ~Packet(); | 108 virtual ~Packet(); |
| 108 | 109 |
| 110 Packet& operator=(const Packet&) = default; |
| 111 |
| 109 private: | 112 private: |
| 110 struct ExtensionInfo { | 113 struct ExtensionInfo { |
| 111 ExtensionType type; | 114 ExtensionType type; |
| 112 uint16_t offset; | 115 uint16_t offset; |
| 113 uint8_t length; | 116 uint8_t length; |
| 114 }; | 117 }; |
| 115 | 118 |
| 116 // Helper function for Parse. Fill header fields using data in given buffer, | 119 // Helper function for Parse. Fill header fields using data in given buffer, |
| 117 // but does not touch packet own buffer, leaving packet in invalid state. | 120 // but does not touch packet own buffer, leaving packet in invalid state. |
| 118 bool ParseBuffer(const uint8_t* buffer, size_t size); | 121 bool ParseBuffer(const uint8_t* buffer, size_t size); |
| (...skipping 26 matching lines...) Expand all Loading... |
| 145 uint8_t padding_size_; | 148 uint8_t padding_size_; |
| 146 uint16_t sequence_number_; | 149 uint16_t sequence_number_; |
| 147 uint32_t timestamp_; | 150 uint32_t timestamp_; |
| 148 uint32_t ssrc_; | 151 uint32_t ssrc_; |
| 149 size_t payload_offset_; // Match header size with csrcs and extensions. | 152 size_t payload_offset_; // Match header size with csrcs and extensions. |
| 150 size_t payload_size_; | 153 size_t payload_size_; |
| 151 | 154 |
| 152 uint8_t num_extensions_ = 0; | 155 uint8_t num_extensions_ = 0; |
| 153 ExtensionInfo extension_entries_[kMaxExtensionHeaders]; | 156 ExtensionInfo extension_entries_[kMaxExtensionHeaders]; |
| 154 uint16_t extensions_size_ = 0; // Unaligned. | 157 uint16_t extensions_size_ = 0; // Unaligned. |
| 155 rtc::Buffer buffer_; | 158 rtc::CopyOnWriteBuffer buffer_; |
| 156 | 159 |
| 157 RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(Packet); | 160 Packet() = delete; |
| 158 }; | 161 }; |
| 159 | 162 |
| 160 template <typename Extension, typename... Values> | 163 template <typename Extension, typename... Values> |
| 161 bool Packet::GetExtension(Values... values) const { | 164 bool Packet::GetExtension(Values... values) const { |
| 162 uint16_t offset = 0; | 165 uint16_t offset = 0; |
| 163 if (!FindExtension(Extension::kId, Extension::kValueSizeBytes, &offset)) | 166 if (!FindExtension(Extension::kId, Extension::kValueSizeBytes, &offset)) |
| 164 return false; | 167 return false; |
| 165 return Extension::Parse(data() + offset, values...); | 168 return Extension::Parse(data() + offset, values...); |
| 166 } | 169 } |
| 167 | 170 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 178 uint16_t offset = 0; | 181 uint16_t offset = 0; |
| 179 if (!AllocateExtension(Extension::kId, Extension::kValueSizeBytes, &offset)) | 182 if (!AllocateExtension(Extension::kId, Extension::kValueSizeBytes, &offset)) |
| 180 return false; | 183 return false; |
| 181 memset(WriteAt(offset), 0, Extension::kValueSizeBytes); | 184 memset(WriteAt(offset), 0, Extension::kValueSizeBytes); |
| 182 return true; | 185 return true; |
| 183 } | 186 } |
| 184 } // namespace rtp | 187 } // namespace rtp |
| 185 } // namespace webrtc | 188 } // namespace webrtc |
| 186 | 189 |
| 187 #endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ | 190 #endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_RTP_PACKET_H_ |
| OLD | NEW |