#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
#include <signal.h>

int s;

void rip(int n) {
  exit(1);
}

int main(int argc, char *argv[]) {
  int i, c=0, r[10]={-1,-1,-1,-1,-1,-1,-1,-1,-1,-1}, size;
  fd_set readfds;
  struct sockaddr_in saddr, raddr;

  signal(SIGINT, rip);
  s = socket(AF_INET, SOCK_STREAM, 0);

  saddr.sin_family = AF_INET;
  saddr.sin_port = htons(31331);
  saddr.sin_addr.s_addr = INADDR_ANY;

  bind(s, (const struct sockaddr *)&saddr, sizeof(saddr));
  listen(s, 10);

  while(1) {
    FD_ZERO(&readfds);
    FD_SET(s, &readfds);
    for (i=0; i<c; i++) {
      printf("setting r[%d]\n", i);
      FD_SET(r[i], &readfds);
    }

    select((r[(!c)?0:c-1]!=-1)?r[c-1]+1:s+1, &readfds, 0, 0, 0);

    if (FD_ISSET(s, &readfds)) {
	r[c++] = accept(s, (struct sockaddr *)&raddr, (socklen_t *)&size);
	printf("got connection from %d\n", raddr.sin_addr.s_addr);
    }

    for (i=0; i<c; i++) if (FD_ISSET(r[i], &readfds)) {
	int j;
	char buffer[1024];

	memset(buffer, 0, 1024);
	printf("(r[%d] = %d) is set\n", i, r[i]);
	read(r[i], buffer, 1024);
	for (j=0; j<c; j++) if (r[i] != r[j]) write(r[j], buffer, 1024);
    }
  }

  return 0;
}