From d542aa0b24d5607ee066a2a50f5e29064c1aaef7 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Wed, 1 Aug 2018 14:15:43 -0400
Subject: ...

---
 include/salticidae/msg.h | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

(limited to 'include/salticidae/msg.h')

diff --git a/include/salticidae/msg.h b/include/salticidae/msg.h
index bc1a633..80b468a 100644
--- a/include/salticidae/msg.h
+++ b/include/salticidae/msg.h
@@ -61,7 +61,8 @@ class MsgBase {
     template<typename MsgType,
             typename = typename std::enable_if<
                 !std::is_same<MsgType, MsgBase>::value &&
-                !std::is_same<MsgType, uint8_t *>::value>::type>
+                !std::is_same<MsgType, bytearray_t>::value &&
+                !std::is_same<MsgType, DataStream>::value>::type>
     MsgBase(const MsgType &msg): magic(0x0) {
         set_opcode(MsgType::opcode);
         set_payload(std::move(msg.serialized));
@@ -87,15 +88,13 @@ class MsgBase {
             payload(std::move(other.payload)),
             no_payload(other.no_payload) {}
 
-    MsgBase(const uint8_t *raw_header): no_payload(true) {
+    MsgBase(DataStream &&s): no_payload(true) {
         uint32_t _magic;
         opcode_t _opcode;
         uint32_t _length;
 #ifndef SALTICIDAE_NOCHECKSUM
         uint32_t _checksum;
 #endif
-        DataStream s(raw_header, raw_header + MsgBase::header_size);
-
         s >> _magic
           >> _opcode
           >> _length
-- 
cgit v1.2.3