3a1bc92778
This fixes building the test code on MSVC 2013. (Actually building all of the tests on windows still is tricky due to the OpenSSL dependency for SHA1, but there's a patchset in review to take care of that.)
83 lines
1.4 KiB
C++
83 lines
1.4 KiB
C++
#ifndef __BUFFEREDDATA_H__
|
|
#define __BUFFEREDDATA_H__
|
|
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <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__
|