#include <stdio.h>
#include <assert.h>
#include "libtrace.h"
#include <map>

struct libtrace_packet_t packet;
int psize;

/** parse an option
 * @param ptr	the pointer to the current option
 * @param plen	the length of the remaining buffer
 * @param type	the type of the option
 * @param optlen the length of the option
 * @param data	the data of the option
 *
 * @returns bool true if there is another option (and the fields are filled in)
 */
int get_next_option(unsigned char **ptr,int *len,
			unsigned char *type,
			unsigned char *optlen,
			unsigned char **data)
{
	if (*len<=0) {
//		printf("Missing End of Options\n");
		return 0;
	}
	*type=**ptr;
	switch(*type) {
		case 0: /* End of options */
	//		printf("End of option\n");
			return 0;
		case 1: /* Pad */
			(*ptr)++;
			(*len)--;
			return 1;
		default:
		case 6: // ECHO (obsolete)
		case 7: // ECHO Reply (obsolete)
		case 9: // Partial ordering
		case 10: // Partial ordering service profile
		case 11: // CC
		case 13: // CC.ECHO
		case 14: // Alternative checksum request
		case 15: // Alternative checksum data
		case 16: // Skeeter
		case 17: // Bubba
		case 18: // Trailer checksum
		case 19: // Md5 signature
		case 20: // SCPS capability
		case 21: // Selective NACK
		case 22: // Record boundary
		case 23: // Corruption experienced
		case 24: // SNAP
		case 25: // Unassigned
		case 26: // TCP Compression filter
			printf("Unknown option type (%i)\n",*type);
		case 2: // MSS
		case 3: // WS
		case 4: // SACK permitted
		case 5: // SACK
		case 8: // Timestamp
		case 12: // CC.new
			*optlen = *(*ptr+1);
			if (*optlen<2) {
				printf("Optlen <2?! %i\n",*optlen);
				return 0; // I have no idea wtf is going on
					  // with these packets
			}
			(*len)-=(unsigned int)*optlen;
			(*data)=(*ptr+2);
			(*ptr)+=*optlen;
			if (*len<0) {
				printf("Option longer than option area (%i > %i)\n",*optlen,*len+*optlen);
				return 0;
			}
			return 1;
	}
	assert(0);
}

int main(int argc, char *argv[])
{
	struct libtrace_t *trace;
	double last = 0;

	trace = trace_create(argv[1]);

	for (;;) {
		int psize;

		if ((psize = trace_read_packet(trace, &packet)) <= 0) {
			break;
		}

		struct libtrace_ip *ipptr = trace_get_ip(&packet);

		if (!ipptr) 
			continue;

		struct libtrace_tcp *tcpptr = trace_get_tcp(&packet);

		if (!tcpptr)
			continue;

		int plen = (tcpptr->doff*4-sizeof *tcpptr);
		int plen2;

		// The len of the section that isn't the tcp header
		plen2 = ((char*)tcpptr - (char*)packet.buffer);
		// Minus the length of the tcp header
		plen2 += sizeof *tcpptr;
		// Now, how much do we have left?
		plen2 = psize - plen2;
		
		if (plen != plen2) {
			printf("Snapped wrong (%i != %i)\n",plen,plen2);
			printf(" %i != %i\n",psize-plen,psize-plen2);
			printf("iph %i tcp->doff %i\n",ipptr->ip_hl,tcpptr->doff);
			printf("psize = %i totlen=%i\n",psize,htons(ipptr->ip_len));
		}


		double now=trace_get_seconds(&packet);

		if (now-last>60) {
			printf(".");
			fflush(stdout);
			last=now;
		}

	}

	return 0;
}