#include <netdb.h>
#include <inttypes.h>
#include <lt_inttypes.h>
#include <stdio.h>
#include "libtrace.h"
#include "tracereport.h"

static stat_t ttl_stat[3][256] = {{{0,0}}} ;
static bool suppress[3] = {true,true,true};

void ttl_per_packet(struct libtrace_packet_t *packet)
{
	struct libtrace_ip *ip = trace_get_ip(packet);
	libtrace_direction_t dir = trace_get_direction(packet);
	
	if (!ip)
		return;
	
	if (dir != TRACE_DIR_INCOMING && dir != TRACE_DIR_OUTGOING)
		dir = TRACE_DIR_OTHER;
	
	ttl_stat[dir][ip->ip_ttl].count++;
	ttl_stat[dir][ip->ip_ttl].bytes+=trace_get_wire_length(packet);
	suppress[dir] = false;
}

void ttl_suppress()
{
	int i;
	printf("%-20s","Direction:");
	for(i=0;i<3;i++){
		if(!suppress[i]){
			switch(i){
				case 0:
					printf("\t%24s", "Outbound   ");
					break;
				case 1:
					printf("\t%24s", "Inbound   ");
					break;
				case 2:
					printf("\t%24s", "Undefined   ");
					break;
				default:
					break;
			}
		}
	}
	printf("\n");
	printf("%-20s","TTL");
	for(i=0;i<3;i++){
		if(!suppress[i]){
			printf("\t%12s\t%12s", "bytes","packets");
		}
	}
	printf("\n");
}

void ttl_report(void)
{
	int i,j;
	printf("# TTL breakdown:\n");
	ttl_suppress();
	for(i=0;i<256;++i) {
		if (ttl_stat[0][i].count==0 && 
			ttl_stat[1][i].count==0 && ttl_stat[2][i].count==0)
			continue;
		printf("%20i:",i);
		for(j=0;j<3;j++){
			if (ttl_stat[j][i].count==0){
				if(!suppress[j])
					printf("\t%24s"," ");
				continue;
			}
			printf("\t%12" PRIu64 "\t%12" PRIu64,
				ttl_stat[j][i].bytes,
				ttl_stat[j][i].count);
		}
		printf("\n");
	}
}