zeek/tools/binpac/lib/binpac_bytestring.h

#ifndef binpac_bytestring_h
#define binpac_bytestring_h

#include <string.h>
#include <string>

#include "binpac.h"

namespace binpac
	{

template <class T> class datastring;

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

	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_;
	};

typedef const_datastring<uint8> const_bytestring;

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)
		{
		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_ = 0;
		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_;
	};

typedef datastring<uint8> bytestring;

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

inline std::string std_str(const_bytestring const& s)
	{
	return std::string((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