From 7a269e185fc59e5f81e7c91e6e75891ba2ca4b3b Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 6 Jun 2021 23:43:52 +0800
Subject: import(life): ...

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

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

diff --git a/works/life/computer-network-experiment/client.cpp b/works/life/computer-network-experiment/client.cpp
index aeb352f..f209171 100644
--- a/works/life/computer-network-experiment/client.cpp
+++ b/works/life/computer-network-experiment/client.cpp
@@ -8,6 +8,11 @@
 #ifdef WIN32
 #include <Windows.h>
 #include <winsock.h>
+#else
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+
 #endif
 
 const auto connect_address = "127.0.0.1"; // control connect address
@@ -16,7 +21,7 @@ const u_short port = 1234;                // control connect port
 int Main() {
   int client_socket;
 
-  if ((client_socket = socket(AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET) {
+  if ((client_socket = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
     PrintErrorMessageAndExit(CRUT("Failed to create socket!\n"));
   }
 
@@ -28,7 +33,7 @@ int Main() {
   memset(&(server_address.sin_zero), 0, sizeof(server_address.sin_zero));
 
   if (connect(client_socket, (sockaddr *)&server_address, sizeof(sockaddr)) ==
-      SOCKET_ERROR) {
+      -1) {
     PrintErrorMessageAndExit(CRUT("Failed to connect!"));
   }
 
@@ -45,6 +50,6 @@ int Main() {
 
   SendOutput(CRUT("Received message:\n"));
 
-  closesocket(client_socket);
+  Close(client_socket);
   return 0;
 }
-- 
cgit v1.2.3