stats.c

Go to the documentation of this file.

#include "evl.h"

/* Data structure for storing statistics */

typedef struct Uc_Stats_t {
  int totalPkts;
  int totalBytes;
  Hash_t *allIps;
  util_timing_t lastPrintTime;
  util_timing_t printInterval;
} Uc_Stats_t;

/**AutomaticStart */

// prototypes 

static void Uc_PrintStats( Uc_Stats_t *stats );
static Uc_Stats_t * Uc_InitStats( char *argument );
static void Uc_UpdateStats( Evl_Manager_t *mgr, Pkt_ProcessPkt_t *pp, Uc_Stats_t *stats );

/**AutomaticEnd */


/* \brief Code for gathering traffic statistics.  */

int
stats (Evl_Manager_t * mgr, Pkt_ProcessPkt_t * pp, void **mystate, void *argument )
{
  static Uc_Stats_t *stats;

  if (*mystate == NULL) {
    printf( "First call to Uc_Stats\n");
    stats = Uc_InitStats( (char *) argument );
    *mystate = (void *) stats;
  }

  Uc_UpdateStats( mgr, pp, stats );
  Uc_PrintStats( stats );

  return 0;
}


/** \brief Allocate and initialize the stats structure */

static Uc_Stats_t *
Uc_InitStats( 
  char *argument
)
{
  Uc_Stats_t *result = ( Uc_Stats_t * ) malloc( sizeof( Uc_Stats_t ) );

  result->totalPkts = 0;
  result->totalBytes = 0;

  result->allIps = Hash_InitTable( Hash_NumCmp, Hash_NumHash ); 

  util_timing_set( & result->lastPrintTime );

  double printInterval;
  sscanf( argument, "%lf", & printInterval );
  result->printInterval = util_timing_pair( printInterval );

  return result;
}


/** \brief Update stats field */

static void 
Uc_UpdateStats(
  Evl_Manager_t *mgr,
  Pkt_ProcessPkt_t *pp,
  Uc_Stats_t *stats 
)
{
  stats->totalPkts++;
  stats->totalBytes += pp->length;

  if ( Pkt_EthernetReadL3Type( pp->pkt ) == Pkt_L3ProtIp_c ) {
    Pkt_IpHdr_t *ipPkt = Pkt_EthernetExtractIp( pp->pkt );
    Hash_Insert( stats->allIps, ipPkt->sourceIp, 0 );
    Hash_Insert( stats->allIps, ipPkt->destIp, 0 );
  }
}


/** \brief Print current state of stats structure 

   Ensures that at least printInterval time has elapsed
   between printings.

*/

static void
Uc_PrintStats(
  Uc_Stats_t *stats
)
{
  util_timing_t currentTime;
  util_timing_set( & currentTime );
  util_timing_t diffTime = util_timing_diff( currentTime, stats->lastPrintTime );

  if ( util_timing_compare( diffTime, stats->printInterval ) ) {
    return;
  }
  else {
    stats->lastPrintTime = currentTime;
  }

  printf("Packets:\t%d\n", stats->totalPkts );
  printf("Bytes:\t%d\n", stats->totalBytes );

  printf("Num IPs:\t%d\n", Hash_Count( stats->allIps ) );
  printf("---\n\n");
}