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