/*
	We are trying to overload firelisten to see how it reacts.
	We could have done a script, but it is too slow to really push firelisten.

	It is talking to a firelisten on port 999 on localhost. It uses the id
	"jack". You need an entry in the /etc/userfirewall/suppliers.conf
*/
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <time.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/wait.h>

static void test_random(int num1, int num2)
{
	printf ("Generating %d challenges\n",num1*num2);
	for (int i=0; i<num1; i++){
		for (int j=0; j<num2; j++){
			int fd = open ("/dev/urandom",O_RDONLY);
			if (fd == -1){
				fprintf (stderr,"Can't open /dev/urandom (%s)",strerror(errno));
			}else{
				char tmp[8];
				if (read (fd,tmp,8)!=8){
					fprintf (stderr,"Can't read 8 bytes from /dev/urandom");
				}else{
					char challenge[30];
					for (int i=0; i<8; i++){
						sprintf (challenge+i*2,"%02x",tmp[i]);
					}
					time_t ti = time(NULL);
					sprintf (challenge+16,"%lu",ti);
				}
				close (fd);
			}
		}
	}
	printf ("done\n");
}

int main (int argc, char *argv[])
{
	if (argc != 6){
		fprintf (stderr
			,"testload num1 num2 host port chunk\n"
			 "\n"
			 "Will perform a firewall login of many pseudo users\n"
			 "originating from pseudo IP number.\n"
			 "It will perform num1 x num2 login.\n"
			 "It will wait for one second between each request chunks\n"
			);
	}else{
		{
			char tmp[1000];
			sprintf (tmp,"/usr/lib/linuxconf/lib/firesendid --id jack --fhost %s --port %s reset"
				,argv[3],argv[4]);
			system (tmp);
		}
		int num1 = atoi(argv[1]);
		int num2 = atoi(argv[2]);
		int chunk = atoi(argv[5]);
		test_random(num1,num2);
		for (int i=1; i<=num1; i++){
			for (int j=1; j<=num2; j++){
				static int nb=0;
				nb++;
				if ((nb % chunk) == 0){
					fprintf (stderr,"Sleeping..");
					fflush (stderr);
					sleep(1);
					fprintf (stderr,"\n");
				}
				if (fork()==0){
					char user[100],ip[100];
					sprintf (user,"user-%03d-%03d",i,j);
					sprintf (ip,"10.1.%d.%d",i,j);
					execl ("/usr/lib/linuxconf/lib/firesendid","firesendid"
						,"--id","jack"
						,"--fhost",argv[3],"--port",argv[4]
						,"login",user,ip,NULL);
					fprintf (stderr,"Can't exec\n");
					_exit (0);
				}
			}
		}
		int status;
		while (wait(&status)!=-1);
	}
	return 0;
}