NSD v3.2.4

1 files changed, 131 insertions, 0 deletions

diff --git a/usr.sbin/nsd/buffer.c b/usr.sbin/nsd/buffer.c
new file mode 100644
index 00000000000..67b11183309
--- /dev/null
+++ b/usr.sbin/nsd/buffer.c
@@ -0,0 +1,131 @@
+/*
+ * buffer.c -- generic memory buffer .
+ *
+ * Copyright (c) 2001-2006, NLnet Labs. All rights reserved.
+ *
+ * See LICENSE for the license.
+ *
+ */
+
+#include <config.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+#include "buffer.h"
+
+static void
+buffer_cleanup(void *arg)
+{
+	buffer_type *buffer = (buffer_type *) arg;
+	assert(!buffer->_fixed);
+	free(buffer->_data);
+}
+
+buffer_type *
+buffer_create(region_type *region, size_t capacity)
+{
+	buffer_type *buffer
+		= (buffer_type *) region_alloc(region, sizeof(buffer_type));
+	if (!buffer)
+		return NULL;
+	
+	buffer->_data = (uint8_t *) xalloc(capacity);
+	buffer->_position = 0;
+	buffer->_limit = buffer->_capacity = capacity;
+	buffer->_fixed = 0;
+	buffer_invariant(buffer);
+	
+	region_add_cleanup(region, buffer_cleanup, buffer);
+
+	return buffer;
+}
+
+void
+buffer_create_from(buffer_type *buffer, void *data, size_t size)
+{
+	assert(data);
+
+	buffer->_position = 0;
+	buffer->_limit = buffer->_capacity = size;
+	buffer->_data = (uint8_t *) data;
+	buffer->_fixed = 1;
+	
+	buffer_invariant(buffer);
+}
+
+void
+buffer_clear(buffer_type *buffer)
+{
+	buffer_invariant(buffer);
+	
+	buffer->_position = 0;
+	buffer->_limit = buffer->_capacity;
+}
+
+void
+buffer_flip(buffer_type *buffer)
+{
+	buffer_invariant(buffer);
+	
+	buffer->_limit = buffer->_position;
+	buffer->_position = 0;
+}
+
+void
+buffer_rewind(buffer_type *buffer)
+{
+	buffer_invariant(buffer);
+	
+	buffer->_position = 0;
+}
+
+void
+buffer_set_capacity(buffer_type *buffer, size_t capacity)
+{
+	buffer_invariant(buffer);
+	assert(buffer->_position <= capacity);
+	buffer->_data = (uint8_t *) xrealloc(buffer->_data, capacity);
+	buffer->_limit = buffer->_capacity = capacity;
+}
+
+void
+buffer_reserve(buffer_type *buffer, size_t amount)
+{
+	buffer_invariant(buffer);
+	assert(!buffer->_fixed);
+	if (buffer->_capacity < buffer->_position + amount) {
+		size_t new_capacity = buffer->_capacity * 3 / 2;
+		if (new_capacity < buffer->_position + amount) {
+			new_capacity = buffer->_position + amount;
+		}
+		buffer_set_capacity(buffer, new_capacity);
+	}
+	buffer->_limit = buffer->_capacity;
+}
+
+int
+buffer_printf(buffer_type *buffer, const char *format, ...)
+{
+	va_list args;
+	int written;
+	size_t remaining;
+	
+	buffer_invariant(buffer);
+	assert(buffer->_limit == buffer->_capacity);
+
+	remaining = buffer_remaining(buffer);
+	va_start(args, format);
+	written = vsnprintf((char *) buffer_current(buffer), remaining,
+			    format, args);
+	va_end(args);
+	if (written >= 0 && (size_t) written >= remaining) {
+		buffer_reserve(buffer, written + 1);
+		va_start(args, format);
+		written = vsnprintf((char *) buffer_current(buffer),
+				    buffer_remaining(buffer),
+				    format, args);
+		va_end(args);
+	}
+	buffer->_position += written;
+	return written;
+}