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
|
/*
* libnbt++ - A library for the Minecraft Named Binary Tag format.
* Copyright (C) 2013, 2015 ljfa-ag
*
* This file is part of libnbt++.
*
* libnbt++ is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* libnbt++ is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with libnbt++. If not, see <http://www.gnu.org/licenses/>.
*/
#include "microtest.h"
#include "io/stream_reader.h"
#include <sstream>
using namespace nbt;
void test_stream_reader_big()
{
const char input[] {
1, //tag_type::Byte
0, //tag_type::End
11, //tag_type::Int_Array
0x0a, 0x0b, 0x0c, 0x0d, //0x0a0b0c0d in Big Endian
0x00, 0x06, //String length in Big Endian
'f', 'o', 'o', 'b', 'a', 'r',
0 //tag_type::End (invalid with allow_end = false)
};
std::istringstream is(std::string(input, sizeof input));
nbt::io::stream_reader reader(is);
ASSERT(&reader.get_istr() == &is);
ASSERT(reader.get_endian() == endian::big);
ASSERT(reader.read_type() == tag_type::Byte);
ASSERT(reader.read_type(true) == tag_type::End);
ASSERT(reader.read_type(false) == tag_type::Int_Array);
int32_t i;
reader.read_num(i);
ASSERT(i == 0x0a0b0c0dL);
ASSERT(reader.read_string() == "foobar");
EXPECT_EXCEPTION(reader.read_type(false), io::stream_reader::input_error);
ASSERT(!is);
is.clear();
//Test for invalid tag type 12
is.str("\x0c");
EXPECT_EXCEPTION(reader.read_type(), io::stream_reader::input_error);
ASSERT(!is);
is.clear();
//Test for unexpcted EOF on numbers (input too short for int32_t)
is.str("\x03\x04");
reader.read_num(i);
ASSERT(!is);
}
void test_stream_reader_little()
{
const char input[] {
0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, //0x0d0c0b0a09080706 in Little Endian
0x06, 0x00, //String length in Little Endian
'f', 'o', 'o', 'b', 'a', 'r',
0x10, 0x00, //String length (intentionally too large)
'a', 'b', 'c', 'd' //unexpected EOF
};
std::istringstream is(std::string(input, sizeof input));
nbt::io::stream_reader reader(is, endian::little);
ASSERT(reader.get_endian() == endian::little);
int64_t i;
reader.read_num(i);
ASSERT(i == 0x0d0c0b0a09080706LL);
ASSERT(reader.read_string() == "foobar");
EXPECT_EXCEPTION(reader.read_string(), io::stream_reader::input_error);
ASSERT(!is);
}
int main()
{
test_stream_reader_big();
test_stream_reader_little();
}
|
