vm_arm: add pcpu list for core pinning

Adds a pcpu list to the VM_Arm component to allow pinning of vcpus to specific physical cpus within a CAmkES configuration. Signed-off-by: Alex Pavey <[email protected]>
seL4 · Nov 17, 2022 · eadf2e8 · eadf2e8
1 parent 379f2cf
commit eadf2e8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/components/VM_Arm/configurations/vm.h b/components/VM_Arm/configurations/vm.h
@@ -65,6 +65,7 @@
     provides VMDTBPassthrough dtb; \
     attribute int base_prio; \
     attribute int num_vcpus = 1; \
+    attribute int pcpus[] = []; \
     attribute int num_extra_frame_caps; \
     attribute int extra_frame_map_address; \
     attribute { \

diff --git a/components/VM_Arm/src/main.c b/components/VM_Arm/src/main.c
@@ -1173,8 +1173,27 @@ int main_continued(void)
         vm_vcpu_t *new_vcpu = create_vmm_plat_vcpu(&vm, VM_PRIO - 1);
         assert(new_vcpu);
     }
+
+    int pcpu_assignments = 0;
+    if (get_instance_size_pcpus_list() < NUM_VCPUS) {
+        pcpu_assignments = get_instance_size_pcpus_list();
+    } else {
+        pcpu_assignments = NUM_VCPUS;
+    }
+    /* Assign vcpus to explicit pcpus */
+    for (int j = 0; j < pcpu_assignments; j++) {
+        vm.vcpus[j]->target_cpu = pcpus[j];
+    }
+
     vm_vcpu_t *vm_vcpu = vm.vcpus[BOOT_VCPU];
-    err = vm_assign_vcpu_target(vm_vcpu, 0);
+
+    /* Use affinity as boot core if pcpus are not specified */
+    if (0 == pcpu_assignments) {
+        err = vm_assign_vcpu_target(vm_vcpu, get_instance_affinity());
+    } else {
+        err = vm_assign_vcpu_target(vm_vcpu, vm_vcpu->target_cpu);
+    }
+
     if (err) {
         return -1;
     }