source: tools/tracereport/ecn_report.c @ dd1217b

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

static stat_t ecn_stat[3][4] = {{{0,0}}} ;

void ecn_per_packet(struct libtrace_packet_t *packet)
{
        struct libtrace_ip *ip = trace_get_ip(packet);
        libtrace_direction_t dir = trace_get_direction(packet);
        int ecn;
        
        if (!ip)
                return;
        
        if (dir != TRACE_DIR_INCOMING && dir != TRACE_DIR_OUTGOING)
                dir = TRACE_DIR_OTHER;
        
        ecn = ip->ip_tos & 0x2;
        ecn_stat[dir][ecn].count++;
        ecn_stat[dir][ecn].bytes+=trace_get_wire_length(packet);
}

void ecn_report(void)
{
        int i,j;
        int total = 0;
        
        FILE *out = fopen("ecn.rpt", "w");
        if (!out) {
                perror("fopen");
                return;
        }

        /* Put some headings up for human-readability */
        fprintf(out, "%-12s\t%10s\t%16s %16s\n",
                        "ECN",
                        "DIRECTION",
                        "BYTES",
                        "PACKETS");
        
        for(i=0;i<4;++i) {
                if (ecn_stat[0][i].count==0 && 
                        ecn_stat[1][i].count==0 && ecn_stat[2][i].count==0)
                        continue;
                switch(i){
                        case 1:
                                fprintf(out, "%12s", "ECN CAPE |");
                                break;
                        case 2:
                                fprintf(out, "%12s", "ECN CAPE |");
                                break;
                        case 3:
                                fprintf(out, "%12s", "CONG EXP |");
                                break;
                        default:
                                fprintf(out, "%12s", "NO ECN |");       
                }
                
                for(j=0;j<3;j++){
                        if (j != 0) {
                                fprintf(out, "%12s", " |");
                        }
                        
                        switch (j) {
                                case 0:
                                        fprintf(out, "\t%10s", "Outbound");
                                        break;
                                case 1:
                                        fprintf(out, "\t%10s", "Inbound");
                                        break;
                                case 2:
                                        fprintf(out, "\t%10s", "Unknown");
                                        break;
                        }
                        fprintf(out, "\t%16" PRIu64 " %16" PRIu64 "\n",
                                ecn_stat[j][i].bytes,
                                ecn_stat[j][i].count);
                }
        }
        
        for(i=0;i<4;i++){
                for(j=1;j<4;j++)
                        total += ecn_stat[i][j].count;
        }
        fprintf(out, "%s: %i\n", "Total ECN", total);
        fclose(out);
}