/* vim: tabstop=4 shiftwidth=4 noexpandtab
 * This file is part of ToaruOS and is released under the terms
 * of the NCSA / University of Illinois License - see LICENSE.md
 * Copyright (C) 2014-2018 K. Lange
 */
#include <kernel/module.h>
#include <kernel/pci.h>
#include <kernel/printf.h>
#include <kernel/logging.h>
#include <kernel/mod/shell.h>

static uint32_t hub_device = 0;

static void find_usb_device(uint32_t device, uint16_t vendorid, uint16_t deviceid, void * extra) {
	if (pci_find_type(device) == 0xc03) {
		int prog_if = (int)pci_read_field(device, PCI_PROG_IF, 1);
		if (prog_if == 0) {
			*((uint32_t *)extra)= device;
		}
	}
}

DEFINE_SHELL_FUNCTION(usb, "Enumerate USB devices (UHCI)") {

	pci_scan(&find_usb_device, -1, &hub_device);

	if (!hub_device) {
		fprintf(tty, "Failed to locate a UHCI controller.\n");
		return 1;
	}

	fprintf(tty, "Located UHCI controller: %2x:%2x.%d\n",
			(int)pci_extract_bus (hub_device),
			(int)pci_extract_slot(hub_device),
			(int)pci_extract_func(hub_device));

	return 0;
}

static int install(void) {
	BIND_SHELL_FUNCTION(usb);
	return 0;
}

static int uninstall(void) {
	return 0;
}

MODULE_DEF(usbuhci, install, uninstall);
MODULE_DEPENDS(debugshell);