#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <netdb.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <iostream.h>

#define MAX_MSG 1000
#define OUTPUT_FILE "congestC.txt"

int main (int argc, char** argv)
{
  if (argc != 5)
    {
      cout << "congestC <socktype> <max packets> <server ip address> <port>\n";
      cout << "\twhere socktype: \n\t\t1 -> SOCK_STREAM\n\t\t2 -> SOCK_CLUSTER\n";
      return 1;
    }
  
  int socktype = atoi(argv[1]);
  if ((socktype <= 0) || (socktype > 2))
    {
      cout << "Socktype must be 1 for SOCK_STREAM or 2 for SOCK_CLUSTER\n";
      return 1;
    }
  
  int sockDeclVal = 0;
  switch(socktype)
    {
    case 1:
      sockDeclVal = SOCK_STREAM;
      break;
    case 2:
      sockDeclVal = SOCK_CLUSTER;
      break;
    }
  
  FILE* OutputFile = fopen(OUTPUT_FILE, "w+");
  if (!OutputFile)
    {
      cout << "Failed to write to file" <<  OUTPUT_FILE << "\n";
      return 1;
    }
  char* machineName = argv[3];
  int port = atoi(argv[4]);
  int max = atoi(argv[2]);
  
  struct hostent* host = gethostbyname(machineName);
  
  struct sockaddr_in sockStruct;
  sockStruct.sin_family =  AF_INET;
  sockStruct.sin_port = port;
  sockStruct.sin_addr.s_addr = inet_addr(machineName);
  
  int desc = socket(PF_INET, sockDeclVal, 0); //0->ip
  if (desc == -1)
    {
      cout << "Error setting protocol\n";
      return 1;
    }
  
  
  if ( connect(desc, (const sockaddr*)&sockStruct, sizeof(sockStruct)) == -1)
    {
      cout << "Failed to connect, " << strerror(errno) << "\n";
      return 1;
    }
  
  
  char* sentMsg = new char[MAX_MSG+1];
  char msgSizeStr[5];
  int msgSize = MAX_MSG;
  cout << "waiting, press enter to continue\n";
  cout.flush();
  cin.getline(sentMsg, 4); 
 
  for(int i=0; i < max; i++)
    {
      int got = recv(desc, (void*)sentMsg, 1,  MSG_WAITALL );
      if (got != 1)
	{
	  cout << "Only received " << got << "  chars\n";
	 break;
	  // return 1;
	}
      sentMsg[2] = 0;
      fprintf(OutputFile, "%s", sentMsg);
      fflush(OutputFile);
    }
 
  delete sentMsg;
  fclose(OutputFile);
  close(desc);
  
  return 0;
};