Huge library overhaul

3 files changed, 69 insertions, 0 deletions

diff --git a/lib/array/array_free.c b/lib/array/array_free.c
new file mode 100644
index 0000000..0e5560e
--- /dev/null
+++ b/lib/array/array_free.c
@@ -0,0 +1,12 @@
+#include <stdint.h>
+
+#include "array.h"
+#include "alloc.h"
+
+void
+(array_free)(void *vptr, ptrdiff_t elemsz, ptrdiff_t align)
+{
+	ptrdiff_t pad = -sizeof(_mlib_arr_hdr_t) & (align - 1);
+	_mlib_arr_hdr_t *hdr = _mlib_array_hdr(vptr, align);
+	delete(hdr->mem, (uint8_t *)hdr, sizeof(*hdr) + pad + hdr->cap*elemsz);
+}
diff --git a/lib/array/array_new.c b/lib/array/array_new.c
new file mode 100644
index 0000000..7339ba5
--- /dev/null
+++ b/lib/array/array_new.c
@@ -0,0 +1,29 @@
+#include <stdckdint.h>
+#include <stddef.h>
+#include <stdint.h>
+#include <string.h>
+
+#include "array.h"
+#include "alloc.h"
+
+void *
+(array_new)(allocator_t mem, ptrdiff_t nmemb, ptrdiff_t elemsz, ptrdiff_t align)
+{
+	ptrdiff_t total, pad = -sizeof(_mlib_arr_hdr_t) & (align - 1);
+
+	/* Overflow; just call the allocator with the maximum size to force
+	   proper error handling */
+	if (ckd_mul(&total, nmemb, elemsz)
+		|| ckd_add(&total, total, pad)
+	    || ckd_add(&total, total, sizeof(_mlib_arr_hdr_t)))
+	{
+		mem.alloc(mem, ALLOC_NEW, nullptr, 0,
+			PTRDIFF_MAX, PTRDIFF_MAX, PTRDIFF_MAX);
+		unreachable();
+	}
+
+	uint8_t *buf = new(mem, uint8_t, total);
+	_mlib_arr_hdr_t hdr = {0, nmemb, mem};
+	memcpy(buf, &hdr, sizeof(hdr));
+	return buf + sizeof(_mlib_arr_hdr_t) + pad;
+}
diff --git a/lib/array/array_resz.c b/lib/array/array_resz.c
new file mode 100644
index 0000000..d28a7a0
--- /dev/null
+++ b/lib/array/array_resz.c
@@ -0,0 +1,28 @@
+#include <stdckdint.h>
+#include <stddef.h>
+#include <stdint.h>
+
+#include "array.h"
+#include "alloc.h"
+
+void *
+(array_resz)(void *ptr, ptrdiff_t ncap, ptrdiff_t elemsz, ptrdiff_t align)
+{
+	ptrdiff_t newsz, pad = -sizeof(_mlib_arr_hdr_t) & (align - 1);
+	_mlib_arr_hdr_t *hdr = _mlib_array_hdr(ptr, align);
+	/* Overflow; just call the allocator with the maximum size to force
+	   proper error handling */
+	if (ckd_mul(&newsz, ncap, elemsz)
+	 || ckd_add(&newsz, newsz, pad)
+	 || ckd_add(&newsz, newsz, sizeof(_mlib_arr_hdr_t)))
+	{
+		hdr->mem.alloc(hdr->mem, ALLOC_NEW, nullptr, 0,
+			PTRDIFF_MAX, PTRDIFF_MAX, PTRDIFF_MAX);
+		unreachable();
+	}
+
+	ptrdiff_t oldsz = sizeof(_mlib_arr_hdr_t) + pad + hdr->cap*elemsz;
+	hdr = (_mlib_arr_hdr_t *)resz(hdr->mem, (uint8_t *)hdr, oldsz, newsz);
+	hdr->cap = ncap;
+	return (uint8_t *)hdr + sizeof(*hdr) + pad;
+}