/****************************************************************/
/*  mdso: midipix dso scavenger                                 */
/*  Copyright (C) 2015--2017  Z. Gilboa                         */
/*  Released under GPLv2 and GPLv3; see COPYING.MDSO.           */
/****************************************************************/

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <mdso/mdso.h>
#include "mdso_errinfo_impl.h"

static void mdso_free_uctx_vector(
	struct mdso_unit_ctx **	uctxv,
	const char **		symv,
	int *			stype,
	FILE *			fout)
{
	struct mdso_unit_ctx **	puctx;

	if (symv)
		free(symv);

	if (stype)
		free(stype);

	for (puctx=uctxv; *puctx; puctx++)
		mdso_free_unit_ctx(*puctx);

	free(uctxv);
	fclose(fout);
}

int  mdso_create_implib_archive(const struct mdso_driver_ctx * dctx)
{
	int			ret;
	FILE *			fout;
	size_t			nsym;
	struct mdso_unit_ctx **	uctxv;
	struct mdso_unit_ctx **	puctx;
	const char * const *	dsym;
	const char **		unit;
	const char **		psym;
	const char **		symv;
	int *			stype;

	if (!dctx->cctx->implib)
		return MDSO_CUSTOM_ERROR(dctx,MDSO_ERR_INVALID_NAME);

	for (unit=dctx->units; *unit; unit++)
		(void)0;

	if (!(uctxv = calloc(++unit - dctx->units,sizeof(*uctxv))))
		return MDSO_SYSTEM_ERROR(dctx);

	if (!(fout = mdso_create_archive(dctx,dctx->cctx->implib)))
		return MDSO_NESTED_ERROR(dctx);

	for (puctx=uctxv,unit=dctx->units; *unit; unit++) {
		if (mdso_get_unit_ctx(dctx,*unit,puctx)) {
			mdso_free_uctx_vector(uctxv,0,0,fout);
			return MDSO_NESTED_ERROR(dctx);
		}
	}

	for (nsym=0,puctx=uctxv; *puctx; puctx++)
		for (dsym=puctx[0]->syms; *dsym; dsym++)
			nsym++;

	if (!(symv = calloc(nsym+1,sizeof(const char *)))) {
		mdso_free_uctx_vector(uctxv,0,0,fout);
		return MDSO_SYSTEM_ERROR(dctx);
	}

	if (!(stype = calloc(nsym+1,sizeof(int)))) {
		mdso_free_uctx_vector(uctxv,symv,0,fout);
		return MDSO_SYSTEM_ERROR(dctx);
	}

	for (psym=symv,puctx=uctxv; *puctx; puctx++) {
		for (dsym=puctx[0]->syms; *dsym; dsym++) {
			stype[psym-symv] = puctx[0]->stype[dsym-puctx[0]->syms];
			*psym++ = *dsym;
		}
	}

	ret = mdso_argen_common(dctx,symv,stype,fout,0);
	mdso_free_uctx_vector(uctxv,symv,stype,fout);

	return ret ? MDSO_NESTED_ERROR(dctx) : 0;
}