From 800923717d1f95cd16b270ad8f1e430f77f33f57 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 6 Jun 2021 23:43:52 +0800
Subject: ...

---
 computer-network-experiment/Common.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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

diff --git a/computer-network-experiment/Common.cpp b/computer-network-experiment/Common.cpp
index 2e01dcb..59c2f0f 100644
--- a/computer-network-experiment/Common.cpp
+++ b/computer-network-experiment/Common.cpp
@@ -5,6 +5,11 @@
 #ifdef WIN32
 #include <Windows.h>
 #include <winsock.h>
+#else
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+#include <sys/unistd.h>
 #endif
 
 [[noreturn]] void PrintErrorMessageAndExit(StringView message,
@@ -46,6 +51,14 @@ void InitWSA() {
 }
 #endif
 
+int Close(int socket) {
+#ifdef WIN32
+  return closesocket(socket);
+#else
+  return close(socket);
+#endif
+}
+
 int main() {
 #ifdef WIN32
   InitWSA();
-- 
cgit v1.2.3