Convert Vp8 Rtp headers to frame references.

webrtc/modules/video_coding/packet_buffer_unittest.cc

 /*
  *  Copyright (c) 2016 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 <cstring>
 #include <limits>
 
 #include "webrtc/modules/video_coding/frame_object.h"
 #include "webrtc/modules/video_coding/packet_buffer.h"
 
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webrtc/base/random.h"
 
 namespace webrtc {
 namespace video_coding {
 
 class TestPacketBuffer : public ::testing::Test,
                          public OnCompleteFrameCallback {
  protected:
   TestPacketBuffer()
-      : rand_(0x8739211), packet_buffer_(kStartSize, kMaxSize, this) {}
+      : rand_(0x8739211),
+        packet_buffer_(new PacketBuffer(kStartSize, kMaxSize, this)) {}
 
   uint16_t Rand() { return rand_.Rand(std::numeric_limits<uint16_t>::max()); }
 
   void OnCompleteFrame(std::unique_ptr<FrameObject> frame) override {
-    frames_from_callback_.emplace_back(std::move(frame));
+    uint16_t pid = frame->picture_id;
+    auto frame_it = frames_from_callback_.find(pid);
+    if (frame_it != frames_from_callback_.end()) {
+      ADD_FAILURE() << "Already received frame with picture id: " << pid;
+      return;
+    }
+
+    frames_from_callback_.insert(
+        make_pair(frame->picture_id, std::move(frame)));
   }
 
   void TearDown() override {
-    // All FrameObjects must be destroyed before the PacketBuffer since
-    // a FrameObject will try to remove itself from the packet buffer
+    // All frame objects must be destroyed before the packet buffer since
+    // a frame object will try to remove itself from the packet buffer
     // upon destruction.
     frames_from_callback_.clear();
   }
 
+  // Insert a generic packet into the packet buffer.
+  void InsertGeneric(uint16_t seq_num,              // packet sequence number
+                     bool keyframe,                 // is keyframe
+                     bool first,                    // is first packet of frame
+                     bool last,                     // is last packet of frame
+                     size_t data_size = 0,          // size of data
+                     uint8_t* data = nullptr) {     // data pointer
+    VCMPacket packet;
+    packet.codec = kVideoCodecGeneric;
+    packet.seqNum = seq_num;
+    packet.frameType = keyframe ? kVideoFrameKey : kVideoFrameDelta;
+    packet.isFirstPacket = first;
+    packet.markerBit = last;
+    packet.sizeBytes = data_size;
+    packet.dataPtr = data;
+
+    EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
+  }
+
+  // Insert a Vp8 packet into the packet buffer.
+  void InsertVp8(uint16_t seq_num,              // packet sequence number
+                 bool keyframe,                 // is keyframe
+                 bool first,                    // is first packet of frame
+                 bool last,                     // is last packet of frame
+                 bool sync = false,             // is sync frame
+                 int32_t pid = kNoPictureId,    // picture id
+                 uint8_t tid = kNoTemporalIdx,  // temporal id
+                 int32_t tl0 = kNoTl0PicIdx,    // tl0 pic index
+                 size_t data_size = 0,          // size of data
+                 uint8_t* data = nullptr) {     // data pointer
+    VCMPacket packet;
+    packet.codec = kVideoCodecVP8;
+    packet.seqNum = seq_num;
+    packet.frameType = keyframe ? kVideoFrameKey : kVideoFrameDelta;
+    packet.isFirstPacket = first;
+    packet.markerBit = last;
+    packet.sizeBytes = data_size;
+    packet.dataPtr = data;
+    packet.codecSpecificHeader.codecHeader.VP8.pictureId = pid % (1 << 15);
+    packet.codecSpecificHeader.codecHeader.VP8.temporalIdx = tid;
+    packet.codecSpecificHeader.codecHeader.VP8.tl0PicIdx = tl0;
+    packet.codecSpecificHeader.codecHeader.VP8.layerSync = sync;
+
+    EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
+  }
+
+  // Check if a frame with picture id |pid| has been delivered from the packet
+  // buffer, and if so, if it has the references specified by |refs|.
+  template <typename... T>
+  void CheckReferences(uint16_t pid, T... refs) const {
+    auto frame_it = frames_from_callback_.find(pid);
+    if (frame_it == frames_from_callback_.end()) {
+      ADD_FAILURE() << "Could not find frame with picture id " << pid;
+      return;
+    }
+
+    std::set<uint16_t> actual_refs;
+    for (uint8_t r = 0; r < frame_it->second->num_references; ++r) {
+      actual_refs.insert(frame_it->second->references[r]);
+    }
+
+    std::set<uint16_t> expected_refs;
+    RefsToSet(&expected_refs, refs...);
+
+    ASSERT_EQ(expected_refs, actual_refs);
+  }
+
+  template <typename... T>
+  void RefsToSet(std::set<uint16_t>* m, uint16_t ref, T... refs) const {
+    m->insert(ref);
+    RefsToSet(m, refs...);
+  }
+
+  void RefsToSet(std::set<uint16_t>* m) const {}
+
   const int kStartSize = 16;
   const int kMaxSize = 64;
 
   Random rand_;
-  PacketBuffer packet_buffer_;
-  std::vector<std::unique_ptr<FrameObject>> frames_from_callback_;
+  std::unique_ptr<PacketBuffer> packet_buffer_;
+  std::map<uint16_t, std::unique_ptr<FrameObject>> frames_from_callback_;
 };
 
 TEST_F(TestPacketBuffer, InsertOnePacket) {
   VCMPacket packet;
   packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 }
 
 TEST_F(TestPacketBuffer, InsertMultiplePackets) {
   VCMPacket packet;
   packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 }
 
 TEST_F(TestPacketBuffer, InsertDuplicatePacket) {
   VCMPacket packet;
   packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 }
 
 TEST_F(TestPacketBuffer, ExpandBuffer) {
-  VCMPacket packet;
-  packet.seqNum = Rand();
+  uint16_t seq_num = Rand();
 
   for (int i = 0; i < kStartSize + 1; ++i) {
-    EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-    ++packet.seqNum;
+    //            seq_num    , keyframe, first, last
+    InsertGeneric(seq_num + i, true    , true , true);
   }
 }
 
 TEST_F(TestPacketBuffer, ExpandBufferOverflow) {
-  VCMPacket packet;
-  packet.seqNum = Rand();
+  uint16_t seq_num = Rand();
 
   for (int i = 0; i < kMaxSize; ++i) {
-    EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-    ++packet.seqNum;
+    //            seq_num    , keyframe, first, last
+    InsertGeneric(seq_num + i, true    , true , true);
   }
 
-  EXPECT_FALSE(packet_buffer_.InsertPacket(packet));
+  VCMPacket packet;
+  packet.seqNum = seq_num + kMaxSize + 1;
+  packet.sizeBytes = 1;
+  EXPECT_FALSE(packet_buffer_->InsertPacket(packet));
 }
 
-TEST_F(TestPacketBuffer, OnePacketOneFrame) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.markerBit = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+TEST_F(TestPacketBuffer, GenericOnePacketOneFrame) {
+  //            seq_num, keyframe, first, last
+  InsertGeneric(Rand() , true    , true , true);
+  ASSERT_EQ(1UL, frames_from_callback_.size());
+}
+
+TEST_F(TestPacketBuffer, GenericTwoPacketsTwoFrames) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyframe, first, last
+  InsertGeneric(seq_num    , true    , true , true);
+  InsertGeneric(seq_num + 1, true    , true , true);
+
+  EXPECT_EQ(2UL, frames_from_callback_.size());
+}
+
+TEST_F(TestPacketBuffer, GenericTwoPacketsOneFrames) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyframe, first, last
+  InsertGeneric(seq_num    , true    , true , false);
+  InsertGeneric(seq_num + 1, true    , false, true);
+
   EXPECT_EQ(1UL, frames_from_callback_.size());
 }
 
-TEST_F(TestPacketBuffer, TwoPacketsTwoFrames) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.markerBit = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(2UL, frames_from_callback_.size());
-}
+TEST_F(TestPacketBuffer, GenericThreePacketReorderingOneFrame) {
+  uint16_t seq_num = Rand();
 
-TEST_F(TestPacketBuffer, TwoPacketsOneFrames) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  packet.markerBit = true;
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  //            seq_num    , keyframe, first, last
+  InsertGeneric(seq_num    , true    , true , false);
+  InsertGeneric(seq_num + 2, true    , false, true);
+  InsertGeneric(seq_num + 1, true    , false, false);
+
   EXPECT_EQ(1UL, frames_from_callback_.size());
 }
 
-TEST_F(TestPacketBuffer, ThreePacketReorderingOneFrame) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.isFirstPacket = false;
-  packet.markerBit = true;
-  packet.seqNum += 2;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.markerBit = false;
-  packet.seqNum -= 1;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(1UL, frames_from_callback_.size());
-}
-
-TEST_F(TestPacketBuffer, IndexWrapOneFrame) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.seqNum = kStartSize - 1;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.isFirstPacket = false;
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.markerBit = true;
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(1UL, frames_from_callback_.size());
-}
-
 TEST_F(TestPacketBuffer, DiscardOldPacket) {
   uint16_t seq_num = Rand();
   VCMPacket packet;
   packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   packet.seqNum += 2;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 
   for (int i = 3; i < kMaxSize; ++i) {
     ++packet.seqNum;
-    EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+    EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   }
 
   ++packet.seqNum;
-  EXPECT_FALSE(packet_buffer_.InsertPacket(packet));
-  packet_buffer_.ClearTo(seq_num + 1);
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_FALSE(packet_buffer_->InsertPacket(packet));
+  packet_buffer_->ClearTo(seq_num + 1);
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 }
 
 TEST_F(TestPacketBuffer, DiscardMultipleOldPackets) {
   uint16_t seq_num = Rand();
   VCMPacket packet;
   packet.seqNum = seq_num;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   packet.seqNum += 2;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
 
   for (int i = 3; i < kMaxSize; ++i) {
     ++packet.seqNum;
-    EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+    EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   }
 
-  packet_buffer_.ClearTo(seq_num + 15);
+  packet_buffer_->ClearTo(seq_num + 15);
   for (int i = 0; i < 15; ++i) {
     ++packet.seqNum;
-    EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+    EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   }
   for (int i = 15; i < kMaxSize; ++i) {
     ++packet.seqNum;
-    EXPECT_FALSE(packet_buffer_.InsertPacket(packet));
+    EXPECT_FALSE(packet_buffer_->InsertPacket(packet));
   }
 }
 
+TEST_F(TestPacketBuffer, GenericFrames) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyf , first, last
+  InsertGeneric(seq_num    , true , true , true);
+  InsertGeneric(seq_num + 1, false, true , true);
+  InsertGeneric(seq_num + 2, false, true , true);
+  InsertGeneric(seq_num + 3, false, true , true);
+
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+  CheckReferences(seq_num);
+  CheckReferences(seq_num + 1, seq_num);
+  CheckReferences(seq_num + 2, seq_num + 1);
+  CheckReferences(seq_num + 3, seq_num + 2);
+}
+
+TEST_F(TestPacketBuffer, GenericFramesReordered) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyf , first, last
+  InsertGeneric(seq_num + 1, false, true , true);
+  InsertGeneric(seq_num    , true , true , true);
+  InsertGeneric(seq_num + 3, false, true , true);
+  InsertGeneric(seq_num + 2, false, true , true);
+
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+  CheckReferences(seq_num);
+  CheckReferences(seq_num + 1, seq_num);
+  CheckReferences(seq_num + 2, seq_num + 1);
+  CheckReferences(seq_num + 3, seq_num + 2);
+}
+
 TEST_F(TestPacketBuffer, GetBitstreamFromFrame) {
   // "many bitstream, such data" with null termination.
   uint8_t many[] = {0x6d, 0x61, 0x6e, 0x79, 0x20};
   uint8_t bitstream[] = {0x62, 0x69, 0x74, 0x73, 0x74, 0x72,
                          0x65, 0x61, 0x6d, 0x2c, 0x20};
   uint8_t such[] = {0x73, 0x75, 0x63, 0x68, 0x20};
   uint8_t data[] = {0x64, 0x61, 0x74, 0x61, 0x0};
   uint8_t
       result[sizeof(many) + sizeof(bitstream) + sizeof(such) + sizeof(data)];
 
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyf , first, last , data_size        , data
+  InsertGeneric(seq_num    , true , true , false, sizeof(many)     , many);
+  InsertGeneric(seq_num + 1, false, false, false, sizeof(bitstream), bitstream);
+  InsertGeneric(seq_num + 2, false, false, false, sizeof(such)     , such);
+  InsertGeneric(seq_num + 3, false, false, true , sizeof(data)     , data);
+
+  ASSERT_EQ(1UL, frames_from_callback_.size());
+  CheckReferences(seq_num + 3);
+  EXPECT_TRUE(frames_from_callback_[seq_num + 3]->GetBitstream(result));
+  EXPECT_EQ(std::strcmp("many bitstream, such data",
+            reinterpret_cast<char*>(result)),
+            0);
+}
+
+TEST_F(TestPacketBuffer, FreeSlotsOnFrameDestruction) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyf , first, last
+  InsertGeneric(seq_num    , true , true , false);
+  InsertGeneric(seq_num + 1, false, false, false);
+  InsertGeneric(seq_num + 2, false, false, true);
+  EXPECT_EQ(1UL, frames_from_callback_.size());
+
+  frames_from_callback_.clear();
+
+  //            seq_num    , keyf , first, last
+  InsertGeneric(seq_num    , true , true , false);
+  InsertGeneric(seq_num + 1, false, false, false);
+  InsertGeneric(seq_num + 2, false, false, true);
+  EXPECT_EQ(1UL, frames_from_callback_.size());
+}
+
+TEST_F(TestPacketBuffer, Flush) {
+  uint16_t seq_num = Rand();
+
+  //            seq_num    , keyf , first, last
+  InsertGeneric(seq_num    , true , true , false);
+  InsertGeneric(seq_num + 1, false, false, false);
+  InsertGeneric(seq_num + 2, false, false, true);
+  EXPECT_EQ(1UL, frames_from_callback_.size());
+
+  packet_buffer_->Flush();
+
+  //            seq_num                 , keyf , first, last
+  InsertGeneric(seq_num + kStartSize    , true , true , false);
+  InsertGeneric(seq_num + kStartSize + 1, false, false, false);
+  InsertGeneric(seq_num + kStartSize + 2, false, false, true);
+  EXPECT_EQ(2UL, frames_from_callback_.size());
+}
+
+TEST_F(TestPacketBuffer, InvalidateFrameByFlushing) {
   VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.seqNum = 0xfffe;
-  packet.dataPtr = many;
-  packet.sizeBytes = sizeof(many);
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  packet.isFirstPacket = false;
-  ++packet.seqNum;
-  packet.dataPtr = bitstream;
-  packet.sizeBytes = sizeof(bitstream);
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  ++packet.seqNum;
-  packet.dataPtr = such;
-  packet.sizeBytes = sizeof(such);
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  packet.markerBit = true;
-  ++packet.seqNum;
-  packet.dataPtr = data;
-  packet.sizeBytes = sizeof(data);
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  ASSERT_EQ(1UL, frames_from_callback_.size());
-
-  EXPECT_TRUE(frames_from_callback_[0]->GetBitstream(result));
-  EXPECT_EQ(
-      std::strcmp("many bitstream, such data", reinterpret_cast<char*>(result)),
-      0);
-}
-
-TEST_F(TestPacketBuffer, FreeSlotsOnFrameDestruction) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.isFirstPacket = false;
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  ++packet.seqNum;
-  packet.markerBit = true;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(1UL, frames_from_callback_.size());
-
-  frames_from_callback_.clear();
-
-  packet.isFirstPacket = true;
-  packet.markerBit = false;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  packet.isFirstPacket = false;
-  ++packet.seqNum;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(0UL, frames_from_callback_.size());
-  ++packet.seqNum;
-  packet.markerBit = true;
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(1UL, frames_from_callback_.size());
-}
-
-TEST_F(TestPacketBuffer, Flush) {
-  VCMPacket packet;
+  packet.codec = kVideoCodecGeneric;
+  packet.frameType = kVideoFrameKey;
   packet.isFirstPacket = true;
   packet.markerBit = true;
   packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  packet_buffer_.Flush();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
-  EXPECT_EQ(2UL, frames_from_callback_.size());
-}
-
-TEST_F(TestPacketBuffer, InvalidateFrameByFlushing) {
-  VCMPacket packet;
-  packet.isFirstPacket = true;
-  packet.markerBit = true;
-  packet.seqNum = Rand();
-  EXPECT_TRUE(packet_buffer_.InsertPacket(packet));
+  EXPECT_TRUE(packet_buffer_->InsertPacket(packet));
   ASSERT_EQ(1UL, frames_from_callback_.size());
 
-  packet_buffer_.Flush();
-  EXPECT_FALSE(frames_from_callback_[0]->GetBitstream(nullptr));
+  packet_buffer_->Flush();
+  EXPECT_FALSE(frames_from_callback_.begin()->second->GetBitstream(nullptr));
+}
+
+TEST_F(TestPacketBuffer, Vp8NoPictureId) {
+  uint16_t seq_num = Rand();
+
+  //        seq_num     , keyf , first, last
+  InsertVp8(seq_num     , true , true , false);
+  InsertVp8(seq_num + 1 , false, false, false);
+  InsertVp8(seq_num + 2 , false, false, true);
+  ASSERT_EQ(1UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 3 , false, true , false);
+  InsertVp8(seq_num + 4 , false, false, true);
+  ASSERT_EQ(2UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 5 , false, true , false);
+  InsertVp8(seq_num + 6 , false, false, false);
+  InsertVp8(seq_num + 7 , false, false, false);
+  InsertVp8(seq_num + 8 , false, false, true);
+  ASSERT_EQ(3UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 9 , false, true , true);
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 10, false, true , false);
+  InsertVp8(seq_num + 11, false, false, true);
+  ASSERT_EQ(5UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 12, true , true , true);
+  ASSERT_EQ(6UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 13, false, true , false);
+  InsertVp8(seq_num + 14, false, false, false);
+  InsertVp8(seq_num + 15, false, false, false);
+  InsertVp8(seq_num + 16, false, false, false);
+  InsertVp8(seq_num + 17, false, false, true);
+  ASSERT_EQ(7UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 18, false, true , true);
+  ASSERT_EQ(8UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 19, false, true , false);
+  InsertVp8(seq_num + 20, false, false, true);
+  ASSERT_EQ(9UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 21, false, true , true);
+
+  ASSERT_EQ(10UL, frames_from_callback_.size());
+  CheckReferences(seq_num + 2);
+  CheckReferences(seq_num + 4, seq_num + 2);
+  CheckReferences(seq_num + 8, seq_num + 4);
+  CheckReferences(seq_num + 9, seq_num + 8);
+  CheckReferences(seq_num + 11, seq_num + 9);
+  CheckReferences(seq_num + 12);
+  CheckReferences(seq_num + 17, seq_num + 12);
+  CheckReferences(seq_num + 18, seq_num + 17);
+  CheckReferences(seq_num + 20, seq_num + 18);
+  CheckReferences(seq_num + 21, seq_num + 20);
+}
+
+TEST_F(TestPacketBuffer, Vp8NoPictureIdReordered) {
+  uint16_t seq_num = 0xfffa;
+
+  //        seq_num     , keyf , first, last
+  InsertVp8(seq_num + 1 , false, false, false);
+  InsertVp8(seq_num     , true , true , false);
+  InsertVp8(seq_num + 2 , false, false, true);
+  InsertVp8(seq_num + 4 , false, false, true);
+  InsertVp8(seq_num + 6 , false, false, false);
+  InsertVp8(seq_num + 3 , false, true , false);
+  InsertVp8(seq_num + 7 , false, false, false);
+  InsertVp8(seq_num + 5 , false, true , false);
+  InsertVp8(seq_num + 9 , false, true , true);
+  InsertVp8(seq_num + 10, false, true , false);
+  InsertVp8(seq_num + 8 , false, false, true);
+  InsertVp8(seq_num + 13, false, true , false);
+  InsertVp8(seq_num + 14, false, false, false);
+  InsertVp8(seq_num + 12, true , true , true);
+  InsertVp8(seq_num + 11, false, false, true);
+  InsertVp8(seq_num + 16, false, false, false);
+  InsertVp8(seq_num + 19, false, true , false);
+  InsertVp8(seq_num + 15, false, false, false);
+  InsertVp8(seq_num + 17, false, false, true);
+  InsertVp8(seq_num + 20, false, false, true);
+  InsertVp8(seq_num + 21, false, true , true);
+  InsertVp8(seq_num + 18, false, true , true);
+
+  ASSERT_EQ(10UL, frames_from_callback_.size());
+  CheckReferences(seq_num + 2);
+  CheckReferences(seq_num + 4, seq_num + 2);
+  CheckReferences(seq_num + 8, seq_num + 4);
+  CheckReferences(seq_num + 9, seq_num + 8);
+  CheckReferences(seq_num + 11, seq_num + 9);
+  CheckReferences(seq_num + 12);
+  CheckReferences(seq_num + 17, seq_num + 12);
+  CheckReferences(seq_num + 18, seq_num + 17);
+  CheckReferences(seq_num + 20, seq_num + 18);
+  CheckReferences(seq_num + 21, seq_num + 20);
+}
+
+
+TEST_F(TestPacketBuffer, Vp8KeyFrameReferences) {
+  uint16_t pid = Rand();
+  //        seq_num, keyf, first, last, sync , pid, tid, tl0
+  InsertVp8(Rand() , true, true , true, false, pid, 0  , 0);
+
+  ASSERT_EQ(1UL, frames_from_callback_.size());
+  CheckReferences(pid);
+}
+
+// Test with 1 temporal layer.
+TEST_F(TestPacketBuffer, Vp8TemporalLayers_0) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num    , keyf , first, last, sync , pid    , tid, tl0
+  InsertVp8(seq_num    , true , true , true, false, pid    , 0  , 1);
+  InsertVp8(seq_num + 1, false, true , true, false, pid + 1, 0  , 2);
+  InsertVp8(seq_num + 2, false, true , true, false, pid + 2, 0  , 3);
+  InsertVp8(seq_num + 3, false, true , true, false, pid + 3, 0  , 4);
+
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid + 1);
+  CheckReferences(pid + 3, pid + 2);
+}
+
+// Test with 1 temporal layer.
+TEST_F(TestPacketBuffer, Vp8TemporalLayersReordering_0) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num    , keyf , first, last, sync , pid    , tid, tl0
+  InsertVp8(seq_num    , true , true , true, false, pid    , 0  , 1);
+  InsertVp8(seq_num + 1, false, true , true, false, pid + 1, 0  , 2);
+  InsertVp8(seq_num + 3, false, true , true, false, pid + 3, 0  , 4);
+  InsertVp8(seq_num + 2, false, true , true, false, pid + 2, 0  , 3);
+  InsertVp8(seq_num + 5, false, true , true, false, pid + 5, 0  , 6);
+  InsertVp8(seq_num + 6, false, true , true, false, pid + 6, 0  , 7);
+  InsertVp8(seq_num + 4, false, true , true, false, pid + 4, 0  , 5);
+
+  ASSERT_EQ(7UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid + 1);
+  CheckReferences(pid + 3, pid + 2);
+  CheckReferences(pid + 4, pid + 3);
+  CheckReferences(pid + 5, pid + 4);
+  CheckReferences(pid + 6, pid + 5);
+}
+
+// Test with 2 temporal layers in a 01 pattern.
+TEST_F(TestPacketBuffer, Vp8TemporalLayers_01) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num    , keyf , first, last, sync , pid    , tid, tl0
+  InsertVp8(seq_num    , true , true , true, false, pid    , 0, 255);
+  InsertVp8(seq_num + 1, false, true , true, true , pid + 1, 1, 255);
+  InsertVp8(seq_num + 2, false, true , true, false, pid + 2, 0, 0);
+  InsertVp8(seq_num + 3, false, true , true, false, pid + 3, 1, 0);
+
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid);
+  CheckReferences(pid + 3, pid + 1, pid + 2);
+}
+
+// Test with 2 temporal layers in a 01 pattern.
+TEST_F(TestPacketBuffer, Vp8TemporalLayersReordering_01) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num    , keyf , first, last, sync , pid    , tid, tl0
+  InsertVp8(seq_num + 1, false, true , true, true , pid + 1, 1  , 255);
+  InsertVp8(seq_num    , true , true , true, false, pid    , 0  , 255);
+  InsertVp8(seq_num + 3, false, true , true, false, pid + 3, 1  , 0);
+  InsertVp8(seq_num + 5, false, true , true, false, pid + 5, 1  , 1);
+  InsertVp8(seq_num + 2, false, true , true, false, pid + 2, 0  , 0);
+  InsertVp8(seq_num + 4, false, true , true, false, pid + 4, 0  , 1);
+  InsertVp8(seq_num + 6, false, true , true, false, pid + 6, 0  , 2);
+  InsertVp8(seq_num + 7, false, true , true, false, pid + 7, 1  , 2);
+
+  ASSERT_EQ(8UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid);
+  CheckReferences(pid + 3, pid + 1, pid + 2);
+  CheckReferences(pid + 4, pid + 2);
+  CheckReferences(pid + 5, pid + 3, pid + 4);
+  CheckReferences(pid + 6, pid + 4);
+  CheckReferences(pid + 7, pid + 5, pid + 6);
+}
+
+// Test with 3 temporal layers in a 0212 pattern.
+TEST_F(TestPacketBuffer, Vp8TemporalLayers_0212) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num     , keyf , first, last, sync  , pid     , tid, tl0
+  InsertVp8(seq_num     , true , true , true , false, pid     , 0  , 55);
+  InsertVp8(seq_num + 1 , false, true , true , true , pid + 1 , 2  , 55);
+  InsertVp8(seq_num + 2 , false, true , true , true , pid + 2 , 1  , 55);
+  InsertVp8(seq_num + 3 , false, true , true , false, pid + 3 , 2  , 55);
+  InsertVp8(seq_num + 4 , false, true , true , false, pid + 4 , 0  , 56);
+  InsertVp8(seq_num + 5 , false, true , true , false, pid + 5 , 2  , 56);
+  InsertVp8(seq_num + 6 , false, true , true , false, pid + 6 , 1  , 56);
+  InsertVp8(seq_num + 7 , false, true , true , false, pid + 7 , 2  , 56);
+  InsertVp8(seq_num + 8 , false, true , true , false, pid + 8 , 0  , 57);
+  InsertVp8(seq_num + 9 , false, true , true , true , pid + 9 , 2  , 57);
+  InsertVp8(seq_num + 10, false, true , true , true , pid + 10, 1  , 57);
+  InsertVp8(seq_num + 11, false, true , true , false, pid + 11, 2  , 57);
+
+  ASSERT_EQ(12UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1 , pid);
+  CheckReferences(pid + 2 , pid);
+  CheckReferences(pid + 3 , pid, pid + 1, pid + 2);
+  CheckReferences(pid + 4 , pid);
+  CheckReferences(pid + 5 , pid + 2, pid + 3, pid + 4);
+  CheckReferences(pid + 6 , pid + 2, pid + 4);
+  CheckReferences(pid + 7 , pid + 4, pid + 5, pid + 6);
+  CheckReferences(pid + 8 , pid + 4);
+  CheckReferences(pid + 9 , pid + 8);
+  CheckReferences(pid + 10, pid + 8);
+  CheckReferences(pid + 11, pid + 8, pid + 9, pid + 10);
+}
+
+// Test with 3 temporal layers in a 0212 pattern.
+TEST_F(TestPacketBuffer, Vp8TemporalLayersReordering_0212) {
+  uint16_t pid = 126;
+  uint16_t seq_num = Rand();
+
+  //        seq_num     , keyf , first, last, sync , pid     , tid, tl0
+  InsertVp8(seq_num + 1 , false, true , true, true , pid + 1 , 2  , 55);
+  InsertVp8(seq_num     , true , true , true, false, pid     , 0  , 55);
+  InsertVp8(seq_num + 2 , false, true , true, true , pid + 2 , 1  , 55);
+  InsertVp8(seq_num + 4 , false, true , true, false, pid + 4 , 0  , 56);
+  InsertVp8(seq_num + 5 , false, true , true, false, pid + 5 , 2  , 56);
+  InsertVp8(seq_num + 3 , false, true , true, false, pid + 3 , 2  , 55);
+  InsertVp8(seq_num + 7 , false, true , true, false, pid + 7 , 2  , 56);
+  InsertVp8(seq_num + 9 , false, true , true, true , pid + 9 , 2  , 57);
+  InsertVp8(seq_num + 6 , false, true , true, false, pid + 6 , 1  , 56);
+  InsertVp8(seq_num + 8 , false, true , true, false, pid + 8 , 0  , 57);
+  InsertVp8(seq_num + 11, false, true , true, false, pid + 11, 2  , 57);
+  InsertVp8(seq_num + 10, false, true , true, true , pid + 10, 1  , 57);
+
+  ASSERT_EQ(12UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1 , pid);
+  CheckReferences(pid + 2 , pid);
+  CheckReferences(pid + 3 , pid, pid + 1, pid + 2);
+  CheckReferences(pid + 4 , pid);
+  CheckReferences(pid + 5 , pid + 2, pid + 3, pid + 4);
+  CheckReferences(pid + 6 , pid + 2, pid + 4);
+  CheckReferences(pid + 7 , pid + 4, pid + 5, pid + 6);
+  CheckReferences(pid + 8 , pid + 4);
+  CheckReferences(pid + 9 , pid + 8);
+  CheckReferences(pid + 10, pid + 8);
+  CheckReferences(pid + 11, pid + 8, pid + 9, pid + 10);
+}
+
+TEST_F(TestPacketBuffer, Vp8InsertManyFrames_0212) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  const int keyframes_to_insert = 50;
+  const int frames_per_keyframe = 120;  // Should be a multiple of 4.
+  uint8_t tl0 = 128;
+
+  for (int k = 0; k < keyframes_to_insert; ++k) {
+    //        seq_num    , keyf , first, last , sync , pid    , tid, tl0
+    InsertVp8(seq_num    , true , true , true , false, pid    , 0  , tl0);
+    InsertVp8(seq_num + 1, false, true , true , true , pid + 1, 2  , tl0);
+    InsertVp8(seq_num + 2, false, true , true , true , pid + 2, 1  , tl0);
+    InsertVp8(seq_num + 3, false, true , true , false, pid + 3, 2  , tl0);
+    CheckReferences(pid);
+    CheckReferences(pid + 1, pid);
+    CheckReferences(pid + 2, pid);
+    CheckReferences(pid + 3, pid, pid + 1, pid + 2);
+    frames_from_callback_.clear();
+    ++tl0;
+
+    for (int f = 4; f < frames_per_keyframe; f += 4) {
+      uint16_t sf = seq_num + f;
+      uint16_t pidf = pid + f;
+
+      //        seq_num, keyf , first, last, sync , pid     , tid, tl0
+      InsertVp8(sf     , false, true , true, false, pidf    , 0  , tl0);
+      InsertVp8(sf + 1 , false, true , true, false, pidf + 1, 2  , tl0);
+      InsertVp8(sf + 2 , false, true , true, false, pidf + 2, 1  , tl0);
+      InsertVp8(sf + 3 , false, true , true, false, pidf + 3, 2  , tl0);
+      CheckReferences(pidf, pidf - 4);
+      CheckReferences(pidf + 1, pidf, pidf - 1, pidf - 2);
+      CheckReferences(pidf + 2, pidf, pidf - 2);
+      CheckReferences(pidf + 3, pidf, pidf + 1, pidf + 2);
+      frames_from_callback_.clear();
+      ++tl0;
+    }
+
+    pid += frames_per_keyframe;
+    seq_num += frames_per_keyframe;
+  }
+}
+
+TEST_F(TestPacketBuffer, Vp8LayerSync) {
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  //        seq_num     , keyf , first, last, sync , pid     , tid, tl0
+  InsertVp8(seq_num     , true , true , true, false, pid     , 0  , 0);
+  InsertVp8(seq_num + 1 , false, true , true, true , pid + 1 , 1  , 0);
+  InsertVp8(seq_num + 2 , false, true , true, false, pid + 2 , 0  , 1);
+  ASSERT_EQ(3UL, frames_from_callback_.size());
+
+  InsertVp8(seq_num + 4 , false, true , true, false, pid + 4 , 0  , 2);
+  InsertVp8(seq_num + 5 , false, true , true, true , pid + 5 , 1  , 2);
+  InsertVp8(seq_num + 6 , false, true , true, false, pid + 6 , 0  , 3);
+  InsertVp8(seq_num + 7 , false, true , true, false, pid + 7 , 1  , 3);
+
+  ASSERT_EQ(7UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid);
+  CheckReferences(pid + 4, pid + 2);
+  CheckReferences(pid + 5, pid + 4);
+  CheckReferences(pid + 6, pid + 4);
+  CheckReferences(pid + 7, pid + 6, pid + 5);
+}
+
+TEST_F(TestPacketBuffer, Vp8InsertLargeFrames) {
+  packet_buffer_.reset(new PacketBuffer(1 << 3, 1 << 12, this));
+  uint16_t pid = Rand();
+  uint16_t seq_num = Rand();
+
+  const uint16_t packets_per_frame = 1000;
+  uint16_t current = seq_num;
+  uint16_t end = current + packets_per_frame;
+
+  //        seq_num  , keyf , first, last , sync , pid, tid, tl0
+  InsertVp8(current++, true , true , false, false, pid, 0  , 0);
+  while (current != end)
+    InsertVp8(current++, false, false, false, false, pid, 0  , 0);
+  InsertVp8(current++, false, false, true , false, pid, 0  , 0);
+  end = current + packets_per_frame;
+
+  for (int f = 1; f < 4; ++f) {
+    InsertVp8(current++, false, true , false, false, pid + f, 0, f);
+    while (current != end)
+      InsertVp8(current++, false, false, false, false, pid + f, 0, f);
+    InsertVp8(current++, false, false, true , false, pid + f, 0, f);
+    end = current + packets_per_frame;
+  }
+
+  ASSERT_EQ(4UL, frames_from_callback_.size());
+  CheckReferences(pid);
+  CheckReferences(pid + 1, pid);
+  CheckReferences(pid + 2, pid + 1);
+  CheckReferences(pid + 3, pid + 2);
 }
 
 }  // namespace video_coding
 }  // namespace webrtc