1 files changed, 151 insertions, 0 deletions
diff --git a/src/btparser/strbuf.c b/src/btparser/strbuf.c
new file mode 100644
index 00000000..43d834fd
--- /dev/null
+++ b/src/btparser/strbuf.c
@@ -0,0 +1,151 @@
+/*
+    strbuf.c - string buffer
+
+    Copyright (C) 2010  Red Hat, Inc.
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License along
+    with this program; if not, write to the Free Software Foundation, Inc.,
+    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+*/
+#include "strbuf.h"
+#include "utils.h"
+#include <stdlib.h>
+#include <stdio.h>
+#include <assert.h>
+#include <string.h>
+#include <stdarg.h>
+
+struct btp_strbuf *
+btp_strbuf_new()
+{
+    struct btp_strbuf *buf = btp_malloc(sizeof(struct btp_strbuf));
+    buf->alloc = 8;
+    buf->len = 0;
+    buf->buf = btp_malloc(buf->alloc);
+    buf->buf[buf->len] = '\0';
+    return buf;
+}
+
+void
+btp_strbuf_free(struct btp_strbuf *strbuf)
+{
+    if (!strbuf)
+        return;
+
+    free(strbuf->buf);
+    free(strbuf);
+}
+
+char *
+btp_strbuf_free_nobuf(struct btp_strbuf *strbuf)
+{
+    char *buf = strbuf->buf;
+    free(strbuf);
+    return buf;
+}
+
+
+void
+btp_strbuf_clear(struct btp_strbuf *strbuf)
+{
+    assert(strbuf->alloc > 0);
+    strbuf->len = 0;
+    strbuf->buf[0] = '\0';
+}
+
+/* Ensures that the buffer can be extended by num characters
+ * without touching malloc/realloc.
+ */
+void
+btp_strbuf_grow(struct btp_strbuf *strbuf, int num)
+{
+    if (strbuf->len + num + 1 > strbuf->alloc)
+    {
+	while (strbuf->len + num + 1 > strbuf->alloc)
+	    strbuf->alloc *= 2; /* huge grow = infinite loop */
+
+	strbuf->buf = realloc(strbuf->buf, strbuf->alloc);
+	if (!strbuf->buf)
+	{
+	    puts("Error while allocating memory for string buffer.");
+	    exit(5);
+	}
+    }
+}
+
+struct btp_strbuf *
+btp_strbuf_append_char(struct btp_strbuf *strbuf,
+                       char c)
+{
+    btp_strbuf_grow(strbuf, 1);
+    strbuf->buf[strbuf->len++] = c;
+    strbuf->buf[strbuf->len] = '\0';
+    return strbuf;
+}
+
+struct btp_strbuf *
+btp_strbuf_append_str(struct btp_strbuf *strbuf,
+                      const char *str)
+{
+    int len = strlen(str);
+    btp_strbuf_grow(strbuf, len);
+    assert(strbuf->len + len < strbuf->alloc);
+    strcpy(strbuf->buf + strbuf->len, str);
+    strbuf->len += len;
+    return strbuf;
+}
+
+struct btp_strbuf *
+btp_strbuf_prepend_str(struct btp_strbuf *strbuf,
+                       const char *str)
+{
+    int len = strlen(str);
+    btp_strbuf_grow(strbuf, len);
+    assert(strbuf->len + len < strbuf->alloc);
+    memmove(strbuf->buf + len, strbuf->buf, strbuf->len + 1);
+    memcpy(strbuf->buf, str, len);
+    strbuf->len += len;
+    return strbuf;
+}
+
+struct btp_strbuf *
+btp_strbuf_append_strf(struct btp_strbuf *strbuf,
+                       const char *format, ...)
+{
+    va_list p;
+    char *string_ptr;
+
+    va_start(p, format);
+    string_ptr = btp_vasprintf(format, p);
+    va_end(p);
+
+    btp_strbuf_append_str(strbuf, string_ptr);
+    free(string_ptr);
+    return strbuf;
+}
+
+struct btp_strbuf *
+btp_strbuf_prepend_strf(struct btp_strbuf *strbuf,
+                        const char *format, ...)
+{
+    va_list p;
+    char *string_ptr;
+
+    va_start(p, format);
+    string_ptr = btp_vasprintf(format, p);
+    va_end(p);
+
+    btp_strbuf_prepend_str(strbuf, string_ptr);
+    free(string_ptr);
+    return strbuf;
+}