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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
#include "nbt.h"
#include "../lib.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
static bool nbt_read_data(tag_t *tag, const stream_t *stream);
static bool nbt_read_header(tag_t *tag, const stream_t *stream, bool named) {
bool ok = true;
if (stream_read_i8(stream, &tag->type) == false)
return false;
if (tag->type == TAG_END)
named = false;
if (named) {
if (stream_read_u16(stream, &tag->name_len) == false)
return false;
} else {
tag->name_len = 0;
}
if (tag->name_len < 1) {
tag->name = "";
} else {
tag->name = xalloc(tag->name_len);
ok = stream_read(stream, tag->name, tag->name_len);
}
if (!ok)
return false;
return true;
}
static bool nbt_read_byte_array(tagdata_t *data, const stream_t *stream) {
if (stream_read_i32(stream, &data->b_arr.size) == false)
return false;
if (data->b_arr.size == 0) {
data->b_arr.data = NULL;
return true;
}
data->b_arr.data = xalloc(data->b_arr.size * sizeof(int8_t));
for (int32_t i = 0; i < data->b_arr.size; i++)
if (stream_read_i8(stream, &data->b_arr.data[i]) == false)
return false;
return true;
}
static bool nbt_read_int_array(tagdata_t *data, const stream_t *stream) {
if (stream_read_i32(stream, &data->i_arr.size) == false)
return false;
if (data->i_arr.size == 0) {
data->i_arr.data = NULL;
return true;
}
data->i_arr.data = xalloc(data->i_arr.size * sizeof(int32_t));
for (int32_t i = 0; i < data->i_arr.size; i++)
if (stream_read_i32(stream, &data->i_arr.data[i]) == false)
return false;
return true;
}
static bool nbt_read_long_array(tagdata_t *data, const stream_t *stream) {
if (stream_read_i32(stream, &data->l_arr.size) == false)
return false;
if (data->l_arr.size == 0) {
data->l_arr.data = NULL;
return true;
}
data->l_arr.data = xalloc(data->l_arr.size * sizeof(int64_t));
for (int32_t i = 0; i < data->l_arr.size; i++)
if (stream_read_i64(stream, &data->l_arr.data[i]) == false)
return false;
return true;
}
static bool nbt_read_string(tagdata_t *data, const stream_t *stream) {
if (stream_read_u16(stream, &data->string.size) == false)
return false;
if (data->string.size < 1) {
data->string.data = NULL;
return true;
}
data->string.data = xalloc(data->string.size);
if (stream_read(stream, data->string.data, data->string.size) == false)
return false;
return true;
}
static bool nbt_read_list(tagdata_t *data, const stream_t *stream) {
if (stream_read_i8(stream, &data->list.type) == false)
return false;
if (stream_read_i32(stream, &data->list.size) == false)
return false;
if (data->list.size <= 0) {
data->list.tags = NULL;
return true;
} else if (data->list.type == 0) {
// tag end is not allowed to be used with non empty list
return false;
}
data->list.tags = xalloc(data->list.size * sizeof(tag_t));
for (int32_t i = 0; i < data->list.size; i++) {
tag_t tag;
tag.type = data->list.type;
tag.name = "";
tag.name_len = 0;
if (nbt_read_data(&tag, stream) == false)
return false;
data->list.tags[i] = tag;
}
return true;
}
static bool nbt_read_compound(tagdata_t *data, const stream_t *stream) {
map_t map;
map_init(&map);
while (1) {
tag_t tag;
if (nbt_read_header(&tag, stream, true) == false) {
map_free(&map);
return false;
}
if (tag.type == TAG_END)
break;
if (tag.name_len < 1) {
map_free(&map);
return false;
}
if (nbt_read_data(&tag, stream) == false) {
map_free(&map);
return false;
}
map_put(&map, &tag);
}
data->compound = map;
return true;
}
static bool nbt_read_data(tag_t *tag, const stream_t *stream) {
bool ok = true;
switch (tag->type) {
case TAG_END:
// tag has no data
break;
case TAG_BYTE:
ok = stream_read_i8(stream, &tag->data.b);
break;
case TAG_SHORT:
ok = stream_read_i16(stream, &tag->data.s);
break;
case TAG_FLOAT:
case TAG_INT:
ok = stream_read_i32(stream, &tag->data.i);
break;
case TAG_DOUBLE:
case TAG_LONG:
ok = stream_read_i64(stream, &tag->data.l);
break;
case TAG_BYTE_ARRAY:
ok = nbt_read_byte_array(&tag->data, stream);
break;
case TAG_STRING:
ok = nbt_read_string(&tag->data, stream);
break;
case TAG_LIST:
ok = nbt_read_list(&tag->data, stream);
break;
case TAG_COMPOUND:
ok = nbt_read_compound(&tag->data, stream);
break;
case TAG_INT_ARRAY:
ok = nbt_read_int_array(&tag->data, stream);
break;
case TAG_LONG_ARRAY:
ok = nbt_read_long_array(&tag->data, stream);
break;
break;
};
return ok;
}
bool nbt_read(tag_t *tag, const stream_t *stream) {
memset(tag, 0, sizeof(tag_t));
if (nbt_read_header(tag, stream, true) == false)
return false;
if (nbt_read_data(tag, stream) == false)
return false;
return true;
}
|
