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
use std::io::{Read, Result, Error, ErrorKind};
use byteorder::{BigEndian, ReadBytesExt};
use serde_json::{Map, Value};
fn parse_utf8<R: Read>(r: &mut R) -> Result<String> {
let length = r.read_u8()?;
let mut buf = vec![0; length as usize];
r.read_exact(&mut buf)?;
String::from_utf8(buf).map_err(|e| Error::new(ErrorKind::InvalidData, e))
}
fn parse_val<R: Read>(r: &mut R) -> Result<Value> {
match r.read_u8()? {
0x53 => match r.read_u8()? {
0x55 => Ok(Value::String(parse_utf8(r)?)),
c => Err(err!("Expected 0x55 for string length, but got: {}", c)),
},
0x6c => Ok(Value::Number(serde_json::Number::from(r.read_i32::<BigEndian>()?))),
0x7b => Ok(Value::Object(parse_map(r)?)),
c => Err(err!("unexpected UBJSON value type: {}", c)),
}
}
fn parse_key<R: Read>(r: &mut R) -> Result<Option<String>> {
match r.read_u8()? {
0x55 => Ok(Some(parse_utf8(r)?)),
0x7d => Ok(None),
c => Err(err!("unexpected UBJSON key type: {}", c)),
}
}
pub fn parse_map<R: Read>(r: &mut R) -> Result<Map<String, Value>> {
let mut m = Map::new();
while match parse_key(r)? {
Some(k) => {m.insert(k, parse_val(r)?); true},
None => false,
} {}
Ok(m)
}