From 7875db76986b79904f76c0fe45903fbda8baec25 Mon Sep 17 00:00:00 2001
From: teodor <teodor>
Date: Thu, 16 Sep 2010 15:58:20 +0000
Subject: [PATCH] extend initStringBuffer

---
 tmalloc.c | 48 ++++++++++++++++++++++++++++++++++++++++--------
 tmalloc.h |  7 ++++++-
 2 files changed, 46 insertions(+), 9 deletions(-)

diff --git a/tmalloc.c b/tmalloc.c
index 2d45bc6..e93a46c 100644
--- a/tmalloc.c
+++ b/tmalloc.c
@@ -299,19 +299,51 @@ mcnstrdup(MemoryContext *cntx, char *src, int len) {
 
 /*********StringBuffer********/
 
-#define SBALLOC(s)		( ( (s)->mc ) ? mcalloc((s)->mc, (s)->len) : tmalloc((s)->len) )
-#define SBREALLOC(s)	( ( (s)->mc ) ? mcrealloc((s)->buf, (s)->len) : trealloc((s)->buf, (s)->len) )
+static inline void*
+StringBufferAlloc(StringBuffer* state, void *ptr, size_t size) {
+	if (state->mc)
+		return (ptr) ? mcrealloc(ptr, size) : mcalloc(state->mc, size);
+	else if (state->memalloc)
+		return state->memalloc(ptr, size);
+	else
+		return (ptr) ? trealloc(ptr, size) : tmalloc(size);
+}
 
-StringBuffer*
-initStringBuffer(StringBuffer* state, MemoryContext *mc, int initsize) {
+static StringBuffer*
+initStringBufferInternal(StringBuffer* state, int initsize) {
     state->len = (initsize>0) ? initsize : 1024;
-	state->mc = mc;
-	state->ptr = state->buf = (char*)SBALLOC(state);
+	state->ptr = state->buf = (char*)StringBufferAlloc(state, NULL, initsize);
+	if (!state->ptr)
+		return NULL;
+
 	*(state->ptr) ='\0';
 
 	return state;
 }
 
+StringBuffer*
+initStringBuffer(StringBuffer* state, int initsize) {
+	memset(state, 0, sizeof(*state));
+	
+	return initStringBufferInternal(state, initsize);
+}
+
+StringBuffer*
+initStringBufferMC(StringBuffer* state, MemoryContext *mc, int initsize) {
+	memset(state, 0, sizeof(*state));
+	state->mc = mc;
+
+	return initStringBufferInternal(state, initsize);
+}
+
+StringBuffer*
+initStringBufferMA(StringBuffer* state, MemAlloc memalloc, int initsize) {
+	memset(state, 0, sizeof(*state));
+	state->memalloc = memalloc;
+
+	return initStringBufferInternal(state, initsize);
+}
+
 StringBuffer*
 appendStringBuffer( StringBuffer *state, char *string, int stringlen) {
     if ( string ) {
@@ -327,7 +359,7 @@ appendStringBuffer( StringBuffer *state, char *string, int stringlen) {
 			int	diff = state->ptr - state->buf;
 
 			state->len *= 2;
-			state->buf = (char*)SBREALLOC(state);
+			state->buf = (char*)StringBufferAlloc(state, state->buf, state->len);
 			state->ptr = state->buf + diff;
 	}
 
@@ -358,7 +390,7 @@ printStringBuffer( StringBuffer *state, const char *format, ...) {
 			state->len *= 2;
 		} while( state->ptr - state->buf + printedlen >= state->len );
 
-		state->buf = (char*)SBREALLOC(state);
+		state->buf = (char*)StringBufferAlloc(state, state->buf, state->len);
 		state->ptr = state->buf + curlen;
 		va_start(args, format);
 		printedlen = vsnprintf(state->ptr, printedlen+1, format, args);
diff --git a/tmalloc.h b/tmalloc.h
index f1abcee..89648de 100644
--- a/tmalloc.h
+++ b/tmalloc.h
@@ -85,14 +85,19 @@ void	mcfree(void * ptr);
 char * mcstrdup(MemoryContext *cntx, char * src);
 char * mcnstrdup(MemoryContext *cntx, char *src, int len);
 
+typedef void* (*MemAlloc)(void *ptr, size_t size);
+
 typedef struct {
     char            *buf;
 	char            *ptr;
 	u_int32_t       len;
 	MemoryContext   *mc;
+	MemAlloc		memalloc;
 } StringBuffer;
 
-StringBuffer* initStringBuffer(StringBuffer* state, MemoryContext *mc, int initsize);
+StringBuffer* initStringBuffer(StringBuffer* state, int initsize);
+StringBuffer* initStringBufferMC(StringBuffer* state, MemoryContext *mc, int initsize);
+StringBuffer* initStringBufferMA(StringBuffer* state, MemAlloc memalloc, int initsize);
 StringBuffer* appendStringBuffer( StringBuffer *state, char *string, int stringlen);
 StringBuffer* printStringBuffer( StringBuffer *state, const char *format, ...);
 
-- 
2.37.3