#include <stdio.h>
#include <stdlib.h>
//#include "socketfun.h"

int inout(int in, int out)
{
  char s[1000];
  int i;

  i = 0;
  while(read(in, s+i, 1) != 0) {
    if (s[i] == '\n') {
      write(out, s, i+1);
      return i+1;
    }
    i++;
  }
  return 0;
}

main(int argc, char **argv)
{
  char *hn, *un;
  int port, sock, fd;
  int i;
  char s[1000];

  if (argc != 3) {
    fprintf(stderr, "usage: serve1 hostname port\n");
    exit(1);
  }

  hn = argv[1];
  port = atoi(argv[2]);
  if (port < 5000) {
    fprintf(stderr, "usage: serve1 hostname port\n");
    fprintf(stderr, "       port must be > 5000\n");
    exit(1);
  }
  un = getenv("USER");

  sock = serve_socket(hn, port);

  fd = accept_connection(sock);

  printf("Connection established.  Sending `Server: %s'\n", un);
  sprintf(s, "Server: %s\n", un);
  write(fd, s, strlen(s));

  printf("Receiving:\n");

  inout(fd, 1);
}