604 lines · 21.8 KB · Rust
RawDownload
// SPDX-License-Identifier: GPL-2.0 //! Abstractions for the serial device bus.//!//! C header: [`include/linux/serdev.h`](srctree/include/linux/serdev.h) use crate::{    acpi,    device,    driver,    error::{        from_result,        to_result,        VTABLE_DEFAULT_ERROR, //    },    new_mutex,    of,    prelude::*,    sync::{        aref::AlwaysRefCounted,        Mutex, //    },    time::Jiffies,    types::{        Opaque,        ScopeGuard, //    }, //}; use core::{    cell::UnsafeCell,    marker::PhantomData,    mem::{offset_of, MaybeUninit},    ptr::NonNull, //}; /// Parity bit to use with a serial device.#[repr(u32)]pub enum Parity {    /// No parity bit.    None = bindings::serdev_parity_SERDEV_PARITY_NONE,    /// Even partiy.    Even = bindings::serdev_parity_SERDEV_PARITY_EVEN,    /// Odd parity.    Odd = bindings::serdev_parity_SERDEV_PARITY_ODD,} /// An adapter for the registration of serial device bus device drivers.pub struct Adapter<T: Driver>(T); // SAFETY:// - `bindings::serdev_device_driver` is a C type declared as `repr(C)`.// - `PrivateData<'bound, T>` is the type of the driver's device private data.// - `struct serdev_device_driver` embeds a `struct device_driver`.// - `DEVICE_DRIVER_OFFSET` is the correct byte offset to the embedded `struct device_driver`.unsafe impl<T: Driver> driver::DriverLayout for Adapter<T> {    type DriverType = bindings::serdev_device_driver;    type DriverData<'bound> = PrivateData<'bound, T>;    const DEVICE_DRIVER_OFFSET: usize = core::mem::offset_of!(Self::DriverType, driver);} // SAFETY: A call to `unregister` for a given instance of `DriverType` is guaranteed to be valid if// a preceding call to `register` has been successful.unsafe impl<T: Driver> driver::RegistrationOps for Adapter<T> {    unsafe fn register(        sdrv: &Opaque<Self::DriverType>,        name: &'static CStr,        module: &'static ThisModule,    ) -> Result {        let of_table = match T::OF_ID_TABLE {            Some(table) => table.as_ptr(),            None => core::ptr::null(),        };         let acpi_table = match T::ACPI_ID_TABLE {            Some(table) => table.as_ptr(),            None => core::ptr::null(),        };         // SAFETY: It's safe to set the fields of `struct serdev_device_driver` on initialization.        unsafe {            (*sdrv.get()).driver.name = name.as_char_ptr();            (*sdrv.get()).probe = Some(Self::probe_callback);            (*sdrv.get()).remove = Some(Self::remove_callback);            (*sdrv.get()).driver.of_match_table = of_table;            (*sdrv.get()).driver.acpi_match_table = acpi_table;        }         // SAFETY: `sdrv` is guaranteed to be a valid `DriverType`.        to_result(unsafe { bindings::__serdev_device_driver_register(sdrv.get(), module.as_ptr()) })    }     unsafe fn unregister(sdrv: &Opaque<Self::DriverType>) {        // SAFETY: `sdrv` is guaranteed to be a valid `DriverType`.        unsafe { bindings::serdev_device_driver_unregister(sdrv.get()) };    }} #[doc(hidden)]#[pin_data(PinnedDrop)]pub struct PrivateData<'bound, T: Driver> {    sdev: &'bound Device<device::Bound>,    #[pin]    driver: UnsafeCell<MaybeUninit<T::Data<'bound>>>,    open: UnsafeCell<bool>,    /// Whether `receive_buf_callback` is allowed to call `Driver::receive`.    ///    /// If locked, the receive_buf_callback will be blocked on data reception.    /// This is the case while the driver is being probed or while [`PrivateData`] is being dropped.    /// This is necessary, because we need to open the serdev device before the driver has been    /// probed in order to allow it to be configured, which allows `receive_buf_callback` to be    /// called. Thus we need to block data until probe completes and the driver data becomes    /// initialized.    ///    /// If unlocked and true, the receive_buf_callback will forward the data to    /// `Driver::receive`. This is the normal state of operation.    ///    /// If unlocked and false, the receive_buf_callback will throw away the data.    /// This is only the case, if the serdev device is open and    /// - the driver returned an error in probe    /// or    /// - the driver data already has been dropped, because it was unbound.    #[pin]    active: Mutex<bool>,} #[pinned_drop]impl<T: Driver> PinnedDrop for PrivateData<'_, T> {    fn drop(self: Pin<&mut Self>) {        let mut active = self.active.lock();        if *active {            // SAFETY:            // - We have exclusive access to `self.driver`.            // - `self.driver` is guaranteed to be initialized.            unsafe { (*self.driver.get()).assume_init_drop() };            *active = false;        }        drop(active);         // SAFETY: We have exclusive access to `self.open`.        if unsafe { *self.open.get() } {            // SAFETY: `self.sdev.as_raw()` is guaranteed to be a pointer to a valid            // `struct serdev_device`.            unsafe { bindings::serdev_device_close(self.sdev.as_raw()) };        }    }} impl<T: Driver> Adapter<T> {    const OPS: &'static bindings::serdev_device_ops = &bindings::serdev_device_ops {        receive_buf: if T::HAS_RECEIVE {            Some(Self::receive_buf_callback)        } else {            None        },        write_wakeup: Some(bindings::serdev_device_write_wakeup),    };     extern "C" fn probe_callback(sdev: *mut bindings::serdev_device) -> kernel::ffi::c_int {        // SAFETY: The serial device bus only ever calls the probe callback with a valid pointer to        // a `struct serdev_device`.        //        // INVARIANT: `sdev` is valid for the duration of `probe_callback()`.        let sdev = unsafe { &*sdev.cast::<Device<device::CoreInternal<'_>>>() };        // SAFETY: `sdev` matched data is of type `Self::IdInfo`.        let info = unsafe { <Self as driver::Adapter>::id_info(sdev.as_ref()) };         from_result(|| {            sdev.as_ref().set_drvdata(try_pin_init!(PrivateData::<T> {                sdev: &**sdev,                driver: MaybeUninit::<T::Data<'_>>::zeroed().into(),                open: false.into(),                active <- new_mutex!(false),            }))?;            // SAFETY: We just set drvdata to `PrivateData<'_, T>`.            let private_data = unsafe { sdev.as_ref().drvdata_borrow::<PrivateData<'_, T>>() };            let private_data = ScopeGuard::new_with_data(private_data, |_| {                // SAFETY: We just set drvdata to `PrivateData<'_, T>`.                drop(unsafe { sdev.as_ref().drvdata_obtain::<PrivateData<'_, T>>() });            });            let mut active = private_data.active.lock();             // SAFETY: `sdev.as_raw()` is guaranteed to be a valid pointer to `serdev_device`.            unsafe { bindings::serdev_device_set_client_ops(sdev.as_raw(), Self::OPS) };             // SAFETY: The serial device bus only ever calls the probe callback with a valid pointer            // to a `serdev_device`.            to_result(unsafe { bindings::serdev_device_open(sdev.as_raw()) })?;             // SAFETY: We have exclusive access to `private_data.open`.            unsafe { *private_data.open.get() = true };             let data = T::probe(sdev, info);             // SAFETY: We have exclusive access to `private_data.driver`.            let driver = unsafe { &mut *private_data.driver.get() };            // SAFETY:            // - `driver.as_mut_ptr()` is a valid pointer to uninitialized data.            // - `private_data.driver` is pinned.            let result = unsafe { pin_init::raw_try_init(driver.as_mut_ptr(), data) };             *active = result.is_ok();             drop(active);             result.map(|()| {                private_data.dismiss();                0            })        })    }     extern "C" fn remove_callback(sdev: *mut bindings::serdev_device) {        // SAFETY: The serial device bus only ever calls the remove callback with a valid pointer        // to a `struct serdev_device`.        //        // INVARIANT: `sdev` is valid for the duration of `remove_callback()`.        let sdev = unsafe { &*sdev.cast::<Device<device::CoreInternal<'_>>>() };         // SAFETY: `remove_callback` is only ever called after a successful call to        // `probe_callback`, hence it's guaranteed that `Device::set_drvdata()` has been called        // and stored a `Pin<KBox<PrivateData<'_, T>>>`.        let private_data = unsafe { sdev.as_ref().drvdata_borrow::<PrivateData<'_, T>>() };         // SAFETY: No one has exclusive access to `private_data.driver`.        let data = unsafe { &*private_data.driver.get() };        // SAFETY:        // - `private_data.driver` is pinned.        // - `remove_callback` is only ever called after a successful call to `probe_callback`,        //   hence it's guaranteed that `private_data.driver` was initialized.        let data_pinned = unsafe { Pin::new_unchecked(data.assume_init_ref()) };         T::unbind(sdev, data_pinned);    }     extern "C" fn receive_buf_callback(        sdev: *mut bindings::serdev_device,        buf: *const u8,        length: usize,    ) -> usize {        // SAFETY: The serial device bus only ever calls the receive buf callback with a valid        // pointer to a `struct serdev_device`.        //        // INVARIANT: `sdev` is valid for the duration of `receive_buf_callback()`.        let sdev = unsafe { &*sdev.cast::<Device<device::BoundInternal>>() };         // SAFETY: `receive_buf_callback` is only ever called after a successful call to        // `probe_callback`, hence it's guaranteed that `Device::set_drvdata()` has been called        // and stored a `Pin<KBox<PrivateData<'_, T>>>`.        let private_data = unsafe { sdev.as_ref().drvdata_borrow::<PrivateData<'_, T>>() };        let active = private_data.active.lock();         if !*active {            return length;        }         // SAFETY: No one has exclusive access to `private_data.driver`.        let data = unsafe { &*private_data.driver.get() };        // SAFETY:        // - `private_data.driver` is pinned.        // - `receive_buf_callback` is only ever called after a successful call to `probe_callback`,        //   hence it's guaranteed that `private_data.driver` was initialized.        let data_pinned = unsafe { Pin::new_unchecked(data.assume_init_ref()) };         // SAFETY: `buf` is guaranteed to be non-null and has the size of `length`.        let buf = unsafe { core::slice::from_raw_parts(buf, length) };         T::receive(sdev, data_pinned, buf)    }} impl<T: Driver> driver::Adapter for Adapter<T> {    type IdInfo = T::IdInfo;     fn of_id_table() -> Option<of::IdTable<Self::IdInfo>> {        T::OF_ID_TABLE    }     fn acpi_id_table() -> Option<acpi::IdTable<Self::IdInfo>> {        T::ACPI_ID_TABLE    }} /// Declares a kernel module that exposes a single serial device bus device driver.////// # Examples////// ```ignore/// kernel::module_serdev_device_driver! {///     type: MyDriver,///     name: "Module name",///     authors: ["Author name"],///     description: "Description",///     license: "GPL v2",/// }/// ```#[macro_export]macro_rules! module_serdev_device_driver {    ($($f:tt)*) => {        $crate::module_driver!(<T>, $crate::serdev::Adapter<T>, { $($f)* });    };} /// The serial device bus device driver trait.////// Drivers must implement this trait in order to get a serial device bus device driver registered.////// # Examples//////```/// # use kernel::{///     acpi,///     bindings,///     device::{///         Bound,///         Core, /////     },///     of,///     serdev, ///// };////// struct MyDriver;////// kernel::of_device_table!(///     OF_TABLE,///     <MyDriver as serdev::Driver>::IdInfo,///     [///         (of::DeviceId::new(c"test,device"), ())///     ]/// );////// kernel::acpi_device_table!(///     ACPI_TABLE,///     <MyDriver as serdev::Driver>::IdInfo,///     [///         (acpi::DeviceId::new(c"LNUXBEEF"), ())///     ]/// );////// #[vtable]/// impl serdev::Driver for MyDriver {///     type IdInfo = ();///     type Data<'bound> = Self;///     const OF_ID_TABLE: Option<of::IdTable<Self::IdInfo>> = Some(&OF_TABLE);///     const ACPI_ID_TABLE: Option<acpi::IdTable<Self::IdInfo>> = Some(&ACPI_TABLE);//////     fn probe<'bound>(///         sdev: &'bound serdev::Device<Core<'_>>,///         _id_info: Option<&'bound Self::IdInfo>,///     ) -> impl PinInit<Self::Data<'bound>, Error> + 'bound {///         sdev.set_baudrate(115200);///         sdev.write_all(b"Hello\n", 0)?;///         Ok(MyDriver)///     }/// }///```#[vtable]pub trait Driver {    /// The type holding driver private data about each device id supported by the driver.    // TODO: Use associated_type_defaults once stabilized:    //    // ```    // type IdInfo: 'static = ();    // ```    type IdInfo: 'static;     /// The type of the driver's bus device private data.    type Data<'bound>: Send + Sync + 'bound;     /// The table of OF device ids supported by the driver.    const OF_ID_TABLE: Option<of::IdTable<Self::IdInfo>> = None;     /// The table of ACPI device ids supported by the driver.    const ACPI_ID_TABLE: Option<acpi::IdTable<Self::IdInfo>> = None;     /// Serial device bus device driver probe.    ///    /// Called when a new serial device bus device is added or discovered.    /// Implementers should attempt to initialize the device here.    fn probe<'bound>(        sdev: &'bound Device<device::Core<'_>>,        id_info: Option<&'bound Self::IdInfo>,    ) -> impl PinInit<Self::Data<'bound>, Error> + 'bound;     /// Serial device bus device driver unbind.    ///    /// Called when a [`Device`] is unbound from its bound [`Driver`]. Implementing this callback    /// is optional.    ///    /// This callback serves as a place for drivers to perform teardown operations that require a    /// `&Device<Core>` or `&Device<Bound>` reference. For instance.    ///    /// Otherwise, release operations for driver resources should be performed in `Drop`.    fn unbind<'bound>(sdev: &'bound Device<device::Core<'_>>, this: Pin<&Self::Data<'bound>>) {        let _ = (sdev, this);    }     /// Serial device bus device data receive callback.    ///    /// Called when data got received from device.    ///    /// Returns the number of bytes accepted.    fn receive<'bound>(        sdev: &'bound Device<device::Bound>,        this: Pin<&Self::Data<'bound>>,        data: &[u8],    ) -> usize {        let _ = (sdev, this, data);        build_error!(VTABLE_DEFAULT_ERROR)    }} /// The serial device bus device representation.////// This structure represents the Rust abstraction for a C `struct serdev_device`. The/// implementation abstracts the usage of an already existing C `struct serdev_device` within Rust/// code that we get passed from the C side.////// # Invariants////// A [`Device`] instance represents a valid `struct serdev_device` created by the C portion of/// the kernel.#[repr(transparent)]pub struct Device<Ctx: device::DeviceContext = device::Normal>(    Opaque<bindings::serdev_device>,    PhantomData<Ctx>,); impl<Ctx: device::DeviceContext> Device<Ctx> {    #[inline]    fn as_raw(&self) -> *mut bindings::serdev_device {        self.0.get()    }} impl Device<device::Bound> {    /// Set the baudrate in bits per second.    ///    /// Common baudrates are 115200, 9600, 19200, 57600, 4800.    ///    /// Use [`Device::write_flush`] before calling this if you have written data prior to this call.    #[inline]    pub fn set_baudrate(&self, speed: u32) -> Result<(), u32> {        // SAFETY: `self.as_raw()` is guaranteed to be a pointer to a valid `serdev_device`.        let ret = unsafe { bindings::serdev_device_set_baudrate(self.as_raw(), speed) };        if ret == speed {            Ok(())        } else {            Err(ret)        }    }     /// Set if flow control should be enabled.    ///    /// Use [`Device::write_flush`] before calling this if you have written data prior to this call.    #[inline]    pub fn set_flow_control(&self, enable: bool) {        // SAFETY: `self.as_raw()` is guaranteed to be a pointer to a valid `serdev_device`.        unsafe { bindings::serdev_device_set_flow_control(self.as_raw(), enable) };    }     /// Set parity to use.    ///    /// Use [`Device::write_flush`] before calling this if you have written data prior to this call.    #[inline]    pub fn set_parity(&self, parity: Parity) -> Result {        // SAFETY: `self.as_raw()` is guaranteed to be a pointer to a valid `serdev_device`.        to_result(unsafe { bindings::serdev_device_set_parity(self.as_raw(), parity as u32) })    }     /// Write data to the serial device until the controller has accepted all the data or has    /// been interrupted by a timeout or signal.    ///    /// Note that any accepted data has only been buffered by the controller. Use    /// [`Device::wait_until_sent`] to make sure the controller write buffer has actually been    /// emptied.    ///    /// Use a timeout of 0 to wait indefinitely.    ///    /// Returns the number of bytes written (less than `data.len()` if interrupted).    /// [`kernel::error::code::ETIMEDOUT`] or [`kernel::error::code::ERESTARTSYS`] if interrupted    /// before any bytes were written. [`kernel::error::code::EINVAL`] if `data.len() > i32::MAX`.    #[inline]    pub fn write_all(&self, data: &[u8], timeout: Jiffies) -> Result<usize> {        if data.len() > i32::MAX as usize {            return Err(EINVAL);        }         // SAFETY:        // - `self.as_raw()` is guaranteed to be a pointer to a valid `serdev_device`.        // - `data.as_ptr()` is guaranteed to be a valid array pointer with the size of        //   `data.len()`.        let ret = unsafe {            bindings::serdev_device_write(                self.as_raw(),                data.as_ptr(),                data.len(),                isize::try_from(timeout).unwrap_or_default(),            )        };

Lines 1–500 of 604.