feat: 添加 pa_bits 字段到 HCpu 和 VmMachineUninit 结构体，增强物理地址处理能力

Author: ZR233

src/arch/aarch64/cpu.rs

@@ -23,6 +23,7 @@ pub struct HCpu {
     vpercpu: Aarch64PerCpu,
     max_guest_page_table_levels: usize,
     pub pa_range: core::ops::Range<usize>,
+    pub pa_bits: usize,
 }
 
 impl HCpu {
@@ -38,13 +39,15 @@ impl HCpu {
             vpercpu,
             max_guest_page_table_levels: 0,
             pa_range: 0..0,
+            pa_bits: 0,
         }
     }
 
     pub fn init(&mut self) -> anyhow::Result<()> {
         self.vpercpu.hardware_enable();
         self.max_guest_page_table_levels = self.vpercpu.max_guest_page_table_levels();
         self.pa_range = self.vpercpu.pa_range();
+        self.pa_bits = self.vpercpu.pa_bits();
         Ok(())
     }
 
@@ -102,7 +105,6 @@ impl VCpu {
         let vcpu = arm_vcpu::Aarch64VCpu::new(Aarch64VCpuCreateConfig {
             mpidr_el1: hard_id.raw() as u64,
             dtb_addr: dtb_addr.as_usize(),
-            pt_level: 4,
         })
         .unwrap();
         Ok(VCpu { vcpu, common })
@@ -111,6 +113,10 @@ impl VCpu {
     pub fn set_pt_level(&mut self, level: usize) {
         self.vcpu.pt_level = level;
     }
+
+    pub fn set_pa_bits(&mut self, pa_bits: usize) {
+        self.vcpu.pa_bits = pa_bits;
+    }
 }
 
 impl VCpuOp for VCpu {

src/arch/aarch64/vm/unint.rs

@@ -15,6 +15,7 @@ pub struct VmMachineUninit {
     config: AxVMConfig,
     pt_levels: usize,
     pa_max: usize,
+    pa_bits: usize,
 }
 
 impl VmMachineUninitOps for VmMachineUninit {
@@ -25,6 +26,7 @@ impl VmMachineUninitOps for VmMachineUninit {
             config,
             pt_levels: 4,
             pa_max: usize::MAX,
+            pa_bits: 48,
         }
     }
 
@@ -63,23 +65,32 @@ impl VmMachineUninit {
         let vcpu_count = vcpus.len();
 
         for vcpu in &vcpus {
-            let (max_levels, max_pa) =
-                vcpu.with_hcpu(|cpu| (cpu.max_guest_page_table_levels(), cpu.pa_range.end));
+            let (max_levels, max_pa, pa_bits) = vcpu.with_hcpu(|cpu| {
+                (
+                    cpu.max_guest_page_table_levels(),
+                    cpu.pa_range.end,
+                    cpu.pa_bits,
+                )
+            });
             if max_levels < self.pt_levels {
                 self.pt_levels = max_levels;
             }
             if max_pa < self.pa_max {
                 self.pa_max = max_pa;
             }
+
+            if pa_bits < self.pa_bits {
+                self.pa_bits = pa_bits;
+            }
         }
 
         if self.pt_levels == 3 {
             self.pa_max = self.pa_max.min(0x8000000000);
         }
 
         debug!(
-            "VM {} ({}) vCPU count: {}, \n  Max Guest Page Table Levels: {}\n  Max PA: {:#x}",
-            self.config.id, self.config.name, vcpu_count, self.pt_levels, self.pa_max
+            "VM {} ({}) vCPU count: {}, \n  Max Guest Page Table Levels: {}\n  Max PA: {:#x}\n  PA Bits: {}",
+            self.config.id, self.config.name, vcpu_count, self.pt_levels, self.pa_max, self.pa_bits
         );
         Ok(vcpus)
     }
@@ -121,6 +132,7 @@ impl VmMachineUninit {
             vcpu.vcpu.set_entry(kernel_entry).unwrap();
             vcpu.vcpu.set_dtb_addr(dtb_addr).unwrap();
             vcpu.set_pt_level(self.pt_levels);
+            vcpu.set_pa_bits(self.pa_bits);
 
             let setup_config = Aarch64VCpuSetupConfig {
                 passthrough_interrupt: self.config.interrupt_mode()