1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
/***************************************************************/
/* perk: PE Resource Kit */
/* Copyright (C) 2015--2016 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.PERK. */
/***************************************************************/
#include <stdint.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <perk/perk.h>
#include "perk_driver_impl.h"
#include "perk_errinfo_impl.h"
static int pe_free_unit_ctx_impl(struct pe_unit_ctx_impl * ctx, int ret)
{
if (ctx) {
pe_free_image_meta(ctx->meta);
pe_unmap_raw_image(&ctx->map);
free(ctx);
}
return ret;
}
int pe_get_unit_ctx(
const struct pe_driver_ctx * dctx,
const char * path,
struct pe_unit_ctx ** pctx)
{
struct pe_unit_ctx_impl * ctx;
int prot;
if (!dctx)
return PERK_CUSTOM_ERROR(dctx,0);
else if (!(ctx = calloc(1,sizeof(*ctx))))
return PERK_BUFFER_ERROR(dctx);
prot = (dctx->cctx->actflags & PERK_ACTION_MAP_READWRITE)
? PROT_READ | PROT_WRITE
: PROT_READ;
if (pe_map_raw_image(dctx,dctx->cctx->ioctx->fdin,path,prot,&ctx->map))
return pe_free_unit_ctx_impl(ctx,
PERK_SYSTEM_ERROR(dctx));
if (pe_get_image_meta(dctx,&ctx->map,&ctx->meta))
return pe_free_unit_ctx_impl(ctx,
PERK_NESTED_ERROR(dctx));
memcpy(&ctx->cctx,dctx->cctx,
sizeof(ctx->cctx));
ctx->path = path;
ctx->ioctx.prot = prot;
ctx->uctx.path = &ctx->path;
ctx->uctx.map = &ctx->map;
ctx->uctx.meta = ctx->meta;
ctx->uctx.cctx = &ctx->cctx;
*pctx = &ctx->uctx;
return 0;
}
void pe_free_unit_ctx(struct pe_unit_ctx * ctx)
{
struct pe_unit_ctx_impl * ictx;
uintptr_t addr;
if (ctx) {
addr = (uintptr_t)ctx - offsetof(struct pe_unit_ctx_impl,uctx);
ictx = (struct pe_unit_ctx_impl *)addr;
pe_free_unit_ctx_impl(ictx,0);
}
}
|
