| 1 | |
| 2 |
|
| 3 | |
| 4 |
|
| 5 | |
| 6 | |
| 7 | |
| 8 | |
| 9 | |
| 10 |
|
| 11 | use crate::{
|
| 12 | bindings,
|
| 13 | device::Device,
|
| 14 | error::{
|
| 15 | to_result,
|
| 16 | VTABLE_DEFAULT_ERROR, |
| 17 | },
|
| 18 | fs::{
|
| 19 | File,
|
| 20 | Kiocb, |
| 21 | },
|
| 22 | iov::{
|
| 23 | IovIterDest,
|
| 24 | IovIterSource, |
| 25 | },
|
| 26 | mm::virt::VmaNew,
|
| 27 | prelude::*,
|
| 28 | seq_file::SeqFile,
|
| 29 | types::{
|
| 30 | ForeignOwnable,
|
| 31 | Opaque, |
| 32 | },
|
| 33 | };
|
| 34 | use core::marker::PhantomData;
|
| 35 |
|
| 36 | |
| 37 | #[derive(Copy, Clone)]
|
| 38 | pub struct MiscDeviceOptions {
|
| 39 | |
| 40 | pub name: &'static CStr,
|
| 41 | }
|
| 42 |
|
| 43 | impl MiscDeviceOptions {
|
| 44 | |
| 45 | pub const fn into_raw<T: MiscDevice>(self) -> bindings::miscdevice {
|
| 46 | let mut result: bindings::miscdevice = pin_init::zeroed();
|
| 47 | result.minor = bindings::MISC_DYNAMIC_MINOR as ffi::c_int;
|
| 48 | result.name = crate::str::as_char_ptr_in_const_context(self.name);
|
| 49 | result.fops = MiscdeviceVTable::<T>::build();
|
| 50 | result
|
| 51 | }
|
| 52 | }
|
| 53 |
|
| 54 | |
| 55 | |
| 56 | |
| 57 | |
| 58 | |
| 59 | |
| 60 | |
| 61 | |
| 62 | |
| 63 | |
| 64 | |
| 65 | #[repr(transparent)]
|
| 66 | #[pin_data(PinnedDrop)]
|
| 67 | pub struct MiscDeviceRegistration<T> {
|
| 68 | #[pin]
|
| 69 | inner: Opaque<bindings::miscdevice>,
|
| 70 | _t: PhantomData<T>,
|
| 71 | }
|
| 72 |
|
| 73 | |
| 74 | |
| 75 | unsafe impl<T> Send for MiscDeviceRegistration<T> {}
|
| 76 | |
| 77 | |
| 78 | unsafe impl<T> Sync for MiscDeviceRegistration<T> {}
|
| 79 |
|
| 80 | impl<T: MiscDevice> MiscDeviceRegistration<T> {
|
| 81 | |
| 82 | pub fn register(opts: MiscDeviceOptions) -> impl PinInit<Self, Error> {
|
| 83 | try_pin_init!(Self {
|
| 84 | inner <- Opaque::try_ffi_init(move |slot: *mut bindings::miscdevice| {
|
| 85 | |
| 86 | unsafe { slot.write(opts.into_raw::<T>()) };
|
| 87 |
|
| 88 | |
| 89 | |
| 90 | |
| 91 | |
| 92 | |
| 93 | to_result(unsafe { bindings::misc_register(slot) })
|
| 94 | }),
|
| 95 | _t: PhantomData,
|
| 96 | })
|
| 97 | }
|
| 98 |
|
| 99 | |
| 100 | pub fn as_raw(&self) -> *mut bindings::miscdevice {
|
| 101 | self.inner.get()
|
| 102 | }
|
| 103 |
|
| 104 | |
| 105 | pub fn device(&self) -> &Device {
|
| 106 | |
| 107 | |
| 108 | |
| 109 | |
| 110 | |
| 111 | unsafe { Device::from_raw((*self.as_raw()).this_device) }
|
| 112 | }
|
| 113 | }
|
| 114 |
|
| 115 | #[pinned_drop]
|
| 116 | impl<T> PinnedDrop for MiscDeviceRegistration<T> {
|
| 117 | fn drop(self: Pin<&mut Self>) {
|
| 118 | |
| 119 | unsafe { bindings::misc_deregister(self.inner.get()) };
|
| 120 | }
|
| 121 | }
|
| 122 |
|
| 123 | |
| 124 | #[vtable]
|
| 125 | pub trait MiscDevice: Sized {
|
| 126 | |
| 127 | type Ptr: ForeignOwnable + Send + Sync;
|
| 128 |
|
| 129 | |
| 130 | |
| 131 | |
| 132 | fn open(_file: &File, _misc: &MiscDeviceRegistration<Self>) -> Result<Self::Ptr>;
|
| 133 |
|
| 134 | |
| 135 | fn release(device: Self::Ptr, _file: &File) {
|
| 136 | drop(device);
|
| 137 | }
|
| 138 |
|
| 139 | |
| 140 | |
| 141 | |
| 142 | |
| 143 | |
| 144 | |
| 145 | |
| 146 | |
| 147 | fn mmap(
|
| 148 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 149 | _file: &File,
|
| 150 | _vma: &VmaNew,
|
| 151 | ) -> Result {
|
| 152 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 153 | }
|
| 154 |
|
| 155 | |
| 156 | fn read_iter(_kiocb: Kiocb<'_, Self::Ptr>, _iov: &mut IovIterDest<'_>) -> Result<usize> {
|
| 157 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 158 | }
|
| 159 |
|
| 160 | |
| 161 | fn write_iter(_kiocb: Kiocb<'_, Self::Ptr>, _iov: &mut IovIterSource<'_>) -> Result<usize> {
|
| 162 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 163 | }
|
| 164 |
|
| 165 | |
| 166 | |
| 167 | |
| 168 | |
| 169 | |
| 170 | fn ioctl(
|
| 171 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 172 | _file: &File,
|
| 173 | _cmd: u32,
|
| 174 | _arg: usize,
|
| 175 | ) -> Result<isize> {
|
| 176 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 177 | }
|
| 178 |
|
| 179 | |
| 180 | |
| 181 | |
| 182 | |
| 183 | |
| 184 | |
| 185 | |
| 186 | #[cfg(CONFIG_COMPAT)]
|
| 187 | fn compat_ioctl(
|
| 188 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 189 | _file: &File,
|
| 190 | _cmd: u32,
|
| 191 | _arg: usize,
|
| 192 | ) -> Result<isize> {
|
| 193 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 194 | }
|
| 195 |
|
| 196 | |
| 197 | fn show_fdinfo(
|
| 198 | _device: <Self::Ptr as ForeignOwnable>::Borrowed<'_>,
|
| 199 | _m: &SeqFile,
|
| 200 | _file: &File,
|
| 201 | ) {
|
| 202 | build_error!(VTABLE_DEFAULT_ERROR)
|
| 203 | }
|
| 204 | }
|
| 205 |
|
| 206 | |
| 207 | struct MiscdeviceVTable<T: MiscDevice>(PhantomData<T>);
|
| 208 |
|
| 209 | impl<T: MiscDevice> MiscdeviceVTable<T> {
|
| 210 | |
| 211 | |
| 212 | |
| 213 | |
| 214 | unsafe extern "C" fn open(inode: *mut bindings::inode, raw_file: *mut bindings::file) -> c_int {
|
| 215 | |
| 216 | let ret = unsafe { bindings::generic_file_open(inode, raw_file) };
|
| 217 | if ret != 0 {
|
| 218 | return ret;
|
| 219 | }
|
| 220 |
|
| 221 | |
| 222 | let misc_ptr = unsafe { (*raw_file).private_data };
|
| 223 |
|
| 224 | |
| 225 | |
| 226 | |
| 227 | |
| 228 | let misc = unsafe { &*misc_ptr.cast::<MiscDeviceRegistration<T>>() };
|
| 229 |
|
| 230 | |
| 231 | |
| 232 | |
| 233 | let file = unsafe { File::from_raw_file(raw_file) };
|
| 234 |
|
| 235 | let ptr = match T::open(file, misc) {
|
| 236 | Ok(ptr) => ptr,
|
| 237 | Err(err) => return err.to_errno(),
|
| 238 | };
|
| 239 |
|
| 240 | |
| 241 | |
| 242 | |
| 243 | |
| 244 | |
| 245 | |
| 246 | unsafe { (*raw_file).private_data = ptr.into_foreign() };
|
| 247 |
|
| 248 | 0
|
| 249 | }
|
| 250 |
|
| 251 | |
| 252 | |
| 253 | |
| 254 | |
| 255 | unsafe extern "C" fn release(_inode: *mut bindings::inode, file: *mut bindings::file) -> c_int {
|
| 256 | |
| 257 | let private = unsafe { (*file).private_data };
|
| 258 | |
| 259 | let ptr = unsafe { <T::Ptr as ForeignOwnable>::from_foreign(private) };
|
| 260 |
|
| 261 | |
| 262 | |
| 263 | |
| 264 | T::release(ptr, unsafe { File::from_raw_file(file) });
|
| 265 |
|
| 266 | 0
|
| 267 | }
|
| 268 |
|
| 269 | |
| 270 | |
| 271 | |
| 272 | |
| 273 | unsafe extern "C" fn read_iter(
|
| 274 | kiocb: *mut bindings::kiocb,
|
| 275 | iter: *mut bindings::iov_iter,
|
| 276 | ) -> isize {
|
| 277 | |
| 278 | |
| 279 | let kiocb = unsafe { Kiocb::from_raw(kiocb) };
|
| 280 | |
| 281 | let iov = unsafe { IovIterDest::from_raw(iter) };
|
| 282 |
|
| 283 | match T::read_iter(kiocb, iov) {
|
| 284 | Ok(res) => res as isize,
|
| 285 | Err(err) => err.to_errno() as isize,
|
| 286 | }
|
| 287 | }
|
| 288 |
|
| 289 | |
| 290 | |
| 291 | |
| 292 | |
| 293 | unsafe extern "C" fn write_iter(
|
| 294 | kiocb: *mut bindings::kiocb,
|
| 295 | iter: *mut bindings::iov_iter,
|
| 296 | ) -> isize {
|
| 297 | |
| 298 | |
| 299 | let kiocb = unsafe { Kiocb::from_raw(kiocb) };
|
| 300 | |
| 301 | let iov = unsafe { IovIterSource::from_raw(iter) };
|
| 302 |
|
| 303 | match T::write_iter(kiocb, iov) {
|
| 304 | Ok(res) => res as isize,
|
| 305 | Err(err) => err.to_errno() as isize,
|
| 306 | }
|
| 307 | }
|
| 308 |
|
| 309 | |
| 310 | |
| 311 | |
| 312 | |
| 313 | unsafe extern "C" fn mmap(
|
| 314 | file: *mut bindings::file,
|
| 315 | vma: *mut bindings::vm_area_struct,
|
| 316 | ) -> c_int {
|
| 317 | |
| 318 | let private = unsafe { (*file).private_data };
|
| 319 | |
| 320 | |
| 321 | |
| 322 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private.cast()) };
|
| 323 | |
| 324 | let area = unsafe { VmaNew::from_raw(vma) };
|
| 325 | |
| 326 | |
| 327 | |
| 328 | let file = unsafe { File::from_raw_file(file) };
|
| 329 |
|
| 330 | match T::mmap(device, file, area) {
|
| 331 | Ok(()) => 0,
|
| 332 | Err(err) => err.to_errno(),
|
| 333 | }
|
| 334 | }
|
| 335 |
|
| 336 | |
| 337 | |
| 338 | |
| 339 | unsafe extern "C" fn ioctl(file: *mut bindings::file, cmd: c_uint, arg: c_ulong) -> c_long {
|
| 340 | |
| 341 | let private = unsafe { (*file).private_data };
|
| 342 | |
| 343 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private) };
|
| 344 |
|
| 345 | |
| 346 | |
| 347 | |
| 348 | let file = unsafe { File::from_raw_file(file) };
|
| 349 |
|
| 350 | match T::ioctl(device, file, cmd, arg) {
|
| 351 | Ok(ret) => ret as c_long,
|
| 352 | Err(err) => err.to_errno() as c_long,
|
| 353 | }
|
| 354 | }
|
| 355 |
|
| 356 | |
| 357 | |
| 358 | |
| 359 | #[cfg(CONFIG_COMPAT)]
|
| 360 | unsafe extern "C" fn compat_ioctl(
|
| 361 | file: *mut bindings::file,
|
| 362 | cmd: c_uint,
|
| 363 | arg: c_ulong,
|
| 364 | ) -> c_long {
|
| 365 | |
| 366 | let private = unsafe { (*file).private_data };
|
| 367 | |
| 368 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private) };
|
| 369 |
|
| 370 | |
| 371 | |
| 372 | |
| 373 | let file = unsafe { File::from_raw_file(file) };
|
| 374 |
|
| 375 | match T::compat_ioctl(device, file, cmd, arg) {
|
| 376 | Ok(ret) => ret as c_long,
|
| 377 | Err(err) => err.to_errno() as c_long,
|
| 378 | }
|
| 379 | }
|
| 380 |
|
| 381 | |
| 382 | |
| 383 | |
| 384 | |
| 385 | unsafe extern "C" fn show_fdinfo(seq_file: *mut bindings::seq_file, file: *mut bindings::file) {
|
| 386 | |
| 387 | let private = unsafe { (*file).private_data };
|
| 388 | |
| 389 | let device = unsafe { <T::Ptr as ForeignOwnable>::borrow(private) };
|
| 390 | |
| 391 | |
| 392 | |
| 393 | let file = unsafe { File::from_raw_file(file) };
|
| 394 | |
| 395 | |
| 396 | let m = unsafe { SeqFile::from_raw(seq_file) };
|
| 397 |
|
| 398 | T::show_fdinfo(device, m, file);
|
| 399 | }
|
| 400 |
|
| 401 | const VTABLE: bindings::file_operations = bindings::file_operations {
|
| 402 | open: Some(Self::open),
|
| 403 | release: Some(Self::release),
|
| 404 | mmap: if T::HAS_MMAP { Some(Self::mmap) } else { None },
|
| 405 | read_iter: if T::HAS_READ_ITER {
|
| 406 | Some(Self::read_iter)
|
| 407 | } else {
|
| 408 | None
|
| 409 | },
|
| 410 | write_iter: if T::HAS_WRITE_ITER {
|
| 411 | Some(Self::write_iter)
|
| 412 | } else {
|
| 413 | None
|
| 414 | },
|
| 415 | unlocked_ioctl: if T::HAS_IOCTL {
|
| 416 | Some(Self::ioctl)
|
| 417 | } else {
|
| 418 | None
|
| 419 | },
|
| 420 | #[cfg(CONFIG_COMPAT)]
|
| 421 | compat_ioctl: if T::HAS_COMPAT_IOCTL {
|
| 422 | Some(Self::compat_ioctl)
|
| 423 | } else if T::HAS_IOCTL {
|
| 424 | bindings::compat_ptr_ioctl
|
| 425 | } else {
|
| 426 | None
|
| 427 | },
|
| 428 | show_fdinfo: if T::HAS_SHOW_FDINFO {
|
| 429 | Some(Self::show_fdinfo)
|
| 430 | } else {
|
| 431 | None
|
| 432 | },
|
| 433 | ..pin_init::zeroed()
|
| 434 | };
|
| 435 |
|
| 436 | const fn build() -> &'static bindings::file_operations {
|
| 437 | &Self::VTABLE
|
| 438 | }
|
| 439 | }
|