buffer.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_BUFFER_HH
#define SPONGE_LIBSPONGE_BUFFER_HH
 
#include <algorithm>
#include <deque>
#include <memory>
#include <numeric>
#include <stdexcept>
#include <string>
#include <string_view>
#include <sys/uio.h>
#include <vector>
 
class Buffer {
  private:
    std::shared_ptr<std::string> _storage{};
    size_t _starting_offset{};
 
  public:
    Buffer() = default;
 
    Buffer(std::string &&str) noexcept : _storage(std::make_shared<std::string>(std::move(str))) {}
 
    std::string_view str() const {
        if (not _storage) {
            return {};
        }
        return {_storage->data() + _starting_offset, _storage->size() - _starting_offset};
    }
 
    operator std::string_view() const { return str(); }
 
    uint8_t at(const size_t n) const { return str().at(n); }
 
    size_t size() const { return str().size(); }
 
    std::string copy() const { return std::string(str()); }
 
    void remove_prefix(const size_t n);
};
 
class BufferList {
  private:
    std::deque<Buffer> _buffers{};
 
  public:
 
    BufferList() = default;
 
    BufferList(Buffer buffer) : _buffers{buffer} {}
 
    BufferList(std::string &&str) noexcept {
        Buffer buf{std::move(str)};
        append(buf);
    }
 
    const std::deque<Buffer> &buffers() const { return _buffers; }
 
    void append(const BufferList &other);
 
    operator Buffer() const;
 
    void remove_prefix(size_t n);
 
    size_t size() const;
 
    std::string concatenate() const;
};
 
class BufferViewList {
    std::deque<std::string_view> _views{};
 
  public:
 
    BufferViewList(const std::string &str) : BufferViewList(std::string_view(str)) {}
 
    BufferViewList(const char *s) : BufferViewList(std::string_view(s)) {}
 
    BufferViewList(const BufferList &buffers);
 
    BufferViewList(std::string_view str) { _views.push_back({const_cast<char *>(str.data()), str.size()}); }
 
    void remove_prefix(size_t n);
 
    size_t size() const;
 
    std::vector<iovec> as_iovecs() const;
};
 
#endif  // SPONGE_LIBSPONGE_BUFFER_HH