#include "TFTPClient.h"

int main(int argc, char** argv)
{
    /*argc = 4;
    argv[1] = "222.205.41.160";
    argv[2] = "GET";
    argv[3] = "bb.txt";*/
    /*argc = 5;
    argv[1] = "-i";
    argv[2] = "222.205.41.160";
    argv[3] = "GET";
    argv[4] = "Cantor.zip";*/
    //freopen("out.txt", "w", stdout);
    if(argc < 4)
    {
  help();
  exit(1);
    }
    parseCmd(argc, argv);//Parse the command line commands

    SOCKET sock = initWinsock();
    sockaddr_in servAddr;
    servAddr.sin_family = AF_INET;
    servAddr.sin_port = htons(SERV_PORT);
    servAddr.sin_addr.s_addr = cmd.hostaddr;
    char buf[MAX_BUF];
    int startTime = int(GetTickCount());
    fileSize = 0;

    if(cmd.put == 1)
    {
  //Begin initialize the WRQ packet
  WRQNode node;
  memset(&node, 0, sizeof(node));
  strcpy(node.fileName, cmd.fileName.c_str());
  node.opCode = 2;
  if(cmd.binFlag == 0)
      strcpy(node.mode, "netascii");
  else
  if(cmd.binFlag == 1)
      strcpy(node.mode, "octet");
  //End initialize the WRQ packet

  //Begin sending the WRQ packet
  sockaddr_in addr;
  int len = sizeof(addr);
  ACKNode ackNode;
  do
  {
      int size = makeWRQ(node, buf, sizeof(buf));
      if(sendto(sock, buf, size, 0, (sockaddr*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
      {
    cerr << "Error at sendto(): " << WSAGetLastError() << endl;
    WSACleanup();
    return 1;
      }
      memset(buf, 0, sizeof(buf));
      int ret = recvfrom(sock, buf, sizeof(buf), 0, (sockaddr*)&addr, &len);
      if(ret == SOCKET_ERROR)
      {
    cerr << "Error at recvfrom(): " << WSAGetLastError() << endl;
    WSACleanup();
    return 1;
      }
      if (getOp(buf) == ERR)
      {
    ERRNode errNode;
    parseERR(buf, errNode);
    cout << "Error: " << errNode.errMsg << endl;
    return 1;
      }
      else
    parseACK(buf, ackNode);
  }while(ackNode.opCode != 4 || ackNode.blockNumber != 0);
  //End sending the WRQ packet and received the ACK packet successfully
  
  //Begin sending the file
  if(sendFile(sock, (sockaddr*)&addr))
  {
      int endTime = int(GetTickCount());
      printInfo(cmd.fileName.c_str(), endTime - startTime, fileSize);
  }
  //End sending the file and successfully received last ACK packet

  //Close the connection
  closesocket(sock);
  WSACleanup();
  return 0;
    }
    else
    if(cmd.get == 1)
    {
  //Begin initialize the RRQNode
  RRQNode rrqNode;
  memset(&rrqNode, 0, sizeof(rrqNode));
  rrqNode.opCode = 1;
  strcpy(rrqNode.fileName, cmd.fileName.c_str());
  if(cmd.binFlag == 0)
      strcpy(rrqNode.mode, "netascii");
  else
  if(cmd.binFlag == 1)
      strcpy(rrqNode.mode, "octet");
  //End initialize the RRQNode

  //Begin send the RRQNode
  int ret = makeRRQ(rrqNode, buf, sizeof(buf));
  if(sendto(sock, buf, ret, 0, (sockaddr*)&servAddr, sizeof(servAddr)) == SOCKET_ERROR)
  {
      cout << "Error at sendto(): " << WSAGetLastError() << endl;
      WSACleanup();
      return 1;
  }
  //End send the RRQNode

  //Begin receive the file
  if(recvFile(sock))
  {
      int endTime = int(GetTickCount());
      printInfo(cmd.fileName.c_str(), endTime - startTime, fileSize);
  }
  //End receive the file

  //Close the connection
  closesocket(sock);
  WSACleanup();
  return 0;
    }
    return 0;
}