/*
NAME:           TCPCleint1 
SYNOPSIS:       TCPCleint1 <hostid> <portnumber> 
DESCRIPTION:    The program creates a stream socket in the inet domain, 
                Connect to TCPServer1, 
                Get messages typed by a user and 
                Send them to TCPServer1 running on hostid
                Then it waits for a reply from  the TCPServer1
                and show it back to the user, with a message
                indicating if there is an error during the round trip. 
*/

#include "def"
#define MAXHOSTNAME 80
#define BUFSIZE 1024
void exit(int);
char buf[BUFSIZE];
char rbuf[BUFSIZE];
void GetUserInput();
  int rc, cc;
  int   sd;

main( argc, argv )
int argc;
char *argv[];
{
  int childpid;
  struct   sockaddr_in server;
  struct   sockaddr_in client;
  struct  hostent *hp, *gethostbyname();
  struct  servent *sp;
  struct sockaddr_in from;
  struct sockaddr_in addr;
  int fromlen;
  int length;
  char ThisHost[80];

  sp = getservbyname("echo", "tcp");

/** get TCPClient1 Host information, NAME and INET ADDRESS */

  gethostname(ThisHost, MAXHOSTNAME);
  /* OR strcpy(ThisHost,"localhost"); */

  printf("----TCP/Cleint running at host NAME: %s\n", ThisHost);
  if  ( (hp = gethostbyname(ThisHost)) == NULL ) {
    fprintf(stderr, "Can't find host %s\n", argv[1]);
    exit(-1);
  }
  bcopy ( hp->h_addr, &(server.sin_addr), hp->h_length);
  printf("    (TCP/Cleint INET ADDRESS is: %s )\n", inet_ntoa(server.sin_addr));

/** get TCPServer1 Host information, NAME and INET ADDRESS */

  if  ( (hp = gethostbyname(argv[1])) == NULL ) {
    addr.sin_addr.s_addr = inet_addr(argv[1]);
    if ((hp = gethostbyaddr((char *) &addr.sin_addr.s_addr,
        sizeof(addr.sin_addr.s_addr),AF_INET)) == NULL) {
      fprintf(stderr, "Can't find host %s\n", argv[1]);
      exit(-1);
    }
  }
  printf("----TCP/Server running at host NAME: %s\n", hp->h_name);
  bcopy ( hp->h_addr, &(server.sin_addr), hp->h_length);
  printf("    (TCP/Server INET ADDRESS is: %s )\n", inet_ntoa(server.sin_addr));

  /* Construct name of socket to send to. */
    server.sin_family = AF_INET; 
  /* OR server.sin_family = hp->h_addrtype; */

  server.sin_port = htons(atoi(argv[2]));
  /*OR    server.sin_port = sp->s_port; */

  /*   Create socket on which to send  and receive */

  sd = socket (AF_INET,SOCK_STREAM,0); 
  /* sd = socket (hp->h_addrtype,SOCK_STREAM,0); */
  
  if (sd<0) {
    perror("opening stream socket");
    exit(-1);
  }

/** for cs779 assignment #1  */
/*
  client.sin_port = htons(atoi(argv[2]));
  if ( bind( sd, (SA *) &client, sizeof(client) ) ) {
    close(sd);
    perror("binding name to stream socket");
    exit(-1);
  }
*/
/** Connect to TCPServer1 */
  if ( connect(sd, (SA *) &server, sizeof(server)) < 0 ) {
    close(sd);
    perror("connecting stream socket");
    exit(0);
  }
  fromlen = sizeof(from);
  if (getpeername(sd,(SA *)&from,&fromlen)<0){
    perror("could't get peername\n");
    exit(1);
  }
  printf("Connected to TCPServer1: ");
  printf("%s:%d\n", inet_ntoa(from.sin_addr),
      ntohs(from.sin_port));
  if ((hp = gethostbyaddr((char *) &from.sin_addr.s_addr,
      sizeof(from.sin_addr.s_addr),AF_INET)) == NULL)
    fprintf(stderr, "Can't find host %s\n", inet_ntoa(from.sin_addr));
  else
    printf("(Name is : %s)\n", hp->h_name);
 childpid = fork();
 if (childpid == 0) {
   GetUserInput();
 }

/** get data from USER, send it SERVER,
      receive it from SERVER, display it back to USER  */

  for(;;) {
    cleanup(rbuf);
    if( (rc=recv(sd, rbuf, sizeof(buf), 0)) < 0){
      perror("receiving stream  message");
      exit(-1);
    }
    if (rc > 0){
      rbuf[rc]=NULL;
      printf("Received: %s\n", rbuf);
    }else {
      printf("Disconnected..\n");
      close (sd);
      exit(0);
    }


  }
}
cleanup(buf)
char *buf;
{
  int i;
  for(i=0; i<BUFSIZE; i++) buf[i]=NULL;
}

void GetUserInput()
{
for(;;) {
    printf("\nType anything followed by RETURN, or type CTRL-D to exit\n");
    cleanup(buf);
    rc=read(0,buf, sizeof(buf));
    if (rc == 0) break;
    if (send(sd, buf, rc, 0) <0 )
      perror("sending stream message");
  }
  printf ("EOF... exit\n");
  close(sd);
  kill(getppid(), 9);
  exit (0);
}