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 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 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 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