Fix packetization logic to leave space for extensions in the last packet

webrtc/modules/rtp_rtcp/source/rtp_sender_video.cc

 /*
  *  Copyright (c) 2012 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.
  */
 
@@ skipping 282 matching lines @@
                                const RTPFragmentationHeader* fragmentation,
                                const RTPVideoHeader* video_header) {
   if (payload_size == 0)
     return false;
 
   // Create header that will be reused in all packets.
   std::unique_ptr<RtpPacketToSend> rtp_header = rtp_sender_->AllocatePacket();
   rtp_header->SetPayloadType(payload_type);
   rtp_header->SetTimestamp(rtp_timestamp);
   rtp_header->set_capture_time_ms(capture_time_ms);
+  std::unique_ptr<RtpPacketToSend> rtp_header_with_extensions(

[danilchap, 2017/05/12 08:34:40]

this name makes false impression rtp_header doesn't contain any extensions
may be call it last_packet?

[ilnik, 2017/05/12 09:17:57]

Done.

+      new RtpPacketToSend(*rtp_header));

[danilchap, 2017/05/12 08:34:40]

in base/ptr_util.h there is helper for this scenario of using new:
auto last_packet = rtc::MakeUnique<RtpPacketToSend>(*rtp_header);

[ilnik, 2017/05/12 09:17:58]

Done.

 
   size_t fec_packet_overhead;
   bool red_enabled;
   int32_t retransmission_settings;
   {
     rtc::CritScope cs(&crit_);
     // According to
     // http://www.etsi.org/deliver/etsi_ts/126100_126199/126114/12.07.00_60/
     // ts_126114v120700p.pdf Section 7.4.5:
     // The MTSI client shall add the payload bytes as defined in this clause
     // onto the last RTP packet in each group of packets which make up a key
     // frame (I-frame or IDR frame in H.264 (AVC), or an IRAP picture in H.265
     // (HEVC)). The MTSI client may also add the payload bytes onto the last RTP
     // packet in each group of packets which make up another type of frame
     // (e.g. a P-Frame) only if the current value is different from the previous
     // value sent.
     if (video_header) {
       // Set rotation when key frame or when changed (to follow standard).
       // Or when different from 0 (to follow current receiver implementation).
       VideoRotation current_rotation = video_header->rotation;
       if (frame_type == kVideoFrameKey || current_rotation != last_rotation_ ||
           current_rotation != kVideoRotation_0)
-        rtp_header->SetExtension<VideoOrientation>(current_rotation);
+        rtp_header_with_extensions->SetExtension<VideoOrientation>(
+            current_rotation);
       last_rotation_ = current_rotation;
       // Report content type only for key frames.
       if (frame_type == kVideoFrameKey &&
           video_header->content_type != VideoContentType::UNSPECIFIED) {
-        rtp_header->SetExtension<VideoContentTypeExtension>(
+        rtp_header_with_extensions->SetExtension<VideoContentTypeExtension>(
             video_header->content_type);
       }
     }
 
     // FEC settings.
     const FecProtectionParams& fec_params =
         frame_type == kVideoFrameKey ? key_fec_params_ : delta_fec_params_;
     if (flexfec_enabled())
       flexfec_sender_->SetFecParameters(fec_params);
     if (ulpfec_enabled())
       ulpfec_generator_.SetFecParameters(fec_params);
 
     fec_packet_overhead = CalculateFecPacketOverhead();
     red_enabled = this->red_enabled();
     retransmission_settings = retransmission_settings_;
   }
 
   size_t packet_capacity = rtp_sender_->MaxRtpPacketSize() -
                            fec_packet_overhead -
                            (rtp_sender_->RtxStatus() ? kRtxHeaderSize : 0);
   RTC_DCHECK_LE(packet_capacity, rtp_header->capacity());
   RTC_DCHECK_GT(packet_capacity, rtp_header->headers_size());
+  RTC_DCHECK_GT(packet_capacity, rtp_header_with_extensions->headers_size());
   size_t max_data_payload_length = packet_capacity - rtp_header->headers_size();
+  size_t last_packet_extensions_len =
+      rtp_header_with_extensions->headers_size() - rtp_header->headers_size();
 
   std::unique_ptr<RtpPacketizer> packetizer(RtpPacketizer::Create(
-      video_type, max_data_payload_length,
+      video_type, max_data_payload_length, last_packet_extensions_len,
       video_header ? &(video_header->codecHeader) : nullptr, frame_type));
   // Media packet storage.
   StorageType storage = packetizer->GetStorageType(retransmission_settings);
 
   // TODO(changbin): we currently don't support to configure the codec to
   // output multiple partitions for VP8. Should remove below check after the
   // issue is fixed.
   const RTPFragmentationHeader* frag =
       (video_type == kRtpVideoVp8) ? nullptr : fragmentation;
   packetizer->SetPayloadData(payload_data, payload_size, frag);
 
   bool first_frame = first_frame_sent_();
-  bool first = true;
   bool last = false;

[danilchap, 2017/05/12 08:34:40]

is it used outside for loop?

[ilnik, 2017/05/12 09:17:57]

No. Moved inside.

-  while (!last) {
-    std::unique_ptr<RtpPacketToSend> packet(new RtpPacketToSend(*rtp_header));
-
-    if (!packetizer->NextPacket(packet.get(), &last))
+  size_t total_packets = packetizer->TotalPackets();

[danilchap, 2017/05/12 08:34:40]

may be num_packets

[ilnik, 2017/05/12 09:17:57]

Done.

+  for (size_t current_packet = 0; current_packet < total_packets;

[danilchap, 2017/05/12 08:34:40]

may be name iterator to show it is not a packet (specially that there lot's of
variables nearby that are), but an index.
Simple 'i' might work.

[ilnik, 2017/05/12 09:17:57]

Done.

+       ++current_packet) {
+    RtpPacketToSend* rtp_header_to_use = rtp_header.get();
+    last = (current_packet + 1) == total_packets;
+    if (last)
+      rtp_header_to_use = rtp_header_with_extensions.get();
+    std::unique_ptr<RtpPacketToSend> packet(
+        new RtpPacketToSend(*rtp_header_to_use));
+    if (!packetizer->NextPacket(packet.get()))
       return false;
-    RTC_DCHECK_LE(packet->payload_size(), max_data_payload_length);
+    if (last) {

[danilchap, 2017/05/12 08:34:40]

may be instead of branch
RTC_DCHECK_LE(packet->size(), packet_capacity);

[ilnik, 2017/05/12 09:17:57]

Great idea. Done.

+      RTC_DCHECK_LE(packet->payload_size(),
+                    max_data_payload_length - last_packet_extensions_len);
+    } else {
+      RTC_DCHECK_LE(packet->payload_size(), max_data_payload_length);
+    }
 
     if (!rtp_sender_->AssignSequenceNumber(packet.get()))
       return false;
 
     const bool protect_packet =
         (packetizer->GetProtectionType() == kProtectedPacket);
     if (flexfec_enabled()) {
       // TODO(brandtr): Remove the FlexFEC code path when FlexfecSender
       // is wired up to PacedSender instead.
       SendVideoPacketWithFlexfec(std::move(packet), storage, protect_packet);
     } else if (red_enabled) {
       SendVideoPacketAsRedMaybeWithUlpfec(std::move(packet), storage,
                                           protect_packet);
     } else {
       SendVideoPacket(std::move(packet), storage);
     }
 
     if (first_frame) {
-      if (first) {
+      if (current_packet == 0) {
         LOG(LS_INFO)
             << "Sent first RTP packet of the first video frame (pre-pacer)";
       }
       if (last) {
         LOG(LS_INFO)
             << "Sent last RTP packet of the first video frame (pre-pacer)";
       }
     }
-    first = false;
   }
 
   TRACE_EVENT_ASYNC_END1("webrtc", "Video", capture_time_ms, "timestamp",
                          rtp_timestamp);
   return true;
 }
 
 uint32_t RTPSenderVideo::VideoBitrateSent() const {
   rtc::CritScope cs(&stats_crit_);
   return video_bitrate_.Rate(clock_->TimeInMilliseconds()).value_or(0);
 }
 
 uint32_t RTPSenderVideo::FecOverheadRate() const {
   rtc::CritScope cs(&stats_crit_);
   return fec_bitrate_.Rate(clock_->TimeInMilliseconds()).value_or(0);
 }
 
 int RTPSenderVideo::SelectiveRetransmissions() const {
   rtc::CritScope cs(&crit_);
   return retransmission_settings_;
 }
 
 void RTPSenderVideo::SetSelectiveRetransmissions(uint8_t settings) {
   rtc::CritScope cs(&crit_);
   retransmission_settings_ = settings;
 }
 
 }  // namespace webrtc