#include <system.h>

typedef struct {
	char *name;
	uint16_t id;
} vendor_t;

static vendor_t **vendor_list;
static uint64_t num_vendors;

static char *find_vendor(uint16_t id) {
	for (uint64_t i = 0; i < num_vendors; i++) {
		if (vendor_list[i]->id == id) { return vendor_list[i]->name; }
	}
	return "Нет в базе";
}

static inline uint16_t pci_read_word(uint16_t bus, uint16_t slot, uint16_t func, uint16_t offset) {
	uint64_t address;
	uint64_t lbus = (uint64_t)bus;
	uint64_t lslot = (uint64_t)slot;
	uint64_t lfunc = (uint64_t)func;
	uint16_t tmp = 0;
	address = (uint64_t)((lbus << 16) | (lslot << 11) | (lfunc << 8) | (offset & 0xFC) | ((uint32_t)0x80000000));
	outl(0xCF8, address);
	tmp = (uint16_t)((inl(0xCFC) >> ((offset & 2) * 8)) & 0xFFFF);
	return (tmp);
}

static inline uint16_t get_vendor_id(uint16_t bus, uint16_t slot, uint16_t function) {
	uint32_t r0 = pci_read_word(bus, slot, function, 0);
	return r0;
}

static inline uint16_t get_device_id(uint16_t bus, uint16_t slot, uint16_t function) {
	uint32_t r0 = pci_read_word(bus, slot, function, 2);
	return r0;
}

static inline uint16_t get_class_id(uint16_t bus, uint16_t slot, uint16_t function) {
	uint32_t r0 = pci_read_word(bus, slot, function, 0xA);
	return (r0 & ~0x00FF) >> 8;
}

static inline uint16_t get_sub_class_id(uint16_t bus, uint16_t slot, uint16_t function) {
	uint32_t r0 = pci_read_word(bus, slot, function, 0xA);
	return (r0 & ~0xFF00);
}

static char *get_class_name(uint16_t number) {
	switch (number) {
		case 0x0: return "Неклассифицирован";
		case 0x1: return "Контроллер устройства хранения";
		case 0x2: return "Контроллер сети";
		case 0x3: return "Контроллер дисплея";
		case 0x4: return "Мультимедийный контроллер";
		case 0x5: return "Контроллер памяти";
		case 0x6: return "Мост";
		case 0x7: return "Простой коммуникационный контроллер";
		case 0x8: return "Периферийное устройство базовой системы";
		case 0x9: return "Устройство ввода";
		case 0xA: return "Док-станция";
		case 0xB: return "Процессор";
		case 0xC: return "Контроллер последовательной шины";
		case 0xD: return "Беспроводной контроллер";
		case 0xE: return "Интеллектуальный контроллер ввода/вывода";
		case 0xF: return "Контроллер спутниковой связи";
		case 0x10: return "Контроллер шифрования/дешифрования";
		case 0x11: return "Контроллер сбора данных и обработки сигналов";
		default: return "Неизвестный";
	}
}

static inline void scan( ) {
	uint64_t devices = 0;
	for (uint32_t bus = 0; bus < 256; bus++) {
		for (uint32_t slot = 0; slot < 32; slot++) {
			for (uint32_t function = 0; function < 8; function++) {
				uint16_t vendor = get_vendor_id(bus, slot, function);

				if (vendor == 0xFFFF) { continue; }

				uint16_t device_id = get_device_id(bus, slot, function);
				uint16_t class_id = get_class_id(bus, slot, function);

				uint16_t status = pci_read_word(bus, slot, function, 0x6);
				/*
				uint32_t mem_addr_0 = pci_read_word(bus, slot, function, 0x1C);
				uint32_t mem_addr_1 = pci_read_word(bus, slot, function, 0x24);
				uint32_t mem_lim_0 = pci_read_word(bus, slot, function, 0x20);
				uint32_t mem_lim_1 = pci_read_word(bus, slot, function, 0x28);
				uint32_t io_addr_0 = pci_read_word(bus, slot, function, 0x2C);
				uint32_t io_addr_1 = pci_read_word(bus, slot, function, 0x34);
				uint32_t io_lim_0 = pci_read_word(bus, slot, function, 0x30);
				uint32_t io_lim_1 = pci_read_word(bus, slot, function, 0x38);
				*/

				char *name = find_vendor(vendor);
				log_printf("[%4u] %4x [%10s], устройство: %x, %u.%u.%u | ", devices, vendor, name, device_id, bus, slot,
				           function);
				log_printf("%32s", get_class_name(class_id));

				/*
				log_printf(" | %8x : %8x", mem_addr_0, mem_lim_0);
				log_printf(" | %8x : %8x", mem_addr_1, mem_lim_1);
				log_printf(" | %8x : %8x", io_addr_0, io_lim_0);
				log_printf(" | %8x : %8x", io_addr_1, io_lim_1);
				*/

				log_printf(" | %4x\n", status);
				devices++;
			}
		}
	}
}

void __attribute__((section(".minit"))) init(env_t *env) {
	init_env(env);

	log_printf("pci_data %x\n", 1);
	module_info_t *pci_data = get_module("[PCI][ADAPTER]");
	log_printf("pci_data %x\n", pci_data);

	if (pci_data == NULL) {
		log_printf("Адаптер PCI данных не найден!\n");
		num_vendors = 0;
	} else {
		num_vendors = pci_data->data_size - 1;
		log_printf("Записей в базе PCI: %u\n", pci_data->data_size);
		vendor_list = (vendor_t **)pci_data->data;
	}

	scan( );
	env->ret = &((module_info_t){ .name = "[PCI]",
	                              .message = "PCI драйвер",
	                              .type = 0,
	                              .data_size = 0,
	                              .data = 0,
	                              .err_code = 0,
	                              .module_id = 0,
	                              .irq = 0,
	                              .irq_handler = 0,
	                              .get_func = 0,
	                              .after_init = 0 });
	mod_update_info(env);
	delete_thread( );
}