#include <stdint.h>

#include <stdlib.h>

#include <string.h>

#include <sys/mman.h>

#include <perk/perk.h>

static int pe_free_unit_ctx_impl(struct pe_unit_ctx * ctx, int status)

{

	if (ctx) {

		pe_free_image_meta(ctx->meta);

		pe_unmap_raw_image(&ctx->map);

		free(ctx);

	}

	return status;

}

int pe_get_unit_ctx(

	struct pe_driver_ctx *	dctx,

	const char *		path,

	struct pe_unit_ctx **	pctx)

{

	struct pe_unit_ctx *	ctx;

	int			prot;

	if (!dctx || !(ctx = calloc(sizeof(*ctx),1)))

		return -1;

	prot = (dctx->cctx.actflags & PERK_ACTION_FLAG_MAP_READWRITE)

		? PROT_READ | PROT_WRITE

		: PROT_READ;

	if (pe_map_raw_image(dctx->cctx.fdin,path,prot,&ctx->map))

		return pe_free_unit_ctx_impl(ctx,-1);

	if (pe_get_image_meta(&ctx->map,&ctx->meta))

		return pe_free_unit_ctx_impl(ctx,-1);

	memcpy(&ctx->cctx,&dctx->cctx,

		sizeof(ctx->cctx));

	ctx->path	= path;

	ctx->cctx.prot	= prot;

	*pctx = ctx;

	return 0;

}

void pe_free_unit_ctx(struct pe_unit_ctx * ctx)

{

	pe_free_unit_ctx_impl(ctx,0);

}