summaryrefslogtreecommitdiff
path: root/src/tag_read.c
blob: 5f43e7ce9be6815c54239a35655ff4e9fe7ccc85 (plain)
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
#include "tag.h"
#include "lib.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

bool tag_read_header(tag_t *tag, 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 tag_read_byte_array(tagdata_t *data, 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 tag_read_int_array(tagdata_t *data, 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 tag_read_long_array(tagdata_t *data, 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 tag_read_string(tagdata_t *data, 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 tag_read_list(tagdata_t *data, 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 (tag_read_data(&tag, stream) == false)
			return false;
		data->list.tags[i] = tag;
	}
	return true;
}

static bool tag_read_compound(tagdata_t *data, stream_t *stream) {
	int32_t size = 0;
	int32_t capacity = 8;
	tag_t *tags = xalloc(capacity * sizeof(tag_t));

	while (1) {

		tag_t tag;

		if (tag_read_header(&tag, stream, true) == false)
			return false;

		if (tag.type == TAG_END)
			break;

		if (tag_read_data(&tag, stream) == false)
			return false;
		
		if (size == capacity) {
			capacity *= 2;
			tags = xrealloc(tags, capacity * sizeof(tag_t));
		}

		tags[size++] = tag;
	}

	data->compound.size = size;
	data->compound.tags = xalloc(size * sizeof(tag_t));
	memcpy(data->compound.tags, tags, size * sizeof(tag_t));
	free(tags);

	return true;
}
bool tag_read_data(tag_t *tag, 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 = tag_read_byte_array(&tag->data, stream);
			break;
        case TAG_STRING:
			ok = tag_read_string(&tag->data, stream);
			break;
        case TAG_LIST:
			ok = tag_read_list(&tag->data, stream);
			break;
        case TAG_COMPOUND:
			ok = tag_read_compound(&tag->data, stream);
			break;
        case TAG_INT_ARRAY:
			ok = tag_read_int_array(&tag->data, stream);
			break;
        case TAG_LONG_ARRAY:
			ok = tag_read_long_array(&tag->data, stream);
			break;
          break;
        };
    return ok;
}

bool tag_read(tag_t *tag, stream_t *stream, bool named) {
	memset(tag, 0, sizeof(tag_t));
	if (tag_read_header(tag, stream, named) == false)
		return false;
	if (tag_read_data(tag, stream) == false)
		return false;
	return true;
}