firasuke / cross / slibtool

Forked from cross/slibtool 3 months ago
Clone

Blame src/logic/slbt_exec_link.c

2bd749
/*******************************************************************/
2bd749
/*  slibtool: a skinny libtool implementation, written in C        */
2bd749
/*  Copyright (C) 2016  Z. Gilboa                                  */
2bd749
/*  Released under the Standard MIT License; see COPYING.SLIBTOOL. */
2bd749
/*******************************************************************/
2bd749
2bd749
#include <string.h>
2bd749
#include <stdbool.h>
2bd749
#include <fcntl.h>
2bd749
#include <errno.h>
2bd749
#include <sys/stat.h>
2bd749
2bd749
#include <slibtool/slibtool.h>
2bd749
#include "slibtool_spawn_impl.h"
2bd749
5cc3b3
/*******************************************************************/
5cc3b3
/*                                                                 */
5cc3b3
/* -o <ltlib>  switches              input   result                */
5cc3b3
/* ----------  --------------------- -----   ------                */
b07789
/* libfoo.a    [-shared|-static]     bar.lo  libfoo.a              */
5cc3b3
/*                                                                 */
6d048d
/* ar cru libfoo.a bar.o                                           */
5cc3b3
/* ranlib libfoo.a                                                 */
5cc3b3
/*                                                                 */
5cc3b3
/*******************************************************************/
5cc3b3
a0c318
/*******************************************************************/
a0c318
/*                                                                 */
a0c318
/* -o <ltlib>  switches              input   result                */
a0c318
/* ----------  --------------------- -----   ------                */
a0c318
/* libfoo.la   -shared               bar.lo  libfoo.la             */
a0c318
/*                                           .libs/libfoo.a        */
a0c318
/*                                           .libs/libfoo.la (lnk) */
a0c318
/*                                                                 */
a0c318
/* ar cru .libs/libfoo.a .libs/bar.o                               */
a0c318
/* ranlib .libs/libfoo.a                                           */
a0c318
/* (generate libfoo.la)                                            */
a0c318
/* ln -s ../libfoo.la .libs/libfoo.la                              */
f730ac
/*                                                                 */
f730ac
/*******************************************************************/
f730ac
f730ac
/*******************************************************************/
f730ac
/*                                                                 */
f730ac
/* -o <ltlib>  switches              input   result                */
f730ac
/* ----------  --------------------- -----   ------                */
f730ac
/* libfoo.la   -static               bar.lo  libfoo.la             */
f730ac
/*                                           .libs/libfoo.a        */
f730ac
/*                                           .libs/libfoo.la (lnk) */
f730ac
/*                                                                 */
f730ac
/* ar cru .libs/libfoo.a bar.o                                     */
f730ac
/* ranlib .libs/libfoo.a                                           */
f730ac
/* (generate libfoo.la)                                            */
f730ac
/* ln -s ../libfoo.la .libs/libfoo.la                              */
a0c318
/*                                                                 */
a0c318
/*******************************************************************/
a0c318
5cc3b3
static bool slbt_adjust_input_argument(char * arg, bool fpic)
5cc3b3
{
53aa85
	char *	slash;
5cc3b3
	char *	dot;
53aa85
	char	base[PATH_MAX];
5cc3b3
5cc3b3
	if (*arg == '-')
5cc3b3
		return false;
5cc3b3
5cc3b3
	if (!(dot = strrchr(arg,'.')))
5cc3b3
		return false;
5cc3b3
5cc3b3
	if (strcmp(dot,".lo"))
5cc3b3
		return false;
5cc3b3
5cc3b3
	if (fpic) {
53aa85
		if ((slash = strrchr(arg,'/')))
53aa85
			slash++;
53aa85
		else
53aa85
			slash = arg;
53aa85
53aa85
		if ((size_t)snprintf(base,sizeof(base),"%s",
53aa85
				slash) >= sizeof(base))
53aa85
			return false;
53aa85
53aa85
		sprintf(slash,".libs/%s",base);
53aa85
		dot = strrchr(arg,'.');
5cc3b3
	}
53aa85
53aa85
	dot[1] = 'o';
53aa85
	dot[2] = '\0';
53aa85
	return true;
5cc3b3
}
5cc3b3
b3940a
static bool slbt_adjust_linker_argument(
b3940a
	char *		arg,
b3940a
	bool		fpic,
b3940a
	const char *	dsosuffix,
b3940a
	const char *	arsuffix)
b3940a
{
b3940a
	int	fdlib;
b3940a
	char *	slash;
b3940a
	char *	dot;
b3940a
	char	base[PATH_MAX];
b3940a
b3940a
	if (*arg == '-')
b3940a
		return false;
b3940a
b3940a
	if (!(dot = strrchr(arg,'.')))
b3940a
		return false;
b3940a
b3940a
	if (strcmp(dot,".la"))
b3940a
		return false;
b3940a
b3940a
	if (fpic) {
b3940a
		if ((slash = strrchr(arg,'/')))
b3940a
			slash++;
b3940a
		else
b3940a
			slash = arg;
b3940a
b3940a
		if ((size_t)snprintf(base,sizeof(base),"%s",
b3940a
				slash) >= sizeof(base))
b3940a
			return false;
b3940a
b3940a
		sprintf(slash,".libs/%s",base);
b3940a
		dot = strrchr(arg,'.');
b3940a
	}
b3940a
b3940a
	/* shared library dependency? */
b3940a
	if (fpic) {
b3940a
		sprintf(dot,"%s",dsosuffix);
b3940a
b3940a
		if ((fdlib = open(arg,O_RDONLY)) >= 0)
b3940a
			close(fdlib);
b3940a
		else
b3940a
			sprintf(dot,"%s",arsuffix);
b3940a
b3940a
		return true;
b3940a
	}
b3940a
b3940a
	/* input archive */
b3940a
	sprintf(dot,"%s",arsuffix);
b3940a
	return true;
b3940a
}
b3940a
628cde
static int slbt_exec_link_remove_file(
628cde
	const struct slbt_driver_ctx *	dctx,
628cde
	struct slbt_exec_ctx *		ectx,
628cde
	const char *			target)
628cde
{
628cde
	/* remove target (if any) */
628cde
	if (!(unlink(target)) || (errno == ENOENT))
628cde
		return 0;
628cde
628cde
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
628cde
		strerror(errno);
628cde
628cde
	return -1;
628cde
}
628cde
3b5e3d
static int slbt_exec_link_create_archive(
5cc3b3
	const struct slbt_driver_ctx *	dctx,
a51ace
	struct slbt_exec_ctx *		ectx,
3b5e3d
	const char *			arfilename,
3b5e3d
	bool				fpic)
5cc3b3
{
5cc3b3
	char ** 	aarg;
5cc3b3
	char ** 	parg;
5cc3b3
	char *		ranlib[3];
3c594d
	char		program[PATH_MAX];
3c594d
	char		output [PATH_MAX];
5cc3b3
e7ddb2
	/* initial state */
e7ddb2
	slbt_reset_arguments(ectx);
e7ddb2
5cc3b3
	/* placeholders */
5cc3b3
	slbt_reset_placeholders(ectx);
5cc3b3
5cc3b3
	/* alternate program (ar, ranlib) */
5cc3b3
	ectx->program = program;
5cc3b3
a51ace
	/* output */
a51ace
	if ((size_t)snprintf(output,sizeof(output),"%s",
a51ace
			arfilename) >= sizeof(output))
a51ace
		return -1;
a51ace
5cc3b3
	/* ar alternate argument vector */
5cc3b3
	if ((size_t)snprintf(program,sizeof(program),"%s",
5cc3b3
			dctx->cctx->host.ar) >= sizeof(program))
5cc3b3
		return -1;
5cc3b3
5cc3b3
	aarg    = ectx->altv;
5cc3b3
	*aarg++ = program;
5cc3b3
	*aarg++ = "cru";
a51ace
	*aarg++ = output;
5cc3b3
5cc3b3
	/* input argument adjustment */
5cc3b3
	for (parg=ectx->cargv; *parg; parg++)
3b5e3d
		if (slbt_adjust_input_argument(*parg,fpic))
5cc3b3
			*aarg++ = *parg;
5cc3b3
5cc3b3
	*aarg = 0;
5cc3b3
	ectx->argv = ectx->altv;
5cc3b3
5cc3b3
	/* step output */
5cc3b3
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
5cc3b3
		if (slbt_output_link(dctx,ectx))
5cc3b3
			return -1;
5cc3b3
628cde
	/* remove old archive as needed */
628cde
	if (slbt_exec_link_remove_file(dctx,ectx,output))
628cde
		return -1;
628cde
5cc3b3
	/* ar spawn */
5cc3b3
	if ((slbt_spawn(ectx,true) < 0) || ectx->exitcode)
5cc3b3
		return -1;
5cc3b3
5cc3b3
	/* ranlib argv */
5cc3b3
	if ((size_t)snprintf(program,sizeof(program),"%s",
5cc3b3
			dctx->cctx->host.ranlib) >= sizeof(program))
5cc3b3
		return -1;
5cc3b3
5cc3b3
	ranlib[0] = program;
a51ace
	ranlib[1] = output;
5cc3b3
	ranlib[2] = 0;
5cc3b3
	ectx->argv = ranlib;
5cc3b3
5cc3b3
	/* step output */
5cc3b3
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
5cc3b3
		if (slbt_output_link(dctx,ectx))
5cc3b3
			return -1;
5cc3b3
5cc3b3
	/* ranlib spawn */
5cc3b3
	if ((slbt_spawn(ectx,true) < 0) || ectx->exitcode)
5cc3b3
		return -1;
5cc3b3
5cc3b3
	return 0;
5cc3b3
}
5cc3b3
b3940a
static int slbt_exec_link_create_library(
b3940a
	const struct slbt_driver_ctx *	dctx,
b3940a
	struct slbt_exec_ctx *		ectx,
b3940a
	const char *			dsofilename)
b3940a
{
b3940a
	char ** parg;
b3940a
	char	output [PATH_MAX];
b3940a
b3940a
	/* initial state */
b3940a
	slbt_reset_arguments(ectx);
b3940a
b3940a
	/* placeholders */
b3940a
	slbt_reset_placeholders(ectx);
b3940a
b3940a
	/* using default argument vector */
b3940a
	ectx->argv    = ectx->cargv;
b3940a
	ectx->program = ectx->cargv[0];
b3940a
b3940a
	/* input argument adjustment */
b3940a
	for (parg=ectx->cargv; *parg; parg++)
b3940a
		slbt_adjust_input_argument(*parg,true);
b3940a
b3940a
	/* linker argument adjustment */
b3940a
	for (parg=ectx->cargv; *parg; parg++)
b3940a
		slbt_adjust_linker_argument(
b3940a
			*parg,true,
b3940a
			dctx->cctx->settings.dsosuffix,
b3940a
			dctx->cctx->settings.arsuffix);
b3940a
b3940a
	/* --no-undefined */
b3940a
	if (dctx->cctx->drvflags & SLBT_DRIVER_NO_UNDEFINED)
b3940a
		*ectx->noundef = "-Wl,--no-undefined";
b3940a
b3940a
	/* shared object */
b3940a
	*ectx->dpic = "-shared";
b3940a
	*ectx->fpic = "-fPIC";
b3940a
b3940a
	/* output */
b3940a
	if ((size_t)snprintf(output,sizeof(output),"%s.%d.%d.%d",
b3940a
				dsofilename,
b3940a
				dctx->cctx->verinfo.major,
b3940a
				dctx->cctx->verinfo.minor,
b3940a
				dctx->cctx->verinfo.revision)
b3940a
			>= sizeof(output))
b3940a
		return -1;
b3940a
b3940a
	*ectx->lout[0] = "-o";
b3940a
	*ectx->lout[1] = output;
b3940a
b3940a
	/* step output */
b3940a
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
b3940a
		if (slbt_output_link(dctx,ectx))
b3940a
			return -1;
b3940a
b3940a
	/* spawn */
b3940a
	if ((slbt_spawn(ectx,true) < 0) || ectx->exitcode)
b3940a
		return -1;
b3940a
b3940a
	return 0;
b3940a
}
b3940a
f6ccbe
static int slbt_exec_link_create_executable(
f6ccbe
	const struct slbt_driver_ctx *	dctx,
f6ccbe
	struct slbt_exec_ctx *		ectx,
f6ccbe
	const char *			exefilename)
f6ccbe
{
f6ccbe
	char ** parg;
f6ccbe
	char	output [PATH_MAX];
f6ccbe
f6ccbe
	/* initial state */
f6ccbe
	slbt_reset_arguments(ectx);
f6ccbe
f6ccbe
	/* placeholders */
f6ccbe
	slbt_reset_placeholders(ectx);
f6ccbe
f6ccbe
	/* using default argument vector */
f6ccbe
	ectx->argv    = ectx->cargv;
f6ccbe
	ectx->program = ectx->cargv[0];
f6ccbe
f6ccbe
	/* input argument adjustment */
f6ccbe
	for (parg=ectx->cargv; *parg; parg++)
f6ccbe
		slbt_adjust_input_argument(*parg,true);
f6ccbe
f6ccbe
	/* linker argument adjustment */
f6ccbe
	for (parg=ectx->cargv; *parg; parg++)
f6ccbe
		slbt_adjust_linker_argument(
f6ccbe
			*parg,true,
f6ccbe
			dctx->cctx->settings.dsosuffix,
f6ccbe
			dctx->cctx->settings.arsuffix);
f6ccbe
f6ccbe
	/* --no-undefined */
f6ccbe
	if (dctx->cctx->drvflags & SLBT_DRIVER_NO_UNDEFINED)
f6ccbe
		*ectx->noundef = "-Wl,--no-undefined";
f6ccbe
f6ccbe
	/* output */
f6ccbe
	if ((size_t)snprintf(output,sizeof(output),"%s",
f6ccbe
				exefilename)
f6ccbe
			>= sizeof(output))
f6ccbe
		return -1;
f6ccbe
f6ccbe
	*ectx->lout[0] = "-o";
f6ccbe
	*ectx->lout[1] = output;
f6ccbe
f6ccbe
	/* step output */
f6ccbe
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
f6ccbe
		if (slbt_output_link(dctx,ectx))
f6ccbe
			return -1;
f6ccbe
f6ccbe
	/* spawn */
f6ccbe
	if ((slbt_spawn(ectx,true) < 0) || ectx->exitcode)
f6ccbe
		return -1;
f6ccbe
f6ccbe
	return 0;
f6ccbe
}
f6ccbe
4c76f7
static int slbt_exec_link_create_symlink(
4c76f7
	const struct slbt_driver_ctx *	dctx,
4c76f7
	struct slbt_exec_ctx *		ectx,
4c76f7
	const char *			target,
d1091b
	char *				lnkname,
d1091b
	bool				flawrapper)
4c76f7
{
4c76f7
	const char *	slash;
628cde
	char *		ln[5];
d1091b
	char *		dotdot;
4c76f7
	char		atarget[PATH_MAX];
4c76f7
4c76f7
	/* atarget */
4c76f7
	if ((slash = strrchr(target,'/')))
4c76f7
		slash++;
4c76f7
	else
4c76f7
		slash = target;
4c76f7
d1091b
	dotdot = flawrapper ? "../" : "";
d1091b
d1091b
	if ((size_t)snprintf(atarget,sizeof(atarget),"%s%s",
d1091b
			dotdot,slash) >= sizeof(atarget))
4c76f7
		return -1;
4c76f7
4c76f7
	/* ln argv (fake) */
4c76f7
	ln[0] = "ln";
628cde
	ln[1] = "-s";
628cde
	ln[2] = atarget;
628cde
	ln[3] = lnkname;
a34bef
	ln[4] = 0;
4c76f7
	ectx->argv = ln;
4c76f7
4c76f7
	/* step output */
4c76f7
	if (!(dctx->cctx->drvflags & SLBT_DRIVER_SILENT))
4c76f7
		if (slbt_output_link(dctx,ectx))
4c76f7
			return -1;
4c76f7
628cde
	/* remove old symlink as needed */
628cde
	if (slbt_exec_link_remove_file(dctx,ectx,lnkname))
628cde
		return -1;
4c76f7
4c76f7
	/* create symlink */
4c76f7
	return symlink(atarget,lnkname);
4c76f7
}
5cc3b3
b3940a
static int slbt_exec_link_create_library_symlink(
b3940a
	const struct slbt_driver_ctx *	dctx,
b3940a
	struct slbt_exec_ctx *		ectx,
b3940a
	bool				fmajor)
b3940a
{
b3940a
	char	target[PATH_MAX];
b3940a
	char	lnkname[PATH_MAX];
b3940a
b3940a
	sprintf(target,"%s.%d.%d.%d",
b3940a
		ectx->dsofilename,
b3940a
		dctx->cctx->verinfo.major,
b3940a
		dctx->cctx->verinfo.minor,
b3940a
		dctx->cctx->verinfo.revision);
b3940a
b3940a
	if (fmajor)
b3940a
		sprintf(lnkname,"%s.%d",
b3940a
			ectx->dsofilename,
b3940a
			dctx->cctx->verinfo.major);
b3940a
b3940a
	else
b3940a
		strcpy(lnkname,ectx->dsofilename);
b3940a
b3940a
	return slbt_exec_link_create_symlink(
b3940a
		dctx,ectx,
b3940a
		target,lnkname,
b3940a
		false);
b3940a
}
b3940a
2bd749
int slbt_exec_link(
2bd749
	const struct slbt_driver_ctx *	dctx,
2bd749
	struct slbt_exec_ctx *		ectx)
2bd749
{
2bd749
	int			ret;
2bd749
	int			fdlibs;
a51ace
	const char *		output;
2bd749
	char *			dot;
2bd749
	FILE *			fout;
2bd749
	struct slbt_exec_ctx *	actx;
2bd749
2bd749
	/* context */
2bd749
	if (ectx)
2bd749
		actx = 0;
2bd749
	else if ((ret = slbt_get_exec_ctx(dctx,&ectx)))
2bd749
		return ret;
2bd749
	else
2bd749
		actx = ectx;
2bd749
5cc3b3
	/* output suffix */
a51ace
	output = dctx->cctx->output;
a51ace
	dot    = strrchr(output,'.');
5cc3b3
2bd749
	/* .libs directory */
2bd749
	if (dctx->cctx->drvflags & SLBT_DRIVER_SHARED) {
2bd749
		if ((fdlibs = open(ectx->ldirname,O_DIRECTORY)) >= 0)
2bd749
			close(fdlibs);
2bd749
		else if ((errno != ENOENT) || mkdir(ectx->ldirname,0777)) {
2bd749
			slbt_free_exec_ctx(actx);
2bd749
			return -1;
2bd749
		}
2bd749
	}
2bd749
5cc3b3
	/* non-pic libfoo.a */
5cc3b3
	if (dot && !strcmp(dot,".a"))
3b5e3d
		if (slbt_exec_link_create_archive(dctx,ectx,output,false)) {
5cc3b3
			slbt_free_exec_ctx(actx);
5cc3b3
			return -1;
5cc3b3
		}
5cc3b3
a0c318
	/* pic libfoo.a */
a0c318
	if (dot && !strcmp(dot,".la"))
a0c318
		if (slbt_exec_link_create_archive(
a0c318
				dctx,ectx,
a0c318
				ectx->arfilename,
5fff27
				dctx->cctx->drvflags & SLBT_DRIVER_SHARED)) {
a0c318
			slbt_free_exec_ctx(actx);
a0c318
			return -1;
a0c318
		}
a0c318
b3940a
	/* dynamic library */
b3940a
	if (dot && !strcmp(dot,".la") && dctx->cctx->rpath) {
b3940a
		/* linking: libfoo.so.x.y.z */
b3940a
		if (slbt_exec_link_create_library(
b3940a
				dctx,ectx,
b3940a
				ectx->dsofilename)) {
b3940a
			slbt_free_exec_ctx(actx);
b3940a
			return -1;
b3940a
		}
b3940a
b3940a
		/* symlink: libfoo.so.x --> libfoo.so.x.y.z */
b3940a
		if (slbt_exec_link_create_library_symlink(
b3940a
				dctx,ectx,
b3940a
				true)) {
b3940a
			slbt_free_exec_ctx(actx);
b3940a
			return -1;
b3940a
		}
b3940a
b3940a
		/* symlink: libfoo.so --> libfoo.so.x.y.z */
b3940a
		if (slbt_exec_link_create_library_symlink(
b3940a
				dctx,ectx,
b3940a
				false)) {
b3940a
			slbt_free_exec_ctx(actx);
b3940a
			return -1;
b3940a
		}
b3940a
	}
b3940a
f6ccbe
	/* executable */
f6ccbe
	if (!dctx->cctx->rpath && !dctx->cctx->libname) {
f6ccbe
		/* linking: .libs/exefilename */
f6ccbe
		if (slbt_exec_link_create_executable(
f6ccbe
				dctx,ectx,
f6ccbe
				ectx->exefilename)) {
f6ccbe
			slbt_free_exec_ctx(actx);
f6ccbe
			return -1;
f6ccbe
		}
f6ccbe
	}
f6ccbe
2bd749
	/* no wrapper? */
5cc3b3
	if (!dot || strcmp(dot,".la")) {
2bd749
		slbt_free_exec_ctx(actx);
2bd749
		return 0;
2bd749
	}
2bd749
2bd749
	/* hey, yo, let's rap it up */
687567
	if (!(fout = fopen(output,"w"))) {
2bd749
		slbt_free_exec_ctx(actx);
2bd749
		return -1;
2bd749
	}
2bd749
2bd749
	ret = fprintf(fout,
2bd749
		"# slibtool (pre-alphe) generated file\n\n");
2bd749
4c76f7
	/* wrapper symlink */
4c76f7
	if (slbt_exec_link_create_symlink(
4c76f7
			dctx,ectx,
4c76f7
			output,
d1091b
			ectx->lafilename,
d1091b
			true)) {
4c76f7
		slbt_free_exec_ctx(actx);
4c76f7
		return -1;
4c76f7
	}
4c76f7
2bd749
	/* all done */
2bd749
	fclose(fout);
2bd749
	slbt_free_exec_ctx(actx);
2bd749
2bd749
	return (ret > 0) ? 0 : -1;
2bd749
}