summaryrefslogtreecommitdiff
path: root/perk/perk.h
blob: 7b7a5f7bcf5d79f975b4f5f9cee62e2e8e945cdd (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
#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_image_dos_hdr *	ados;
	struct pe_coff_file_hdr *	acoff;
	union  pe_opt_hdr *		aopt;

	struct pe_meta_image_dos_hdr	dos;
	struct pe_meta_coff_file_hdr	coff;
	struct pe_meta_opt_hdr		opt;
};

/* 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		(char * image, struct pe_image_meta ** meta);
int pe_free_image_meta		(struct pe_image_meta * meta);

/* 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 perk_main(int argc, char * argv[], char * envp[]);

#endif