| 1 | |
| 2 |
|
| 3 | |
| 4 |
|
| 5 | |
| 6 | |
| 7 | |
| 8 | |
| 9 | |
| 10 |
|
| 11 | use crate::{
|
| 12 | bindings,
|
| 13 | error::{to_result, Error, Result, VTABLE_DEFAULT_ERROR},
|
| 14 | prelude::*,
|
| 15 | str::CStr,
|
| 16 | types::{ForeignOwnable, Opaque},
|
| 17 | };
|
| 18 | use core::{
|
| 19 | ffi::{c_int, c_long, c_uint, c_ulong},
|
| 20 | marker::PhantomData,
|
| 21 | mem::MaybeUninit,
|
| 22 | pin::Pin,
|
| 23 | };
|
| 24 |
|
| 25 | |
| 26 | #[derive(Copy, Clone)]
|
| 27 | pub struct MiscDeviceOptions {
|
| 28 | |
| 29 | pub name: &'static CStr,
|
| 30 | }
|
| 31 |
|
| 32 | impl MiscDeviceOptions {
|
| 33 | |
| 34 | pub const fn into_raw<T: MiscDevice>(self) -> bindings::miscdevice {
|
| 35 | |
| 36 | let mut result: bindings::miscdevice = unsafe { MaybeUninit::zeroed().assume_init() };
|
| 37 | result.minor = bindings::MISC_DYNAMIC_MINOR as _;
|
| 38 | result.name = self.name.as_char_ptr();
|
| 39 | result.fops = create_vtable::<T>();
|
| 40 | result
|
| 41 | }
|
| 42 | }
|
| 43 |
|
| 44 | |
| 45 | |
| 46 | |
| 47 | |
| 48 | |
| 49 | #[repr(transparent)]
|
| 50 | #[pin_data(PinnedDrop)]
|
| 51 | pub struct MiscDeviceRegistration<T> {
|
| 52 | #[pin]
|
| 53 | inner: Opaque<bindings::miscdevice>,
|
| 54 | _t: PhantomData<T>,
|
| 55 | }
|
| 56 |
|
| 57 | |
| 58 | |
| 59 | unsafe impl<T> Send for MiscDeviceRegistration<T> {}
|
| 60 | |
| 61 | |
| 62 | unsafe impl<T> Sync for MiscDeviceRegistration<T> {}
|
| 63 |
|
| 64 | impl<T: MiscDevice> MiscDeviceRegistration<T> {
|
| 65 | |
| 66 | pub fn register(opts: MiscDeviceOptions) -> impl PinInit<Self, Error> {
|
| 67 | try_pin_init!(Self {
|
| 68 | inner <- Opaque::try_ffi_init(move |slot: *mut bindings::miscdevice| {
|
| 69 | |
| 70 | unsafe { slot.write(opts.into_raw::<T>()) };
|
| 71 |
|
| 72 | |
| 73 | |
| 74 | |
| 75 | |
| 76 | |
| 77 | to_result(unsafe { bindings::misc_register(slot) })
|
| 78 | }),
|
| 79 | _t: PhantomData,
|
| 80 | })
|
| 81 | }
|
| 82 |
|
| 83 | |
| 84 | pub fn as_raw(&self) -> *mut bindings::miscdevice {
|
| 85 | self.inner.get()
|
| 86 | }
|
| 87 | }
|
| 88 |
|
| 89 | #[pinned_drop]
|
| 90 | impl<T> PinnedDrop for MiscDeviceRegistration<T> {
|
| 91 | fn drop(self: Pin<&mut Self>) {
|
| 92 | |
| 93 | unsafe { bindings::misc_deregister(self.inner.get()) };
|
| 94 | }
|
| 95 | }
|
| 96 |
|
| 97 | |
| 98 | #[vtable]
|
| 99 | pub trait MiscDevice {
|
| 100 | |
| 101 | type Ptr: ForeignOwnable + Send + Sync;
|
| 102 |
|
| 103 | |
| 104 | |
| 105 | |
| 106 | fn open() -> Result<Self::Ptr>;
|
| 107 |
|
| 108 | |
| 109 | fn release(device: Self::Ptr) {
|
| 110 | drop(device);
|
| 111 | }
|
| 112 |
|
| 113 | |
| 114 | |
| 115 | |
| 116 | |
| 117 | |
| 118 | fn ioctl(
|
| 119 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 120 | _cmd: u32,
|
| 121 | _arg: usize,
|
| 122 | ) -> Result<isize> {
|
| 123 | kernel::build_error(VTABLE_DEFAULT_ERROR)
|
| 124 | }
|
| 125 |
|
| 126 | |
| 127 | |
| 128 | |
| 129 | |
| 130 | |
| 131 | |
| 132 | |
| 133 | #[cfg(CONFIG_COMPAT)]
|
| 134 | fn compat_ioctl(
|
| 135 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 136 | _cmd: u32,
|
| 137 | _arg: usize,
|
| 138 | ) -> Result<isize> {
|
| 139 | kernel::build_error(VTABLE_DEFAULT_ERROR)
|
| 140 | }
|
| 141 | }
|
| 142 |
|
| 143 | const fn create_vtable<T: MiscDevice>() -> &'static bindings::file_operations {
|
| 144 | const fn maybe_fn<T: Copy>(check: bool, func: T) -> Option<T> {
|
| 145 | if check {
|
| 146 | Some(func)
|
| 147 | } else {
|
| 148 | None
|
| 149 | }
|
| 150 | }
|
| 151 |
|
| 152 | struct VtableHelper<T: MiscDevice> {
|
| 153 | _t: PhantomData<T>,
|
| 154 | }
|
| 155 | impl<T: MiscDevice> VtableHelper<T> {
|
| 156 | const VTABLE: bindings::file_operations = bindings::file_operations {
|
| 157 | open: Some(fops_open::<T>),
|
| 158 | release: Some(fops_release::<T>),
|
| 159 | unlocked_ioctl: maybe_fn(T::HAS_IOCTL, fops_ioctl::<T>),
|
| 160 | #[cfg(CONFIG_COMPAT)]
|
| 161 | compat_ioctl: if T::HAS_COMPAT_IOCTL {
|
| 162 | Some(fops_compat_ioctl::<T>)
|
| 163 | } else if T::HAS_IOCTL {
|
| 164 | Some(bindings::compat_ptr_ioctl)
|
| 165 | } else {
|
| 166 | None
|
| 167 | },
|
| 168 | |
| 169 | ..unsafe { MaybeUninit::zeroed().assume_init() }
|
| 170 | };
|
| 171 | }
|
| 172 |
|
| 173 | &VtableHelper::<T>::VTABLE
|
| 174 | }
|
| 175 |
|
| 176 | |
| 177 | |
| 178 | |
| 179 | |
| 180 | unsafe extern "C" fn fops_open<T: MiscDevice>(
|
| 181 | inode: *mut bindings::inode,
|
| 182 | file: *mut bindings::file,
|
| 183 | ) -> c_int {
|
| 184 | |
| 185 | let ret = unsafe { bindings::generic_file_open(inode, file) };
|
| 186 | if ret != 0 {
|
| 187 | return ret;
|
| 188 | }
|
| 189 |
|
| 190 | let ptr = match T::open() {
|
| 191 | Ok(ptr) => ptr,
|
| 192 | Err(err) => return err.to_errno(),
|
| 193 | };
|
| 194 |
|
| 195 | |
| 196 | unsafe { (*file).private_data = ptr.into_foreign().cast_mut() };
|
| 197 |
|
| 198 | 0
|
| 199 | }
|
| 200 |
|
| 201 | |
| 202 | |
| 203 | |
| 204 | |
| 205 | unsafe extern "C" fn fops_release<T: MiscDevice>(
|
| 206 | _inode: *mut bindings::inode,
|
| 207 | file: *mut bindings::file,
|
| 208 | ) -> c_int {
|
| 209 | |
| 210 | let private = unsafe { (*file).private_data };
|
| 211 | |
| 212 | let ptr = unsafe { <T::Ptr as ForeignOwnable>::from_foreign(private) };
|
| 213 |
|
| 214 | T::release(ptr);
|
| 215 |
|
| 216 | 0
|
| 217 | }
|
| 218 |
|
| 219 | |
| 220 | |
| 221 | |
| 222 | unsafe extern "C" fn fops_ioctl<T: MiscDevice>(
|
| 223 | file: *mut bindings::file,
|
| 224 | cmd: c_uint,
|
| 225 | arg: c_ulong,
|
| 226 | ) -> c_long {
|
| 227 | |
| 228 | let private = unsafe { (*file).private_data };
|
| 229 | |
| 230 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private) };
|
| 231 |
|
| 232 | match T::ioctl(device, cmd, arg as usize) {
|
| 233 | Ok(ret) => ret as c_long,
|
| 234 | Err(err) => err.to_errno() as c_long,
|
| 235 | }
|
| 236 | }
|
| 237 |
|
| 238 | |
| 239 | |
| 240 | |
| 241 | #[cfg(CONFIG_COMPAT)]
|
| 242 | unsafe extern "C" fn fops_compat_ioctl<T: MiscDevice>(
|
| 243 | file: *mut bindings::file,
|
| 244 | cmd: c_uint,
|
| 245 | arg: c_ulong,
|
| 246 | ) -> c_long {
|
| 247 | |
| 248 | let private = unsafe { (*file).private_data };
|
| 249 | |
| 250 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private) };
|
| 251 |
|
| 252 | match T::compat_ioctl(device, cmd, arg as usize) {
|
| 253 | Ok(ret) => ret as c_long,
|
| 254 | Err(err) => err.to_errno() as c_long,
|
| 255 | }
|
| 256 | }
|