@@ -21,6 +21,7 @@
#include <drm/drm_gem_framebuffer_helper.h>
#include <drm/drm_gem_shmem_helper.h>
#include <drm/drm_plane.h>
+#include <drm/drm_probe_helper.h>
#include <drm/drm_ioctl.h>
#include <kunit/test.h>
#include <linux/blk-mq.h>
@@ -8,15 +8,20 @@
pub mod plane;
use crate::{
- device,
+ container_of, device,
drm::{device::Device, drv::Driver},
error::to_result,
prelude::*,
private::Sealed,
+ sync::{Mutex, MutexGuard},
types::*,
};
use bindings;
-use core::{marker::PhantomData, ops::Deref, ptr::NonNull};
+use core::{
+ marker::PhantomData,
+ ops::Deref,
+ ptr::{self, addr_of_mut, NonNull},
+};
/// The C vtable for a [`Device`].
///
@@ -191,6 +196,23 @@ pub struct ModeConfigInfo {
pub preferred_fourcc: Option<u32>,
}
+impl<T: KmsDriver> Device<T> {
+ /// Retrieve a pointer to the mode_config mutex
+ #[inline]
+ pub(crate) fn mode_config_mutex(&self) -> &Mutex<()> {
+ // SAFETY: This lock is initialized for as long as `Device<T>` is exposed to users
+ unsafe { Mutex::from_raw(addr_of_mut!((*self.as_raw()).mode_config.mutex)) }
+ }
+
+ /// Acquire the [`mode_config.mutex`] for this [`Device`].
+ #[inline]
+ pub fn mode_config_lock(&self) -> ModeConfigGuard<'_, T> {
+ // INVARIANT: We're locking mode_config.mutex, fulfilling our invariant that this lock is
+ // held throughout ModeConfigGuard's lifetime.
+ ModeConfigGuard(self.mode_config_mutex().lock(), PhantomData)
+ }
+}
+
/// A modesetting object in DRM.
///
/// This is any type of object where the underlying C object contains a [`struct drm_mode_object`].
@@ -314,3 +336,67 @@ unsafe fn dec_ref(obj: NonNull<Self>) {
unsafe { bindings::drm_mode_object_put(obj.as_ref().raw_mode_obj()) }
}
}
+
+/// A mode config guard.
+///
+/// This is an exclusive primitive that represents when [`drm_device.mode_config.mutex`] is held - as
+/// some modesetting operations (particularly ones related to [`connectors`](connector)) are still
+/// protected under this single lock. The lock will be dropped once this object is dropped.
+///
+/// # Invariants
+///
+/// - `self.0` is contained within a [`struct drm_mode_config`], which is contained within a
+/// [`struct drm_device`].
+/// - The [`KmsDriver`] implementation of that [`struct drm_device`] is always `T`.
+/// - This type proves that [`drm_device.mode_config.mutex`] is acquired.
+///
+/// [`struct drm_mode_config`]: (srctree/include/drm/drm_device.h)
+/// [`drm_device.mode_config.mutex`]: (srctree/include/drm/drm_device.h)
+/// [`struct drm_device`]: (srctree/include/drm/drm_device.h)
+pub struct ModeConfigGuard<'a, T: KmsDriver>(MutexGuard<'a, ()>, PhantomData<T>);
+
+impl<'a, T: KmsDriver> ModeConfigGuard<'a, T> {
+ /// Construct a new [`ModeConfigGuard`].
+ ///
+ /// # Safety
+ ///
+ /// The caller must ensure that [`drm_device.mode_config.mutex`] is acquired.
+ ///
+ /// [`drm_device.mode_config.mutex`]: (srctree/include/drm/drm_device.h)
+ pub(crate) unsafe fn new(drm: &'a Device<T>) -> Self {
+ // SAFETY: Our safety contract fulfills the requirements of `MutexGuard::new()`
+ // INVARIANT: And our safety contract ensures that this type proves that
+ // `drm_device.mode_config.mutex` is acquired.
+ Self(
+ unsafe { MutexGuard::new(drm.mode_config_mutex(), ()) },
+ PhantomData,
+ )
+ }
+
+ /// Return the [`Device`] that this [`ModeConfigGuard`] belongs to.
+ pub fn drm_dev(&self) -> &'a Device<T> {
+ // SAFETY:
+ // - `self` is embedded within a `drm_mode_config` via our type invariants
+ // - `self.0.lock` has an equivalent data type to `mutex` via its type invariants.
+ let mode_config = unsafe { container_of!(self.0.lock, bindings::drm_mode_config, mutex) };
+
+ // SAFETY: And that `drm_mode_config` lives in a `drm_device` via type invariants.
+ unsafe {
+ Device::borrow(container_of!(
+ mode_config,
+ bindings::drm_device,
+ mode_config
+ ))
+ }
+ }
+
+ /// Assert that the given device is the owner of this mode config guard.
+ ///
+ /// # Panics
+ ///
+ /// Panics if `dev` is different from the owning device for this mode config guard.
+ #[inline]
+ pub(crate) fn assert_owner(&self, dev: &Device<T>) {
+ assert!(ptr::eq(self.drm_dev(), dev));
+ }
+}
@@ -4,7 +4,7 @@
//!
//! C header: [`include/drm/drm_connector.h`](srctree/include/drm/drm_connector.h)
-use super::{encoder::*, KmsDriver, ModeObject, RcModeObject};
+use super::{encoder::*, KmsDriver, ModeConfigGuard, ModeObject, RcModeObject};
use crate::{
alloc::KBox,
bindings,
@@ -17,7 +17,7 @@
};
use core::{
marker::*,
- mem,
+ mem::{self, ManuallyDrop},
ops::*,
ptr::{addr_of_mut, null_mut},
stringify,
@@ -106,7 +106,7 @@ pub trait DriverConnector: Send + Sync + Sized {
destroy: Some(connector_destroy_callback::<Self>),
force: None,
detect: None,
- fill_modes: None,
+ fill_modes: Some(bindings::drm_helper_probe_single_connector_modes),
debugfs_init: None,
oob_hotplug_event: None,
atomic_duplicate_state: Some(atomic_duplicate_state_callback::<Self::State>),
@@ -114,7 +114,7 @@ pub trait DriverConnector: Send + Sync + Sized {
helper_funcs: bindings::drm_connector_helper_funcs {
mode_valid: None,
atomic_check: None,
- get_modes: None,
+ get_modes: Some(get_modes_callback::<Self>),
detect_ctx: None,
enable_hpd: None,
disable_hpd: None,
@@ -145,6 +145,12 @@ pub trait DriverConnector: Send + Sync + Sized {
///
/// Drivers may use this to instantiate their [`DriverConnector`] object.
fn new(device: &Device<Self::Driver>, args: Self::Args) -> impl PinInit<Self, Error>;
+
+ /// Retrieve a list of available display modes for this [`Connector`].
+ fn get_modes<'a>(
+ connector: ConnectorGuard<'a, Self>,
+ guard: &ModeConfigGuard<'a, Self::Driver>,
+ ) -> i32;
}
/// The generated C vtable for a [`DriverConnector`].
@@ -196,6 +202,21 @@ fn deref(&self) -> &Self::Target {
}
}
+impl<T: DriverConnector> Connector<T> {
+ /// Acquire a [`ConnectorGuard`] for this connector from a [`ModeConfigGuard`].
+ ///
+ /// This verifies using the provided reference that the given guard is actually for the same
+ /// device as this connector's parent.
+ ///
+ /// # Panics
+ ///
+ /// Panics if `guard` is not a [`ModeConfigGuard`] for this connector's parent [`Device`].
+ pub fn guard<'a>(&'a self, guard: &ModeConfigGuard<'a, T::Driver>) -> ConnectorGuard<'a, T> {
+ guard.assert_owner(self.drm_dev());
+ ConnectorGuard(self)
+ }
+}
+
/// A trait implemented by any type that acts as a [`struct drm_connector`] interface.
///
/// This is implemented internally by DRM.
@@ -392,6 +413,39 @@ pub fn attach_encoder(&self, encoder: &impl AsRawEncoder) -> Result {
drop(unsafe { KBox::from_raw(connector as *mut Connector<T>) });
}
+unsafe extern "C" fn get_modes_callback<T: DriverConnector>(
+ connector: *mut bindings::drm_connector,
+) -> core::ffi::c_int {
+ // SAFETY: This is safe via `DriverConnector`s type invariants.
+ let connector = unsafe { Connector::<T>::from_raw(connector) };
+
+ // SAFETY: This FFI callback is only called while `mode_config.lock` is held
+ // We use ManuallyDrop here to prevent the lock from being released after the callback
+ // completes, as that should be handled by DRM.
+ let guard = ManuallyDrop::new(unsafe { ModeConfigGuard::new(connector.drm_dev()) });
+
+ T::get_modes(connector.guard(&guard), &guard)
+}
+
+/// A privileged [`Connector`] obtained while holding a [`ModeConfigGuard`].
+///
+/// This provides access to various methods for [`Connector`] that must happen under lock, such as
+/// setting resolution preferences and adding display modes.
+///
+/// # Invariants
+///
+/// Shares the invariants of [`ModeConfigGuard`].
+#[derive(Copy, Clone)]
+pub struct ConnectorGuard<'a, T: DriverConnector>(&'a Connector<T>);
+
+impl<T: DriverConnector> Deref for ConnectorGuard<'_, T> {
+ type Target = Connector<T>;
+
+ fn deref(&self) -> &Self::Target {
+ self.0
+ }
+}
+
// SAFETY: DRM expects this struct to be zero-initialized
unsafe impl Zeroable for bindings::drm_connector_state {}