#include <stdio.h>
#include <stdint.h>
#include "pcap.h"
#include "stream.h"

#define IPPROTO_TCP 6
#define IPPROTO_UDP 17
#define IPPROTO_ICMP 1

#define TH_FIN 0x01
#define TH_SYN 0x02
#define TH_RST 0x04
#define TH_PSH 0x08
#define TH_ACK 0x10
#define TH_URG 0x20

void
ip4_addr(char *addr_s, uint32_t addr)
{
	snprintf(addr_s, 16, "%u.%u.%u.%u", (addr >> 24) & 0xff, (addr >> 16) & 0xff, (addr >> 8) & 0xff, (addr >> 0) & 0xff);
}

void
print_payload(struct stream *s)
{
	while (s->len) {
		printf("%02x", read_uint8(s));
	}
}

void
process_tcp(struct stream *s, char *saddr_s, char *daddr_s)
{
	uint16_t sport = read_uint16be(s);
	uint16_t dport = read_uint16be(s);
	uint32_t seq = read_uint32be(s);
	uint32_t ack = read_uint32be(s);
	uint8_t off = read_uint8(s);
	uint8_t hlen = (off >> 4) * 4;
	uint8_t flags = read_uint8(s);
	uint16_t window = read_uint16be(s);
	uint16_t chksum = read_uint16be(s);
	uint16_t urgent = read_uint16be(s);

	if (hlen < 20) {
		printf("!");
	}

	printf("TCP %s,%u,%u %s,%u,%u ", saddr_s, sport, seq, daddr_s, dport, ack);
}

void
process_udp(struct stream *s, char *saddr_s, char *daddr_s)
{
	uint16_t sport = read_uint16be(s);
	uint16_t dport = read_uint16be(s);
	uint16_t len = read_uint16be(s);
	uint16_t chksum = read_uint16be(s);

	printf("UDP %s,%u %s,%u ", saddr_s, sport, daddr_s, dport);
}

void
process_icmp(struct stream *s, char *saddr_s, char *daddr_s)
{
	uint8_t type = read_uint8(s);
	uint8_t code = read_uint8(s);
	uint16_t checksum = read_uint16be(s);
	
	printf("ICMP %s %s %d ", saddr_s, daddr_s, code);
}

void
process_ip4(struct stream *s)
{
	uint8_t vhl = read_uint8(s);
	uint8_t ihl = (vhl & 0x0f) * 4;
	uint8_t tos = read_uint8(s);
	uint16_t length = read_uint16be(s);
	uint16_t id = read_uint16be(s);
	uint16_t off = read_uint16be(s);
	uint8_t ttl = read_uint8(s);
	uint8_t proto = read_uint8(s);
	uint16_t chksum = read_uint16be(s);
	uint32_t saddr = read_uint32be(s);
	uint32_t daddr = read_uint32be(s);

	char saddr_s[20];
	char daddr_s[20];

	ip4_addr(saddr_s, saddr);
	ip4_addr(daddr_s, daddr);

	// Ignore options
	sskip(s, 20 - ihl);

	// Force stream length to IP payload length
	s->len = length - ihl;

	switch (proto) {
		case IPPROTO_TCP:
			process_tcp(s, saddr_s, daddr_s);
			break;
		case IPPROTO_UDP:
			process_udp(s, saddr_s, daddr_s);
			break;
		case IPPROTO_ICMP:
			process_icmp(s, saddr_s, daddr_s);
			break;
		default:
			printf("P%d %s %s ", proto, saddr_s, daddr_s);
			break;
	}

	print_payload(s);
}


void
print_ethernet(struct stream *s)
{
	uint8_t saddr[6];
	uint8_t daddr[6];
	uint16_t ethertype;

	sread(s, &saddr, sizeof(saddr));
	sread(s, &daddr, sizeof(daddr));
	ethertype = read_uint16be(s);

	if (ethertype == 0x8100) {
		// VLAN 
		read_uint16be(s);
		ethertype = read_uint16be(s);
	}

	switch (ethertype) {
		case 0x0800:	// IPv4
			process_ip4(s);
			break;
	}
}

void
print_frame(struct pcap_file *p, struct pcap_pkthdr *hdr, char const *frame)
{
	struct stream streambuf;
	struct stream *s = &streambuf;

	sinit(s, frame, hdr->caplen);
	printf("%u.%u ", hdr->ts.tv_sec, hdr->ts.tv_usec);
	switch (p->linktype) {
		case LINKTYPE_ETHERNET:
			print_ethernet(s);
			break;
		case LINKTYPE_RAW:
			process_ip4(s);
			break;
	}
	printf("\n");
}

void
pcat(FILE * f)
{
	struct pcap_file p;

	if (-1 == pcap_open_in(&p, f))
		return;

	for (;;) {
		struct pcap_pkthdr hdr;
		char frame[MAXFRAME];

		if (-1 == pcap_read_pkthdr(&p, &hdr)) {
			break;
		}

		if (1 != fread(frame, hdr.caplen, 1, f)) {
			break;
		}

		print_frame(&p, &hdr, frame);
	}
}


int
main(int argc, char *argv[])
{
	pcat(stdin);

	return 0;
}