/*
Compile:
  gcc -Wall -o saltrowrrdreg saltrowrrdreg.c

Run:
./saltrowrrdreg <options>

Options:
--bcast enable broadcast (write only) default off
--fec <fec> DTC link front end card" (default 0)
--chip <chip> 0-7 (default 0)
--channel <channel> 0-15 (default 0)
--expected <value to expect> (if verify read and no write)
--register <register in hex>
--write <value in hex> (default off)
--read read specified register (default off)
--loop <number of loops, 0 = infinity> (default 1)
--verify verify written value (default off, read is forced)

*/
#include <getopt.h>
#include <signal.h>
#include <stdio.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>

#define FEC 0
#define CHIP 0
#define CHANNEL 0

//#define LOCAL_IP "130.235.185.193"
//#define SRU_IP "130.235.185.193"
#define LOCAL_IP "192.168.4.11"
#define SRU_IP "192.168.4.4"

#define SRU_WAIT_WRITE 500


unsigned int  valuerd,valuewr;
int fec;
int loop;
int chip;
int channel;
int reg;
int verify;
int pattern;
int signal_stop = 0;
int dowrite;
int doread;
int expected;
int randomflag = 0;
int broadcast = 0;

typedef struct {
  char controlip[64];
  int controlport;
  struct sockaddr_in si_sru;
  int scsocket; // socket for slow control
  int listenport;
  char readoutip[64];
  unsigned int readoutipnb;
  int status; // status from configfile
  int linkup; // it it answered ping
  int active; // status & linkup is set and FMVER could be read
} SRUSERVER;


typedef struct {
  struct sockaddr_in si_controlme;
  int controlsocket; // socket for control
  int listenport; // port for listen on control
  int nbrdpackets; // Number of packets expected back
  SRUSERVER sruserver;
} SRU;

/* SRU part */
int scSruConnect(SRU *srus)
{
  int s;
  struct timeval timeout;
  timeout.tv_sec = 0;
  timeout.tv_usec = 500000;

  if ((s=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP))==-1) {
    printf("%s: socket error %s\n",__FUNCTION__,strerror(errno));
    return -1;
  }
  setsockopt(s, SOL_SOCKET, SO_RCVTIMEO,&timeout,sizeof(timeout));
  memset((char *) &srus->si_controlme, 0, sizeof(struct sockaddr));
  srus->si_controlme.sin_family = AF_INET;
  srus->si_controlme.sin_port = htons(srus->listenport);
  if (inet_aton(LOCAL_IP, &srus->si_controlme.sin_addr)==0) {
    printf("%s: inet_aton error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    return -2;
  }
  if (bind(s, (struct sockaddr *)&srus->si_controlme, sizeof(struct sockaddr)) != 0) {
    printf("%s: bind error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    return -3;
  }
  srus->controlsocket = s;

  memset((char *) &srus->sruserver.si_sru, 0, sizeof(struct sockaddr_in));
  srus->sruserver.si_sru.sin_family = AF_INET;
  srus->sruserver.si_sru.sin_port = htons(srus->sruserver.controlport);
  if (inet_aton(srus->sruserver.controlip, &srus->sruserver.si_sru.sin_addr)==0) {
    printf("%s: sru inet_aton error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    srus->controlsocket = 0;
    return -4;
  }
  return 0;
}


/* SALTRO part */
int saltroWriteRegister(int handle,SRUSERVER *srv,int broadcast, int node, int chip, int channel, int reg, unsigned int value)
{
  unsigned int buf[4];
  int s_sent,s_bytes;
  socklen_t s_length=sizeof(srv->si_sru);
  unsigned int myreg;


  if (node >= 20) {
    buf[0] = htonl(1 << (node - 20));
    buf[1] = 0;
  }
  else {
    buf[0] = 0;
    buf[1] = htonl(1 << node);
  }
  myreg = (broadcast << 18) | (node << 12) | (chip << 9) | (channel << 5) | reg;
  buf[2] = htonl(myreg | 0x40000000);
  buf[3] = htonl(value);
  s_bytes = 16;
  if ( (s_sent = sendto(handle, buf, 16, 0, (struct sockaddr *)&srv->si_sru, s_length)) ==-1) {
    return -1;
  }
  if (s_sent != s_bytes) {
    return -2;
  }
  usleep(SRU_WAIT_WRITE);
  return 0;
}

int saltroReadRegister(int handle, SRUSERVER *srv, int node, int chip, int channel, int reg, unsigned int *value)
{
  unsigned int buf[20];
  int s_sent,s_bytes;
  socklen_t s_length=sizeof(srv->si_sru);
  struct sockaddr_in si_from;
  socklen_t si_fromlen=sizeof(si_from);
  unsigned int myreg,mynode;
  unsigned int myregrd;
  if (node >= 20) {
    buf[0] = htonl(1 << (node - 20));
    buf[1] = 0;
  }
  else {
    buf[0] = 0;
    buf[1] = htonl(1 << node);
 }

  myregrd = (node << 12) | (chip << 9) | (channel << 5) | reg;
  buf[2] = htonl(myregrd | 0xC0000000);
  buf[3] = 0;
  s_bytes = 16;
  if ( (s_sent = sendto(handle, buf, 16, 0, (struct sockaddr *)&srv->si_sru, s_length)) ==-1) {
    return -1;
  }
  if (s_sent != s_bytes) {
    return -2;
  }
  if ((s_bytes = recvfrom(handle, buf,sizeof(buf), 0, (struct sockaddr *) &si_from, &si_fromlen))==-1) {
    perror("-3");
    return -3;
  }
  if (s_bytes != 12) {
    return -4;
  }

  if (si_from.sin_addr.s_addr != srv->si_sru.sin_addr.s_addr) {
    printf("%s: Received illegal packet (lengh = %d) from %s: %d\n",__FUNCTION__,s_bytes, 
	   inet_ntoa(si_from.sin_addr), ntohs(si_from.sin_port));
    return -5;
  }
 
  mynode = ntohl(buf[0]);
  if (mynode != node) {
    printf("%s: Received not expected node %08X expected %08X\n",
	   __FUNCTION__,mynode,node);
    return -6;
  }
  myreg = ntohl(buf[1]);
  if (myreg != (myregrd | 0xC0000000)) {
    printf("%s: Received not expected register %08X expected %08X\n",
	   __FUNCTION__,myreg,myregrd | 0xC0000000);
    return -7;
  }
  *value = ntohl(buf[2]);

  return 0;
}


void signal_handler(int signum)
{
  switch(signum) {
  case SIGQUIT:
  case SIGTERM:
  case SIGINT:
    puts("\nSIGTERM");
    signal_stop = 1; break;
    /*
  case SIGTERM:
  case SIGQUIT: break;
    */
  }
}

int decodeParameters(int argc, char **argv)
{
  int c;
  int option_index = 0;
  char *short_options = "bC:c:de:f:hl:p:r:Rvw:";
  const struct option long_options[]=
  {
    {"bcast",0,0,'b'},
    {"chip",1,0,'C'},
    {"channel",1,0,'c'},
    {"fec",1,0,'f'},
    {"expected",1,0,'e'},
    {"help",0,0,'h'},
    {"loop",1,0,'l'},
    {"pattern",1,0,'p'},
    {"register",1,0,'r'},
    {"random",0,0,'R'},
    {"write",1,0,'w'},
    {"read",0,0,'d'},
    {"verify",0,0,'v'},
    {0, 0, 0, 0}
  };
  channel = CHANNEL;
  chip = CHIP;
  fec = FEC;
  loop = 1;
  verify = 0;
  valuewr = 0;
  expected = 0;
  dowrite = 0;
  doread = 0;
  pattern = 0;
  broadcast = 0;
  if (argc > 1)
  {
    while ((c = getopt_long(argc, argv, short_options,
                                        long_options, &option_index)) != EOF)
    {
      switch (c)
      {
      case 'b': broadcast = 1; break;
      case 'C': sscanf(optarg,"%d",&chip); break;
      case 'c': sscanf(optarg,"%d",&channel); break;
      case 'd': doread = 1; break;
      case 'e': sscanf(optarg,"%d",&expected); break;
      case 'f': sscanf(optarg,"%d",&fec); break;
      case 'h':
    printf("Options: \n \
	   --bcast enable broadcast (write only) default off\n \
	   --fec <fec> 0-39 DTC link front end card (default 0)\n \
	   --chip <chip> 0-7 (default 0)\n \
	   --channel <channel> 0-15 (default 0)\n \
	   --expected <value to expect> (only if verify read and no write)\n \
	   --read read specified register\n \
	   --register <register in hex>\n \
	   --write <value in hex> (of no write then read only) \n \
	   --loop <number of loops, 0 = infinity> (default 1)\n \
	   --verify verify written value (default off, read is forced)\n \
	 ");
	return 1;
	break;
      case 'l': sscanf(optarg,"%d",&loop); break;
      case 'p': sscanf(optarg,"%d",&pattern); break;
      case 'r': sscanf(optarg,"%X",&reg); break;
      case 'R': randomflag = 1; break;
      case 'v': doread = 1; verify = 1; break;
      case 'w': sscanf(optarg,"%X",&valuewr); dowrite = 1; break;
      default: printf("Unkown paramater\n"); return 1;
      }
    }
  }
  else {
    printf("Options: \n \
	   --bcast enable broadcast (write only) default off\n \
	   --fec <fec> 0-39 DTC link front end card (default 0)\n \
	   --chip <chip> 0-7 (default 0)\n \
	   --channel <channel> 0-15 (default 0)\n \
	   --expected <value to expect> (only if verify read and no write)\n \
	   --read read specified register \n \
	   --register <register in hex>\n \
	   --write <value in hex>\n \
	   --loop <number of loops, 0 = infinity> (default 1)\n \
	   --verify verify written value (default off, read is forced)\n \
	 ");
    return 1;
  }
  return 0;
}


int main(int argc, char **argv){

  SRU srus;
  int cont,ret1,ret2;
  
  int errst;

    signal(SIGINT,signal_handler);
    signal(SIGQUIT,signal_handler);
    signal(SIGTERM,signal_handler);

    strcpy(srus.sruserver.controlip,SRU_IP);
    srus.sruserver.controlport = 4097;
    srus.listenport = 4100;

    if (decodeParameters(argc,argv)) return 0;
    printf("Parameters: FEC= %d CHIP %d CHANNEL %d REG %04X BCAST %d LOOP %d VER %d READ %d WRITE(%d) %X EXP %X\n",
	   fec,chip,channel,reg,broadcast,loop,verify,doread,dowrite,valuewr,expected);

    errst = scSruConnect(&srus);
    if (errst != 0) return(0);
    
    cont = 1;
    while(cont) {
      if (dowrite != 0) {
	if (randomflag != 0) {
	  //	  valuewr = rand() & 0x3FF;
	  valuewr = loop & 0x3FF;
	  //	  printf("VALUE %X\n",valuewr);
	}
	ret1 = saltroWriteRegister(srus.controlsocket,&srus.sruserver,broadcast,fec,chip,channel,reg,valuewr);
	if ((reg != 0x1C) && (reg != 0x1D)) {
	  if (doread != 0) {
	    ret2 = saltroReadRegister(srus.controlsocket,&srus.sruserver,fec,chip,channel,reg,&valuerd);
	    if (verify) {
	      if ((valuewr != valuerd) || (ret1 != 0) || (ret2 != 0)) {
		printf("ERROR %d %d RD %08X != WR %08X\n",ret1,ret2,valuerd,valuewr);
	      }
	    }
	  }
	}
      }
      else if (doread != 0) {
	ret2 = saltroReadRegister(srus.controlsocket,&srus.sruserver,fec,chip,channel,reg,&valuerd);
	if (verify && ((ret2 != 0) || (expected != valuerd))) {
	  printf("RD %08X EXPECTED %08X err %d\n",valuerd,expected,ret2);
	}
	else if (verify != 0) {
	    printf("Read value %X\n",valuerd);
	}
      }
      if ((++cont % 1000) == 0) printf("LOOP %8d\n",cont);
      if (--loop == 0) cont = 0;
      if (loop < 0) loop = 0;
      if (signal_stop) cont = 0;
    } 

    close(srus.controlsocket);
    
    puts("\nEXIT");
    return 0;
}