Method assign

crates/openshell-driver-vm/src/gpu.rs:67–115

Assign a GPU to a sandbox. Returns the assignment details including BDF.

(&mut self, sandbox_id: &str, gpu_device: &str)

    /// Assign a GPU to a sandbox. Returns the assignment details including BDF.
    pub fn assign(&mut self, sandbox_id: &str, gpu_device: &str) -> Result<GpuAssignment, String> {
        let slot_idx = if gpu_device.is_empty() {
            self.slots
                .iter()
                .position(|s| s.assigned_to.is_none())
                .ok_or_else(|| "all GPUs are currently assigned to other sandboxes".to_string())?
        } else if let Ok(idx) = gpu_device.parse::<usize>() {
            if idx >= self.slots.len() {
                return Err(format!(
                    "GPU index {idx} out of range (have {} GPUs)",
                    self.slots.len()
                ));
            }
            if self.slots[idx].assigned_to.is_some() {
                return Err(format!(
                    "GPU at index {idx} ({}) is already assigned to another sandbox",
                    self.slots[idx].info.bdf
                ));
            }
            idx
        } else {
            validate_bdf(gpu_device).map_err(|e| e.to_string())?;
            let idx = self
                .slots
                .iter()
                .position(|s| s.info.bdf == gpu_device)
                .ok_or_else(|| format!("GPU {gpu_device} not found in inventory"))?;
            if self.slots[idx].assigned_to.is_some() {
                return Err(format!(
                    "GPU {gpu_device} is already assigned to another sandbox"
                ));
            }
            idx
        };

        let bdf = self.slots[slot_idx].info.bdf.clone();
        let guard = prepare_gpu_for_passthrough(&self.sysfs, &bdf)
            .map_err(|e| format!("failed to prepare GPU {bdf} for passthrough: {e}"))?;

        self.slots[slot_idx].assigned_to = Some(sandbox_id.to_string());
        self.slots[slot_idx].bind_guard = Some(guard);
        self.persist_state();

        Ok(GpuAssignment {
            bdf,
            name: self.slots[slot_idx].info.name.clone(),
            iommu_group: self.slots[slot_idx].info.iommu_group,
        })
    }

    /// Release a GPU assignment. The `GpuBindGuard` is dropped, restoring the GPU.
    pub fn release(&mut self, sandbox_id: &str) {