| /* | 
 |  *  Copyright (c) 2011 The WebRTC project authors. All Rights Reserved. | 
 |  * | 
 |  *  Use of this source code is governed by a BSD-style license | 
 |  *  that can be found in the LICENSE file in the root of the source | 
 |  *  tree. An additional intellectual property rights grant can be found | 
 |  *  in the file PATENTS.  All contributing project authors may | 
 |  *  be found in the AUTHORS file in the root of the source tree. | 
 |  */ | 
 |  | 
 | #include "webrtc/modules/rtp_rtcp/source/rtp_format_vp8.h" | 
 |  | 
 | #include <assert.h>  // assert | 
 | #include <string.h>  // memcpy | 
 |  | 
 | #include <vector> | 
 |  | 
 | #include "webrtc/base/logging.h" | 
 | #include "webrtc/modules/rtp_rtcp/source/vp8_partition_aggregator.h" | 
 |  | 
 | namespace webrtc { | 
 | namespace { | 
 | int ParseVP8PictureID(RTPVideoHeaderVP8* vp8, | 
 |                       const uint8_t** data, | 
 |                       size_t* data_length, | 
 |                       size_t* parsed_bytes) { | 
 |   assert(vp8 != NULL); | 
 |   if (*data_length == 0) | 
 |     return -1; | 
 |  | 
 |   vp8->pictureId = (**data & 0x7F); | 
 |   if (**data & 0x80) { | 
 |     (*data)++; | 
 |     (*parsed_bytes)++; | 
 |     if (--(*data_length) == 0) | 
 |       return -1; | 
 |     // PictureId is 15 bits | 
 |     vp8->pictureId = (vp8->pictureId << 8) + **data; | 
 |   } | 
 |   (*data)++; | 
 |   (*parsed_bytes)++; | 
 |   (*data_length)--; | 
 |   return 0; | 
 | } | 
 |  | 
 | int ParseVP8Tl0PicIdx(RTPVideoHeaderVP8* vp8, | 
 |                       const uint8_t** data, | 
 |                       size_t* data_length, | 
 |                       size_t* parsed_bytes) { | 
 |   assert(vp8 != NULL); | 
 |   if (*data_length == 0) | 
 |     return -1; | 
 |  | 
 |   vp8->tl0PicIdx = **data; | 
 |   (*data)++; | 
 |   (*parsed_bytes)++; | 
 |   (*data_length)--; | 
 |   return 0; | 
 | } | 
 |  | 
 | int ParseVP8TIDAndKeyIdx(RTPVideoHeaderVP8* vp8, | 
 |                          const uint8_t** data, | 
 |                          size_t* data_length, | 
 |                          size_t* parsed_bytes, | 
 |                          bool has_tid, | 
 |                          bool has_key_idx) { | 
 |   assert(vp8 != NULL); | 
 |   if (*data_length == 0) | 
 |     return -1; | 
 |  | 
 |   if (has_tid) { | 
 |     vp8->temporalIdx = ((**data >> 6) & 0x03); | 
 |     vp8->layerSync = (**data & 0x20) ? true : false;  // Y bit | 
 |   } | 
 |   if (has_key_idx) { | 
 |     vp8->keyIdx = (**data & 0x1F); | 
 |   } | 
 |   (*data)++; | 
 |   (*parsed_bytes)++; | 
 |   (*data_length)--; | 
 |   return 0; | 
 | } | 
 |  | 
 | int ParseVP8Extension(RTPVideoHeaderVP8* vp8, | 
 |                       const uint8_t* data, | 
 |                       size_t data_length) { | 
 |   assert(vp8 != NULL); | 
 |   assert(data_length > 0); | 
 |   size_t parsed_bytes = 0; | 
 |   // Optional X field is present. | 
 |   bool has_picture_id = (*data & 0x80) ? true : false;   // I bit | 
 |   bool has_tl0_pic_idx = (*data & 0x40) ? true : false;  // L bit | 
 |   bool has_tid = (*data & 0x20) ? true : false;          // T bit | 
 |   bool has_key_idx = (*data & 0x10) ? true : false;      // K bit | 
 |  | 
 |   // Advance data and decrease remaining payload size. | 
 |   data++; | 
 |   parsed_bytes++; | 
 |   data_length--; | 
 |  | 
 |   if (has_picture_id) { | 
 |     if (ParseVP8PictureID(vp8, &data, &data_length, &parsed_bytes) != 0) { | 
 |       return -1; | 
 |     } | 
 |   } | 
 |  | 
 |   if (has_tl0_pic_idx) { | 
 |     if (ParseVP8Tl0PicIdx(vp8, &data, &data_length, &parsed_bytes) != 0) { | 
 |       return -1; | 
 |     } | 
 |   } | 
 |  | 
 |   if (has_tid || has_key_idx) { | 
 |     if (ParseVP8TIDAndKeyIdx( | 
 |             vp8, &data, &data_length, &parsed_bytes, has_tid, has_key_idx) != | 
 |         0) { | 
 |       return -1; | 
 |     } | 
 |   } | 
 |   return static_cast<int>(parsed_bytes); | 
 | } | 
 |  | 
 | int ParseVP8FrameSize(RtpDepacketizer::ParsedPayload* parsed_payload, | 
 |                       const uint8_t* data, | 
 |                       size_t data_length) { | 
 |   assert(parsed_payload != NULL); | 
 |   if (parsed_payload->frame_type != kVideoFrameKey) { | 
 |     // Included in payload header for I-frames. | 
 |     return 0; | 
 |   } | 
 |   if (data_length < 10) { | 
 |     // For an I-frame we should always have the uncompressed VP8 header | 
 |     // in the beginning of the partition. | 
 |     return -1; | 
 |   } | 
 |   parsed_payload->type.Video.width = ((data[7] << 8) + data[6]) & 0x3FFF; | 
 |   parsed_payload->type.Video.height = ((data[9] << 8) + data[8]) & 0x3FFF; | 
 |   return 0; | 
 | } | 
 | }  // namespace | 
 |  | 
 | // Define how the VP8PacketizerModes are implemented. | 
 | // Modes are: kStrict, kAggregate, kEqualSize. | 
 | const RtpPacketizerVp8::AggregationMode RtpPacketizerVp8::aggr_modes_ | 
 |     [kNumModes] = {kAggrNone, kAggrPartitions, kAggrFragments}; | 
 | const bool RtpPacketizerVp8::balance_modes_[kNumModes] = {true, true, true}; | 
 | const bool RtpPacketizerVp8::separate_first_modes_[kNumModes] = {true, false, | 
 |                                                                  false}; | 
 |  | 
 | RtpPacketizerVp8::RtpPacketizerVp8(const RTPVideoHeaderVP8& hdr_info, | 
 |                                    size_t max_payload_len, | 
 |                                    VP8PacketizerMode mode) | 
 |     : payload_data_(NULL), | 
 |       payload_size_(0), | 
 |       vp8_fixed_payload_descriptor_bytes_(1), | 
 |       aggr_mode_(aggr_modes_[mode]), | 
 |       balance_(balance_modes_[mode]), | 
 |       separate_first_(separate_first_modes_[mode]), | 
 |       hdr_info_(hdr_info), | 
 |       num_partitions_(0), | 
 |       max_payload_len_(max_payload_len), | 
 |       packets_calculated_(false) { | 
 | } | 
 |  | 
 | RtpPacketizerVp8::RtpPacketizerVp8(const RTPVideoHeaderVP8& hdr_info, | 
 |                                    size_t max_payload_len) | 
 |     : payload_data_(NULL), | 
 |       payload_size_(0), | 
 |       part_info_(), | 
 |       vp8_fixed_payload_descriptor_bytes_(1), | 
 |       aggr_mode_(aggr_modes_[kEqualSize]), | 
 |       balance_(balance_modes_[kEqualSize]), | 
 |       separate_first_(separate_first_modes_[kEqualSize]), | 
 |       hdr_info_(hdr_info), | 
 |       num_partitions_(0), | 
 |       max_payload_len_(max_payload_len), | 
 |       packets_calculated_(false) { | 
 | } | 
 |  | 
 | RtpPacketizerVp8::~RtpPacketizerVp8() { | 
 | } | 
 |  | 
 | void RtpPacketizerVp8::SetPayloadData( | 
 |     const uint8_t* payload_data, | 
 |     size_t payload_size, | 
 |     const RTPFragmentationHeader* fragmentation) { | 
 |   payload_data_ = payload_data; | 
 |   payload_size_ = payload_size; | 
 |   if (fragmentation) { | 
 |     part_info_.CopyFrom(*fragmentation); | 
 |     num_partitions_ = fragmentation->fragmentationVectorSize; | 
 |   } else { | 
 |     part_info_.VerifyAndAllocateFragmentationHeader(1); | 
 |     part_info_.fragmentationLength[0] = payload_size; | 
 |     part_info_.fragmentationOffset[0] = 0; | 
 |     num_partitions_ = part_info_.fragmentationVectorSize; | 
 |   } | 
 | } | 
 |  | 
 | bool RtpPacketizerVp8::NextPacket(uint8_t* buffer, | 
 |                                   size_t* bytes_to_send, | 
 |                                   bool* last_packet) { | 
 |   if (!packets_calculated_) { | 
 |     int ret = 0; | 
 |     if (aggr_mode_ == kAggrPartitions && balance_) { | 
 |       ret = GeneratePacketsBalancedAggregates(); | 
 |     } else { | 
 |       ret = GeneratePackets(); | 
 |     } | 
 |     if (ret < 0) { | 
 |       return false; | 
 |     } | 
 |   } | 
 |   if (packets_.empty()) { | 
 |     return false; | 
 |   } | 
 |   InfoStruct packet_info = packets_.front(); | 
 |   packets_.pop(); | 
 |  | 
 |   int bytes = WriteHeaderAndPayload(packet_info, buffer, max_payload_len_); | 
 |   if (bytes < 0) { | 
 |     return false; | 
 |   } | 
 |   *bytes_to_send = static_cast<size_t>(bytes); | 
 |  | 
 |   *last_packet = packets_.empty(); | 
 |   return true; | 
 | } | 
 |  | 
 | ProtectionType RtpPacketizerVp8::GetProtectionType() { | 
 |   bool protect = | 
 |       hdr_info_.temporalIdx == 0 || hdr_info_.temporalIdx == kNoTemporalIdx; | 
 |   return protect ? kProtectedPacket : kUnprotectedPacket; | 
 | } | 
 |  | 
 | StorageType RtpPacketizerVp8::GetStorageType(uint32_t retransmission_settings) { | 
 |   if (hdr_info_.temporalIdx == 0 && | 
 |       !(retransmission_settings & kRetransmitBaseLayer)) { | 
 |     return kDontRetransmit; | 
 |   } | 
 |   if (hdr_info_.temporalIdx != kNoTemporalIdx && | 
 |              hdr_info_.temporalIdx > 0 && | 
 |              !(retransmission_settings & kRetransmitHigherLayers)) { | 
 |     return kDontRetransmit; | 
 |   } | 
 |   return kAllowRetransmission; | 
 | } | 
 |  | 
 | std::string RtpPacketizerVp8::ToString() { | 
 |   return "RtpPacketizerVp8"; | 
 | } | 
 |  | 
 | size_t RtpPacketizerVp8::CalcNextSize(size_t max_payload_len, | 
 |                                       size_t remaining_bytes, | 
 |                                       bool split_payload) const { | 
 |   if (max_payload_len == 0 || remaining_bytes == 0) { | 
 |     return 0; | 
 |   } | 
 |   if (!split_payload) { | 
 |     return max_payload_len >= remaining_bytes ? remaining_bytes : 0; | 
 |   } | 
 |  | 
 |   if (balance_) { | 
 |     // Balance payload sizes to produce (almost) equal size | 
 |     // fragments. | 
 |     // Number of fragments for remaining_bytes: | 
 |     size_t num_frags = remaining_bytes / max_payload_len + 1; | 
 |     // Number of bytes in this fragment: | 
 |     return static_cast<size_t>( | 
 |         static_cast<double>(remaining_bytes) / num_frags + 0.5); | 
 |   } else { | 
 |     return max_payload_len >= remaining_bytes ? remaining_bytes | 
 |                                               : max_payload_len; | 
 |   } | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::GeneratePackets() { | 
 |   if (max_payload_len_ < vp8_fixed_payload_descriptor_bytes_ + | 
 |                              PayloadDescriptorExtraLength() + 1) { | 
 |     // The provided payload length is not long enough for the payload | 
 |     // descriptor and one payload byte. Return an error. | 
 |     return -1; | 
 |   } | 
 |   size_t total_bytes_processed = 0; | 
 |   bool start_on_new_fragment = true; | 
 |   bool beginning = true; | 
 |   size_t part_ix = 0; | 
 |   while (total_bytes_processed < payload_size_) { | 
 |     size_t packet_bytes = 0;    // How much data to send in this packet. | 
 |     bool split_payload = true;  // Splitting of partitions is initially allowed. | 
 |     size_t remaining_in_partition = part_info_.fragmentationOffset[part_ix] - | 
 |                                  total_bytes_processed + | 
 |                                  part_info_.fragmentationLength[part_ix]; | 
 |     size_t rem_payload_len = | 
 |         max_payload_len_ - | 
 |         (vp8_fixed_payload_descriptor_bytes_ + PayloadDescriptorExtraLength()); | 
 |     size_t first_partition_in_packet = part_ix; | 
 |  | 
 |     while (size_t next_size = CalcNextSize( | 
 |                rem_payload_len, remaining_in_partition, split_payload)) { | 
 |       packet_bytes += next_size; | 
 |       rem_payload_len -= next_size; | 
 |       remaining_in_partition -= next_size; | 
 |  | 
 |       if (remaining_in_partition == 0 && !(beginning && separate_first_)) { | 
 |         // Advance to next partition? | 
 |         // Check that there are more partitions; verify that we are either | 
 |         // allowed to aggregate fragments, or that we are allowed to | 
 |         // aggregate intact partitions and that we started this packet | 
 |         // with an intact partition (indicated by first_fragment_ == true). | 
 |         if (part_ix + 1 < num_partitions_ && | 
 |             ((aggr_mode_ == kAggrFragments) || | 
 |              (aggr_mode_ == kAggrPartitions && start_on_new_fragment))) { | 
 |           assert(part_ix < num_partitions_); | 
 |           remaining_in_partition = part_info_.fragmentationLength[++part_ix]; | 
 |           // Disallow splitting unless kAggrFragments. In kAggrPartitions, | 
 |           // we can only aggregate intact partitions. | 
 |           split_payload = (aggr_mode_ == kAggrFragments); | 
 |         } | 
 |       } else if (balance_ && remaining_in_partition > 0) { | 
 |         break; | 
 |       } | 
 |     } | 
 |     if (remaining_in_partition == 0) { | 
 |       ++part_ix;  // Advance to next partition. | 
 |     } | 
 |     assert(packet_bytes > 0); | 
 |  | 
 |     QueuePacket(total_bytes_processed, | 
 |                 packet_bytes, | 
 |                 first_partition_in_packet, | 
 |                 start_on_new_fragment); | 
 |     total_bytes_processed += packet_bytes; | 
 |     start_on_new_fragment = (remaining_in_partition == 0); | 
 |     beginning = false;  // Next packet cannot be first packet in frame. | 
 |   } | 
 |   packets_calculated_ = true; | 
 |   assert(total_bytes_processed == payload_size_); | 
 |   return 0; | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::GeneratePacketsBalancedAggregates() { | 
 |   if (max_payload_len_ < vp8_fixed_payload_descriptor_bytes_ + | 
 |                              PayloadDescriptorExtraLength() + 1) { | 
 |     // The provided payload length is not long enough for the payload | 
 |     // descriptor and one payload byte. Return an error. | 
 |     return -1; | 
 |   } | 
 |   std::vector<int> partition_decision; | 
 |   const size_t overhead = | 
 |       vp8_fixed_payload_descriptor_bytes_ + PayloadDescriptorExtraLength(); | 
 |   const size_t max_payload_len = max_payload_len_ - overhead; | 
 |   int min_size, max_size; | 
 |   AggregateSmallPartitions(&partition_decision, &min_size, &max_size); | 
 |  | 
 |   size_t total_bytes_processed = 0; | 
 |   size_t part_ix = 0; | 
 |   while (part_ix < num_partitions_) { | 
 |     if (partition_decision[part_ix] == -1) { | 
 |       // Split large partitions. | 
 |       size_t remaining_partition = part_info_.fragmentationLength[part_ix]; | 
 |       size_t num_fragments = Vp8PartitionAggregator::CalcNumberOfFragments( | 
 |           remaining_partition, max_payload_len, overhead, min_size, max_size); | 
 |       const size_t packet_bytes = | 
 |           (remaining_partition + num_fragments - 1) / num_fragments; | 
 |       for (size_t n = 0; n < num_fragments; ++n) { | 
 |         const size_t this_packet_bytes = packet_bytes < remaining_partition | 
 |                                              ? packet_bytes | 
 |                                              : remaining_partition; | 
 |         QueuePacket( | 
 |             total_bytes_processed, this_packet_bytes, part_ix, (n == 0)); | 
 |         remaining_partition -= this_packet_bytes; | 
 |         total_bytes_processed += this_packet_bytes; | 
 |         if (static_cast<int>(this_packet_bytes) < min_size) { | 
 |           min_size = this_packet_bytes; | 
 |         } | 
 |         if (static_cast<int>(this_packet_bytes) > max_size) { | 
 |           max_size = this_packet_bytes; | 
 |         } | 
 |       } | 
 |       assert(remaining_partition == 0); | 
 |       ++part_ix; | 
 |     } else { | 
 |       size_t this_packet_bytes = 0; | 
 |       const size_t first_partition_in_packet = part_ix; | 
 |       const int aggregation_index = partition_decision[part_ix]; | 
 |       while (part_ix < partition_decision.size() && | 
 |              partition_decision[part_ix] == aggregation_index) { | 
 |         // Collect all partitions that were aggregated into the same packet. | 
 |         this_packet_bytes += part_info_.fragmentationLength[part_ix]; | 
 |         ++part_ix; | 
 |       } | 
 |       QueuePacket(total_bytes_processed, | 
 |                   this_packet_bytes, | 
 |                   first_partition_in_packet, | 
 |                   true); | 
 |       total_bytes_processed += this_packet_bytes; | 
 |     } | 
 |   } | 
 |   packets_calculated_ = true; | 
 |   return 0; | 
 | } | 
 |  | 
 | void RtpPacketizerVp8::AggregateSmallPartitions(std::vector<int>* partition_vec, | 
 |                                                 int* min_size, | 
 |                                                 int* max_size) { | 
 |   assert(min_size && max_size); | 
 |   *min_size = -1; | 
 |   *max_size = -1; | 
 |   assert(partition_vec); | 
 |   partition_vec->assign(num_partitions_, -1); | 
 |   const size_t overhead = | 
 |       vp8_fixed_payload_descriptor_bytes_ + PayloadDescriptorExtraLength(); | 
 |   const size_t max_payload_len = max_payload_len_ - overhead; | 
 |   size_t first_in_set = 0; | 
 |   size_t last_in_set = 0; | 
 |   int num_aggregate_packets = 0; | 
 |   // Find sets of partitions smaller than max_payload_len_. | 
 |   while (first_in_set < num_partitions_) { | 
 |     if (part_info_.fragmentationLength[first_in_set] < max_payload_len) { | 
 |       // Found start of a set. | 
 |       last_in_set = first_in_set; | 
 |       while (last_in_set + 1 < num_partitions_ && | 
 |              part_info_.fragmentationLength[last_in_set + 1] < | 
 |                  max_payload_len) { | 
 |         ++last_in_set; | 
 |       } | 
 |       // Found end of a set. Run optimized aggregator. It is ok if start == end. | 
 |       Vp8PartitionAggregator aggregator(part_info_, first_in_set, last_in_set); | 
 |       if (*min_size >= 0 && *max_size >= 0) { | 
 |         aggregator.SetPriorMinMax(*min_size, *max_size); | 
 |       } | 
 |       Vp8PartitionAggregator::ConfigVec optimal_config = | 
 |           aggregator.FindOptimalConfiguration(max_payload_len, overhead); | 
 |       aggregator.CalcMinMax(optimal_config, min_size, max_size); | 
 |       for (size_t i = first_in_set, j = 0; i <= last_in_set; ++i, ++j) { | 
 |         // Transfer configuration for this set of partitions to the joint | 
 |         // partition vector representing all partitions in the frame. | 
 |         (*partition_vec)[i] = num_aggregate_packets + optimal_config[j]; | 
 |       } | 
 |       num_aggregate_packets += optimal_config.back() + 1; | 
 |       first_in_set = last_in_set; | 
 |     } | 
 |     ++first_in_set; | 
 |   } | 
 | } | 
 |  | 
 | void RtpPacketizerVp8::QueuePacket(size_t start_pos, | 
 |                                    size_t packet_size, | 
 |                                    size_t first_partition_in_packet, | 
 |                                    bool start_on_new_fragment) { | 
 |   // Write info to packet info struct and store in packet info queue. | 
 |   InfoStruct packet_info; | 
 |   packet_info.payload_start_pos = start_pos; | 
 |   packet_info.size = packet_size; | 
 |   packet_info.first_partition_ix = first_partition_in_packet; | 
 |   packet_info.first_fragment = start_on_new_fragment; | 
 |   packets_.push(packet_info); | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WriteHeaderAndPayload(const InfoStruct& packet_info, | 
 |                                             uint8_t* buffer, | 
 |                                             size_t buffer_length) const { | 
 |   // Write the VP8 payload descriptor. | 
 |   //       0 | 
 |   //       0 1 2 3 4 5 6 7 8 | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |   //      |X| |N|S| PART_ID | | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |   // X:   |I|L|T|K|         | (mandatory if any of the below are used) | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |   // I:   |PictureID (8/16b)| (optional) | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |   // L:   |   TL0PIC_IDX    | (optional) | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |   // T/K: |TID:Y|  KEYIDX   | (optional) | 
 |   //      +-+-+-+-+-+-+-+-+-+ | 
 |  | 
 |   assert(packet_info.size > 0); | 
 |   buffer[0] = 0; | 
 |   if (XFieldPresent()) | 
 |     buffer[0] |= kXBit; | 
 |   if (hdr_info_.nonReference) | 
 |     buffer[0] |= kNBit; | 
 |   if (packet_info.first_fragment) | 
 |     buffer[0] |= kSBit; | 
 |   buffer[0] |= (packet_info.first_partition_ix & kPartIdField); | 
 |  | 
 |   const int extension_length = WriteExtensionFields(buffer, buffer_length); | 
 |   if (extension_length < 0) | 
 |     return -1; | 
 |  | 
 |   memcpy(&buffer[vp8_fixed_payload_descriptor_bytes_ + extension_length], | 
 |          &payload_data_[packet_info.payload_start_pos], | 
 |          packet_info.size); | 
 |  | 
 |   // Return total length of written data. | 
 |   return packet_info.size + vp8_fixed_payload_descriptor_bytes_ + | 
 |          extension_length; | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WriteExtensionFields(uint8_t* buffer, | 
 |                                            size_t buffer_length) const { | 
 |   size_t extension_length = 0; | 
 |   if (XFieldPresent()) { | 
 |     uint8_t* x_field = buffer + vp8_fixed_payload_descriptor_bytes_; | 
 |     *x_field = 0; | 
 |     extension_length = 1;  // One octet for the X field. | 
 |     if (PictureIdPresent()) { | 
 |       if (WritePictureIDFields( | 
 |               x_field, buffer, buffer_length, &extension_length) < 0) { | 
 |         return -1; | 
 |       } | 
 |     } | 
 |     if (TL0PicIdxFieldPresent()) { | 
 |       if (WriteTl0PicIdxFields( | 
 |               x_field, buffer, buffer_length, &extension_length) < 0) { | 
 |         return -1; | 
 |       } | 
 |     } | 
 |     if (TIDFieldPresent() || KeyIdxFieldPresent()) { | 
 |       if (WriteTIDAndKeyIdxFields( | 
 |               x_field, buffer, buffer_length, &extension_length) < 0) { | 
 |         return -1; | 
 |       } | 
 |     } | 
 |     assert(extension_length == PayloadDescriptorExtraLength()); | 
 |   } | 
 |   return static_cast<int>(extension_length); | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WritePictureIDFields(uint8_t* x_field, | 
 |                                            uint8_t* buffer, | 
 |                                            size_t buffer_length, | 
 |                                            size_t* extension_length) const { | 
 |   *x_field |= kIBit; | 
 |   assert(buffer_length >= | 
 |       vp8_fixed_payload_descriptor_bytes_ + *extension_length); | 
 |   const int pic_id_length = WritePictureID( | 
 |       buffer + vp8_fixed_payload_descriptor_bytes_ + *extension_length, | 
 |       buffer_length - vp8_fixed_payload_descriptor_bytes_ - *extension_length); | 
 |   if (pic_id_length < 0) | 
 |     return -1; | 
 |   *extension_length += pic_id_length; | 
 |   return 0; | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WritePictureID(uint8_t* buffer, | 
 |                                      size_t buffer_length) const { | 
 |   const uint16_t pic_id = static_cast<uint16_t>(hdr_info_.pictureId); | 
 |   size_t picture_id_len = PictureIdLength(); | 
 |   if (picture_id_len > buffer_length) | 
 |     return -1; | 
 |   if (picture_id_len == 2) { | 
 |     buffer[0] = 0x80 | ((pic_id >> 8) & 0x7F); | 
 |     buffer[1] = pic_id & 0xFF; | 
 |   } else if (picture_id_len == 1) { | 
 |     buffer[0] = pic_id & 0x7F; | 
 |   } | 
 |   return static_cast<int>(picture_id_len); | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WriteTl0PicIdxFields(uint8_t* x_field, | 
 |                                            uint8_t* buffer, | 
 |                                            size_t buffer_length, | 
 |                                            size_t* extension_length) const { | 
 |   if (buffer_length < | 
 |       vp8_fixed_payload_descriptor_bytes_ + *extension_length + 1) { | 
 |     return -1; | 
 |   } | 
 |   *x_field |= kLBit; | 
 |   buffer[vp8_fixed_payload_descriptor_bytes_ + *extension_length] = | 
 |       hdr_info_.tl0PicIdx; | 
 |   ++*extension_length; | 
 |   return 0; | 
 | } | 
 |  | 
 | int RtpPacketizerVp8::WriteTIDAndKeyIdxFields(uint8_t* x_field, | 
 |                                               uint8_t* buffer, | 
 |                                               size_t buffer_length, | 
 |                                               size_t* extension_length) const { | 
 |   if (buffer_length < | 
 |       vp8_fixed_payload_descriptor_bytes_ + *extension_length + 1) { | 
 |     return -1; | 
 |   } | 
 |   uint8_t* data_field = | 
 |       &buffer[vp8_fixed_payload_descriptor_bytes_ + *extension_length]; | 
 |   *data_field = 0; | 
 |   if (TIDFieldPresent()) { | 
 |     *x_field |= kTBit; | 
 |     assert(hdr_info_.temporalIdx <= 3); | 
 |     *data_field |= hdr_info_.temporalIdx << 6; | 
 |     *data_field |= hdr_info_.layerSync ? kYBit : 0; | 
 |   } | 
 |   if (KeyIdxFieldPresent()) { | 
 |     *x_field |= kKBit; | 
 |     *data_field |= (hdr_info_.keyIdx & kKeyIdxField); | 
 |   } | 
 |   ++*extension_length; | 
 |   return 0; | 
 | } | 
 |  | 
 | size_t RtpPacketizerVp8::PayloadDescriptorExtraLength() const { | 
 |   size_t length_bytes = PictureIdLength(); | 
 |   if (TL0PicIdxFieldPresent()) | 
 |     ++length_bytes; | 
 |   if (TIDFieldPresent() || KeyIdxFieldPresent()) | 
 |     ++length_bytes; | 
 |   if (length_bytes > 0) | 
 |     ++length_bytes;  // Include the extension field. | 
 |   return length_bytes; | 
 | } | 
 |  | 
 | size_t RtpPacketizerVp8::PictureIdLength() const { | 
 |   if (hdr_info_.pictureId == kNoPictureId) { | 
 |     return 0; | 
 |   } | 
 |   if (hdr_info_.pictureId <= 0x7F) { | 
 |     return 1; | 
 |   } | 
 |   return 2; | 
 | } | 
 |  | 
 | bool RtpPacketizerVp8::XFieldPresent() const { | 
 |   return (TIDFieldPresent() || TL0PicIdxFieldPresent() || PictureIdPresent() || | 
 |           KeyIdxFieldPresent()); | 
 | } | 
 |  | 
 | bool RtpPacketizerVp8::TIDFieldPresent() const { | 
 |   assert((hdr_info_.layerSync == false) || | 
 |          (hdr_info_.temporalIdx != kNoTemporalIdx)); | 
 |   return (hdr_info_.temporalIdx != kNoTemporalIdx); | 
 | } | 
 |  | 
 | bool RtpPacketizerVp8::KeyIdxFieldPresent() const { | 
 |   return (hdr_info_.keyIdx != kNoKeyIdx); | 
 | } | 
 |  | 
 | bool RtpPacketizerVp8::TL0PicIdxFieldPresent() const { | 
 |   return (hdr_info_.tl0PicIdx != kNoTl0PicIdx); | 
 | } | 
 |  | 
 | // | 
 | // VP8 format: | 
 | // | 
 | // Payload descriptor | 
 | //       0 1 2 3 4 5 6 7 | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | //      |X|R|N|S|PartID | (REQUIRED) | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | // X:   |I|L|T|K|  RSV  | (OPTIONAL) | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | // I:   |   PictureID   | (OPTIONAL) | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | // L:   |   TL0PICIDX   | (OPTIONAL) | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | // T/K: |TID:Y| KEYIDX  | (OPTIONAL) | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | // | 
 | // Payload header (considered part of the actual payload, sent to decoder) | 
 | //       0 1 2 3 4 5 6 7 | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | //      |Size0|H| VER |P| | 
 | //      +-+-+-+-+-+-+-+-+ | 
 | //      |      ...      | | 
 | //      +               + | 
 | bool RtpDepacketizerVp8::Parse(ParsedPayload* parsed_payload, | 
 |                                const uint8_t* payload_data, | 
 |                                size_t payload_data_length) { | 
 |   assert(parsed_payload != NULL); | 
 |   if (payload_data_length == 0) { | 
 |     LOG(LS_ERROR) << "Empty payload."; | 
 |     return false; | 
 |   } | 
 |  | 
 |   // Parse mandatory first byte of payload descriptor. | 
 |   bool extension = (*payload_data & 0x80) ? true : false;               // X bit | 
 |   bool beginning_of_partition = (*payload_data & 0x10) ? true : false;  // S bit | 
 |   int partition_id = (*payload_data & 0x0F);  // PartID field | 
 |  | 
 |   parsed_payload->type.Video.width = 0; | 
 |   parsed_payload->type.Video.height = 0; | 
 |   parsed_payload->type.Video.isFirstPacket = | 
 |       beginning_of_partition && (partition_id == 0); | 
 |   parsed_payload->type.Video.simulcastIdx = 0; | 
 |   parsed_payload->type.Video.codec = kRtpVideoVp8; | 
 |   parsed_payload->type.Video.codecHeader.VP8.nonReference = | 
 |       (*payload_data & 0x20) ? true : false;  // N bit | 
 |   parsed_payload->type.Video.codecHeader.VP8.partitionId = partition_id; | 
 |   parsed_payload->type.Video.codecHeader.VP8.beginningOfPartition = | 
 |       beginning_of_partition; | 
 |   parsed_payload->type.Video.codecHeader.VP8.pictureId = kNoPictureId; | 
 |   parsed_payload->type.Video.codecHeader.VP8.tl0PicIdx = kNoTl0PicIdx; | 
 |   parsed_payload->type.Video.codecHeader.VP8.temporalIdx = kNoTemporalIdx; | 
 |   parsed_payload->type.Video.codecHeader.VP8.layerSync = false; | 
 |   parsed_payload->type.Video.codecHeader.VP8.keyIdx = kNoKeyIdx; | 
 |  | 
 |   if (partition_id > 8) { | 
 |     // Weak check for corrupt payload_data: PartID MUST NOT be larger than 8. | 
 |     return false; | 
 |   } | 
 |  | 
 |   // Advance payload_data and decrease remaining payload size. | 
 |   payload_data++; | 
 |   if (payload_data_length <= 1) { | 
 |     LOG(LS_ERROR) << "Error parsing VP8 payload descriptor!"; | 
 |     return false; | 
 |   } | 
 |   payload_data_length--; | 
 |  | 
 |   if (extension) { | 
 |     const int parsed_bytes = | 
 |         ParseVP8Extension(&parsed_payload->type.Video.codecHeader.VP8, | 
 |                           payload_data, | 
 |                           payload_data_length); | 
 |     if (parsed_bytes < 0) | 
 |       return false; | 
 |     payload_data += parsed_bytes; | 
 |     payload_data_length -= parsed_bytes; | 
 |     if (payload_data_length == 0) { | 
 |       LOG(LS_ERROR) << "Error parsing VP8 payload descriptor!"; | 
 |       return false; | 
 |     } | 
 |   } | 
 |  | 
 |   // Read P bit from payload header (only at beginning of first partition). | 
 |   if (beginning_of_partition && partition_id == 0) { | 
 |     parsed_payload->frame_type = | 
 |         (*payload_data & 0x01) ? kVideoFrameDelta : kVideoFrameKey; | 
 |   } else { | 
 |     parsed_payload->frame_type = kVideoFrameDelta; | 
 |   } | 
 |  | 
 |   if (ParseVP8FrameSize(parsed_payload, payload_data, payload_data_length) != | 
 |       0) { | 
 |     return false; | 
 |   } | 
 |  | 
 |   parsed_payload->payload = payload_data; | 
 |   parsed_payload->payload_length = payload_data_length; | 
 |   return true; | 
 | } | 
 | }  // namespace webrtc |