223
|
1 /* |
|
2 * main.c -- binary to C/C++ arrays converter |
|
3 * |
|
4 * Copyright (c) 2020 David Demelier <markand@malikania.fr> |
|
5 * |
|
6 * Permission to use, copy, modify, and/or distribute this software for any |
|
7 * purpose with or without fee is hereby granted, provided that the above |
|
8 * copyright notice and this permission notice appear in all copies. |
|
9 * |
|
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
|
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
|
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
|
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
|
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
|
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
|
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
|
17 */ |
|
18 |
|
19 #include <errno.h> |
|
20 #include <stdarg.h> |
|
21 #include <stdbool.h> |
|
22 #include <stdio.h> |
|
23 #include <stdlib.h> |
|
24 #include <stdnoreturn.h> |
|
25 #include <string.h> |
|
26 #include <unistd.h> |
|
27 |
|
28 static const char *charset = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_"; |
|
29 static char findentchar = '\t'; |
|
30 static int findent = 1; |
|
31 static bool fconst; |
|
32 static bool fnull; |
|
33 static bool fstatic; |
|
34 static bool funsigned; |
|
35 |
|
36 noreturn static void |
|
37 usage(void) |
|
38 { |
|
39 fprintf(stderr, "usage: bcc [-0csu] [-I tab-indent] [-i space-indent] input variable\n"); |
|
40 exit(1); |
|
41 } |
|
42 |
|
43 noreturn static void |
|
44 die(const char *fmt, ...) |
|
45 { |
|
46 va_list ap; |
|
47 |
|
48 va_start(ap, fmt); |
|
49 fputs("abort: ", stderr); |
|
50 vfprintf(stderr, fmt, ap); |
|
51 va_end(ap); |
|
52 exit(1); |
|
53 } |
|
54 |
|
55 static char * |
|
56 mangle(char *variable) |
|
57 { |
|
58 char *p; |
|
59 size_t pos; |
|
60 |
|
61 /* Remove extension. */ |
|
62 if ((p = strrchr(variable, '.'))) |
|
63 *p = '\0'; |
|
64 |
|
65 /* Remove disallowed characters. */ |
|
66 while ((pos = strspn(variable, charset)) != strlen(variable)) |
|
67 variable[pos] = '_'; |
|
68 |
|
69 return variable; |
|
70 } |
|
71 |
|
72 static void |
|
73 indent(void) |
|
74 { |
|
75 for (int i = 0; i < findent; ++i) |
|
76 putchar(findentchar); |
|
77 } |
|
78 |
|
79 static void |
|
80 put(int ch) |
|
81 { |
|
82 if (funsigned) |
|
83 printf("0x%02hhx", (unsigned char)ch); |
|
84 else |
|
85 printf("%hhd", (signed char)ch); |
|
86 } |
|
87 |
|
88 static void |
|
89 process(const char *input, const char *variable) |
|
90 { |
|
91 FILE *fp; |
|
92 int ch, col = 0; |
|
93 |
|
94 if (strcmp(input, "-") == 0) |
|
95 fp = stdin; |
|
96 else if (!(fp = fopen(input, "rb"))) |
|
97 die("%s: %s\n", input, strerror(errno)); |
|
98 |
|
99 if (fstatic) |
|
100 printf("static "); |
|
101 if (fconst) |
|
102 printf("const "); |
|
103 |
|
104 printf(funsigned ? "unsigned " : "signed "); |
|
105 printf("char %s[] = {\n", variable); |
|
106 |
|
107 for (ch = fgetc(fp); ch != EOF; ) { |
|
108 if (col == 0) |
|
109 indent(); |
|
110 |
|
111 put(ch); |
|
112 |
|
113 if ((ch = fgetc(fp)) != EOF || fnull) |
|
114 printf(",%s", col < 3 ? " " : ""); |
|
115 |
|
116 if (++col == 4) { |
|
117 col = 0; |
|
118 putchar('\n'); |
|
119 } |
|
120 |
|
121 /* Add final '\0' if required. */ |
|
122 if (ch == EOF && fnull) { |
|
123 if (col++ == 0) |
|
124 indent(); |
|
125 |
|
126 put(0); |
|
127 } |
|
128 } |
|
129 |
|
130 if (col != 0) |
|
131 printf("\n"); |
|
132 |
|
133 puts("};"); |
|
134 fclose(fp); |
|
135 } |
|
136 |
|
137 int |
|
138 main(int argc, char **argv) |
|
139 { |
|
140 int ch; |
|
141 |
|
142 while ((ch = getopt(argc, argv, "0cI:i:su")) != -1) { |
|
143 switch (ch) { |
|
144 case '0': |
|
145 fnull = true; |
|
146 break; |
|
147 case 'c': |
|
148 fconst = true; |
|
149 break; |
|
150 case 'I': |
|
151 findentchar = '\t'; |
|
152 findent = atoi(optarg); |
|
153 break; |
|
154 case 'i': |
|
155 findentchar = ' '; |
|
156 findent = atoi(optarg); |
|
157 break; |
|
158 case 's': |
|
159 fstatic = true; |
|
160 break; |
|
161 case 'u': |
|
162 funsigned = true; |
|
163 break; |
|
164 default: |
|
165 break; |
|
166 } |
|
167 } |
|
168 |
|
169 argc -= optind; |
|
170 argv += optind; |
|
171 |
|
172 if (argc < 2) |
|
173 usage(); |
|
174 |
|
175 process(argv[0], mangle(argv[1])); |
|
176 } |