diff --git a/include/slibtool/slibtool.h b/include/slibtool/slibtool.h index bd5b31d..e0e92d3 100644 --- a/include/slibtool/slibtool.h +++ b/include/slibtool/slibtool.h @@ -226,6 +226,10 @@ struct slbt_exec_ctx { char * aobjname; char * ltobjname; + char * dlunit; + char * dlopensrc; + char * dlopenobj; + char * arfilename; char * lafilename; char * laifilename; diff --git a/src/internal/slibtool_driver_impl.h b/src/internal/slibtool_driver_impl.h index b540e6a..d838aea 100644 --- a/src/internal/slibtool_driver_impl.h +++ b/src/internal/slibtool_driver_impl.h @@ -149,6 +149,7 @@ struct slbt_driver_ctx_impl { struct slbt_map_info lconf; struct slbt_txtfile_ctx * lconfctx; struct slbt_obj_list * objlistv; + const char ** dlopenv; size_t ndlopen; @@ -177,12 +178,15 @@ struct slbt_exec_ctx_impl { const struct slbt_driver_ctx * dctx; struct slbt_symlist_ctx * sctx; struct slbt_exec_ctx ctx; + struct slbt_archive_ctx ** dlactxv; + char ** dlargv; int argc; char * args; char * shadow; char * dsoprefix; size_t size; int fdwrapper; + char sbuf[PATH_MAX]; char ** lout[2]; char ** mout[2]; char ** vbuffer; diff --git a/src/logic/slbt_exec_ctx.c b/src/logic/slbt_exec_ctx.c index ec8ea3e..583001d 100644 --- a/src/logic/slbt_exec_ctx.c +++ b/src/logic/slbt_exec_ctx.c @@ -11,11 +11,17 @@ #include #include "slibtool_driver_impl.h" +#include "slibtool_linkcmd_impl.h" #include "slibtool_errinfo_impl.h" +#include "slibtool_ar_impl.h" -#define SLBT_ECTX_LIB_EXTRAS 20 +#define SLBT_ECTX_LIB_EXTRAS 24 #define SLBT_ECTX_SPARE_PTRS 16 +static int slbt_ectx_free_exec_ctx_impl( + struct slbt_exec_ctx_impl * ictx, + int status); + static size_t slbt_parse_comma_separated_flags( const char * str, int * argc) @@ -75,6 +81,8 @@ static struct slbt_exec_ctx_impl * slbt_exec_ctx_alloc( char * shadow; char * csrc; char ** parg; + struct slbt_archive_ctx ** dlactxv; + size_t ndlopen; /* internal driver context for host-specific tool arguments */ ctx = slbt_get_driver_ictx(dctx); @@ -137,19 +145,31 @@ static struct slbt_exec_ctx_impl * slbt_exec_ctx_alloc( argc += SLBT_ECTX_SPARE_PTRS; /* ctx alloc and vector alloc: argv, xargv, and altv, where we */ - /* assume -Wl,--whole-archive arg -Wl,--no-whole-archive */ + /* assume -Wl,--whole-archive arg -Wl,--no-whole-archive; */ + /* and also dlargv for compiling dlunit.dlopen.c */ if (!(ictx = calloc(1,sizeof(*ictx)))) { free(args); free(shadow); return 0; } - if (!(ictx->vbuffer = calloc(5*(argc+1),sizeof(char *)))) { + if (!(ictx->vbuffer = calloc(6*(argc+1),sizeof(char *)))) { free(args); free(shadow); free(ictx); } + if ((ndlopen = (slbt_get_driver_ictx(dctx))->ndlopen)) { + if (!(dlactxv = calloc(ndlopen+1,sizeof(*dlactxv)))) { + free(ictx->vbuffer); + free(ictx); + free(args); + free(shadow); + } + + ictx->dlactxv = dlactxv; + } + /* all ready */ ictx->dctx = dctx; ictx->args = args; @@ -172,6 +192,7 @@ int slbt_ectx_get_exec_ctx( struct slbt_exec_ctx ** ectx) { struct slbt_exec_ctx_impl * ictx; + struct slbt_driver_ctx_impl * idctx; char ** parg; char ** src; char ** dst; @@ -179,12 +200,18 @@ int slbt_ectx_get_exec_ctx( char * mark; const char * dmark; char * slash; + char * arname; + struct slbt_archive_ctx ** dlactxv; + const char ** dlopenv; const char * arprefix; const char * dsoprefix; const char * impprefix; const char * ref; int i; + /* internal driver context */ + idctx = slbt_get_driver_ictx(dctx); + /* alloc */ if (!(ictx = slbt_exec_ctx_alloc(dctx))) return SLBT_NESTED_ERROR(dctx); @@ -194,6 +221,7 @@ int slbt_ectx_get_exec_ctx( ictx->ctx.argv = ictx->vbuffer; ictx->ctx.xargv = &ictx->ctx.argv [ictx->argc + 1]; ictx->ctx.altv = &ictx->ctx.xargv[ictx->argc + 1]; + ictx->dlargv = &ictx->ctx.altv [ictx->argc + 1]; /* */ ictx->ctx.compiler = dctx->cctx->cargv[0]; @@ -500,6 +528,63 @@ int slbt_ectx_get_exec_ctx( dctx->cctx->settings.dsosuffix); ch++; } + + /* dlopensrc, dlopenobj */ + if (idctx->ndlopen) { + ictx->ctx.dlopensrc = ch; + ch += sprintf(ch,"%s%s%s.dlopen.c", + ictx->ctx.ldirname, + dsoprefix, + dctx->cctx->libname); + + ch++; + + ictx->ctx.dlopenobj = ch; + ch += sprintf(ch,"%s%s%s.dlopen.o", + ictx->ctx.ldirname, + dsoprefix, + dctx->cctx->libname); + + ch++; + + ictx->ctx.dlunit = ch; + ch += sprintf(ch,"%s%s", + dsoprefix, + dctx->cctx->libname); + + ch++; + } + } + + /* dlopen, dlpreopen */ + if ((dlopenv = idctx->dlopenv), (dlactxv = ictx->dlactxv)) { + for (; *dlopenv; ) { + arname = ictx->sbuf; + strcpy(arname,*dlopenv); + slbt_adjust_wrapper_argument(arname,true); + + if (slbt_ar_get_archive_ctx(dctx,arname,dlactxv) < 0) + return slbt_ectx_free_exec_ctx_impl( + ictx, + SLBT_NESTED_ERROR(dctx)); + + if (slbt_ar_update_syminfo(*dlactxv,&ictx->ctx) < 0) + return slbt_ectx_free_exec_ctx_impl( + ictx, + SLBT_NESTED_ERROR(dctx)); + + dlopenv++; + dlactxv++; + } + + if (slbt_ar_create_dlsyms( + ictx->dlactxv, + ictx->ctx.dlunit, + ictx->ctx.dlopensrc, + 0644) < 0) + return slbt_ectx_free_exec_ctx_impl( + ictx, + SLBT_NESTED_ERROR(dctx)); } /* linking: exefilename */ @@ -547,12 +632,21 @@ static int slbt_ectx_free_exec_ctx_impl( struct slbt_exec_ctx_impl * ictx, int status) { + struct slbt_archive_ctx ** dlactxv; + if (ictx->sctx) slbt_lib_free_symlist_ctx(ictx->sctx); if (ictx->fdwrapper >= 0) close(ictx->fdwrapper); + if (ictx->dlactxv) { + for (dlactxv=ictx->dlactxv; *dlactxv; dlactxv++) + slbt_ar_free_archive_ctx(*dlactxv); + + free(ictx->dlactxv); + } + free(ictx->args); free(ictx->shadow); free(ictx->vbuffer);