#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <time.h>
#include <limits.h>

#define MAXLINE 1024
#define PACKET_SIZE 1038
#define SERV_PORT 10010

/* Global Variables */
int sequence;
char string[MAXLINE], packet[PACKET_SIZE];
int sequenceNumber;
short totalMessageLength;
long timestamp;

void addSequenceNumber() {
	sequenceNumber = htonl(sequence);
	memcpy(packet + sizeof(short), &sequenceNumber, sizeof(int));
}

void deconstructPacket() {
	memcpy(&totalMessageLength, packet, sizeof(short));
	totalMessageLength = ntohs(totalMessageLength);
	memcpy(&sequenceNumber, packet + sizeof(short), sizeof(int));
	sequenceNumber = ntohl(sequenceNumber);
	memcpy(&timestamp, packet + sizeof(short) + sizeof(int), sizeof(long));
	timestamp = be64toh(timestamp);
	memcpy(&string, packet + sizeof(short) + sizeof(int) + sizeof(long), MAXLINE);
}

int main (int arc, char **argv) {
	int dataSocket, n;
	socklen_t clientLength;
	struct sockaddr_in clientSocketAddress, serverSocketAddress;

	/* Create the socket with error detection */
	if ((dataSocket = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
		perror("Problem creating the socket");
		exit(2);
	}

	/* Prepare for the socket address */
	memset(&serverSocketAddress, 0, sizeof(serverSocketAddress));
	memset(&clientSocketAddress, 0, sizeof(clientSocketAddress));
	serverSocketAddress.sin_family = AF_INET;
	serverSocketAddress.sin_addr.s_addr = INADDR_ANY;
	serverSocketAddress.sin_port = htons(SERV_PORT);
	clientLength = sizeof(clientSocketAddress);

	if (bind(dataSocket, (struct sockaddr *) &serverSocketAddress, sizeof(serverSocketAddress)) < 0) {
		perror("Problem binding the socket");
		exit(3);
	}

	printf("%s\n", "Server running... waiting for connections.");

	sequence = 0;
	/* Client handler */
	for ( ; ; ) {

		/* Recieve from a client */
		n = recvfrom(dataSocket, packet, PACKET_SIZE, MSG_WAITALL, (struct sockaddr *) &clientSocketAddress, &clientLength);
		printf("%s", "String received from and resent to the client: ");

		deconstructPacket();
		puts(string);

		/* Send back to a client */
		addSequenceNumber();
		if (n = sendto(dataSocket, packet, totalMessageLength, 0, (struct sockaddr *) &clientSocketAddress, sizeof(clientSocketAddress)) < 0) {
			perror("Problem adding the packet to the buffer");
			exit(4);
		}

		/* Increment sequence number wrapping around to 0 at (2^32 - 1) */
		if (sequence < UINT_MAX) {
			sequence++;
		}
		else {
			sequence = 0;
		}
	}
	close(dataSocket);
}