From 1d7fc82e7c19c4617a921db260fdcd94cfccfd07 Mon Sep 17 00:00:00 2001
From: Tindy X <49061470+tindy2013@users.noreply.github.com>
Date: Fri, 25 Jun 2021 13:23:58 +0800
Subject: [PATCH] Bugs fix

Fix 'Packet #1 came too late' warning after first connection.
Remove "press 'q' to continue" after device disconnection in TCP mode.
---
 src/main.cpp | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index 57940d9..8b5b555 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -466,6 +466,7 @@ int main(int argc, char* argv[])
         while(_getwch() != L'q');
         //std::cout << "Exiting gracefully..." << std::endl;
         printErr("[INFO] Exiting gracefully...\n");
+        last_input_packet_id = 0;
         EXIT_FLAG = true;
         LEDThread.join();
         InputThread.join();
@@ -497,13 +498,14 @@ int main(int argc, char* argv[])
             EXIT_FLAG = false;
             auto LEDThread = std::thread(TCPLEDBroadcast, acc_socket, memory);
             auto InputThread = std::thread(InputReceive, acc_socket, memory);
-            while(_getwch() != L'q');
-            //std::cout << "Exiting gracefully..." << std::endl;
-            printErr("[INFO] Exiting gracefully...\n");
-            EXIT_FLAG = true;
-            CONNECTED = false;
+            //while(_getwch() != L'q');
             LEDThread.join();
             InputThread.join();
+            //std::cout << "Exiting gracefully..." << std::endl;
+            printErr("[INFO] Exiting gracefully...\n");
+            last_input_packet_id = 0;
+            EXIT_FLAG = true;
+            CONNECTED = false;
         }
     }
     return 0;