#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <unistd.h>
#include <netdb.h>
#include <pthread.h>

#define MAXHOSTNAMELEN 128

void* client_thread(void* arg);

int open_listen(int port)
{
    int fd;
    struct sockaddr_in serverAddr;
    int optVal;

    fd = socket(AF_INET, SOCK_STREAM, 0);
    if(fd < 0) {
	perror("Error creating socket");
	exit(1);
    }

    optVal = 1;
    if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &optVal, sizeof(int)) < 0) {
	perror("Error setting socket option");
	exit(1);
    }

    serverAddr.sin_family = AF_INET;	
    serverAddr.sin_port = htons(port);
    serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);

    if(bind(fd, (struct sockaddr*)&serverAddr, sizeof(struct sockaddr_in)) < 0){
	perror("Error binding socket to port");
	exit(1);
    }

    if(listen(fd, SOMAXCONN) < 0) {
	perror("Error listening");
	exit(1);
    }

    return fd;
}

char* capitalise(char* buffer, int len)
{
    int i;

    for(i=0; i<len; i++) {
	buffer[i] = (char)toupper((int)buffer[i]);
    }
    return buffer;
}

void process_connections(int fdServer)
{
    int fd;
    struct sockaddr_in fromAddr;
    socklen_t fromAddrSize;
    int error;
    char hostname[MAXHOSTNAMELEN];
    pthread_t threadId;

    while(1) {
	fromAddrSize = sizeof(struct sockaddr_in);
	
	/* Block, wait for a connection request */
	fd = accept(fdServer, (struct sockaddr*)&fromAddr,  &fromAddrSize);
	if(fd < 0) {
	    perror("Error accepting connection");
	    exit(1);
	}

	/* Determine hostname and print it out */
	error = getnameinfo((struct sockaddr*)&fromAddr, fromAddrSize, hostname,
		MAXHOSTNAMELEN, NULL, 0, 0);
	if(error) {
	    fprintf(stderr, "Error getting hostname: %s\n", 
		    gai_strerror(error));
	} else {
	    printf("Accepted connection from %s (%s), port %d\n", 
		    inet_ntoa(fromAddr.sin_addr),
		    hostname,
		    ntohs(fromAddr.sin_port));
	}

	/* Start a thread to handle communication over this connection. */
	pthread_create(&threadId, NULL, client_thread, (void*)fd);
	pthread_detach(threadId);
    }
}

/* Thread - process text coming in over a connection */
void* client_thread(void* arg)
{
    int fd;
    char buffer[1024];
    ssize_t numBytesRead;

    /* Assume integer fits in something of size void* */
    fd = (int)arg;

    /* Repeatedly read characters from the connection, convert to 
    ** upper case and send them back 
    */
    while((numBytesRead = read(fd, buffer, 1024)) > 0) {
	capitalise(buffer, numBytesRead);
	write(fd, buffer, numBytesRead);
    }
    /* EOF or error */

    if(numBytesRead < 0) {
	perror("Error reading from socket");
	exit(1);
    }
    printf("Done\n");
    fflush(stdout);
    close(fd);
    /* pthread_exit(NULL);  - don't really need this -
     * next line does the same */
    return NULL;
}

int main(int argc, char* argv[])
{
    int portnum;
    int fdServer;

    if(argc != 2) {
	fprintf(stderr, "Usage: %s port-num\n", argv[0]);
	exit(1);
    }

    portnum = atoi(argv[1]);
    if(portnum < 1024 || portnum > 65535) {
	fprintf(stderr,  "Invalid port number: %s\n", argv[1]);
	exit(1);
    }

    fdServer = open_listen(portnum);
    process_connections(fdServer);
    return 0;
}