1 files changed, 117 insertions, 0 deletions
diff --git a/src/arbits/slbt_archive_ctx.c b/src/arbits/slbt_archive_ctx.c
new file mode 100644
index 0000000..575374c
--- /dev/null
+++ b/src/arbits/slbt_archive_ctx.c
@@ -0,0 +1,117 @@
+/*******************************************************************/
+/*  slibtool: a strong libtool implementation, written in C        */
+/*  Copyright (C) 2016--2024  SysDeer Technologies, LLC            */
+/*  Released under the Standard MIT License; see COPYING.SLIBTOOL. */
+/*******************************************************************/
+
+#include <stdint.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/mman.h>
+
+#include <slibtool/slibtool.h>
+#include "slibtool_driver_impl.h"
+#include "slibtool_errinfo_impl.h"
+#include "slibtool_ar_impl.h"
+
+static int slbt_map_raw_archive(
+	const struct slbt_driver_ctx *	dctx,
+	int				fd,
+	const char *			path,
+	int				prot,
+	struct slbt_raw_archive *	map)
+{
+	struct slbt_input mapinfo = {0,0};
+
+	if (slbt_fs_map_input(dctx,fd,path,prot,&mapinfo) < 0)
+		return SLBT_NESTED_ERROR(dctx);
+
+	if (mapinfo.size == 0)
+		return SLBT_CUSTOM_ERROR(
+			dctx,
+			SLBT_ERR_AR_EMPTY_FILE);
+
+	map->map_addr = mapinfo.addr;
+	map->map_size = mapinfo.size;
+
+	return 0;
+}
+
+static int slbt_unmap_raw_archive(struct slbt_raw_archive * map)
+{
+	struct slbt_input mapinfo;
+
+	mapinfo.addr = map->map_addr;
+	mapinfo.size = map->map_size;
+
+	return slbt_fs_unmap_input(&mapinfo);
+}
+
+static int slbt_ar_free_archive_ctx_impl(struct slbt_archive_ctx_impl * ctx, int ret)
+{
+	if (ctx) {
+		slbt_ar_free_archive_meta(ctx->meta);
+		slbt_unmap_raw_archive(&ctx->map);
+		free(ctx->pathbuf);
+		free(ctx);
+	}
+
+	return ret;
+}
+
+int slbt_ar_get_archive_ctx(
+	const struct slbt_driver_ctx *	dctx,
+	const char *			path,
+	struct slbt_archive_ctx **		pctx)
+{
+	struct slbt_archive_ctx_impl *	ctx;
+	struct slbt_archive_meta_impl * mctx;
+	int				prot;
+
+	if (!(ctx = calloc(1,sizeof(*ctx))))
+		return SLBT_BUFFER_ERROR(dctx);
+
+	slbt_driver_set_arctx(
+		dctx,0,path);
+
+	prot = (dctx->cctx->actflags & SLBT_ACTION_MAP_READWRITE)
+		? PROT_READ | PROT_WRITE
+		: PROT_READ;
+
+	if (slbt_map_raw_archive(dctx,-1,path,prot,&ctx->map))
+		return slbt_ar_free_archive_ctx_impl(ctx,
+			SLBT_NESTED_ERROR(dctx));
+
+	if (slbt_ar_get_archive_meta(dctx,&ctx->map,&ctx->meta))
+		return slbt_ar_free_archive_ctx_impl(ctx,
+			SLBT_NESTED_ERROR(dctx));
+
+	if (!(ctx->pathbuf = strdup(path)))
+		return slbt_ar_free_archive_ctx_impl(ctx,
+			SLBT_NESTED_ERROR(dctx));
+
+	mctx = slbt_archive_meta_ictx(ctx->meta);
+
+	ctx->dctx       = dctx;
+	ctx->path	= ctx->pathbuf;
+	ctx->actx.path	= &ctx->path;
+	ctx->actx.map	= &ctx->map;
+	ctx->actx.meta	= ctx->meta;
+	mctx->actx      = &ctx->actx;
+
+	*pctx = &ctx->actx;
+	return 0;
+}
+
+void slbt_ar_free_archive_ctx(struct slbt_archive_ctx * ctx)
+{
+	struct slbt_archive_ctx_impl *	ictx;
+	uintptr_t			addr;
+
+	if (ctx) {
+		addr = (uintptr_t)ctx - offsetof(struct slbt_archive_ctx_impl,actx);
+		ictx = (struct slbt_archive_ctx_impl *)addr;
+		slbt_ar_free_archive_ctx_impl(ictx,0);
+	}
+}