588 lines · 22.0 KB · Rust
RawDownload
// SPDX-License-Identifier: GPL-2.0 OR MIT //! DRM device.//!//! C header: [`include/drm/drm_device.h`](srctree/include/drm/drm_device.h) use crate::{    alloc::allocator::Kmalloc,    bindings,    device,    drm::{        self,        driver::AllocImpl,        private::Sealed, //    },    error::from_err_ptr,    prelude::*,    sync::aref::{        ARef,        AlwaysRefCounted, //    },    types::{        NotThreadSafe,        Opaque, //    },    workqueue::{        HasDelayedWork,        HasWork,        Work,        WorkItem, //    }, //};use core::{    alloc::Layout,    cell::UnsafeCell,    marker::PhantomData,    mem,    ops::Deref,    ptr::{        self,        NonNull, //    },}; #[cfg(CONFIG_DRM_LEGACY)]macro_rules! drm_legacy_fields {    ( $($field:ident: $val:expr),* $(,)? ) => {        bindings::drm_driver {            $( $field: $val ),*,            firstopen: None,            preclose: None,            dma_ioctl: None,            dma_quiescent: None,            context_dtor: None,            irq_handler: None,            irq_preinstall: None,            irq_postinstall: None,            irq_uninstall: None,            get_vblank_counter: None,            enable_vblank: None,            disable_vblank: None,            dev_priv_size: 0,        }    }} #[cfg(not(CONFIG_DRM_LEGACY))]macro_rules! drm_legacy_fields {    ( $($field:ident: $val:expr),* $(,)? ) => {        bindings::drm_driver {            $( $field: $val ),*        }    }} /// A trait implemented by all possible contexts a [`Device`] can be used in.////// A [`Device`] can be in one of the following contexts:////// - [`Normal`]: The general-purpose, reference-counted context. A [`Device`] in this context may///   or may not be registered with userspace./// - [`Ioctl`]: The device has been registered with userspace at some point; used in ioctl///   dispatch context./// - [`Registered`]: The device is currently registered with userspace and the parent bus device///   is bound.////// Both `Device<T, Ioctl>` and `Device<T, Registered>` dereference to `Device<T>` ([`Normal`]),/// so any method available on a [`Normal`] device is also available in the other contexts.pub trait DeviceContext: Sealed + Send + Sync + 'static {} /// The general-purpose, reference-counted [`DeviceContext`].////// A [`Device`] in this context may or may not be registered with userspace. This context is used/// for reference-counted device handles and during device setup via [`UnregisteredDevice`].////// [`AlwaysRefCounted`] is only implemented for `Device<T, Normal>`, making this the required/// context for [`ARef`]-based device handles.pub struct Normal; impl Sealed for Normal {}impl DeviceContext for Normal {} /// The [`DeviceContext`] of a [`Device`] that is currently registered with userspace.////// A [`Device`] in this context is guaranteed to be registered and its parent bus device is/// guaranteed to be bound. This is enforced at runtime by [`RegistrationGuard`], which holds a/// `drm_dev_enter()` / `drm_dev_exit()` SRCU critical section.////// # Invariants////// The parent bus device is bound for the duration of any reference to a `Device<T, Registered>`.pub struct Registered; impl Sealed for Registered {}impl DeviceContext for Registered {} /// The [`DeviceContext`] of a [`Device`] that has been registered with userspace previously.////// A [`Device`] in this context has been registered at some point, but may be concurrently/// unregistering or already unregistered. `drm_dev_enter()` can guard against this, ensuring the/// device remains registered for the duration of the critical section.////// # Invariants////// A [`Device`] in this context has been registered with userspace via `drm_dev_register()` at/// some point.pub struct Ioctl; impl Sealed for Ioctl {}impl DeviceContext for Ioctl {} /// A [`Device`] which is known at compile-time to be unregistered with userspace.////// This type allows performing operations which are only safe to do before userspace registration,/// and can be used to create a [`Registration`](drm::driver::Registration) once the driver is ready/// to register the device with userspace.////// Since DRM device initialization must be single-threaded, this object is not thread-safe.////// # Invariants////// The device in `self.0` is guaranteed to be a newly created [`Device`] that has not yet been/// registered with userspace until this type is dropped.pub struct UnregisteredDevice<T: drm::Driver>(ARef<Device<T, Normal>>, NotThreadSafe); impl<T: drm::Driver> Deref for UnregisteredDevice<T> {    type Target = Device<T, Normal>;     fn deref(&self) -> &Self::Target {        &self.0    }} impl<T: drm::Driver> UnregisteredDevice<T> {    const fn compute_features() -> u32 {        let mut features = drm::driver::FEAT_GEM;         if T::FEAT_RENDER {            features |= drm::driver::FEAT_RENDER;        }         features    }     const VTABLE: bindings::drm_driver = drm_legacy_fields! {        load: None,        open: Some(drm::File::<T::File>::open_callback),        postclose: Some(drm::File::<T::File>::postclose_callback),        unload: None,        release: Some(Device::<T>::release),        master_set: None,        master_drop: None,        debugfs_init: None,         gem_create_object: T::Object::ALLOC_OPS.gem_create_object,        prime_handle_to_fd: T::Object::ALLOC_OPS.prime_handle_to_fd,        prime_fd_to_handle: T::Object::ALLOC_OPS.prime_fd_to_handle,        gem_prime_import: T::Object::ALLOC_OPS.gem_prime_import,        gem_prime_import_sg_table: T::Object::ALLOC_OPS.gem_prime_import_sg_table,        dumb_create: T::Object::ALLOC_OPS.dumb_create,        dumb_map_offset: T::Object::ALLOC_OPS.dumb_map_offset,         show_fdinfo: None,        fbdev_probe: None,         major: T::INFO.major,        minor: T::INFO.minor,        patchlevel: T::INFO.patchlevel,        name: crate::str::as_char_ptr_in_const_context(T::INFO.name).cast_mut(),        desc: crate::str::as_char_ptr_in_const_context(T::INFO.desc).cast_mut(),         driver_features: Self::compute_features(),        ioctls: T::IOCTLS.as_ptr(),        num_ioctls: T::IOCTLS.len() as i32,        fops: &Self::GEM_FOPS,    };     const GEM_FOPS: bindings::file_operations =        drm::gem::create_fops(crate::module::this_module::<T::OwnerModule>().as_ptr());     /// Create a new `UnregisteredDevice` for a `drm::Driver`.    ///    /// This can be used to create a [`Registration`](kernel::drm::Registration).    pub fn new(        dev: &T::ParentDevice<device::Bound>,        data: impl PinInit<T::Data, Error>,    ) -> Result<Self> {        // `__drm_dev_alloc` uses `kmalloc()` to allocate memory, hence ensure a `kmalloc()`        // compatible `Layout`.        let layout = Kmalloc::aligned_layout(Layout::new::<Device<T, Normal>>());         // Use a temporary vtable without a `release` callback until `data` is initialized, so        // init failure can release the DRM device without dropping uninitialized fields.        let alloc_vtable = bindings::drm_driver {            release: None,            ..Self::VTABLE        };         // SAFETY:        // - `alloc_vtable` reference remains valid until no longer used,        // - `dev` is valid by its type invarants,        let raw_drm: *mut Device<T, Normal> = unsafe {            bindings::__drm_dev_alloc(                dev.as_ref().as_raw(),                &alloc_vtable,                layout.size(),                mem::offset_of!(Device<T, Normal>, dev),            )        }        .cast();        let raw_drm = NonNull::new(from_err_ptr(raw_drm)?).ok_or(ENOMEM)?;         // SAFETY: `raw_drm` is a valid pointer to `Self`, given that `__drm_dev_alloc` was        // successful.        let drm_dev = unsafe { Device::into_drm_device(raw_drm) };         // SAFETY: `raw_drm` is a valid pointer to `Self`.        let raw_data = unsafe { ptr::addr_of_mut!((*raw_drm.as_ptr()).data) };         // SAFETY:        // - `raw_data` is a valid pointer to uninitialized memory.        // - `raw_data` will not move until it is dropped.        unsafe { pin_init::raw_try_init(raw_data, data) }.inspect_err(|_| {            // SAFETY: `__drm_dev_alloc()` was successful, hence `drm_dev` must be valid and the            // refcount must be non-zero.            unsafe { bindings::drm_dev_put(drm_dev) };        })?;         // SAFETY: `drm_dev` is still private to this function.        unsafe { (*drm_dev).driver = const { &Self::VTABLE } };         // SAFETY: `raw_drm` is valid; no concurrent access before registration.        unsafe { (*raw_drm.as_ptr()).registration_data = UnsafeCell::new(NonNull::dangling()) };         // SAFETY: The reference count is one, and now we take ownership of that reference as a        // `drm::Device`.        // INVARIANT: We just created the device above, but have yet to call `drm_dev_register`.        // `Self` cannot be copied or sent to another thread - ensuring that `drm_dev_register`        // won't be called during its lifetime and that the device is unregistered.        Ok(Self(unsafe { ARef::from_raw(raw_drm) }, NotThreadSafe))    }} /// A typed DRM device with a specific [`drm::Driver`] implementation and [`DeviceContext`].////// A device in the [`Registered`] context is currently registered with userspace and its parent/// bus device is bound. The [`Normal`] context is the general-purpose, reference-counted context.////// # Invariants////// * `self.dev` is a valid instance of a `struct device`./// * The data layout of `Self` remains the same across all implementations of `C`./// * Any invariants for `C` also apply.#[repr(C)]pub struct Device<T: drm::Driver, C: DeviceContext = Normal> {    dev: Opaque<bindings::drm_device>,    data: T::Data,    pub(super) registration_data: UnsafeCell<NonNull<T::RegistrationData<'static>>>,    _ctx: PhantomData<C>,} impl<T: drm::Driver, C: DeviceContext> Device<T, C> {    pub(crate) fn as_raw(&self) -> *mut bindings::drm_device {        self.dev.get()    }     /// # Safety    ///    /// `ptr` must be a valid pointer to a `struct device` embedded in `Self`.    unsafe fn from_drm_device(ptr: *const bindings::drm_device) -> *mut Self {        // SAFETY: By the safety requirements of this function `ptr` is a valid pointer to a        // `struct drm_device` embedded in `Self`.        unsafe { crate::container_of!(Opaque::cast_from(ptr), Self, dev) }.cast_mut()    }     /// # Safety    ///    /// `ptr` must be a valid pointer to `Self`.    unsafe fn into_drm_device(ptr: NonNull<Self>) -> *mut bindings::drm_device {        // SAFETY: By the safety requirements of this function, `ptr` is a valid pointer to `Self`.        unsafe { &raw mut (*ptr.as_ptr()).dev }.cast()    }     /// Not intended to be called externally, except via declare_drm_ioctls!()    ///    /// # Safety    ///    /// * Callers must ensure that `ptr` is valid, non-null, and has a non-zero reference count,    ///   i.e. it must be ensured that the reference count of the C `struct drm_device` `ptr` points    ///   to can't drop to zero, for the duration of this function call and the entire duration when    ///   the returned reference exists.    /// * Additionally, callers must ensure that the `struct device`, `ptr` is pointing to, is    ///   embedded in `Self`.    /// * Callers promise that any type invariants of `C` will be upheld.    #[doc(hidden)]    pub unsafe fn from_raw<'a>(ptr: *const bindings::drm_device) -> &'a Self {        // SAFETY: By the safety requirements of this function `ptr` is a valid pointer to a        // `struct drm_device` embedded in `Self`.        let ptr = unsafe { Self::from_drm_device(ptr) };         // SAFETY: `ptr` is valid by the safety requirements of this function.        unsafe { &*ptr.cast() }    }     extern "C" fn release(ptr: *mut bindings::drm_device) {        // SAFETY: `ptr` is a valid pointer to a `struct drm_device` and embedded in `Self`.        let this = unsafe { Self::from_drm_device(ptr) };         // SAFETY:        // - When `release` runs it is guaranteed that there is no further access to `this`.        // - `this` is valid for dropping.        unsafe { core::ptr::drop_in_place(this) };    }     /// Change the [`DeviceContext`] for a [`Device`].    ///    /// # Safety    ///    /// The caller promises that `self` fulfills all of the guarantees provided by the given    /// [`DeviceContext`].    pub(crate) unsafe fn assume_ctx<NewCtx: DeviceContext>(&self) -> &Device<T, NewCtx> {        // SAFETY: The data layout is identical via our type invariants.        unsafe { mem::transmute(self) }    }} impl<T: drm::Driver> Device<T, Ioctl> {    /// Guard against the parent bus device being unbound.    ///    /// Returns a [`RegistrationGuard`] if the device has not been unplugged, [`None`] otherwise.    ///    /// While [`RegistrationGuard`] is held the parent device is guaranteed to be bound.    #[must_use]    pub fn registration_guard(&self) -> Option<RegistrationGuard<'_, T>> {        let mut idx: i32 = 0;        // SAFETY: `self.as_raw()` is a valid pointer to a `struct drm_device`.        if unsafe { bindings::drm_dev_enter(self.as_raw(), &mut idx) } {            // INVARIANT:            // - `idx` is the SRCU index from the successful `drm_dev_enter()` above.            // - The parent bus device is bound: `drm_dev_enter()` succeeded, meaning            //   `drm_dev_unplug()` has not completed; since it is only called from            //   `Registration::drop()` during parent unbind, the parent is still bound.            Some(RegistrationGuard {                // SAFETY: See INVARIANT above; the `Registered` context invariant holds.                dev: unsafe { self.assume_ctx() },                idx,                _not_send: NotThreadSafe,            })        } else {            None        }    }} /// A guard proving the DRM device is registered and the parent bus device is bound.////// The guard dereferences to [`Device<T, Registered>`], providing access to the DRM device with/// the guarantee that the parent bus device is bound for the entire duration of the critical/// section.////// Internally this is backed by a `drm_dev_enter()` / `drm_dev_exit()` SRCU critical section.////// # Invariants////// - `idx` is the SRCU read lock index returned by a successful `drm_dev_enter()` call./// - The parent bus device of `dev` is bound for the lifetime of this guard.#[must_use]pub struct RegistrationGuard<'a, T: drm::Driver> {    dev: &'a Device<T, Registered>,    idx: i32,    _not_send: NotThreadSafe,} impl<T: drm::Driver> Device<T, Registered> {    /// Returns a reference to the registration data with lifetime shortened from `'static`.    ///    /// # Safety    ///    /// The returned reference must not be exposed to code that can choose a concrete lifetime for    /// it, as that would be unsound for types that are invariant over their lifetime parameter    /// (e.g. it must be passed through an HRTB-bounded closure).    #[inline]    unsafe fn registration_data_unchecked(&self) -> &T::RegistrationData<'_> {        // SAFETY:        // - `Registered` guarantees the parent bus device is bound, hence the pointer is valid.        // - The pointer cast from `Of<'static>` to `Of<'_>` is layout-compatible since lifetimes        //   are erased at runtime.        // - Caller guarantees the reference is only used behind an HRTB, making the lifetime        //   shortening sound regardless of variance.        unsafe { (*self.registration_data.get()).cast::<_>().as_ref() }    }     /// Access the registration data through a closure, with the lifetime tied to the closure    /// scope.    ///    /// The data is owned by [`Registration`](drm::Registration) and is guaranteed to remain valid    /// as long as the device is registered, since [`Registration`](drm::Registration)'s `drop`    /// calls `drm_dev_unplug()` which waits for all `drm_dev_enter()` critical sections to    /// complete.    #[inline]    pub fn registration_data_with<R, F>(&self, f: F) -> R    where        F: for<'a> FnOnce(&'a T::RegistrationData<'a>) -> R,    {        // SAFETY: `Registered` guarantees the device is registered and the parent bus device is        // bound. The closure's HRTB `for<'a>` prevents the caller from smuggling in references        // with a concrete short lifetime, satisfying the lifetime requirement of        // `registration_data_unchecked`.        f(unsafe { self.registration_data_unchecked() })    }} impl<T: drm::Driver> Deref for RegistrationGuard<'_, T> {    type Target = Device<T, Registered>;     #[inline]    fn deref(&self) -> &Self::Target {        self.dev    }} impl<T: drm::Driver> Drop for RegistrationGuard<'_, T> {    #[inline]    fn drop(&mut self) {        // SAFETY: `self.idx` was returned by a successful `drm_dev_enter()` call, as guaranteed        // by the type invariants of `RegistrationGuard`.        unsafe { bindings::drm_dev_exit(self.idx) };    }} impl<T: drm::Driver> Deref for Device<T> {    type Target = T::Data;     fn deref(&self) -> &Self::Target {        &self.data    }} impl<T: drm::Driver> Deref for Device<T, Registered> {    type Target = Device<T>;     #[inline]    fn deref(&self) -> &Self::Target {        // SAFETY: The caller holds a `Device<T, Registered>`, which guarantees all invariants        // of the weaker `Normal` context.        unsafe { self.assume_ctx() }    }} impl<T: drm::Driver> Deref for Device<T, Ioctl> {    type Target = Device<T>;     #[inline]    fn deref(&self) -> &Self::Target {        // SAFETY: The caller holds a `Device<T, Ioctl>`, which guarantees all invariants        // of the weaker `Normal` context.        unsafe { self.assume_ctx() }    }} // SAFETY: DRM device objects are always reference counted and the get/put functions// satisfy the requirements.unsafe impl<T: drm::Driver> AlwaysRefCounted for Device<T> {    fn inc_ref(&self) {        // SAFETY: The existence of a shared reference guarantees that the refcount is non-zero.        unsafe { bindings::drm_dev_get(self.as_raw()) };    }     unsafe fn dec_ref(obj: NonNull<Self>) {        // SAFETY: `obj` is a valid pointer to `Self`.        let drm_dev = unsafe { Self::into_drm_device(obj) };         // SAFETY: The safety requirements guarantee that the refcount is non-zero.        unsafe { bindings::drm_dev_put(drm_dev) };    }} impl<T: drm::Driver> AsRef<T::ParentDevice<device::Normal>> for Device<T> {    fn as_ref(&self) -> &T::ParentDevice<device::Normal> {        // SAFETY: `bindings::drm_device::dev` is valid as long as the DRM device itself is valid,

Lines 1–500 of 588.