mirror of
https://github.com/hasherezade/bearparser
synced 2026-06-08 14:32:38 +00:00
hasherezade
103 lines
3.5 KiB
C++
103 lines
3.5 KiB
C++
#include "ByteBuffer.h"
|
|
#include <iostream>
|
|
|
|
ByteBuffer::ByteBuffer(bufsize_t v_size, bufsize_t v_padding)
|
|
: content(nullptr), contentSize(0), padding(v_padding),
|
|
originalSize(0)
|
|
{
|
|
if (v_size == 0) throw BufferException("Zero size requested");
|
|
|
|
this->content = allocContent(v_size, v_padding);
|
|
this->contentSize = v_size;
|
|
this->originalSize = v_size;
|
|
}
|
|
|
|
ByteBuffer::ByteBuffer(BYTE *v_content, bufsize_t v_size, bufsize_t v_padding)
|
|
: content(nullptr), contentSize(0), padding(v_padding),
|
|
originalSize(0)
|
|
{
|
|
if (v_size == 0) throw BufferException("Zero size requested");
|
|
|
|
this->content = allocContent(v_size, v_padding);
|
|
if (this->content) {
|
|
this->contentSize = v_size;
|
|
this->originalSize = v_size;
|
|
::memcpy(this->content, v_content, v_size);
|
|
}
|
|
}
|
|
|
|
ByteBuffer::ByteBuffer(AbstractByteBuffer *v_parent, offset_t v_offset, bufsize_t v_size, bufsize_t v_padding)
|
|
: content(NULL), contentSize(0), padding(0),
|
|
originalSize(0)
|
|
{
|
|
if (!v_parent) throw BufferException("Cannot make subBuffer for NULL buffer!");
|
|
if (!v_size) throw BufferException("Cannot make 0 size buffer!");
|
|
|
|
bufsize_t parentSize = v_parent->getContentSize();
|
|
|
|
bufsize_t copySize = v_size < parentSize ? v_size : parentSize;
|
|
bufsize_t allocSize = v_size > parentSize ? v_size : parentSize;
|
|
|
|
BYTE *bContent = v_parent->getContentAt(v_offset, copySize);
|
|
if (!bContent) throw BufferException("Cannot make Buffer for NULL content!");
|
|
|
|
this->content = allocContent(allocSize, v_padding);
|
|
if (this->content) {
|
|
this->contentSize = allocSize;
|
|
this->originalSize = this->contentSize;
|
|
::memcpy(this->content, bContent, copySize);
|
|
}
|
|
}
|
|
|
|
BYTE* ByteBuffer::allocContent(bufsize_t v_size, bufsize_t v_padding)
|
|
{
|
|
if (!v_size) throw BufferException("Zero size requested");
|
|
|
|
const bufsize_t allocSize = v_size + v_padding;
|
|
std::cerr << "Trying to resize. Ptr: " << std::hex << (void*)content << " New size: " << allocSize << std::endl;
|
|
|
|
const bufsize_t sizeDiff = (allocSize > this->contentSize) ? (allocSize - this->contentSize) : 0;
|
|
BYTE* content = reinterpret_cast<BYTE*>(::realloc((void*)this->content, allocSize));
|
|
|
|
if (!content) {
|
|
std::cerr << "Error!" << std::endl;
|
|
throw BufferException("Cannot allocate buffer of size: 0x" + QString::number(allocSize, 16));
|
|
}
|
|
if (sizeDiff) {
|
|
std::cerr << "Ptr: " << std::hex << (void*)content << " Additional size: " << sizeDiff << " BaseContentSize: " << this->contentSize << std::endl;
|
|
::memset(content + this->contentSize, 0, sizeDiff);
|
|
} else {
|
|
std::cerr << "Ptr: " << std::hex << (void*)content << " New size: " << allocSize << std::endl;
|
|
}
|
|
/*if (sizeDiff) {
|
|
// if some memory has been added to the existing buffer, initialize it with 0
|
|
::memset(content + this->contentSize, 0, sizeDiff);
|
|
}*/
|
|
return content;
|
|
}
|
|
|
|
bool ByteBuffer::resize(bufsize_t newSize)
|
|
{
|
|
if (newSize == this->contentSize) {
|
|
return true;
|
|
}
|
|
BYTE *newContent = nullptr;
|
|
bool isOk = true;
|
|
try {
|
|
newContent = allocContent(newSize, this->padding);
|
|
if (newContent) {
|
|
this->content = newContent;
|
|
this->contentSize = newSize;
|
|
}
|
|
} catch (const BufferException&) {
|
|
isOk = false;
|
|
}
|
|
return isOk;
|
|
}
|
|
|
|
ByteBuffer::~ByteBuffer()
|
|
{
|
|
std::cerr << "Ptr: " << std::hex << (void*)content << " BaseContentSize: " << this->contentSize << " : " << __FUNCTION__ << std::endl;
|
|
::free(this->content);
|
|
}
|