mirror of
https://github.com/giantpune/mailboxbomb.git
synced 2024-11-17 17:49:16 +01:00
107 lines
3.1 KiB
C++
107 lines
3.1 KiB
C++
#ifndef BUFFER_H
|
|
#define BUFFER_H
|
|
|
|
#include <string>
|
|
#include "types.h"
|
|
|
|
//quick'n'dirty (tm) buffer class
|
|
// (c) giantpune 2011
|
|
//! uses malloc, so no data alignment is performed
|
|
//! no memory sharing takes place among buffers
|
|
//! no "over allocation" occurs. it is reallocated whenever the size is changed
|
|
class Buffer
|
|
{
|
|
public:
|
|
|
|
//create a Buffer object with the given data up to the first NULL byte or until len
|
|
//! this makes a copy of the data
|
|
explicit Buffer( const void* stuff, u32 size = 0 );
|
|
Buffer();
|
|
Buffer( const Buffer &other );
|
|
|
|
//! creates a buffer with the given size
|
|
Buffer( u32 size );
|
|
Buffer( u32 size, char fillChar );
|
|
~Buffer();
|
|
|
|
//de-allocate memory
|
|
void Free();
|
|
|
|
//returns true if len or ptr is 0
|
|
bool IsEmpty() const { return !( len && ptr ); }
|
|
|
|
//for debugging purposes
|
|
//! hexdump data to printf, from start to end.
|
|
//! if size == 0, it just dumps till the end
|
|
void Dump( u32 start = 0, u32 size = 0 );
|
|
|
|
//number of bytes
|
|
u32 Size() const { return len; }
|
|
|
|
//access functions to the data
|
|
unsigned char* Data() { return (unsigned char*)ptr; }
|
|
const unsigned char* ConstData() const { return (unsigned char*)ptr; }
|
|
|
|
//copies the data to this buffer and returns a reference to itself
|
|
Buffer &SetData( const void* stuff, u32 size = 0 );
|
|
|
|
//add more data to the end
|
|
Buffer &Append( char c );
|
|
Buffer &Append( const void* stuff, u32 size = 0 );
|
|
Buffer &Append( const Buffer &other );
|
|
|
|
//add data to the middle
|
|
Buffer &Insert( u32 pos, char c );
|
|
Buffer &Insert( u32 pos, const void* stuff, u32 size = 0 );
|
|
Buffer &Insert( u32 pos, const Buffer &other );
|
|
|
|
//add data to the start
|
|
Buffer &Prepend( char c ) { return Insert( 0, c ); }
|
|
Buffer &Prepend( const void* stuff, u32 size ) { return Insert( 0, stuff, size ); }
|
|
Buffer &Prepend( const Buffer &other ) { return Insert( 0, other ); }
|
|
|
|
|
|
//change the size of the allocated memory for this buffer
|
|
Buffer &Resize( u32 size );
|
|
|
|
//operators to append data to this buffer and return a reference it itself
|
|
Buffer &operator=(const Buffer &);
|
|
Buffer &operator=( const char* stuff );
|
|
|
|
Buffer &operator+=(char c);
|
|
Buffer &operator+=( const u8* stuff );
|
|
Buffer &operator+=( const char* stuff );
|
|
Buffer &operator+=( const Buffer & );
|
|
|
|
Buffer &operator<<(char c);
|
|
Buffer &operator<<( const u8* stuff );
|
|
Buffer &operator<<( const char* stuff );
|
|
Buffer &operator<<( const Buffer & );
|
|
|
|
Buffer ToHex() const;
|
|
|
|
static Buffer FromHex( const std::string &hexEncoded );
|
|
private:
|
|
void* ptr;
|
|
u32 len;
|
|
};
|
|
inline Buffer &Buffer::operator+=(char c)
|
|
{ return Append( c ); }
|
|
inline Buffer &Buffer::operator+=(const u8* stuff)
|
|
{ return Append( stuff ); }
|
|
inline Buffer &Buffer::operator+=(const char* stuff)
|
|
{ return Append( stuff ); }
|
|
inline Buffer &Buffer::operator+=( const Buffer &other )
|
|
{ return Append( other ); }
|
|
|
|
inline Buffer &Buffer::operator<<(char c)
|
|
{ return Append( c ); }
|
|
inline Buffer &Buffer::operator<<(const char* stuff)
|
|
{ return Append( stuff ); }
|
|
inline Buffer &Buffer::operator<<(const u8* stuff)
|
|
{ return Append( stuff ); }
|
|
inline Buffer &Buffer::operator<<( const Buffer &other )
|
|
{ return Append( other ); }
|
|
|
|
#endif // BUFFER_H
|