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
|
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <libelf.h>
#include <gelf.h>
#include "elfhandle.h"
#include "options.h"
#include "printing.h"
int main(int argc, char **argv)
{
CLIOpts opts = { 0 };
ELFHandles hIn = { 0 };
ELFHandles hOut = { 0 };
int exitval = EXIT_SUCCESS;
/* Is libelf alive and well? */
if (elf_version(EV_CURRENT) == EV_NONE) {
fprintf(stderr, "libelf init error: %s\n", elf_errmsg(-1));
}
/* Parse and validate user input */
parseOptions(&opts, argc, argv);
/* Open input/output files */
openElf(&hIn, opts.fnInput, ELF_C_READ);
if (!hIn.e) {
exitval = EXIT_FAILURE;
goto EXIT;
}
if (opts.fnOutput) {
openElf(&hOut, opts.fnOutput, ELF_C_WRITE);
if (!hOut.e) {
exitval = EXIT_FAILURE;
goto EXIT;
}
}
/* Now that we have a (hopefully) sane environment, execute commands */
if (opts.printHeader) {
printHeader(hIn.e);
}
if (opts.printSegments) {
printSegments(hIn.e);
}
if (opts.printSections) {
printSections(hIn.e);
}
EXIT:
if (hOut.e) {
if (elf_update(hOut.e, ELF_C_WRITE) < 0) {
fprintf(stderr, "elf_update() failed: %s\n", elf_errmsg(-1));
}
closeElf(&hOut);
}
if (hIn.e) {
closeElf(&hIn);
}
return (exitval);
}
|