1 /*
2 * *****************************************************************************
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 *
6 * Copyright (c) 2018-2021 Gavin D. Howard and contributors.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are met:
10 *
11 * * Redistributions of source code must retain the above copyright notice, this
12 * list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above copyright notice,
15 * this list of conditions and the following disclaimer in the documentation
16 * and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 *
30 * *****************************************************************************
31 *
32 * Generates a const array from a bc script.
33 *
34 */
35
36 #include <stdbool.h>
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <string.h>
40
41 #include <errno.h>
42
43 #ifndef _WIN32
44 #include <libgen.h>
45 #endif // _WIN32
46
47 static const char* const bc_gen_header =
48 "// Copyright (c) 2018-2021 Gavin D. Howard and contributors.\n"
49 "// Licensed under the 2-clause BSD license.\n"
50 "// *** AUTOMATICALLY GENERATED FROM %s. DO NOT MODIFY. ***\n\n";
51
52 static const char* const bc_gen_label = "const char *%s = \"%s\";\n\n";
53 static const char* const bc_gen_label_extern = "extern const char *%s;\n\n";
54 static const char* const bc_gen_ifdef = "#if %s\n";
55 static const char* const bc_gen_endif = "#endif // %s\n";
56 static const char* const bc_gen_name = "const char %s[] = {\n";
57 static const char* const bc_gen_name_extern = "extern const char %s[];\n\n";
58
59 #define IO_ERR (1)
60 #define INVALID_INPUT_FILE (2)
61 #define INVALID_PARAMS (3)
62
63 #define MAX_WIDTH (74)
64
open_file(FILE ** f,const char * filename,const char * mode)65 static void open_file(FILE** f, const char* filename, const char* mode) {
66
67 #ifndef _WIN32
68 *f = fopen(filename, mode);
69 #else // _WIN32
70 *f = NULL;
71 fopen_s(f, filename, mode);
72 #endif // _WIN32
73 }
74
output_label(FILE * out,const char * label,const char * name)75 static int output_label(FILE* out, const char* label, const char* name) {
76
77 #ifndef _WIN32
78 return fprintf(out, bc_gen_label, label, name);
79 #else // _WIN32
80
81 size_t i, count = 0, len = strlen(name);
82 char* buf;
83 int ret;
84
85 for (i = 0; i < len; ++i) {
86 count += (name[i] == '\\');
87 }
88
89 buf = (char*) malloc(len + 1 + count);
90 if (buf == NULL) return -1;
91
92 count = 0;
93
94 for (i = 0; i < len; ++i) {
95 buf[i + count] = name[i];
96 if (name[i] == '\\') {
97 count += 1;
98 buf[i + count] = name[i];
99 }
100 }
101
102 buf[i + count] = '\0';
103
104 ret = fprintf(out, bc_gen_label, label, buf);
105
106 free(buf);
107
108 return ret;
109
110 #endif // _WIN32
111 }
112
main(int argc,char * argv[])113 int main(int argc, char *argv[]) {
114
115 FILE *in, *out;
116 char *label, *define, *name;
117 int c, count, slashes, err = IO_ERR;
118 bool has_label, has_define, remove_tabs;
119
120 if (argc < 5) {
121 printf("usage: %s input output name header [label [define [remove_tabs]]]\n", argv[0]);
122 return INVALID_PARAMS;
123 }
124
125 name = argv[3];
126
127 has_label = (argc > 4 && strcmp("", argv[4]) != 0);
128 label = has_label ? argv[4] : "";
129
130 has_define = (argc > 5 && strcmp("", argv[5]) != 0);
131 define = has_define ? argv[5] : "";
132
133 remove_tabs = (argc > 6);
134
135 open_file(&in, argv[1], "r");
136 if (!in) return INVALID_INPUT_FILE;
137
138 open_file(&out, argv[2], "w");
139 if (!out) goto out_err;
140
141 if (fprintf(out, bc_gen_header, argv[1]) < 0) goto err;
142 if (has_label && fprintf(out, bc_gen_label_extern, label) < 0) goto err;
143 if (fprintf(out, bc_gen_name_extern, name) < 0) goto err;
144 if (has_define && fprintf(out, bc_gen_ifdef, define) < 0) goto err;
145 if (has_label && output_label(out, label, argv[1]) < 0) goto err;
146 if (fprintf(out, bc_gen_name, name) < 0) goto err;
147
148 c = count = slashes = 0;
149
150 while (slashes < 2 && (c = fgetc(in)) >= 0) {
151 slashes += (slashes == 1 && c == '/' && fgetc(in) == '\n');
152 slashes += (!slashes && c == '/' && fgetc(in) == '*');
153 }
154
155 if (c < 0) {
156 err = INVALID_INPUT_FILE;
157 goto err;
158 }
159
160 while ((c = fgetc(in)) == '\n');
161
162 while (c >= 0) {
163
164 int val;
165
166 if (!remove_tabs || c != '\t') {
167
168 if (!count && fputc('\t', out) == EOF) goto err;
169
170 val = fprintf(out, "%d,", c);
171 if (val < 0) goto err;
172
173 count += val;
174
175 if (count > MAX_WIDTH) {
176 count = 0;
177 if (fputc('\n', out) == EOF) goto err;
178 }
179 }
180
181 c = fgetc(in);
182 }
183
184 if (!count && (fputc(' ', out) == EOF || fputc(' ', out) == EOF)) goto err;
185 if (fprintf(out, "0\n};\n") < 0) goto err;
186
187 err = (has_define && fprintf(out, bc_gen_endif, define) < 0);
188
189 err:
190 fclose(out);
191 out_err:
192 fclose(in);
193 return err;
194 }
195