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

int skt; // Global.  This is the socket that the server will listen on
int current_client;

#define DATALEN (1024*16)

void* string_sender(void *data){
	ssize_t write_result;
	while(1){
		write_result = write(current_client, data, DATALEN);
		printf("write for %c:  %ld bytes\n", ((char*)data)[0], write_result);
	}
}

void close_server(int signal_number){
	printf("Closing down the server\n");
	close(skt);
}

int main(int argc, char ** argv){
	struct sockaddr_in sad;
	sad.sin_port = htons(5142);
	sad.sin_addr.s_addr = INADDR_ANY;
	sad.sin_family = AF_INET;

	skt = socket(AF_INET, SOCK_STREAM, 0);
	if(skt == -1){
		perror("socket");
		return 1;
	}
	struct sigaction close_action;
	close_action.sa_handler = close_server;
	if(sigaction(SIGINT, &close_action, 0)){
		perror("sigaction");
		return 1;
	}

	if( bind(skt, (struct sockaddr *)(&sad), sizeof(struct sockaddr_in)) ){
		perror("bind");
		return 1;
	}
	if( listen(skt, 5) ){
		perror("listen");
		return 1;
	}	
	int client_fd;
	struct sockaddr_in client_address;
	socklen_t address_size = sizeof(struct sockaddr_in);
	char a_buffer[DATALEN];
	char b_buffer[DATALEN];
	memset(a_buffer, 'a', DATALEN);
	memset(b_buffer, 'Z', DATALEN);

	for(;;){
		client_fd = accept(skt, (struct sockaddr *)(&client_address), &address_size);
		if(client_fd == -1){
			perror("accept");
			break;
		}
		printf("Connection made from address %s\n", inet_ntoa(client_address.sin_addr));
		pthread_t t1, t2;
		current_client = client_fd;
		pthread_create(&t1, 0, string_sender, a_buffer);
		pthread_create(&t2, 0, string_sender, b_buffer);
		pthread_join(t1, 0);
		pthread_join(t2, 0);
		close(client_fd);
	}
	return 0;
}