From 3ec69b6d97b2486fe4fa53a348c0110aec589000 Mon Sep 17 00:00:00 2001
From: rustdesk <info@rustdesk.com>
Date: Sat, 5 Feb 2022 01:19:24 +0800
Subject: [PATCH] refactor video_ack_required

---
 libs/hbb_common/protos/message.proto | 1 +
 src/server/connection.rs             | 4 +---
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/libs/hbb_common/protos/message.proto b/libs/hbb_common/protos/message.proto
index d11c48407..16470a753 100644
--- a/libs/hbb_common/protos/message.proto
+++ b/libs/hbb_common/protos/message.proto
@@ -54,6 +54,7 @@ message LoginRequest {
     FileTransfer file_transfer = 7;
     PortForward port_forward = 8;
   }
+  bool video_ack_required = 9;
 }
 
 message ChatMessage { string text = 1; }
diff --git a/src/server/connection.rs b/src/server/connection.rs
index b03abe60a..9c1fa92b7 100644
--- a/src/server/connection.rs
+++ b/src/server/connection.rs
@@ -684,6 +684,7 @@ impl Connection {
             if let Some(o) = lr.option.as_ref() {
                 self.update_option(o).await;
             }
+            self.video_ack_required = lr.video_ack_required;
             if self.authorized {
                 return true;
             }
@@ -932,9 +933,6 @@ impl Connection {
                         }
                     }
                     Some(misc::Union::video_received(_)) => {
-                        if !self.video_ack_required {
-                            self.video_ack_required = true;
-                        }
                         video_service::notify_video_frame_feched(self.inner.id, Some(Instant::now().into()));
                     }
                     _ => {}