1 files changed, 31 insertions, 7 deletions
diff --git a/ByteBuffer.h b/ByteBuffer.h
index a7e9f9f..444943d 100644
--- a/ByteBuffer.h
+++ b/ByteBuffer.h
@@ -3,25 +3,49 @@
 
 #include <cstdint>
 #include <cstdlib>
+#include <string>
+#include <stdexcept>
+
+class ByteBufferException : public std::runtime_error {
+public:
+    ByteBufferException(std::string const &what) : std::runtime_error(what) {
+    }
+};
 
 class ByteBuffer {
 public:
-    ByteBuffer(uint8_t *bytes, ssize_t size, ssize_t zero = 0);
+    ByteBuffer(uint8_t *bytes, size_t capacity, size_t size, size_t zero = 0);
+
+    inline size_t getSize() {
+        return size;
+    }
+
+    inline void setCursor(size_t newCursor) {
+        cursor = newCursor;
+    }
 
     ByteBuffer &add8(uint8_t value);
 
     ByteBuffer &add16le(uint16_t value);
 
-    uint8_t get8(ssize_t index);
+    uint8_t get8(size_t index);
+
+    uint8_t get8();
+
+    uint16_t get16le();
+
+    void copy(uint8_t *bytes, size_t length);
 
 private:
-    void canAccessNextBytes(ssize_t count);
-    void canAccessIndex(ssize_t count);
+    void checkAndUpdateSize(size_t count);
+
+    void canAccessIndex(size_t count);
 
     uint8_t *bytes;
-    ssize_t zero;
-    ssize_t size;
-    ssize_t ptr;
+    size_t zero;
+    size_t size;
+    size_t capacity;
+    size_t cursor;
 };
 
 #endif