1a381e482SR0ronoa use core::hint::spin_loop; 2a381e482SR0ronoa 3a381e482SR0ronoa use alloc::{ 4a381e482SR0ronoa string::ToString, 5a381e482SR0ronoa sync::{Arc, Weak}, 6e2841179SLoGin vec::Vec, 7a381e482SR0ronoa }; 8a381e482SR0ronoa use kdepends::ringbuffer::{AllocRingBuffer, RingBuffer}; 9*2eab6dd7S曾俊 use log::{debug, error}; 10a381e482SR0ronoa use system_error::SystemError; 11a381e482SR0ronoa 12a381e482SR0ronoa use crate::{ 13a381e482SR0ronoa arch::{io::PortIOArch, CurrentIrqArch, CurrentPortIOArch}, 14a381e482SR0ronoa driver::{ 15a381e482SR0ronoa base::{ 16a381e482SR0ronoa class::Class, 17a381e482SR0ronoa device::{ 18a381e482SR0ronoa bus::Bus, device_manager, device_number::DeviceNumber, driver::Driver, Device, 19a381e482SR0ronoa DeviceType, IdTable, 20a381e482SR0ronoa }, 21a381e482SR0ronoa kobject::{KObjType, KObject, KObjectState, LockedKObjectState}, 22a381e482SR0ronoa kset::KSet, 23a381e482SR0ronoa }, 249993c0fcSR0ronoa input::{ 259993c0fcSR0ronoa ps2_dev::ps2_device::Ps2Device, 269993c0fcSR0ronoa serio::serio_device::{serio_device_manager, SerioDevice}, 279993c0fcSR0ronoa }, 28a381e482SR0ronoa }, 29a381e482SR0ronoa exception::InterruptArch, 30a381e482SR0ronoa filesystem::{ 31a381e482SR0ronoa devfs::{devfs_register, DevFS, DeviceINode}, 32a381e482SR0ronoa kernfs::KernFSInode, 33a381e482SR0ronoa vfs::{ 341074eb34SSamuel Dai core::generate_inode_id, syscall::ModeType, utils::DName, FilePrivateData, FileSystem, 351074eb34SSamuel Dai FileType, IndexNode, Metadata, 36a381e482SR0ronoa }, 37a381e482SR0ronoa }, 38a381e482SR0ronoa libs::{ 39a381e482SR0ronoa rwlock::{RwLockReadGuard, RwLockWriteGuard}, 40dfe53cf0SGnoCiYeH spinlock::{SpinLock, SpinLockGuard}, 41a381e482SR0ronoa }, 426fc066acSJomo time::PosixTimeSpec, 43a381e482SR0ronoa }; 44a381e482SR0ronoa 45a381e482SR0ronoa static mut PS2_MOUSE_DEVICE: Option<Arc<Ps2MouseDevice>> = None; 46a381e482SR0ronoa 47a381e482SR0ronoa pub fn ps2_mouse_device() -> Option<Arc<Ps2MouseDevice>> { 48a381e482SR0ronoa unsafe { PS2_MOUSE_DEVICE.clone() } 49a381e482SR0ronoa } 50a381e482SR0ronoa 51a381e482SR0ronoa const ADDRESS_PORT_ADDRESS: u16 = 0x64; 52a381e482SR0ronoa const DATA_PORT_ADDRESS: u16 = 0x60; 53a381e482SR0ronoa 54a381e482SR0ronoa const KEYBOARD_COMMAND_ENABLE_PS2_MOUSE_PORT: u8 = 0xa8; 55a381e482SR0ronoa const KEYBOARD_COMMAND_SEND_TO_PS2_MOUSE: u8 = 0xd4; 56a381e482SR0ronoa 57a381e482SR0ronoa const MOUSE_BUFFER_CAPACITY: usize = 15; 58a381e482SR0ronoa 59a381e482SR0ronoa bitflags! { 60a381e482SR0ronoa /// Represents the flags currently set for the mouse. 61a381e482SR0ronoa #[derive(Default)] 62a381e482SR0ronoa pub struct MouseFlags: u8 { 63a381e482SR0ronoa /// Whether or not the left mouse button is pressed. 64a381e482SR0ronoa const LEFT_BUTTON = 0b0000_0001; 65a381e482SR0ronoa 66a381e482SR0ronoa /// Whether or not the right mouse button is pressed. 67a381e482SR0ronoa const RIGHT_BUTTON = 0b0000_0010; 68a381e482SR0ronoa 69a381e482SR0ronoa /// Whether or not the middle mouse button is pressed. 70a381e482SR0ronoa const MIDDLE_BUTTON = 0b0000_0100; 71a381e482SR0ronoa 72a381e482SR0ronoa /// Whether or not the packet is valid or not. 73a381e482SR0ronoa const ALWAYS_ONE = 0b0000_1000; 74a381e482SR0ronoa 75a381e482SR0ronoa /// Whether or not the x delta is negative. 76a381e482SR0ronoa const X_SIGN = 0b0001_0000; 77a381e482SR0ronoa 78a381e482SR0ronoa /// Whether or not the y delta is negative. 79a381e482SR0ronoa const Y_SIGN = 0b0010_0000; 80a381e482SR0ronoa 81a381e482SR0ronoa /// Whether or not the x delta overflowed. 82a381e482SR0ronoa const X_OVERFLOW = 0b0100_0000; 83a381e482SR0ronoa 84a381e482SR0ronoa /// Whether or not the y delta overflowed. 85a381e482SR0ronoa const Y_OVERFLOW = 0b1000_0000; 86a381e482SR0ronoa } 87a381e482SR0ronoa } 88a381e482SR0ronoa 89a381e482SR0ronoa #[derive(Debug)] 90a381e482SR0ronoa enum PsMouseCommand { 91a381e482SR0ronoa SampleRate(u8), 92a381e482SR0ronoa EnablePacketStreaming, 93a381e482SR0ronoa // SetDefaults = 0xF6, 94a381e482SR0ronoa InitKeyboard, 95a381e482SR0ronoa GetMouseId, 96a381e482SR0ronoa SetSampleRate, 97a381e482SR0ronoa } 98a381e482SR0ronoa 99b5b571e0SLoGin impl From<PsMouseCommand> for u8 { 100b5b571e0SLoGin fn from(val: PsMouseCommand) -> Self { 101b5b571e0SLoGin match val { 102b5b571e0SLoGin PsMouseCommand::SampleRate(x) => x, 103b5b571e0SLoGin PsMouseCommand::EnablePacketStreaming => 0xf4, 104b5b571e0SLoGin PsMouseCommand::InitKeyboard => 0x47, 105b5b571e0SLoGin PsMouseCommand::GetMouseId => 0xf2, 106b5b571e0SLoGin PsMouseCommand::SetSampleRate => 0xf3, 107a381e482SR0ronoa } 108a381e482SR0ronoa } 109a381e482SR0ronoa } 110a381e482SR0ronoa 111a381e482SR0ronoa #[derive(Debug)] 112a381e482SR0ronoa pub struct MouseState { 113a381e482SR0ronoa flags: MouseFlags, 114a381e482SR0ronoa x: i16, 115a381e482SR0ronoa y: i16, 116a381e482SR0ronoa } 117a381e482SR0ronoa 118a381e482SR0ronoa #[allow(dead_code)] 119a381e482SR0ronoa impl MouseState { 120a381e482SR0ronoa /// Returns a new `MouseState`. 121a381e482SR0ronoa pub const fn new() -> MouseState { 122a381e482SR0ronoa MouseState { 123a381e482SR0ronoa flags: MouseFlags::empty(), 124a381e482SR0ronoa x: 0, 125a381e482SR0ronoa y: 0, 126a381e482SR0ronoa } 127a381e482SR0ronoa } 128a381e482SR0ronoa 129a381e482SR0ronoa /// Returns true if the left mouse button is currently down. 130a381e482SR0ronoa pub fn left_button_down(&self) -> bool { 131a381e482SR0ronoa self.flags.contains(MouseFlags::LEFT_BUTTON) 132a381e482SR0ronoa } 133a381e482SR0ronoa 134a381e482SR0ronoa /// Returns true if the left mouse button is currently up. 135a381e482SR0ronoa pub fn left_button_up(&self) -> bool { 136a381e482SR0ronoa !self.flags.contains(MouseFlags::LEFT_BUTTON) 137a381e482SR0ronoa } 138a381e482SR0ronoa 139a381e482SR0ronoa /// Returns true if the right mouse button is currently down. 140a381e482SR0ronoa pub fn right_button_down(&self) -> bool { 141a381e482SR0ronoa self.flags.contains(MouseFlags::RIGHT_BUTTON) 142a381e482SR0ronoa } 143a381e482SR0ronoa 144a381e482SR0ronoa /// Returns true if the right mouse button is currently up. 145a381e482SR0ronoa pub fn right_button_up(&self) -> bool { 146a381e482SR0ronoa !self.flags.contains(MouseFlags::RIGHT_BUTTON) 147a381e482SR0ronoa } 148a381e482SR0ronoa 149a381e482SR0ronoa /// Returns true if the x axis has moved. 150a381e482SR0ronoa pub fn x_moved(&self) -> bool { 151a381e482SR0ronoa self.x != 0 152a381e482SR0ronoa } 153a381e482SR0ronoa 154a381e482SR0ronoa /// Returns true if the y axis has moved. 155a381e482SR0ronoa pub fn y_moved(&self) -> bool { 156a381e482SR0ronoa self.y != 0 157a381e482SR0ronoa } 158a381e482SR0ronoa 159a381e482SR0ronoa /// Returns true if the x or y axis has moved. 160a381e482SR0ronoa pub fn moved(&self) -> bool { 161a381e482SR0ronoa self.x_moved() || self.y_moved() 162a381e482SR0ronoa } 163a381e482SR0ronoa 164a381e482SR0ronoa /// Returns the x delta of the mouse state. 165a381e482SR0ronoa pub fn get_x(&self) -> i16 { 166a381e482SR0ronoa self.x 167a381e482SR0ronoa } 168a381e482SR0ronoa 169a381e482SR0ronoa /// Returns the y delta of the mouse state. 170a381e482SR0ronoa pub fn get_y(&self) -> i16 { 171a381e482SR0ronoa self.y 172a381e482SR0ronoa } 173a381e482SR0ronoa } 174a381e482SR0ronoa 175a381e482SR0ronoa #[derive(Debug)] 176a381e482SR0ronoa #[cast_to([sync] Device, SerioDevice)] 177a381e482SR0ronoa pub struct Ps2MouseDevice { 178a381e482SR0ronoa inner: SpinLock<InnerPs2MouseDevice>, 179a381e482SR0ronoa kobj_state: LockedKObjectState, 180a381e482SR0ronoa } 181a381e482SR0ronoa 182a381e482SR0ronoa impl Ps2MouseDevice { 183a381e482SR0ronoa pub const NAME: &'static str = "psmouse"; 184a381e482SR0ronoa pub fn new() -> Self { 185a381e482SR0ronoa let r = Self { 186a381e482SR0ronoa inner: SpinLock::new(InnerPs2MouseDevice { 187a381e482SR0ronoa bus: None, 188a381e482SR0ronoa class: None, 189a381e482SR0ronoa driver: None, 190a381e482SR0ronoa kern_inode: None, 191a381e482SR0ronoa parent: None, 192a381e482SR0ronoa kset: None, 193a381e482SR0ronoa kobj_type: None, 194a381e482SR0ronoa current_packet: 0, 195a381e482SR0ronoa current_state: MouseState::new(), 196a381e482SR0ronoa buf: AllocRingBuffer::new(MOUSE_BUFFER_CAPACITY), 197a381e482SR0ronoa devfs_metadata: Metadata { 198a381e482SR0ronoa dev_id: 1, 199a381e482SR0ronoa inode_id: generate_inode_id(), 200a381e482SR0ronoa size: 4096, 201a381e482SR0ronoa blk_size: 0, 202a381e482SR0ronoa blocks: 0, 2036fc066acSJomo atime: PosixTimeSpec::default(), 2046fc066acSJomo mtime: PosixTimeSpec::default(), 2056fc066acSJomo ctime: PosixTimeSpec::default(), 206a381e482SR0ronoa file_type: FileType::CharDevice, // 文件夹,block设备,char设备 207a381e482SR0ronoa mode: ModeType::from_bits_truncate(0o644), 208a381e482SR0ronoa nlinks: 1, 209a381e482SR0ronoa uid: 0, 210a381e482SR0ronoa gid: 0, 211a381e482SR0ronoa raw_dev: DeviceNumber::default(), // 这里用来作为device number 212a381e482SR0ronoa }, 213a381e482SR0ronoa device_inode_fs: None, 214a381e482SR0ronoa }), 215a381e482SR0ronoa kobj_state: LockedKObjectState::new(None), 216a381e482SR0ronoa }; 217a381e482SR0ronoa return r; 218a381e482SR0ronoa } 219a381e482SR0ronoa 220a381e482SR0ronoa pub fn init(&self) -> Result<(), SystemError> { 221a381e482SR0ronoa let _irq_guard = unsafe { CurrentIrqArch::save_and_disable_irq() }; 222a381e482SR0ronoa 223a381e482SR0ronoa self.write_control_port(KEYBOARD_COMMAND_ENABLE_PS2_MOUSE_PORT)?; 224a381e482SR0ronoa for _i in 0..1000 { 225a381e482SR0ronoa for _j in 0..1000 { 226a381e482SR0ronoa spin_loop(); 227a381e482SR0ronoa } 228a381e482SR0ronoa } 229a381e482SR0ronoa self.read_data_port().ok(); 230a381e482SR0ronoa 231a381e482SR0ronoa self.send_command_to_ps2mouse(PsMouseCommand::EnablePacketStreaming) 232a381e482SR0ronoa .map_err(|e| { 233*2eab6dd7S曾俊 error!("ps2 mouse init error: {:?}", e); 234a381e482SR0ronoa e 235a381e482SR0ronoa })?; 236a381e482SR0ronoa self.read_data_port().ok(); 237a381e482SR0ronoa for _i in 0..1000 { 238a381e482SR0ronoa for _j in 0..1000 { 239a381e482SR0ronoa spin_loop(); 240a381e482SR0ronoa } 241a381e482SR0ronoa } 242a381e482SR0ronoa 243a381e482SR0ronoa // self.send_command_to_ps2mouse(PsMouseCommand::InitKeyboard)?; 244a381e482SR0ronoa self.do_send_command(DATA_PORT_ADDRESS as u8, PsMouseCommand::InitKeyboard.into())?; 245a381e482SR0ronoa self.read_data_port().ok(); 246a381e482SR0ronoa for _i in 0..1000 { 247a381e482SR0ronoa for _j in 0..1000 { 248a381e482SR0ronoa spin_loop(); 249a381e482SR0ronoa } 250a381e482SR0ronoa } 251a381e482SR0ronoa 252a381e482SR0ronoa self.set_sample_rate(20)?; 253a381e482SR0ronoa // self.get_mouse_id()?; 254a381e482SR0ronoa Ok(()) 255a381e482SR0ronoa } 256a381e482SR0ronoa 257a381e482SR0ronoa #[allow(dead_code)] 258a381e482SR0ronoa pub fn get_mouse_id(&self) -> Result<(), SystemError> { 259a381e482SR0ronoa self.send_command_to_ps2mouse(PsMouseCommand::GetMouseId)?; 260a381e482SR0ronoa let _mouse_id = self.read_data_port()?; 261a381e482SR0ronoa Ok(()) 262a381e482SR0ronoa } 263a381e482SR0ronoa 264a381e482SR0ronoa /// 设置鼠标采样率 265a381e482SR0ronoa /// 266a381e482SR0ronoa /// `hz` 合法值为 10,20,40,60,80,100,200 267a381e482SR0ronoa pub fn set_sample_rate(&self, hz: u8) -> Result<(), SystemError> { 268a381e482SR0ronoa const SAMPLE_RATE: [u8; 7] = [10, 20, 40, 60, 80, 100, 200]; 269a381e482SR0ronoa if !SAMPLE_RATE.contains(&hz) { 270a381e482SR0ronoa return Err(SystemError::EINVAL); 271a381e482SR0ronoa } 272a381e482SR0ronoa 273a381e482SR0ronoa self.send_command_to_ps2mouse(PsMouseCommand::SetSampleRate)?; 274a381e482SR0ronoa self.read_data_port().ok(); 275a381e482SR0ronoa for _i in 0..1000 { 276a381e482SR0ronoa for _j in 0..1000 { 277a381e482SR0ronoa spin_loop(); 278a381e482SR0ronoa } 279a381e482SR0ronoa } 280a381e482SR0ronoa 281a381e482SR0ronoa self.send_command_to_ps2mouse(PsMouseCommand::SampleRate(hz))?; 282a381e482SR0ronoa for _i in 0..1000 { 283a381e482SR0ronoa for _j in 0..1000 { 284a381e482SR0ronoa spin_loop(); 285a381e482SR0ronoa } 286a381e482SR0ronoa } 287a381e482SR0ronoa self.read_data_port().ok(); 288a381e482SR0ronoa Ok(()) 289a381e482SR0ronoa } 290a381e482SR0ronoa 291a381e482SR0ronoa /// # 函数的功能 292a381e482SR0ronoa /// 鼠标设备处理数据包 293a381e482SR0ronoa pub fn process_packet(&self) -> Result<(), SystemError> { 294a381e482SR0ronoa let packet = self.read_data_port()?; 295a381e482SR0ronoa let mut guard = self.inner.lock(); 296a381e482SR0ronoa guard.buf.push(packet); // 更新缓冲区 297a381e482SR0ronoa match guard.current_packet { 298a381e482SR0ronoa 0 => { 299a381e482SR0ronoa let flags: MouseFlags = MouseFlags::from_bits_truncate(packet); 300a381e482SR0ronoa if !flags.contains(MouseFlags::ALWAYS_ONE) { 301a381e482SR0ronoa return Ok(()); 302a381e482SR0ronoa } 303a381e482SR0ronoa guard.current_state.flags = flags; 304a381e482SR0ronoa } 305a381e482SR0ronoa 1 => { 306b5b571e0SLoGin let flags = guard.current_state.flags; 307a381e482SR0ronoa if !flags.contains(MouseFlags::X_OVERFLOW) { 308a381e482SR0ronoa guard.current_state.x = self.get_x_movement(packet, flags); 309a381e482SR0ronoa } 310a381e482SR0ronoa } 311a381e482SR0ronoa 2 => { 312b5b571e0SLoGin let flags = guard.current_state.flags; 313a381e482SR0ronoa if !flags.contains(MouseFlags::Y_OVERFLOW) { 314a381e482SR0ronoa guard.current_state.y = self.get_y_movement(packet, flags); 315a381e482SR0ronoa } 316a381e482SR0ronoa 317*2eab6dd7S曾俊 // debug!( 318a381e482SR0ronoa // "Ps2MouseDevice packet : flags:{}, x:{}, y:{}\n", 319a381e482SR0ronoa // guard.current_state.flags.bits, 320a381e482SR0ronoa // guard.current_state.x, 321a381e482SR0ronoa // guard.current_state.y 3229993c0fcSR0ronoa // ); 323a381e482SR0ronoa } 324a381e482SR0ronoa _ => unreachable!(), 325a381e482SR0ronoa } 326a381e482SR0ronoa guard.current_packet = (guard.current_packet + 1) % 3; 327a381e482SR0ronoa Ok(()) 328a381e482SR0ronoa } 329a381e482SR0ronoa 330a381e482SR0ronoa fn get_x_movement(&self, packet: u8, flags: MouseFlags) -> i16 { 331a381e482SR0ronoa if flags.contains(MouseFlags::X_SIGN) { 332a381e482SR0ronoa return self.sign_extend(packet); 333a381e482SR0ronoa } else { 334a381e482SR0ronoa return packet as i16; 335a381e482SR0ronoa } 336a381e482SR0ronoa } 337a381e482SR0ronoa 338a381e482SR0ronoa fn get_y_movement(&self, packet: u8, flags: MouseFlags) -> i16 { 339a381e482SR0ronoa if flags.contains(MouseFlags::Y_SIGN) { 340a381e482SR0ronoa return self.sign_extend(packet); 341a381e482SR0ronoa } else { 342a381e482SR0ronoa return packet as i16; 343a381e482SR0ronoa } 344a381e482SR0ronoa } 345a381e482SR0ronoa 346a381e482SR0ronoa fn sign_extend(&self, packet: u8) -> i16 { 347a381e482SR0ronoa ((packet as u16) | 0xFF00) as i16 348a381e482SR0ronoa } 349a381e482SR0ronoa 350a381e482SR0ronoa fn read_data_port(&self) -> Result<u8, SystemError> { 351a381e482SR0ronoa self.wait_for_write()?; 352a381e482SR0ronoa let cmd = unsafe { CurrentPortIOArch::in8(ADDRESS_PORT_ADDRESS) }; 353a381e482SR0ronoa if (cmd & 0x21) == 0x21 { 354a381e482SR0ronoa let data = unsafe { CurrentPortIOArch::in8(DATA_PORT_ADDRESS) }; 355a381e482SR0ronoa return Ok(data); 356a381e482SR0ronoa } else { 357a381e482SR0ronoa return Err(SystemError::ENODATA); 358a381e482SR0ronoa } 359a381e482SR0ronoa } 360a381e482SR0ronoa 361a381e482SR0ronoa #[inline(never)] 362a381e482SR0ronoa fn send_command_to_ps2mouse(&self, command: PsMouseCommand) -> Result<(), SystemError> { 363a381e482SR0ronoa self.do_send_command(KEYBOARD_COMMAND_SEND_TO_PS2_MOUSE, command.into())?; 364a381e482SR0ronoa Ok(()) 365a381e482SR0ronoa } 366a381e482SR0ronoa 367a381e482SR0ronoa #[inline(never)] 368a381e482SR0ronoa fn do_send_command(&self, ctrl: u8, command: u8) -> Result<(), SystemError> { 369a381e482SR0ronoa self.write_control_port(ctrl)?; 370a381e482SR0ronoa self.write_data_port(command)?; 371a381e482SR0ronoa return Ok(()); 372a381e482SR0ronoa } 373a381e482SR0ronoa 374a381e482SR0ronoa fn write_data_port(&self, data: u8) -> Result<(), SystemError> { 375a381e482SR0ronoa self.wait_for_write()?; 376a381e482SR0ronoa unsafe { 377a381e482SR0ronoa CurrentPortIOArch::out8(DATA_PORT_ADDRESS, data); 378a381e482SR0ronoa } 379a381e482SR0ronoa Ok(()) 380a381e482SR0ronoa } 381a381e482SR0ronoa 382a381e482SR0ronoa fn write_control_port(&self, command: u8) -> Result<(), SystemError> { 383a381e482SR0ronoa self.wait_for_write()?; 384a381e482SR0ronoa unsafe { 385a381e482SR0ronoa CurrentPortIOArch::out8(ADDRESS_PORT_ADDRESS, command); 386a381e482SR0ronoa } 387a381e482SR0ronoa Ok(()) 388a381e482SR0ronoa } 389a381e482SR0ronoa 390a381e482SR0ronoa fn wait_for_read(&self) -> Result<(), SystemError> { 391a381e482SR0ronoa let timeout = 100_000; 392a381e482SR0ronoa for _ in 0..timeout { 393a381e482SR0ronoa let value = unsafe { CurrentPortIOArch::in8(ADDRESS_PORT_ADDRESS) }; 394a381e482SR0ronoa if (value & 0x1) == 0x1 { 395a381e482SR0ronoa return Ok(()); 396a381e482SR0ronoa } 397a381e482SR0ronoa } 398a381e482SR0ronoa Err(SystemError::ETIMEDOUT) 399a381e482SR0ronoa } 400a381e482SR0ronoa 401a381e482SR0ronoa fn wait_for_write(&self) -> Result<(), SystemError> { 402a381e482SR0ronoa let timeout = 100_000; 403a381e482SR0ronoa for _ in 0..timeout { 404a381e482SR0ronoa let value = unsafe { CurrentPortIOArch::in8(ADDRESS_PORT_ADDRESS) }; 405a381e482SR0ronoa if (value & 0x2) == 0 { 406a381e482SR0ronoa return Ok(()); 407a381e482SR0ronoa } 408a381e482SR0ronoa } 409a381e482SR0ronoa Err(SystemError::ETIMEDOUT) 410a381e482SR0ronoa } 411a381e482SR0ronoa } 412a381e482SR0ronoa 413a381e482SR0ronoa #[derive(Debug)] 414a381e482SR0ronoa struct InnerPs2MouseDevice { 415a381e482SR0ronoa bus: Option<Weak<dyn Bus>>, 4164256da7fSLoGin class: Option<Weak<dyn Class>>, 417a381e482SR0ronoa driver: Option<Weak<dyn Driver>>, 418a381e482SR0ronoa kern_inode: Option<Arc<KernFSInode>>, 419a381e482SR0ronoa parent: Option<Weak<dyn KObject>>, 420a381e482SR0ronoa kset: Option<Arc<KSet>>, 421a381e482SR0ronoa kobj_type: Option<&'static dyn KObjType>, 422a381e482SR0ronoa 423a381e482SR0ronoa /// 鼠标数据 424a381e482SR0ronoa current_state: MouseState, 425a381e482SR0ronoa current_packet: u8, 426a381e482SR0ronoa /// 鼠标数据环形缓冲区 427a381e482SR0ronoa buf: AllocRingBuffer<u8>, 428a381e482SR0ronoa 429a381e482SR0ronoa /// device inode要求的字段 430a381e482SR0ronoa device_inode_fs: Option<Weak<DevFS>>, 431a381e482SR0ronoa devfs_metadata: Metadata, 432a381e482SR0ronoa } 433a381e482SR0ronoa 434a381e482SR0ronoa impl Device for Ps2MouseDevice { 435a381e482SR0ronoa fn is_dead(&self) -> bool { 436a381e482SR0ronoa false 437a381e482SR0ronoa } 438a381e482SR0ronoa 439a381e482SR0ronoa fn dev_type(&self) -> DeviceType { 440a381e482SR0ronoa DeviceType::Char 441a381e482SR0ronoa } 442a381e482SR0ronoa 443a381e482SR0ronoa fn id_table(&self) -> IdTable { 444a381e482SR0ronoa IdTable::new(self.name().to_string(), None) 445a381e482SR0ronoa } 446a381e482SR0ronoa 447a381e482SR0ronoa fn set_bus(&self, bus: Option<alloc::sync::Weak<dyn Bus>>) { 448a381e482SR0ronoa self.inner.lock_irqsave().bus = bus; 449a381e482SR0ronoa } 450a381e482SR0ronoa 4514256da7fSLoGin fn set_class(&self, class: Option<alloc::sync::Weak<dyn Class>>) { 452a381e482SR0ronoa self.inner.lock_irqsave().class = class; 453a381e482SR0ronoa } 454a381e482SR0ronoa 455a381e482SR0ronoa fn driver(&self) -> Option<alloc::sync::Arc<dyn Driver>> { 456a381e482SR0ronoa self.inner.lock_irqsave().driver.clone()?.upgrade() 457a381e482SR0ronoa } 458a381e482SR0ronoa 459a381e482SR0ronoa fn set_driver(&self, driver: Option<alloc::sync::Weak<dyn Driver>>) { 460a381e482SR0ronoa self.inner.lock_irqsave().driver = driver; 461a381e482SR0ronoa } 462a381e482SR0ronoa 463a381e482SR0ronoa fn can_match(&self) -> bool { 464a381e482SR0ronoa true 465a381e482SR0ronoa } 466a381e482SR0ronoa 467a381e482SR0ronoa fn set_can_match(&self, _can_match: bool) {} 468a381e482SR0ronoa 469a381e482SR0ronoa fn state_synced(&self) -> bool { 470a381e482SR0ronoa true 471a381e482SR0ronoa } 472a381e482SR0ronoa 473a381e482SR0ronoa fn bus(&self) -> Option<alloc::sync::Weak<dyn Bus>> { 474a381e482SR0ronoa self.inner.lock_irqsave().bus.clone() 475a381e482SR0ronoa } 476a381e482SR0ronoa 477a381e482SR0ronoa fn class(&self) -> Option<Arc<dyn Class>> { 4784256da7fSLoGin let mut guard = self.inner.lock_irqsave(); 4794256da7fSLoGin let r = guard.class.clone()?.upgrade(); 4804256da7fSLoGin if r.is_none() { 4814256da7fSLoGin guard.class = None; 4824256da7fSLoGin } 4834256da7fSLoGin 4844256da7fSLoGin return r; 485a381e482SR0ronoa } 486a381e482SR0ronoa } 487a381e482SR0ronoa 488a381e482SR0ronoa impl SerioDevice for Ps2MouseDevice { 489a381e482SR0ronoa fn write( 490a381e482SR0ronoa &self, 491a381e482SR0ronoa _device: &alloc::sync::Arc<dyn SerioDevice>, 492a381e482SR0ronoa _data: u8, 493a381e482SR0ronoa ) -> Result<(), system_error::SystemError> { 494a381e482SR0ronoa todo!() 495a381e482SR0ronoa } 496a381e482SR0ronoa 497a381e482SR0ronoa fn open( 498a381e482SR0ronoa &self, 499a381e482SR0ronoa _device: &alloc::sync::Arc<dyn SerioDevice>, 500a381e482SR0ronoa ) -> Result<(), system_error::SystemError> { 501a381e482SR0ronoa todo!() 502a381e482SR0ronoa } 503a381e482SR0ronoa 504a381e482SR0ronoa fn close( 505a381e482SR0ronoa &self, 506a381e482SR0ronoa _device: &alloc::sync::Arc<dyn SerioDevice>, 507a381e482SR0ronoa ) -> Result<(), system_error::SystemError> { 508a381e482SR0ronoa todo!() 509a381e482SR0ronoa } 510a381e482SR0ronoa 511a381e482SR0ronoa fn start( 512a381e482SR0ronoa &self, 513a381e482SR0ronoa _device: &alloc::sync::Arc<dyn SerioDevice>, 514a381e482SR0ronoa ) -> Result<(), system_error::SystemError> { 515a381e482SR0ronoa todo!() 516a381e482SR0ronoa } 517a381e482SR0ronoa 518a381e482SR0ronoa fn stop( 519a381e482SR0ronoa &self, 520a381e482SR0ronoa _device: &alloc::sync::Arc<dyn SerioDevice>, 521a381e482SR0ronoa ) -> Result<(), system_error::SystemError> { 522a381e482SR0ronoa todo!() 523a381e482SR0ronoa } 524a381e482SR0ronoa } 525a381e482SR0ronoa 526a381e482SR0ronoa impl KObject for Ps2MouseDevice { 527a381e482SR0ronoa fn as_any_ref(&self) -> &dyn core::any::Any { 528a381e482SR0ronoa self 529a381e482SR0ronoa } 530a381e482SR0ronoa 531a381e482SR0ronoa fn set_inode(&self, inode: Option<alloc::sync::Arc<KernFSInode>>) { 532a381e482SR0ronoa self.inner.lock_irqsave().kern_inode = inode; 533a381e482SR0ronoa } 534a381e482SR0ronoa 535a381e482SR0ronoa fn inode(&self) -> Option<alloc::sync::Arc<KernFSInode>> { 536a381e482SR0ronoa self.inner.lock_irqsave().kern_inode.clone() 537a381e482SR0ronoa } 538a381e482SR0ronoa 539a381e482SR0ronoa fn parent(&self) -> Option<alloc::sync::Weak<dyn KObject>> { 540a381e482SR0ronoa self.inner.lock_irqsave().parent.clone() 541a381e482SR0ronoa } 542a381e482SR0ronoa 543a381e482SR0ronoa fn set_parent(&self, parent: Option<alloc::sync::Weak<dyn KObject>>) { 544a381e482SR0ronoa self.inner.lock_irqsave().parent = parent 545a381e482SR0ronoa } 546a381e482SR0ronoa 547a381e482SR0ronoa fn kset(&self) -> Option<alloc::sync::Arc<KSet>> { 548a381e482SR0ronoa self.inner.lock_irqsave().kset.clone() 549a381e482SR0ronoa } 550a381e482SR0ronoa 551a381e482SR0ronoa fn set_kset(&self, kset: Option<alloc::sync::Arc<KSet>>) { 552a381e482SR0ronoa self.inner.lock_irqsave().kset = kset; 553a381e482SR0ronoa } 554a381e482SR0ronoa 555a381e482SR0ronoa fn kobj_type(&self) -> Option<&'static dyn KObjType> { 556b5b571e0SLoGin self.inner.lock_irqsave().kobj_type 557a381e482SR0ronoa } 558a381e482SR0ronoa 559a381e482SR0ronoa fn set_kobj_type(&self, ktype: Option<&'static dyn KObjType>) { 560a381e482SR0ronoa self.inner.lock_irqsave().kobj_type = ktype; 561a381e482SR0ronoa } 562a381e482SR0ronoa 563a381e482SR0ronoa fn name(&self) -> alloc::string::String { 564a381e482SR0ronoa Self::NAME.to_string() 565a381e482SR0ronoa } 566a381e482SR0ronoa 567a381e482SR0ronoa fn set_name(&self, _name: alloc::string::String) {} 568a381e482SR0ronoa 569a381e482SR0ronoa fn kobj_state(&self) -> RwLockReadGuard<KObjectState> { 570a381e482SR0ronoa self.kobj_state.read() 571a381e482SR0ronoa } 572a381e482SR0ronoa 573a381e482SR0ronoa fn kobj_state_mut(&self) -> RwLockWriteGuard<KObjectState> { 574a381e482SR0ronoa self.kobj_state.write() 575a381e482SR0ronoa } 576a381e482SR0ronoa 577a381e482SR0ronoa fn set_kobj_state(&self, state: KObjectState) { 578a381e482SR0ronoa *self.kobj_state.write() = state; 579a381e482SR0ronoa } 580a381e482SR0ronoa } 581a381e482SR0ronoa 582a381e482SR0ronoa impl DeviceINode for Ps2MouseDevice { 583a381e482SR0ronoa fn set_fs(&self, fs: Weak<DevFS>) { 584a381e482SR0ronoa self.inner.lock_irqsave().device_inode_fs = Some(fs); 585a381e482SR0ronoa } 586a381e482SR0ronoa } 587a381e482SR0ronoa 588a381e482SR0ronoa impl IndexNode for Ps2MouseDevice { 589a381e482SR0ronoa fn open( 590a381e482SR0ronoa &self, 591dfe53cf0SGnoCiYeH _data: SpinLockGuard<FilePrivateData>, 592a381e482SR0ronoa _mode: &crate::filesystem::vfs::file::FileMode, 593a381e482SR0ronoa ) -> Result<(), SystemError> { 594a381e482SR0ronoa let mut guard = self.inner.lock_irqsave(); 595a381e482SR0ronoa guard.buf.clear(); 596a381e482SR0ronoa Ok(()) 597a381e482SR0ronoa } 598a381e482SR0ronoa 599dfe53cf0SGnoCiYeH fn close(&self, _data: SpinLockGuard<FilePrivateData>) -> Result<(), SystemError> { 600a381e482SR0ronoa let mut guard = self.inner.lock_irqsave(); 601a381e482SR0ronoa guard.buf.clear(); 602a381e482SR0ronoa Ok(()) 603a381e482SR0ronoa } 604a381e482SR0ronoa 605a381e482SR0ronoa fn read_at( 606a381e482SR0ronoa &self, 607a381e482SR0ronoa _offset: usize, 608a381e482SR0ronoa _len: usize, 609a381e482SR0ronoa buf: &mut [u8], 610dfe53cf0SGnoCiYeH _data: SpinLockGuard<FilePrivateData>, 611a381e482SR0ronoa ) -> Result<usize, SystemError> { 612a381e482SR0ronoa let mut guard = self.inner.lock_irqsave(); 613a381e482SR0ronoa 614a381e482SR0ronoa if guard.buf.len() >= 3 { 615b5b571e0SLoGin for item in buf.iter_mut().take(3) { 616b5b571e0SLoGin *item = guard.buf.dequeue().unwrap(); 617a381e482SR0ronoa } 618a381e482SR0ronoa return Ok(3); 619a381e482SR0ronoa } else { 620a381e482SR0ronoa return Ok(0); 621a381e482SR0ronoa } 622a381e482SR0ronoa } 623a381e482SR0ronoa 624a381e482SR0ronoa fn write_at( 625a381e482SR0ronoa &self, 626a381e482SR0ronoa _offset: usize, 627a381e482SR0ronoa _len: usize, 628a381e482SR0ronoa _buf: &[u8], 629dfe53cf0SGnoCiYeH _data: SpinLockGuard<FilePrivateData>, 630a381e482SR0ronoa ) -> Result<usize, SystemError> { 6311074eb34SSamuel Dai return Err(SystemError::ENOSYS); 632a381e482SR0ronoa } 633a381e482SR0ronoa 634a381e482SR0ronoa fn fs(&self) -> Arc<dyn FileSystem> { 635a381e482SR0ronoa self.inner 636a381e482SR0ronoa .lock_irqsave() 637a381e482SR0ronoa .device_inode_fs 638a381e482SR0ronoa .as_ref() 639a381e482SR0ronoa .unwrap() 640a381e482SR0ronoa .upgrade() 641a381e482SR0ronoa .unwrap() 642a381e482SR0ronoa } 643a381e482SR0ronoa 644a381e482SR0ronoa fn as_any_ref(&self) -> &dyn core::any::Any { 645a381e482SR0ronoa self 646a381e482SR0ronoa } 647a381e482SR0ronoa 648e2841179SLoGin fn list(&self) -> Result<Vec<alloc::string::String>, SystemError> { 649a381e482SR0ronoa todo!() 650a381e482SR0ronoa } 651a381e482SR0ronoa 652a381e482SR0ronoa fn metadata(&self) -> Result<Metadata, SystemError> { 653a381e482SR0ronoa Ok(self.inner.lock_irqsave().devfs_metadata.clone()) 654a381e482SR0ronoa } 655a381e482SR0ronoa 656a381e482SR0ronoa fn resize(&self, _len: usize) -> Result<(), SystemError> { 657a381e482SR0ronoa Ok(()) 658a381e482SR0ronoa } 6591074eb34SSamuel Dai 6601074eb34SSamuel Dai fn dname(&self) -> Result<DName, SystemError> { 6611074eb34SSamuel Dai Ok(DName::from(self.name())) 6621074eb34SSamuel Dai } 663a381e482SR0ronoa } 664a381e482SR0ronoa 6659993c0fcSR0ronoa impl Ps2Device for Ps2MouseDevice {} 6669993c0fcSR0ronoa 6679993c0fcSR0ronoa pub fn rs_ps2_mouse_device_init(parent: Arc<dyn KObject>) -> Result<(), SystemError> { 668*2eab6dd7S曾俊 debug!("ps2_mouse_device initializing..."); 669a381e482SR0ronoa let psmouse = Arc::new(Ps2MouseDevice::new()); 670a381e482SR0ronoa 671a381e482SR0ronoa device_manager().device_default_initialize(&(psmouse.clone() as Arc<dyn Device>)); 6729993c0fcSR0ronoa psmouse.set_parent(Some(Arc::downgrade(&parent))); 6739993c0fcSR0ronoa serio_device_manager().register_port(psmouse.clone() as Arc<dyn SerioDevice>)?; 674a381e482SR0ronoa 675a381e482SR0ronoa devfs_register(&psmouse.name(), psmouse.clone()).map_err(|e| { 676*2eab6dd7S曾俊 error!( 677a381e482SR0ronoa "register psmouse device '{}' to devfs failed: {:?}", 678a381e482SR0ronoa psmouse.name(), 679a381e482SR0ronoa e 680a381e482SR0ronoa ); 681a381e482SR0ronoa device_manager().remove(&(psmouse.clone() as Arc<dyn Device>)); 682a381e482SR0ronoa e 683a381e482SR0ronoa })?; 684a381e482SR0ronoa 685a381e482SR0ronoa unsafe { PS2_MOUSE_DEVICE = Some(psmouse) }; 686a381e482SR0ronoa return Ok(()); 687a381e482SR0ronoa } 688