summaryrefslogtreecommitdiff
path: root/perk/perk.h
blob: 28bee7701976dcaca32109a2bf5f21cccf94bfcb (plain)
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
#ifndef PERK_H
#define PERK_H

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

#include "perk_consts.h"
#include "perk_structs.h"
#include "perk_meta.h"

/* flags */
#define PERK_HELP		0x01

/* errors */
#define PERK_OK			0x00
#define PERK_BAD_OPT		0xA1
#define PERK_BAD_OPT_VAL	0xA2
#define PERK_IO_ERROR		0xA3
#define PERK_MAP_ERROR		0xA4
#define PERK_BAD_DOS_HEADER	0xA5
#define PERK_BAD_COFF_HEADER	0xA6
#define PERK_BAD_PE_IMAGE_TYPE	0xA7

struct pe_raw_image {
	char *	addr;
	size_t	size;
};

struct pe_image_meta {
	struct pe_raw_image		image;

	struct pe_meta_image_dos_hdr	dos;
	struct pe_meta_coff_file_hdr	coff;
	struct pe_meta_opt_hdr		opt;
	struct pe_meta_sec_hdr *	sectbl;

	struct pe_image_dos_hdr *	ados;
	struct pe_coff_file_hdr *	acoff;
	union  pe_opt_hdr *		aopt;
	struct pe_sec_hdr *		asectbl;

	struct pe_meta_export_hdr	edata;
	struct pe_export_hdr *		aedata;
	struct pe_meta_sec_hdr *	hedata;
};

/* utility api */
int perk_main			(int argc, char * argv[], char * envp[]);

/* high-level api */
int pe_map_raw_image		(int fd, char * fname,		struct pe_raw_image *);
int pe_unmap_raw_image		(struct pe_raw_image *);

int pe_get_image_meta		(struct pe_raw_image * image,	struct pe_image_meta ** meta);
int pe_free_image_meta		(struct pe_image_meta * meta);

int pe_get_named_section_index	(struct pe_image_meta * meta,	char * name);
int pe_output_export_symbols	(struct pe_image_meta * meta,	FILE * stream);

/* low-level api */
int pe_read_dos_header		(struct pe_image_dos_hdr * p,	struct pe_meta_image_dos_hdr * m);
int pe_read_coff_header		(struct pe_coff_file_hdr * p,	struct pe_meta_coff_file_hdr * m);
int pe_read_optional_header	(union  pe_opt_hdr * p,		struct pe_meta_opt_hdr * m);
int pe_read_section_header	(struct pe_sec_hdr * p,		struct pe_meta_sec_hdr * m);
int pe_read_export_header	(struct pe_export_hdr * p,	struct pe_meta_export_hdr * m);

#endif