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
|
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <libelf/libelf.h>
#include <libelf/gelf.h>
#include <libelfu/libelfu.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;
ElfuElf *me;
/* 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;
}
/* Now that we have a (hopefully) sane environment, execute commands. */
me = elfu_mFromElf(hIn.e);
if (me) {
closeElf(&hIn);
printf("Model successfully loaded.\n");
elfu_mDumpElf(me);
elfu_mCheck(me);
printf("Input model checked.\n");
} else {
printf("Failed to load model, aborting.\n");
goto EXIT;
}
/* Copy the input ELF to the output file if the latter is specified.
* Perform requested transformations on the memory model on-the-fly. */
if (!opts.fnOutput) {
printf("No output file specified - no further operations performed.\n");
} else {
if (opts.expandNobitsOffs) {
elfu_mExpandNobits(me, opts.expandNobitsOffs);
}
if (opts.insertBeforeSz) {
elfu_mInsertSpaceBefore(me, opts.insertBeforeOffs, opts.insertBeforeSz);
}
if (opts.insertAfterSz) {
elfu_mInsertSpaceAfter(me, opts.insertAfterOffs, opts.insertAfterSz);
}
if (opts.fnReladd) {
ELFHandles hRel = { 0 };
ElfuElf *mrel = NULL;
openElf(&hRel, opts.fnReladd, ELF_C_READ);
if (!hRel.e) {
printf("--reladd: Failed to open file for --reladd, skipping operation.\n");
} else {
mrel = elfu_mFromElf(hRel.e);
closeElf(&hRel);
if (!me) {
printf("--reladd: Failed to load model for --reladd, skipping operation.\n");
} else {
printf("--reladd: Model successfully loaded.\n");
elfu_mCheck(mrel);
printf("--reladd: Input model checked.\n");
elfu_mReladd(me, mrel);
}
}
}
elfu_mCheck(me);
printf("Output model checked.\n");
openElf(&hOut, opts.fnOutput, ELF_C_WRITE);
if (!hOut.e) {
printf("Failed to open output file. Aborting.\n");
exitval = EXIT_FAILURE;
goto EXIT;
}
elfu_mToElf(me, hOut.e);
printf("Model converted to ELF, ready to be written.\n");
}
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);
}
|
