1 use alloc::sync::Arc;
2
3 use crate::driver::base::device::{
4 bus::Bus,
5 driver::{driver_manager, Driver},
6 };
7
8 use system_error::SystemError;
9
10 use super::{platform_bus, platform_device::PlatformDevice};
11
12 /// @brief: 实现该trait的设备驱动实例应挂载在platform总线上，
13 /// 同时应该实现Driver trait
14 ///
15 /// ## 注意
16 ///
17 /// 应当在所有实现这个trait的结构体上方，添加 `#[cast_to([sync] PlatformDriver)]`，
18 /// 否则运行时将报错“该对象不是PlatformDriver”
19 #[allow(dead_code)]
20 pub trait PlatformDriver: Driver {
21 /// 检测设备是否能绑定到这个驱动
22 ///
23 /// 如果能，则把设备的driver字段指向这个驱动。
24 /// 请注意，这个函数不应该把driver加入驱动的devices列表，相关工作会在外部的函数里面处理。
probe(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>25 fn probe(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>;
remove(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>26 fn remove(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>;
shutdown(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>27 fn shutdown(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>;
suspend(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>28 fn suspend(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>;
resume(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>29 fn resume(&self, device: &Arc<dyn PlatformDevice>) -> Result<(), SystemError>;
30 }
31
32 #[inline(always)]
platform_driver_manager() -> &'static PlatformDriverManager33 pub fn platform_driver_manager() -> &'static PlatformDriverManager {
34 &PlatformDriverManager
35 }
36
37 #[derive(Debug)]
38 pub struct PlatformDriverManager;
39
40 impl PlatformDriverManager {
41 /// 注册平台设备驱动
42 ///
43 /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/platform.c?fi=__platform_driver_register#861
register(&self, driver: Arc<dyn PlatformDriver>) -> Result<(), SystemError>44 pub fn register(&self, driver: Arc<dyn PlatformDriver>) -> Result<(), SystemError> {
45 driver.set_bus(Some(Arc::downgrade(&(platform_bus() as Arc<dyn Bus>))));
46 return driver_manager().register(driver as Arc<dyn Driver>);
47 }
48
49 /// 卸载平台设备驱动
50 #[allow(dead_code)]
unregister(&self, driver: &Arc<dyn PlatformDriver>)51 pub fn unregister(&self, driver: &Arc<dyn PlatformDriver>) {
52 driver_manager().unregister(&(driver.clone() as Arc<dyn Driver>));
53 }
54 }
55