Kernel: Allow USB devices to handle STALL conditions

kernel/include/kernel/USB/Device.h

@@ -21,6 +21,7 @@ namespace Kernel
 
 		virtual BAN::ErrorOr<void> initialize() { return {}; };
 
+		virtual void handle_stall(uint8_t endpoint_id) = 0;
 		virtual void handle_input_data(size_t byte_count, uint8_t endpoint_id) = 0;
 	};
 
@@ -71,6 +72,7 @@ namespace Kernel
 		static USB::SpeedClass determine_speed_class(uint64_t bits_per_second);
 
 	protected:
+		void handle_stall(uint8_t endpoint_id);
 		void handle_input_data(size_t byte_count, uint8_t endpoint_id);
 		virtual BAN::ErrorOr<void> initialize_control_endpoint() = 0;
 

kernel/include/kernel/USB/HID/HIDDriver.h

@@ -75,6 +75,7 @@ namespace Kernel
 		};
 
 	public:
+		void handle_stall(uint8_t endpoint_id) override;
 		void handle_input_data(size_t byte_count, uint8_t endpoint_id) override;
 
 	private:

kernel/include/kernel/USB/MassStorage/MassStorageDriver.h

@@ -15,6 +15,7 @@ namespace Kernel
 		BAN_NON_MOVABLE(USBMassStorageDriver);
 
 	public:
+		void handle_stall(uint8_t endpoint_id) override;
 		void handle_input_data(size_t byte_count, uint8_t endpoint_id) override;
 
 		BAN::ErrorOr<size_t> send_bytes(paddr_t, size_t count);

kernel/kernel/USB/Device.cpp

@@ -328,6 +328,12 @@ namespace Kernel
 		return BAN::move(configuration);
 	}
 
+	void USBDevice::handle_stall(uint8_t endpoint_id)
+	{
+		for (auto& driver : m_class_drivers)
+			driver->handle_stall(endpoint_id);
+	}
+
 	void USBDevice::handle_input_data(size_t byte_count, uint8_t endpoint_id)
 	{
 		for (auto& driver : m_class_drivers)

kernel/kernel/USB/HID/HIDDriver.cpp

@@ -272,6 +272,12 @@ namespace Kernel
 		return BAN::move(result);
 	}
 
+	void USBHIDDriver::handle_stall(uint8_t endpoint_id)
+	{
+		(void)endpoint_id;
+		// FIXME: do something :)
+	}
+
 	void USBHIDDriver::handle_input_data(size_t byte_count, uint8_t endpoint_id)
 	{
 		if (m_data_endpoint_id != endpoint_id)

kernel/kernel/USB/MassStorage/MassStorageDriver.cpp

@@ -169,6 +169,12 @@ namespace Kernel
 		return static_cast<size_t>(bytes_recv);
 	}
 
+	void USBMassStorageDriver::handle_stall(uint8_t endpoint_id)
+	{
+		(void)endpoint_id;
+		// FIXME: do something :)
+	}
+
 	void USBMassStorageDriver::handle_input_data(size_t byte_count, uint8_t endpoint_id)
 	{
 		if (endpoint_id != m_in_endpoint_id && endpoint_id != m_out_endpoint_id)

kernel/kernel/USB/XHCI/Device.cpp

@@ -295,15 +295,19 @@ namespace Kernel
 	void XHCIDevice::on_interrupt_or_bulk_endpoint_event(XHCI::TRB trb)
 	{
 		ASSERT(trb.trb_type == XHCI::TRBType::TransferEvent);
+
+		const uint32_t endpoint_id = trb.transfer_event.endpoint_id;
+		auto& endpoint = m_endpoints[endpoint_id - 1];
+
+		if (trb.transfer_event.completion_code == 6)
+			return handle_stall(endpoint_id);
+
 		if (trb.transfer_event.completion_code != 1 && trb.transfer_event.completion_code != 13)
 		{
 			dwarnln("Interrupt or bulk endpoint got transfer event with completion code {}", +trb.transfer_event.completion_code);
 			return;
 		}
 
-		const uint32_t endpoint_id = trb.transfer_event.endpoint_id;
-		auto& endpoint = m_endpoints[endpoint_id - 1];
-
 		const auto* transfer_trb_arr = reinterpret_cast<volatile XHCI::TRB*>(endpoint.transfer_ring->vaddr());
 		const uint32_t transfer_trb_index = (trb.transfer_event.trb_pointer - endpoint.transfer_ring->paddr()) / sizeof(XHCI::TRB);
 		const uint32_t original_len = transfer_trb_arr[transfer_trb_index].normal.trb_transfer_length;