devmgr_srv/
lib.rs

1#![feature(naked_functions)]
2
3use devmgr::{DriverSpec, OwnedDevice};
4use pci_types::device_type::DeviceType;
5use twizzler::{
6    collections::vec::VecObject,
7    object::{ObjID, ObjectBuilder},
8};
9use twizzler_abi::kso::{KactionCmd, KactionFlags};
10use twizzler_driver::{
11    bus::pcie::{PcieDeviceInfo, PcieFunctionHeader, PcieKactionSpecific},
12    device::{BusType, Device},
13};
14use twizzler_rt_abi::error::TwzError;
15use volatile::map_field;
16
17fn get_pcie_offset(bus: u8, device: u8, function: u8) -> usize {
18    ((bus as usize * 256) + (device as usize * 8) + function as usize) * 4096
19}
20
21fn start_pcie_device(seg: &Device, bus: u8, device: u8, function: u8) {
22    let kr = seg.kaction(
23        KactionCmd::Specific(PcieKactionSpecific::RegisterDevice.into()),
24        ((bus as u64) << 16) | ((device as u64) << 8) | (function as u64),
25        KactionFlags::empty(),
26        0,
27    );
28    match kr {
29        Ok(_) => {}
30        Err(_) => eprintln!(
31            "failed to register pcie device {:x}.{:x}.{:x}",
32            bus, device, function
33        ),
34    }
35}
36
37fn start_pcie(seg: Device) {
38    tracing::info!("[devmgr] scanning PCIe bus");
39    let mmio = seg.get_mmio(0).unwrap();
40
41    for bus in 0..=255 {
42        for device in 0..32 {
43            let off = get_pcie_offset(bus, device, 0);
44            let cfg = unsafe { mmio.get_mmio_offset::<PcieFunctionHeader>(off) };
45            let cfg = cfg.as_ptr();
46            if map_field!(cfg.vendor_id).read() != 0xffff
47                && map_field!(cfg.device_id).read() != 0xffff
48                && map_field!(cfg.vendor_id).read() != 0
49            {
50                let mf = if map_field!(cfg.header_type).read() & 0x80 != 0 {
51                    7
52                } else {
53                    0
54                };
55                for function in 0..=mf {
56                    let off = get_pcie_offset(bus, device, function);
57                    let cfg = unsafe { mmio.get_mmio_offset::<PcieFunctionHeader>(off) };
58                    let cfg = cfg.as_ptr();
59                    if map_field!(cfg.vendor_id).read() != 0xffff {
60                        let dt = DeviceType::from((
61                            map_field!(cfg.class).read(),
62                            map_field!(cfg.subclass).read(),
63                        ));
64                        tracing::info!(
65                            "pcie device: {:02x}:{:02x}.{:02x} -- {:?}",
66                            bus,
67                            device,
68                            function,
69                            dt
70                        );
71                        start_pcie_device(&seg, bus, device, function)
72                    }
73                }
74            }
75        }
76    }
77}
78
79#[secgate::secure_gate]
80pub fn devmgr_start() -> Result<(), TwzError> {
81    tracing::subscriber::set_global_default(
82        tracing_subscriber::fmt()
83            .with_max_level(tracing::Level::INFO)
84            .without_time()
85            .finish(),
86    )
87    .unwrap();
88
89    let device_root = twizzler_driver::get_bustree_root();
90    for device in device_root.children() {
91        if device.is_bus() && device.bus_type() == BusType::Pcie {
92            start_pcie(device);
93        }
94    }
95    Ok(())
96}
97
98#[secgate::secure_gate]
99pub fn get_devices(spec: DriverSpec) -> Result<ObjID, TwzError> {
100    match spec.supported {
101        devmgr::Supported::PcieClass(class, subclass, progif) => {
102            let device_root = twizzler_driver::get_bustree_root();
103            let mut ids = Vec::new();
104            for device in device_root.children() {
105                if device.is_bus() && device.bus_type() == BusType::Pcie {
106                    for child in device.children() {
107                        let info = unsafe { child.get_info::<PcieDeviceInfo>(0).unwrap() };
108                        if info.get_data().class == class
109                            && info.get_data().subclass == subclass
110                            && info.get_data().progif == progif
111                        {
112                            ids.push(child.id());
113                        }
114                    }
115                }
116            }
117
118            tracing::debug!("found devices {:?} for spec {:?}", ids, spec);
119            let mut owned_devices_object = VecObject::new(ObjectBuilder::default())?;
120            for id in ids {
121                owned_devices_object.push(OwnedDevice { id })?;
122            }
123            // TODO: on-drop for this object.
124            Ok(owned_devices_object.object().id())
125        }
126        devmgr::Supported::Vendor(vendor_code, device_code) => {
127            let device_root = twizzler_driver::get_bustree_root();
128            let mut ids = Vec::new();
129            for device in device_root.children() {
130                if device.is_bus() && device.bus_type() == BusType::Pcie {
131                    for child in device.children() {
132                        let info = unsafe { child.get_info::<PcieDeviceInfo>(0).unwrap() };
133                        if info.get_data().device_id == device_code
134                            && info.get_data().vendor_id == vendor_code
135                        {
136                            ids.push(child.id());
137                        }
138                    }
139                }
140            }
141
142            tracing::debug!("found devices {:?} for spec {:?}", ids, spec);
143            let mut owned_devices_object = VecObject::new(ObjectBuilder::default())?;
144            for id in ids {
145                owned_devices_object.push(OwnedDevice { id })?;
146            }
147            // TODO: on-drop for this object.
148            Ok(owned_devices_object.object().id())
149        }
150    }
151}
devmgr_srv/lib.rs

devmgr_srv/
lib.rs
