From 41dae9b3c7061267fd7048350e895efcdda4c655 Mon Sep 17 00:00:00 2001 From: Liav A Date: Thu, 10 Feb 2022 10:50:37 +0200 Subject: [PATCH] Kernel: Convert i8042 code to use the ErrorOr pattern more broadly Not only does it makes the code more robust and correct as it allows error propagation, it allows us to enforce timeouts on waiting loops so we don't hang forever, by waiting for the i8042 controller to respond to us. Therefore, it makes the i8042 more resilient against faulty hardware and bad behaving chipsets out there. --- Kernel/Devices/HID/HIDManagement.cpp | 16 ++- Kernel/Devices/HID/HIDManagement.h | 2 +- Kernel/Devices/HID/I8042Controller.cpp | 133 ++++++++++++++----------- Kernel/Devices/HID/I8042Controller.h | 47 +++++---- 4 files changed, 117 insertions(+), 81 deletions(-) diff --git a/Kernel/Devices/HID/HIDManagement.cpp b/Kernel/Devices/HID/HIDManagement.cpp index adfd30ecb4..873e5f8070 100644 --- a/Kernel/Devices/HID/HIDManagement.cpp +++ b/Kernel/Devices/HID/HIDManagement.cpp @@ -99,30 +99,36 @@ void HIDManagement::set_maps(NonnullOwnPtr character_map_name, Keyboard dbgln("New Character map '{}' passed in by client.", m_character_map_name); } -UNMAP_AFTER_INIT void HIDManagement::enumerate() +UNMAP_AFTER_INIT ErrorOr HIDManagement::enumerate() { // FIXME: When we have USB HID support, we should ensure that we disable // emulation of the PS/2 controller if it was set by the BIOS. // If ACPI indicates we have an i8042 controller and the USB controller was // set to emulate PS/2, we should not initialize the PS/2 controller. if (kernel_command_line().disable_ps2_controller()) - return; + return {}; if (ACPI::Parser::the() && !ACPI::Parser::the()->have_8042()) - return; + return {}; m_i8042_controller = I8042Controller::initialize(); - m_i8042_controller->detect_devices(); + + // Note: If we happen to not have i8042 just return "gracefully" for now. + if (!m_i8042_controller->check_existence({})) + return {}; + TRY(m_i8042_controller->detect_devices()); if (m_i8042_controller->mouse()) m_hid_devices.append(m_i8042_controller->mouse().release_nonnull()); if (m_i8042_controller->keyboard()) m_hid_devices.append(m_i8042_controller->keyboard().release_nonnull()); + return {}; } UNMAP_AFTER_INIT void HIDManagement::initialize() { VERIFY(!s_the.is_initialized()); s_the.ensure_instance(); - s_the->enumerate(); + // FIXME: Propagate errors back to init to deal with them. + MUST(s_the->enumerate()); } HIDManagement& HIDManagement::the() diff --git a/Kernel/Devices/HID/HIDManagement.h b/Kernel/Devices/HID/HIDManagement.h index d4caf53782..5201fb9891 100644 --- a/Kernel/Devices/HID/HIDManagement.h +++ b/Kernel/Devices/HID/HIDManagement.h @@ -37,7 +37,7 @@ public: static void initialize(); static HIDManagement& the(); - void enumerate(); + ErrorOr enumerate(); StringView keymap_name() const { return m_character_map_name->view(); } Keyboard::CharacterMapData const& character_map() const { return m_character_map; } diff --git a/Kernel/Devices/HID/I8042Controller.cpp b/Kernel/Devices/HID/I8042Controller.cpp index 9bec82f5f1..8b5b0379d1 100644 --- a/Kernel/Devices/HID/I8042Controller.cpp +++ b/Kernel/Devices/HID/I8042Controller.cpp @@ -31,7 +31,7 @@ UNMAP_AFTER_INIT I8042Controller::I8042Controller() { } -UNMAP_AFTER_INIT bool I8042Controller::check_existence() +UNMAP_AFTER_INIT bool I8042Controller::check_existence(Badge) { { SpinlockLocker lock(m_lock); @@ -50,59 +50,60 @@ UNMAP_AFTER_INIT bool I8042Controller::check_existence() } } -UNMAP_AFTER_INIT void I8042Controller::detect_devices() +UNMAP_AFTER_INIT ErrorOr I8042Controller::detect_devices() { - if (!check_existence()) - return; u8 configuration; { SpinlockLocker lock(m_lock); - drain_output_buffer(); + TRY(drain_output_buffer()); - do_wait_then_write(I8042Port::Command, I8042Command::DisableFirstPS2Port); - do_wait_then_write(I8042Port::Command, I8042Command::DisableSecondPS2Port); // ignored if it doesn't exist + TRY(do_wait_then_write(I8042Port::Command, I8042Command::DisableFirstPS2Port)); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::DisableSecondPS2Port)); // ignored if it doesn't exist - do_wait_then_write(I8042Port::Command, I8042Command::ReadConfiguration); - configuration = do_wait_then_read(I8042Port::Buffer); - do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::ReadConfiguration)); + configuration = TRY(do_wait_then_read(I8042Port::Buffer)); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration)); configuration &= ~I8042ConfigurationFlag::FirstPS2PortInterrupt; configuration &= ~I8042ConfigurationFlag::SecondPS2PortInterrupt; - do_wait_then_write(I8042Port::Buffer, configuration); + TRY(do_wait_then_write(I8042Port::Buffer, configuration)); m_is_dual_channel = (configuration & I8042ConfigurationFlag::SecondPS2PortClock) != 0; dbgln("I8042: {} channel controller", m_is_dual_channel ? "Dual" : "Single"); // Perform controller self-test - do_wait_then_write(I8042Port::Command, I8042Command::TestPS2Controller); - if (do_wait_then_read(I8042Port::Buffer) == I8042Response::ControllerTestPassed) { + TRY(do_wait_then_write(I8042Port::Command, I8042Command::TestPS2Controller)); + auto self_test_result = TRY(do_wait_then_read(I8042Port::Buffer)); + if (self_test_result == I8042Response::ControllerTestPassed) { // Restore configuration in case the controller reset - do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration); - do_wait_then_write(I8042Port::Buffer, configuration); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration)); + TRY(do_wait_then_write(I8042Port::Buffer, configuration)); } else { dbgln("I8042: Controller self test failed"); } // Test ports and enable them if available - do_wait_then_write(I8042Port::Command, I8042Command::TestFirstPS2Port); - m_first_port_available = (do_wait_then_read(I8042Port::Buffer) == 0); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::TestFirstPS2Port)); + auto first_port_test_result = TRY(do_wait_then_read(I8042Port::Buffer)); + m_first_port_available = (first_port_test_result == 0); if (m_first_port_available) { - do_wait_then_write(I8042Port::Command, I8042Command::EnableFirstPS2Port); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::EnableFirstPS2Port)); configuration |= I8042ConfigurationFlag::FirstPS2PortInterrupt; configuration &= ~I8042ConfigurationFlag::FirstPS2PortClock; } else { dbgln("I8042: Keyboard port not available"); } - drain_output_buffer(); + TRY(drain_output_buffer()); if (m_is_dual_channel) { - do_wait_then_write(I8042Port::Command, I8042Command::TestSecondPS2Port); - m_second_port_available = (do_wait_then_read(I8042Port::Buffer) == 0); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::TestSecondPS2Port)); + auto test_second_port_result = TRY(do_wait_then_read(I8042Port::Buffer)); + m_second_port_available = (test_second_port_result == 0); if (m_second_port_available) { - do_wait_then_write(I8042Port::Command, I8042Command::EnableSecondPS2Port); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::EnableSecondPS2Port)); configuration |= I8042ConfigurationFlag::SecondPS2PortInterrupt; configuration &= ~I8042ConfigurationFlag::SecondPS2PortClock; } else { @@ -114,8 +115,8 @@ UNMAP_AFTER_INIT void I8042Controller::detect_devices() if (m_first_port_available || m_second_port_available) { configuration &= ~I8042ConfigurationFlag::FirstPS2PortClock; configuration &= ~I8042ConfigurationFlag::SecondPS2PortClock; - do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration); - do_wait_then_write(I8042Port::Buffer, configuration); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration)); + TRY(do_wait_then_write(I8042Port::Buffer, configuration)); } } @@ -128,8 +129,8 @@ UNMAP_AFTER_INIT void I8042Controller::detect_devices() configuration &= ~I8042ConfigurationFlag::FirstPS2PortInterrupt; configuration |= I8042ConfigurationFlag::FirstPS2PortClock; SpinlockLocker lock(m_lock); - do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration); - do_wait_then_write(I8042Port::Buffer, configuration); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration)); + TRY(do_wait_then_write(I8042Port::Buffer, configuration)); } } if (m_second_port_available) { @@ -141,8 +142,8 @@ UNMAP_AFTER_INIT void I8042Controller::detect_devices() m_second_port_available = false; configuration |= I8042ConfigurationFlag::SecondPS2PortClock; SpinlockLocker lock(m_lock); - do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration); - do_wait_then_write(I8042Port::Buffer, configuration); + TRY(do_wait_then_write(I8042Port::Command, I8042Command::WriteConfiguration)); + TRY(do_wait_then_write(I8042Port::Buffer, configuration)); } } } @@ -152,6 +153,7 @@ UNMAP_AFTER_INIT void I8042Controller::detect_devices() m_keyboard_device->enable_interrupts(); if (m_mouse_device) m_mouse_device->enable_interrupts(); + return {}; } bool I8042Controller::irq_process_input_buffer(HIDDevice::Type instrument_type) @@ -175,29 +177,39 @@ bool I8042Controller::irq_process_input_buffer(HIDDevice::Type instrument_type) return false; } -void I8042Controller::drain_output_buffer() +ErrorOr I8042Controller::drain_output_buffer() { - for (;;) { + for (int attempt = 0; attempt < 5; attempt++) { u8 status = IO::in8(I8042Port::Status); if (!(status & I8042StatusFlag::OutputBuffer)) - return; + return {}; IO::in8(I8042Port::Buffer); + + IO::delay(100); } + return Error::from_errno(EBUSY); } -bool I8042Controller::do_reset_device(HIDDevice::Type device) +ErrorOr I8042Controller::do_reset_device(HIDDevice::Type device) { VERIFY(device != HIDDevice::Type::Unknown); VERIFY(m_lock.is_locked()); VERIFY(!Processor::current_in_irq()); - if (do_send_command(device, I8042Command::Reset) != I8042Response::Acknowledge) - return false; + auto reset_result = TRY(do_send_command(device, I8042Command::Reset)); + // FIXME: Is this the correct errno value for this? + if (reset_result != I8042Response::Acknowledge) + return Error::from_errno(EIO); // Wait until we get the self-test result - return do_wait_then_read(I8042Port::Buffer) == I8042Response::Success; + auto self_test_result = TRY(do_wait_then_read(I8042Port::Buffer)); + + // FIXME: Is this the correct errno value for this? + if (self_test_result != I8042Response::Success) + return Error::from_errno(EIO); + return {}; } -u8 I8042Controller::do_send_command(HIDDevice::Type device, u8 command) +ErrorOr I8042Controller::do_send_command(HIDDevice::Type device, u8 command) { VERIFY(device != HIDDevice::Type::Unknown); VERIFY(m_lock.is_locked()); @@ -207,20 +219,20 @@ u8 I8042Controller::do_send_command(HIDDevice::Type device, u8 command) return do_write_to_device(device, command); } -u8 I8042Controller::do_send_command(HIDDevice::Type device, u8 command, u8 data) +ErrorOr I8042Controller::do_send_command(HIDDevice::Type device, u8 command, u8 data) { VERIFY(device != HIDDevice::Type::Unknown); VERIFY(m_lock.is_locked()); VERIFY(!Processor::current_in_irq()); - u8 response = do_write_to_device(device, command); + u8 response = TRY(do_write_to_device(device, command)); if (response == I8042Response::Acknowledge) - response = do_write_to_device(device, data); + response = TRY(do_write_to_device(device, data)); return response; } -u8 I8042Controller::do_write_to_device(HIDDevice::Type device, u8 data) +ErrorOr I8042Controller::do_write_to_device(HIDDevice::Type device, u8 data) { VERIFY(device != HIDDevice::Type::Unknown); VERIFY(m_lock.is_locked()); @@ -231,50 +243,56 @@ u8 I8042Controller::do_write_to_device(HIDDevice::Type device, u8 data) u8 response; do { if (device != HIDDevice::Type::Keyboard) { - prepare_for_output(); + TRY(prepare_for_output()); IO::out8(I8042Port::Command, I8042Command::WriteSecondPS2PortInputBuffer); } - prepare_for_output(); + TRY(prepare_for_output()); IO::out8(I8042Port::Buffer, data); - response = do_wait_then_read(I8042Port::Buffer); + response = TRY(do_wait_then_read(I8042Port::Buffer)); } while (response == I8042Response::Resend && ++attempts < 3); if (attempts >= 3) dbgln("Failed to write byte to device, gave up"); return response; } -u8 I8042Controller::do_read_from_device(HIDDevice::Type device) +ErrorOr I8042Controller::do_read_from_device(HIDDevice::Type device) { VERIFY(device != HIDDevice::Type::Unknown); - prepare_for_input(device); + TRY(prepare_for_input(device)); return IO::in8(I8042Port::Buffer); } -void I8042Controller::prepare_for_input(HIDDevice::Type device) +ErrorOr I8042Controller::prepare_for_input(HIDDevice::Type device) { VERIFY(m_lock.is_locked()); u8 const second_port_flag = device == HIDDevice::Type::Keyboard ? 0 : I8042StatusFlag::SecondPS2PortOutputBuffer; - for (;;) { + for (int attempt = 0; attempt < 5; attempt++) { u8 status = IO::in8(I8042Port::Status); - if (!(status & I8042StatusFlag::OutputBuffer)) + if (!(status & I8042StatusFlag::OutputBuffer)) { + IO::delay(100); continue; + } if (device == HIDDevice::Type::Unknown) - return; + return {}; if ((status & I8042StatusFlag::SecondPS2PortOutputBuffer) == second_port_flag) - return; + return {}; + IO::delay(100); } + return Error::from_errno(EBUSY); } -void I8042Controller::prepare_for_output() +ErrorOr I8042Controller::prepare_for_output() { VERIFY(m_lock.is_locked()); - for (;;) { + for (int attempt = 0; attempt < 5; attempt++) { u8 status = IO::in8(I8042Port::Status); if (!(status & I8042StatusFlag::InputBuffer)) - return; + return {}; + IO::delay(100); } + return Error::from_errno(EBUSY); } UNMAP_AFTER_INIT void I8042Controller::do_write(u8 port, u8 data) @@ -289,17 +307,18 @@ UNMAP_AFTER_INIT u8 I8042Controller::do_read(u8 port) return IO::in8(port); } -void I8042Controller::do_wait_then_write(u8 port, u8 data) +ErrorOr I8042Controller::do_wait_then_write(u8 port, u8 data) { VERIFY(m_lock.is_locked()); - prepare_for_output(); + TRY(prepare_for_output()); IO::out8(port, data); + return {}; } -u8 I8042Controller::do_wait_then_read(u8 port) +ErrorOr I8042Controller::do_wait_then_read(u8 port) { VERIFY(m_lock.is_locked()); - prepare_for_input(HIDDevice::Type::Unknown); + TRY(prepare_for_input(HIDDevice::Type::Unknown)); return IO::in8(port); } diff --git a/Kernel/Devices/HID/I8042Controller.h b/Kernel/Devices/HID/I8042Controller.h index da3d4c3851..98b85a9e85 100644 --- a/Kernel/Devices/HID/I8042Controller.h +++ b/Kernel/Devices/HID/I8042Controller.h @@ -81,6 +81,7 @@ protected: class PS2KeyboardDevice; class PS2MouseDevice; +class HIDManagement; class I8042Controller : public RefCounted { friend class PS2KeyboardDevice; friend class PS2MouseDevice; @@ -88,66 +89,76 @@ class I8042Controller : public RefCounted { public: static NonnullRefPtr initialize(); - void detect_devices(); + ErrorOr detect_devices(); bool reset_device(HIDDevice::Type device) { SpinlockLocker lock(m_lock); - return do_reset_device(device); + // FIXME: Propagate errors properly + if (auto result = do_reset_device(device); result.is_error()) + return false; + return true; } u8 send_command(HIDDevice::Type device, u8 command) { SpinlockLocker lock(m_lock); - return do_send_command(device, command); + // FIXME: Propagate errors properly + return MUST(do_send_command(device, command)); } u8 send_command(HIDDevice::Type device, u8 command, u8 data) { SpinlockLocker lock(m_lock); - return do_send_command(device, command, data); + // FIXME: Propagate errors properly + return MUST(do_send_command(device, command, data)); } u8 read_from_device(HIDDevice::Type device) { SpinlockLocker lock(m_lock); - return do_read_from_device(device); + // FIXME: Propagate errors properly + return MUST(do_read_from_device(device)); } void wait_then_write(u8 port, u8 data) { SpinlockLocker lock(m_lock); - do_wait_then_write(port, data); + // FIXME: Propagate errors properly + MUST(do_wait_then_write(port, data)); } u8 wait_then_read(u8 port) { SpinlockLocker lock(m_lock); - return do_wait_then_read(port); + // FIXME: Propagate errors properly + return MUST(do_wait_then_read(port)); } - void prepare_for_output(); - void prepare_for_input(HIDDevice::Type); + ErrorOr prepare_for_output(); + ErrorOr prepare_for_input(HIDDevice::Type); bool irq_process_input_buffer(HIDDevice::Type); RefPtr mouse() const; RefPtr keyboard() const; + // Note: This function exists only for the initialization process of the controller + bool check_existence(Badge); + private: I8042Controller(); - bool do_reset_device(HIDDevice::Type); - u8 do_send_command(HIDDevice::Type type, u8 data); - u8 do_send_command(HIDDevice::Type device, u8 command, u8 data); - u8 do_write_to_device(HIDDevice::Type device, u8 data); - u8 do_read_from_device(HIDDevice::Type device); - void do_wait_then_write(u8 port, u8 data); - u8 do_wait_then_read(u8 port); - void drain_output_buffer(); + ErrorOr do_reset_device(HIDDevice::Type); + ErrorOr do_send_command(HIDDevice::Type type, u8 data); + ErrorOr do_send_command(HIDDevice::Type device, u8 command, u8 data); + ErrorOr do_write_to_device(HIDDevice::Type device, u8 data); + ErrorOr do_read_from_device(HIDDevice::Type device); + ErrorOr do_wait_then_write(u8 port, u8 data); + ErrorOr do_wait_then_read(u8 port); + ErrorOr drain_output_buffer(); // Note: These functions exist only for the initialization process of the controller void do_write(u8 port, u8 data); u8 do_read(u8 port); - bool check_existence(); Spinlock m_lock; bool m_first_port_available { false };