aml: subsystem

Signed-off-by: Andy-Python-Programmer <[email protected]>

Author: Andy-Python-Programmer

src/Cargo.lock

@@ -0,0 +1,29 @@
+use alloc::sync::Arc;
+use spin::Once;
+
+/// ## Reference
+/// * [ACPI Sleeping States](https://uefi.org/specs/ACPI/6.4/16_Waking_and_Sleeping/sleeping-states.html)
+#[repr(u8)]
+pub enum SleepState {
+    S5 = 5,
+}
+
+pub trait AmlSubsystem: Send + Sync {
+    fn enter_state(&self, state: SleepState);
+}
+
+static AML_SUBSYSTEM: Once<Arc<dyn AmlSubsystem>> = Once::new();
+
+pub fn get_subsystem() -> Arc<dyn AmlSubsystem> {
+    AML_SUBSYSTEM.get().unwrap().clone()
+}
+
+pub fn init(subsystem: Arc<dyn AmlSubsystem>) {
+    assert!(
+        AML_SUBSYSTEM.get().is_none(),
+        "aml: subsystem already initialized"
+    );
+
+    AML_SUBSYSTEM.call_once(|| subsystem);
+    log::debug!("aml: subsystem initialized");
+}

src/aero_kernel/src/acpi/aml.rs

@@ -31,6 +31,7 @@ use crate::{
 
 use self::{hpet::Hpet, madt::Madt, mcfg::Mcfg, sdt::Sdt};
 
+pub mod aml;
 pub mod fadt;
 pub mod hpet;
 pub mod madt;

src/aero_kernel/src/acpi/mod.rs

@@ -1,5 +1,6 @@
 use alloc::sync::Arc;
 
+use crate::acpi::aml;
 use crate::acpi::fadt;
 use crate::acpi::get_acpi_table;
 
@@ -98,6 +99,14 @@ impl lai::Host for LaiHost {
     }
 }
 
+struct LaiSubsystem;
+
+impl aml::AmlSubsystem for LaiSubsystem {
+    fn enter_state(&self, state: aml::SleepState) {
+        lai::enter_sleep(state as u8)
+    }
+}
+
 pub fn init_lai() {
     let lai_host = Arc::new(LaiHost);
     lai::init(lai_host);
@@ -106,6 +115,9 @@ pub fn init_lai() {
     lai::create_namespace();
 
     lai::enable_acpi(1);
+
+    let subsystem = Arc::new(LaiSubsystem);
+    aml::init(subsystem);
 }
 
 crate::module_init!(init_lai);

src/aero_kernel/src/drivers/lai.rs

@@ -22,6 +22,7 @@ use aero_syscall::{AeroSyscallError, MMapFlags, MMapProt, SysInfo, Utsname};
 use alloc::string::String;
 use spin::{Mutex, Once};
 
+use crate::acpi::aml;
 use crate::fs;
 use crate::fs::Path;
 
@@ -311,7 +312,7 @@ pub fn shutdown() -> ! {
     fs::cache::clear_dir_cache();
 
     let _guard = IrqGuard::new();
-    lai::enter_sleep(5);
+    aml::get_subsystem().enter_state(aml::SleepState::S5);
 
     unreachable!("aml: failed to shutdown (enter state S5)")
 }