axplat-dyn 0.6.2

extern crate alloc;

use alloc::format;

use pcie::CommandRegister;
use rdrive::{
    PlatformDevice, module_driver,
    probe::{
        OnProbeError,
        pci::{EndpointRc, FnOnProbe},
    },
};

use super::{PlatformDeviceUsbHost, USB_KERNEL, align_up_4k, resolve_pci_irq_from_fdt};
use crate::drivers::iomap;

const DRIVER_NAME: &str = "usb-xhci-pci";

module_driver!(
    name: "USB xHCI PCI",
    level: ProbeLevel::PostKernel,
    priority: ProbePriority::DEFAULT,
    probe_kinds: &[ProbeKind::Pci {
        on_probe: probe as FnOnProbe
    }],
);

fn probe(endpoint: &mut EndpointRc, plat_dev: PlatformDevice) -> Result<(), OnProbeError> {
    let class = endpoint.revision_and_class();
    if (class.base_class, class.sub_class, class.interface) != (0x0c, 0x03, 0x30) {
        return Err(OnProbeError::NotMatch);
    }

    let Some(bar) = endpoint.bar_mmio(0) else {
        return Err(OnProbeError::other("xHCI BAR0 MMIO region missing"));
    };

    endpoint.update_command(|mut cmd| {
        cmd.insert(CommandRegister::MEMORY_ENABLE | CommandRegister::BUS_MASTER_ENABLE);
        cmd
    });

    let mmio = iomap(bar.start.into(), align_up_4k(bar.count().max(1)))?;
    let irq_num = Some(
        resolve_pci_irq_from_fdt(endpoint)
            .map_err(|err| OnProbeError::other(alloc::format!("{err}")))?,
    );
    let host = crab_usb::USBHost::new_xhci(mmio, &USB_KERNEL).map_err(|err| {
        OnProbeError::other(format!(
            "failed to create xHCI host for PCI endpoint {}: {err}",
            endpoint.address()
        ))
    })?;

    plat_dev.register_usb_host(DRIVER_NAME, host, irq_num);
    info!(
        "xHCI PCI host registered successfully at {} with irq {:?}",
        endpoint.address(),
        irq_num
    );
    Ok(())
}