From 7f9467d7593bd3438a697fd359bbcefe3616f7f9 Mon Sep 17 00:00:00 2001
From: loki <loki@fakeemail.com>
Date: Wed, 7 Jul 2021 17:10:06 +0200
Subject: [PATCH] Add support for IDR frame request for encrypted control

---
 sunshine/stream.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/sunshine/stream.cpp b/sunshine/stream.cpp
index 90b4512c..6646a882 100644
--- a/sunshine/stream.cpp
+++ b/sunshine/stream.cpp
@@ -26,7 +26,6 @@ extern "C" {
 #include "utility.h"
 
 #define IDX_START_A 0
-#define IDX_REQUEST_IDR_FRAME 0
 #define IDX_START_B 1
 #define IDX_INVALIDATE_REF_FRAMES 2
 #define IDX_LOSS_STATS 3
@@ -34,7 +33,8 @@ extern "C" {
 #define IDX_RUMBLE_DATA 6
 #define IDX_TERMINATION 7
 #define IDX_PERIODIC_PING 8
-#define IDX_ENCRYPTED 9
+#define IDX_REQUEST_IDR_FRAME 9
+#define IDX_ENCRYPTED 10
 
 static const short packetTypes[] = {
   0x0305, // Start A
@@ -46,6 +46,7 @@ static const short packetTypes[] = {
   0x010b, // Rumble data
   0x0100, // Termination
   0x0200, // Periodic Ping
+  0x0302, // IDR frame
   0x0001, // fully encrypted
 };
 
@@ -475,6 +476,12 @@ void controlBroadcastThread(control_server_t *server) {
       << "---end stats---";
   });
 
+  server->map(packetTypes[IDX_REQUEST_IDR_FRAME], [&](session_t *session, const std::string_view &payload) {
+    BOOST_LOG(debug) << "type [IDX_REQUEST_IDR_FRAME]"sv;
+
+    session->video.idr_events->raise(true);
+  });
+
   server->map(packetTypes[IDX_INVALIDATE_REF_FRAMES], [&](session_t *session, const std::string_view &payload) {
     auto frames     = (std::int64_t *)payload.data();
     auto firstFrame = frames[0];