/[gentoo-projects]/pax-utils/scanelf.c
Gentoo

Contents of /pax-utils/scanelf.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.45 - (show annotations) (download) (as text)
Sat May 14 00:18:56 2005 UTC (9 years, 2 months ago) by solar
Branch: MAIN
Changes since 1.44: +51 -6 lines
File MIME type: text/x-csrc
- add option to read from streams

1 /*
2 * Copyright 2003 Ned Ludd <solar@gentoo.org>
3 * Copyright 1999-2005 Gentoo Foundation
4 * Distributed under the terms of the GNU General Public License v2
5 * $Header: /var/cvsroot/gentoo-projects/pax-utils/scanelf.c,v 1.44 2005/05/10 22:54:25 vapier Exp $
6 *
7 ********************************************************************
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU General Public License as
10 * published by the Free Software Foundation; either version 2 of the
11 * License, or (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful, but
14 * WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
21 * MA 02111-1307, USA.
22 */
23
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <sys/types.h>
27 #define __USE_GNU
28 #include <string.h>
29 #include <errno.h>
30 #include <unistd.h>
31 #include <sys/stat.h>
32 #include <dirent.h>
33 #include <getopt.h>
34 #include <assert.h>
35
36 #include "paxelf.h"
37
38 static const char *rcsid = "$Id: scanelf.c,v 1.44 2005/05/10 22:54:25 vapier Exp $";
39 #define argv0 "scanelf"
40
41
42
43 /* prototypes */
44 static void scanelf_file(const char *filename);
45 static void scanelf_dir(const char *path);
46 static void scanelf_ldpath();
47 static void scanelf_envpath();
48 static void usage(int status);
49 static void parseargs(int argc, char *argv[]);
50 static char *xstrdup(char *s);
51 static void *xmalloc(size_t size);
52 static void xstrcat(char **dst, const char *src, size_t *curr_len);
53 static inline void xchrcat(char **dst, const char append, size_t *curr_len);
54 static int xemptybuffer(const char *buff);
55
56 /* variables to control behavior */
57 static char scan_ldpath = 0;
58 static char scan_envpath = 0;
59 static char scan_symlink = 1;
60 static char dir_recurse = 0;
61 static char dir_crossmount = 1;
62 static char show_pax = 0;
63 static char show_stack = 0;
64 static char show_textrel = 0;
65 static char show_rpath = 0;
66 static char show_needed = 0;
67 static char show_interp = 0;
68 static char show_banner = 1;
69 static char be_quiet = 0;
70 static char be_verbose = 0;
71 static char *find_sym = NULL, *versioned_symname = NULL;
72 static char *out_format = NULL;
73
74
75
76 /* sub-funcs for scanelf_file() */
77 static char *scanelf_file_pax(elfobj *elf, char *found_pax)
78 {
79 static char *paxflags;
80
81 if (!show_pax) return NULL;
82
83 paxflags = pax_short_hf_flags(PAX_FLAGS(elf));
84 if (!be_quiet || (be_quiet && strncmp(paxflags, "PeMRxS", 6))) {
85 *found_pax = 1;
86 return paxflags;
87 }
88
89 return NULL;
90 }
91 static char *scanelf_file_stack(elfobj *elf, char *found_stack, char *found_relro)
92 {
93 static char ret[8];
94 char *found;
95 unsigned long i, off, shown;
96
97 if (!show_stack) return NULL;
98
99 shown = 0;
100 strcpy(ret, "--- ---");
101
102 if (elf->phdr) {
103 #define SHOW_STACK(B) \
104 if (elf->elf_class == ELFCLASS ## B) { \
105 Elf ## B ## _Ehdr *ehdr = EHDR ## B (elf->ehdr); \
106 Elf ## B ## _Phdr *phdr = PHDR ## B (elf->phdr); \
107 for (i = 0; i < EGET(ehdr->e_phnum); i++) { \
108 if (EGET(phdr[i].p_type) == PT_GNU_STACK) { \
109 found = found_stack; \
110 off = 0; \
111 } else if (EGET(phdr[i].p_type) == PT_GNU_RELRO) { \
112 found = found_relro; \
113 off = 3; \
114 } else \
115 continue; \
116 if (be_quiet && !(EGET(phdr[i].p_flags) & PF_X)) \
117 continue; \
118 memcpy(ret+off, gnu_short_stack_flags(EGET(phdr[i].p_flags)), 3); \
119 *found = 1; \
120 ++shown; \
121 } \
122 }
123 SHOW_STACK(32)
124 SHOW_STACK(64)
125 }
126
127 if (be_quiet && !shown)
128 return NULL;
129 else
130 return ret;
131 }
132 static char *scanelf_file_textrel(elfobj *elf, char *found_textrel)
133 {
134 static char *ret = "TEXTREL";
135 unsigned long i;
136
137 if (!show_textrel) return NULL;
138
139 if (elf->phdr) {
140 #define SHOW_TEXTREL(B) \
141 if (elf->elf_class == ELFCLASS ## B) { \
142 Elf ## B ## _Dyn *dyn; \
143 Elf ## B ## _Ehdr *ehdr = EHDR ## B (elf->ehdr); \
144 Elf ## B ## _Phdr *phdr = PHDR ## B (elf->phdr); \
145 Elf ## B ## _Off offset; \
146 for (i = 0; i < EGET(ehdr->e_phnum); i++) { \
147 if (phdr[i].p_type != PT_DYNAMIC) continue; \
148 offset = EGET(phdr[i].p_offset); \
149 if (offset >= elf->len - sizeof(Elf ## B ## _Dyn)) continue; \
150 dyn = DYN ## B (elf->data + offset); \
151 while (EGET(dyn->d_tag) != DT_NULL) { \
152 if (EGET(dyn->d_tag) == DT_TEXTREL) { /*dyn->d_tag != DT_FLAGS)*/ \
153 *found_textrel = 1; \
154 /*if (dyn->d_un.d_val & DF_TEXTREL)*/ \
155 return ret; \
156 } \
157 ++dyn; \
158 } \
159 } }
160 SHOW_TEXTREL(32)
161 SHOW_TEXTREL(64)
162 }
163
164 if (be_quiet)
165 return NULL;
166 else
167 return " - ";
168 }
169 static void scanelf_file_rpath(elfobj *elf, char *found_rpath, char **ret, size_t *ret_len)
170 {
171 /* TODO: if be_quiet, only output RPATH's which aren't in /etc/ld.so.conf */
172 unsigned long i;
173 char *rpath, *runpath;
174 void *strtbl_void;
175
176 if (!show_rpath) return;
177
178 strtbl_void = elf_findsecbyname(elf, ".dynstr");
179 rpath = runpath = NULL;
180
181 if (elf->phdr && strtbl_void) {
182 #define SHOW_RPATH(B) \
183 if (elf->elf_class == ELFCLASS ## B) { \
184 Elf ## B ## _Dyn *dyn; \
185 Elf ## B ## _Ehdr *ehdr = EHDR ## B (elf->ehdr); \
186 Elf ## B ## _Phdr *phdr = PHDR ## B (elf->phdr); \
187 Elf ## B ## _Shdr *strtbl = SHDR ## B (strtbl_void); \
188 Elf ## B ## _Off offset; \
189 for (i = 0; i < EGET(ehdr->e_phnum); i++) { \
190 if (EGET(phdr[i].p_type) != PT_DYNAMIC) continue; \
191 offset = EGET(phdr[i].p_offset); \
192 if (offset >= elf->len - sizeof(Elf ## B ## _Dyn)) continue; \
193 dyn = DYN ## B (elf->data + offset); \
194 while (EGET(dyn->d_tag) != DT_NULL) { \
195 if (EGET(dyn->d_tag) == DT_RPATH) { \
196 if (rpath) warn("ELF has multiple DT_RPATH's !?"); \
197 offset = EGET(strtbl->sh_offset) + EGET(dyn->d_un.d_ptr); \
198 if (offset >= elf->len) continue; \
199 rpath = (char*)(elf->data + offset); \
200 *found_rpath = 1; \
201 } else if (EGET(dyn->d_tag) == DT_RUNPATH) { \
202 if (runpath) warn("ELF has multiple DT_RUNPATH's !?"); \
203 offset = EGET(strtbl->sh_offset) + EGET(dyn->d_un.d_ptr); \
204 if (offset >= elf->len) continue; \
205 runpath = (char*)(elf->data + offset); \
206 *found_rpath = 1; \
207 } \
208 ++dyn; \
209 } \
210 } }
211 SHOW_RPATH(32)
212 SHOW_RPATH(64)
213 }
214
215 if (rpath && runpath) {
216 if (!strcmp(rpath, runpath)) {
217 xstrcat(ret, runpath, ret_len);
218 } else {
219 fprintf(stderr, "RPATH [%s] != RUNPATH [%s]\n", rpath, runpath);
220 xchrcat(ret, '{', ret_len);
221 xstrcat(ret, rpath, ret_len);
222 xchrcat(ret, ',', ret_len);
223 xstrcat(ret, runpath, ret_len);
224 xchrcat(ret, '}', ret_len);
225 }
226 } else if (rpath || runpath)
227 xstrcat(ret, (runpath ? runpath : rpath), ret_len);
228 else if (!be_quiet)
229 xstrcat(ret, " - ", ret_len);
230 }
231 static void scanelf_file_needed(elfobj *elf, char *found_needed, char **ret, size_t *ret_len)
232 {
233 unsigned long i;
234 char *needed;
235 void *strtbl_void;
236
237 if (!show_needed) return;
238
239 strtbl_void = elf_findsecbyname(elf, ".dynstr");
240
241 if (elf->phdr && strtbl_void) {
242 #define SHOW_NEEDED(B) \
243 if (elf->elf_class == ELFCLASS ## B) { \
244 Elf ## B ## _Dyn *dyn; \
245 Elf ## B ## _Ehdr *ehdr = EHDR ## B (elf->ehdr); \
246 Elf ## B ## _Phdr *phdr = PHDR ## B (elf->phdr); \
247 Elf ## B ## _Shdr *strtbl = SHDR ## B (strtbl_void); \
248 Elf ## B ## _Off offset; \
249 for (i = 0; i < EGET(ehdr->e_phnum); i++) { \
250 if (EGET(phdr[i].p_type) != PT_DYNAMIC) continue; \
251 offset = EGET(phdr[i].p_offset); \
252 if (offset >= elf->len - sizeof(Elf ## B ## _Dyn)) continue; \
253 dyn = DYN ## B (elf->data + offset); \
254 while (EGET(dyn->d_tag) != DT_NULL) { \
255 if (EGET(dyn->d_tag) == DT_NEEDED) { \
256 offset = EGET(strtbl->sh_offset) + EGET(dyn->d_un.d_ptr); \
257 if (offset >= elf->len) continue; \
258 needed = (char*)(elf->data + offset); \
259 if (*found_needed) xchrcat(ret, ',', ret_len); \
260 xstrcat(ret, needed, ret_len); \
261 *found_needed = 1; \
262 } \
263 ++dyn; \
264 } \
265 } }
266 SHOW_NEEDED(32)
267 SHOW_NEEDED(64)
268 }
269 }
270 static char *scanelf_file_interp(elfobj *elf, char *found_interp)
271 {
272 void *strtbl_void;
273
274 if (!show_interp) return NULL;
275
276 strtbl_void = elf_findsecbyname(elf, ".interp");
277
278 if (strtbl_void) {
279 #define SHOW_INTERP(B) \
280 if (elf->elf_class == ELFCLASS ## B) { \
281 Elf ## B ## _Shdr *strtbl = SHDR ## B (strtbl_void); \
282 *found_interp = 1; \
283 return elf->data + EGET(strtbl->sh_offset); \
284 }
285 SHOW_INTERP(32)
286 SHOW_INTERP(64)
287 }
288 return NULL;
289 }
290 static char *scanelf_file_sym(elfobj *elf, char *found_sym, const char *filename)
291 {
292 unsigned long i;
293 void *symtab_void, *strtab_void;
294
295 if (!find_sym) return NULL;
296
297 symtab_void = elf_findsecbyname(elf, ".symtab");
298 strtab_void = elf_findsecbyname(elf, ".strtab");
299
300 if (symtab_void && strtab_void) {
301 #define FIND_SYM(B) \
302 if (elf->elf_class == ELFCLASS ## B) { \
303 Elf ## B ## _Shdr *symtab = SHDR ## B (symtab_void); \
304 Elf ## B ## _Shdr *strtab = SHDR ## B (strtab_void); \
305 Elf ## B ## _Sym *sym = SYM ## B (elf->data + EGET(symtab->sh_offset)); \
306 unsigned long cnt = EGET(symtab->sh_size) / EGET(symtab->sh_entsize); \
307 char *symname; \
308 for (i = 0; i < cnt; ++i) { \
309 if (sym->st_name) { \
310 symname = (char *)(elf->data + EGET(strtab->sh_offset) + EGET(sym->st_name)); \
311 if (*find_sym == '*') { \
312 printf("%s(%s) %5lX %15s %s\n", \
313 ((*found_sym == 0) ? "\n\t" : "\t"), \
314 (char *)basename(filename), \
315 (long)sym->st_size, \
316 (char *)get_elfstttype(sym->st_info), \
317 symname); \
318 *found_sym = 1; \
319 } else if ((strcmp(find_sym, symname) == 0) || \
320 (strcmp(symname, versioned_symname) == 0)) \
321 (*found_sym)++; \
322 } \
323 ++sym; \
324 } }
325 FIND_SYM(32)
326 FIND_SYM(64)
327 }
328 if (*find_sym != '*' && *found_sym)
329 return find_sym;
330 if (be_quiet)
331 return NULL;
332 else
333 return " - ";
334 }
335 /* scan an elf file and show all the fun stuff */
336 #define prints(str) fputs(str, stdout)
337 static void scanelf_file(const char *filename)
338 {
339 unsigned long i;
340 char found_pax, found_stack, found_relro, found_textrel,
341 found_rpath, found_needed, found_interp, found_sym,
342 found_file;
343 elfobj *elf;
344 struct stat st;
345 static char *out_buffer = NULL;
346 static size_t out_len;
347
348 /* make sure 'filename' exists */
349 if (lstat(filename, &st) == -1) {
350 if (be_verbose > 2) printf("%s: does not exist\n", filename);
351 return;
352 }
353 /* always handle regular files and handle symlinked files if no -y */
354 if (!(S_ISREG(st.st_mode) || (S_ISLNK(st.st_mode) && scan_symlink))) {
355 if (be_verbose > 2) printf("%s: skipping non-file\n", filename);
356 return;
357 }
358
359 found_pax = found_stack = found_relro = found_textrel = \
360 found_rpath = found_needed = found_interp = found_sym = \
361 found_file = 0;
362
363 /* verify this is real ELF */
364 if ((elf = readelf(filename)) == NULL) {
365 if (be_verbose > 2) printf("%s: not an ELF\n", filename);
366 return;
367 }
368
369 if (be_verbose > 1)
370 printf("%s: scanning file {%s,%s}\n", filename,
371 get_elfeitype(elf, EI_CLASS, elf->elf_class),
372 get_elfeitype(elf, EI_DATA, elf->data[EI_DATA]));
373 else if (be_verbose)
374 printf("%s: scanning file\n", filename);
375
376 /* init output buffer */
377 if (!out_buffer) {
378 out_len = sizeof(char) * 80;
379 out_buffer = (char*)xmalloc(out_len);
380 }
381 *out_buffer = '\0';
382
383 /* show the header */
384 if (!be_quiet && show_banner) {
385 for (i=0; out_format[i]; ++i) {
386 if (out_format[i] != '%') continue;
387
388 switch (out_format[++i]) {
389 case '%': break;
390 case 'F': prints("FILE "); break;
391 case 'o': prints(" TYPE "); break;
392 case 'x': prints(" PAX "); break;
393 case 'e': prints("STK/REL "); break;
394 case 't': prints("TEXTREL "); break;
395 case 'r': prints("RPATH "); break;
396 case 'n': prints("NEEDED "); break;
397 case 'i': prints("INTERP "); break;
398 case 's': prints("SYM "); break;
399 }
400 }
401 prints("\n");
402 show_banner = 0;
403 }
404
405 /* dump all the good stuff */
406 for (i=0; out_format[i]; ++i) {
407 const char *out;
408
409 /* make sure we trim leading spaces in quiet mode */
410 if (be_quiet && *out_buffer == ' ' && !out_buffer[1])
411 *out_buffer = '\0';
412
413 if (out_format[i] != '%') {
414 xchrcat(&out_buffer, out_format[i], &out_len);
415 continue;
416 }
417
418 out = NULL;
419 switch (out_format[++i]) {
420 case '%': xchrcat(&out_buffer, '%', &out_len); break;
421 case 'F': found_file = 1; xstrcat(&out_buffer, filename, &out_len); break;
422 case 'o': out = get_elfetype(elf); break;
423 case 'x': out = scanelf_file_pax(elf, &found_pax); break;
424 case 'e': out = scanelf_file_stack(elf, &found_stack, &found_relro); break;
425 case 't': out = scanelf_file_textrel(elf, &found_textrel); break;
426 case 'r': scanelf_file_rpath(elf, &found_rpath, &out_buffer, &out_len); break;
427 case 'n': scanelf_file_needed(elf, &found_needed, &out_buffer, &out_len); break;
428 case 'i': out = scanelf_file_interp(elf, &found_interp); break;
429 case 's': out = scanelf_file_sym(elf, &found_sym, filename); break;
430 }
431 if (out) xstrcat(&out_buffer, out, &out_len);
432 }
433
434 if (!found_file) {
435 if (!be_quiet || found_pax || found_stack || found_textrel || \
436 found_rpath || found_needed || found_interp || found_sym)
437 xstrcat(&out_buffer, filename, &out_len);
438 }
439 if (!(be_quiet && xemptybuffer(out_buffer)))
440 puts(out_buffer);
441
442 unreadelf(elf);
443 }
444
445 /* scan a directory for ET_EXEC files and print when we find one */
446 static void scanelf_dir(const char *path)
447 {
448 register DIR *dir;
449 register struct dirent *dentry;
450 struct stat st_top, st;
451 char buf[_POSIX_PATH_MAX];
452 size_t pathlen = 0, len = 0;
453
454 /* make sure path exists */
455 if (lstat(path, &st_top) == -1) {
456 if (be_verbose > 2) printf("%s: does not exist\n", path);
457 return;
458 }
459
460 /* ok, if it isn't a directory, assume we can open it */
461 if (!S_ISDIR(st_top.st_mode)) {
462 scanelf_file(path);
463 return;
464 }
465
466 /* now scan the dir looking for fun stuff */
467 if ((dir = opendir(path)) == NULL) {
468 warnf("could not opendir %s: %s", path, strerror(errno));
469 return;
470 }
471 if (be_verbose) printf("%s: scanning dir\n", path);
472
473 pathlen = strlen(path);
474 while ((dentry = readdir(dir))) {
475 if (!strcmp(dentry->d_name, ".") || !strcmp(dentry->d_name, ".."))
476 continue;
477 len = (pathlen + 1 + strlen(dentry->d_name) + 1);
478 if (len >= sizeof(buf)) {
479 warnf("Skipping '%s': len > sizeof(buf); %d > %d\n", path, (int)len, (int)sizeof(buf));
480 continue;
481 }
482 sprintf(buf, "%s/%s", path, dentry->d_name);
483 if (lstat(buf, &st) != -1) {
484 if (S_ISREG(st.st_mode))
485 scanelf_file(buf);
486 else if (dir_recurse && S_ISDIR(st.st_mode)) {
487 if (dir_crossmount || (st_top.st_dev == st.st_dev))
488 scanelf_dir(buf);
489 }
490 }
491 }
492 closedir(dir);
493 }
494
495 int scanelf_from_file(char *filename) {
496 FILE *fp = NULL;
497 char *p;
498 char path[_POSIX_PATH_MAX];
499
500 if (((strcmp(filename, "-")) == 0) && (ttyname(0) == NULL))
501 fp = stdin;
502 else if ((fp = fopen(filename, "r")) == NULL)
503 return 1;
504
505 while ((fgets(path, _POSIX_PATH_MAX, fp)) != NULL) {
506 if ((p = strchr(path, '\n')) != NULL)
507 *p = 0;
508 scanelf_dir(path);
509 }
510 if (fp != stdin)
511 fclose(fp);
512 return 0;
513 }
514
515 /* scan /etc/ld.so.conf for paths */
516 static void scanelf_ldpath()
517 {
518 char scan_l, scan_ul, scan_ull;
519 char *path, *p;
520 FILE *fp;
521
522 if ((fp = fopen("/etc/ld.so.conf", "r")) == NULL)
523 err("Unable to open ld.so.conf: %s", strerror(errno));
524
525 scan_l = scan_ul = scan_ull = 0;
526
527 path = (char*)xmalloc(_POSIX_PATH_MAX);
528 while ((fgets(path, _POSIX_PATH_MAX, fp)) != NULL)
529 if (*path == '/') {
530 if ((p = strrchr(path, '\r')) != NULL)
531 *p = 0;
532 if ((p = strrchr(path, '\n')) != NULL)
533 *p = 0;
534 if (!scan_l && !strcmp(path, "/lib")) scan_l = 1;
535 if (!scan_ul && !strcmp(path, "/usr/lib")) scan_ul = 1;
536 if (!scan_ull && !strcmp(path, "/usr/local/lib")) scan_ull = 1;
537 scanelf_dir(path);
538 }
539 free(path);
540 fclose(fp);
541
542 if (!scan_l) scanelf_dir("/lib");
543 if (!scan_ul) scanelf_dir("/usr/lib");
544 if (!scan_ull) scanelf_dir("/usr/local/lib");
545 }
546
547 /* scan env PATH for paths */
548 static void scanelf_envpath()
549 {
550 char *path, *p;
551
552 path = getenv("PATH");
553 if (!path)
554 err("PATH is not set in your env !");
555 path = xstrdup(path);
556
557 while ((p = strrchr(path, ':')) != NULL) {
558 scanelf_dir(p + 1);
559 *p = 0;
560 }
561
562 free(path);
563 }
564
565
566
567 /* usage / invocation handling functions */
568 #define PARSE_FLAGS "plRmyxetrnis:aqvF:f:o:BhV"
569 #define a_argument required_argument
570 static struct option const long_opts[] = {
571 {"path", no_argument, NULL, 'p'},
572 {"ldpath", no_argument, NULL, 'l'},
573 {"recursive", no_argument, NULL, 'R'},
574 {"mount", no_argument, NULL, 'm'},
575 {"symlink", no_argument, NULL, 'y'},
576 {"pax", no_argument, NULL, 'x'},
577 {"header", no_argument, NULL, 'e'},
578 {"textrel", no_argument, NULL, 't'},
579 {"rpath", no_argument, NULL, 'r'},
580 {"needed", no_argument, NULL, 'n'},
581 {"interp", no_argument, NULL, 'i'},
582 {"symbol", a_argument, NULL, 's'},
583 {"all", no_argument, NULL, 'a'},
584 {"quiet", no_argument, NULL, 'q'},
585 {"verbose", no_argument, NULL, 'v'},
586 {"format", a_argument, NULL, 'F'},
587 {"from", a_argument, NULL, 'f'},
588 {"file", a_argument, NULL, 'o'},
589 {"nobanner", no_argument, NULL, 'B'},
590 {"help", no_argument, NULL, 'h'},
591 {"version", no_argument, NULL, 'V'},
592 {NULL, no_argument, NULL, 0x0}
593 };
594 static char *opts_help[] = {
595 "Scan all directories in PATH environment",
596 "Scan all directories in /etc/ld.so.conf",
597 "Scan directories recursively",
598 "Don't recursively cross mount points",
599 "Don't scan symlinks\n",
600 "Print PaX markings",
601 "Print GNU_STACK markings",
602 "Print TEXTREL information",
603 "Print RPATH information",
604 "Print NEEDED information",
605 "Print INTERP information",
606 "Find a specified symbol",
607 "Print all scanned info (-x -e -t -r)\n",
608 "Only output 'bad' things",
609 "Be verbose (can be specified more than once)",
610 "Use specified format for output",
611 "Read input stream from a filename",
612 "Write output stream to a filename",
613 "Don't display the header",
614 "Print this help and exit",
615 "Print version and exit",
616 NULL
617 };
618
619 /* display usage and exit */
620 static void usage(int status)
621 {
622 unsigned long i;
623 printf(" Scan ELF binaries for stuff\n\n"
624 "Usage: %s [options] <dir1/file1> [dir2 dirN fileN ...]\n\n", argv0);
625 printf("Options: -[%s]\n", PARSE_FLAGS);
626 for (i = 0; long_opts[i].name; ++i)
627 if (long_opts[i].has_arg == no_argument)
628 printf(" -%c, --%-13s %s\n", long_opts[i].val,
629 long_opts[i].name, opts_help[i]);
630 else
631 printf(" -%c, --%-6s <arg> %s\n", long_opts[i].val,
632 long_opts[i].name, opts_help[i]);
633
634 if (status != EXIT_SUCCESS)
635 exit(status);
636
637 puts("\nThe format modifiers for the -F option are");
638 puts(" %F Filename");
639 puts(" %x PaX Flags");
640 puts(" %e STACK/RELRO");
641 puts(" %t TEXTREL");
642 puts(" %r RPATH");
643 puts(" %n NEEDED");
644 puts(" %i INTERP");
645 puts(" %s symbol");
646
647 exit(status);
648 }
649
650 /* parse command line arguments and preform needed actions */
651 static void parseargs(int argc, char *argv[])
652 {
653 int flag;
654 char *from_file = NULL;
655
656 opterr = 0;
657 while ((flag=getopt_long(argc, argv, PARSE_FLAGS, long_opts, NULL)) != -1) {
658 switch (flag) {
659
660 case 'V':
661 printf("%s compiled %s\n%s\n"
662 "%s written for Gentoo Linux by <solar and vapier @ gentoo.org>\n",
663 __FILE__, __DATE__, rcsid, argv0);
664 exit(EXIT_SUCCESS);
665 break;
666 case 'h': usage(EXIT_SUCCESS); break;
667 case 'f':
668 if (from_file == NULL)
669 from_file = xstrdup(optarg);
670 break;
671 case 'o': {
672 FILE *fp = NULL;
673 fp = freopen(optarg, "w", stdout);
674 if (fp == NULL)
675 err("Could not open output stream '%s': %s", optarg, strerror(errno));
676 stdout = fp;
677 break;
678 }
679
680 case 's': {
681 size_t len;
682 find_sym = xstrdup(optarg);
683 len = strlen(find_sym) + 1;
684 versioned_symname = (char*)xmalloc(sizeof(char) * (len+1));
685 sprintf(versioned_symname, "%s@", find_sym);
686 break;
687 }
688
689 case 'F': {
690 out_format = xstrdup(optarg);
691 break;
692 }
693
694 case 'y': scan_symlink = 0; break;
695 case 'B': show_banner = 0; break;
696 case 'l': scan_ldpath = 1; break;
697 case 'p': scan_envpath = 1; break;
698 case 'R': dir_recurse = 1; break;
699 case 'm': dir_crossmount = 0; break;
700 case 'x': show_pax = 1; break;
701 case 'e': show_stack = 1; break;
702 case 't': show_textrel = 1; break;
703 case 'r': show_rpath = 1; break;
704 case 'n': show_needed = 1; break;
705 case 'i': show_interp = 1; break;
706 case 'q': be_quiet = 1; break;
707 case 'v': be_verbose = (be_verbose % 20) + 1; break;
708 case 'a': show_pax = show_stack = show_textrel = show_rpath = show_needed = show_interp = 1; break;
709
710 case ':':
711 warn("Option missing parameter\n");
712 usage(EXIT_FAILURE);
713 break;
714 case '?':
715 warn("Unknown option\n");
716 usage(EXIT_FAILURE);
717 break;
718 default:
719 err("Unhandled option '%c'", flag);
720 break;
721 }
722 }
723
724 if (be_quiet && be_verbose)
725 err("You can be quiet or you can be verbose, not both, stupid");
726
727 /* let the format option override all other options */
728 if (out_format) {
729 show_pax = show_stack = show_textrel = show_rpath = show_needed = show_interp = 0;
730 for (flag=0; out_format[flag]; ++flag) {
731 if (out_format[flag] != '%') continue;
732
733 switch (out_format[++flag]) {
734 case '%': break;
735 case 'F': break;
736 case 's': break;
737 case 'o': break;
738 case 'x': show_pax = 1; break;
739 case 'e': show_stack = 1; break;
740 case 't': show_textrel = 1; break;
741 case 'r': show_rpath = 1; break;
742 case 'n': show_needed = 1; break;
743 case 'i': show_interp = 1; break;
744 default:
745 err("Invalid format specifier '%c' (byte %i)",
746 out_format[flag], flag+1);
747 }
748 }
749
750 /* construct our default format */
751 } else {
752 size_t fmt_len = 30;
753 out_format = (char*)xmalloc(sizeof(char) * fmt_len);
754 if (!be_quiet) xstrcat(&out_format, "%o ", &fmt_len);
755 if (show_pax) xstrcat(&out_format, "%x ", &fmt_len);
756 if (show_stack) xstrcat(&out_format, "%e ", &fmt_len);
757 if (show_textrel) xstrcat(&out_format, "%t ", &fmt_len);
758 if (show_rpath) xstrcat(&out_format, "%r ", &fmt_len);
759 if (show_needed) xstrcat(&out_format, "%n ", &fmt_len);
760 if (show_interp) xstrcat(&out_format, "%i ", &fmt_len);
761 if (find_sym) xstrcat(&out_format, "%s ", &fmt_len);
762 if (!be_quiet) xstrcat(&out_format, "%F ", &fmt_len);
763 }
764 if (be_verbose > 2) printf("Format: %s\n", out_format);
765
766 /* now lets actually do the scanning */
767 if (scan_ldpath) scanelf_ldpath();
768 if (scan_envpath) scanelf_envpath();
769 if (from_file) {
770 scanelf_from_file(from_file);
771 free(from_file);
772 from_file = *argv;
773 }
774 if (optind == argc && !scan_ldpath && !scan_envpath && !from_file)
775 err("Nothing to scan !?");
776 while (optind < argc)
777 scanelf_dir(argv[optind++]);
778
779 /* clean up */
780 if (find_sym) {
781 free(find_sym);
782 free(versioned_symname);
783 }
784 if (out_format) free(out_format);
785 }
786
787
788
789 /* utility funcs */
790 static char *xstrdup(char *s)
791 {
792 char *ret = strdup(s);
793 if (!ret) err("Could not strdup(): %s", strerror(errno));
794 return ret;
795 }
796 static void *xmalloc(size_t size)
797 {
798 void *ret = malloc(size);
799 if (!ret) err("Could not malloc() %li bytes", (unsigned long)size);
800 return ret;
801 }
802 static void xstrcat(char **dst, const char *src, size_t *curr_len)
803 {
804 long new_len;
805
806 new_len = strlen(*dst) + strlen(src);
807 if (*curr_len <= new_len) {
808 *curr_len = new_len + (*curr_len / 2);
809 *dst = realloc(*dst, *curr_len);
810 if (!*dst)
811 err("could not realloc %li bytes", (unsigned long)*curr_len);
812 }
813
814 strcat(*dst, src);
815 }
816 static inline void xchrcat(char **dst, const char append, size_t *curr_len)
817 {
818 static char my_app[2];
819 my_app[0] = append;
820 my_app[1] = '\0';
821 xstrcat(dst, my_app, curr_len);
822 }
823 static int xemptybuffer(const char *buff)
824 {
825 long i;
826 for (i=0; buff[i]; ++i)
827 if (buff[i] != ' ')
828 return 0;
829 return 1;
830 }
831
832
833
834 int main(int argc, char *argv[])
835 {
836 if (argc < 2)
837 usage(EXIT_FAILURE);
838 parseargs(argc, argv);
839 fclose(stdout);
840 return EXIT_SUCCESS;
841 }

  ViewVC Help
Powered by ViewVC 1.1.20