1 files changed, 79 insertions, 0 deletions
diff --git a/Kernel/Bus/VirtIO/Console.h b/Kernel/Bus/VirtIO/Console.h
new file mode 100644
index 0000000000..a9bb01cb50
--- /dev/null
+++ b/Kernel/Bus/VirtIO/Console.h
@@ -0,0 +1,79 @@
+/*
+ * Copyright (c) 2021, Kyle Pereira <hey@xylepereira.me>
+ * Copyright (c) 2021, the SerenityOS developers.
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <Kernel/Bus/VirtIO/ConsolePort.h>
+#include <Kernel/Bus/VirtIO/Device.h>
+#include <Kernel/Memory/RingBuffer.h>
+
+namespace Kernel::VirtIO {
+class Console
+    : public VirtIO::Device
+    , public RefCounted<Console> {
+    friend VirtIO::ConsolePort;
+
+public:
+    Console(PCI::Address);
+    virtual ~Console() override = default;
+
+    virtual StringView purpose() const override { return class_name(); }
+
+    unsigned device_id() const
+    {
+        return m_device_id;
+    }
+
+private:
+    virtual StringView class_name() const override { return "VirtIOConsole"; }
+    enum class ControlEvent : u16 {
+        DeviceReady = 0,
+        DeviceAdd = 1,
+        PortReady = 3,
+        ConsolePort = 4,
+        PortOpen = 6,
+    };
+    struct [[gnu::packed]] ControlMessage {
+        u32 id;
+        u16 event;
+        u16 value;
+
+        enum class Status : u16 {
+            Success = 1,
+            Failure = 0
+        };
+
+        enum class PortStatus : u16 {
+            Open = 1,
+            Close = 0
+        };
+    };
+
+    constexpr static u16 CONTROL_RECEIVEQ = 2;
+    constexpr static u16 CONTROL_TRANSMITQ = 3;
+    constexpr static size_t CONTROL_MESSAGE_SIZE = sizeof(ControlMessage);
+    constexpr static size_t CONTROL_BUFFER_SIZE = CONTROL_MESSAGE_SIZE * 32;
+
+    virtual bool handle_device_config_change() override;
+    virtual void handle_queue_update(u16 queue_index) override;
+
+    Vector<RefPtr<ConsolePort>> m_ports;
+    void setup_multiport();
+    void process_control_message(ControlMessage message);
+    void write_control_message(ControlMessage message);
+    void send_open_control_message(unsigned port_number, bool open);
+
+    unsigned m_device_id;
+
+    OwnPtr<Memory::RingBuffer> m_control_transmit_buffer;
+    OwnPtr<Memory::RingBuffer> m_control_receive_buffer;
+
+    WaitQueue m_control_wait_queue;
+
+    static unsigned next_device_id;
+};
+}