ref: 23cdf06e5ec6e217dcd91a1414dfa7801d453044
dir: /tools/scan_includes.c/
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <stdbool.h> #include <getopt.h> void usage(void) { printf("Usage: scan_includes [-h] [-s] filename\n" "-h, --help\n" " Print usage and exit\n" "-s, --strict\n" " Fail if a file cannot be read\n"); } struct Options { bool help; bool strict; }; struct Options Options = {0}; void scan_file(char* filename) { FILE *f = fopen(filename, "rb"); if (!f) { if (Options.strict) { fprintf(stderr, "Could not open file: '%s'\n", filename); exit(1); } else { return; } } fseek(f, 0, SEEK_END); long size = ftell(f); rewind(f); char *buffer = malloc(size + 1); char *orig = buffer; size = fread(buffer, 1, size, f); buffer[size] = '\0'; fclose(f); for (; buffer && (buffer - orig < size); buffer++) { bool is_include = false; bool is_incbin = false; switch (*buffer) { case ';': buffer = strchr(buffer, '\n'); if (!buffer) { fprintf(stderr, "%s: no newline at end of file\n", filename); break; } break; case '"': buffer++; buffer = strchr(buffer, '"'); if (!buffer) { fprintf(stderr, "%s: unterminated string\n", filename); break; } buffer++; break; case 'i': case 'I': if ((strncmp(buffer, "INCBIN", 6) == 0) || (strncmp(buffer, "incbin", 6) == 0)) { is_incbin = true; } else if ((strncmp(buffer, "INCLUDE", 7) == 0) || (strncmp(buffer, "include", 7) == 0)) { is_include = true; } if (is_incbin || is_include) { buffer = strchr(buffer, '"'); if (!buffer) { break; } buffer++; int length = strcspn(buffer, "\""); char *include = malloc(length + 1); strncpy(include, buffer, length); include[length] = '\0'; printf("%s ", include); if (is_include) { scan_file(include); } free(include); buffer = strchr(buffer, '"'); } break; } if (!buffer) { break; } } free(orig); } int main(int argc, char* argv[]) { int i = 0; struct option long_options[] = { {"strict", no_argument, 0, 's'}, {"help", no_argument, 0, 'h'}, {0} }; int opt = -1; while ((opt = getopt_long(argc, argv, "sh", long_options, &i)) != -1) { switch (opt) { case 's': Options.strict = true; break; case 'h': Options.help = true; break; default: usage(); exit(1); break; } } argc -= optind; argv += optind; if (Options.help) { usage(); return 0; } if (argc < 1) { usage(); exit(1); } scan_file(argv[0]); return 0; }