diff options
Diffstat (limited to 'Src/replicant/nu/AutoBuffer.h')
-rw-r--r-- | Src/replicant/nu/AutoBuffer.h | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/Src/replicant/nu/AutoBuffer.h b/Src/replicant/nu/AutoBuffer.h new file mode 100644 index 00000000..a972f327 --- /dev/null +++ b/Src/replicant/nu/AutoBuffer.h @@ -0,0 +1,49 @@ +#pragma once +#include "foundation/error.h" +#include <stdlib.h> + +/* an automatically growing buffer */ + +class AutoBuffer +{ +public: + AutoBuffer() + { + buffer=0; + buffer_length=0; + } + + ~AutoBuffer() + { + free(buffer); + } + + operator void *() + { + return buffer; + } + + template <class ptr_t> + ptr_t Get() + { + return (ptr_t)buffer; + } + + int Reserve(size_t new_size) + { + if (new_size <= buffer_length) + return NErr_Success; + + void *new_buffer = realloc(buffer, new_size); + if (!new_buffer) + return NErr_OutOfMemory; + + buffer = new_buffer; + buffer_length = new_size; + return NErr_Success; + } + +private: + void *buffer; + size_t buffer_length; +}; |