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

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

#include <stdbool.h>

#include <fcntl.h>

#include <errno.h>

#include <sys/stat.h>

#include <slibtool/slibtool.h>

#include "slibtool_spawn_impl.h"

#include "slibtool_mkdir_impl.h"

#include "slibtool_errinfo_impl.h"

#include "slibtool_metafile_impl.h"

static int slbt_exec_compile_remove_file(

	const struct slbt_driver_ctx *	dctx,

	struct slbt_exec_ctx *		ectx,

	const char *			target)

{

	int fdcwd;

	(void)ectx;

	/* fdcwd */

	fdcwd = slbt_driver_fdcwd(dctx);

	/* remove target (if any) */

	if (!unlinkat(fdcwd,target,0) || (errno == ENOENT))

		return 0;

	return SLBT_SYSTEM_ERROR(dctx,0);

}

static int slbt_exec_compile_finalize_argument_vector(

	const struct slbt_driver_ctx *	dctx,

	struct slbt_exec_ctx *		ectx)

{

	char *		sargv[1024];

	char **		sargvbuf;

	char **		base;

	char **		parg;

	char **		aarg;

	char **		aargv;

	char **		cap;

	char **		src;

	char **		dst;

	char **		cmp;

	char *		ccwrap;

	char *          custom;

	/* vector size */

	base = ectx->argv;

	parg = ectx->argv;

	for (; *parg; )

		parg++;

	/* buffer */

	if (parg - base < 1024) {

		aargv    = sargv;

		aarg     = aargv;

		sargvbuf = 0;

	} else if (!(sargvbuf = calloc(parg-base+1,sizeof(char *)))) {

		return SLBT_SYSTEM_ERROR(dctx,0);

	} else {

		aargv = sargvbuf;

		aarg  = aargv;

	}

	/* (program name) */

	parg = &base[1];

	/* avoid -I deduplication with project specific drivers */

	custom = strchr(ectx->program,'/');

	/* split object args from all other args, record output */

	/* annotation, and remove redundant -l arguments       */

	for (; *parg; ) {

		if (ectx->lout[0] == parg) {

			ectx->lout[0] = &aarg[0];

			ectx->lout[1] = &aarg[1];

		}

		if (ectx->mout[0] == parg) {

			ectx->mout[0] = &aarg[0];

			ectx->mout[1] = &aarg[1];

		}

		/* placeholder argument? */

		if (!strncmp(*parg,"-USLIBTOOL_PLACEHOLDER_",23)) {

			parg++;

		} else {

			*aarg++ = *parg++;

		}

	}

	/* program name, ccwrap */

	if ((ccwrap = (char *)dctx->cctx->ccwrap)) {

		base[1] = base[0];

		base[0] = ccwrap;

		base++;

	}

	/* join all other args, starting with de-duplicated -I arguments, */

	/* and filter out all -f switches when compiling in --tag=RC mode */

	src = aargv;

	cap = aarg;

	dst = &base[1];

	for (; !custom && src

		if (((*src)[0] == '-') && ((*src)[1] == 'I')) {

			cmp = &base[1];

			for (; cmp && cmp

				if (!strcmp(*src,*cmp)) {

					cmp = 0;

				} else {

					cmp++;

				}

			}

			if (cmp)

				*dst++ = *src;

		}

		src++;

	}

	src = aargv;

	for (; src

		if ((dctx->cctx->tag == SLBT_TAG_RC)

				&& ((*src)[0] == '-')

				&& ((*src)[1] == 'f'))

			(void)0;

		else if (((*src)[0] != '-') || ((*src)[1] != 'I'))

			*dst++ = *src;

		else if (custom)

			*dst++ = *src;

		src++;

	}

	/* properly null-terminate argv, accounting for redundant arguments */

	*dst = 0;

	/* output annotation */

	if (ectx->lout[0]) {

		ectx->lout[0] = &base[1] + (ectx->lout[0] - aargv);

		ectx->lout[1] = ectx->lout[0] + 1;

	}

	if (ectx->mout[0]) {

		ectx->mout[0] = &base[1] + (ectx->mout[0] - aargv);

		ectx->mout[1] = ectx->mout[0] + 1;

	}

	/* all done */

	if (sargvbuf)

		free(sargvbuf);

	return 0;

}

int  slbt_exec_compile(const struct slbt_driver_ctx * dctx)

{

	int				ret;

	char *				fpic;

	char *				ccwrap;

	bool                            fshared;

	bool                            fstatic;

	struct slbt_exec_ctx *		ectx;

	const struct slbt_common_ctx *	cctx = dctx->cctx;

	/* dry run */

	if (cctx->drvflags & SLBT_DRIVER_DRY_RUN)

		return 0;

	/* context */

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

		return SLBT_NESTED_ERROR(dctx);

	/* remove old .lo wrapper */

	if (slbt_exec_compile_remove_file(dctx,ectx,ectx->ltobjname))

		return SLBT_NESTED_ERROR(dctx);

	/* fshared, fstatic */

	fshared = (cctx->drvflags & (SLBT_DRIVER_SHARED | SLBT_DRIVER_PREFER_SHARED));

	fstatic = (cctx->drvflags & (SLBT_DRIVER_STATIC | SLBT_DRIVER_PREFER_STATIC));

	/* .libs directory */

	if (fshared)

		if (slbt_mkdir(dctx,ectx->ldirname)) {

			ret = SLBT_SYSTEM_ERROR(dctx,ectx->ldirname);

			slbt_ectx_free_exec_ctx(ectx);

			return ret;

		}

	/* compile mode */

	ccwrap        = (char *)cctx->ccwrap;

	ectx->program = ccwrap ? ccwrap : ectx->compiler;

	ectx->argv    = ectx->cargv;

	/* -fpic */

	switch (cctx->tag) {

		case SLBT_TAG_CC:

		case SLBT_TAG_CXX:

		case SLBT_TAG_F77:

		case SLBT_TAG_FC:

			fpic = cctx->settings.picswitch

				? cctx->settings.picswitch

				: *ectx->fpic;

			break;

		default:

			fpic = *ectx->fpic;

	}

	/* shared library object */

	if (fshared) {

		if (!(cctx->drvflags & SLBT_DRIVER_ANTI_PIC)) {

			*ectx->dpic = "-DPIC";

			*ectx->fpic = fpic;

		}

		*ectx->lout[0] = "-o";

		*ectx->lout[1] = ectx->lobjname;

		if (slbt_exec_compile_finalize_argument_vector(dctx,ectx))

			return SLBT_NESTED_ERROR(dctx);

		if (!(cctx->drvflags & SLBT_DRIVER_SILENT)) {

			if (slbt_output_compile(ectx)) {

				slbt_ectx_free_exec_ctx(ectx);

				return SLBT_NESTED_ERROR(dctx);

			}

		}

		if ((slbt_spawn(ectx,true) < 0) && (ectx->pid < 0)) {

			slbt_ectx_free_exec_ctx(ectx);

			return SLBT_SYSTEM_ERROR(dctx,0);

		} else if (ectx->exitcode) {

			slbt_ectx_free_exec_ctx(ectx);

			return SLBT_CUSTOM_ERROR(dctx,SLBT_ERR_COMPILE_ERROR);

		}

		if (fstatic)

			slbt_ectx_reset_argvector(ectx);

	}

	/* static archive object */

	if (fstatic) {

		slbt_reset_placeholders(ectx);

		if (cctx->drvflags & SLBT_DRIVER_PRO_PIC) {

			*ectx->dpic = "-DPIC";

			*ectx->fpic = fpic;

		}

		*ectx->lout[0] = "-o";

		*ectx->lout[1] = ectx->aobjname;

		if (slbt_exec_compile_finalize_argument_vector(dctx,ectx))

			return SLBT_NESTED_ERROR(dctx);

		if (!(cctx->drvflags & SLBT_DRIVER_SILENT)) {

			if (slbt_output_compile(ectx)) {

				slbt_ectx_free_exec_ctx(ectx);

				return SLBT_NESTED_ERROR(dctx);

			}

		}

		if ((slbt_spawn(ectx,true) < 0) && (ectx->pid < 0)) {

			slbt_ectx_free_exec_ctx(ectx);

			return SLBT_SYSTEM_ERROR(dctx,0);

		} else if (ectx->exitcode) {

			slbt_ectx_free_exec_ctx(ectx);

			return SLBT_CUSTOM_ERROR(dctx,SLBT_ERR_COMPILE_ERROR);

		}

	}

	ret = slbt_create_object_wrapper(dctx,ectx);

	slbt_ectx_free_exec_ctx(ectx);

	return ret ? SLBT_NESTED_ERROR(dctx) : 0;

}