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

/*  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"

#include "perk_structs.h"

static void mdso_argen_common_hdr(

	struct pe_raw_archive_common_hdr *	arhdr,

	char *					file_id,

	size_t					size)

{

	size_t	slen;

	char	sbuf[10];

	memset(arhdr,0x20,sizeof(*arhdr));

	slen = strlen(file_id);

	memcpy(arhdr->ar_file_id,file_id,slen);

	arhdr->ar_file_id[slen] = '/';

	arhdr->ar_uid[0] = '0';

	arhdr->ar_gid[0] = '0';

	arhdr->ar_file_mode[0] = '0';

	slen = sprintf(sbuf,"%zu",size);

	memcpy(arhdr->ar_file_size,sbuf,slen);

	arhdr->ar_end_tag[0] = 0x60;

	arhdr->ar_end_tag[1] = 0x0a;

}

static void mdso_write_big_endian_long(unsigned char * ch, uint32_t val)

{

	ch[3] = val;

	ch[2] = val >> 8;

	ch[1] = val >> 16;

	ch[0] = val >> 24;

}

int  mdso_argen_common(

	const struct mdso_driver_ctx *	dctx,

	const char **			symv,

	const int *			stype,

	FILE *				fout,

	struct mdso_object *		vobj)

{

	int					ret;

	const char **				psym;

	int					nsym;

	int					nobj;

	int					ndata;

	uint32_t				objlen;

	uint32_t				hdrlen;

	uint32_t				mapstrsnum;

	uint32_t				mapstrslen;

	uint32_t				symidx;

	unsigned char *				ar;

	unsigned char *				mark;

	unsigned char *				idx;

	char *					mapstrs;

	struct pe_raw_archive_common_hdr *	arhdr;

	struct mdso_object *			pobj;

	struct mdso_object *			aobj;

	struct mdso_object			sobj[256];

	char					objname[16];

	/* init */

	memset (sobj,0,sizeof(sobj));

	for (nsym=0,ndata=0,psym=symv; *psym; psym++,nsym++)

		ndata += (stype[psym-symv] == MDSO_SYMBOL_TYPE_DATA);

	if ((nobj = 1 + (2*nsym) - ndata) < 256)

		aobj = sobj;

	else if (nobj > 1024*1024)

		return MDSO_CUSTOM_ERROR(dctx,MDSO_ERR_INVALID_VECTOR);

	else if (!(aobj = calloc(1,nobj*sizeof(*aobj))))

		return MDSO_SYSTEM_ERROR(dctx);

	/* objlen: archive signature, index header */

	objlen  = 8;

	objlen += sizeof(struct pe_raw_archive_common_hdr);

	/* objlen: member headers */

	ret = mdso_objgen_dsometa(dctx,0,aobj);

	aobj->size += 1;

	aobj->size |= 1;

	aobj->size ^= 1;

	objlen += aobj->size;

	objlen += sizeof(struct pe_raw_archive_common_hdr);

	mapstrslen = aobj->mapstrslen;

	mapstrsnum = aobj->mapstrsnum;

	for (psym=symv,pobj=&aobj[1]; *psym && !ret; psym++) {

		if (stype[psym-symv] == MDSO_SYMBOL_TYPE_CODE) {

			ret  = mdso_objgen_symfn(dctx,*psym,0,pobj);

			pobj->size += 1;

			pobj->size |= 1;

			pobj->size ^= 1;

			objlen += pobj->size;

			objlen += sizeof(struct pe_raw_archive_common_hdr);

			mapstrslen += pobj->mapstrslen;

			mapstrsnum += pobj->mapstrsnum;

			pobj++;

		}

		ret |= mdso_objgen_symentry(dctx,*psym,0,pobj);

		pobj->size += 1;

		pobj->size |= 1;

		pobj->size ^= 1;

		objlen += pobj->size;

		objlen += sizeof(struct pe_raw_archive_common_hdr);

		mapstrslen += pobj->mapstrslen;

		mapstrsnum += pobj->mapstrsnum;

		pobj++;

	}

	if (ret && (aobj != sobj))

		free(aobj);

	if (ret)

		return ret;

	/* index: string block alignment */

	mapstrslen += 1;

	mapstrslen |= 1;

	mapstrslen ^= 1;

	/* objlen: index size, padding */

	objlen += sizeof(uint32_t) * (1 + mapstrsnum);

	objlen += mapstrslen;

	objlen += 15;

	objlen |= 15;

	objlen ^= 15;

	if (vobj && vobj->addr && (vobj->size < objlen))

		return MDSO_BUFFER_ERROR(dctx);

	if (vobj && !vobj->addr) {

		vobj->size = objlen;

		vobj->mapstrslen = mapstrslen;

		vobj->mapstrsnum = mapstrsnum;

		return 0;

	}

	if (vobj)

		ar = (unsigned char *)vobj->addr;

	else if (!(ar = calloc(1,objlen))) {

		if (aobj != sobj)

			free(aobj);

		return MDSO_SYSTEM_ERROR(dctx);

	}

	/* archive signature */

	memcpy(ar,"!<arch>\n",8);

	mark = &ar[8];

	/* archive header */

	arhdr  = (struct pe_raw_archive_common_hdr *)mark;

	hdrlen = sizeof(uint32_t) * (1 + mapstrsnum) + mapstrslen;

	mdso_argen_common_hdr(arhdr,"",hdrlen);

	mark += sizeof(*arhdr);

	/* archive index initialization */

	mdso_write_big_endian_long(mark,mapstrsnum);

	mark += sizeof(uint32_t);

	idx   = mark;

	mark += sizeof(uint32_t) * mapstrsnum;

	mapstrs = (char *)mark;

	mark   += mapstrslen;

	/* .dsometa object */

	aobj->mapstrs  = mapstrs;

	aobj->arhdrpos = (uint32_t)(mark - ar);

	aobj->arhdrlen = sizeof(struct pe_raw_archive_common_hdr);

	aobj->addr     = &mark[aobj->arhdrlen];

	for (symidx=0; symidx<aobj->mapstrsnum; symidx++) {

		mdso_write_big_endian_long(idx,aobj->arhdrpos);

		idx += sizeof(uint32_t);

	}

	ret = mdso_objgen_dsometa(dctx,0,aobj);

	mdso_argen_common_hdr(

		(struct pe_raw_archive_common_hdr *)mark,

		".dsometa.o",aobj->size);

	mark    += aobj->arhdrlen + aobj->size;

	mapstrs += aobj->mapstrslen;

	/* archive symfn and symentry objects */

	for (psym=symv,pobj=&aobj[1]; *psym && !ret; psym++) {

		/* symfn object */

		if (stype[psym-symv] == MDSO_SYMBOL_TYPE_CODE) {

			pobj->mapstrs  = mapstrs;

			pobj->arhdrpos = (uint32_t)(mark - ar);

			pobj->arhdrlen = sizeof(struct pe_raw_archive_common_hdr);

			pobj->addr     = &mark[pobj->arhdrlen];

			for (symidx=0; symidx<pobj->mapstrsnum; symidx++) {

				mdso_write_big_endian_long(idx,pobj->arhdrpos);

				idx += sizeof(uint32_t);

			}

			ret = mdso_objgen_symfn(dctx,*psym,0,pobj);

			sprintf(

				objname,"f%06zu.o",

				psym - symv);

			mdso_argen_common_hdr(

				(struct pe_raw_archive_common_hdr *)mark,

				objname,pobj->size);

			mark    += pobj->arhdrlen + pobj->size;

			mapstrs += pobj->mapstrslen;

			pobj++;

		}

		/* symentry object */

		pobj->mapstrs  = mapstrs;

		pobj->arhdrpos = (uint32_t)(mark - ar);

		pobj->arhdrlen = sizeof(struct pe_raw_archive_common_hdr);

		pobj->addr     = &mark[pobj->arhdrlen];

		for (symidx=0; symidx<pobj->mapstrsnum; symidx++) {

			mdso_write_big_endian_long(idx,pobj->arhdrpos);

			idx += sizeof(uint32_t);

		}

		sprintf(

			objname,"s%06zu.o",

			psym - symv);

		ret = mdso_objgen_symentry(dctx,*psym,0,pobj);

		mdso_argen_common_hdr(

			(struct pe_raw_archive_common_hdr *)mark,

			objname,pobj->size);

		mark    += pobj->arhdrlen + pobj->size;

		mapstrs += pobj->mapstrslen;

		pobj++;

	}

	if (ret) {

		if (aobj != sobj)

			free(aobj);

		if (!vobj)

			free(ar);

		return ret;

	}

	/* tada */

	if (fout)

		ret = fwrite(ar,objlen,1,fout);

	if (aobj != sobj)

		free(aobj);

	if (!vobj)

		free(ar);

	return (ret == 0)

		? MDSO_FILE_ERROR(dctx)

		: 0;

}