source: network-game/client/Client/main.cpp@ 5e693e8

#include "../../common/compiler.h"

#include <sys/types.h>

#if defined WINDOWS
   #include <winsock2.h>
   #include <WS2tcpip.h>
#elif defined LINUX
   #include <sys/types.h>
   #include <unistd.h>
   #include <sys/socket.h>
   #include <netinet/in.h>
   #include <netdb.h>
   #include <cstring>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string>

#include <iostream>

#include <boost/lambda/lambda.hpp>

#include "../../common/message.h"

#ifdef WINDOWS
        #pragma comment(lib, "ws2_32.lib")
#endif

using namespace std;

void initWinSock();
void shutdownWinSock();
void error(const char *);

int main(int argc, char *argv[])
{
   int sock, n;
   struct sockaddr_in server, from;
   struct hostent *hp;
   char buffer[256];
   NETWORK_MSG msgTo, msgFrom;

   if (argc != 3) {
      cout << "Usage: server port" << endl;
      exit(1);
   }

   initWinSock();
        
   sock = socket(AF_INET, SOCK_DGRAM, 0);
   if (sock < 0)
      error("socket");

   server.sin_family = AF_INET;
   hp = gethostbyname(argv[1]);
   if (hp==0)
      error("Unknown host");

   memcpy((char *)&server.sin_addr, (char *)hp->h_addr, hp->h_length);
   server.sin_port = htons(atoi(argv[2]));

   strcpy(msgTo.buffer, "Hello");
   n=sendMessage(&msgTo, sock, &server);
   if (n < 0)
      error("sendMessage");

   n = receiveMessage(&msgFrom, sock, &from);
   if (n < 0)
      error("receiveMessage");
        
   cout << msgFrom.buffer << endl;

   while(true) {
      cout << "Please enter a message (or q to quit): ";
      cin.getline(msgTo.buffer, 256);
                
      if (strcmp(msgTo.buffer, "q") == 0) {
         break;
      }

      n=sendMessage(&msgTo, sock, &server);
      if (n < 0)
         error("sendMessage");

      n = receiveMessage(&msgFrom, sock, &from);
      if (n < 0)
         error("receiveMessage");

      cout << msgFrom.buffer << endl;
   }

#if defined WINDOWS
   closesocket(sock);
#elif defined LINUX
   close(sock);
#endif

   shutdownWinSock();

   cout << "Thank you for playing!" << endl;
   getchar();

   return 0;
}

void initWinSock()
{
#if defined WINDOWS
   WORD wVersionRequested;
   WSADATA wsaData;
   int wsaerr;

   wVersionRequested = MAKEWORD(2, 2);
   wsaerr = WSAStartup(wVersionRequested, &wsaData);
        
   if (wsaerr != 0) {
      cout << "The Winsock dll not found." << endl;
      exit(1);
   }else
      cout << "The Winsock dll was found." << endl;
#endif
}

void shutdownWinSock()
{
#if defined WINDOWS
   WSACleanup();
#endif
}

// need to make a function like this that works on windows
void error(const char *msg)
{
   perror(msg);
   shutdownWinSock();
   exit(1);
}