Home
strip.c - scc - simple c99 compiler
HTML git clone git://git.simple-cc.org/scc
DIR Log
DIR Files
DIR Refs
DIR Submodules
DIR README
DIR LICENSE
---
strip.c (1952B)
---
1 #include <errno.h>
2 #include <stdarg.h>
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6
7 #include <scc/arg.h>
8 #include <scc/mach.h>
9
10 static int status;
11 static char tmpname[FILENAME_MAX];
12 static char *filename;
13 char *argv0;
14
15 static void
16 error(char *fmt, ...)
17 {
18 va_list va;
19
20 va_start(va, fmt);
21 fprintf(stderr, "strip: %s: ", filename);
22 vfprintf(stderr, fmt, va);
23 putc('\n', stderr);
24 va_end(va);
25
26 status = EXIT_FAILURE;
27 }
28
29 static void
30 doit(char *fname)
31 {
32 int type;
33 size_t r;
34 FILE *src, *dst;
35 Map *map;
36 Obj *obj;
37
38 errno = 0;
39 filename = fname;
40 r = snprintf(tmpname, sizeof(tmpname), "%s.tmp", fname);
41 if (r >= sizeof(tmpname)) {
42 error("%s: name too long", fname);
43 return;
44 }
45 if ((src = fopen(fname, "rb")) == NULL) {
46 error("opening src file");
47 return;
48 }
49 if ((type = objprobe(src, NULL)) < 0) {
50 error("getting object type");
51 goto err1;
52 }
53 if ((obj = newobj(type)) == NULL) {
54 error("creating object");
55 goto err1;
56 }
57 if (readobj(obj, src) < 0) {
58 error("reading object");
59 goto err2;
60 }
61 if ((map = loadmap(obj, src)) == NULL) {
62 error("loading map");
63 goto err3;
64 }
65 if (strip(obj) < 0) {
66 error("stripping");
67 goto err3;
68 }
69 if ((dst = fopen(tmpname, "wb")) == NULL) {
70 error("opening dst");
71 goto err3;
72 }
73 if (writeobj(obj, map, dst) < 0) {
74 error("writing object");
75 goto err5;
76 }
77 if (fclose(dst) == EOF) {
78 error("closing dst");
79 goto err4;
80 }
81 if (remove(fname) == EOF) {
82 error("removing fname");
83 goto err4;
84 }
85 if (rename(tmpname, fname) == EOF) {
86 error("rename fname");
87 goto err4;
88 }
89
90 goto err3;
91
92 err5:
93 fclose(dst);
94 err4:
95 remove(tmpname);
96 err3:
97 free(map);
98 err2:
99 delobj(obj);
100 err1:
101 fclose(src);
102 }
103
104 static void
105 usage(void)
106 {
107 fputs("usage: strip [file...]\n", stderr);
108 exit(EXIT_FAILURE);
109 }
110
111 int
112 main(int argc, char *argv[])
113 {
114 ARGBEGIN {
115 default:
116 usage();
117 } ARGEND
118
119 if (argc == 0) {
120 doit("a.out");
121 } else {
122 for (; *argv; ++argv)
123 doit(*argv);
124 }
125
126 return status;
127 }