From 8dae26aa52348a9cdf792eb5c04b73d5153c250f Mon Sep 17 00:00:00 2001
From: Cameron Gutman <aicommander@gmail.com>
Date: Fri, 9 Jan 2015 00:04:00 -0500
Subject: [PATCH] Request an IDR frame after 120 dropped frames

---
 limelight-common/VideoDepacketizer.c | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/limelight-common/VideoDepacketizer.c b/limelight-common/VideoDepacketizer.c
index 0d53969..c4b6ea6 100644
--- a/limelight-common/VideoDepacketizer.c
+++ b/limelight-common/VideoDepacketizer.c
@@ -15,6 +15,9 @@ static int gotNextFrameStart;
 static int lastPacketInStream;
 static int decodingFrame;
 
+#define CONSECUTIVE_DROP_LIMIT 120
+static int consecutiveFrameDrops;
+
 static LINKED_BLOCKING_QUEUE decodeUnitQueue;
 static unsigned int nominalPacketDataLength;
 
@@ -55,6 +58,21 @@ static void cleanupAvcFrameState(void) {
 /* Cleanup AVC frame state and set that we're waiting for an IDR Frame*/
 static void dropAvcFrameState(void) {
 	waitingForIdrFrame = 1;
+    
+    // Count the number of consecutive frames dropped
+    consecutiveFrameDrops++;
+    
+    // If we reach our limit, immediately request an IDR frame and reset
+    if (consecutiveFrameDrops == CONSECUTIVE_DROP_LIMIT) {
+        Limelog("Reached consecutive drop limit");
+        
+        // Restart the count
+        consecutiveFrameDrops = 0;
+        
+        // Request an IDR frame
+        connectionDetectedFrameLoss(0, 0);
+    }
+    
 	cleanupAvcFrameState();
 }
 
@@ -161,6 +179,9 @@ static void reassembleAvcFrame(int frameNumber) {
 
 			// Notify the control connection
 			connectionReceivedFrame(frameNumber);
+            
+            // Clear frame drops
+            consecutiveFrameDrops = 0;
 		}
 	}
 }