Initial commit of OpenSPARC T2 design and verification files.

[OpenSPARC-T2-DV] / verif / env / common / vera / niu_ippktgen / C / wrapper / pgLibnet.c

#if (HAVE_CONFIG_H)
#include "../include/config.h"
#endif

#include "include/libnet.h"

#include "include/pg.h"
#include "include/packet.h"

/* AJXXXX */
libnet_t * libnet_init_pkt(int injection_type, char *device, char *err_buf);
int libnet_write_pkt(libnet_t *l, u_int8_t **packet, u_int32_t *len,
                    u_int32_t align);
void libnet_print_pkt(u_int8_t *pktbuf, u_int32_t pktlen);


extern unsigned char * genPayload (int data_type, unsigned int seed,
                                  int data_length);
extern unsigned char * addPayload (unsigned char *p1, int l1,
                                  unsigned char *p2, int l2) ;
extern unsigned char * addCrc (unsigned char *p1, int l1, unsigned int crc) ;
extern unsigned char *  addL2Pad(unsigned char * pktbuf,int pktlen,int l2_pad_length) ;

extern int libnet_gen_l4_packet(struct packet_desc *packet_vars, libnet_t *l);
extern int libnet_gen_l3_packet(struct packet_desc *packet_vars, libnet_t *l);
extern int libnet_gen_l2_packet(struct packet_desc *packet_vars, libnet_t *l);



char * libnet_call(struct packet_desc *packet_vars, u_int32_t *total_len) {

   int c;
   libnet_t *l;
   char errbuf[LIBNET_ERRBUF_SIZE];
   /* AJXXXX */
   struct libnet_stats ls;
   u_int32_t pktlen=0;
   u_int32_t pktalign=0;
   u_int8_t *pktbuf=NULL;
   u_int32_t crc;
   int status;
   int l2_length;
   int payload_start_ptr;


/*
    *  Initialize the library.  Root priviledges are required.
    */
   /* AJXXXX */
   l = libnet_init_pkt(
           LIBNET_LINK,                            /* injection type */
           NULL,                                   /* network interface */
           errbuf);                                /* error buffer */

   if (l == NULL)
   {
       fprintf(stderr, "libnet_init() failed: %s", errbuf);
       exit(EXIT_FAILURE);
   }


   // The guts of packet gen

   if(packet_vars->l2_pkt_type==L2_RUNT) {
           l2_length = packet_vars->payload.data_length;
           if(l2_length>4) {
                   l2_length= l2_length - 4;
                   // generate payload
                   pktbuf = genPayload(packet_vars->payload.data_type,
                                       packet_vars->payload.data_seed,
                                       l2_length);
                   pktlen = l2_length;
                   crc = libnet_compute_crc(pktbuf,pktlen);
                   pktbuf = addCrc(pktbuf,pktlen,crc); 
                   pktlen+=4;
                   *total_len = pktlen;
           } else { /*NOCRC*/
                   // generate payload
                   pktbuf = genPayload(packet_vars->payload.data_type,
                                       packet_vars->payload.data_seed,
                                       l2_length);
                   pktlen = l2_length;
                   *total_len = pktlen;

           }
   } else {
     status = libnet_gen_l4_packet(packet_vars, l); 
     if(status == -1) {
      printf("ERROR libnet_gen_l4_packet failed \n");
      return(NULL);
     }


     status = libnet_gen_l3_packet(packet_vars, l); 
     if(status == -1) {
      printf("ERROR libnet_gen_l3_packet failed \n");
      return(NULL);
     }
     status = libnet_gen_l2_packet(packet_vars, l); 
     if(status == -1) {
      printf("ERROR libnet_gen_l2_packet failed \n");
      return(NULL);
     }

     /* sbehera - need this for IP header alignment */
     pktalign = 2;
     c = libnet_write_pkt(l, &pktbuf, &pktlen, pktalign);
     if (c == -1)
     { 
      fprintf(stderr, "Write error: %s\n", libnet_geterror(l));
      return(NULL);
     }
     payload_start_ptr = packet_vars->l2_hdr+packet_vars->l3_hdr+ packet_vars->l4_hdr;

     if( ( packet_vars->payload.error_code & PG_CHKSUM_ERR) ){
      if (packet_vars->l4_pkt_type & L4_PKT_TYPE_TCP) { 
           printf("Libnet CheckSum - %x%x\n",pktbuf[payload_start_ptr-4],pktbuf[payload_start_ptr-3]);
           pktbuf[payload_start_ptr-4] =  pktbuf[payload_start_ptr-4]^1;
      } else if(packet_vars->l4_pkt_type & L4_PKT_TYPE_UDP) {
           printf("Libnet CheckSum - %x%x\n",pktbuf[payload_start_ptr-2],pktbuf[payload_start_ptr-1]);
           pktbuf[payload_start_ptr-1] =  pktbuf[payload_start_ptr-1]^1;
      }
     }


    pktbuf = addL2Pad(pktbuf,pktlen,packet_vars->payload.l2_pad_length); 
    pktlen+=packet_vars->payload.l2_pad_length;

    crc = libnet_compute_crc(pktbuf,pktlen);
    
    printf(" Libnet CRC - %x \n",crc);
    if(packet_vars->payload.error_code & PG_CRC_ERR) {
     crc = crc^0x01010101;
     printf("Libnet CRCError - %x \n",crc);
     }
     
    pktbuf = addCrc(pktbuf,pktlen,crc); 
    pktlen+=4;
    *total_len = pktlen;
   }


#if 1 /* AJXXXX */
   // c = libnet_write_pkt(l, &pktbuf, &pktlen, pktalign);
   // crc = libnet_compute_crc(pktbuf,pktlen);
   // pktbuf = addCrc(pktbuf,pktlen,crc);
   // pktlen+=4;
#endif

   if (c == -1) {
       fprintf(stderr, "Write error: %s\n", libnet_geterror(l));
       return(NULL);
   } else {
       fprintf(stderr, "Wrote %d byte packet; check the wire.\n", pktlen);
#if 1 /* AJXXXX */
       // libnet_print_pkt(pktbuf, pktlen);
       return(pktbuf);
#endif
   }

}