/*
	Utility to test the performance of the solution
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <ctype.h>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <tlmplib.h>
#include <tlmpnet.h>

static long long getnow()
{
	struct timeval tv;
	gettimeofday (&tv,NULL);
	return tv.tv_sec*1000000L+tv.tv_usec;
}

#define _TLMP_dofork

struct _F_dofork{
	#define _F_dofork_process(x) void x process(int noproc, int &ret)
	virtual _F_dofork_process( )=0;
};

static int dofork (_F_dofork &c, int nbfork)
{
	int ret = 0;
	if (nbfork == 1){
		c.process (0,ret);
	}else{
		for (int i=0; i<nbfork; i++){
			pid_t pid = fork();
			if (pid == (pid_t)-1){
				tlmp_error ("Can't fork (%s)\n",strerror(errno));
				exit (-1);
			}else if (pid == (pid_t)0){
				c.process (i,ret);
				_exit (0);
			}else{
			}
		}
		int status;
		while (wait(&status)!=-1);
	}
	return ret;
}


<mod>
int main (int argc, char *argv[])
{
	glocal int nbtest = 1000;
	glocal int nbfork = 1;
	glocal const char *host = "localhost";
	glocal const char *url = "/trunk/index.php";
	glocal const char *ddosport = "10000";
	glocal bool keepconnect = false;
	glocal int ret = -1;
	glocal.ret = <call tlmpprogram>(argc,argv);
	<f init>
		setarg (' ',"host","Target host for the test",glocal.host,false);
		setarg (' ',"nbtest","Number of repetition",glocal.nbtest,false);
		setarg (' ',"nbfork","Number of sub-process to use",glocal.nbfork,false);
		setarg (' ',"url","URL used for the tests",glocal.url,false);
		setarg (' ',"ddosport","Port for ddosproof",glocal.ddosport,false);
		setarg (' ',"keepconnect","Keep connection open with ddosproof",glocal.keepconnect,false);
	</f>
	<f main>
		int ret = -1;
		if (argc != 1){
			usage();
		}else if (strcmp(argv[0],"testget")==0){
			ret = <call dofork>(glocal.nbfork);
			<f process>
				long long start = getnow();
				int i;
				for (i=0; i<glocal.nbtest; i++){
					glocal bool work = false;
					<call tcpconnect>(glocal.host,"80",1);
					<f init>
						sendf ("GET %s\n\n",glocal.url);
					</f>
					<f oneline>
						//printf ("line %s\n",line);
						if (strstr(line,"var work =")!=NULL) glocal.work = true;
					</f>
					<f end>
					</f>
					</call>
					if (!glocal.work){
						tlmp_error ("No work retrieved at iteration %d, exiting\n",i);
						ret = -1;
						break;
					}
				}
				long long end = getnow();
				printf ("testget %lf get/s\n",i/((end-start)/1000000.0));
			</f>
			</call>
		}else if (strcmp(argv[0],"testwork")==0){
			ret = <call dofork>(glocal.nbfork);
			<f process>
				long long start = getnow();
				glocal int count = 0; 
				int i;
				for (i=0; i<glocal.nbtest; i++){
					glocal bool work = false;
					<call tcpconnect>(glocal.host,glocal.ddosport,1);
					<f init>
						send ("getwork 127.0.0.1 -\n");
					</f>
					<f oneline>
						//printf ("line %s\n",line);
						if (strncmp(line,"201 ",4)==0){
							glocal.work = true;
							glocal.count++;
							if (glocal.keepconnect){
								if (glocal.count < glocal.nbtest){
									send ("getwork 127.0.0.1 -\n");
								}else{
									end = true;
								}
							}else{
								end = true;
							}
						}else if (isdigit(line[0])){
							// some error
							end = true;
						}
					</f>
					<f end>
					</f>
					</call>
					if (!glocal.work){
						tlmp_error ("No work retrieved at iteration %d, exiting\n",i);
						ret = -1;
						break;
					}
					if (glocal.keepconnect) break;
				}
				long long end = getnow();
				printf ("testwork %lf get/s\n",glocal.count/((end-start)/1000000.0));
			</f>
			</call>
		}else{
			usage();
		}
		return ret;
	</f>
	</call>
	return glocal.ret;
}
</mod>