1
 2#include <stdio.h>
 3#include <stdint.h>
 4#include <stdarg.h>
 5#include <stdlib.h>
 6#include <string.h>
 7#include <errno.h>
 8#include <endian.h>
 9#include <elf.h>
10
11#include "relocs.h"
12
13void die(char *fmt, ...)
14{
15	va_list ap;
16
17	va_start(ap, fmt);
18	vfprintf(stderr, fmt, ap);
19	va_end(ap);
20	exit(1);
21}
22
23static void usage(void)
24{
25	die("relocs [--reloc-info|--text|--bin|--keep] vmlinux\n");
26}
27
28int main(int argc, char **argv)
29{
30	int show_reloc_info, as_text, as_bin, keep_relocs;
31	const char *fname;
32	FILE *fp;
33	int i;
34	unsigned char e_ident[EI_NIDENT];
35
36	show_reloc_info = 0;
37	as_text = 0;
38	as_bin = 0;
39	keep_relocs = 0;
40	fname = NULL;
41	for (i = 1; i < argc; i++) {
42		char *arg = argv[i];
43
44		if (*arg == '-') {
45			if (strcmp(arg, "--reloc-info") == 0) {
46				show_reloc_info = 1;
47				continue;
48			}
49			if (strcmp(arg, "--text") == 0) {
50				as_text = 1;
51				continue;
52			}
53			if (strcmp(arg, "--bin") == 0) {
54				as_bin = 1;
55				continue;
56			}
57			if (strcmp(arg, "--keep") == 0) {
58				keep_relocs = 1;
59				continue;
60			}
61		} else if (!fname) {
62			fname = arg;
63			continue;
64		}
65		usage();
66	}
67	if (!fname)
68		usage();
69
70	fp = fopen(fname, "r+");
71	if (!fp)
72		die("Cannot open %s: %s\n", fname, strerror(errno));
73
74	if (fread(&e_ident, 1, EI_NIDENT, fp) != EI_NIDENT)
75		die("Cannot read %s: %s", fname, strerror(errno));
76
77	rewind(fp);
78	if (e_ident[EI_CLASS] == ELFCLASS64)
79		process_64(fp, as_text,  as_bin, show_reloc_info, keep_relocs);
80	else
81		process_32(fp, as_text, as_bin, show_reloc_info, keep_relocs);
82	fclose(fp);
83	return 0;
84}