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

webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h

Index: webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h
diff --git a/webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h b/webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h
index 2036ce49f4b2b9d9b75c0c262427fd09db3c7119..7e30005cff6eae47721756e72d0ba90dd9bb50d6 100644
--- a/webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h
+++ b/webrtc/modules/rtp_rtcp/source/rtp_format_vp9.h
@@ -33,7 +33,9 @@ namespace webrtc {
 
 class RtpPacketizerVp9 : public RtpPacketizer {
  public:
-  RtpPacketizerVp9(const RTPVideoHeaderVP9& hdr, size_t max_payload_length);
+  RtpPacketizerVp9(const RTPVideoHeaderVP9& hdr,
+                   size_t max_payload_length,
+                   size_t last_packet_extension_len);
 
   virtual ~RtpPacketizerVp9();
 
@@ -50,10 +52,11 @@ class RtpPacketizerVp9 : public RtpPacketizer {
 
   // Gets the next payload with VP9 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* packet, bool* last_packet) override;
+  bool NextPacket(RtpPacketToSend* packet) override;
+
+  // Returns total number of packets will be produced.
+  size_t TotalPackets() override;
 
   typedef struct {
     size_t payload_start_pos;
@@ -69,10 +72,11 @@ class RtpPacketizerVp9 : public RtpPacketizer {
 
   // Writes the payload descriptor header and copies payload to the |buffer|.
   // |packet_info| determines which part of the payload to write.
-  // |bytes_to_send| contains the number of written bytes to the buffer.
+  // |last| indicates if the packet is last.
   // Returns true on success, false otherwise.
   bool WriteHeaderAndPayload(const PacketInfo& packet_info,
-                             RtpPacketToSend* packet) const;
+                             RtpPacketToSend* packet,
+                             bool last) const;
 
   // Writes payload descriptor header to |buffer|.
   // Returns true on success, false otherwise.
@@ -84,6 +88,7 @@ class RtpPacketizerVp9 : public RtpPacketizer {
   const size_t max_payload_length_;  // The max length in bytes of one packet.
   const uint8_t* payload_;           // The payload data to be packetized.
   size_t payload_size_;              // The size in bytes of the payload data.
+  const size_t last_packet_extension_len_;
   PacketInfoQueue packets_;
 
   RTC_DISALLOW_COPY_AND_ASSIGN(RtpPacketizerVp9);