server

#include

#include

#include

#include

#include

#include

#pragma comment$($lib$,"$Ws$2\_32.$lib"$)$

#define DEFAULT$\_$BUFLEN $512$

#define DEFAULT$\_$PORT $"27015"$

#define MAX$\_$CLIENTS $10$

std::vector ClientSockets;

std::mutex mtx;

int $\_\_$cdecl main$($void$)\{$

WSADATA wsaData;

int iResult;

SOCKET ListenSocket $=$ INVALID$\_$SOCKET;

struct addrinfo $*$result $=$ NULL, hints;

char recvbuf$[$DEFAULT$\_$BUFLEN$]$;

int recvbuflen $=$ DEFAULT$\_$BUFLEN;

fd$\_$set readfds;

int max$\_$sd$,$ sd$,$ activity;

iResult $=$ WSAStartup$($MAKEWORD$(2,2),$ &wsaData$)$;

if $($iResult $!=0)\{$

printf$("$WSAStartup failed with error: $\%$d

$",$ iResult$)$;

return $1$;

$\}$

ZeroMemory$($&hints, sizeof$($hints$))$;

hints.ai$\_$family $=$ AF$\_$INET;

hints.ai$\_$socktype $=$ SOCK$\_$STREAM;

hints.ai$\_$protocol $=$ IPPROTO$\_$TCP;

hints.ai$\_$flags $=$ AI$\_$PASSIVE;

iResult $=$ getaddrinfo$($NULL$,$ DEFAULT$\_$PORT, &hints, &result$)$;

if $($iResult $!=0)\{$

printf$("$getaddrinfo failed with error: $\%$d

$",$ iResult$)$;

WSACleanup$()$;

return $1$;

$\}$

ListenSocket $=$ socket$($result$->$ai$\_$family, result$->$ai$\_$socktype, result$->$ai$\_$protocol$)$;

if $($ListenSocket $==$ INVALID$\_$SOCKET$)\{$

printf$("$socket failed with error: $\%$ld

$",$ WSAGetLastError$())$;

freeaddrinfo$($result$)$;

WSACleanup$()$;

return $1$;

$\}$

u$\_$long mode $=1$; $//1$ to enable non$-$blocking mode

iResult $=$ ioctlsocket$($ListenSocket$,$ FIONBIO, &mode$)$;

if $($iResult $==$ SOCKET$\_$ERROR$)\{$

printf$("$ioctlsocket failed with error: $\%$d

$",$ WSAGetLastError$())$;

closesocket$($ListenSocket$)$;

WSACleanup$()$;

return $1$;

$\}$

iResult $=$ bind$($ListenSocket$,$ result$->$ai$\_$addr, $($int$)$result$->$ai$\_$addrlen$)$;

if $($iResult $==$ SOCKET$\_$ERROR$)\{$

printf$("$bind failed with error: $\%$d

$",$ WSAGetLastError$())$;

freeaddrinfo$($result$)$;

closesocket$($ListenSocket$)$;

WSACleanup$()$;

return $1$;

$\}$

freeaddrinfo$($result$)$;

iResult $=$ listen$($ListenSocket$,$ SOMAXCONN$)$;

if $($iResult $==$ SOCKET$\_$ERROR$)\{$

printf$("$listen failed with error: $\%$d

$",$ WSAGetLastError$())$;

closesocket$($ListenSocket$)$;

WSACleanup$()$;

return $1$;

$\}$

printf$("$Server is running. Waiting for connections...

$")$;

while $(1)\{$

FD$\_$ZERO$($&readfds$)$;

FD$\_$SET$($ListenSocket$,$ &readfds$)$;

max$\_$sd $=$ ListenSocket;

$//$ Lock the mutex before accessing the shared list

mtx$.$lock$()$;

for $($SOCKET &sock : ClientSockets$)\{$

sd $=$ sock;

if $($sd $>0)$ FD$\_$SET$($sd$,$ &readfds$)$;

if $($sd $>$ max$\_$sd$)$ max$\_$sd $=$ sd;

$\}$

mtx$.$unlock$()$; $//$ Unlock the mutex after accessing the shared list

activity $=$ select$($max$\_$sd $+1,$ &readfds, NULL, NULL, NULL$)$;

if $($activity $==$ SOCKET$\_$ERROR$)\{$

printf$("$select failed with error: $\%$d

$",$ WSAGetLastError$())$;

break;

$\}$

if $($FD$\_$ISSET$($ListenSocket$,$ &readfds$))\{$

SOCKET ClientSocket $=$ accept$($ListenSocket$,$ NULL, NULL$)$;

if $($ClientSocket $==$ INVALID$\_$SOCKET$)\{$

printf$("$accept failed with error: $\%$d

$",$ WSAGetLastError$())$;

continue;

$\}$

u$\_$long mode $=1$; $//1$ to enable non$-$blocking mode

iResult $=$ ioctlsocket$($ClientSocket$,$ FIONBIO, &mode$)$;

if $($iResult $==$ SOCKET$\_$ERROR$)\{$

printf$("$ioctlsocket failed with error: $\%$d

$",$ WSAGetLastError$())$;

closesocket$($ClientSocket$)$;

continue;

$\}$

printf$("$New connection accepted

$")$;

$//$ Lock the mutex before modifying the shared list

mtx$.$lock$()$;

ClientSockets.push$\_$back$($ClientSocket$)$;

mtx$.$unlock$()$; $//$ Unlock the mutex after modification

$\}$

mtx$.$lock$()$;

for $($auto it $=$ ClientSockets.begin$()$; it $!=$ ClientSockets.end$()$; $)\{$

sd $=*$it;

if $($FD$\_$ISSET$($sd$,$ &readfds$))\{$

iResult $=$ recv$($sd$,$ recvbuf, recvbuflen, $0)$;

if $($iResult $>0)\{$

recvbuf$[$iResult$]=\text{'}\backslash 0\text{'}$;

printf$("$Client $\%$d: $\%$s

$",($int$)($it $-$ ClientSockets.begin$()),$ recvbuf$)$;

send$($sd$,$ recvbuf, iResult, $0)$;

$\}$ else if $($iResult $==0)\{$

printf$("$Client $\%$d disconnected

$",($int$)($it $-$ ClientSockets.begin$()))$;

closesocket$($sd$)$;

it $=$ ClientSockets.erase$($it$)$; $//$ Erase the client from the list

continue;

$\}$ else $\{$

printf$("$recv failed with error: $\%$d

$",$ WSAGetLastError$())$;

closesocket$($sd$)$;

it $=$ ClientSockets.erase$($it$)$; $//$ Erase the client from the list

continue;

$\}$

$\}$

$++$it;

$\}$

mtx$.$unlock$()$;

$\}$

closesocket$($ListenSocket$)$;

mtx$.$lock$()$;

for $($SOCKET &sock : ClientSockets$)\{$

if $($sock $!=$ INVALID$\_$SOCKET$)\{$

closesocket$($sock$)$;

$\}$

$\}$

mtx$.$unlock$()$;

WSACleanup$()$;

return $0$;

$\}$

client

#include

#include

#include

#include

#include

#pragma comment$($lib$,"$Ws$2\_32.$lib"$)$

#define DEFAULT$\_$BUFLEN $512$

#define DEFAULT$\_$PORT $"27015"$

int $\_\_$cdecl main$($int argc, char $**$argv$)\{$

WSADATA wsaData;

SOCKET ConnectSocket $=$ INVALID$\_$SOCKET;

struct addrinfo $*$result $=$ NULL, $*$ptr $=$ NULL, hints;

char recvbuf$[$DEFAULT$\_$BUFLEN$]$;

int iResult;

int recvbuflen $=$ DEFAULT$\_$BUFLEN;

if $($argc $!=2)\{$

printf$("$usage: $\%$s server$-$name

$",$ argv$[0])$;

return $1$;

$\}$

iResult $=$ WSAStartup$($MAKEWORD$(2,2),$ &wsaData$)$;

if $($iResult $!=0)\{$

printf$("$WSAStartup failed with error: $\%$d

$",$ iResult$)$;

return $1$;

$\}$

ZeroMemory$($&hints, sizeof$($hints$))$;

hints.ai$\_$family $=$ AF$\_$UNSPEC;

hints.ai$\_$socktype $=$ SOCK$\_$STREAM;

hints.ai$\_$protocol $=$ IPPROTO$\_$TCP;

iResult $=$ getaddrinfo$($argv$[1],$ DEFAULT$\_$PORT, &hints, &result$)$;

if $($iResult $!=0)\{$

printf$("$getaddrinfo failed with error: $\%$d

$",$ iResult$)$;

WSACleanup$()$;

return $1$;

$\}$

for $($ptr $=$ result; ptr $!=$ NULL; ptr $=$ ptr$->$ai$\_$next$)\{$

ConnectSocket $=$ socket$($ptr$->$ai$\_$family, ptr$->$ai$\_$socktype, ptr$->$ai$\_$protocol$)$;

if $($ConnectSocket $==$ INVALID$\_$SOCKET$)\{$

printf$("$socket failed with error: $\%$ld

$",$ WSAGetLastError$())$;

WSACleanup$()$;

return $1$;

$\}$

u$\_$long mode $=1$; $//1$ to enable non$-$blocking mode

iResult $=$ ioctlsocket$($ConnectSocket$,$ FIONBIO, &mode$)$;

if $($iResult $==$ SOCKET$\_$ERROR$)\{$

printf$("$ioctlsocket failed with error: $\%$d

$",$ WSAGetLastError$())$;

closesocket$($ConnectSocket$)$;

WSACleanup$()$;

return