/*******************************************************************/

/*  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 <slibtool/slibtool.h>

#include "slibtool_ar_impl.h"

#include "slibtool_driver_impl.h"

#include "slibtool_snprintf_impl.h"

#include "slibtool_errinfo_impl.h"

static const char * slbt_strong_symname(

	const char *    symname,

	bool            fcoff,

	char            (*strbuf)[4096])

{

	const char *    dot;

	const char *    mark;

	char *          sym;

	if (fcoff) {

		if (!strncmp(symname,"__imp_",6))

			return 0;

		if (strncmp(symname,".weak.",6))

			return symname;

		sym  = *strbuf;

		mark = &symname[6];

		dot  = strchr(mark,'.');

		strncpy(sym,mark,dot-mark);

		sym[dot-mark] = '\0';

		return sym;

	}

	return symname;

}

static int slbt_ar_dlsyms_define_by_type(

	int                                 fdout,

	const char *                        arname,

	struct slbt_archive_meta_impl *     mctx,

	const char *                        desc,

	const char                          stype)

{

	uint64_t      idx;

	uint64_t      nsyms;

	bool          fcoff;

	const char *  symname;

	char          strbuf[4096];

	for (idx=0,nsyms=0; idx<mctx->armaps.armap_nsyms; idx++)

		if (mctx->syminfv[idx]->ar_symbol_type[0] == stype)

			nsyms++;

	if (nsyms == 0)

		return 0;

	fcoff  = slbt_host_objfmt_is_coff(mctx->dctx);

	fcoff |= (mctx->ofmtattr & AR_OBJECT_ATTR_COFF);

	if (slbt_dprintf(fdout,"/* %s (%s) */\n",desc,arname) < 0)

		return SLBT_SYSTEM_ERROR(mctx->dctx,0);

	for (idx=0; idx<mctx->armaps.armap_nsyms; idx++)

		if (mctx->syminfv[idx]->ar_symbol_type[0] == stype)

			if ((symname = slbt_strong_symname(

					mctx->syminfv[idx]->ar_symbol_name,

					fcoff,&strbuf)))

				if (slbt_dprintf(fdout,

						(stype == 'T')

							? "extern int %s();\n"

							: "extern char %s[];\n",

						symname) < 0)

					return SLBT_SYSTEM_ERROR(mctx->dctx,0);

	if (slbt_dprintf(fdout,"\n") < 0)

		return SLBT_SYSTEM_ERROR(mctx->dctx,0);

	return 0;

}

static int slbt_ar_dlsyms_get_max_len_by_type(

	int                                 mlen,

	struct slbt_archive_meta_impl *     mctx,

	const char                          stype)

{

	int           len;

	uint64_t      idx;

	bool          fcoff;

	const char *  symname;

	char          strbuf[4096];

	fcoff  = slbt_host_objfmt_is_coff(mctx->dctx);

	fcoff |= (mctx->ofmtattr & AR_OBJECT_ATTR_COFF);

	for (idx=0; idx<mctx->armaps.armap_nsyms; idx++)

		if (mctx->syminfv[idx]->ar_symbol_type[0] == stype)

			if ((symname = slbt_strong_symname(

					mctx->syminfv[idx]->ar_symbol_name,

					fcoff,&strbuf)))

				if ((len = strlen(symname)) > mlen)

					mlen = len;

	return mlen;

}

static int slbt_ar_dlsyms_add_by_type(

	int                                 fdout,

	struct slbt_archive_meta_impl *     mctx,

	const char *                        fmt,

	const char                          stype,

	char                                (*namebuf)[4096])

{

	uint64_t      idx;

	uint64_t      nsyms;

	bool          fcoff;

	const char *  symname;

	char          strbuf[4096];

	nsyms   = 0;

	symname = *namebuf;

	fcoff  = slbt_host_objfmt_is_coff(mctx->dctx);

	fcoff |= (mctx->ofmtattr & AR_OBJECT_ATTR_COFF);

	for (idx=0; idx<mctx->armaps.armap_nsyms; idx++)

		if (mctx->syminfv[idx]->ar_symbol_type[0] == stype)

			nsyms++;

	if (nsyms == 0)

		return 0;

	if (slbt_dprintf(fdout,"\n") < 0)

		return SLBT_SYSTEM_ERROR(mctx->dctx,0);

	for (idx=0; idx<mctx->armaps.armap_nsyms; idx++) {

		if (mctx->syminfv[idx]->ar_symbol_type[0] == stype) {

			symname = slbt_strong_symname(

				mctx->syminfv[idx]->ar_symbol_name,

				fcoff,&strbuf);

			if (symname) {

				if (slbt_snprintf(*namebuf,sizeof(*namebuf),

						"%s\",",symname) < 0)

					return SLBT_SYSTEM_ERROR(mctx->dctx,0);

				if (slbt_dprintf(fdout,fmt,

						*namebuf,

						(stype == 'T') ? "&" : "",

						symname) < 0)

					return SLBT_NESTED_ERROR(mctx->dctx);

			}

		}

	}

	return 0;

}

static int slbt_ar_output_dlsyms_impl(

	int                                 fdout,

	const struct slbt_driver_ctx *      dctx,

	struct slbt_archive_ctx **          arctxv,

	const char *                        dsounit)

{

	int                                 ret;

	int                                 idx;

	unsigned                            len;

	unsigned                            cmp;

	const char *                        arname;

	const char *                        soname;

	struct slbt_archive_ctx *           actx;

	struct slbt_archive_ctx **          parctx;

	struct slbt_archive_ctx_impl *      ictx;

	struct slbt_archive_meta_impl *     mctx;

	const struct slbt_source_version *  verinfo;

	char                                dlsymfmt[32];

	char                                cline[6][73];

	char                                symname[4096];

	/* init */

	actx    = arctxv[0];

	verinfo = slbt_api_source_version();

	/* preamble */

	memset(cline[0],'*',72);

	memset(cline[1],' ',72);

	memset(cline[2],' ',72);

	memset(cline[3],' ',72);

	memset(cline[4],'*',72);

	memset(cline[5],0,72);

	cline[5][0] = '\n';

	len = snprintf(&cline[1][3],69,

		"backward-compatible dlsym table");

	cline[1][3+len] = ' ';

	len = snprintf(&cline[2][3],69,

		"Generated by %s (slibtool %d.%d.%d)",

		dctx->program,

		verinfo->major,verinfo->minor,verinfo->revision);

	cline[2][3+len] = ' ';

	len = snprintf(&cline[3][3],69,

		"[commit reference: %s]",

		verinfo->commit);

	cline[3][3+len] = ' ';

	for (idx=0; idx<5; idx++) {

		cline[idx][0] = '/';

		cline[idx][1] = '*';

		cline[idx][70] = '*';

		cline[idx][71] = '/';

		cline[idx][72] = '\n';

	}

	if (slbt_dprintf(fdout,"%s",&cline[0]) < 0)

		return SLBT_SYSTEM_ERROR(dctx,0);

	if (slbt_dprintf(fdout,

			"#ifdef __cplusplus\n"

			"extern \"C\" {\n"

			"#endif\n\n") < 0)

		return SLBT_SYSTEM_ERROR(dctx,0);

	/* declarations */

	for (parctx=arctxv; *parctx; parctx++) {

		actx   = *parctx;

		ictx   = slbt_get_archive_ictx(actx);

		mctx   = slbt_archive_meta_ictx(ictx->meta);

		if ((arname = strrchr(*actx->path,'/')))

			arname++;

		if (!arname)

			arname = *actx->path;

		ret  = slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Absolute Values",     'A');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: BSS Section",         'B');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Common Section",      'C');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Initialized Data",    'D');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Small Globals",       'G');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Indirect References", 'I');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Read-Only Section",   'R');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Small Objects",       'S');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Data Symbols: Weak Symbols",        'W');

		ret |= slbt_ar_dlsyms_define_by_type(fdout,arname,mctx,"Text Section: Public Interfaces",   'T');

		if (ret < 0)

			return SLBT_NESTED_ERROR(dctx);

	}

	/* vtable struct definition */

	if (slbt_dprintf(fdout,

			"/* name-address Public ABI struct definition */\n"

			"struct lt_dlsym_symdef {\n"

			"\tconst char *   dlsym_name;\n"

			"\tvoid *         dlsym_addr;\n"

			"};\n\n") < 0)

		return SLBT_NESTED_ERROR(dctx);

	soname = (strcmp(dsounit,"@PROGRAM@")) ? dsounit : "_PROGRAM_";

	if (slbt_dprintf(fdout,

			"/* dlsym vtable */\n"

			"extern const struct lt_dlsym_symdef "

			"lt_%s_LTX_preloaded_symbols[];\n\n"

			"const struct lt_dlsym_symdef "

			"lt_%s_LTX_preloaded_symbols[] = {\n",

			soname,soname) < 0)

		return SLBT_NESTED_ERROR(dctx);

	/* align dlsym_name and dlsym_addr columsn (because we can) */

	for (parctx=arctxv,len=0; *parctx; parctx++) {

		actx = *parctx;

		ictx = slbt_get_archive_ictx(actx);

		mctx = slbt_archive_meta_ictx(ictx->meta);

		if ((arname = strrchr(*actx->path,'/')))

			arname++;

		if (!arname)

			arname = *actx->path;

		if (len < (cmp = strlen(arname)))

			len = cmp;

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'A');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'B');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'C');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'D');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'G');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'I');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'R');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'S');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'T');

		len  = slbt_ar_dlsyms_get_max_len_by_type(len,mctx,'W');

	}

	/* quote, comma */

	len += 2;

	if (len >= sizeof(symname))

		return SLBT_CUSTOM_ERROR(

			dctx,

			SLBT_ERR_FLOW_ERROR);

	/* aligned print format */

	snprintf(dlsymfmt,sizeof(dlsymfmt),"\t{\"%%-%ds %%s%%s},\n",len);

	/* dso unit */

	if (slbt_snprintf(symname,sizeof(symname),"%s\",",dsounit) < 0)

		return SLBT_SYSTEM_ERROR(dctx,0);

	if (slbt_dprintf(fdout,dlsymfmt,symname,"","0") < 0)

		return SLBT_NESTED_ERROR(dctx);

	/* (-dlopen force) */

	if (!arctxv[0]->meta->a_memberv)

		if (!strcmp(*arctxv[0]->path,"@PROGRAM@"))

			arctxv++;

	/* at long last */

	for (parctx=arctxv; *parctx; parctx++) {

		actx = *parctx;

		ictx = slbt_get_archive_ictx(actx);

		mctx = slbt_archive_meta_ictx(ictx->meta);

		if ((arname = strrchr(*actx->path,'/')))

			arname++;

		if (!arname)

			arname = *actx->path;

		if (slbt_dprintf(fdout,"\n") < 0)

			return SLBT_NESTED_ERROR(mctx->dctx);

		if (slbt_snprintf(symname,sizeof(symname),"%s\",",arname) < 0)

			return SLBT_SYSTEM_ERROR(mctx->dctx,0);

		if (slbt_dprintf(fdout,dlsymfmt,symname,"","0") < 0)

			return SLBT_NESTED_ERROR(mctx->dctx);

		ret  = slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'A',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'B',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'C',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'D',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'G',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'I',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'R',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'S',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'S',&symname);

		ret |= slbt_ar_dlsyms_add_by_type(fdout,mctx,dlsymfmt,'T',&symname);

		if (ret < 0)

			return SLBT_NESTED_ERROR(dctx);

	}

	/* null-terminate the vtable */

	if (slbt_dprintf(fdout,"\n\t{%d,%*c%d}\n",0,len,' ',0) < 0)

		return SLBT_NESTED_ERROR(mctx->dctx);

	/* close vtable, wrap translation unit */

	if (slbt_dprintf(fdout,

			"};\n\n"

			"#ifdef __cplusplus\n"

			"}\n"

			"#endif\n") < 0)

		return SLBT_SYSTEM_ERROR(dctx,0);

	return 0;

}

static int slbt_ar_create_dlsyms_impl(

	struct slbt_archive_ctx **        arctxv,

	const char *                      dlunit,

	const char *                      path,

	mode_t                            mode)

{

	int                               ret;

	struct slbt_archive_ctx **        actx;

	struct slbt_exec_ctx *            ectx;

	struct slbt_archive_meta_impl *   mctx;

	const struct slbt_driver_ctx *    dctx;

	struct slbt_fd_ctx                fdctx;

	int                               fdout;

	mctx = slbt_archive_meta_ictx(arctxv[0]->meta);

	dctx = mctx->dctx;

	ectx = 0;

	if (slbt_lib_get_driver_fdctx(dctx,&fdctx) < 0)

		return SLBT_NESTED_ERROR(dctx);

	if (path) {

		if ((fdout = openat(

				fdctx.fdcwd,path,

				O_WRONLY|O_CREAT|O_TRUNC,

				mode)) < 0)

			return SLBT_SYSTEM_ERROR(dctx,path);

	} else {

		fdout = fdctx.fdout;

	}

	for (actx=arctxv; *actx; actx++) {

		mctx = slbt_archive_meta_ictx((*actx)->meta);

		if (!mctx->syminfo && !ectx)

			if (slbt_ectx_get_exec_ctx(dctx,&ectx) < 0)

				return SLBT_NESTED_ERROR(dctx);

		if (!mctx->syminfo)

			if (slbt_ar_update_syminfo(*actx,ectx) < 0)

				return SLBT_NESTED_ERROR(dctx);

	}

	if (ectx)

		slbt_ectx_free_exec_ctx(ectx);

	ret = slbt_ar_output_dlsyms_impl(

		fdout,dctx,arctxv,dlunit);

	if (path) {

		close(fdout);

	}

	return ret;

}

int slbt_ar_create_dlsyms(

	struct slbt_archive_ctx **  arctxv,

	const char *                dlunit,

	const char *                path,

	mode_t                      mode)

{

	return slbt_ar_create_dlsyms_impl(arctxv,dlunit,path,mode);

}