pokecrystal/tools/scan_includes.c

105 lines
2.1 KiB
C

#include "common.h"
void usage() {
fputs("Usage: scan_includes [-h|--help] [-s|--strict] filename.asm\n", stderr);
}
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 = file_size_verbose(filename, f);
char *contents = malloc_verbose(size + 1);
fread_verbose((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;
}
break;
case '"':
ptr++;
ptr = strchr(ptr, '"');
if (!ptr) {
fprintf(stderr, "%s: unterminated string\n", filename);
break;
}
ptr++;
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) {
break;
}
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;
}