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

webrtc/modules/rtp_rtcp/source/rtp_format_h264.h

Index: webrtc/modules/rtp_rtcp/source/rtp_format_h264.h
diff --git a/webrtc/modules/rtp_rtcp/source/rtp_format_h264.h b/webrtc/modules/rtp_rtcp/source/rtp_format_h264.h
index fe7b37832aa90f1208be3aa4c6dd48a5d6ed2c50..2219bf2ae2819ce29dd7b2a5ddc1b5ca0c93bc07 100644
--- a/webrtc/modules/rtp_rtcp/source/rtp_format_h264.h
+++ b/webrtc/modules/rtp_rtcp/source/rtp_format_h264.h
@@ -27,6 +27,7 @@ class RtpPacketizerH264 : public RtpPacketizer {
   // Initialize with payload from encoder.
   // The payload_data must be exactly one encoded H264 frame.
   RtpPacketizerH264(size_t max_payload_len,
+                    size_t last_packet_extension_len,
                     H264PacketizationMode packetization_mode);
 
   virtual ~RtpPacketizerH264();
@@ -37,10 +38,10 @@ class RtpPacketizerH264 : public RtpPacketizer {
 
   // Get the next payload with H264 payload header.
   // Write payload and set marker bit of the |packet|.
-  // The parameter |last_packet| is true for the last packet of the frame, false
-  // otherwise (i.e., call the function again to get the next packet).
   // Returns true on success, false otherwise.
-  bool NextPacket(RtpPacketToSend* rtp_packet, bool* last_packet) override;
+  bool NextPacket(RtpPacketToSend* rtp_packet) override;
+
+  size_t TotalPackets() override;
 
   ProtectionType GetProtectionType() override;
 
@@ -88,10 +89,12 @@ class RtpPacketizerH264 : public RtpPacketizer {
   void PacketizeFuA(size_t fragment_index);
   size_t PacketizeStapA(size_t fragment_index);
   void PacketizeSingleNalu(size_t fragment_index);
-  void NextAggregatePacket(RtpPacketToSend* rtp_packet);
+  void NextAggregatePacket(RtpPacketToSend* rtp_packet, bool last);
   void NextFragmentPacket(RtpPacketToSend* rtp_packet);
 
   const size_t max_payload_len_;
+  const size_t last_packet_extensions_len_;
+  size_t total_packets_;
   const H264PacketizationMode packetization_mode_;
   std::deque<Fragment> input_fragments_;
   std::queue<PacketUnit> packets_;