xref: /DragonOS/kernel/src/driver/base/device/bus.rs (revision 2eab6dd743e94a86a685f1f3c01e599adf86610a)
1 use super::{
2     driver::{Driver, DriverMatchName, DriverMatcher},
3     sys_devices_kset, Device, DeviceMatchName, DeviceMatcher, DeviceState,
4 };
5 use crate::{
6     driver::base::{
7         device::{device_manager, driver::driver_manager},
8         kobject::{KObjType, KObject, KObjectManager},
9         kset::KSet,
10         subsys::SubSysPrivate,
11     },
12     filesystem::{
13         sysfs::{
14             file::sysfs_emit_str, sysfs_instance, Attribute, AttributeGroup, SysFSOps,
15             SysFSOpsSupport, SYSFS_ATTR_MODE_RW, SYSFS_ATTR_MODE_WO,
16         },
17         vfs::syscall::ModeType,
18     },
19     libs::rwlock::RwLock,
20 };
21 use alloc::{
22     string::{String, ToString},
23     sync::{Arc, Weak},
24 };
25 use core::{ffi::CStr, fmt::Debug, intrinsics::unlikely};
26 use hashbrown::HashMap;
27 use intertrait::cast::CastArc;
28 use log::{debug, error, info};
29 use system_error::SystemError;
30 
31 /// `/sys/bus`的kset
32 static mut BUS_KSET_INSTANCE: Option<Arc<KSet>> = None;
33 /// `/sys/devices/system`的kset
34 static mut DEVICES_SYSTEM_KSET_INSTANCE: Option<Arc<KSet>> = None;
35 
36 static mut BUS_MANAGER_INSTANCE: Option<BusManager> = None;
37 
38 #[inline(always)]
39 pub fn sys_bus_kset() -> Arc<KSet> {
40     unsafe { BUS_KSET_INSTANCE.clone().unwrap() }
41 }
42 
43 #[inline(always)]
44 #[allow(dead_code)]
45 pub fn sys_devices_system_kset() -> Arc<KSet> {
46     unsafe { DEVICES_SYSTEM_KSET_INSTANCE.clone().unwrap() }
47 }
48 
49 #[inline(always)]
50 pub fn bus_manager() -> &'static BusManager {
51     unsafe { BUS_MANAGER_INSTANCE.as_ref().unwrap() }
52 }
53 
54 #[inline(always)]
55 pub fn subsystem_manager() -> &'static SubSystemManager {
56     &SubSystemManager
57 }
58 
59 /// @brief: 总线状态
60 #[derive(Debug, Copy, Clone)]
61 pub enum BusState {
62     NotInitialized = 0, // 未初始化
63     Initialized = 1,    // 已初始化
64     UnDefined = 2,      // 未定义的
65 }
66 
67 /// @brief: 将u32类型转换为总线状态类型
68 impl From<u32> for BusState {
69     fn from(state: u32) -> Self {
70         match state {
71             0 => BusState::NotInitialized,
72             1 => BusState::Initialized,
73             _ => BusState::UnDefined,
74         }
75     }
76 }
77 
78 /// @brief: 将总线状态类型转换为u32类型
79 impl From<DeviceState> for BusState {
80     fn from(state: DeviceState) -> Self {
81         match state {
82             DeviceState::Initialized => BusState::Initialized,
83             DeviceState::NotInitialized => BusState::NotInitialized,
84             DeviceState::UnDefined => BusState::UnDefined,
85         }
86     }
87 }
88 
89 /// @brief: 将总线状态类型转换为设备状态类型
90 impl From<BusState> for DeviceState {
91     fn from(state: BusState) -> Self {
92         match state {
93             BusState::Initialized => DeviceState::Initialized,
94             BusState::NotInitialized => DeviceState::NotInitialized,
95             BusState::UnDefined => DeviceState::UnDefined,
96         }
97     }
98 }
99 
100 /// 总线子系统的trait,所有总线都应实现该trait
101 ///
102 /// 请注意,这个trait是用于实现总线子系统的,而不是总线驱动/总线设备。
103 /// https://code.dragonos.org.cn/xref/linux-6.1.9/include/linux/device/bus.h#84
104 pub trait Bus: Debug + Send + Sync {
105     fn name(&self) -> String;
106     /// Used for subsystems to enumerate devices like ("foo%u", dev->id).
107     fn dev_name(&self) -> String;
108     fn root_device(&self) -> Option<Weak<dyn Device>> {
109         None
110     }
111 
112     fn set_root_device(&self, _dev: Option<Weak<dyn Device>>) {}
113 
114     /// 总线上的设备的默认属性组
115     fn dev_groups(&self) -> &'static [&'static dyn AttributeGroup] {
116         &[]
117     }
118 
119     /// 总线的默认属性组
120     fn bus_groups(&self) -> &'static [&'static dyn AttributeGroup] {
121         &[]
122     }
123 
124     /// 总线上的驱动的默认属性组
125     fn drv_groups(&self) -> &'static [&'static dyn AttributeGroup] {
126         &[]
127     }
128 
129     /// 检查设备是否可以被总线绑定,如果可以,就绑定它们。
130     /// 绑定之后,device的driver字段会被设置为驱动实例。
131     ///
132     /// ## 参数
133     ///
134     /// - `device` - 设备实例
135     ///
136     /// ## 默认实现
137     ///
138     /// 如果总线不支持该操作,返回`SystemError::ENOSYS`
139     fn probe(&self, _device: &Arc<dyn Device>) -> Result<(), SystemError> {
140         return Err(SystemError::ENOSYS);
141     }
142     fn remove(&self, _device: &Arc<dyn Device>) -> Result<(), SystemError>;
143     fn sync_state(&self, _device: &Arc<dyn Device>) {}
144     fn shutdown(&self, _device: &Arc<dyn Device>);
145     fn suspend(&self, _device: &Arc<dyn Device>) {
146         // todo: implement suspend
147     }
148 
149     fn resume(&self, device: &Arc<dyn Device>) -> Result<(), SystemError>;
150 
151     /// match device to driver.
152     ///
153     /// ## 参数
154     ///
155     /// * `device` - device
156     /// * `driver` - driver
157     ///
158     /// ## 返回
159     ///
160     /// - `Ok(true)` - 匹配成功
161     /// - `Ok(false)` - 匹配失败
162     /// - `Err(_)` - 由于内部错误导致匹配失败
163     /// - `Err(SystemError::ENOSYS)` - 该总线不支持该操作
164     fn match_device(
165         &self,
166         _device: &Arc<dyn Device>,
167         _driver: &Arc<dyn Driver>,
168     ) -> Result<bool, SystemError> {
169         return Err(SystemError::ENOSYS);
170     }
171 
172     fn subsystem(&self) -> &SubSysPrivate;
173 
174     /// 对当前总线操作的时候需要获取父级总线的锁
175     fn need_parent_lock(&self) -> bool {
176         false
177     }
178 }
179 
180 impl dyn Bus {
181     /// 在bus上,根据条件寻找一个特定的设备
182     ///
183     /// ## 参数
184     ///
185     /// - `matcher` - 匹配器
186     /// - `data` - 传给匹配器的数据
187     pub fn find_device<T: Copy>(
188         &self,
189         matcher: &dyn DeviceMatcher<T>,
190         data: T,
191     ) -> Option<Arc<dyn Device>> {
192         let subsys = self.subsystem();
193         let guard = subsys.devices();
194         for dev in guard.iter() {
195             if matcher.match_device(dev, data) {
196                 return Some(dev.clone());
197             }
198         }
199         return None;
200     }
201 
202     /// 根据名称匹配设备
203     ///
204     /// ## 参数
205     ///
206     /// - name 设备名称
207     pub fn find_device_by_name(&self, name: &str) -> Option<Arc<dyn Device>> {
208         return self.find_device(&DeviceMatchName, name);
209     }
210 
211     /// 在bus上,根据条件寻找一个特定的驱动
212     ///
213     /// ## 参数
214     ///
215     /// - `matcher` - 匹配器
216     /// - `data` - 传给匹配器的数据
217     pub fn find_driver<T: Copy>(
218         &self,
219         matcher: &dyn DriverMatcher<T>,
220         data: T,
221     ) -> Option<Arc<dyn Driver>> {
222         let subsys = self.subsystem();
223         let guard = subsys.drivers();
224         for drv in guard.iter() {
225             if matcher.match_driver(drv, data) {
226                 return Some(drv.clone());
227             }
228         }
229         return None;
230     }
231 
232     /// 根据名称在bus上匹配驱动
233     pub fn find_driver_by_name(&self, name: &str) -> Option<Arc<dyn Driver>> {
234         return self.find_driver(&DriverMatchName, name);
235     }
236 }
237 
238 /// @brief: 总线管理结构体
239 #[derive(Debug)]
240 pub struct BusManager {
241     /// 存储总线bus的kset结构体与bus实例的映射(用于在sysfs callback的时候,根据kset找到bus实例)
242     kset_bus_map: RwLock<HashMap<Arc<KSet>, Arc<dyn Bus>>>,
243 }
244 
245 impl BusManager {
246     pub fn new() -> Self {
247         return Self {
248             kset_bus_map: RwLock::new(HashMap::new()),
249         };
250     }
251 
252     /// 把一个设备添加到总线上
253     ///
254     /// ## 描述
255     ///
256     /// - 添加一个设备的与bus相关的属性
257     /// - 在bus和设备文件夹下,创建软链接
258     /// - 把设备添加到它的总线的设备列表中
259     ///
260     /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_add_device#441
261     ///
262     /// ## 参数
263     ///
264     /// - `dev` - 要被添加的设备
265     pub fn add_device(&self, dev: &Arc<dyn Device>) -> Result<(), SystemError> {
266         let bus = dev.bus().and_then(|bus| bus.upgrade());
267         if let Some(bus) = bus {
268             device_manager().add_groups(dev, bus.dev_groups())?;
269 
270             // 增加符号链接
271             let bus_devices_kset = bus
272                 .subsystem()
273                 .devices_kset()
274                 .expect("bus devices kset is none, maybe bus is not registered");
275             let dev_kobj = dev.clone() as Arc<dyn KObject>;
276 
277             sysfs_instance().create_link(
278                 Some(&bus_devices_kset.as_kobject()),
279                 &dev_kobj,
280                 dev.name(),
281             )?;
282             sysfs_instance().create_link(
283                 Some(&dev_kobj),
284                 &bus.subsystem().subsys().as_kobject(),
285                 "subsystem".to_string(),
286             )?;
287             bus.subsystem().add_device_to_vec(dev)?;
288         }
289         return Ok(());
290     }
291 
292     /// 在总线上添加一个驱动
293     ///
294     /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_add_driver#590
295     pub fn add_driver(&self, driver: &Arc<dyn Driver>) -> Result<(), SystemError> {
296         let bus = driver
297             .bus()
298             .and_then(|bus| bus.upgrade())
299             .ok_or(SystemError::EINVAL)?;
300         debug!("bus '{}' add driver '{}'", bus.name(), driver.name());
301 
302         driver.set_kobj_type(Some(&BusDriverKType));
303         let kobj = driver.clone() as Arc<dyn KObject>;
304         KObjectManager::add_kobj(kobj, bus.subsystem().drivers_kset())?;
305 
306         bus.subsystem().add_driver_to_vec(driver)?;
307         if bus.subsystem().drivers_autoprobe() {
308             let r = driver_manager().driver_attach(driver);
309             if let Err(e) = r {
310                 bus.subsystem().remove_driver_from_vec(driver);
311                 return Err(e);
312             }
313         }
314 
315         driver_manager()
316             .add_groups(driver, bus.drv_groups())
317             .map_err(|e| {
318                 error!(
319                     "BusManager::add_driver: driver '{:?}' add_groups failed, err: '{:?}",
320                     driver.name(),
321                     e
322                 );
323                 e
324             })
325             .ok();
326 
327         if !driver.suppress_bind_attrs() {
328             self.add_bind_files(driver)
329                 .map_err(|e| {
330                     error!(
331                         "BusManager::add_driver: driver '{:?}' add_bind_files failed, err: '{:?}",
332                         driver.name(),
333                         e
334                     );
335                     e
336                 })
337                 .ok();
338         }
339 
340         return Ok(());
341     }
342 
343     ///
344     /// bus_register - register a driver-core subsystem
345     ///
346     /// ## 参数
347     /// - `bus` - bus to register
348     ///
349     /// Once we have that, we register the bus with the kobject
350     /// infrastructure, then register the children subsystems it has:
351     /// the devices and drivers that belong to the subsystem.
352     ///
353     /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_register#783
354     ///
355     /// todo: 增加错误处理逻辑
356     pub fn register(&self, bus: Arc<dyn Bus>) -> Result<(), SystemError> {
357         bus.subsystem().set_bus(Some(Arc::downgrade(&bus)));
358 
359         let subsys_kset = bus.subsystem().subsys();
360         subsys_kset.set_name(bus.name());
361         bus.subsystem().set_drivers_autoprobe(true);
362 
363         subsys_kset.register(Some(sys_bus_kset()))?;
364 
365         let devices_kset =
366             KSet::new_and_add("devices".to_string(), None, Some(subsys_kset.clone()))?;
367         bus.subsystem().set_devices_kset(devices_kset);
368         let drivers_kset =
369             KSet::new_and_add("drivers".to_string(), None, Some(subsys_kset.clone()))?;
370         bus.subsystem().set_drivers_kset(drivers_kset);
371 
372         self.add_probe_files(&bus)?;
373         let bus_groups = bus.bus_groups();
374         self.add_groups(&bus, bus_groups)?;
375         // 把bus实例添加到总线管理器中(方便在sysfs callback的时候,根据kset找到bus实例)
376         self.kset_bus_map.write().insert(subsys_kset, bus.clone());
377         return Ok(());
378     }
379 
380     pub fn unregister(&self, _bus: Arc<dyn Bus>) -> Result<(), SystemError> {
381         todo!("bus_unregister")
382     }
383 
384     fn add_probe_files(&self, bus: &Arc<dyn Bus>) -> Result<(), SystemError> {
385         self.create_file(bus, &BusAttrDriversProbe)?;
386         let r = self.create_file(bus, &BusAttrDriversAutoprobe);
387 
388         if r.is_err() {
389             self.remove_file(bus, &BusAttrDriversProbe);
390         }
391         return r;
392     }
393 
394     #[allow(dead_code)]
395     fn remove_probe_files(&self, bus: &Arc<dyn Bus>) {
396         self.remove_file(bus, &BusAttrDriversAutoprobe);
397         self.remove_file(bus, &BusAttrDriversProbe);
398     }
399 
400     fn create_file(
401         &self,
402         bus: &Arc<dyn Bus>,
403         attr: &'static dyn Attribute,
404     ) -> Result<(), SystemError> {
405         let bus_kobj = bus.subsystem().subsys() as Arc<dyn KObject>;
406         return sysfs_instance().create_file(&bus_kobj, attr);
407     }
408 
409     fn remove_file(&self, bus: &Arc<dyn Bus>, attr: &'static dyn Attribute) {
410         let bus_kobj = bus.subsystem().subsys() as Arc<dyn KObject>;
411         sysfs_instance().remove_file(&bus_kobj, attr);
412     }
413 
414     #[inline]
415     fn add_groups(
416         &self,
417         bus: &Arc<dyn Bus>,
418         groups: &[&'static dyn AttributeGroup],
419     ) -> Result<(), SystemError> {
420         let bus_kobj = bus.subsystem().subsys() as Arc<dyn KObject>;
421         return sysfs_instance().create_groups(&bus_kobj, groups);
422     }
423 
424     /// 根据bus的kset找到bus实例
425     fn get_bus_by_kset(&self, kset: &Arc<KSet>) -> Option<Arc<dyn Bus>> {
426         return self.kset_bus_map.read().get(kset).cloned();
427     }
428 
429     /// 为bus上的设备选择可能的驱动程序
430     ///
431     /// 这个函数会扫描总线上的所有没有驱动的设备,然后为它们选择可能的驱动程序。
432     ///
433     /// ## 参数
434     ///
435     /// - `bus` - bus实例
436     #[allow(dead_code)]
437     pub fn rescan_devices(&self, bus: &Arc<dyn Bus>) -> Result<(), SystemError> {
438         for dev in bus.subsystem().devices().iter() {
439             rescan_devices_helper(dev)?;
440         }
441         return Ok(());
442     }
443 
444     /// 为新设备探测驱动
445     ///
446     /// Automatically probe for a driver if the bus allows it.
447     pub fn probe_device(&self, dev: &Arc<dyn Device>) {
448         let bus = dev.bus().and_then(|bus| bus.upgrade());
449         if bus.is_none() {
450             return;
451         }
452         let bus = bus.unwrap();
453         if bus.subsystem().drivers_autoprobe() {
454             device_manager().device_initial_probe(dev).ok();
455         }
456         for interface in bus.subsystem().interfaces() {
457             interface.add_device(dev).ok();
458         }
459     }
460 
461     /// 从总线上移除一个驱动
462     ///
463     /// Detach the driver from the devices it controls, and remove
464     /// it from its bus's list of drivers. Finally, we drop the reference
465     /// to the bus.
466     ///
467     /// ## 参数
468     ///
469     /// - `driver` - 驱动实例
470     ///
471     /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_remove_driver#666
472     pub fn remove_driver(&self, _driver: &Arc<dyn Driver>) {
473         todo!("BusManager::remove_driver")
474     }
475 
476     fn add_bind_files(&self, driver: &Arc<dyn Driver>) -> Result<(), SystemError> {
477         driver_manager().create_attr_file(driver, &DriverAttrUnbind)?;
478 
479         driver_manager()
480             .create_attr_file(driver, &DriverAttrBind)
481             .map_err(|e| {
482                 driver_manager().remove_attr_file(driver, &DriverAttrUnbind);
483                 e
484             })?;
485 
486         return Ok(());
487     }
488 }
489 
490 /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?r=&mo=5649&fi=241#684
491 fn rescan_devices_helper(dev: &Arc<dyn Device>) -> Result<(), SystemError> {
492     if dev.driver().is_none() {
493         let need_parent_lock = dev
494             .bus()
495             .map(|bus| bus.upgrade().unwrap().need_parent_lock())
496             .unwrap_or(false);
497         if unlikely(need_parent_lock) {
498             // todo: lock device parent
499             unimplemented!()
500         }
501         device_manager().device_attach(dev)?;
502     }
503     return Ok(());
504 }
505 
506 ///
507 /// bus_register - register a driver-core subsystem
508 ///
509 /// ## 参数
510 /// - `bus` - bus to register
511 ///
512 /// Once we have that, we register the bus with the kobject
513 /// infrastructure, then register the children subsystems it has:
514 /// the devices and drivers that belong to the subsystem.
515 pub fn bus_register(bus: Arc<dyn Bus>) -> Result<(), SystemError> {
516     return bus_manager().register(bus);
517 }
518 
519 /// @brief: 总线注销,并在sys/bussys/devices下删除文件夹
520 /// @parameter bus: Bus设备实体
521 /// @return: 成功:()   失败:SystemError
522 #[allow(dead_code)]
523 pub fn bus_unregister(bus: Arc<dyn Bus>) -> Result<(), SystemError> {
524     return bus_manager().unregister(bus);
525 }
526 
527 pub fn buses_init() -> Result<(), SystemError> {
528     let bus_kset = KSet::new("bus".to_string());
529     bus_kset.register(None).expect("bus kset register failed");
530     unsafe {
531         BUS_KSET_INSTANCE = Some(bus_kset);
532     }
533 
534     // 初始化 /sys/devices/system
535     {
536         let devices_system_kset = KSet::new("system".to_string());
537         let parent = sys_devices_kset() as Arc<dyn KObject>;
538         devices_system_kset.set_parent(Some(Arc::downgrade(&parent)));
539         devices_system_kset
540             .register(Some(sys_devices_kset()))
541             .expect("devices system kset register failed");
542         unsafe {
543             DEVICES_SYSTEM_KSET_INSTANCE = Some(devices_system_kset);
544         }
545     }
546 
547     // 初始化总线管理器
548     {
549         let bus_manager = BusManager::new();
550         unsafe {
551             BUS_MANAGER_INSTANCE = Some(bus_manager);
552         }
553     }
554     return Ok(());
555 }
556 
557 /// 把一个设备添加到总线上
558 ///
559 /// ## 描述
560 ///
561 /// - 添加一个设备的与bus相关的属性
562 /// - 在bus和设备文件夹下,创建软链接
563 /// - 把设备添加到它的总线的设备列表中
564 ///
565 /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_add_device#441
566 ///
567 /// ## 参数
568 ///
569 /// - `dev` - 要被添加的设备
570 pub fn bus_add_device(dev: &Arc<dyn Device>) -> Result<(), SystemError> {
571     return bus_manager().add_device(dev);
572 }
573 
574 /// 自动为设备在总线上寻找可用的驱动程序
575 ///
576 /// Automatically probe for a driver if the bus allows it.
577 ///
578 /// ## 参数
579 ///
580 /// - `dev` - 要被添加的设备
581 ///
582 /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=bus_probe_device#478
583 pub fn bus_probe_device(dev: &Arc<dyn Device>) {
584     info!("bus_probe_device: dev: {:?}", dev.name());
585     bus_manager().probe_device(dev);
586 }
587 
588 #[derive(Debug)]
589 struct BusAttrDriversProbe;
590 
591 impl Attribute for BusAttrDriversProbe {
592     fn mode(&self) -> ModeType {
593         return ModeType::S_IWUSR;
594     }
595 
596     fn name(&self) -> &str {
597         return "drivers_probe";
598     }
599 
600     fn support(&self) -> SysFSOpsSupport {
601         return SysFSOpsSupport::ATTR_STORE;
602     }
603 
604     /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?r=&mo=5649&fi=241#241
605     fn store(&self, kobj: Arc<dyn KObject>, buf: &[u8]) -> Result<usize, SystemError> {
606         let kset: Arc<KSet> = kobj.arc_any().downcast().map_err(|_| SystemError::EINVAL)?;
607         let bus = bus_manager()
608             .get_bus_by_kset(&kset)
609             .ok_or(SystemError::EINVAL)?;
610 
611         let name = CStr::from_bytes_with_nul(buf)
612             .map_err(|_| SystemError::EINVAL)?
613             .to_str()
614             .map_err(|_| SystemError::EINVAL)?;
615 
616         let device = bus.find_device_by_name(name).ok_or(SystemError::ENODEV)?;
617 
618         if rescan_devices_helper(&device).is_ok() {
619             return Ok(buf.len());
620         }
621 
622         return Err(SystemError::EINVAL);
623     }
624 }
625 
626 #[derive(Debug)]
627 struct BusAttrDriversAutoprobe;
628 
629 impl Attribute for BusAttrDriversAutoprobe {
630     fn mode(&self) -> ModeType {
631         SYSFS_ATTR_MODE_RW
632     }
633 
634     fn name(&self) -> &str {
635         return "drivers_autoprobe";
636     }
637 
638     fn support(&self) -> SysFSOpsSupport {
639         return SysFSOpsSupport::ATTR_STORE | SysFSOpsSupport::ATTR_SHOW;
640     }
641 
642     /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?r=&mo=5649&fi=241#231
643     fn store(&self, kobj: Arc<dyn KObject>, buf: &[u8]) -> Result<usize, SystemError> {
644         if buf.is_empty() {
645             return Ok(0);
646         }
647 
648         let kset: Arc<KSet> = kobj.arc_any().downcast().map_err(|_| SystemError::EINVAL)?;
649         let bus = bus_manager()
650             .get_bus_by_kset(&kset)
651             .ok_or(SystemError::EINVAL)?;
652 
653         if buf[0] == b'0' {
654             bus.subsystem().set_drivers_autoprobe(false);
655         } else {
656             bus.subsystem().set_drivers_autoprobe(true);
657         }
658 
659         return Ok(buf.len());
660     }
661 
662     /// 参考: https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?r=&mo=5649&fi=241#226
663     fn show(&self, kobj: Arc<dyn KObject>, buf: &mut [u8]) -> Result<usize, SystemError> {
664         let kset: Arc<KSet> = kobj.arc_any().downcast().map_err(|_| SystemError::EINVAL)?;
665         let bus = bus_manager()
666             .get_bus_by_kset(&kset)
667             .ok_or(SystemError::EINVAL)?;
668         let val = if bus.subsystem().drivers_autoprobe() {
669             1
670         } else {
671             0
672         };
673         return sysfs_emit_str(buf, format!("{val}\n").as_str());
674     }
675 }
676 
677 #[allow(dead_code)]
678 #[derive(Debug, Clone, Copy)]
679 pub enum BusNotifyEvent {
680     /// 一个设备被添加到总线上
681     AddDevice,
682     /// 一个设备将要被移除
683     DelDevice,
684     /// 一个设备已经被移除
685     RemovedDevice,
686     /// 一个驱动将要被绑定
687     BindDriver,
688     /// 一个驱动已经被绑定
689     BoundDriver,
690     /// 一个驱动将要被解绑
691     UnbindDriver,
692     /// 一个驱动已经被解绑
693     UnboundDriver,
694     /// 驱动绑定失败
695     DriverNotBound,
696 }
697 
698 #[derive(Debug)]
699 struct BusDriverKType;
700 
701 impl KObjType for BusDriverKType {
702     fn sysfs_ops(&self) -> Option<&dyn SysFSOps> {
703         Some(&BusDriverSysFSOps)
704     }
705 
706     fn attribute_groups(&self) -> Option<&'static [&'static dyn AttributeGroup]> {
707         None
708     }
709 }
710 
711 #[derive(Debug)]
712 struct BusDriverSysFSOps;
713 
714 impl SysFSOps for BusDriverSysFSOps {
715     #[inline]
716     fn show(
717         &self,
718         kobj: Arc<dyn KObject>,
719         attr: &dyn Attribute,
720         buf: &mut [u8],
721     ) -> Result<usize, SystemError> {
722         attr.show(kobj, buf)
723     }
724 
725     #[inline]
726     fn store(
727         &self,
728         kobj: Arc<dyn KObject>,
729         attr: &dyn Attribute,
730         buf: &[u8],
731     ) -> Result<usize, SystemError> {
732         attr.store(kobj, buf)
733     }
734 }
735 
736 #[derive(Debug)]
737 struct DriverAttrUnbind;
738 
739 impl Attribute for DriverAttrUnbind {
740     fn mode(&self) -> ModeType {
741         SYSFS_ATTR_MODE_WO
742     }
743 
744     fn name(&self) -> &str {
745         "unbind"
746     }
747 
748     fn store(&self, kobj: Arc<dyn KObject>, buf: &[u8]) -> Result<usize, SystemError> {
749         let driver = kobj.cast::<dyn Driver>().map_err(|kobj| {
750             error!(
751                 "Intertrait casting not implemented for kobj: {}",
752                 kobj.name()
753             );
754             SystemError::ENOSYS
755         })?;
756 
757         let bus = driver
758             .bus()
759             .and_then(|bus| bus.upgrade())
760             .ok_or(SystemError::ENODEV)?;
761 
762         let s = CStr::from_bytes_with_nul(buf)
763             .map_err(|_| SystemError::EINVAL)?
764             .to_str()
765             .map_err(|_| SystemError::EINVAL)?;
766         let dev = bus.find_device_by_name(s).ok_or(SystemError::ENODEV)?;
767         let p = dev.driver().ok_or(SystemError::ENODEV)?;
768         if Arc::ptr_eq(&p, &driver) {
769             device_manager().device_driver_detach(&dev);
770             return Ok(buf.len());
771         }
772         return Err(SystemError::ENODEV);
773     }
774 
775     fn support(&self) -> SysFSOpsSupport {
776         SysFSOpsSupport::ATTR_STORE
777     }
778 }
779 
780 #[derive(Debug)]
781 struct DriverAttrBind;
782 
783 impl Attribute for DriverAttrBind {
784     fn name(&self) -> &str {
785         "bind"
786     }
787 
788     fn mode(&self) -> ModeType {
789         SYSFS_ATTR_MODE_WO
790     }
791 
792     /*
793      * Manually attach a device to a driver.
794      * Note: the driver must want to bind to the device,
795      * it is not possible to override the driver's id table.
796      */
797     fn store(&self, kobj: Arc<dyn KObject>, buf: &[u8]) -> Result<usize, SystemError> {
798         let driver = kobj.cast::<dyn Driver>().map_err(|kobj| {
799             error!(
800                 "Intertrait casting not implemented for kobj: {}",
801                 kobj.name()
802             );
803             SystemError::ENOSYS
804         })?;
805 
806         let bus = driver
807             .bus()
808             .and_then(|bus| bus.upgrade())
809             .ok_or(SystemError::ENODEV)?;
810         let device = bus
811             .find_device_by_name(
812                 CStr::from_bytes_with_nul(buf)
813                     .map_err(|_| SystemError::EINVAL)?
814                     .to_str()
815                     .map_err(|_| SystemError::EINVAL)?,
816             )
817             .ok_or(SystemError::ENODEV)?;
818 
819         if driver_manager().match_device(&driver, &device)? {
820             device_manager().device_driver_attach(&driver, &device)?;
821             return Ok(buf.len());
822         }
823         return Err(SystemError::ENODEV);
824     }
825     fn support(&self) -> SysFSOpsSupport {
826         SysFSOpsSupport::ATTR_STORE
827     }
828 }
829 
830 #[derive(Debug)]
831 pub struct SubSystemManager;
832 
833 impl SubSystemManager {
834     /// 注册一个子系统,并在`/sys/bus`和指定的父级文件夹下创建子文件夹
835     ///
836     /// ## 参数
837     ///
838     /// - `subsys` - 子系统实例
839     /// - `fake_root_dev` - 该子系统的伪根设备
840     /// - `parent_of_root` - 该子系统的伪根设备的父级节点
841     ///
842     /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/bus.c?fi=subsys_system_register#1078
843     pub fn subsys_register(
844         &self,
845         subsys: &Arc<dyn Bus>,
846         fake_root_dev: &Arc<dyn Device>,
847         parent_of_root: &Arc<dyn KObject>,
848     ) -> Result<(), SystemError> {
849         bus_manager().register(subsys.clone())?;
850         fake_root_dev.set_name(subsys.name());
851         fake_root_dev.set_parent(Some(Arc::downgrade(parent_of_root)));
852 
853         device_manager().register(fake_root_dev.clone())?;
854 
855         subsys.set_root_device(Some(Arc::downgrade(fake_root_dev)));
856         return Ok(());
857     }
858 
859     /// register a subsystem at /sys/devices/system/
860     /// 并且在/sys/bus和/sys/devices下创建文件夹
861     ///
862     /// All 'system' subsystems have a /sys/devices/system/<name> root device
863     /// with the name of the subsystem. The root device can carry subsystem-
864     /// wide attributes. All registered devices are below this single root
865     /// device and are named after the subsystem with a simple enumeration
866     /// number appended. The registered devices are not explicitly named;
867     /// only 'id' in the device needs to be set.
868     pub fn subsys_system_register(
869         &self,
870         subsys: &Arc<dyn Bus>,
871         fake_root_dev: &Arc<dyn Device>,
872     ) -> Result<(), SystemError> {
873         return self.subsys_register(
874             subsys,
875             fake_root_dev,
876             &(sys_devices_system_kset() as Arc<dyn KObject>),
877         );
878     }
879 }
880