1 files changed, 34 insertions, 0 deletions
diff --git a/lib/binarytools/MemoryStream.h b/lib/binarytools/MemoryStream.h
new file mode 100644
index 0000000..9f38169
--- /dev/null
+++ b/lib/binarytools/MemoryStream.h
@@ -0,0 +1,34 @@
+#pragma once
+
+#include <memory>
+#include <vector>
+#include "Stream.h"
+
+namespace LUS {
+class MemoryStream : public Stream {
+  public:
+    MemoryStream();
+    MemoryStream(char* nBuffer, size_t nBufferSize);
+    ~MemoryStream();
+
+    uint64_t GetLength() override;
+
+    void Seek(int32_t offset, SeekOffsetType seekType) override;
+
+    std::unique_ptr<char[]> Read(size_t length) override;
+    void Read(const char* dest, size_t length) override;
+    int8_t ReadByte() override;
+
+    void Write(char* srcBuffer, size_t length) override;
+    void WriteByte(int8_t value) override;
+
+    std::vector<char> ToVector() override;
+
+    void Flush() override;
+    void Close() override;
+
+  protected:
+    std::vector<char> mBuffer;
+    std::size_t mBufferSize;
+};
+} // namespace LUS