msg_serialize.h

Go to the documentation of this file.

 
 
#ifndef _MSG_SERIALIZE_H
#define _MSG_SERIALIZE_H
 
#include <stdint.h>
#include <string.h>
#include <type_traits>
#include <typeinfo>
#include <iostream>
#include <memory>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <string>
 
namespace serialize_traits
{
    template <typename T>
    struct is_shared_ptr : std::false_type {};
 
    template <typename T>
    struct is_shared_ptr<std::shared_ptr<T>> : std::true_type {};
 
    template <typename U>
    struct is_unsupported_container : std::false_type {};
}
 
// Define to enable compile-time checking of unsupported container types
// Undefine for less compile-time include file dependencies
#define CHECK_UNSUPPORTED_CONTAINER
#ifdef CHECK_UNSUPPORTED_CONTAINER
 
#include <deque>
#include <forward_list>
#include <queue>
#include <stack>
#include <unordered_set>
#include <unordered_map>
 
    template <typename U>
    struct serialize_traits::is_unsupported_container<std::multiset<U>> : std::true_type {};
 
    template <typename U, typename V>
    struct serialize_traits::is_unsupported_container<std::pair<U, V>> : std::true_type {};
 
    template <typename T, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::deque<T, Alloc>> : std::true_type {};
 
    template <typename T, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::forward_list<T, Alloc>> : std::true_type {};
 
    template <typename T, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::priority_queue<T, std::vector<T, Alloc>>> : std::true_type {};
 
    template <typename T, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::queue<T, Alloc>> : std::true_type {};
 
    template <typename T, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::stack<T, Alloc>> : std::true_type {};
 
    template <typename T, typename Hash, typename KeyEqual, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::unordered_multiset<T, Hash, KeyEqual, Alloc>> : std::true_type {};
 
    template <typename Key, typename T, typename Hash, typename KeyEqual, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::unordered_map<Key, T, Hash, KeyEqual, Alloc>> : std::true_type {};
 
    template <typename Key, typename T, typename Hash, typename KeyEqual, typename Alloc>
    struct serialize_traits::is_unsupported_container<std::unordered_multimap<Key, T, Hash, KeyEqual, Alloc>> : std::true_type {};
#endif  // CHECK_UNSUPPORTED_CONTAINER
 
class serialize
{
public:
    class I
    {
    public:
        virtual std::ostream& write(serialize& ms, std::ostream& os) = 0;
 
        virtual std::istream& read(serialize& ms, std::istream& is) = 0;
    };
 
    enum class Type 
    {
        UNKNOWN = 0,
        LITERAL = 1,
        STRING = 8,
        WSTRING = 9,
        VECTOR = 20,
        MAP = 21,
        LIST = 22,
        SET = 23,
        ENDIAN = 30,
        USER_DEFINED = 31,
    };
 
    enum class ParsingError
    {
        NONE,
        TYPE_MISMATCH,
        STREAM_ERROR,
        STRING_TOO_LONG,
        CONTAINER_TOO_MANY,
        INVALID_INPUT,
        END_OF_FILE
    };
 
    serialize() = default;
    ~serialize() = default;
 
    bool LE()
    {        
        const static  int n = 1;
        const static  bool le= (* (char *)&n == 1);
        return le;
    }
    
    std::istream& readEndian(std::istream& is, bool& littleEndian)
    {
        if (read_type(is, Type::ENDIAN))
        {
            is.read((char*) &littleEndian, sizeof(littleEndian));
        }
        return is;
    }
    
    void writeEndian(std::ostream& os)
    {
        bool littleEndian = LE();        
        write_type(os, Type::ENDIAN);
        os.write((const char*) &littleEndian, sizeof(littleEndian));
    }
    
    std::istream& read (std::istream& is, I* t_)
    {
        if (check_stop_parse(is))
            return is;
 
        if (check_pointer(is, t_))
        {
            if (read_type(is, Type::USER_DEFINED))
            {
                uint16_t size = 0;
                std::streampos startPos = is.tellg();
 
                read(is, size, false);
 
                // Save the stop parsing position to prevent parsing overrun
                push_stop_parse_pos(startPos + std::streampos(size));
 
                t_->read(*this, is);
 
                pop_stop_parse_pos();
 
                if (is.good())
                {
                    std::streampos endPos = is.tellg();
                    uint16_t rcvdSize = static_cast<uint16_t>(endPos - startPos);
 
                    // Did sender send a larger object than what receiver parsed? 
                    if (rcvdSize < size)
                    {
                        // Skip over the extra received data
                        uint16_t seekOffset = size - rcvdSize; 
                        is.seekg(seekOffset, std::ios_base::cur);
                    }
                }
                return is;
            }
        }
        return is;
    }
 
    std::istream& read (std::istream& is, std::string& s)
    {
        if (check_stop_parse(is))
            return is;
 
        if (read_type(is, Type::STRING))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_slength(is, size))
            {
                s.resize(size);
                parseStatus(typeid(s), s.size());
                read_internal(is, const_cast<char*>(s.c_str()), size, true);
            }
        }
        return  is;
    }
    
    std::istream& read (std::istream& is, std::wstring& s)
    {
        if (check_stop_parse(is))
            return is;
 
        if (read_type(is, Type::WSTRING))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_slength(is, size))
            {
                s.resize(size);
                parseStatus(typeid(s), s.size());
                for (uint16_t ii = 0; ii < size; ii++)
                {
                    wchar_t c;
                    int offset = sizeof(wchar_t) - WCHAR_SIZE;
                    read_internal(is, reinterpret_cast<char*>(&c) + offset, WCHAR_SIZE);
                    s[ii] = c;
                }
            }
        }
        return  is;
    }
 
    std::istream& read (std::istream& is, char* str)
    {
        if (check_stop_parse(is))
            return is;
 
        if (read_type(is, Type::STRING))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_slength(is, size))
            {
                if (check_pointer(is, str))
                {
                    parseStatus(typeid(str), size);
                    read_internal(is, str, size, true);
                }
            }
        }
        return  is;
    }
    
    std::istream& read (std::istream& is, std::vector<bool>& container)
    {
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::VECTOR))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    bool t;
                    read(is, t);
                    container.push_back(t);
                }
            }
        }
        return is;
    }
    
    std::ostream& write (std::ostream& os, I* t_)
    {
        if (check_pointer(os, t_))
        {
            uint16_t elementSize = 0;
 
            write_type(os, Type::USER_DEFINED);
            std::streampos elementSizePos = os.tellp();
            write(os, elementSize, false);
 
            // Write user defined object
            t_->write(*this, os);
 
            if (os.good())
            {
                // Write user defined object size into stream
                std::streampos currentPos = os.tellp();
                os.seekp(elementSizePos);
                elementSize = static_cast<uint16_t>(currentPos - elementSizePos);
                write(os, elementSize, false);
                os.seekp(currentPos);
            }
            return os;
        }
        return os;
    }
 
    std::ostream& write(std::ostream& os, const std::string& s)
    {
        uint16_t size = static_cast<uint16_t>(s.size());
        write_type(os, Type::STRING);
        write(os, size, false);
        if (check_stream(os) && check_slength(os, size))
        {
            write_internal(os, s.c_str(), size, true);
        }
        return os;
    }
 
    std::ostream& write(std::ostream& os, std::string& s)
    {
        return write(os, static_cast<const std::string&>(s));
    }
 
    std::ostream& write (std::ostream& os, const std::wstring& s)
    {
        uint16_t size = static_cast<uint16_t>(s.size());
        write_type(os, Type::WSTRING);
        write(os, size, false);
        if (check_stream(os) && check_slength(os, size))
        {
            for (uint16_t ii = 0; ii < size; ii++)
            {
                wchar_t c = s[ii];
                int offset = sizeof(wchar_t) - WCHAR_SIZE;
                write_internal(os, reinterpret_cast<char*>(&c) + offset, WCHAR_SIZE);
            }
        }
        return os;
    }
 
    std::ostream& write (std::ostream& os, std::wstring& s)
    {
        return write(os, static_cast<const std::wstring&>(s));
    }
 
    std::ostream& write(std::ostream& os, char* str)
    {
        return write(os, static_cast<const char*>(str));
    }
    
    std::ostream& write (std::ostream& os, const char* str)
    {
        if (check_pointer(os, str))
        {
            uint16_t size = static_cast<uint16_t>(strlen(str)) + 1;
            write_type(os, Type::STRING);
            write(os, size, false);
            if (check_stream(os) && check_slength(os, size))
            {
                write_internal (os, str, size, true);
            }
        }
        return os;
    }
    
    std::ostream& write (std::ostream& os, std::vector<bool>& container)
    {
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::VECTOR);
        write(os, size, false);
        if (check_stream(os) && check_container_size(os, size))
        {
            for (const bool& c : container) 
            {
                write(os, c);
            }
        }
        return os;
    }
 
    template<typename T>
    std::istream& read(std::istream& is, T &t_, bool readPrependedType = true)
    {   
        static_assert(!serialize_traits::is_unsupported_container<T>::value, "Unsupported C++ container type");
 
        static_assert(!(std::is_pointer<T>::value &&
            (std::is_arithmetic<typename std::remove_pointer<T>::type>::value ||
                std::is_class<typename std::remove_pointer<T>::type>::value)),
            "T cannot be a pointer to a built-in or custom data type");
 
        if (check_stop_parse(is))
            return is;
 
        // Is T a built-in data type (e.g. float, int, ...)?
        if (std::is_class<T>::value == false)
        {
            // Is T is not a pointer type
            if (std::is_pointer<T>::value == false)
            {
                if (readPrependedType)
                {
                    if(!read_type(is, Type::LITERAL))
                    {
                        return is;
                    }
                }
 
                if (readPrependedType)
                    parseStatus(typeid(t_));
                read_internal(is, (char*)&t_, sizeof (t_));
                return is;
            }
            else
            {
                // Can't read pointers to built-in type
                raiseError(ParsingError::INVALID_INPUT, __LINE__, __FILE__);
                is.setstate(std::ios::failbit);
                return is;
            }
        }
        // Else T is a user defined data type (e.g. MyData)
        else
        {
            parseStatus(typeid(t_));
            read(is, (serialize::I*)&t_);
            return is;
        }
    }
 
    template<typename T>
    std::ostream& write(std::ostream& os, T &t_, bool prependType = true)
    {
        static_assert(!serialize_traits::is_unsupported_container<T>::value, "Unsupported C++ container type");
 
        static_assert(!(std::is_pointer<T>::value &&
            (std::is_arithmetic<typename std::remove_pointer<T>::type>::value ||
                std::is_class<typename std::remove_pointer<T>::type>::value)),
            "T cannot be a pointer to a built-in or custom data type");
 
        // If C++17 or higher
#if defined(_MSVC_LANG) && _MSVC_LANG >= 201703L || __cplusplus >= 201703L 
        // Check if T is a user-defined class
        if constexpr (std::is_class<T>::value)
        {
            // Ensure that the class T inherits from serialize::I
            static_assert(std::is_base_of<serialize::I, T>::value, "T must inherit from serialize::I");
 
            // Proceed with serialization for user-defined type
            return write(os, (serialize::I*)&t_);
        }
        else
        {
            // If T is not a class, handle built-in data types
            if (std::is_pointer<T>::value == false)
            {
                if (prependType)
                {
                    write_type(os, Type::LITERAL);
                }
                return write_internal(os, (const char*)&t_, sizeof(t_));
            }
            else
            {
                // Can't write pointers to built-in types
                raiseError(ParsingError::INVALID_INPUT, __LINE__, __FILE__);
                os.setstate(std::ios::failbit);
                return os;
            }
        }
#else
        // Is T type a built-in data type (e.g. float, int, ...)?
        if (std::is_class<T>::value == false)
        {
            // Is T is not a pointer type
            if (std::is_pointer<T>::value == false)
            {
                if (prependType)
                {
                    write_type(os, Type::LITERAL);
                }
                return write_internal(os, (const char*)&t_, sizeof(t_));
            }
            else
            {
                // Can't write pointers to built-in type
                raiseError(ParsingError::INVALID_INPUT, __LINE__, __FILE__);
                os.setstate(std::ios::failbit);
                return os;
            }
        }
        // Else T type is a user defined data type (e.g. MyData)
        else
        {
            return write(os, (serialize::I*)&t_);
        }
#endif
    }
 
    template <class T>
    std::ostream& write(std::ostream& os, std::vector<T>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::VECTOR);
        write(os, size, false);
        if (check_stream(os) && check_container_size(os, size))
        {
            for (const auto& item : container) 
            {
                write(os, item, false);
            }
        }
        return os;
    }
 
    template <class T>
    std::istream& read(std::istream& is, std::vector<T>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::VECTOR))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    T t;
                    read(is, t, false);
                    container.push_back(t);
                }
            }
        }
        return is;
    }
 
    template <class T>
    std::ostream& write(std::ostream& os, std::vector<T*>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::VECTOR);
        write(os, size, false);
 
        if (check_stream(os) && check_container_size(os, size))
        {
            for (auto* ptr : container) 
            {
                if (ptr != nullptr) 
                {
                    bool notNULL = true;
                    write(os, notNULL, false);
 
                    auto* i = static_cast<I*>(ptr);
                    write(os, i);
                }
                else
                {
                    bool notNULL = false;
                    write(os, notNULL, false);
                }
            }
 
        }
        return os;
    }
 
    template <class T>
    std::istream& read(std::istream& is, std::vector<T*>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::VECTOR))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    bool notNULL = false;
                    read(is, notNULL, false);
 
                    if (notNULL)
                    {
                        T *object = new T;
                        auto *i = static_cast<I*>(object);
                        read(is, i);
                        container.push_back(object);
                    }
                    else
                    {
                        container.push_back(nullptr);
                    }
                }
            }
        }
        return is;
    }
 
    template <class K, class V, class P>
    std::ostream& write(std::ostream& os, std::map<K, V, P>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<V>::value, "Type V must not be a shared_ptr type");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::MAP);
        write(os, size, false);
        if (check_stream(os) && check_container_size(os, size))
        {
            for (const auto& entry : container) 
            {
                write(os, entry.first, false);
                write(os, entry.second, false);
            }
        }
        return os;
    }
 
    template <class K, class V, class P>
    std::istream& read(std::istream& is, std::map<K, V, P>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<V>::value, "Type V must not be a shared_ptr type");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::MAP))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    K key;
                    V value;
                    read(is, key, false);
 
                    read(is, value, false);
                    container[key] = value;
                }
            }
        }
        return is;
    }
 
    template <class K, class V, class P>
    std::ostream& write(std::ostream& os, std::map<K, V*, P>& container)
    {
        static_assert(std::is_base_of<serialize::I, V>::value, "Type V must be derived from serialize::I");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::MAP);
        write(os, size, false);
 
        if (check_stream(os) && check_container_size(os, size))
        {
            for (auto& entry : container) 
            {
                write(os, entry.first, false);
 
                if (entry.second != nullptr) 
                {
                    bool notNULL = true;
                    write(os, notNULL, false);
 
                    auto* i = static_cast<I*>(entry.second);
                    write(os, i);
                }
                else 
                {
                    bool notNULL = false;
                    write(os, notNULL, false);
                }
            }
        }
        return os;
    }
 
    template <class K, class V, class P>
    std::istream& read(std::istream& is, std::map<K, V*, P>& container)
    {
        static_assert(std::is_base_of<serialize::I, V>::value, "Type V must be derived from serialize::I");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::MAP))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    K key;
                    read(is, key, false);
                    bool notNULL;
                    read(is, notNULL, false);
                    if (notNULL)
                    {
                        V *object = new V;
                        auto *i = static_cast<I*>(object);
                        read(is, i);
                        container[key] = (V*) object;
                    }
                    else
                    {
                        container[key] = nullptr;
                    }
                }
            }
        }
        return is;
    }
 
    template <class T, class P>
    std::ostream& write(std::ostream& os, std::set<T, P>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::SET);
        write(os, size, false);
 
        if (check_stream(os) && check_container_size(os, size))
        {
            for (const auto& item : container) 
            {
                write(os, item, false);
            }
        }
        return os;
    }
 
    template <class T, class P>
    std::istream& read(std::istream& is, std::set<T, P>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::SET))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    T t;
                    read(is, t, false);
                    container.insert(t);
                }
            }
        }
        return is;
    }
 
    template <class T, class P>
    std::ostream& write(std::ostream& os, std::set<T*, P>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::SET);
        write(os, size, false);
        if (check_stream(os) && check_container_size(os, size))
        {
            for (auto ptr : container) 
            {
                if (ptr != nullptr) 
                {
                    bool notNULL = true;
                    write(os, notNULL, false);
 
                    auto* i = static_cast<I*>(ptr);
                    write(os, i);
                }
                else 
                {
                    bool notNULL = false;
                    write(os, notNULL, false);
                }
            }
        }
        return os;
    }
 
    template <class T, class P>
    std::istream& read(std::istream& is, std::set<T*, P>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::SET))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    bool notNULL = false;
                    read(is, notNULL, false);
                    if (notNULL)
                    {
                        T *object = new T;
                        auto *i = static_cast<I*>(object);
                        read(is, i);
                        container.insert(object);
                    }
                    else
                    {
                        container.insert(nullptr);
                    }
                }
            }
        }
        return is;
    }
 
    template <class T>
    std::ostream& write(std::ostream& os, std::list<T>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::LIST);
        write(os, size, false);
 
        if (check_stream(os) && check_container_size(os, size))
        {
            for (const auto& item : container) 
            {
                write(os, item, false);
            }
        }
        return os;
    }
 
    template <class T>
    std::istream& read(std::istream& is, std::list<T>& container)
    {
        static_assert(!serialize_traits::is_shared_ptr<T>::value, "Type T must not be a shared_ptr type");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::LIST))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    T t;
                    read(is, t, false);
                    container.push_back(t);
                }
            }            
        }
        return is;
    }
 
    template <class T>
    std::ostream& write(std::ostream& os, std::list<T*>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        uint16_t size = static_cast<uint16_t>(container.size());
        write_type(os, Type::LIST);
        write(os, size, false);
 
        if (check_stream(os) && check_container_size(os, size))
        {
            for (auto* ptr : container) 
            {
                if (ptr != nullptr) 
                {
                    bool notNULL = true;
                    write(os, notNULL, false);
 
                    auto* i = static_cast<I*>(ptr);
                    write(os, i);
                }
                else 
                {
                    bool notNULL = false;
                    write(os, notNULL, false);
                }
            }
        }
        return os;
    }
 
    template <class T>
    std::istream& read(std::istream& is, std::list<T*>& container)
    {
        static_assert(std::is_base_of<serialize::I, T>::value, "Type T must be derived from serialize::I");
 
        if (check_stop_parse(is))
            return is;
 
        container.clear();
        if (read_type(is, Type::LIST))
        {
            uint16_t size = 0;
            read(is, size, false);
            if (check_stream(is) && check_container_size(is, size))
            {
                parseStatus(typeid(container), size);
                for (uint16_t i = 0; i < size; ++i)
                {
                    bool notNULL = false;
                    read(is, notNULL, false);
                    if (notNULL)
                    {
                        T *object = new T;
                        auto *i = static_cast<I*>(object);
                        read(is, i);
                        container.push_back(object);
                    }
                    else
                    {
                        container.push_back(nullptr);
                    }
                }
            }
        }
        return is;
    }
 
    typedef void (*ErrorHandler)(ParsingError error, int line, const char* file);
    void setErrorHandler(ErrorHandler error_handler_)
    {
        error_handler = error_handler_;
    }
 
    ParsingError getLastError() const { return lastError; }
    void clearLastError() { lastError = ParsingError::NONE; }
 
    typedef void (*ParseHandler)(const std::type_info& typeId, size_t size);
    void setParseHandler(ParseHandler parse_handler_)
    {
        parse_handler = parse_handler_;
    }
 
private:
    std::istream& read_internal(std::istream& is, char* p, uint32_t size, bool no_swap = false)
    {
        if (check_stop_parse(is))
        {
            return is;
        }
 
        if (!check_pointer(is, p))
        {
            return is;
        }
        if (LE() && !no_swap)
        {
            // If little endian, read as little endian
            for (int i = size - 1; i >= 0; --i)
            {
                is.read(p + i, 1);
            }
        }
        else
        {
            // Read as big endian
            is.read(p, size);
        }
 
        return  is;
    }
 
    std::ostream& write_internal(std::ostream& os, const char* p, uint32_t size, bool no_swap = false)
    {
        if (!check_pointer(os, p))
        {
            return os;
        }
        if (LE() && !no_swap)
        {
            // If little endian, write as little endian
            for (int i = size - 1; i >= 0; --i)
            {
                os.write(p + i, 1);
            }
        }
        else
        {
            // Write as big endian
            os.write(p, size);
        }
        return  os;
    }
 
    // Maximum sizes allowed by parser
    static const uint16_t MAX_STRING_SIZE = 256;
    static const uint16_t MAX_CONTAINER_SIZE = 200;
 
    // Keep wchar_t serialize size consistent on any platform
    static const size_t WCHAR_SIZE = 2;
 
    // Used to stop parsing early if not enough data to continue
    std::list<std::streampos> stopParsePosStack;
 
    ErrorHandler error_handler = nullptr;
    ParsingError lastError = ParsingError::NONE;
    void raiseError(ParsingError error, int line, const char* file)
    {
        lastError = error;
        if (error_handler)
            error_handler(error, line, file);
    }
 
    ParseHandler parse_handler = nullptr;
    void parseStatus(const std::type_info& typeId, size_t size = 0)
    {
        if (parse_handler)
            parse_handler(typeId, size);
    }
 
    void write_type(std::ostream& os, Type type_)
    {
        uint8_t type = static_cast<uint8_t>(type_);
        write_internal(os, (const char*) &type, sizeof(type));
    }
 
    bool read_type(std::istream& is, Type type_)
    {
        Type type = static_cast<Type>(is.peek());
        if (type == type_)
        {
            uint8_t typeByte = 0;
            read_internal(is, (char*) &typeByte, sizeof(typeByte));
            return true;
        }
        else
        {
            raiseError(ParsingError::TYPE_MISMATCH, __LINE__, __FILE__);
            is.setstate(std::ios::failbit);
            return false;
        }
    }
 
    bool check_stream(std::ios& stream)
    {
        if (!stream.good())
        {
            raiseError(ParsingError::STREAM_ERROR, __LINE__, __FILE__);
            stream.setstate(std::ios::failbit);
        }
        return stream.good();
    }
 
    bool check_slength(std::ios& stream, int stringSize)
    {
        bool sizeOk = stringSize <= MAX_STRING_SIZE;
        if (!sizeOk)
        {
            raiseError(ParsingError::STRING_TOO_LONG, __LINE__, __FILE__);
            stream.setstate(std::ios::failbit);
        }
        if (stringSize == 0)
            return false;
        return sizeOk;
    }
 
    bool check_container_size(std::ios& stream, int containerSize)
    {
        bool sizeOk = containerSize <= MAX_CONTAINER_SIZE;
        if (!sizeOk)
        {
            raiseError(ParsingError::CONTAINER_TOO_MANY, __LINE__, __FILE__);
            stream.setstate(std::ios::failbit);
        }
        return sizeOk;
    }
 
    bool check_pointer(std::ios& stream, const void* ptr)
    {
        if (!ptr)
        {
            raiseError(ParsingError::INVALID_INPUT, __LINE__, __FILE__);
            stream.setstate(std::ios::failbit);
        }
        return ptr != NULL;
    }
 
    void push_stop_parse_pos(std::streampos stopParsePos)
    {
        stopParsePosStack.push_front(stopParsePos);
    }
 
    std::streampos pop_stop_parse_pos()
    {
        std::streampos stopParsePos = stopParsePosStack.front();
        stopParsePosStack.pop_front();
        return stopParsePos;
    }
 
    bool check_stop_parse(std::istream& is)
    {
        if (is.eof())
        {
            raiseError(ParsingError::END_OF_FILE, __LINE__, __FILE__);
            return true;
        }
        if (stopParsePosStack.size() > 0)
        {
            std::streampos stopParsePos = stopParsePosStack.front();
            if (is.tellg() >= stopParsePos)
            {
                return true;
            }
        }
        return false;
    }
};
 
#endif // _SERIALIZE_H