#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <arpa/inet.h>
#include <netinet/ip.h>
#include <netinet/tcp.h>
#include <linux/if.h>
#include <linux/if_packet.h>
#include <linux/if_ether.h>
#include <sys/wait.h>
#include <sys/time.h>

void printError(char msg[128]);
void transmit(int port, int protocol, int requests, int buffersize);

struct sockaddr_in remote_addr;
struct ifreq ifreq;
socklen_t s_size;
FILE* logfile;

int main(int argc, char* argv[]) {
	pid_t pid;
	int i, port, clients, requests, buffersize, protocol;

	if (argc < 7) printError("usage: testcli <ip> <port> <protocol> <clients> <requests> <bytes>\n");

	logfile = fopen("log.txt","w");

	if (inet_addr(argv[1]) == -1) printError("invalid ip address\n");
	if ((port = atoi(argv[2])) < 1) printError("port < 1\n");
	if ((clients = atoi(argv[4])) < 1) printError("clients < 1\n");
	if ((requests = atoi(argv[5])) < 1) printError("requests < 1\n");
	if ((buffersize = atoi(argv[6])) < 1 || buffersize > 8192) printError("bytes parameter must be between 1 and 8192\n");
	if (!strcmp(argv[3],"UDP")) protocol = SOCK_DGRAM;
	else if (!strcmp(argv[3],"TCP")) protocol = SOCK_STREAM;
	else printError("invalid protocol specified");

	remote_addr.sin_family = AF_INET;
	remote_addr.sin_port = htons(port);
	inet_pton(AF_INET, argv[1], &remote_addr.sin_addr.s_addr);

	strcpy (ifreq.ifr_name, "eth0");
	s_size = sizeof(remote_addr);

	if (getpid() == getppid()) {
		fprintf(logfile,"LOG: %i bytes\n", buffersize);
		fprintf(logfile,"LOG: %i clients\n", clients);
		fprintf(logfile,"-------------------\n");
	}

	for (i = 0; i < clients; i++) {
		pid = fork();
		switch(pid) {
			case -1: printError("Fork error\n");
			case 0: {
				transmit(port, protocol, requests, buffersize);
				break;
			}
		}
	}
	for (i = 0; i < clients; i++) wait(0);

	fclose(logfile);
	return 0;
}

void transmit(int port, int protocol, int requests, int buffersize) {
	int i, sock, msec;
	char buffer[buffersize], rbuffer[buffersize];
	
	for (i = 0; i < buffersize; i++) { buffer[i] = (char)(rand() % 92 + 70); }

	struct timeval start, finish;
  	gettimeofday (&start, NULL);
	switch (protocol) {
		case SOCK_STREAM:{
			if ((sock = socket(AF_INET, protocol, 0)) == -1) printError("Can't create socket\n");
			if (connect(sock, (struct sockaddr *)&remote_addr, s_size) == -1) printError("Can't connect\n");
			printf("Forked TCP process (pid: %i) connected\n", getpid());
			for (i = 0; i < requests; i++) {
				if (send(sock, buffer, buffersize, 0) == -1) printf("Forked process, Send error (pid: %i)\n", getpid());
				if (recv(sock, rbuffer, buffersize, 0) == -1) printf("Forked process, Recv error (pid: %i)\n", getpid());
				else printf("Forked process %i sent:	%s\n		received:	%s\n", getpid(), buffer, rbuffer);
			}
			close(sock);
			gettimeofday (&finish, NULL);
			msec = finish.tv_sec * 1000 + finish.tv_usec / 1000;
	  		msec -= start.tv_sec * 1000 + start.tv_usec / 1000;
			printf("Forked process (pid: %i) killed\n", getpid());
			fprintf(logfile," - Duration: %5.3fs\n", (double)msec/1000);
			exit(0);
			break;
		}
		case SOCK_DGRAM:{
			if ((sock = socket(AF_INET, protocol, 0)) == -1) printError("Can't create socket\n");
			printf("Forked UDP process (pid: %i) created\n", getpid());
			for (i = 0; i < requests; i++) {
				if (sendto(sock, buffer, buffersize, 0, (struct sockaddr*)&remote_addr, s_size) == -1) printf("Forked process, Sendto error (pid: %i)\n", getpid());
				if (recvfrom(sock, rbuffer, buffersize, 0, (struct sockaddr*)&remote_addr, &s_size) == -1) printf("Forked process, Recvfrom error (pid: %i)\n", getpid());
				else printf("Forked process %i sent:	%s\n		received:	%s\n", getpid(), buffer, rbuffer);
			}
			close(sock);
			gettimeofday (&finish, NULL);
			msec = finish.tv_sec * 1000 + finish.tv_usec / 1000;
	  		msec -= start.tv_sec * 1000 + start.tv_usec / 1000;
			printf("Forked process (pid: %i) killed\n", getpid());
			fprintf(logfile," - Duration: %5.3fs\n", (double)msec/1000);
			exit(0);
			break;
		}
	}
}
void printError(char msg[128]) {
	printf(msg);
	exit(1);
}