mailboxbomb/source/buffer.h

107 lines
3.1 KiB
C
Raw Normal View History

2015-09-03 05:13:18 +02:00
#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