From 0cd800307043a8dc580e499f57f14f3887f185f1 Mon Sep 17 00:00:00 2001
From: Tony Yang <t510599@gmail.com>
Date: Tue, 28 Dec 2021 00:08:48 +0800
Subject: [PATCH] [Fix] should close socket descriptor before exit

---
 109062273_cli.cpp | 3 +++
 109062273_ser.cpp | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/109062273_cli.cpp b/109062273_cli.cpp
index 7b52342..5af596a 100644
--- a/109062273_cli.cpp
+++ b/109062273_cli.cpp
@@ -3,6 +3,7 @@
 #include <string>
 #include <cstring>
 #include <cstdlib>
+#include <unistd.h>
 #include <sys/socket.h>
 #include <sys/types.h>
 #include <netinet/in.h>
@@ -45,6 +46,7 @@ int main(int argc, char const *argv[]) {
 
     if (connect(sockfd, (struct sockaddr *) &info, sizeof(info)) == -1) {
         cerr << "Connection failed." << endl;
+        close(sockfd);
         return CONN_FAILED_RET;
     }
 
@@ -106,5 +108,6 @@ int main(int argc, char const *argv[]) {
         cout << endl;
     }
 
+    close(sockfd);
     return 0;
 }
\ No newline at end of file
diff --git a/109062273_ser.cpp b/109062273_ser.cpp
index 0d6bb17..4745d4e 100644
--- a/109062273_ser.cpp
+++ b/109062273_ser.cpp
@@ -4,6 +4,7 @@
 #include <cstring>
 #include <cstdlib>
 #include <ctime>
+#include <unistd.h>
 #include <sys/socket.h>
 #include <sys/types.h>
 #include <netinet/in.h>
@@ -97,6 +98,7 @@ int main(int argc, char const *argv[]) {
     // bind
     if (bind(sockfd, (struct sockaddr *) &server_info, sizeof(server_info)) == -1) {
         cerr << "Cannot bind socker." << endl;
+        close(sockfd);
         return BINDERR_RET;
     }
 
@@ -173,5 +175,6 @@ int main(int argc, char const *argv[]) {
         }
     }
 
+    close(sockfd);
     return 0;
 }
\ No newline at end of file