driver: added slbt_lib_get_symlist_ctx(): symbol string vector from symlist.

4 files changed, 168 insertions, 0 deletions

diff --git a/include/slibtool/slibtool.h b/include/slibtool/slibtool.h
index c1e96c2..d80a00e 100644
--- a/include/slibtool/slibtool.h
+++ b/include/slibtool/slibtool.h
@@ -338,6 +338,11 @@ struct slbt_archive_ctx {
 	void *				any;
 };
 
+struct slbt_symlist_ctx {
+	const char * const *            path;
+	const char **                   symstrv;
+};
+
 /* raw input api */
 slbt_api int  slbt_fs_map_input         (const struct slbt_driver_ctx *,
                                          int, const char *, int,
@@ -356,6 +361,10 @@ slbt_api int  slbt_lib_get_driver_fdctx (const struct slbt_driver_ctx *, struct
 
 slbt_api int  slbt_lib_set_driver_fdctx (struct slbt_driver_ctx *, const struct slbt_fd_ctx *);
 
+slbt_api int  slbt_lib_get_symlist_ctx  (const struct slbt_driver_ctx *, const char *, struct slbt_symlist_ctx **);
+
+slbt_api void slbt_lib_free_symlist_ctx (struct slbt_symlist_ctx *);
+
 /* command execution context api */
 slbt_api int  slbt_ectx_get_exec_ctx    (const struct slbt_driver_ctx *, struct slbt_exec_ctx **);
 slbt_api void slbt_ectx_free_exec_ctx   (struct slbt_exec_ctx *);
diff --git a/project/common.mk b/project/common.mk
index 13a3e2d..e82ddc1 100644
--- a/project/common.mk
+++ b/project/common.mk
@@ -16,6 +16,7 @@ API_SRCS = \
 	src/driver/slbt_driver_ctx.c \
 	src/driver/slbt_link_params.c \
 	src/driver/slbt_split_argv.c \
+	src/driver/slbt_symlist_ctx.c \
 	src/driver/slbt_version_info.c \
 	src/host/slbt_host_params.c \
 	src/util/slbt_archive_import.c \
diff --git a/src/driver/slbt_symlist_ctx.c b/src/driver/slbt_symlist_ctx.c
new file mode 100644
index 0000000..2a4997d
--- /dev/null
+++ b/src/driver/slbt_symlist_ctx.c
@@ -0,0 +1,149 @@
+/*******************************************************************/
+/*  slibtool: a skinny libtool implementation, written in C        */
+/*  Copyright (C) 2016--2024  SysDeer Technologies, LLC            */
+/*  Released under the Standard MIT License; see COPYING.SLIBTOOL. */
+/*******************************************************************/
+
+#include <ctype.h>
+#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"
+
+static int slbt_lib_free_symlist_ctx_impl(
+	struct slbt_symlist_ctx_impl *  ctx,
+	struct slbt_input *             mapinfo,
+	int                             ret)
+{
+	if (mapinfo)
+		slbt_fs_unmap_input(mapinfo);
+
+	if (ctx) {
+		if (ctx->pathbuf)
+			free(ctx->pathbuf);
+
+		if (ctx->symstrs)
+			free(ctx->symstrs);
+
+		if (ctx->symstrv)
+			free(ctx->symstrv);
+
+		free(ctx);
+	}
+
+	return ret;
+}
+
+int slbt_lib_get_symlist_ctx(
+	const struct slbt_driver_ctx *  dctx,
+	const char *                    path,
+	struct slbt_symlist_ctx **      pctx)
+{
+	struct slbt_symlist_ctx_impl *  ctx;
+	struct slbt_input               mapinfo;
+	size_t                          nsyms;
+	char *                          ch;
+	char *                          cap;
+	char *                          src;
+	const char **                   psym;
+	char                            dummy;
+	bool                            fvalid;
+
+	/* map symlist file temporarily */
+	if (slbt_fs_map_input(dctx,-1,path,PROT_READ,&mapinfo) < 0)
+		return SLBT_NESTED_ERROR(dctx);
+
+	/* alloc context */
+	if (!(ctx = calloc(1,sizeof(*ctx))))
+		return slbt_lib_free_symlist_ctx_impl(
+			ctx,&mapinfo,
+			SLBT_BUFFER_ERROR(dctx));
+
+	/* count symbols */
+	src = mapinfo.size ? mapinfo.addr : &dummy;
+	cap = &src[mapinfo.size];
+
+	for (; (src<cap) && isspace(*src); )
+		src++;
+
+	for (ch=src,nsyms=0; ch<cap; nsyms++) {
+		for (; (ch<cap) && !isspace(*ch); )
+			ch++;
+
+		fvalid = false;
+
+		for (; (ch<cap) && isspace(*ch); )
+			fvalid = (*ch++ == '\n') || fvalid;
+
+		if (!fvalid)
+			return slbt_lib_free_symlist_ctx_impl(
+				ctx,&mapinfo,
+				SLBT_CUSTOM_ERROR(
+					dctx,
+					SLBT_ERR_FLOW_ERROR));
+	}
+
+	/* clone path, alloc string buffer and symbol vector */
+	if (!(ctx->pathbuf = strdup(path)))
+		return slbt_lib_free_symlist_ctx_impl(
+			ctx,&mapinfo,
+			SLBT_SYSTEM_ERROR(dctx,0));
+
+	if (!(ctx->symstrs = calloc(mapinfo.size+1,1)))
+		return slbt_lib_free_symlist_ctx_impl(
+			ctx,&mapinfo,
+			SLBT_SYSTEM_ERROR(dctx,0));
+
+	if (!(ctx->symstrv = calloc(nsyms+1,sizeof(char *))))
+		return slbt_lib_free_symlist_ctx_impl(
+			ctx,&mapinfo,
+			SLBT_SYSTEM_ERROR(dctx,0));
+
+	/* copy the source to the allocated string buffer */
+	memcpy(ctx->symstrs,mapinfo.addr,mapinfo.size);
+	slbt_fs_unmap_input(&mapinfo);
+
+	/* populate the symbol vector, handle whitespace */
+	src = ctx->symstrs;
+	cap = &src[mapinfo.size];
+
+	for (; (src<cap) && isspace(*src); )
+		src++;
+
+	for (ch=src,psym=ctx->symstrv; ch<cap; psym++) {
+		*psym = ch;
+
+		for (; (ch<cap) && !isspace(*ch); )
+			ch++;
+
+		for (; (ch<cap) && isspace(*ch); )
+			*ch++ = '\0';
+	}
+
+	/* all done */
+	ctx->dctx         = dctx;
+	ctx->path         = ctx->pathbuf;
+	ctx->sctx.path	  = &ctx->path;
+	ctx->sctx.symstrv = ctx->symstrv;
+
+	*pctx = &ctx->sctx;
+
+	return 0;
+}
+
+void slbt_lib_free_symlist_ctx(struct slbt_symlist_ctx * ctx)
+{
+	struct slbt_symlist_ctx_impl *	ictx;
+	uintptr_t			addr;
+
+	if (ctx) {
+		addr = (uintptr_t)ctx - offsetof(struct slbt_symlist_ctx_impl,sctx);
+		ictx = (struct slbt_symlist_ctx_impl *)addr;
+		slbt_lib_free_symlist_ctx_impl(ictx,0,0);
+	}
+}
diff --git a/src/internal/slibtool_driver_impl.h b/src/internal/slibtool_driver_impl.h
index e30514d..8420532 100644
--- a/src/internal/slibtool_driver_impl.h
+++ b/src/internal/slibtool_driver_impl.h
@@ -185,6 +185,15 @@ struct slbt_archive_ctx_impl {
 	struct slbt_archive_ctx		actx;
 };
 
+struct slbt_symlist_ctx_impl {
+	const struct slbt_driver_ctx *	dctx;
+	const char *                    path;
+	char *                          pathbuf;
+	char *                          symstrs;
+	const char **                   symstrv;
+	struct slbt_symlist_ctx         sctx;
+};
+
 
 const char * slbt_program_name(const char *);