// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2019 Ramon Fried <rfried.dev@gmail.com>
 */

#include <net.h>
#include <net/pcap.h>
#include <time.h>
#include <linux/errno.h>
#include <asm/io.h>

#define LINKTYPE_ETHERNET	1

static bool initialized;
static bool running;
static bool buffer_full;
static void *buf;
static unsigned int max_size;
static unsigned int pos;

static unsigned long incoming_count;
static unsigned long outgoing_count;

struct pcap_header {
	u32 magic;
	u16 version_major;
	u16 version_minor;
	s32 thiszone;
	u32 sigfigs;
	u32 snaplen;
	u32 network;
};

struct pcap_packet_header {
	u32 ts_sec;
	u32 ts_usec;
	u32 incl_len;
	u32 orig_len;
};

static struct pcap_header file_header = {
	.magic = 0xa1b2c3d4,
	.version_major = 2,
	.version_minor = 4,
	.snaplen = 65535,
	.network = LINKTYPE_ETHERNET,
};

int pcap_init(phys_addr_t paddr, unsigned long size)
{
	buf = map_physmem(paddr, size, 0);
	if (!buf) {
		printf("Failed mapping PCAP memory\n");
		return -ENOMEM;
	}

	printf("PCAP capture initialized: addr: 0x%lx max length: %lu\n",
	       (unsigned long)buf, size);

	memcpy(buf, &file_header, sizeof(file_header));
	pos = sizeof(file_header);
	max_size = size;
	initialized = true;
	running = false;
	buffer_full = false;
	incoming_count = 0;
	outgoing_count = 0;
	return 0;
}

int pcap_start_stop(bool start)
{
	if (!initialized) {
		printf("error: pcap was not initialized\n");
		return -ENODEV;
	}

	running = start;

	return 0;
}

int pcap_clear(void)
{
	if (!initialized) {
		printf("error: pcap was not initialized\n");
		return -ENODEV;
	}

	pos = sizeof(file_header);
	incoming_count = 0;
	outgoing_count = 0;
	buffer_full = false;

	printf("pcap capture cleared\n");
	return 0;
}

int pcap_post(const void *packet, size_t len, bool outgoing)
{
	struct pcap_packet_header header;
	u64 cur_time = timer_get_us();

	if (!initialized || !running || !buf)
		return -ENODEV;

	if (buffer_full)
		return -ENOMEM;

	if ((pos + len + sizeof(header)) >= max_size) {
		buffer_full = true;
		printf("\n!!! Buffer is full, consider increasing buffer size !!!\n");
		return -ENOMEM;
	}

	header.ts_sec = cur_time / 1000000;
	header.ts_usec = cur_time % 1000000;
	header.incl_len = len;
	header.orig_len = len;

	memcpy(buf + pos, &header, sizeof(header));
	pos += sizeof(header);
	memcpy(buf + pos, packet, len);
	pos += len;

	if (outgoing)
		outgoing_count++;
	else
		incoming_count++;

	env_set_hex("pcapsize", pos);

	return 0;
}

int pcap_print_status(void)
{
	if (!initialized) {
		printf("pcap was not initialized\n");
		return -ENODEV;
	}
	printf("PCAP status:\n");
	printf("\tInitialized addr: 0x%lx\tmax length: %u\n",
	       (unsigned long)buf, max_size);
	printf("\tStatus: %s.\t file size: %u\n", running ? "Active" : "Idle",
	       pos);
	printf("\tIncoming packets: %lu Outgoing packets: %lu\n",
	       incoming_count, outgoing_count);

	return 0;
}

bool pcap_active(void)
{
	return running;
}