#include <stdlib.h>
#include <string.h>
#include <ctype.h>
+#include <stdarg.h>
#include "tlog.h"
#include "tmalloc.h"
return dest;
}
+/*********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) )
+
+StringBuffer*
+initStringBuffer(StringBuffer* state, MemoryContext *mc, int initsize) {
+ state->len = (initsize>0) ? initsize : 1024;
+ state->mc = mc;
+ state->ptr = state->buf = (char*)SBALLOC(state);
+ *(state->ptr) ='\0';
+
+ return state;
+}
+
+StringBuffer*
+appendStringBuffer( StringBuffer *state, char *string, int stringlen) {
+ if ( string ) {
+ if ( stringlen <= 0 )
+ stringlen = strlen(string);
+ } else
+ stringlen = 0;
+
+ if ( stringlen == 0 )
+ return state;
+
+ while ( state->len - ( state->ptr - state->buf ) < stringlen + 1 ) {
+ int diff = state->ptr - state->buf;
+
+ state->len *= 2;
+ state->buf = (char*)SBREALLOC(state);
+ state->ptr = state->buf + diff;
+ }
+
+ memcpy(state->ptr, string, stringlen);
+ state->ptr += stringlen;
+ *state->ptr = '\0';
+ return state;
+}
+
+StringBuffer*
+printStringBuffer( StringBuffer *state, const char *format, ...) {
+ va_list args;
+ int printedlen;
+ int buffreelen;
+
+ buffreelen = state->len - ( state->ptr - state->buf ) - 1;
+
+ va_start(args, format);
+ printedlen = vsnprintf(state->ptr, buffreelen, format, args);
+ va_end(args);
+ /*
+ * if buffer too short, resize buffer and
+ * print it again
+ */
+ if ( buffreelen<=printedlen ) {
+ u_int32_t curlen = state->ptr - state->buf;
+ do {
+ state->len *= 2;
+ } while( state->ptr - state->buf + printedlen >= state->len );
+
+ state->buf = (char*)SBREALLOC(state);
+ state->ptr = state->buf + curlen;
+ va_start(args, format);
+ printedlen = vsnprintf(state->ptr, printedlen+1, format, args);
+ va_end(args);
+ }
+ state->ptr += printedlen;
+ return state;
+}
+