Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 1 | /* |
Gavin Howard | b5904bf | 2018-02-20 13:28:18 -0700 | [diff] [blame] | 2 | * ***************************************************************************** |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 3 | * |
Gavin Howard | b5904bf | 2018-02-20 13:28:18 -0700 | [diff] [blame] | 4 | * Copyright 2018 Gavin D. Howard |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 5 | * |
| 6 | * Permission to use, copy, modify, and/or distribute this software for any |
| 7 | * purpose with or without fee is hereby granted. |
| 8 | * |
| 9 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH |
| 10 | * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY |
| 11 | * AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, |
| 12 | * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM |
| 13 | * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR |
| 14 | * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR |
| 15 | * PERFORMANCE OF THIS SOFTWARE. |
| 16 | * |
Gavin Howard | b5904bf | 2018-02-20 13:28:18 -0700 | [diff] [blame] | 17 | * ***************************************************************************** |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 18 | * |
| 19 | * Generates a const array from a bc script. |
| 20 | * |
| 21 | */ |
| 22 | |
| 23 | #include <stdio.h> |
Gavin Howard | 207053c | 2018-02-20 11:12:43 -0700 | [diff] [blame] | 24 | #include <stdlib.h> |
| 25 | #include <string.h> |
| 26 | |
| 27 | #include <libgen.h> |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 28 | |
Gavin Howard | 3f68df7 | 2018-03-22 20:30:27 -0600 | [diff] [blame] | 29 | const char *bc_gen_usage = "usage: gen bc_script output [header]\n"; |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 30 | |
| 31 | #define INVALID_PARAMS (1) |
Gavin Howard | 207053c | 2018-02-20 11:12:43 -0700 | [diff] [blame] | 32 | #define MALLOC_FAIL (2) |
| 33 | #define INVALID_INPUT_FILE (3) |
| 34 | #define INVALID_OUTPUT_FILE (4) |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 35 | #define INVALID_HEADER_FILE (5) |
| 36 | #define IO_ERR (6) |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 37 | |
Gavin Howard | 7edf829 | 2018-03-09 02:07:59 -0700 | [diff] [blame] | 38 | #define MAX_WIDTH (74) |
| 39 | |
Gavin Howard | 8d1f1db | 2018-02-23 11:29:41 -0700 | [diff] [blame] | 40 | int main(int argc, char *argv[]) { |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 41 | |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 42 | FILE *in; |
| 43 | FILE *out; |
| 44 | FILE *header; |
| 45 | char *header_name; |
| 46 | char *header_buf; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 47 | int c; |
| 48 | int count; |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 49 | char *buffer; |
| 50 | size_t len, read; |
Gavin Howard | 8d1f1db | 2018-02-23 11:29:41 -0700 | [diff] [blame] | 51 | char *buf; |
| 52 | char *base; |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 53 | int err; |
Gavin Howard | 7edf829 | 2018-03-09 02:07:59 -0700 | [diff] [blame] | 54 | int slashes; |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 55 | |
| 56 | err = 0; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 57 | |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 58 | if (argc < 3) { |
Gavin Howard | d8a2b50 | 2018-02-24 13:26:48 -0700 | [diff] [blame] | 59 | printf("%s", bc_gen_usage); |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 60 | return INVALID_PARAMS; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 61 | } |
| 62 | |
Gavin Howard | 207053c | 2018-02-20 11:12:43 -0700 | [diff] [blame] | 63 | buf = malloc(strlen(argv[1]) + 1); |
| 64 | |
| 65 | if (!buf) return MALLOC_FAIL; |
| 66 | |
| 67 | strcpy(buf, argv[1]); |
| 68 | |
| 69 | base = basename(buf); |
| 70 | |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 71 | in = fopen(argv[1], "r"); |
| 72 | |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 73 | if (!in) { |
| 74 | err = INVALID_INPUT_FILE; |
| 75 | goto in_err; |
| 76 | } |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 77 | |
| 78 | out = fopen(argv[2], "w"); |
| 79 | |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 80 | if (!out) { |
| 81 | err = INVALID_OUTPUT_FILE; |
| 82 | goto out_err; |
| 83 | } |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 84 | |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 85 | if (argc >= 4) header_name = argv[3]; |
Gavin Howard | 6a11d79 | 2018-03-24 06:47:28 -0600 | [diff] [blame] | 86 | else header_name = "lib/header.c"; |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 87 | |
| 88 | header = fopen(header_name, "r"); |
| 89 | |
| 90 | if (!header) { |
| 91 | err = INVALID_HEADER_FILE; |
| 92 | goto header_err; |
| 93 | } |
| 94 | |
| 95 | fseek(header, 0, SEEK_END); |
| 96 | len = ftell(header); |
| 97 | fseek(header, 0, SEEK_SET); |
| 98 | |
| 99 | buffer = malloc(len + 1); |
| 100 | |
| 101 | if (!buffer) { |
| 102 | err = MALLOC_FAIL; |
| 103 | goto buffer_err; |
| 104 | } |
| 105 | |
| 106 | read = fread(buffer, 1, len, header); |
| 107 | |
| 108 | if (read != len) { |
| 109 | err = IO_ERR; |
| 110 | goto header_buf_err; |
| 111 | } |
| 112 | |
| 113 | buffer[len] = '\0'; |
| 114 | |
| 115 | header_buf = malloc(len + strlen(base) + 1); |
| 116 | |
| 117 | if (!header_buf) { |
| 118 | err= MALLOC_FAIL; |
| 119 | goto header_buf_err; |
| 120 | } |
| 121 | |
| 122 | if (sprintf(header_buf, buffer, base) < 0) { |
| 123 | err = IO_ERR; |
| 124 | goto error; |
| 125 | } |
| 126 | |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 127 | count = 0; |
| 128 | |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 129 | if (fputs(header_buf, out) == EOF) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 130 | err = IO_ERR; |
| 131 | goto error; |
| 132 | } |
Gavin Howard | 207053c | 2018-02-20 11:12:43 -0700 | [diff] [blame] | 133 | |
Gavin Howard | 8d1f1db | 2018-02-23 11:29:41 -0700 | [diff] [blame] | 134 | if (fprintf(out, "const char *bc_lib_name = \"%s\";\n\n", base) < 0) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 135 | err = IO_ERR; |
| 136 | goto error; |
| 137 | } |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 138 | |
Gavin Howard | eb9a822 | 2018-03-13 09:25:56 -0600 | [diff] [blame] | 139 | if (fprintf(out, "const char bc_lib[] = {\n") < 0) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 140 | err = IO_ERR; |
| 141 | goto error; |
| 142 | } |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 143 | |
Gavin Howard | 7edf829 | 2018-03-09 02:07:59 -0700 | [diff] [blame] | 144 | slashes = 0; |
| 145 | |
| 146 | while (slashes < 2 && (c = fgetc(in)) >= 0) { |
| 147 | if (slashes == 1 && c == '/' && fgetc(in) == '\n') ++slashes; |
| 148 | if (!slashes && c == '/' && fgetc(in) == '*') ++slashes; |
| 149 | } |
| 150 | |
| 151 | if (c < 0) { |
| 152 | err = INVALID_INPUT_FILE; |
| 153 | goto error; |
| 154 | } |
| 155 | |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 156 | while ((c = fgetc(in)) >= 0) { |
| 157 | |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 158 | int val; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 159 | |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 160 | if (!count) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 161 | if (fprintf(out, " ") < 0) { |
| 162 | err = IO_ERR; |
| 163 | goto error; |
| 164 | } |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 165 | } |
| 166 | |
| 167 | val = fprintf(out, "%d,", c); |
| 168 | |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 169 | if (val < 0) { |
| 170 | err = IO_ERR; |
| 171 | goto error; |
| 172 | } |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 173 | |
| 174 | count += val; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 175 | |
Gavin Howard | 7edf829 | 2018-03-09 02:07:59 -0700 | [diff] [blame] | 176 | if (count > MAX_WIDTH) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 177 | |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 178 | count = 0; |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 179 | |
| 180 | if (fputc('\n', out) == EOF) { |
| 181 | err = IO_ERR; |
| 182 | goto error; |
| 183 | } |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 184 | } |
| 185 | } |
| 186 | |
Gavin Howard | 8bdbb1f | 2018-02-20 16:17:16 -0700 | [diff] [blame] | 187 | if (!count) { |
Gavin Howard | 7edf829 | 2018-03-09 02:07:59 -0700 | [diff] [blame] | 188 | if (fputc(' ', out) == EOF || fputc(' ', out) == EOF) { |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 189 | err = IO_ERR; |
| 190 | goto error; |
| 191 | } |
Gavin Howard | fe2351d | 2018-02-19 14:10:15 -0700 | [diff] [blame] | 192 | } |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 193 | if (fprintf(out, "0\n};\n") < 0) { |
| 194 | err = IO_ERR; |
| 195 | goto error; |
| 196 | } |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 197 | |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 198 | error: |
| 199 | |
Gavin Howard | 1ba6bee | 2018-03-08 11:52:45 -0700 | [diff] [blame] | 200 | free(header_buf); |
| 201 | |
| 202 | header_buf_err: |
| 203 | |
| 204 | free(buffer); |
| 205 | |
| 206 | buffer_err: |
| 207 | |
| 208 | fclose(header); |
| 209 | |
| 210 | header_err: |
| 211 | |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 212 | fclose(out); |
| 213 | |
Gavin Howard | d571acf | 2018-02-20 17:13:07 -0700 | [diff] [blame] | 214 | out_err: |
| 215 | |
| 216 | fclose(in); |
| 217 | |
| 218 | in_err: |
| 219 | |
| 220 | free(buf); |
| 221 | |
| 222 | return err; |
Gavin Howard | f833399 | 2018-02-19 13:55:25 -0700 | [diff] [blame] | 223 | } |