tool: Set VM VCPU TCB domain to its PDs domain

JE-Archer · Nov 13, 2024 · ef76b9b · ef76b9b
1 parent 04378bf
commit ef76b9b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tool/microkit/src/main.rs b/tool/microkit/src/main.rs
@@ -2711,6 +2711,22 @@ fn build_system(
                     tcb: pd_tcb_objs[pd_idx].cap_addr,
                 },
             ));
+            if let Some(pd_vm) = &pd.virtual_machine {
+                for (vm_idx, vm) in virtual_machines.iter().enumerate() {
+                    if pd_vm.name == vm.name {
+                        for vcpu_idx in 0..vm.vcpus.len() {
+                            system_invocations.push(Invocation::new(
+                                config,
+                                InvocationArgs::DomainSetSet {
+                                    domain_set: DOMAIN_CAP_ADDRESS,
+                                    domain: domain_id as u8,
+                                    tcb: vcpu_tcb_objs[vm_idx + vcpu_idx].cap_addr,
+                                },
+                            ));
+                        }
+                    }
+                }
+            }
         }
     }