1 files changed, 51 insertions, 0 deletions
diff --git a/Kernel/Bus/USB/USBTransfer.h b/Kernel/Bus/USB/USBTransfer.h
new file mode 100644
index 0000000000..b54d3a73b3
--- /dev/null
+++ b/Kernel/Bus/USB/USBTransfer.h
@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) 2021, Jesse Buhagiar <jooster669@gmail.com>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <AK/OwnPtr.h>
+#include <AK/RefPtr.h>
+#include <Kernel/Bus/USB/PacketTypes.h>
+#include <Kernel/Bus/USB/USBPipe.h>
+#include <Kernel/VM/ContiguousVMObject.h>
+#include <Kernel/VM/PhysicalPage.h>
+#include <Kernel/VM/Region.h>
+
+// TODO: Callback stuff in this class please!
+namespace Kernel::USB {
+
+class Transfer : public RefCounted<Transfer> {
+public:
+    static RefPtr<Transfer> try_create(Pipe& pipe, u16 len);
+
+public:
+    Transfer() = delete;
+    Transfer(Pipe& pipe, u16 len, ContiguousVMObject&);
+    ~Transfer();
+
+    void set_setup_packet(const USBRequestData& request);
+    void set_complete() { m_complete = true; }
+    void set_error_occurred() { m_error_occurred = true; }
+
+    // `const` here makes sure we don't blow up by writing to a physical address
+    const USBRequestData& request() const { return m_request; }
+    const Pipe& pipe() const { return m_pipe; }
+    Pipe& pipe() { return m_pipe; }
+    VirtualAddress buffer() const { return m_data_buffer->vaddr(); }
+    PhysicalAddress buffer_physical() const { return m_data_buffer->physical_page(0)->paddr(); }
+    u16 transfer_data_size() const { return m_transfer_data_size; }
+    bool complete() const { return m_complete; }
+    bool error_occurred() const { return m_error_occurred; }
+
+private:
+    Pipe& m_pipe;                    // Pipe that initiated this transfer
+    USBRequestData m_request;        // USB request
+    OwnPtr<Region> m_data_buffer;    // DMA Data buffer for transaction
+    u16 m_transfer_data_size { 0 };  // Size of the transfer's data stage
+    bool m_complete { false };       // Has this transfer been completed?
+    bool m_error_occurred { false }; // Did an error occur during this transfer?
+};
+}