Add usage and avoid execution when there is no argument
[elfdbg/.git] / elfdbg.c
1 /*-
2  * Copyright (c) 2015 Sofian Brabez <sbz@FreeBSD.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer
10  *    in this position and unchanged.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 #include <err.h>
27 #include <fcntl.h>
28 #include <gelf.h>
29 #include <stdint.h>
30 #include <string.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <sysexits.h>
34 #include <unistd.h>
35
36 extern char *__progname;
37
38 static int
39 elf_debug_sections(Elf *e)
40 {
41         Elf_Scn *scn = NULL;
42         GElf_Shdr shdr;
43         size_t n, shstrndx, sz;
44         char *name;
45         int has_debug = 0;
46
47         if (elf_getshdrstrndx(e, &shstrndx) != 0)
48                 errx(EX_SOFTWARE, "elf_getshdrstrndx() failed : %s . ",
49                     elf_errmsg(-1));
50
51         while ((scn = elf_nextscn(e, scn)) != NULL) {
52                 gelf_getshdr(scn, &shdr);
53
54                 name = elf_strptr(e, shstrndx, shdr.sh_name);
55                 if (!strstr(name, "debug_"))
56                         continue;
57
58                 has_debug++;
59         }
60
61         return (has_debug > 0);
62 }
63
64 static void
65 usage(void) {
66         fprintf(stderr, "Usage: %s file\n", __progname);
67         exit(EX_USAGE);
68 }
69
70 int
71 main(int argc, char *argv[])
72 {
73         const char *filename;
74         int fd, rc;
75         Elf *e;
76         int has_debug;
77
78         if (argc == 1)
79                 usage();
80
81         filename = argv[1];
82
83         if (elf_version(EV_CURRENT) == EV_NONE)
84                 errx(EX_SOFTWARE, "ELF library initialization failed : %s ",
85                     elf_errmsg(-1));
86
87         if ((fd = open(filename, O_RDONLY, 0)) < 0)
88                 err(EX_NOINPUT, "open %s failed ", filename);
89         if ((e = elf_begin(fd, ELF_C_READ, NULL)) == NULL)
90                 errx(EX_SOFTWARE, "elf_begin() failed : %s", elf_errmsg(-1));
91         if (elf_kind(e) != ELF_K_ELF)
92                 errx(EX_DATAERR, "%s is not an ELF object", filename);
93
94         has_debug = elf_debug_sections(e);
95         printf(has_debug ? "HAS DEBUG\n" : "NO DEBUG\n");
96
97         rc = close(fd);
98         rc = elf_end(e);
99
100         return (rc);
101 }