From da6c0e6194578538ce0bcd1b9815696b96153f6b Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Mon, 7 Jun 2021 14:05:51 +0800
Subject: import(life): ...

---
 works/life/computer-network-experiment/Common.cpp | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

(limited to 'works/life/computer-network-experiment/Common.cpp')

diff --git a/works/life/computer-network-experiment/Common.cpp b/works/life/computer-network-experiment/Common.cpp
index 59c2f0f..fbdb2c8 100644
--- a/works/life/computer-network-experiment/Common.cpp
+++ b/works/life/computer-network-experiment/Common.cpp
@@ -34,9 +34,7 @@
 #endif
   }
 
-#ifdef WIN32
-  WSACleanup();
-#endif
+  BeforeExit();
 
   std::exit(1);
 }
@@ -51,7 +49,7 @@ void InitWSA() {
 }
 #endif
 
-int Close(int socket) {
+int CloseSocket(int socket) {
 #ifdef WIN32
   return closesocket(socket);
 #else
@@ -59,16 +57,21 @@ int Close(int socket) {
 #endif
 }
 
-int main() {
+void BeforeExit() {
 #ifdef WIN32
-  InitWSA();
+  WSACleanup();
 #endif
 
-  int c = Main();
+  SignalAndWaitForOutputThreadStop();
+}
 
+int main() {
 #ifdef WIN32
-  WSACleanup();
+  InitWSA();
 #endif
 
+  int c = Main();
+
+  BeforeExit();
   return c;
 }
-- 
cgit v1.2.3