#include <stdlib.h>

#include "constants.h"
#include "conn_table.h"

extern double lasttime;
extern int CHECKID;

FILE* rfile = 0;
int astart=0, aend=MAX_TIME, aint=0, debug = FALSE;
unsigned int dests[MAX_DESTS];
int nd=0;
unsigned int my_source = 0;
unsigned int my_dest = 0;
unsigned int my_smask;
unsigned int my_dmask;
int batch = FALSE;
char Host[17];
int mask;
int format = HUMAN;
double offset = 0;
double adjust = 0;

void usage(char* prg)
{
  printf("Usage: %s <source trace file>  [-o <output file>] [-v <victim trace file>] [-f offset]"
	 "[-s <sourceIP> <source mask> | -d <dest IP> <dest mask>]"  
	 "[-a <attack start> <attack end>] [-m <monitoring interval>]"
	 "[-b dest_file] [-f human|machine] [-c 0|1] \n", prg);
  exit(0);
}

struct appstats AppStats[MAX_APPLICATIONS];
int numapps = 0;

int main(int argc, char** argv)
{
  int i, pargc=2;
  FILE* output, *dfile;
  char* pargv[3];

  if (argc < 2)
    usage(argv[0]);

  /* Copy input file as the first argument to pcap_read */
  pargv[0] = malloc(strlen(argv[0]));
  strcpy(pargv[0], argv[0]);
  pargv[1] = malloc(strlen(argv[1]));
  strcpy(pargv[1], argv[1]);
  for(i=2; i<argc; i++)
    {
      if (!strcmp(argv[i],"-o"))
	output = fopen(argv[++i],"w");
      else if (!strcmp(argv[i],"-v"))
	{
	  int k;
	  pargv[2] = malloc(strlen(argv[++i]));
	  strcpy(pargv[2],argv[i]);
	  pargc++;
	}
      else if (!strcmp(argv[i],"-s"))
	{
	  my_source = ntohl(inet_addr(argv[++i]));
	  strcpy(Host,argv[i]);
	  my_smask = (~0) << atoi(argv[++i]);	
	  mask = atoi(argv[i]);	
	  printf("Source %s/%d\n", Host, mask);
	}
      else if (!strcmp(argv[i],"-d"))
	{
	  my_dest = ntohl(inet_addr(argv[++i]));
	  strcpy(Host,argv[i]);
	  my_dmask = (~0) << atoi(argv[++i]);	
	  mask = atoi(argv[i]);	
	  printf("Dest %s/%d\n", Host, mask);
	}
      else if (!strcmp(argv[i],"-a"))
	{
	  astart = atoi(argv[++i]);
	  aend = atoi(argv[++i]);
	  printf("Attack start %d end %d\n", astart, aend);
	}
      else if (!strcmp(argv[i],"-m"))
	{
	  aint = atoi(argv[++i]);
	}
      else if (!strcmp(argv[i],"-f"))
      {
	offset = atoi(argv[++i])/1000.0;
      }
      else if (!strcmp(argv[i],"-f"))
	{
	  i++;
	  if (!strcmp(argv[i],"human"))
	    format = HUMAN;
	  else if (!strcmp(argv[i],"machine"))
	    format = MACHINE;
	}
      else if (!strcmp(argv[i],"-c"))
	{
	  CHECKID = atoi(argv[++i]);
	}
      /* Add for batch processing */
      else if (!strcmp(argv[i],"-b"))
	{
	  batch = TRUE;
	  dfile = fopen(argv[++i],"r");
	}
    }
  rfile = fopen("details.txt", "w");
  /* Jelena: Do consistency checks here */
  if ((my_dest == 0 && my_source == 0 && !batch) ||
      (my_dest != 0 && my_source != 0))
    {
      printf("You must specify source or destination address but not both or neither!\n");
      exit(1);
    }
  /* Now initialize connection and packet table */
  create_conntable();
  create_packtable();
  /* Check if we are in batch processing or not.
     If we are then read destinations from a file and
     calculate performance for each destination */
  if (batch)
    {
      nd = 0;
      while(TRUE)
	{
	  fscanf(dfile, "%s", Host);
	  if (feof(dfile))
	    break;
	  dests[nd++] =  ntohl(inet_addr(Host));
	  my_dest = ntohl(inet_addr(Host));
	}
      printf("Read %d dests\n", nd);
    }
  if (TRUE)
    {
      read_pcap(pargc, pargv);
      calculatePft(astart, aend, lasttime);
      printTransactions(output);
      if (aint)
	printTimedStats();
    }
  return 0;
}