wsTCPServer.c

#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif

#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <iphlpapi.h>
#include <stdio.h>


#pragma comment(lib, "Ws2_32.lib")

/* Prototype */
void ShowError(char * msg)  {
  printf("%s\n", msg);
  WSACleanup();
  exit(1);
}


int main(int argc, char *argv[])  {
  WSADATA wsaData;

  char recvmsg[1024];
  int iResult, PORT;
  struct addrinfo *result = NULL, *ptr = NULL, ServerAddr;


/* Passing argument */
  if(argc == 1) {
    PORT = argv[1];
  }
  else  {
    printf("Usage: %s [PORT]", argv[0]);
  }

  /* Passing data to address struct */
  ZeroMemory(&ServerAddr, sizeof(ServerAddr));
  ServerAddr.ai_family = AF_INET;
  ServerAddr.ai_socktype = SOCK_STREAM;
  ServerAddr.ai_protocol = IPPROTO_TCP;
  ServerAddr.ai_flags = AI_PASSIVE;

  /* Get address information */
  if((iResult getaddrinfo(NULL, PORT, &ServerAddr, &result)) != 0)  {
    ShowError("getaddrinfo() failed:");
  }

  /* Initialize Winsock */
  if((iResult = WSAStartup(MAKEWORD(2,2), &wsaData)) != 0)  {
    ShowError("WSAStartup failed!");
  }

  /* Create SOCKET */
  SOCKET ListenSocket = INVALID_SOCKET;
  SOCKET  ClientSock = INVALID_SOCKET;

  if((ListenSocket = socket(result->ai_family, result->ai_socktype, result->ai_protocol)) == INVALID_SOCKET)  {
    ShowError("socket() failed");
  }

  /* Bind socket and port */
  if((iResult = bind(ListenSocket, result->ai_addr, (int)result->ai_addrlen)) == SOCKET_ERROR)  {
    freeaddrinfo(result);
    closesocket(ListenSocket);
    ShowError("bind() failed");
  }

  if(listen(ListenSocket, 5) == SOCKET_ERROR) {
    freeaddrinfo(result);
    closesocket(ListenSocket);
    ShowError("Listen() failed");
  }

  /* Accept a client socket */
  if((ClientSock = accept(ListenSocket, NULL, NULL)) == INVALID_SOCKET) {
    closesocket(ListenSocket);
    ShowError("accept() failed");
  }


/* Recv and send */
  do {
    iResult = recv(ClientSock, recvmsg, 1024, 0);
    if(iResult > 0) {
      printf("Handling client: %d\n", ClientSock);

      /* Echo received packet */
      if(send(ClientSock, recvmsg, iResult, 0) == SOCKET_ERROR) {
        printf("send failed %d\n", WSAGetLastError());
        closesocket(ClientSock);
        WSACleanup();
        exit(1);
      }
    }
    else if(iResult == 0) {
      printf("Conneciont closing...\n");
    }
    else {
      printf("recv() failed");
      closesocket(ClientSock);
      WSACleanup();
      exit(1);
    }
  }  while(iResult > 0);

  /* Disconnection Server */


  return 0;
}