/*
	Small echo service. It listen on a port and copy back what is sent
*/
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/wait.h>
#include <sys/ioctl.h>
#include <time.h>
#include <string.h>
#include <tlmplib.h>
#include <tlmpnet.h>

struct CLIENT_INFO: public ARRAY_OBJ{
	long received;
	time_t start;
	CLIENT_INFO(){
		start = time(NULL);
		received = 0;
	}
};
<mod>
int main (int argc, char *argv[])
{
	glocal bool server = false;
	glocal int count = 100000;
	glocal const char *echoserver = NULL;
	glocal bool slowserver = false;
	glocal const char *manyconnect = NULL;
	glocal int nbprocess = 1;
	glocal const char *client = NULL;
	glocal const char *bind = NULL;
	glocal const char *port = NULL;
	glocal const char *message = "ECHO";
	glocal bool pingmode = false;
	glocal bool discard = false;		// Do not analyse the output of the echo server
						// for manyconnect
	int ret = <call tlmpprogram>(argc, argv);
	<f init>
		setarg ('b',"bind","bind on this IP",glocal.bind,false);
		setarg ('p',"port","Setup the service on this port",glocal.port,true);
		setarg ('s',"server","this is the server/listener",glocal.server,false);
		setarg ('c',"client","Client mode, specify the server",glocal.client,false);
		setarg (' ',"echoserver","Client mode, specify the echo server",glocal.echoserver,false);
		setarg (' ',"slowserver","Slow server, receive and usleep",glocal.slowserver,false);
		setarg ('C',"count","Number of packets to send",glocal.count,false);
		setarg ('P',"pingmode","exchange packets between client and server",glocal.pingmode,false);
		setgrouparg ("Manyconnect");
		setarg (' ',"manyconnect","Establish many short connection to the echo server",glocal.manyconnect,false);
		setarg (' ',"processes","Number of processes (for manyconnect)",glocal.nbprocess,false);
		setarg (' ',"discard","Do not parse the response from the server",glocal.discard,false);
	</f>
	<f main_noarg>
		int ret = -1;
		if (glocal.server){
			<obj TCPSERVER s>(glocal.bind,glocal.port,1);
			<f newclient>
				settcpnodelay (true);
				info.data = new CLIENT_INFO();
			</f>
			<f endclient>
				CLIENT_INFO *n = (CLIENT_INFO*)info.data;
				int duration = time(NULL) - n->start;
				printf ("Received %ld in %d seconds\n",n->received,duration);
				if (duration > 1){
					printf ("rate=%lf bytes/s\n",(double)n->received/duration);
				}
			</f>
			<f receive>
				CLIENT_INFO *n = (CLIENT_INFO*)info.data;
				n->received += info.linelen;
				if (glocal.pingmode) send (line,info.linelen);
			</f>
			</obj>
			s.setrawmode (true);
			s.loop();
			ret = 0;
		}else if (glocal.slowserver){
			<obj TCPSERVER s>(glocal.bind,glocal.port,1);
			<f newclient>
				settcpnodelay (true);
				info.data = new CLIENT_INFO();
			</f>
			<f endclient>
				CLIENT_INFO *n = (CLIENT_INFO*)info.data;
				printf ("Received %ld\n",n->received);
			</f>
			<f receive>
				CLIENT_INFO *n = (CLIENT_INFO*)info.data;
				n->received += info.linelen;
				usleep (100000);
			</f>
			</obj>
			s.setrawmode (true);
			s.loop();
			ret = 0;
		}else if (glocal.client != NULL){
			glocal long received = 0;
			<call tcpconnect>(glocal.bind,glocal.client,glocal.port,1);
			<f init>
				settcpnodelay(true);
				setrawmode (true);
				if (glocal.pingmode){
					char buf[100];
					send (buf,sizeof(buf));
				}else{
					char buf[10000];
					for (int i=0; i<glocal.count; i++){
						send (buf,sizeof(buf));
					}
					printf ("Sent %zu\n",glocal.count*sizeof(buf));
					end = true;
				}
			</f>
			<f oneline>
				glocal.received += info.linelen;
				glocal.count--;
				if (glocal.count == 0){
					end = true;
				}else{
					char buf[100];
					send (buf,sizeof(buf));
				}
			</f>
			<f end>
			</f>
			</call>
			printf ("received %ld\n",glocal.received);
			ret = 0;
		}else if (glocal.echoserver != NULL){
			glocal const char *longstr = "1222222222222222222222222222222222222222222222222"
				"22222222222222222222222222222222222222222222222222222222222222222";
			glocal long received = 0;
			glocal bool first = true;
			long expected = glocal.count*(long)(strlen(glocal.longstr)+6);
			<call tcpconnect>(glocal.bind,glocal.echoserver,glocal.port,1);
			<f init>
				settcpnodelay(true);
				if (fork()==0){
					for (int i=0; i<glocal.count; i++){
						sendf ("%s\n",glocal.longstr);
						if ((i % 10000)==0) fprintf (stderr,"i=%d\n",i);
					}
					_exit (0);
				}
			</f>
			<f oneline>
				if (glocal.first){
					glocal.first = false;	// Skip the welcome
				}else{
					glocal.received += info.linelen;
					glocal.count--;
					if (glocal.count == 0){
						end = true;
					}
				}
			</f>
			<f end>
			</f>
			</call>
			printf ("received %ld expected %ld\n",glocal.received,expected);
			if (glocal.received != expected) fprintf (stderr,"****** mismatch ******\n");
			ret = 0;
		}else if (glocal.manyconnect != NULL){
			for (int p=0; p<glocal.nbprocess; p++){
				pid_t pid = fork();
				if (pid == (pid_t)0){
					for (int i=0; i<glocal.count; i++){
						glocal bool welcome_seen = false;
						glocal bool hello_seen = false;
						<call tcpconnect>(glocal.bind,glocal.manyconnect,glocal.port,1);
						<f init>
							settcpnodelay(true);
							send ("hello\n");
						</f>
						<f oneline>
							if (!glocal.discard){
								//printf ("Receive %d %d: %s\n",glocal.welcome_seen,glocal.hello_seen,line);
								if (strncmp(line,"Welcome",7)==0){
									glocal.welcome_seen = true;
								}else if (strncmp(line,"echo hello",10)==0){
									if (!glocal.welcome_seen) tlmp_error ("Received hello without the welcome\n");
									glocal.hello_seen = true;
									end = true;
								}else{
									tlmp_error ("Received unexpected line: %s\n",line);
									end = true;
								}
							}
						</f>
						<f end>
						</f>
						</call>
						if (!glocal.discard) printf ("%d-%d\n",p,i);
					}
					_exit (0);
				}else if (pid == (pid_t)-1){
					tlmp_error ("Can't fork (%s)\n",strerror(errno));
					exit (-1);
				}
			}
			int status;
			while (wait(&status)!=-1);
			ret = 0;
		}else{
			tlmp_error ("You must select either --server or --client or --echoserver servername mode\n");
		}
		return ret;
	</f>
	</call>
	return ret;
}
</mod>