ref: 84ff16c0156e71de20af1cf149f26e3433213174
dir: /test/utils/BufferedData.h/
#ifndef __BUFFEREDDATA_H__ #define __BUFFEREDDATA_H__ #include <stddef.h> #include <stdlib.h> #include "../test_stdint.h" #include <algorithm> class BufferedData { public: BufferedData() : data_(NULL), capacity_(0), length_(0) {} ~BufferedData() { free(data_); } bool PushBack(uint8_t c) { if (!EnsureCapacity(length_ + 1)) { return false; } data_[length_++] = c; return true; } bool PushBack(const uint8_t* data, size_t len) { if (!EnsureCapacity(length_ + len)) { return false; } memcpy(data_ + length_, data, len); length_ += len; return true; } size_t PopFront(uint8_t* ptr, size_t len) { len = std::min(length_, len); memcpy(ptr, data_, len); memmove(data_, data_ + len, length_ - len); SetLength(length_ - len); return len; } void Clear() { length_ = 0; } void SetLength(size_t newLen) { if (EnsureCapacity(newLen)) { length_ = newLen; } } size_t Length() const { return length_; } uint8_t* data() { return data_; } private: bool EnsureCapacity(size_t capacity) { if (capacity > capacity_) { size_t newsize = capacity * 2; uint8_t* data = static_cast<uint8_t*>(realloc(data_, newsize)); if (!data) return false; data_ = data; capacity_ = newsize; return true; } return true; } uint8_t* data_; size_t capacity_; size_t length_; }; #endif //__BUFFEREDDATA_H__