1 files changed, 41 insertions, 1 deletions

diff --git a/src/util/bits.c b/src/util/bits.c
index e2d5d79..05b71aa 100644
--- a/src/util/bits.c
+++ b/src/util/bits.c
@@ -4,7 +4,6 @@
  */
 
 #include "bits.h"
-#include <gen/types.h>
 #include <sha1/sha1.h>
 #include "util/memory.h"
 
@@ -180,6 +179,47 @@ static bits_t *bits_bitwise(const bits_t *one, const bits_t *other,
 	return result;
 }
 
+void bits_concatzv(bits_t *one, va_list valist) {
+	const bits_t *next;
+	while ((next = va_arg(valist, const bits_t *)) != NULL) {
+		if (next->bitlen == 0) continue;
+		size_t new_bitlen = one->bitlen + next->bitlen;
+		size_t new_alloc = BYTE_LEN(new_bitlen);
+		if (new_alloc > one->allocated) {
+			one->bits = try_realloc(one->bits, new_alloc);
+			for (size_t i = one->allocated; i < new_alloc; ++i) {
+				one->bits[i] = 0;
+			}
+			one->allocated = new_alloc;
+		}
+		for (size_t j = 0; j < next->bitlen; ++j) {
+			SET_BIT(one->bits, one->bitlen + j, GET_BIT(next->bits, j));
+		}
+		one->bitlen = new_bitlen;
+	}
+}
+
+void bits_concatz(bits_t *one, ...) {
+	va_list valist;
+	va_start(valist, one);
+
+	bits_concatzv(one, valist);
+
+	va_end(valist);
+}
+
+bits_t *bits_concat(const bits_t *one, ...) {
+	va_list valist;
+	va_start(valist, one);
+
+	bits_t *result = bits_copy(one);
+
+	bits_concatzv(result, valist);
+
+	va_end(valist);
+	return result;
+}
+
 bits_t *bits_or(const bits_t *one, const bits_t *other) {
 	return bits_bitwise(one, other, or_func);
 }