ref: ae79200b1dd8ed693f968dc61048aaf9515db530
dir: /tools/scan_includes.c/
#define PROGRAM_NAME "scan_includes" #define USAGE_OPTS "[-h|--help] [-s|--strict] filename.asm" #include "common.h" void parse_args(int argc, char *argv[], bool *strict) { struct option long_options[] = { {"strict", no_argument, 0, 's'}, {"help", no_argument, 0, 'h'}, {0} }; for (int opt; (opt = getopt_long(argc, argv, "sh", long_options)) != -1;) { switch (opt) { case 's': *strict = true; break; case 'h': usage_exit(0); break; default: usage_exit(1); } } } void scan_file(const char *filename, bool strict) { errno = 0; FILE *f = fopen(filename, "rb"); if (!f) { if (strict) { error_exit("Could not open file \"%s\": %s\n", filename, strerror(errno)); } else { return; } } long size = xfsize(filename, f); char *contents = xmalloc(size + 1); xfread((uint8_t *)contents, size, filename, f); fclose(f); contents[size] = '\0'; for (char *ptr = contents; ptr && ptr - contents < size; ptr++) { bool is_incbin = false, is_include = false; switch (*ptr) { case ';': ptr = strchr(ptr, '\n'); if (!ptr) { fprintf(stderr, "%s: no newline at end of file\n", filename); } break; case '"': ptr++; ptr = strchr(ptr, '"'); if (ptr) { ptr++; } else { fprintf(stderr, "%s: unterminated string\n", filename); } break; case 'I': case 'i': is_incbin = !strncmp(ptr, "INCBIN", 6) || !strncmp(ptr, "incbin", 6); is_include = !strncmp(ptr, "INCLUDE", 7) || !strncmp(ptr, "include", 7); if (is_incbin || is_include) { ptr = strchr(ptr, '"'); if (ptr) { ptr++; char *include_path = ptr; size_t length = strcspn(ptr, "\""); ptr += length + 1; include_path[length] = '\0'; printf("%s ", include_path); if (is_include) { scan_file(include_path, strict); } } } break; } } free(contents); } int main(int argc, char *argv[]) { bool strict = false; parse_args(argc, argv, &strict); argc -= optind; argv += optind; if (argc < 1) { usage_exit(1); } scan_file(argv[0], strict); return 0; }