Store PacketBuffer by value instead of as reference counted object

Bug: None
Change-Id: I5a594972e8a8dad731c927a1a374301e549f5d71
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/153887
Reviewed-by: Philip Eliasson <philipel@webrtc.org>
Reviewed-by: Åsa Persson <asapersson@webrtc.org>
Reviewed-by: Sam Zackrisson <saza@webrtc.org>
Commit-Queue: Danil Chapovalov <danilchap@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#29267}
diff --git a/modules/video_coding/packet_buffer.cc b/modules/video_coding/packet_buffer.cc
index b5f9fa7..a046b54 100644
--- a/modules/video_coding/packet_buffer.cc
+++ b/modules/video_coding/packet_buffer.cc
@@ -22,7 +22,6 @@
 #include "modules/rtp_rtcp/source/rtp_video_header.h"
 #include "modules/video_coding/codecs/h264/include/h264_globals.h"
 #include "modules/video_coding/frame_object.h"
-#include "rtc_base/atomic_ops.h"
 #include "rtc_base/checks.h"
 #include "rtc_base/logging.h"
 #include "rtc_base/numerics/mod_ops.h"
@@ -32,15 +31,6 @@
 namespace webrtc {
 namespace video_coding {
 
-rtc::scoped_refptr<PacketBuffer> PacketBuffer::Create(
-    Clock* clock,
-    size_t start_buffer_size,
-    size_t max_buffer_size,
-    OnAssembledFrameCallback* assembled_frame_callback) {
-  return rtc::scoped_refptr<PacketBuffer>(new PacketBuffer(
-      clock, start_buffer_size, max_buffer_size, assembled_frame_callback));
-}
-
 PacketBuffer::PacketBuffer(Clock* clock,
                            size_t start_buffer_size,
                            size_t max_buffer_size,
@@ -483,18 +473,6 @@
   return &data_buffer_[index];
 }
 
-int PacketBuffer::AddRef() const {
-  return rtc::AtomicOps::Increment(&ref_count_);
-}
-
-int PacketBuffer::Release() const {
-  int count = rtc::AtomicOps::Decrement(&ref_count_);
-  if (!count) {
-    delete this;
-  }
-  return count;
-}
-
 void PacketBuffer::UpdateMissingPackets(uint16_t seq_num) {
   if (!newest_inserted_seq_num_)
     newest_inserted_seq_num_ = seq_num;
diff --git a/modules/video_coding/packet_buffer.h b/modules/video_coding/packet_buffer.h
index 77fa6ad..ebbd974 100644
--- a/modules/video_coding/packet_buffer.h
+++ b/modules/video_coding/packet_buffer.h
@@ -16,7 +16,6 @@
 #include <set>
 #include <vector>
 
-#include "api/scoped_refptr.h"
 #include "api/video/encoded_image.h"
 #include "modules/include/module_common_types.h"
 #include "modules/video_coding/packet.h"
@@ -41,12 +40,11 @@
 
 class PacketBuffer {
  public:
-  static rtc::scoped_refptr<PacketBuffer> Create(
-      Clock* clock,
-      size_t start_buffer_size,
-      size_t max_buffer_size,
-      OnAssembledFrameCallback* frame_callback);
-
+  // Both |start_buffer_size| and |max_buffer_size| must be a power of 2.
+  PacketBuffer(Clock* clock,
+               size_t start_buffer_size,
+               size_t max_buffer_size,
+               OnAssembledFrameCallback* frame_callback);
   virtual ~PacketBuffer();
 
   // Returns true unless the packet buffer is cleared, which means that a key
@@ -65,16 +63,6 @@
   // Returns number of different frames seen in the packet buffer
   int GetUniqueFramesSeen() const;
 
-  int AddRef() const;
-  int Release() const;
-
- protected:
-  // Both |start_buffer_size| and |max_buffer_size| must be a power of 2.
-  PacketBuffer(Clock* clock,
-               size_t start_buffer_size,
-               size_t max_buffer_size,
-               OnAssembledFrameCallback* frame_callback);
-
  private:
   friend RtpFrameObject;
   // Since we want the packet buffer to be as packet type agnostic
@@ -181,8 +169,6 @@
   std::set<uint32_t> rtp_timestamps_history_set_ RTC_GUARDED_BY(crit_);
   // Stores the same unique timestamps in the order of insertion.
   std::queue<uint32_t> rtp_timestamps_history_queue_ RTC_GUARDED_BY(crit_);
-
-  mutable volatile int ref_count_ = 0;
 };
 
 }  // namespace video_coding
diff --git a/modules/video_coding/rtp_frame_reference_finder_unittest.cc b/modules/video_coding/rtp_frame_reference_finder_unittest.cc
index 9d9ba0b..5fe5c09 100644
--- a/modules/video_coding/rtp_frame_reference_finder_unittest.cc
+++ b/modules/video_coding/rtp_frame_reference_finder_unittest.cc
@@ -49,7 +49,6 @@
  protected:
   TestRtpFrameReferenceFinder()
       : rand_(0x8739211),
-        ref_packet_buffer_(new FakePacketBuffer()),
         reference_finder_(new RtpFrameReferenceFinder(this)),
         frames_from_callback_(FrameComp()) {}
 
@@ -78,15 +77,15 @@
     packet.video_header.frame_type = keyframe
                                          ? VideoFrameType::kVideoFrameKey
                                          : VideoFrameType::kVideoFrameDelta;
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
     packet.seqNum = seq_num_end;
     packet.video_header.is_last_packet_in_frame = true;
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
-    std::unique_ptr<RtpFrameObject> frame(
-        new RtpFrameObject(ref_packet_buffer_, seq_num_start, seq_num_end, 0, 0,
-                           0, {}, EncodedImageBuffer::Create(/*size=*/0)));
+    auto frame = std::make_unique<RtpFrameObject>(
+        &packet_buffer_, seq_num_start, seq_num_end, 0, 0, 0, RtpPacketInfos(),
+        EncodedImageBuffer::Create(/*size=*/0));
     reference_finder_->ManageFrame(std::move(frame));
   }
 
@@ -111,17 +110,17 @@
     vp8_header.temporalIdx = tid;
     vp8_header.tl0PicIdx = tl0;
     vp8_header.layerSync = sync;
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
     if (seq_num_start != seq_num_end) {
       packet.seqNum = seq_num_end;
       packet.video_header.is_last_packet_in_frame = true;
-      ref_packet_buffer_->InsertPacket(&packet);
+      packet_buffer_.InsertPacket(&packet);
     }
 
-    std::unique_ptr<RtpFrameObject> frame(
-        new RtpFrameObject(ref_packet_buffer_, seq_num_start, seq_num_end, 0, 0,
-                           0, {}, EncodedImageBuffer::Create(/*size=*/0)));
+    auto frame = std::make_unique<RtpFrameObject>(
+        &packet_buffer_, seq_num_start, seq_num_end, 0, 0, 0, RtpPacketInfos(),
+        EncodedImageBuffer::Create(/*size=*/0));
     reference_finder_->ManageFrame(std::move(frame));
   }
 
@@ -157,18 +156,18 @@
       vp9_header.ss_data_available = true;
       vp9_header.gof = *ss;
     }
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
     if (seq_num_start != seq_num_end) {
       packet.video_header.is_last_packet_in_frame = true;
       vp9_header.ss_data_available = false;
       packet.seqNum = seq_num_end;
-      ref_packet_buffer_->InsertPacket(&packet);
+      packet_buffer_.InsertPacket(&packet);
     }
 
-    std::unique_ptr<RtpFrameObject> frame(
-        new RtpFrameObject(ref_packet_buffer_, seq_num_start, seq_num_end, 0, 0,
-                           0, {}, EncodedImageBuffer::Create(/*size=*/0)));
+    auto frame = std::make_unique<RtpFrameObject>(
+        &packet_buffer_, seq_num_start, seq_num_end, 0, 0, 0, RtpPacketInfos(),
+        EncodedImageBuffer::Create(/*size=*/0));
     reference_finder_->ManageFrame(std::move(frame));
   }
 
@@ -200,17 +199,17 @@
     vp9_header.num_ref_pics = refs.size();
     for (size_t i = 0; i < refs.size(); ++i)
       vp9_header.pid_diff[i] = refs[i];
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
     if (seq_num_start != seq_num_end) {
       packet.seqNum = seq_num_end;
       packet.video_header.is_last_packet_in_frame = true;
-      ref_packet_buffer_->InsertPacket(&packet);
+      packet_buffer_.InsertPacket(&packet);
     }
 
-    std::unique_ptr<RtpFrameObject> frame(
-        new RtpFrameObject(ref_packet_buffer_, seq_num_start, seq_num_end, 0, 0,
-                           0, {}, EncodedImageBuffer::Create(/*size=*/0)));
+    auto frame = std::make_unique<RtpFrameObject>(
+        &packet_buffer_, seq_num_start, seq_num_end, 0, 0, 0, RtpPacketInfos(),
+        EncodedImageBuffer::Create(/*size=*/0));
     reference_finder_->ManageFrame(std::move(frame));
   }
 
@@ -231,17 +230,17 @@
     packet.video_header.frame_marking.temporal_id = tid;
     packet.video_header.frame_marking.tl0_pic_idx = tl0;
     packet.video_header.frame_marking.base_layer_sync = sync;
-    ref_packet_buffer_->InsertPacket(&packet);
+    packet_buffer_.InsertPacket(&packet);
 
     if (seq_num_start != seq_num_end) {
       packet.seqNum = seq_num_end;
       packet.video_header.is_last_packet_in_frame = true;
-      ref_packet_buffer_->InsertPacket(&packet);
+      packet_buffer_.InsertPacket(&packet);
     }
 
-    std::unique_ptr<RtpFrameObject> frame(
-        new RtpFrameObject(ref_packet_buffer_, seq_num_start, seq_num_end, 0, 0,
-                           0, {}, EncodedImageBuffer::Create(/*size=*/0)));
+    auto frame = std::make_unique<RtpFrameObject>(
+        &packet_buffer_, seq_num_start, seq_num_end, 0, 0, 0, RtpPacketInfos(),
+        EncodedImageBuffer::Create(/*size=*/0));
     reference_finder_->ManageFrame(std::move(frame));
   }
 
@@ -299,7 +298,7 @@
   void RefsToSet(std::set<int64_t>* m) const {}
 
   Random rand_;
-  rtc::scoped_refptr<FakePacketBuffer> ref_packet_buffer_;
+  FakePacketBuffer packet_buffer_;
   std::unique_ptr<RtpFrameReferenceFinder> reference_finder_;
   struct FrameComp {
     bool operator()(const std::pair<int64_t, uint8_t> f1,
diff --git a/modules/video_coding/video_packet_buffer_unittest.cc b/modules/video_coding/video_packet_buffer_unittest.cc
index ae41173..efe2ecc 100644
--- a/modules/video_coding/video_packet_buffer_unittest.cc
+++ b/modules/video_coding/video_packet_buffer_unittest.cc
@@ -32,8 +32,7 @@
       : scoped_field_trials_(field_trials),
         rand_(0x7732213),
         clock_(new SimulatedClock(0)),
-        packet_buffer_(
-            PacketBuffer::Create(clock_.get(), kStartSize, kMaxSize, this)) {}
+        packet_buffer_(clock_.get(), kStartSize, kMaxSize, this) {}
 
   uint16_t Rand() { return rand_.Rand<uint16_t>(); }
 
@@ -73,7 +72,7 @@
     packet.sizeBytes = data_size;
     packet.dataPtr = data;
 
-    return packet_buffer_->InsertPacket(&packet);
+    return packet_buffer_.InsertPacket(&packet);
   }
 
   void CheckFrame(uint16_t first_seq_num) {
@@ -98,7 +97,7 @@
 
   Random rand_;
   std::unique_ptr<SimulatedClock> clock_;
-  rtc::scoped_refptr<PacketBuffer> packet_buffer_;
+  PacketBuffer packet_buffer_;
   std::map<uint16_t, std::unique_ptr<RtpFrameObject>> frames_from_callback_;
 };
 
@@ -153,7 +152,7 @@
   ASSERT_TRUE(Insert(seq_num, kKeyFrame, kFirst, kNotLast));
   EXPECT_TRUE(Insert(seq_num + 2, kDeltaFrame, kFirst, kLast));
 
-  packet_buffer_->ClearTo(seq_num + 2);
+  packet_buffer_.ClearTo(seq_num + 2);
   EXPECT_TRUE(Insert(seq_num + 2, kDeltaFrame, kFirst, kLast));
   EXPECT_TRUE(Insert(seq_num + 3, kDeltaFrame, kFirst, kLast));
   ASSERT_EQ(2UL, frames_from_callback_.size());
@@ -170,21 +169,21 @@
   packet.video_header.is_last_packet_in_frame = false;
   packet.timesNacked = 0;
 
-  packet_buffer_->InsertPacket(&packet);
+  packet_buffer_.InsertPacket(&packet);
 
   packet.seqNum++;
   packet.video_header.is_first_packet_in_frame = false;
   packet.timesNacked = 1;
-  packet_buffer_->InsertPacket(&packet);
+  packet_buffer_.InsertPacket(&packet);
 
   packet.seqNum++;
   packet.timesNacked = 3;
-  packet_buffer_->InsertPacket(&packet);
+  packet_buffer_.InsertPacket(&packet);
 
   packet.seqNum++;
   packet.video_header.is_last_packet_in_frame = true;
   packet.timesNacked = 1;
-  packet_buffer_->InsertPacket(&packet);
+  packet_buffer_.InsertPacket(&packet);
 
   ASSERT_EQ(1UL, frames_from_callback_.size());
   RtpFrameObject* frame = frames_from_callback_.begin()->second.get();
@@ -210,34 +209,34 @@
 TEST_F(TestPacketBuffer, CountsUniqueFrames) {
   const uint16_t seq_num = Rand();
 
-  ASSERT_EQ(0, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(0, packet_buffer_.GetUniqueFramesSeen());
 
   EXPECT_TRUE(Insert(seq_num, kKeyFrame, kFirst, kNotLast, 0, nullptr, 100));
-  ASSERT_EQ(1, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(1, packet_buffer_.GetUniqueFramesSeen());
   // Still the same frame.
   EXPECT_TRUE(
       Insert(seq_num + 1, kKeyFrame, kNotFirst, kLast, 0, nullptr, 100));
-  ASSERT_EQ(1, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(1, packet_buffer_.GetUniqueFramesSeen());
 
   // Second frame.
   EXPECT_TRUE(
       Insert(seq_num + 2, kKeyFrame, kFirst, kNotLast, 0, nullptr, 200));
-  ASSERT_EQ(2, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(2, packet_buffer_.GetUniqueFramesSeen());
   EXPECT_TRUE(
       Insert(seq_num + 3, kKeyFrame, kNotFirst, kLast, 0, nullptr, 200));
-  ASSERT_EQ(2, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(2, packet_buffer_.GetUniqueFramesSeen());
 
   // Old packet.
   EXPECT_TRUE(
       Insert(seq_num + 1, kKeyFrame, kNotFirst, kLast, 0, nullptr, 100));
-  ASSERT_EQ(2, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(2, packet_buffer_.GetUniqueFramesSeen());
 
   // Missing middle packet.
   EXPECT_TRUE(
       Insert(seq_num + 4, kKeyFrame, kFirst, kNotLast, 0, nullptr, 300));
   EXPECT_TRUE(
       Insert(seq_num + 6, kKeyFrame, kNotFirst, kLast, 0, nullptr, 300));
-  ASSERT_EQ(3, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(3, packet_buffer_.GetUniqueFramesSeen());
 }
 
 TEST_F(TestPacketBuffer, HasHistoryOfUniqueFrames) {
@@ -250,18 +249,18 @@
     Insert(seq_num + i, kKeyFrame, kFirst, kNotLast, 0, nullptr,
            timestamp + 10 * i);
   }
-  ASSERT_EQ(kNumFrames, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(kNumFrames, packet_buffer_.GetUniqueFramesSeen());
 
   // Old packets within history should not affect number of seen unique frames.
   for (int i = kNumFrames - kRequiredHistoryLength; i < kNumFrames; ++i) {
     Insert(seq_num + i, kKeyFrame, kFirst, kNotLast, 0, nullptr,
            timestamp + 10 * i);
   }
-  ASSERT_EQ(kNumFrames, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(kNumFrames, packet_buffer_.GetUniqueFramesSeen());
 
   // Very old packets should be treated as unique.
   Insert(seq_num, kKeyFrame, kFirst, kNotLast, 0, nullptr, timestamp);
-  ASSERT_EQ(kNumFrames + 1, packet_buffer_->GetUniqueFramesSeen());
+  ASSERT_EQ(kNumFrames + 1, packet_buffer_.GetUniqueFramesSeen());
 }
 
 TEST_F(TestPacketBuffer, ExpandBuffer) {
@@ -360,7 +359,7 @@
   for (int i = 0; i < kMaxSize; ++i)
     EXPECT_TRUE(Insert(seq_num + i, kDeltaFrame, kFirst, kLast));
 
-  packet_buffer_->ClearTo(seq_num);
+  packet_buffer_.ClearTo(seq_num);
   EXPECT_TRUE(Insert(seq_num + kMaxSize, kDeltaFrame, kFirst, kLast));
 }
 
@@ -368,7 +367,7 @@
   for (int i = 0; i < kMaxSize; ++i)
     EXPECT_TRUE(Insert(i, kDeltaFrame, kFirst, kLast));
 
-  packet_buffer_->ClearTo(kMaxSize - 1);
+  packet_buffer_.ClearTo(kMaxSize - 1);
 
   for (int i = kMaxSize; i < 2 * kMaxSize; ++i)
     EXPECT_TRUE(Insert(i, kDeltaFrame, kFirst, kLast));
@@ -376,10 +375,10 @@
 
 TEST_F(TestPacketBuffer, DontClearNewerPacket) {
   EXPECT_TRUE(Insert(0, kKeyFrame, kFirst, kLast));
-  packet_buffer_->ClearTo(0);
+  packet_buffer_.ClearTo(0);
   EXPECT_TRUE(Insert(2 * kStartSize, kKeyFrame, kFirst, kLast));
   EXPECT_TRUE(Insert(3 * kStartSize + 1, kKeyFrame, kFirst, kNotLast));
-  packet_buffer_->ClearTo(2 * kStartSize);
+  packet_buffer_.ClearTo(2 * kStartSize);
   EXPECT_TRUE(Insert(3 * kStartSize + 2, kKeyFrame, kNotFirst, kLast));
 
   ASSERT_EQ(3UL, frames_from_callback_.size());
@@ -577,7 +576,7 @@
     packet.sizeBytes = data_size;
     packet.dataPtr = data;
 
-    return packet_buffer_->InsertPacket(&packet);
+    return packet_buffer_.InsertPacket(&packet);
   }
 
   const bool sps_pps_idr_is_keyframe_;
@@ -599,7 +598,7 @@
 TEST_P(TestPacketBufferH264Parameterized, DontRemoveMissingPacketOnClearTo) {
   EXPECT_TRUE(InsertH264(0, kKeyFrame, kFirst, kLast, 0));
   EXPECT_TRUE(InsertH264(2, kDeltaFrame, kFirst, kNotLast, 2));
-  packet_buffer_->ClearTo(0);
+  packet_buffer_.ClearTo(0);
   EXPECT_TRUE(InsertH264(3, kDeltaFrame, kNotFirst, kLast, 2));
 
   ASSERT_EQ(1UL, frames_from_callback_.size());
@@ -649,7 +648,7 @@
   packet.sizeBytes = sizeof(data_data);
   packet.video_header.is_first_packet_in_frame = true;
   packet.video_header.is_last_packet_in_frame = true;
-  packet_buffer_->InsertPacket(&packet);
+  packet_buffer_.InsertPacket(&packet);
 
   ASSERT_EQ(1UL, frames_from_callback_.size());
   EXPECT_EQ(frames_from_callback_[seq_num]->EncodedImage().size(),
@@ -690,7 +689,7 @@
   EXPECT_EQ(1UL, frames_from_callback_.size());
   CheckFrame(seq_num);
 
-  packet_buffer_->Clear();
+  packet_buffer_.Clear();
 
   EXPECT_TRUE(Insert(seq_num + kStartSize, kKeyFrame, kFirst, kNotLast));
   EXPECT_TRUE(
@@ -703,7 +702,7 @@
 TEST_F(TestPacketBuffer, FramesAfterClear) {
   Insert(9025, kDeltaFrame, kFirst, kLast);
   Insert(9024, kKeyFrame, kFirst, kLast);
-  packet_buffer_->ClearTo(9025);
+  packet_buffer_.ClearTo(9025);
   Insert(9057, kDeltaFrame, kFirst, kLast);
   Insert(9026, kDeltaFrame, kFirst, kLast);
 
@@ -735,7 +734,7 @@
   EXPECT_TRUE(Insert(2, kKeyFrame, kFirst, kNotLast, 5, data2));
 
   // Expect to free data3 upon insertion (old packet).
-  packet_buffer_->ClearTo(1);
+  packet_buffer_.ClearTo(1);
   EXPECT_TRUE(Insert(1, kKeyFrame, kFirst, kNotLast, 5, data3));
 
   // Expect to free data4 upon insertion (packet buffer is full).
@@ -755,15 +754,15 @@
   absl::optional<int64_t> packet_ms;
   absl::optional<int64_t> packet_keyframe_ms;
 
-  packet_ms = packet_buffer_->LastReceivedPacketMs();
-  packet_keyframe_ms = packet_buffer_->LastReceivedKeyframePacketMs();
+  packet_ms = packet_buffer_.LastReceivedPacketMs();
+  packet_keyframe_ms = packet_buffer_.LastReceivedKeyframePacketMs();
   EXPECT_FALSE(packet_ms);
   EXPECT_FALSE(packet_keyframe_ms);
 
   int64_t keyframe_ms = clock_->TimeInMilliseconds();
   EXPECT_TRUE(Insert(100, kKeyFrame, kFirst, kLast));
-  packet_ms = packet_buffer_->LastReceivedPacketMs();
-  packet_keyframe_ms = packet_buffer_->LastReceivedKeyframePacketMs();
+  packet_ms = packet_buffer_.LastReceivedPacketMs();
+  packet_keyframe_ms = packet_buffer_.LastReceivedKeyframePacketMs();
   EXPECT_TRUE(packet_ms);
   EXPECT_TRUE(packet_keyframe_ms);
   EXPECT_EQ(keyframe_ms, *packet_ms);
@@ -772,16 +771,16 @@
   clock_->AdvanceTimeMilliseconds(100);
   int64_t delta_ms = clock_->TimeInMilliseconds();
   EXPECT_TRUE(Insert(101, kDeltaFrame, kFirst, kLast));
-  packet_ms = packet_buffer_->LastReceivedPacketMs();
-  packet_keyframe_ms = packet_buffer_->LastReceivedKeyframePacketMs();
+  packet_ms = packet_buffer_.LastReceivedPacketMs();
+  packet_keyframe_ms = packet_buffer_.LastReceivedKeyframePacketMs();
   EXPECT_TRUE(packet_ms);
   EXPECT_TRUE(packet_keyframe_ms);
   EXPECT_EQ(delta_ms, *packet_ms);
   EXPECT_EQ(keyframe_ms, *packet_keyframe_ms);
 
-  packet_buffer_->Clear();
-  packet_ms = packet_buffer_->LastReceivedPacketMs();
-  packet_keyframe_ms = packet_buffer_->LastReceivedKeyframePacketMs();
+  packet_buffer_.Clear();
+  packet_ms = packet_buffer_.LastReceivedPacketMs();
+  packet_keyframe_ms = packet_buffer_.LastReceivedKeyframePacketMs();
   EXPECT_FALSE(packet_ms);
   EXPECT_FALSE(packet_keyframe_ms);
 }
@@ -798,7 +797,7 @@
   packet.timestamp = 1;
   packet.seqNum = 1;
   packet.video_header.frame_type = VideoFrameType::kVideoFrameKey;
-  EXPECT_TRUE(packet_buffer_->InsertPacket(&packet));
+  EXPECT_TRUE(packet_buffer_.InsertPacket(&packet));
 
   packet.video_header.codec = kVideoCodecH264;
   auto& h264_header =
@@ -806,7 +805,7 @@
   h264_header.nalus_length = 1;
   packet.timestamp = 3;
   packet.seqNum = 3;
-  EXPECT_TRUE(packet_buffer_->InsertPacket(&packet));
+  EXPECT_TRUE(packet_buffer_.InsertPacket(&packet));
 
   packet.video_header.codec = kVideoCodecVP8;
   packet.video_header.video_type_header.emplace<RTPVideoHeaderVP8>();
@@ -814,7 +813,7 @@
   packet.seqNum = 2;
   packet.video_header.frame_type = VideoFrameType::kVideoFrameDelta;
 
-  EXPECT_TRUE(packet_buffer_->InsertPacket(&packet));
+  EXPECT_TRUE(packet_buffer_.InsertPacket(&packet));
 
   EXPECT_EQ(3UL, frames_from_callback_.size());
 }
@@ -832,7 +831,7 @@
   h264_header.nalus_length = kMaxNalusPerPacket;
   packet.sizeBytes = 0;
   packet.dataPtr = nullptr;
-  EXPECT_TRUE(packet_buffer_->InsertPacket(&packet));
+  EXPECT_TRUE(packet_buffer_.InsertPacket(&packet));
 
   EXPECT_EQ(0UL, frames_from_callback_.size());
 }
@@ -894,7 +893,7 @@
   InsertH264(2, kDeltaFrame, kFirst, kLast, 1000);
 
   ASSERT_EQ(1UL, frames_from_callback_.size());
-  packet_buffer_->PaddingReceived(1);
+  packet_buffer_.PaddingReceived(1);
   ASSERT_EQ(2UL, frames_from_callback_.size());
   CheckFrame(0);
   CheckFrame(2);
@@ -928,7 +927,7 @@
       packet_.video_header.video_type_header.emplace<RTPVideoHeaderH264>();
   h264_header.nalus[0].type = H264::NaluType::kIdr;
   h264_header.nalus_length = 1;
-  packet_buffer_->InsertPacket(&packet_);
+  packet_buffer_.InsertPacket(&packet_);
 
   ASSERT_EQ(1u, frames_from_callback_.size());
   EXPECT_EQ(VideoFrameType::kVideoFrameKey,
@@ -943,7 +942,7 @@
   h264_header.nalus[2].type = H264::NaluType::kIdr;
   h264_header.nalus_length = 3;
 
-  packet_buffer_->InsertPacket(&packet_);
+  packet_buffer_.InsertPacket(&packet_);
 
   ASSERT_EQ(1u, frames_from_callback_.size());
   EXPECT_EQ(VideoFrameType::kVideoFrameKey,
@@ -963,7 +962,7 @@
   h264_header.nalus[0].type = H264::NaluType::kIdr;
   h264_header.nalus_length = 1;
 
-  packet_buffer_->InsertPacket(&packet_);
+  packet_buffer_.InsertPacket(&packet_);
 
   ASSERT_EQ(1u, frames_from_callback_.size());
   EXPECT_EQ(VideoFrameType::kVideoFrameDelta,
@@ -977,7 +976,7 @@
   h264_header.nalus[1].type = H264::NaluType::kPps;
   h264_header.nalus_length = 2;
 
-  packet_buffer_->InsertPacket(&packet_);
+  packet_buffer_.InsertPacket(&packet_);
 
   ASSERT_EQ(1u, frames_from_callback_.size());
   EXPECT_EQ(VideoFrameType::kVideoFrameDelta,
@@ -992,7 +991,7 @@
   h264_header.nalus[2].type = H264::NaluType::kIdr;
   h264_header.nalus_length = 3;
 
-  packet_buffer_->InsertPacket(&packet_);
+  packet_buffer_.InsertPacket(&packet_);
 
   ASSERT_EQ(1u, frames_from_callback_.size());
   EXPECT_EQ(VideoFrameType::kVideoFrameKey,