#define WIN32_LEAN_AND_MEAN 
#include <stdio.h> 
#include <stdlib.h> 
#include <winsock2.h> 

//#define CONNECTED_UDP //if commented out, use unconnected semantic; if uncommented, use connected semantics 

#define DEFAULT_PORT 5432
#define DEFAULT_PROTO SOCK_STREAM // TCP 

void Usage(char *progname) 
{ 
  fprintf(stderr,"Usage\n%s -p [protocol] -n [server] -e [endpoint] -l [iterations]\n", progname); 
  fprintf(stderr,"Where:\n\tprotocol is one of TCP or UDP\n"); 
  fprintf(stderr,"\tserver is the IP address or name of server\n"); 
  fprintf(stderr,"\tendpoint is the port to listen on\n"); 
  fprintf(stderr,"\titerations is the number of loops to execute\n"); 
  fprintf(stderr,"\t(-l by itself makes client run in an infinite loop,"); 
  fprintf(stderr," Hit Ctrl-C to terminate it)\n"); 
  fprintf(stderr,"Defaults are TCP , localhost and 5001\n"); 
  WSACleanup(); 
  exit(1); 
} 
void DisplayError (LPTSTR title, DWORD error) 
{ 
  printf ("%s, Error# %d\n\n", title,  error); 
} 

int main(int argc, char **argv) 
{ 
  unsigned char Buffer[128]; 
  char *server_name= "localhost"; 
  unsigned short port = DEFAULT_PORT; 
  int retval, loopflag=0; 
  int i, loopcount,maxloop=-1; 
  unsigned int addr; 
  int socket_type = DEFAULT_PROTO; 
  struct sockaddr_in server; 
  struct hostent *hp; 
  WSADATA wsaData; 
  SOCKET  conn_socket; 
  int j;

  if (argc >1) 
  { 
    for(i=1;i <argc;i++) 
    { 
      if ( (argv[i][0] == '-') || (argv[i][0] == '/') ) 
      { 
        switch(tolower(argv[i][1])) 
        { 
          case 'p': 
            if (!stricmp(argv[i+1], "TCP") ) 
              socket_type = SOCK_STREAM; 
            else 
              Usage(argv[0]); 
            i++; 
            break; 

          case 'n': 
            server_name = argv[++i]; 
            break; 
          case 'e': 
            port = atoi(argv[++i]); 
            break; 
          case 'l': 
            loopflag =1; 
            if (argv[i+1]) 
            { 
              if (argv[i+1][0] != '-')  
              {
                maxloop = atoi(argv[i+1]); 
              }
            } 
            else 
            {
              maxloop = -1; 
            }
            i++; 
            break; 
          default: 
            Usage(argv[0]); 
            break; 
        } 
      } 
      else 
        Usage(argv[0]); 
    } 
  } 

  if ((retval = WSAStartup(0x202,&wsaData)) != 0) 
  { 
    fprintf(stderr,"WSAStartup failed with error %d\n",retval); 
    WSACleanup(); 
    return -1; 
  } 

  if (port == 0)
  { 
    Usage(argv[0]); 
  } 

  if (isalpha(server_name[0])) 
  {
    hp = gethostbyname(server_name); 
  } 
  else  
  {
    addr = inet_addr(server_name); 
    hp = gethostbyaddr((char *)&addr,4,AF_INET); 
  }

  if (hp == NULL ) 
  { 
    fprintf(stderr,"Client: Cannot resolve address [%s]: Error %d\n", server_name,WSAGetLastError()); 
    WSACleanup(); 
    exit(1); 
  } 

  memset(&server,0,sizeof(server)); 
  memcpy(&(server.sin_addr),hp->h_addr,hp->h_length); 
  server.sin_family = hp->h_addrtype; 
  server.sin_port = htons(port); 

  conn_socket = socket(AF_INET,socket_type,0); /* Open a socket */ 
  if (conn_socket <0 ) 
  { 
    fprintf(stderr,"Client: Error Opening socket: Error %d\n", WSAGetLastError()); 
    WSACleanup(); 
    return -1; 
  } 

  if (socket_type == SOCK_STREAM) 
  { 
    printf("Client connecting to: %s\n",hp->h_name); 
    if (connect(conn_socket,(struct sockaddr*)&server,sizeof(server)) == SOCKET_ERROR) 
    { 
      fprintf(stderr,"connect() failed: %d\n",WSAGetLastError()); 
      WSACleanup(); 
      return -1; 
    } 
  } 

  Sleep (200); //Wait a little while to allow PATH, RESV exchange 

  // cook up a string to send 
  // 
  loopcount =0; 
  while(1) 
  { 
    loopcount++;
    Buffer[0] = 102;
    Buffer[1] = 0;

    retval = send(conn_socket,(char*)Buffer,1,0); 
    if (retval == SOCKET_ERROR) 
    { 
      fprintf(stderr,"send() failed: error %d\n",WSAGetLastError()); 
      WSACleanup(); 
      return -1; 
    } 
    Sleep (20); 

    retval = recv(conn_socket,(char*)Buffer,sizeof (Buffer),0 ); 
    if (retval == SOCKET_ERROR) 
    { 
      fprintf(stderr,"recv() failed: error %d\n",WSAGetLastError()); 
      closesocket(conn_socket); 
      continue; 
    } 
    printf("Received %4d bytes, from server.  Data:",retval); 

    for(j=0;j<retval;j++)
    {
      printf("%02x ",Buffer[j]);
    }
    printf("\n");
    Sleep (25); 

    if (!loopflag)
    { 
      printf("Terminating connection\n"); 
      break; 
    } 
    else 
    { 
      if ( (loopcount >= maxloop) && (maxloop >0) ) 
      {
        break; 
      }
    } 
  } 

  closesocket(conn_socket); 
  WSACleanup(); 
  return(0);
}