1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
#include "buffer.h"
#include <string.h>
#include "exceptions.h"
#include "unicode.h"
Buffer::Buffer(int sz, int alloc)
{
allocated = alloc;
size = sz;
if (size > allocated)
allocated = size;
if (allocated < 1024)
allocated = 1024;
data = malloc(allocated);
if (! data)
throw Exception(L"Error allocating memory for Buffer");
}
Buffer::~Buffer()
{
free(data);
}
void Buffer::setSize(size_t sz)
{
if (sz > allocated) {
int newAl = allocated + sz + 1024;
void *d = realloc(data, newAl);
if (! d)
throw Exception(L"Error expanding buffer memory");
data = d;
allocated = newAl;
}
size = sz;
}
size_t Buffer::getSize()
{
return size;
}
size_t Buffer::getAllocated()
{
return allocated;
}
void* Buffer::getData()
{
return data;
}
void Buffer::gotoPos(int offset)
{
currentPos = offset;
}
size_t Buffer::putData(const unsigned char *d, size_t length)
{
if (size < currentPos + length)
setSize(currentPos + length);
memcpy((unsigned char*)data + currentPos, d, length);
currentPos += length;
return length;
}
size_t Buffer::putInteger(int v)
{
unsigned char b[4];
int i, ib;
for (i = 0; i < 4; i++) {
ib = v & 0xFF;
v = v >> 8;
b[i] = ib;
}
return putData(b, 4);
}
size_t Buffer::putUtf8(const std::wstring &string)
{
std::string s(toUtf8(string));
putInteger(s.length());
putData((const unsigned char*)s.c_str(), s.length());
return 4 + s.length();
}
size_t Buffer::putByte(unsigned char value)
{
if (size < (size_t)currentPos + 1)
setSize(currentPos + 1);
((unsigned char*)data)[currentPos] = value;
currentPos++;
return 1;
}
|