From 6fd8decb98ce81fd71cebd0226e30b0024ffd77c Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Thu, 1 Feb 2018 20:56:03 +0100
Subject: Initial commit

---
 sibs/SafeDeserializer.hpp | 90 +++++++++++++++++++++++++++++++++++++++++++++++
 sibs/SafeSerializer.hpp   | 44 +++++++++++++++++++++++
 2 files changed, 134 insertions(+)
 create mode 100644 sibs/SafeDeserializer.hpp
 create mode 100644 sibs/SafeSerializer.hpp

(limited to 'sibs')

diff --git a/sibs/SafeDeserializer.hpp b/sibs/SafeDeserializer.hpp
new file mode 100644
index 0000000..3d83042
--- /dev/null
+++ b/sibs/SafeDeserializer.hpp
@@ -0,0 +1,90 @@
+#pragma once
+
+#include "../types.hpp"
+#include "../utils.hpp"
+#include <cstring>
+#include <stdexcept>
+
+namespace sibs
+{
+    class DeserializeException : public std::runtime_error
+    {
+    public:
+        DeserializeException(const std::string &errMsg) : std::runtime_error(errMsg) {}
+    };
+    
+    /**
+     * Endian independent deserializer
+     */
+    class SafeDeserializer
+    {
+        DISABLE_COPY(SafeDeserializer);
+    public:
+        SafeDeserializer(const u8 *_data, usize _size) : 
+            data(_data),
+            size(_size)
+        {
+            
+        }
+        
+        /*
+         * Throws DeserializeException on failure
+         */
+        template <typename T>
+        T&& extract()
+        {
+            constexpr usize typeSize = sizeof(T);
+            verifyExtractSize(typeSize);
+            size -= typeSize;
+            T result;
+            memcpy(&result, data, typeSize);
+            data += typeSize;
+            return std::move(result);
+        }
+        
+        /*
+         * Throws DeserializeException on failure
+         */
+        void extract(u8 *destination, usize destinationSize)
+        {
+            if(destinationSize > 0)
+            {
+                verifyExtractSize(destinationSize);
+                size -= destinationSize;
+                memcpy(destination, data, destinationSize);
+                data += destinationSize;
+            }
+        }
+        
+        bool empty() const
+        {
+            return size == 0;
+        }
+        
+        const u8* getBuffer()
+        {
+            return data;
+        }
+        
+        usize getSize() const
+        {
+            return size;
+        }
+    private:
+        void verifyExtractSize(usize typeSize) const
+        {
+            if(typeSize > size)
+            {
+                std::string errMsg = "Unable to extract ";
+                errMsg += std::to_string(typeSize);
+                errMsg += " bytes, only ";
+                errMsg += std::to_string(size);
+                errMsg += " bytes left in buffer";
+                throw DeserializeException(errMsg);
+            }
+        }
+    private:
+        const u8 *data;
+        usize size;
+    };
+}
diff --git a/sibs/SafeSerializer.hpp b/sibs/SafeSerializer.hpp
new file mode 100644
index 0000000..94bcdc5
--- /dev/null
+++ b/sibs/SafeSerializer.hpp
@@ -0,0 +1,44 @@
+#pragma once
+
+#include "../types.hpp"
+#include "../utils.hpp"
+#include <vector>
+#include <cstring>
+
+namespace sibs
+{
+    /**
+     * Endian independent serializer
+     */
+    class SafeSerializer
+    {
+        DISABLE_COPY(SafeSerializer);
+    public:
+        SafeSerializer(){}
+        
+        template <typename T>
+        void add(const T &data)
+        {
+            usize offset = buffer.size();
+            buffer.resize(buffer.size() + sizeof(data));
+            memcpy(&buffer[offset], &data, sizeof(data));
+        }
+        
+        void add(const u8 *data, usize size)
+        {
+            if(size > 0)
+            {
+                usize offset = buffer.size();
+                buffer.resize(buffer.size() + size);
+                memcpy(&buffer[offset], data, size);
+            }
+        }
+        
+        std::vector<u8>& getBuffer()
+        {
+            return buffer;
+        }
+    private:
+        std::vector<u8> buffer;
+    };
+}
-- 
cgit v1.2.3