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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
|
#ifndef PERK_H
#define PERK_H
#include <stdint.h>
#include <stdio.h>
#include "perk_api.h"
#include "perk_consts.h"
#include "perk_structs.h"
#include "perk_meta.h"
#ifdef __cplusplus
extern "C" {
#endif
/* pre-alpha */
#ifndef PERK_APP
#ifndef PERK_PRE_ALPHA
#error libperk: pre-alpha: ABI is not final!
#error to use the library, please pass -DPERK_PRE_ALPHA to the compiler.
#endif
#endif
/* status codes */
#define PERK_OK 0x00
#define PERK_USAGE 0x01
#define PERK_ERROR 0x02
/* driver flags */
#define PERK_DRIVER_VERBOSITY_NONE 0x0000
#define PERK_DRIVER_VERBOSITY_ERRORS 0x0001
#define PERK_DRIVER_VERBOSITY_STATUS 0x0002
#define PERK_DRIVER_VERBOSITY_USAGE 0x0004
#define PERK_DRIVER_CLONE_VECTOR 0x0008
#define PERK_DRIVER_VERSION 0x0010
#define PERK_DRIVER_DRY_RUN 0x0020
#define PERK_DRIVER_ANNOTATE_ALWAYS 0x1000
#define PERK_DRIVER_ANNOTATE_NEVER 0x2000
#define PERK_DRIVER_ANNOTATE_FULL 0x4000
/* unit action flags */
#define PERK_ACTION_MAP_READWRITE 0x0001
/* error flags */
#define PERK_ERROR_TOP_LEVEL 0x0001
#define PERK_ERROR_NESTED 0x0002
#define PERK_ERROR_CHILD 0x0004
#define PERK_ERROR_CUSTOM 0x0008
enum pe_custom_error {
PERK_ERR_FLOW_ERROR,
PERK_ERR_FLEE_ERROR,
PERK_ERR_NULL_CONTEXT,
PERK_ERR_NULL_IMAGE,
PERK_ERR_INVALID_CONTEXT,
PERK_ERR_INVALID_IMAGE,
PERK_ERR_IMAGE_SIZE_ZERO,
PERK_ERR_IMAGE_MALFORMED,
PERK_ERR_BAD_DOS_HEADER,
PERK_ERR_BAD_COFF_HEADER,
PERK_ERR_BAD_IMAGE_TYPE,
PERK_ERR_UNSUPPORTED_ABI,
PERK_ERR_CAP,
};
struct pe_raw_image {
void * map_addr;
size_t map_size;
};
struct pe_expsym {
const char * s_name;
void * s_eaddr;
void * s_maddr;
uint32_t s_roffset;
};
struct pe_meta_stats {
int32_t t_nexpsyms;
int32_t t_nimplibs;
int32_t t_ndsolibs;
int32_t t_ndsosyms;
int32_t t_nrelocs;
};
struct pe_image_meta {
struct pe_raw_image r_image;
struct pe_raw_image_dos_hdr * r_dos;
struct pe_raw_coff_image_hdr * r_coff;
struct pe_raw_coff_object_hdr * r_obj;
union pe_raw_opt_hdr * r_opt;
struct pe_raw_sec_hdr * r_sectbl;
struct pe_raw_export_hdr * r_edata;
struct pe_raw_import_hdr * r_idata;
void * r_dsometa;
void * r_dsosyms;
char * r_dsostrs;
void * r_dsodata;
struct pe_meta_stats m_stats;
struct pe_meta_image_dos_hdr m_dos;
struct pe_meta_coff_file_hdr m_coff;
struct pe_meta_opt_hdr m_opt;
struct pe_meta_sec_hdr * m_sectbl;
struct pe_meta_export_hdr m_edata;
struct pe_meta_import_hdr * m_idata;
struct pe_meta_sec_hdr * h_edata;
struct pe_meta_sec_hdr * h_idata;
struct pe_meta_sec_hdr * h_dsometa;
struct pe_meta_sec_hdr * h_dsosyms;
struct pe_meta_sec_hdr * h_dsostrs;
struct pe_meta_sec_hdr * h_dsodata;
};
struct pe_source_version {
int major;
int minor;
int revision;
const char * commit;
};
struct pe_fd_ctx {
int fdin;
int fdout;
int fderr;
int fdlog;
int fdcwd;
int fddst;
};
struct pe_error_info {
const struct pe_driver_ctx * edctx;
const struct pe_unit_ctx * euctx;
const char * eunit;
int esyscode;
int elibcode;
const char * efunction;
int eline;
unsigned eflags;
void * eany;
};
struct pe_common_ctx {
uint64_t drvflags;
uint64_t actflags;
uint64_t fmtflags;
};
struct pe_driver_ctx {
const char ** units;
const char * program;
const char * module;
const struct pe_common_ctx * cctx;
struct pe_error_info ** errv;
void * any;
};
struct pe_unit_ctx {
const char * const * path;
const struct pe_raw_image * map;
const struct pe_image_meta * meta;
void * any;
};
struct pe_info_string {
char buffer[128];
};
/* driver api */
perk_api int pe_get_driver_ctx (char ** argv, char ** envp, uint32_t flags,
const struct pe_fd_ctx *,
struct pe_driver_ctx **);
perk_api void pe_free_driver_ctx (struct pe_driver_ctx *);
perk_api int pe_get_unit_ctx (const struct pe_driver_ctx *, const char * path,
struct pe_unit_ctx **);
perk_api void pe_free_unit_ctx (struct pe_unit_ctx *);
perk_api int pe_get_driver_fdctx (const struct pe_driver_ctx *, struct pe_fd_ctx *);
perk_api int pe_set_driver_fdctx (struct pe_driver_ctx *, const struct pe_fd_ctx *);
/* utility api */
perk_api int pe_main (char **, char **, const struct pe_fd_ctx *);
perk_api int pe_output_image_category (const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_image_sections (const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_image_symbols (const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_image_strings (const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_export_symbols (const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_import_libraries(const struct pe_driver_ctx *, const struct pe_image_meta *);
perk_api int pe_output_mdso_libraries (const struct pe_driver_ctx *, const struct pe_image_meta *);
/* error trace api */
perk_api int pe_output_error_record (const struct pe_driver_ctx *, const struct pe_error_info *);
perk_api int pe_output_error_vector (const struct pe_driver_ctx *);
/* raw image api */
perk_api int pe_map_raw_image (const struct pe_driver_ctx *,
int fd, const char * path, int prot,
struct pe_raw_image *);
perk_api int pe_unmap_raw_image (struct pe_raw_image *);
/* image meta api */
perk_api int pe_get_image_meta (const struct pe_driver_ctx *,
const struct pe_raw_image *,
struct pe_image_meta **);
perk_api void pe_free_image_meta (struct pe_image_meta *);
perk_api int pe_get_named_section_index(const struct pe_image_meta *, const char *);
perk_api int pe_get_block_section_index(const struct pe_image_meta *, const struct pe_block *);
perk_api int pe_get_roffset_from_rva (const struct pe_image_meta *, uint32_t, uint32_t *);
perk_api int pe_get_rva_from_roffset (const struct pe_image_meta *, uint32_t, uint32_t *);
perk_api int pe_get_expsym_by_name (const struct pe_image_meta *, const char *, struct pe_expsym *);
perk_api int pe_get_expsym_by_index (const struct pe_image_meta *, unsigned, struct pe_expsym *);
/* info api */
perk_api int pe_get_image_abi (const struct pe_image_meta *, struct pe_info_string *);
perk_api int pe_get_image_subtype (const struct pe_image_meta *, struct pe_info_string *);
perk_api int pe_get_image_subsystem (const struct pe_image_meta *, struct pe_info_string *);
perk_api int pe_get_image_framework (const struct pe_image_meta *, struct pe_info_string *);
/* low-level api */
perk_api int pe_read_dos_header (const struct pe_raw_image_dos_hdr *, struct pe_meta_image_dos_hdr *);
perk_api int pe_read_coff_header (const struct pe_raw_coff_image_hdr *, struct pe_meta_coff_file_hdr *);
perk_api int pe_read_object_header (const struct pe_raw_coff_object_hdr *, struct pe_meta_coff_file_hdr *);
perk_api int pe_read_optional_header (const union pe_raw_opt_hdr *, struct pe_meta_opt_hdr *);
perk_api int pe_read_section_header (const struct pe_raw_sec_hdr *, struct pe_meta_sec_hdr *);
perk_api int pe_read_export_header (const struct pe_raw_export_hdr *, struct pe_meta_export_hdr *);
perk_api int pe_read_import_header (const struct pe_raw_import_hdr *, struct pe_meta_import_hdr *);
perk_api int pe_read_coff_symbol (const struct pe_raw_coff_symbol *, struct pe_meta_coff_symbol *,
const struct pe_meta_coff_file_hdr *, void * base);
perk_api int pe_read_import_lookup (const union pe_raw_import_lookup *, struct pe_meta_import_lookup *,
uint32_t magic);
/* package info */
perk_api const struct pe_source_version * pe_source_version(void);
#ifdef __cplusplus
}
#endif
#endif
|