#include <rtl.h>
#include <time.h>
#include <rtnet/rtnet.h>
#include <mbuff.h>
#include "rtpr.h"

#define SERV_PORT 7775

#define MSG_SIZE 1500

int sock;
struct sockaddr_in servaddr;
struct sockaddr fromaddr;
hrtime_t *times;
char buffer[MSG_SIZE+1];
int pCount = 0;
int tpCount = 0;
int *c0, *c1;

char *buff;

/* the contents of shared memory may change at any time, thus volatile */
volatile char *shmData, *shmControl;        

int echo_rcv(int s,void *arg)
{
        int ret = 0;
        int fromlen = 0;
        buff = buffer;

        times[pCount] = gethrtime(); 
        c0[0] = pCount;
        pCount++;
        if (pCount == MAX_COUNT) pCount = 0;
        tpCount ++;
        if ( (c1[0] == pCount) && (c1[0] != INIT_MSG) ) {
           rtl_printf("Warning: BUFFER OVERFLOW - writing process too slow\n");
        }
   
        ret = rt_recvfrom(sock, buff, MSG_SIZE,0 ,&fromaddr, &fromlen);   
//rtl_printf("%i %i %x %x %x %x \n",ret, fromlen,fromaddr.sa_data[1],fromaddr.sa_data[2],fromaddr.sa_data[3], fromaddr.sa_data[4]);

	return 0;
}

/* Module initialization - INSMOD */

int init_module(void) {
	int ret;

        // shared memory allocation

        shmData = (volatile char*) mbuff_alloc("recData",MAX_COUNT*sizeof(hrtime_t));   
        if( shmData == NULL ) {
                rtl_printf("mbuff_alloc failed\n");
                return -1;
        }         

        shmControl = (volatile char*) mbuff_alloc("recControl",1024);   
        if( shmControl == NULL) {
                rtl_printf("mbuff_alloc failed\n");
                mbuff_free("recData",(void*)shmData);   
                return -1;
        }          
        // received pointer
        c0 = (int*)&shmControl[0];
        c0[0] = INIT_MSG;

        // write to file pointer
        c1 = (int*)&shmControl[8];
        c1[0] = 0;
        times = (hrtime_t*)shmData;

        // message contents setup - fill with 'A'
        for(ret = 0; ret<MSG_SIZE; ret++) {
           buffer[ret] = 65;
        }
        buffer[ret] = 0;


	// socket setup for sending
	
	sock=rt_socket(AF_INET,SOCK_DGRAM,0);
rtl_printf("sock: %i\n",sock);

	memset(&servaddr,0,sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(SERV_PORT);
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);

	ret=rt_bind(sock,(struct sockaddr *)&servaddr,sizeof(servaddr));

        if (ret<0) {
           rtl_printf("Error in rt_bind\n");
           rt_close(sock);
           return -1;
        }

	// set up receiving

	rt_socket_callback(sock,echo_rcv,NULL);

	//        return pthread_create(&thread, NULL, routine, 0);
        return 0;
}




/* Module clean-up - RMMOD */

void cleanup_module(void) {
   //int i;
   //   int ret;
   // int dif1, dif2;

   // print results - statistics on accuracy
   //for (i=0; i<pCount; i++) {
   //   rtl_printf("%x\n",shmData[i]);
   //}
   rtl_printf("rtpr: %i packets received\n",tpCount);

   // clean-up

   // close sockets
   rt_close(sock);

   // shared memory
   c0[0] = FINISH_MSG;
   mbuff_free("recData",(void*)shmData);       
   mbuff_free("recControl",(void*)shmControl);       

}