Add generic buffer code

Represents a block of memory that can be added to, parsed and so on

2 files changed, 215 insertions, 0 deletions

diff --git a/common/tests/Makefile.am b/common/tests/Makefile.am
index 4badbb5..3b6b986 100644
--- a/common/tests/Makefile.am
+++ b/common/tests/Makefile.am
@@ -21,6 +21,7 @@ CHECK_PROGS = \
 	test-dict \
 	test-array \
 	test-attrs \
+	test-buffer \
 	$(NULL)
 
 noinst_PROGRAMS = \
diff --git a/common/tests/test-buffer.c b/common/tests/test-buffer.c
new file mode 100644
index 0000000..9db5f83
--- /dev/null
+++ b/common/tests/test-buffer.c
@@ -0,0 +1,214 @@
+/*
+ * Copyright (c) 2012 Red Hat Inc.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ *     * Redistributions of source code must retain the above
+ *       copyright notice, this list of conditions and the
+ *       following disclaimer.
+ *     * Redistributions in binary form must reproduce the
+ *       above copyright notice, this list of conditions and
+ *       the following disclaimer in the documentation and/or
+ *       other materials provided with the distribution.
+ *     * The names of contributors to this software may not be
+ *       used to endorse or promote products derived from this
+ *       software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+ * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
+ * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
+ * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
+ * DAMAGE.
+ *
+ * Author: Stef Walter <stef@thewalter.net>
+ */
+
+#include "config.h"
+#include "CuTest.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+
+#include "debug.h"
+#include "buffer.h"
+
+static void
+test_init_uninit (CuTest *tc)
+{
+	p11_buffer buffer;
+
+	p11_buffer_init (&buffer, 10);
+	CuAssertPtrNotNull (tc, buffer.data);
+	CuAssertIntEquals (tc, 0, buffer.len);
+	CuAssertIntEquals (tc, 0, buffer.flags);
+	CuAssertTrue (tc, buffer.size >= 10);
+	CuAssertPtrNotNull (tc, buffer.ffree);
+	CuAssertPtrNotNull (tc, buffer.frealloc);
+
+	p11_buffer_uninit (&buffer);
+}
+
+static void
+test_append (CuTest *tc)
+{
+	p11_buffer buffer;
+
+	p11_buffer_init (&buffer, 10);
+	buffer.len = 5;
+	p11_buffer_append (&buffer, 35);
+	CuAssertIntEquals (tc, 5 + 35, buffer.len);
+	CuAssertTrue (tc, buffer.size >= 35 + 5);
+
+	p11_buffer_append (&buffer, 15);
+	CuAssertIntEquals (tc, 5 + 35 + 15, buffer.len);
+	CuAssertTrue (tc, buffer.size >= 5 + 35 + 15);
+
+	p11_buffer_uninit (&buffer);
+}
+
+static void
+test_null (CuTest *tc)
+{
+	p11_buffer buffer;
+
+	p11_buffer_init_null (&buffer, 10);
+	p11_buffer_add (&buffer, "Blah", -1);
+	p11_buffer_add (&buffer, " blah", -1);
+
+	CuAssertStrEquals (tc, "Blah blah", buffer.data);
+
+	p11_buffer_uninit (&buffer);
+}
+
+static int mock_realloced = 0;
+static int mock_freed = 0;
+
+static void *
+mock_realloc (void *data,
+              size_t size)
+{
+	mock_realloced++;
+	return realloc (data, size);
+}
+
+static void
+mock_free (void *data)
+{
+	mock_freed++;
+	free (data);
+}
+
+static void
+test_init_for_data (CuTest *tc)
+{
+	p11_buffer buffer;
+	unsigned char *ret;
+	size_t len;
+
+	mock_realloced = 0;
+	mock_freed = 0;
+
+	p11_buffer_init_full (&buffer, (unsigned char *)strdup ("blah"), 4, 0,
+	                       mock_realloc, mock_free);
+
+	CuAssertPtrNotNull (tc, buffer.data);
+	CuAssertStrEquals (tc, "blah", (char *)buffer.data);
+	CuAssertIntEquals (tc, 4, buffer.len);
+	CuAssertIntEquals (tc, 0, buffer.flags);
+	CuAssertIntEquals (tc, 4, buffer.size);
+	CuAssertPtrEquals (tc, mock_free, buffer.ffree);
+	CuAssertPtrEquals (tc, mock_realloc, buffer.frealloc);
+
+	CuAssertIntEquals (tc, 0, mock_realloced);
+	CuAssertIntEquals (tc, 0, mock_freed);
+
+	len = buffer.len;
+	ret = p11_buffer_append (&buffer, 1024);
+	CuAssertPtrEquals (tc, (char *)buffer.data + len, ret);
+	CuAssertIntEquals (tc, 1, mock_realloced);
+
+	p11_buffer_uninit (&buffer);
+	CuAssertIntEquals (tc, 1, mock_realloced);
+	CuAssertIntEquals (tc, 1, mock_freed);
+}
+
+static void
+test_steal (CuTest *tc)
+{
+	p11_buffer buffer;
+	char *string;
+	size_t length;
+
+	mock_freed = 0;
+
+	p11_buffer_init_full (&buffer, (unsigned char *)strdup ("blah"), 4,
+	                      P11_BUFFER_NULL, mock_realloc, mock_free);
+
+	CuAssertPtrNotNull (tc, buffer.data);
+	CuAssertStrEquals (tc, "blah", buffer.data);
+
+	p11_buffer_add (&buffer, " yada", -1);
+	CuAssertStrEquals (tc, "blah yada", buffer.data);
+
+	string = p11_buffer_steal (&buffer, &length);
+	p11_buffer_uninit (&buffer);
+
+	CuAssertStrEquals (tc, "blah yada", string);
+	CuAssertIntEquals (tc, 9, length);
+	CuAssertIntEquals (tc, 0, mock_freed);
+
+	free (string);
+}
+
+static void
+test_add (CuTest *tc)
+{
+	p11_buffer buffer;
+
+	p11_buffer_init (&buffer, 10);
+
+	p11_buffer_add (&buffer, (unsigned char *)"Planet Express", 15);
+	CuAssertIntEquals (tc, 15, buffer.len);
+	CuAssertStrEquals (tc, "Planet Express", (char *)buffer.data);
+	CuAssertTrue (tc, p11_buffer_ok (&buffer));
+
+	p11_buffer_uninit (&buffer);
+}
+
+int
+main (void)
+{
+	CuString *output = CuStringNew ();
+	CuSuite* suite = CuSuiteNew ();
+	int ret;
+
+	setenv ("P11_KIT_STRICT", "1", 1);
+	p11_debug_init ();
+
+	SUITE_ADD_TEST (suite, test_init_uninit);
+	SUITE_ADD_TEST (suite, test_init_for_data);
+	SUITE_ADD_TEST (suite, test_append);
+	SUITE_ADD_TEST (suite, test_null);
+	SUITE_ADD_TEST (suite, test_add);
+	SUITE_ADD_TEST (suite, test_steal);
+
+	CuSuiteRun (suite);
+	CuSuiteSummary (suite, output);
+	CuSuiteDetails (suite, output);
+	printf ("%s\n", output->buffer);
+	ret = suite->failCount;
+	CuSuiteDelete (suite);
+	CuStringDelete (output);
+
+	return ret;
+}