zeek/tools/binpac/lib/binpac_bytestring.h

// See the file "COPYING" in the main distribution directory for copyright.

#ifndef binpac_bytestring_h
#define binpac_bytestring_h

#include <cstring>
#include <string>

#include "binpac.h"

namespace binpac {

template<class T>
class datastring;

template<class T>
class const_datastring {
public:
    const_datastring() : begin_(nullptr), end_(nullptr) {}

    const_datastring(T const* data, int length) : begin_(data), end_(data + length) {}

    const_datastring(const T* begin, const T* end) : begin_(begin), end_(end) {}

    const_datastring(datastring<T> const& s) : begin_(s.begin()), end_(s.end()) {}

    void init(const T* data, int length) {
        begin_ = data;
        end_ = data + length;
    }

    T const* begin() const { return begin_; }
    T const* end() const { return end_; }
    int length() const { return end_ - begin_; }

    T const& operator[](int index) const { return begin()[index]; }

    bool operator==(const_datastring<T> const& s) {
        if ( length() != s.length() )
            return false;
        return memcmp((const void*)begin(), (const void*)s.begin(), sizeof(T) * length()) == 0;
    }

    void set_begin(T const* begin) { begin_ = begin; }
    void set_end(T const* end) { end_ = end; }

private:
    T const* begin_;
    T const* end_;
};

using const_bytestring = const_datastring<uint8>;

template<class T>
class datastring {
public:
    datastring() { clear(); }

    datastring(T* data, int len) { set(data, len); }

    datastring(T const* begin, T const* end) { set_const(begin, end - begin); }

    datastring(datastring<T> const& x) : data_(x.data()), length_(x.length()) {}

    explicit datastring(const_datastring<T> const& x) { set_const(x.begin(), x.length()); }

    datastring const& operator=(datastring<T> const& x) {
        if ( this == &x )
            return *this;

        BINPAC_ASSERT(! data_);
        set(x.data(), x.length());
        return *this;
    }

    void init(T const* begin, int length) {
        BINPAC_ASSERT(! data_);
        set_const(begin, length);
    }

    void clear() {
        data_ = nullptr;
        length_ = 0;
    }

    void free() {
        if ( data_ )
            delete[] data_;
        clear();
    }

    void clone() { set_const(begin(), length()); }

    datastring const& operator=(const_datastring<T> const& x) {
        BINPAC_ASSERT(! data_);
        set_const(x.begin(), x.length());
        return *this;
    }

    T const& operator[](int index) const { return begin()[index]; }

    T* data() const { return data_; }
    int length() const { return length_; }

    T const* begin() const { return data_; }
    T const* end() const { return data_ + length_; }

private:
    void set(T* data, int len) {
        data_ = data;
        length_ = len;
    }

    void set_const(T const* data, int len) {
        length_ = len;
        data_ = new T[len + 1];
        memcpy(data_, data, sizeof(T) * len);
        data_[len] = 0;
    }

    T* data_;
    int length_;
};

using bytestring = datastring<uint8>;

inline const char* c_str(bytestring const& s) { return (const char*)s.begin(); }

inline std::string std_str(const_bytestring const& s) { return {(const char*)s.begin(), (const char*)s.end()}; }

inline bool operator==(bytestring const& s1, const char* s2) { return strcmp(c_str(s1), s2) == 0; }

inline void get_pointers(const_bytestring const& s, uint8 const** pbegin, uint8 const** pend) {
    *pbegin = s.begin();
    *pend = s.end();
}

inline void get_pointers(bytestring const* s, uint8 const** pbegin, uint8 const** pend) {
    *pbegin = s->begin();
    *pend = s->end();
}

} // namespace binpac

#endif // binpac_bytestring_h