/* PL155-P controller via ethernet using Spartan 3E board    */
/* Copyright I.Tsurin Liverpool University V1.0 (30.11.2011) */
/*                                                           */
/* Acknowledgment: Paulo Pedreiras, Out.2005 RCAI 05/06      */
/* http://www.ieeta.pt/~pedreiras/resources/rcai/raw-sr.c    */

/* Includes */
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <math.h>
#include <string.h>
#include <sys/time.h>
#include <ctype.h>

#include <signal.h>
#include <sys/ioctl.h>

#include <sys/socket.h>
#include <netinet/in.h>

#include <linux/if.h>
#include <netpacket/packet.h>
#include <net/ethernet.h>  // the L2 protocols
#include <asm/types.h>


#define SAFE_LEN 402 // for 40ms timeout
#define DEVICE "eth0"

int s = 0;		  // Socket descriptor
unsigned char *eth_head;  // byte pointer to ethenet header
unsigned char *eth_data;  // byte pointer to ethernet data field
struct ethhdr *eh;	  // structured pointer to ethernet header

struct sockaddr_ll socket_address;
void *buffer = NULL;

/* Local host NIC MAC address */
unsigned char   src_mac[6] = {0x00, 0x1D, 0x09, 0xB1, 0x46, 0x6F};

/* Front end address template */
unsigned char dest_mac[6] = {0x02, 0x00, 0x00, 0x00, 0x00, 0x00};

void sigint(int signum);
int eth_raw_init();

unsigned char l=1;
unsigned char O=0;

unsigned ePackID;
unsigned eCommID;


/* Board control */

#define RPlength 652

#define MCA   0
#define LU    2

#define MC    l
#define MB    O

#define Reset 0
#define Write 1
#define Read  2


/***************************************/
/*       Ethernet packet sender        */
/***************************************/

void PING(int Device, int Address, int Command, int Dsize)
{
  int i;

  ePackID = (rand()&0x7fff) | ((rand()&0x7fff)<<15);

/* Re-build ethernet header */

  dest_mac[0] =  (   MCA<<4 | Device<<3 | LU)&0xFF;
  dest_mac[1] = (Address<<4 | Command)&0xFF;
  dest_mac[2] = (ePackID>>24)&0xFF;
  dest_mac[3] = (ePackID>>16)&0xFF;
  dest_mac[4] = (ePackID>>8)&0xFF;
  dest_mac[5] = (ePackID>>0)&0xFF;

  eCommID = dest_mac[0]<<8 | (dest_mac[1]&0xF3);

  memcpy((void *)eh->h_dest,  (void*)dest_mac, ETH_ALEN);
  memcpy((void *)eh->h_source,(void*)src_mac,  ETH_ALEN);

/* Update data size */

  eh->h_proto = htons(Dsize);

/* Output data string */

  i = sendto(s, buffer, (Dsize + ETHER_HDR_LEN), 0,
               ((struct sockaddr*)&socket_address), sizeof(socket_address));

  if (i == -1) { perror("sendto():"); exit(1); }
}


void WritePort(unsigned char *ComPtr)
{
  unsigned char *Start;

  int i, ind;
  int LL = strlen(ComPtr);
  int Dsize;

  unsigned char CC;


/* allow 50 ms settling time */

  usleep(50000);

/* reset pointer to beginning of the ethernet data field */

  eth_data = buffer + ETH_HLEN;
  Start = eth_data;


/* Loop over characters in the command */

  for (ind=0; ind<LL; ind++)
     {
       CC = *ComPtr++;

       *(eth_data++) = 0x00;  // Start bit

/* Loop over data bits */

        for (i=0; i<8; i++) *(eth_data++) = ((CC>>i) & 0x01);

       *(eth_data++) = 0x01;  // Stop bit
       *(eth_data++) = 0x01;  // Safety delay

     } /* end for(LL) */

  Dsize = (int)(eth_data - Start);

  if (Dsize<SAFE_LEN) PING(MC, MCA, Write, Dsize);

} /* end WritePort */



void PL155init(void)
{
//  WritePort("*RST\n");

  WritePort("IFLOCK\n");
  WritePort("IRANGE1 2\n");

  WritePort("OCP1 5.000\n");
  WritePort("OVP1 5.000\n");

//  WritePort("I1 0.000\n");
  WritePort("V1 0.000\n");

  WritePort("DELTAI1 0.001\n");
  WritePort("DELTAV1 0.001\n");

  WritePort("OP1 1\n");
}


void PL155end(void)
{
  WritePort("IFUNLOCK\n");
  WritePort("LOCAL\n");
}



/***************************************/
/*             main module             */
/*                                     */
/* send configuration to motherboard N */ 
/***************************************/

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

  float Voltage = 0;

  unsigned char Volt[5];
  unsigned char Comm[9];

  for (i=0; i<5; i++) Volt[i] = 0x00;
  for (i=0; i<9; i++) Comm[i] = 0x00;

  if (argc==2) Voltage = atof(argv[1]);

  sprintf(Volt, "%5.3f", Voltage);

  printf("%s\n", Volt);

  strcat(Comm, "V1 ");
  strcat(Comm,  Volt);
  strcat(Comm, "\n");

/* Create socket */

  eth_raw_init();
  buffer = (void*)malloc(ETHER_MAX_LEN);

/* set structurised pointer to beginning of ethernet header */

  eth_head = buffer;  
  eh = (struct ethhdr *)eth_head;


/* Operate device */

//  PL155init();
  WritePort(Comm);
//  PL155end();

  return 0;

} /* end main */



/***************************************/
/*            eth_raw_init             */
/***************************************/

int eth_raw_init()
{
  int i;
  struct ifreq ifr;
  int ifindex = 0;  // Ethernet Interface index
  
/* Open socket */

  s = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
  if (s == -1) { perror("socket():"); exit(1); }

  printf("Successfully opened socket: %i\n", s);
  
/*Get ethernet interface index */

  strncpy(ifr.ifr_name, DEVICE, IFNAMSIZ);
  if (ioctl(s, SIOCGIFINDEX, &ifr) == -1) { perror("SIOCGIFINDEX"); exit(1); }

  ifindex = ifr.ifr_ifindex;
  printf("Successfully got interface index: %i\n", ifindex);
  
/* retrieve corresponding MAC */

  if (ioctl(s, SIOCGIFHWADDR, &ifr) == -1) { perror("SIOCGIFINDEX"); exit(1); }

  for (i=0; i<6; i++) src_mac[i] = ifr.ifr_hwaddr.sa_data[i];

  printf("Host MAC address: ");
  for (i=0; i<6; i++) printf("%02X:", src_mac[i]);
  printf("\n");
  
/* prepare sockaddr_ll */

  socket_address.sll_family   = PF_PACKET;
  socket_address.sll_protocol = 0x00;
  socket_address.sll_ifindex  = ifindex;
  socket_address.sll_pkttype  = PACKET_BROADCAST;
  socket_address.sll_halen    = ETH_ALEN;
  socket_address.sll_addr[0]  = dest_mac[0];
  socket_address.sll_addr[1]  = dest_mac[1];
  socket_address.sll_addr[2]  = dest_mac[2];
  socket_address.sll_addr[3]  = dest_mac[3];
  socket_address.sll_addr[4]  = dest_mac[4];
  socket_address.sll_addr[5]  = dest_mac[5];
  socket_address.sll_addr[6]  = 0x00; 
  socket_address.sll_addr[7]  = 0x00;
														
  
/* establish signal handler */

  signal(SIGINT, sigint);
  printf("Established signal handler for SIGINT\n");
  
  return 0;
}

/***************************************/
/*          Socket Terminator          */
/***************************************/

void sigint(int signum)
{
  struct ifreq ifr;
	
  if (s == -1) return; // No socket has beeen created
	
  strncpy(ifr.ifr_name, DEVICE, IFNAMSIZ);
  ioctl(s, SIOCGIFFLAGS, &ifr);
  ifr.ifr_flags &= ~IFF_PROMISC;
  ioctl(s, SIOCSIFFLAGS, &ifr);
  close(s);

  free(buffer);
	
  printf("Client terminating...\n");
  exit(0);
}