/*
 * CS360:
 * James S. Plank
 * th_telnet.c - a threaded telnet client.
 */

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "sockettome.h"

/* As input, this procedure takes an array of two FILE *'s, typecast to
   a void *.  Let's call this array "connection."  This procedure will read
   lines of text from from connection[0] and write them to connection[1].
   This is convenient, because it works regardless of whether the FILE *'s 
   are socket connections or stdin/stdout. */

void *process_connection(void *c)
{
  FILE **connection;
  char buf[1000];

  connection = (FILE **) c;
  while (fgets(buf, 1000, connection[0]) != NULL) {
    printf("Read: %s", buf);
    fflush(stdout);
    fputs(buf, connection[1]);
    fflush(connection[1]);
  }
  exit(0);
}

int main(int argc, char **argv)
{
  int fd;
  FILE *fin, *fout;
  FILE *stdin_to_socket[2];
  FILE *socket_to_stdout[2];
  pthread_t tid;

  if (argc != 3 || atoi(argv[2]) <= 0) {
    fprintf(stderr, "usage: th_telnet host port\n");
    exit(1);
  }

  /* Open a socket connection to a server, and convert the file 
     descriptor to two FILE *'s, one for reading and one for 
     writing. */

  fd = request_connection(argv[1], atoi(argv[2]));
  fin = fdopen(fd, "r");
  fout = fdopen(fd, "w");
  
  /* Create arrays of FILE *'s for process_connection. */

  stdin_to_socket[0] = stdin;
  stdin_to_socket[1] = fout;

  socket_to_stdout[0] = fin;
  socket_to_stdout[1] = stdout;

  /* Fork off a thread to read from the socket and print to standard output.
     The main thread will read from standard input and print to the socket. */

  if (pthread_create(&tid, NULL, process_connection, socket_to_stdout) != 0) {
    perror("pthread_create");
    exit(1);
  }

  (void) process_connection(stdin_to_socket);
  return 0;
}