summaryrefslogtreecommitdiff
path: root/src/nbt/print.c
blob: 4eac31fa0e247bbfab503cfb6b59067292e09b2c (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
#include "nbt.h"

static bool nbt_print_header(const tag_t *tag, const stream_t *stream, bool named) {
	if (stream_write_i8(stream, tag->type) == false)
		return false;
	if (!named)
		return true;
	if (stream_write_u16(stream, tag->name_len) == false)
		return false;
	if (tag->name_len > 0)
		if (stream_write(stream, tag->name, tag->name_len) == false)
			return false;
	return true;
}

static bool nbt_print_byte_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->b_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->b_arr.size; i++)
		if (stream_write_i8(stream, data->b_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_int_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->i_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->i_arr.size; i++)
		if (stream_write_i32(stream, data->i_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_long_array(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i32(stream, data->l_arr.size) == false)
		return false;
	for (int32_t i = 0; i < data->l_arr.size; i++)
		if (stream_write_i64(stream, data->l_arr.data[i]) == false)
			return false;
	return true;
}

static bool nbt_print_string(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_u16(stream, data->string.size) == false)
		return false;
	if (data->string.size < 1)
		return true;
	if (stream_write(stream, data->string.data, data->string.size) == false)
		return false;
	return true;
}

static bool nbt_print_data(const tag_t *tag, const stream_t *stream);

static bool nbt_print_list(const tagdata_t *data, const stream_t *stream) {
	if (stream_write_i8(stream, data->list.type) == false)
		return false;
	if (stream_write_i32(stream, data->list.size) == false)
		return false;
	for (int32_t i = 0; i < data->list.size; i++)
		if (nbt_print_data(&data->list.tags[i], stream) == false)
			return false;
	return true;
}

static bool nbt_print_compound(const tagdata_t *data, const stream_t *stream) {
	for (int32_t i = 0; i < data->compound.size; i++)
		if (nbt_print(&data->compound.tags[i], stream) == false)
			return false;
	if (stream_write_i8(stream, TAG_END) == false)
		return false;
	return true;
}

static bool nbt_print_data(const tag_t *tag, const stream_t *stream) {
	bool ok = true;

	switch (tag->type) {
        case TAG_END:
			// tag end has no data
			break;
        case TAG_BYTE:
			ok = stream_write_i8(stream, tag->data.b);
			break;
        case TAG_SHORT:
			ok = stream_write_i16(stream, tag->data.s);
			break;
        case TAG_INT:
        case TAG_FLOAT:
			ok = stream_write_i32(stream, tag->data.i);
			break;
        case TAG_LONG:
        case TAG_DOUBLE:
			ok = stream_write_i64(stream, tag->data.l);
			break;
        case TAG_BYTE_ARRAY:
			ok = nbt_print_byte_array(&tag->data, stream);
			break;
        case TAG_STRING:
			ok = nbt_print_string(&tag->data, stream);
			break;
        case TAG_LIST:
			ok = nbt_print_list(&tag->data, stream);
			break;
        case TAG_COMPOUND:
			ok = nbt_print_compound(&tag->data, stream);
			break;
        case TAG_INT_ARRAY:
			ok = nbt_print_int_array(&tag->data, stream);
			break;
        case TAG_LONG_ARRAY:
			ok = nbt_print_long_array(&tag->data, stream);
			break;
        }

	return ok;
}

bool nbt_print(const tag_t *tag, const stream_t *stream) {
	if (nbt_print_header(tag, stream, true) == false)
		return false;
	if (nbt_print_data(tag, stream) == false)
		return false;
	return true;
}