106d5e247SLoGin use core::intrinsics::unlikely; 206d5e247SLoGin 3a03c4f9dSLoGin use alloc::{string::ToString, sync::Arc}; 4a03c4f9dSLoGin use intertrait::cast::CastArc; 506d5e247SLoGin 6a03c4f9dSLoGin use crate::{ 7a03c4f9dSLoGin driver::base::kobject::KObject, 8a03c4f9dSLoGin filesystem::{ 9a03c4f9dSLoGin sysfs::{file::sysfs_emit_str, sysfs_instance, Attribute, SysFSOpsSupport}, 10a03c4f9dSLoGin vfs::syscall::ModeType, 11a03c4f9dSLoGin }, 12a03c4f9dSLoGin libs::wait_queue::WaitQueue, 13a03c4f9dSLoGin }; 1491e9d4abSLoGin use system_error::SystemError; 1506d5e247SLoGin 16a03c4f9dSLoGin use super::{ 17a03c4f9dSLoGin bus::BusNotifyEvent, 18a03c4f9dSLoGin device_manager, 19a03c4f9dSLoGin driver::{driver_manager, Driver, DriverManager}, 20a03c4f9dSLoGin Device, DeviceManager, 21a03c4f9dSLoGin }; 22a03c4f9dSLoGin 23a03c4f9dSLoGin static PROBE_WAIT_QUEUE: WaitQueue = WaitQueue::INIT; 2406d5e247SLoGin 2506d5e247SLoGin impl DeviceManager { 2606d5e247SLoGin /// 尝试把一个设备与一个驱动匹配 2706d5e247SLoGin /// 2806d5e247SLoGin /// 当前函数会遍历整个bus的驱动列表,并且尝试把设备与每一个驱动进行匹配。 2906d5e247SLoGin /// 一旦有一个驱动匹配成功,就会返回。 3006d5e247SLoGin /// 3106d5e247SLoGin /// ## 参数 3206d5e247SLoGin /// 3306d5e247SLoGin /// - `dev`: 设备 3406d5e247SLoGin /// 3506d5e247SLoGin /// ## 返回 3606d5e247SLoGin /// 3706d5e247SLoGin /// - Ok(true): 匹配成功 3806d5e247SLoGin /// - Ok(false): 没有匹配成功 3906d5e247SLoGin /// - Err(SystemError::ENODEV): 设备还没被注册 4006d5e247SLoGin /// 4106d5e247SLoGin /// ## 参考 4206d5e247SLoGin /// 43*e7071df6SLoGin /// https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#1049 4406d5e247SLoGin pub fn device_attach(&self, dev: &Arc<dyn Device>) -> Result<bool, SystemError> { 4506d5e247SLoGin return self.do_device_attach(dev, false); 4606d5e247SLoGin } 4706d5e247SLoGin 4806d5e247SLoGin pub fn device_initial_probe(&self, dev: &Arc<dyn Device>) -> Result<bool, SystemError> { 4906d5e247SLoGin return self.do_device_attach(dev, true); 5006d5e247SLoGin } 5106d5e247SLoGin 52*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#978 5306d5e247SLoGin fn do_device_attach( 5406d5e247SLoGin &self, 5506d5e247SLoGin dev: &Arc<dyn Device>, 5606d5e247SLoGin allow_async: bool, 5706d5e247SLoGin ) -> Result<bool, SystemError> { 5806d5e247SLoGin if unlikely(allow_async) { 59a03c4f9dSLoGin // todo!("do_device_attach: allow_async") 60a03c4f9dSLoGin kwarn!("do_device_attach: allow_async is true, but currently not supported"); 6106d5e247SLoGin } 6206d5e247SLoGin if dev.is_dead() { 6306d5e247SLoGin return Ok(false); 6406d5e247SLoGin } 6506d5e247SLoGin 66c566df45SLoGin kwarn!("do_device_attach: dev: '{}'", dev.name()); 67c566df45SLoGin 6806d5e247SLoGin let mut do_async = false; 6906d5e247SLoGin let mut r = Ok(false); 7006d5e247SLoGin 7106d5e247SLoGin if dev.driver().is_some() { 7206d5e247SLoGin if self.device_is_bound(dev) { 7306d5e247SLoGin return Ok(true); 7406d5e247SLoGin } 7506d5e247SLoGin 7606d5e247SLoGin if self.device_bind_driver(dev).is_ok() { 7706d5e247SLoGin return Ok(true); 7806d5e247SLoGin } else { 7906d5e247SLoGin dev.set_driver(None); 8006d5e247SLoGin return Ok(false); 8106d5e247SLoGin } 8206d5e247SLoGin } else { 83c566df45SLoGin let bus = dev 84c566df45SLoGin .bus() 85c566df45SLoGin .map(|bus| bus.upgrade()) 86c566df45SLoGin .flatten() 87c566df45SLoGin .ok_or(SystemError::EINVAL)?; 8806d5e247SLoGin let mut data = DeviceAttachData::new(dev.clone(), allow_async, false); 89c566df45SLoGin let mut flag = false; 90a03c4f9dSLoGin for driver in bus.subsystem().drivers().iter() { 9106d5e247SLoGin let r = self.do_device_attach_driver(&driver, &mut data); 9206d5e247SLoGin if unlikely(r.is_err()) { 9306d5e247SLoGin flag = false; 9406d5e247SLoGin break; 95c566df45SLoGin } else { 96c566df45SLoGin if r.unwrap() == true { 97c566df45SLoGin flag = true; 98c566df45SLoGin break; 9906d5e247SLoGin } 10006d5e247SLoGin } 10106d5e247SLoGin } 10206d5e247SLoGin 10306d5e247SLoGin if flag { 10406d5e247SLoGin r = Ok(true); 10506d5e247SLoGin } 10606d5e247SLoGin 10706d5e247SLoGin if !flag && allow_async && data.have_async { 10806d5e247SLoGin // If we could not find appropriate driver 10906d5e247SLoGin // synchronously and we are allowed to do 11006d5e247SLoGin // async probes and there are drivers that 11106d5e247SLoGin // want to probe asynchronously, we'll 11206d5e247SLoGin // try them. 11306d5e247SLoGin 11406d5e247SLoGin do_async = true; 11506d5e247SLoGin kdebug!( 11606d5e247SLoGin "do_device_attach: try scheduling asynchronous probe for device: {}", 11706d5e247SLoGin dev.name() 11806d5e247SLoGin ); 11906d5e247SLoGin } 12006d5e247SLoGin } 12106d5e247SLoGin 12206d5e247SLoGin if do_async { 12306d5e247SLoGin todo!("do_device_attach: do_async") 12406d5e247SLoGin } 12506d5e247SLoGin return r; 12606d5e247SLoGin } 12706d5e247SLoGin 128c566df45SLoGin /// 匹配设备和驱动 129c566df45SLoGin /// 130c566df45SLoGin /// ## 参数 131c566df45SLoGin /// 132c566df45SLoGin /// - `driver`: 驱动 133c566df45SLoGin /// - `data`: 匹配数据 134c566df45SLoGin /// 135c566df45SLoGin /// ## 返回 136c566df45SLoGin /// 137c566df45SLoGin /// - Ok(true): 匹配成功 138c566df45SLoGin /// - Ok(false): 没有匹配成功 139c566df45SLoGin /// - Err(SystemError): 匹配过程中出现意外错误,没有匹配成功 140*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#899 14106d5e247SLoGin fn do_device_attach_driver( 14206d5e247SLoGin &self, 143c566df45SLoGin driver: &Arc<dyn Driver>, 144c566df45SLoGin data: &mut DeviceAttachData, 145c566df45SLoGin ) -> Result<bool, SystemError> { 146c566df45SLoGin if let Some(bus) = driver.bus().map(|bus| bus.upgrade()).flatten() { 147c566df45SLoGin let r = bus.match_device(&data.dev, driver); 148c566df45SLoGin 149c566df45SLoGin if let Err(e) = r { 150c566df45SLoGin // 如果不是ENOSYS,则总线出错 151c566df45SLoGin if e != SystemError::ENOSYS { 152c566df45SLoGin kdebug!( 153c566df45SLoGin "do_device_attach_driver: bus.match_device() failed, dev: '{}', err: {:?}", 154c566df45SLoGin data.dev.name(), 155c566df45SLoGin e 156c566df45SLoGin ); 157c566df45SLoGin return Err(e); 158c566df45SLoGin } 159c566df45SLoGin } else { 160c566df45SLoGin if r.unwrap() == false { 161c566df45SLoGin return Ok(false); 162c566df45SLoGin } 163c566df45SLoGin } 164c566df45SLoGin } 165c566df45SLoGin 166c566df45SLoGin let async_allowed = driver.allows_async_probing(); 167c566df45SLoGin if data.check_async && async_allowed != data.want_async { 168c566df45SLoGin return Ok(false); 169c566df45SLoGin } 170c566df45SLoGin 171c566df45SLoGin return driver_manager() 172c566df45SLoGin .probe_device(driver, &data.dev) 173c566df45SLoGin .map(|_| true); 17406d5e247SLoGin } 17506d5e247SLoGin 17606d5e247SLoGin /// 检查设备是否绑定到驱动程序 17706d5e247SLoGin /// 17806d5e247SLoGin /// ## 参数 17906d5e247SLoGin /// 18006d5e247SLoGin /// - `dev`: 设备 18106d5e247SLoGin /// 18206d5e247SLoGin /// ## 返回 18306d5e247SLoGin /// 18406d5e247SLoGin /// 如果传递的设备已成功完成对驱动程序的探测,则返回true,否则返回false。 18506d5e247SLoGin pub fn device_is_bound(&self, dev: &Arc<dyn Device>) -> bool { 18606d5e247SLoGin if dev.driver().is_some() { 18706d5e247SLoGin return true; 18806d5e247SLoGin } else { 18906d5e247SLoGin return false; 19006d5e247SLoGin } 19106d5e247SLoGin } 19206d5e247SLoGin 19306d5e247SLoGin /// 把一个驱动绑定到设备上 19406d5e247SLoGin /// 19506d5e247SLoGin /// 允许手动绑定驱动到设备上。调用者需要设置好dev.driver(),保证其不为None 19606d5e247SLoGin /// 19706d5e247SLoGin /// ## 参数 19806d5e247SLoGin /// 19906d5e247SLoGin /// - `dev`: 设备 20006d5e247SLoGin /// 20106d5e247SLoGin /// ## 建议 20206d5e247SLoGin /// 20306d5e247SLoGin /// 使用device_manager().driver_attach()会更好 20406d5e247SLoGin /// 205*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#496 20606d5e247SLoGin pub fn device_bind_driver(&self, dev: &Arc<dyn Device>) -> Result<(), SystemError> { 20706d5e247SLoGin let r = driver_manager().driver_sysfs_add(dev); 20806d5e247SLoGin if let Err(e) = r { 20906d5e247SLoGin self.device_links_force_bind(dev); 210a03c4f9dSLoGin driver_manager().driver_bound(dev); 21106d5e247SLoGin return Err(e); 21206d5e247SLoGin } else { 213c566df45SLoGin if let Some(bus) = dev.bus().map(|bus| bus.upgrade()).flatten() { 21406d5e247SLoGin bus.subsystem().bus_notifier().call_chain( 21506d5e247SLoGin BusNotifyEvent::DriverNotBound, 21606d5e247SLoGin Some(dev), 21706d5e247SLoGin None, 21806d5e247SLoGin ); 21906d5e247SLoGin } 22006d5e247SLoGin } 22106d5e247SLoGin return r; 22206d5e247SLoGin } 22306d5e247SLoGin 224*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#528 225a03c4f9dSLoGin fn unbind_cleanup(&self, dev: &Arc<dyn Device>) { 226a03c4f9dSLoGin dev.set_driver(None); 227a03c4f9dSLoGin // todo: 添加更多操作,清理数据 22806d5e247SLoGin } 22906d5e247SLoGin } 23006d5e247SLoGin 231*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#866 23206d5e247SLoGin #[derive(Debug)] 23306d5e247SLoGin #[allow(dead_code)] 23406d5e247SLoGin struct DeviceAttachData { 23506d5e247SLoGin dev: Arc<dyn Device>, 23606d5e247SLoGin 23706d5e247SLoGin /// Indicates whether we are considering asynchronous probing or 23806d5e247SLoGin /// not. Only initial binding after device or driver registration 23906d5e247SLoGin /// (including deferral processing) may be done asynchronously, the 24006d5e247SLoGin /// rest is always synchronous, as we expect it is being done by 24106d5e247SLoGin /// request from userspace. 24206d5e247SLoGin check_async: bool, 24306d5e247SLoGin 24406d5e247SLoGin /// Indicates if we are binding synchronous or asynchronous drivers. 24506d5e247SLoGin /// When asynchronous probing is enabled we'll execute 2 passes 24606d5e247SLoGin /// over drivers: first pass doing synchronous probing and second 24706d5e247SLoGin /// doing asynchronous probing (if synchronous did not succeed - 24806d5e247SLoGin /// most likely because there was no driver requiring synchronous 24906d5e247SLoGin /// probing - and we found asynchronous driver during first pass). 25006d5e247SLoGin /// The 2 passes are done because we can't shoot asynchronous 25106d5e247SLoGin /// probe for given device and driver from bus_for_each_drv() since 25206d5e247SLoGin /// driver pointer is not guaranteed to stay valid once 25306d5e247SLoGin /// bus_for_each_drv() iterates to the next driver on the bus. 25406d5e247SLoGin want_async: bool, 25506d5e247SLoGin 25606d5e247SLoGin /// We'll set have_async to 'true' if, while scanning for matching 25706d5e247SLoGin /// driver, we'll encounter one that requests asynchronous probing. 25806d5e247SLoGin have_async: bool, 25906d5e247SLoGin } 26006d5e247SLoGin 26106d5e247SLoGin impl DeviceAttachData { 26206d5e247SLoGin pub fn new(dev: Arc<dyn Device>, check_async: bool, want_async: bool) -> Self { 26306d5e247SLoGin Self { 26406d5e247SLoGin dev, 26506d5e247SLoGin check_async, 26606d5e247SLoGin want_async, 26706d5e247SLoGin have_async: false, 26806d5e247SLoGin } 26906d5e247SLoGin } 27006d5e247SLoGin 27106d5e247SLoGin #[allow(dead_code)] 27206d5e247SLoGin #[inline(always)] 27306d5e247SLoGin fn set_have_async(&mut self) { 27406d5e247SLoGin self.have_async = true; 27506d5e247SLoGin } 27606d5e247SLoGin } 277a03c4f9dSLoGin 278a03c4f9dSLoGin impl DriverManager { 279a03c4f9dSLoGin /// 尝试把驱动绑定到现有的设备上 280a03c4f9dSLoGin /// 281a03c4f9dSLoGin /// 这个函数会遍历驱动现有的全部设备,然后尝试把他们匹配。 282a03c4f9dSLoGin /// 一旦有一个设备匹配成功,就会返回,并且设备的driver字段会被设置。 283a03c4f9dSLoGin pub fn driver_attach(&self, driver: &Arc<dyn Driver>) -> Result<(), SystemError> { 284c566df45SLoGin let bus = driver 285c566df45SLoGin .bus() 286c566df45SLoGin .map(|bus| bus.upgrade()) 287c566df45SLoGin .flatten() 288c566df45SLoGin .ok_or(SystemError::EINVAL)?; 289a03c4f9dSLoGin for dev in bus.subsystem().devices().iter() { 290a03c4f9dSLoGin if self.do_driver_attach(&dev, &driver) { 291a03c4f9dSLoGin // 匹配成功 292a03c4f9dSLoGin return Ok(()); 293a03c4f9dSLoGin } 294a03c4f9dSLoGin } 295a03c4f9dSLoGin 296a03c4f9dSLoGin return Ok(()); 297a03c4f9dSLoGin } 298a03c4f9dSLoGin 299*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#1134 300a03c4f9dSLoGin fn do_driver_attach(&self, device: &Arc<dyn Device>, driver: &Arc<dyn Driver>) -> bool { 301a03c4f9dSLoGin let r = self.match_device(driver, device).unwrap_or(false); 302a03c4f9dSLoGin if r == false { 303a03c4f9dSLoGin // 不匹配 304a03c4f9dSLoGin return false; 305a03c4f9dSLoGin } 306a03c4f9dSLoGin 307a03c4f9dSLoGin if driver.allows_async_probing() { 308a03c4f9dSLoGin unimplemented!( 309a03c4f9dSLoGin "do_driver_attach: probe driver '{}' asynchronously", 310a03c4f9dSLoGin driver.name() 311a03c4f9dSLoGin ); 312a03c4f9dSLoGin } 313a03c4f9dSLoGin 314a03c4f9dSLoGin if self.probe_device(driver, device).is_err() { 315a03c4f9dSLoGin return false; 316a03c4f9dSLoGin } 317a03c4f9dSLoGin 318a03c4f9dSLoGin return true; 319a03c4f9dSLoGin } 320a03c4f9dSLoGin 321a03c4f9dSLoGin #[inline(always)] 322a03c4f9dSLoGin pub fn match_device( 323a03c4f9dSLoGin &self, 324a03c4f9dSLoGin driver: &Arc<dyn Driver>, 325a03c4f9dSLoGin device: &Arc<dyn Device>, 326a03c4f9dSLoGin ) -> Result<bool, SystemError> { 327c566df45SLoGin return driver 328c566df45SLoGin .bus() 329c566df45SLoGin .map(|bus| bus.upgrade()) 330c566df45SLoGin .flatten() 331c566df45SLoGin .unwrap() 332c566df45SLoGin .match_device(device, driver); 333a03c4f9dSLoGin } 334a03c4f9dSLoGin 335a03c4f9dSLoGin /// 尝试把设备和驱动绑定在一起 336a03c4f9dSLoGin /// 337a03c4f9dSLoGin /// 338a03c4f9dSLoGin /// ## 返回 339a03c4f9dSLoGin /// 340a03c4f9dSLoGin /// - Ok(): 绑定成功 341a03c4f9dSLoGin /// - Err(ENODEV): 设备未注册 342a03c4f9dSLoGin /// - Err(EBUSY): 设备已经绑定到驱动上 343a03c4f9dSLoGin /// 344*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#802 345a03c4f9dSLoGin fn probe_device( 346a03c4f9dSLoGin &self, 347a03c4f9dSLoGin driver: &Arc<dyn Driver>, 348a03c4f9dSLoGin device: &Arc<dyn Device>, 349a03c4f9dSLoGin ) -> Result<(), SystemError> { 350a03c4f9dSLoGin let r = self.do_probe_device(driver, device); 351a03c4f9dSLoGin PROBE_WAIT_QUEUE.wakeup_all(None); 352a03c4f9dSLoGin return r; 353a03c4f9dSLoGin } 354a03c4f9dSLoGin 355a03c4f9dSLoGin fn do_probe_device( 356a03c4f9dSLoGin &self, 357a03c4f9dSLoGin driver: &Arc<dyn Driver>, 358a03c4f9dSLoGin device: &Arc<dyn Device>, 359a03c4f9dSLoGin ) -> Result<(), SystemError> { 360a03c4f9dSLoGin if device.is_dead() || (!device.is_registered()) { 361a03c4f9dSLoGin return Err(SystemError::ENODEV); 362a03c4f9dSLoGin } 363a03c4f9dSLoGin if device.driver().is_some() { 364a03c4f9dSLoGin return Err(SystemError::EBUSY); 365a03c4f9dSLoGin } 366a03c4f9dSLoGin 367a03c4f9dSLoGin device.set_can_match(true); 368a03c4f9dSLoGin 369a03c4f9dSLoGin self.really_probe(driver, device)?; 370a03c4f9dSLoGin 371a03c4f9dSLoGin return Ok(()); 372a03c4f9dSLoGin } 373a03c4f9dSLoGin 374*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#584 375a03c4f9dSLoGin fn really_probe( 376a03c4f9dSLoGin &self, 377a03c4f9dSLoGin driver: &Arc<dyn Driver>, 378a03c4f9dSLoGin device: &Arc<dyn Device>, 379a03c4f9dSLoGin ) -> Result<(), SystemError> { 380a03c4f9dSLoGin let bind_failed = || { 381a03c4f9dSLoGin device_manager().unbind_cleanup(device); 382a03c4f9dSLoGin }; 383a03c4f9dSLoGin 384a03c4f9dSLoGin let sysfs_failed = || { 385c566df45SLoGin if let Some(bus) = device.bus().map(|bus| bus.upgrade()).flatten() { 386a03c4f9dSLoGin bus.subsystem().bus_notifier().call_chain( 387a03c4f9dSLoGin BusNotifyEvent::DriverNotBound, 388a03c4f9dSLoGin Some(device), 389a03c4f9dSLoGin None, 390a03c4f9dSLoGin ); 391a03c4f9dSLoGin } 392a03c4f9dSLoGin }; 393a03c4f9dSLoGin 394a03c4f9dSLoGin let probe_failed = || { 395a03c4f9dSLoGin self.remove_from_sysfs(device); 396a03c4f9dSLoGin }; 397a03c4f9dSLoGin 398a03c4f9dSLoGin let dev_groups_failed = || { 399a03c4f9dSLoGin device_manager().remove(device); 400a03c4f9dSLoGin }; 401a03c4f9dSLoGin 402a03c4f9dSLoGin device.set_driver(Some(Arc::downgrade(driver))); 403a03c4f9dSLoGin 404a03c4f9dSLoGin self.add_to_sysfs(device).map_err(|e| { 405a03c4f9dSLoGin kerror!( 406a03c4f9dSLoGin "really_probe: add_to_sysfs failed, dev: '{}', err: {:?}", 407a03c4f9dSLoGin device.name(), 408a03c4f9dSLoGin e 409a03c4f9dSLoGin ); 410a03c4f9dSLoGin sysfs_failed(); 411a03c4f9dSLoGin bind_failed(); 412a03c4f9dSLoGin e 413a03c4f9dSLoGin })?; 414a03c4f9dSLoGin 415a03c4f9dSLoGin self.call_driver_probe(device, driver).map_err(|e| { 416a03c4f9dSLoGin kerror!( 417a03c4f9dSLoGin "really_probe: call_driver_probe failed, dev: '{}', err: {:?}", 418a03c4f9dSLoGin device.name(), 419a03c4f9dSLoGin e 420a03c4f9dSLoGin ); 421a03c4f9dSLoGin 422a03c4f9dSLoGin probe_failed(); 423a03c4f9dSLoGin sysfs_failed(); 424a03c4f9dSLoGin bind_failed(); 425a03c4f9dSLoGin e 426a03c4f9dSLoGin })?; 427a03c4f9dSLoGin 428a03c4f9dSLoGin device_manager() 429a03c4f9dSLoGin .add_groups(device, driver.dev_groups()) 430a03c4f9dSLoGin .map_err(|e| { 431a03c4f9dSLoGin kerror!( 432a03c4f9dSLoGin "really_probe: add_groups failed, dev: '{}', err: {:?}", 433a03c4f9dSLoGin device.name(), 434a03c4f9dSLoGin e 435a03c4f9dSLoGin ); 436a03c4f9dSLoGin dev_groups_failed(); 437a03c4f9dSLoGin probe_failed(); 438a03c4f9dSLoGin sysfs_failed(); 439a03c4f9dSLoGin bind_failed(); 440a03c4f9dSLoGin e 441a03c4f9dSLoGin })?; 442a03c4f9dSLoGin 443a03c4f9dSLoGin // 我们假设所有的设备都有 sync_state 这个属性。如果没有的话,也创建属性文件。 444a03c4f9dSLoGin device_manager() 445a03c4f9dSLoGin .create_file(device, &DeviceAttrStateSynced) 446a03c4f9dSLoGin .map_err(|e| { 447a03c4f9dSLoGin kerror!( 448a03c4f9dSLoGin "really_probe: create_file failed, dev: '{}', err: {:?}", 449a03c4f9dSLoGin device.name(), 450a03c4f9dSLoGin e 451a03c4f9dSLoGin ); 452a03c4f9dSLoGin dev_groups_failed(); 453a03c4f9dSLoGin probe_failed(); 454a03c4f9dSLoGin sysfs_failed(); 455a03c4f9dSLoGin bind_failed(); 456a03c4f9dSLoGin e 457a03c4f9dSLoGin })?; 458a03c4f9dSLoGin 459a03c4f9dSLoGin self.driver_bound(device); 460a03c4f9dSLoGin 461a03c4f9dSLoGin return Ok(()); 462a03c4f9dSLoGin } 463a03c4f9dSLoGin 464*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#434 465a03c4f9dSLoGin fn add_to_sysfs(&self, device: &Arc<dyn Device>) -> Result<(), SystemError> { 466a03c4f9dSLoGin let driver = device.driver().ok_or(SystemError::EINVAL)?; 467a03c4f9dSLoGin 468c566df45SLoGin if let Some(bus) = device.bus().map(|bus| bus.upgrade()).flatten() { 469a03c4f9dSLoGin bus.subsystem().bus_notifier().call_chain( 470a03c4f9dSLoGin BusNotifyEvent::BindDriver, 471a03c4f9dSLoGin Some(&device), 472a03c4f9dSLoGin None, 473a03c4f9dSLoGin ); 474a03c4f9dSLoGin } 475a03c4f9dSLoGin 476a03c4f9dSLoGin let driver_kobj = driver.clone() as Arc<dyn KObject>; 477a03c4f9dSLoGin let device_kobj = device.clone() as Arc<dyn KObject>; 478a03c4f9dSLoGin 479a03c4f9dSLoGin sysfs_instance().create_link(Some(&driver_kobj), &device_kobj, device.name())?; 480a03c4f9dSLoGin 481a03c4f9dSLoGin let fail_rm_dev_link = || { 482a03c4f9dSLoGin sysfs_instance().remove_link(&driver_kobj, device.name()); 483a03c4f9dSLoGin }; 484a03c4f9dSLoGin 485a03c4f9dSLoGin sysfs_instance() 486a03c4f9dSLoGin .create_link(Some(&device_kobj), &driver_kobj, "driver".to_string()) 487a03c4f9dSLoGin .map_err(|e| { 488a03c4f9dSLoGin fail_rm_dev_link(); 489a03c4f9dSLoGin e 490a03c4f9dSLoGin })?; 491a03c4f9dSLoGin 492a03c4f9dSLoGin device_manager() 493a03c4f9dSLoGin .create_file(device, &DeviceAttrCoredump) 494a03c4f9dSLoGin .map_err(|e| { 495a03c4f9dSLoGin sysfs_instance().remove_link(&device_kobj, "driver".to_string()); 496a03c4f9dSLoGin fail_rm_dev_link(); 497a03c4f9dSLoGin e 498a03c4f9dSLoGin })?; 499a03c4f9dSLoGin 500a03c4f9dSLoGin return Ok(()); 501a03c4f9dSLoGin } 502a03c4f9dSLoGin 503*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c?fi=driver_attach#469 504a03c4f9dSLoGin fn remove_from_sysfs(&self, _device: &Arc<dyn Device>) { 505a03c4f9dSLoGin todo!("remove_from_sysfs") 506a03c4f9dSLoGin } 507a03c4f9dSLoGin 508a03c4f9dSLoGin fn call_driver_probe( 509a03c4f9dSLoGin &self, 510a03c4f9dSLoGin device: &Arc<dyn Device>, 511a03c4f9dSLoGin driver: &Arc<dyn Driver>, 512a03c4f9dSLoGin ) -> Result<(), SystemError> { 513c566df45SLoGin let bus = device 514c566df45SLoGin .bus() 515c566df45SLoGin .map(|bus| bus.upgrade()) 516c566df45SLoGin .flatten() 517c566df45SLoGin .ok_or(SystemError::EINVAL)?; 518a03c4f9dSLoGin let r = bus.probe(device); 519a03c4f9dSLoGin if r == Err(SystemError::EOPNOTSUPP_OR_ENOTSUP) { 520a03c4f9dSLoGin kerror!( 521a03c4f9dSLoGin "call_driver_probe: bus.probe() failed, dev: '{}', err: {:?}", 522a03c4f9dSLoGin device.name(), 523a03c4f9dSLoGin r 524a03c4f9dSLoGin ); 525a03c4f9dSLoGin return r; 526a03c4f9dSLoGin } 527a03c4f9dSLoGin 528a03c4f9dSLoGin if r.is_ok() { 529a03c4f9dSLoGin return Ok(()); 530a03c4f9dSLoGin } 531a03c4f9dSLoGin 532a03c4f9dSLoGin let err = r.unwrap_err(); 533a03c4f9dSLoGin match err { 534a03c4f9dSLoGin SystemError::ENODEV | SystemError::ENXIO => { 535a03c4f9dSLoGin kdebug!( 536a03c4f9dSLoGin "driver'{}': probe of {} rejects match {:?}", 537a03c4f9dSLoGin driver.name(), 538a03c4f9dSLoGin device.name(), 539a03c4f9dSLoGin err 540a03c4f9dSLoGin ); 541a03c4f9dSLoGin } 542a03c4f9dSLoGin 543a03c4f9dSLoGin _ => { 544a03c4f9dSLoGin kwarn!( 545a03c4f9dSLoGin "driver'{}': probe of {} failed with error {:?}", 546a03c4f9dSLoGin driver.name(), 547a03c4f9dSLoGin device.name(), 548a03c4f9dSLoGin err 549a03c4f9dSLoGin ); 550a03c4f9dSLoGin } 551a03c4f9dSLoGin } 552a03c4f9dSLoGin 553a03c4f9dSLoGin return Err(err); 554a03c4f9dSLoGin } 555a03c4f9dSLoGin 556a03c4f9dSLoGin /// 当设备被成功探测,进行了'设备->驱动'绑定后,调用这个函数,完成'驱动->设备'的绑定 557*e7071df6SLoGin /// 参考 https://code.dragonos.org.cn/xref/linux-6.1.9/drivers/base/dd.c#393 558a03c4f9dSLoGin fn driver_bound(&self, device: &Arc<dyn Device>) { 559a03c4f9dSLoGin if self.driver_is_bound(device) { 560a03c4f9dSLoGin kwarn!("driver_bound: device '{}' is already bound.", device.name()); 561a03c4f9dSLoGin return; 562a03c4f9dSLoGin } 563a03c4f9dSLoGin 564a03c4f9dSLoGin let driver = device.driver().unwrap(); 565a03c4f9dSLoGin driver.add_device(device.clone()); 566a03c4f9dSLoGin 567c566df45SLoGin if let Some(bus) = device.bus().map(|bus| bus.upgrade()).flatten() { 568a03c4f9dSLoGin bus.subsystem().bus_notifier().call_chain( 569a03c4f9dSLoGin BusNotifyEvent::BoundDriver, 570a03c4f9dSLoGin Some(device), 571a03c4f9dSLoGin None, 572a03c4f9dSLoGin ); 573a03c4f9dSLoGin } 574a03c4f9dSLoGin 575a03c4f9dSLoGin // todo: 发送kobj bind的uevent 576a03c4f9dSLoGin } 577a03c4f9dSLoGin 578a03c4f9dSLoGin fn driver_is_bound(&self, device: &Arc<dyn Device>) -> bool { 579a03c4f9dSLoGin if let Some(driver) = device.driver() { 580a03c4f9dSLoGin if driver.find_device_by_name(&device.name()).is_some() { 581a03c4f9dSLoGin return true; 582a03c4f9dSLoGin } 583a03c4f9dSLoGin } 584a03c4f9dSLoGin 585a03c4f9dSLoGin return false; 586a03c4f9dSLoGin } 587a03c4f9dSLoGin } 588a03c4f9dSLoGin 589a03c4f9dSLoGin /// 设备文件夹下的`dev`文件的属性 590a03c4f9dSLoGin #[derive(Debug, Clone, Copy)] 591a03c4f9dSLoGin pub struct DeviceAttrStateSynced; 592a03c4f9dSLoGin 593a03c4f9dSLoGin impl Attribute for DeviceAttrStateSynced { 594a03c4f9dSLoGin fn mode(&self) -> ModeType { 595a03c4f9dSLoGin // 0o444 596a03c4f9dSLoGin return ModeType::S_IRUGO; 597a03c4f9dSLoGin } 598a03c4f9dSLoGin 599a03c4f9dSLoGin fn name(&self) -> &str { 600a03c4f9dSLoGin "state_synced" 601a03c4f9dSLoGin } 602a03c4f9dSLoGin 603a03c4f9dSLoGin fn show(&self, kobj: Arc<dyn KObject>, buf: &mut [u8]) -> Result<usize, SystemError> { 604a03c4f9dSLoGin let dev = kobj.cast::<dyn Device>().map_err(|kobj| { 605a03c4f9dSLoGin kerror!( 606a03c4f9dSLoGin "Intertrait casting not implemented for kobj: {}", 607a03c4f9dSLoGin kobj.name() 608a03c4f9dSLoGin ); 609a03c4f9dSLoGin SystemError::EOPNOTSUPP_OR_ENOTSUP 610a03c4f9dSLoGin })?; 611a03c4f9dSLoGin 612a03c4f9dSLoGin let val = dev.state_synced(); 613a03c4f9dSLoGin let val = if val { 1 } else { 0 }; 614a03c4f9dSLoGin return sysfs_emit_str(buf, format!("{}\n", val).as_str()); 615a03c4f9dSLoGin } 616a03c4f9dSLoGin 617a03c4f9dSLoGin fn support(&self) -> SysFSOpsSupport { 618a03c4f9dSLoGin SysFSOpsSupport::SHOW 619a03c4f9dSLoGin } 620a03c4f9dSLoGin } 621a03c4f9dSLoGin 622a03c4f9dSLoGin #[derive(Debug)] 623a03c4f9dSLoGin struct DeviceAttrCoredump; 624a03c4f9dSLoGin 625a03c4f9dSLoGin impl Attribute for DeviceAttrCoredump { 626a03c4f9dSLoGin fn name(&self) -> &str { 627a03c4f9dSLoGin "coredump" 628a03c4f9dSLoGin } 629a03c4f9dSLoGin 630a03c4f9dSLoGin fn mode(&self) -> ModeType { 631a03c4f9dSLoGin ModeType::from_bits_truncate(0o200) 632a03c4f9dSLoGin } 633a03c4f9dSLoGin 634a03c4f9dSLoGin fn support(&self) -> SysFSOpsSupport { 635a03c4f9dSLoGin SysFSOpsSupport::STORE 636a03c4f9dSLoGin } 637a03c4f9dSLoGin 638a03c4f9dSLoGin fn store(&self, kobj: Arc<dyn KObject>, buf: &[u8]) -> Result<usize, SystemError> { 639a03c4f9dSLoGin let dev = kobj.cast::<dyn Device>().map_err(|kobj| { 640a03c4f9dSLoGin kerror!( 641a03c4f9dSLoGin "Intertrait casting not implemented for kobj: {}", 642a03c4f9dSLoGin kobj.name() 643a03c4f9dSLoGin ); 644a03c4f9dSLoGin SystemError::EOPNOTSUPP_OR_ENOTSUP 645a03c4f9dSLoGin })?; 646a03c4f9dSLoGin 647a03c4f9dSLoGin let drv = dev.driver().ok_or(SystemError::EINVAL)?; 648a03c4f9dSLoGin drv.coredump(&dev)?; 649a03c4f9dSLoGin 650a03c4f9dSLoGin return Ok(buf.len()); 651a03c4f9dSLoGin } 652a03c4f9dSLoGin } 653