1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#ifndef NDEBUG
#ifdef _MSC_VER
#define debug_printf(fmt, ...) fprintf(stderr, fmt, __VA_ARGS__)
#else
#define debug_printf(fmt, ...) fprintf(stderr, fmt, ##__VA_ARGS__)
#endif //_MSC_VER
#else
#define debug_printf(fmt, ...) ((void)0)
#endif //NDEBUG
#define FIND_FAIL (-1u)
struct NtrDirHeader {
unsigned offset;
unsigned short first_file;
unsigned short count_or_parent;
};
static char FILEBUF[BUFSIZ];
unsigned find_file(struct NtrDirHeader * fnt, const char * cfilename) {
unsigned file_id = fnt->first_file;
char * filename = strdup(cfilename);
char * tokenizer = filename;
int found = 0;
char * tree = (char *)fnt + fnt->offset;
char * token;
while ((token = strtok(tokenizer, "/")) != NULL) {
tokenizer = NULL;
debug_printf("TOKEN: %s\n", token);
long toklen = strlen(token);
while (*tree) {
char flag = *tree++;
#ifndef NDEBUG
char *entname = malloc((flag & 0x7F) + 1);
*stpncpy(entname, tree, flag & 0x7F) = 0;
debug_printf("testing entry %s...", entname);
free(entname);
#endif //NDEBUG
if ((toklen != (flag & 0x7F)) || strncmp(tree, token, toklen) != 0) {
debug_printf("no; is %s\n", (flag & 0x80) ? "dir" : "file");
// Next entry
tree += (flag & 0x7F);
if (flag & 0x80) {
tree += 2; // skip dir id
}
else {
file_id++;
}
}
else {
debug_printf("yes; is %s\n", (flag & 0x80) ? "dir" : "file");
tree += (flag & 0x7F);
if (flag & 0x80) {
// navigate to next dir
unsigned short dir_id = (unsigned char) tree[0] | ((unsigned char) tree[1] << 8);
file_id = fnt[dir_id & 0xFFF].first_file;
tree = (char *)fnt + fnt[dir_id & 0xFFF].offset;
break;
}
else {
found = 1;
token = strtok(NULL, "/");
goto done;
}
}
}
}
done:
free(filename);
if (!found || token != NULL) {
file_id = FIND_FAIL;
}
return file_id;
}
int main(int argc, char ** argv) {
if (argc < 3) {
fprintf(stderr, "missing required argument: %s\n", (argc == 1) ? "BASEROM" : "FILENAME");
return 1;
}
FILE *baserom = fopen(argv[1], "rb");
if (baserom == NULL) {
fprintf(stderr, "unable to open ROM %s for reading\n", argv[1]);
return 1;
}
debug_printf("opened baserom\n");
// fnt offset, fnt size, fat offset, fat size
unsigned offsets[4];
fseek(baserom, 64, SEEK_SET);
if (fread(offsets, 4, 4, baserom) != 4) {
fprintf(stderr, "failed to read from baserom\n");
fclose(baserom);
return 1;
}
debug_printf("read offsets\n");
// read fnt
struct NtrDirHeader *fnt = malloc(offsets[1]);
if (fnt == NULL) {
fprintf(stderr, "unable to allocate FNT buffer\n");
fclose(baserom);
return 1;
}
fseek(baserom, offsets[0], SEEK_SET);
if (fread(fnt, 1, offsets[1], baserom) != offsets[1]) {
fprintf(stderr, "unable to read FNT\n");
free(fnt);
fclose(baserom);
return 1;
}
debug_printf("read fnt\n");
unsigned file_id = find_file(fnt, argv[2]);
free(fnt);
if (file_id == FIND_FAIL) {
fprintf(stderr, "file not found");
fclose(baserom);
return 1;
}
debug_printf("found file with id %u\n", file_id);
// Extract the file to stdout
if (8 * file_id >= offsets[3]) {
fprintf(stderr, "nitrofs lookup failed");
fclose(baserom);
return 1;
}
FILE *out = fdopen(dup(fileno(stdout)), "wb");
fseek(baserom, offsets[2] + 8 * file_id, SEEK_SET);
fread(offsets, 4, 2, baserom);
fseek(baserom, offsets[0], SEEK_SET);
while (offsets[1] - offsets[0] > BUFSIZ) {
fread(FILEBUF, 1, BUFSIZ, baserom);
fwrite(FILEBUF, 1, BUFSIZ, out);
offsets[0] += BUFSIZ;
}
fread(FILEBUF, 1, offsets[1] - offsets[0], baserom);
fwrite(FILEBUF, 1, offsets[1] - offsets[0], out);
fclose(out);
fclose(baserom);
return 0;
}
|
