DragonOS/kernel/src/misc/events/kprobe/mod.rs

use crate::driver::base::device::bus::Bus;
use crate::driver::base::device::{device_manager, device_register, sys_devices_kset, Device};
use crate::driver::base::kobject::KObject;
use crate::init::initcall::INITCALL_DEVICE;
use crate::misc::events::get_event_source_bus;
use crate::misc::events::kprobe::device::{KprobeAttr, KprobeDevice};
use alloc::sync::Arc;
use system_error::SystemError;
use unified_init::macros::unified_init;

pub mod device;
static mut KPROBE_DEVICE: Option<Arc<KprobeDevice>> = None;

#[unified_init(INITCALL_DEVICE)]
pub fn kprobe_subsys_init() -> Result<(), SystemError> {
    let kprobe_device = KprobeDevice::new(Some(Arc::downgrade(
        &(sys_devices_kset() as Arc<dyn KObject>),
    )));

    let event_source_bus = get_event_source_bus().ok_or(SystemError::EINVAL)?;
    kprobe_device.set_bus(Some(Arc::downgrade(&(event_source_bus as Arc<dyn Bus>))));

    // 注册到/sys/devices下
    device_register(kprobe_device.clone())?;
    unsafe {
        KPROBE_DEVICE = Some(kprobe_device.clone());
    }

    device_manager().create_file(&(kprobe_device as Arc<dyn Device>), &KprobeAttr)?;
    Ok(())
}