#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <errno.h>
#include <stdio.h>
#include <unistd.h>

int main(int argc, char * argv[])
{
  int sockfd, i;
  struct sockaddr_in addr;
  char buf[100];
  fd_set  rset;

  if (argc < 3) {
    printf("Usage: a.out ip_addr port.\n");
    exit(0);
  }

  if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
    perror(": Can't get socket");
    exit(1);
  }

  memset(&addr, 0, sizeof(struct sockaddr_in));
  addr.sin_family = AF_INET;
  addr.sin_port = htons((short)atoi(argv[2]));
  addr.sin_addr.s_addr = inet_addr(argv[1]);

  if (connect(sockfd, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
    perror(": connect");
    exit(1);
  }

  while (1) {
    int maxf;

    FD_ZERO(&rset);
    FD_SET(STDIN_FILENO, &rset);
    FD_SET(sockfd, &rset);    
    if (sockfd > STDIN_FILENO) maxf = sockfd+1;
    else maxf = STDIN_FILENO+ 1;

    select(maxf, &rset, NULL, NULL, NULL);
    if (FD_ISSET(sockfd, &rset)) {
      if (read(sockfd, buf, 100) == 0) {
	printf("server closed connection.\n");
	exit(0);
      }
      printf("%s\n", buf);
    }
    if (FD_ISSET(STDIN_FILENO, &rset)) {
      if (fgets(buf, 100, stdin) == NULL) exit(0);
      write(sockfd, buf, strlen(buf));
    }
  }
}