// MIT License, Copyright (c) 2020 Marvin Borner
// Uses parts of the ToAruOS Project, released under the terms of the NCSA
// Copyright (C) 2011-2018 K. Lange

#include <drivers/cpu.h>
#include <def.h>
#include <drivers/interrupts.h>
#include <mem.h>
#include <net.h>
#include <drivers/pci.h>
#include <print.h>
#include <drivers/rtl8139.h>

static int rtl_irq = 0;
static u8 mac[6] = { 0 };
static u8 *last_packet = NULL;
static u8 *rtl_rx_buffer = NULL;
static u32 rtl_device_pci = 0;
static u32 rtl_iobase = 0;
static u32 cur_rx = 0;

u8 *rtl8139_get_mac(void)
{
	if (!rtl_device_pci)
		return NULL;

	return mac;
}

static void rtl8139_receive_packet(void)
{
	while ((inb(rtl_iobase + RTL_PORT_CMD) & 0x01) == 0) {
		int offset = cur_rx % 0x2000;

		u32 *buf_start = (u32 *)((u32)rtl_rx_buffer + offset);
		u32 rx_status = buf_start[0];
		u32 rx_size = rx_status >> 16;

		if (rx_status & (0x0020 | 0x0010 | 0x0004 | 0x0002)) {
			print("RX Error\n");
		} else {
			u8 *buf_8 = (u8 *)&(buf_start[1]);

			last_packet = malloc(rx_size);

			u32 packet_end = (u32)buf_8 + rx_size;
			if (packet_end > (u32)rtl_rx_buffer + 0x2000) {
				u32 s = ((u32)rtl_rx_buffer + 0x2000) - (u32)buf_8;
				memcpy(last_packet, buf_8, s);
				memcpy((void *)((u32)last_packet + s), rtl_rx_buffer, rx_size - s);
			} else {
				memcpy(last_packet, buf_8, rx_size);
			}

			u16 *t = (u16 *)(rtl_rx_buffer + cur_rx);
			ethernet_handle_packet((struct ethernet_packet *)last_packet, *(t + 1));
		}

		cur_rx = (cur_rx + rx_size + 4 + 3) & ~3;
		outw(rtl_iobase + RTL_PORT_RXPTR, cur_rx - 16);
	}
}

static u8 tsad_array[4] = { 0x20, 0x24, 0x28, 0x2C };
static u8 tsd_array[4] = { 0x10, 0x14, 0x18, 0x1C };
static u8 tx_current = 0;
void rtl8139_send_packet(void *data, u32 len)
{
	if (!rtl_device_pci)
		return;

	outl(rtl_iobase + tsad_array[tx_current], (u32)data);
	outl(rtl_iobase + tsd_array[tx_current++], len);
	if (tx_current > 3)
		tx_current = 0;
}

static void rtl8139_find(u32 device, u16 vendor_id, u16 device_id, void *extra)
{
	if ((vendor_id == 0x10ec) && (device_id == 0x8139)) {
		*((u32 *)extra) = device;
	}
}

static void rtl8139_irq_handler(struct regs *r)
{
	UNUSED(r);
	u16 status = inw(rtl_iobase + RTL_PORT_ISR);
	if (!status)
		return;
	outw(rtl_iobase + RTL_PORT_ISR, status);

	if (status & 0x01 || status & 0x02)
		rtl8139_receive_packet();
}

static void rtl8139_init(void)
{
	if (!rtl_device_pci)
		return;

	u16 command_reg = (u16)pci_read_field(rtl_device_pci, PCI_COMMAND, 4);
	if ((command_reg & (1 << 2)) == 0) {
		command_reg |= (1 << 2);
		pci_write_field(rtl_device_pci, PCI_COMMAND, command_reg);
	}

	rtl_irq = pci_get_interrupt(rtl_device_pci);
	irq_install_handler(rtl_irq, rtl8139_irq_handler);

	u32 rtl_bar0 = pci_read_field(rtl_device_pci, PCI_BAR0, 4);
	/* u32 rtl_bar1 = pci_read_field(rtl_device_pci, PCI_BAR1, 4); */
	rtl_iobase = 0;

	if (rtl_bar0 & 0x00000001)
		rtl_iobase = rtl_bar0 & 0xFFFFFFFC;

	// Get mac address
	for (int i = 0; i < 6; ++i)
		mac[i] = inb(rtl_iobase + RTL_PORT_MAC + i);
	printf("Mac address: %x:%x:%x:%x:%x:%x\n", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);

	// Activate
	outb(rtl_iobase + RTL_PORT_CONFIG, 0x0);

	// Reset
	outb((u16)(rtl_iobase + RTL_PORT_CMD), 0x10);
	while ((inb(rtl_iobase + RTL_PORT_CMD) & 0x10) != 0)
		;

	// Set receive buffer
	rtl_rx_buffer = (u8 *)malloc(0x3000);
	memset(rtl_rx_buffer, 0, 0x3000);
	outl(rtl_iobase + RTL_PORT_RBSTART, (u32)rtl_rx_buffer);

	// Enable IRQs
	outw(rtl_iobase + RTL_PORT_IMR, 0x8000 | /* PCI error */
						0x4000 | /* PCS timeout */
						0x40 | /* Rx FIFO over */
						0x20 | /* Rx underrun */
						0x10 | /* Rx overflow */
						0x08 | /* Tx error */
						0x04 | /* Tx okay */
						0x02 | /* Rx error */
						0x01 /* Rx okay */
	);

	// Configure transmit
	outl(rtl_iobase + RTL_PORT_TCR, 0);

	// Configure receive
	outl(rtl_iobase + RTL_PORT_RCR, (0) | /* 8K receive */
						0x08 | /* broadcast */
						0x01 /* all physical */
	);

	// Enable receive and transmit
	outb(rtl_iobase + RTL_PORT_CMD, 0x08 | 0x04);

	// Reset rx statistics
	outl(rtl_iobase + RTL_PORT_RXMISS, 0);
}

int rtl8139_installed(void)
{
	return rtl_device_pci != 0;
}

int rtl8139_install(void)
{
	pci_scan(&rtl8139_find, -1, &rtl_device_pci);

	if (rtl_device_pci) {
		print("Found rtl8139 card\n");
		rtl8139_init();
	}
	return rtl_device_pci;
}