/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* vim: set ts=2 et sw=2 tw=80: */ /* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this file, * You can obtain one at http://mozilla.org/MPL/2.0/. */ #ifndef databuffer_h__ #define databuffer_h__ #include <algorithm> #include <cstring> #include <iomanip> #include <iostream> namespace nss_test { class DataBuffer { public: DataBuffer() : data_(nullptr), len_(0) {} DataBuffer(const uint8_t* data, size_t len) : data_(nullptr), len_(0) { Assign(data, len); } DataBuffer(const DataBuffer& other) : data_(nullptr), len_(0) { Assign(other); } ~DataBuffer() { delete[] data_; } DataBuffer& operator=(const DataBuffer& other) { if (&other != this) { Assign(other); } return *this; } void Allocate(size_t len) { delete[] data_; data_ = new uint8_t[len ? len : 1]; // Don't depend on new [0]. len_ = len; } void Truncate(size_t len) { len_ = (std::min)(len_, len); } void Assign(const DataBuffer& other) { Assign(other.data(), other.len()); } void Assign(const uint8_t* data, size_t len); // Write will do a new allocation and expand the size of the buffer if needed. // Returns the offset of the end of the write. size_t Write(size_t index, const uint8_t* val, size_t count); size_t Write(size_t index, const DataBuffer& buf) { return Write(index, buf.data(), buf.len()); } // Write an integer, also performing host-to-network order conversion. // Returns the offset of the end of the write. size_t Write(size_t index, uint32_t val, size_t count); // Starting at |index|, remove |remove| bytes and replace them with the // contents of |buf|. void Splice(const DataBuffer& buf, size_t index, size_t remove = 0) { Splice(buf.data(), buf.len(), index, remove); } void Splice(const uint8_t* ins, size_t ins_len, size_t index, size_t remove = 0); void Append(const DataBuffer& buf) { Splice(buf, len_); } bool Read(size_t index, size_t count, uint64_t* val) const; bool Read(size_t index, size_t count, uint32_t* val) const; const uint8_t* data() const { return data_; } uint8_t* data() { return data_; } size_t len() const { return len_; } bool empty() const { return len_ == 0; } static void SetLogLimit(size_t limit); friend std::ostream& operator<<(std::ostream& stream, const DataBuffer& buf); private: static size_t logging_limit; uint8_t* data_; size_t len_; }; inline std::ostream& operator<<(std::ostream& stream, const DataBuffer& buf) { stream << "[" << buf.len() << "] "; for (size_t i = 0; i < buf.len(); ++i) { if (i >= DataBuffer::logging_limit) { stream << "..."; break; } stream << std::hex << std::setfill('0') << std::setw(2) << static_cast<unsigned>(buf.data()[i]); } stream << std::dec; return stream; } inline bool operator==(const DataBuffer& a, const DataBuffer& b) { return (a.empty() && b.empty()) || (a.len() == b.len() && 0 == memcmp(a.data(), b.data(), a.len())); } inline bool operator!=(const DataBuffer& a, const DataBuffer& b) { return !(a == b); } } // namespace nss_test #endif