1 files changed, 70 insertions, 0 deletions
diff --git a/third_party/libwebrtc/video/video_receive_stream_timeout_tracker.h b/third_party/libwebrtc/video/video_receive_stream_timeout_tracker.h
new file mode 100644
index 0000000000..c15aa70e92
--- /dev/null
+++ b/third_party/libwebrtc/video/video_receive_stream_timeout_tracker.h
@@ -0,0 +1,70 @@
+/*
+ *  Copyright (c) 2022 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.
+ */
+
+#ifndef VIDEO_VIDEO_RECEIVE_STREAM_TIMEOUT_TRACKER_H_
+#define VIDEO_VIDEO_RECEIVE_STREAM_TIMEOUT_TRACKER_H_
+
+#include <functional>
+
+#include "api/task_queue/task_queue_base.h"
+#include "api/units/time_delta.h"
+#include "api/units/timestamp.h"
+#include "rtc_base/task_utils/repeating_task.h"
+#include "system_wrappers/include/clock.h"
+
+namespace webrtc {
+
+class VideoReceiveStreamTimeoutTracker {
+ public:
+  struct Timeouts {
+    TimeDelta max_wait_for_keyframe;
+    TimeDelta max_wait_for_frame;
+  };
+
+  using TimeoutCallback = std::function<void(TimeDelta wait)>;
+  VideoReceiveStreamTimeoutTracker(Clock* clock,
+                                   TaskQueueBase* const bookkeeping_queue,
+                                   const Timeouts& timeouts,
+                                   TimeoutCallback callback);
+  ~VideoReceiveStreamTimeoutTracker();
+  VideoReceiveStreamTimeoutTracker(const VideoReceiveStreamTimeoutTracker&) =
+      delete;
+  VideoReceiveStreamTimeoutTracker& operator=(
+      const VideoReceiveStreamTimeoutTracker&) = delete;
+
+  bool Running() const;
+  void Start(bool waiting_for_keyframe);
+  void Stop();
+  void SetWaitingForKeyframe();
+  void OnEncodedFrameReleased();
+  TimeDelta TimeUntilTimeout() const;
+
+  void SetTimeouts(Timeouts timeouts);
+
+ private:
+  TimeDelta TimeoutForNextFrame() const RTC_RUN_ON(bookkeeping_queue_) {
+    return waiting_for_keyframe_ ? timeouts_.max_wait_for_keyframe
+                                 : timeouts_.max_wait_for_frame;
+  }
+  TimeDelta HandleTimeoutTask();
+
+  Clock* const clock_;
+  TaskQueueBase* const bookkeeping_queue_;
+  Timeouts timeouts_ RTC_GUARDED_BY(bookkeeping_queue_);
+  const TimeoutCallback timeout_cb_;
+  RepeatingTaskHandle timeout_task_;
+
+  Timestamp last_frame_ = Timestamp::MinusInfinity();
+  Timestamp timeout_ = Timestamp::MinusInfinity();
+  bool waiting_for_keyframe_;
+};
+}  // namespace webrtc
+
+#endif  // VIDEO_VIDEO_RECEIVE_STREAM_TIMEOUT_TRACKER_H_