From 81b59a08b69d4872b06930aedd06a957c4acb7cc Mon Sep 17 00:00:00 2001
From: Aleksi Lindeman <dec05eba@protonmail.com>
Date: Fri, 23 Nov 2018 15:44:08 +0100
Subject: Add data routing for failed p2p connections

---
 include/sibs/Message.hpp | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

(limited to 'include/sibs/Message.hpp')

diff --git a/include/sibs/Message.hpp b/include/sibs/Message.hpp
index b6eb858..4591221 100644
--- a/include/sibs/Message.hpp
+++ b/include/sibs/Message.hpp
@@ -1,7 +1,7 @@
 #pragma once
 
 #include "../types.hpp"
-#include <vector>
+#include <sibs/SafeSerializer.hpp>
 
 namespace sibs
 {
@@ -20,11 +20,11 @@ namespace sibs
 
         void append(const void *data, const usize size);
 
-        usize getDataSize() const { return rawData.size() - 1; }
-        usize getRawSize() const { return rawData.size(); }
+        usize getDataSize() const { return serializer.getSize() - 1; }
+        usize getRawSize() const { return serializer.getSize(); }
 
-        const u8* data() const { return rawData.data(); }
-    private:
-        std::vector<u8> rawData;
+        const u8* data() { return serializer.getBuffer().data(); }
+
+        sibs::SafeSerializer serializer;
     };
 }
\ No newline at end of file
-- 
cgit v1.2.3