data_buffer.hpp

Go to the documentation of this file.

#ifndef DATA_BUFFER_HPP
#define DATA_BUFFER_HPP
 
#include <stdio.h>
 
#include <cstring>
#include <iostream>
#include <type_traits>
#include <vector>
 
class DataBuffer
{
private:
    std::vector<unsigned char> _buffer;
    mutable size_t             _cursor;
 
public:
    DataBuffer();
    ~DataBuffer() = default;
 
    const std::vector<unsigned char> data() const;
    void                             increaseCursor(size_t amount) const;
    void                             decreaseCursor(size_t amount) const;
 
    void   reset() const;
    void   clear();
    size_t size() const;
 
    void append(const unsigned char* data, size_t len);
 
    DataBuffer&       operator<<(const std::string& value);
    const DataBuffer& operator>>(std::string& value) const;
 
    template <typename T>
    DataBuffer& operator<<(const T& value)
    {
        const unsigned char* ptr = reinterpret_cast<const unsigned char*>(&value);
        _buffer.insert(_buffer.end(), ptr, ptr + sizeof(T));
        return *this;
    }
 
    template <typename T>
    const DataBuffer& operator>>(T& value) const
    {
        if (sizeof(T) + _cursor > _buffer.size())
            throw std::out_of_range("Buffer overflow on read");
 
        std::memcpy(&value, _buffer.data() + _cursor, sizeof(T));
 
        _cursor += sizeof(T);
        return *this;
    }
};
 
#endif