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
|
use crate::helper::{AlignExt, Endianness, ReadExt};
use crate::serialized_file::TypeTreeNode;
use anyhow::Result;
use log::{debug, trace};
use std::io::Seek;
use std::{collections::BTreeMap, io::Read};
#[derive(Debug)]
pub enum Value {
Bool(bool),
U8(u8),
U16(u16),
I16(i16),
U32(u32),
I32(i32),
F32(f32),
I64(i64),
F64(f64),
Array(Vec<Value>),
Object {
class: String,
fields: BTreeMap<String, Value>,
},
String(String),
}
pub fn read_value(
ty: &TypeTreeNode,
e: Endianness,
data: &mut (impl Read + Seek),
) -> Result<Value> {
let r = match ty.type_string.as_str() {
"char" => Ok(Value::U8(data.read_u8()?)),
"int" => Ok(Value::I32(data.read_i32(e)?)),
"unsigned int" => Ok(Value::U32(data.read_u32(e)?)),
"UInt8" => Ok(Value::U8(data.read_u8()?)),
"UInt16" => Ok(Value::U16(data.read_u16(e)?)),
"SInt16" => Ok(Value::I16(data.read_i16(e)?)),
"SInt64" => Ok(Value::I64(data.read_i64(e)?)),
"bool" => Ok(Value::Bool(data.read_u8()? != 0)),
"float" => {
data.align(4)?;
Ok(Value::F32(data.read_f32(e)?))
}
"double" => {
data.align(4)?;
Ok(Value::F64(data.read_f64(e)?))
}
"string" => {
let Value::Array(arr) = read_value(&ty.children[0], e, data)? else {
unreachable!()
};
let bytes = arr
.into_iter()
.map(|e| {
if let Value::U8(x) = e {
x
} else {
unreachable!()
}
})
.collect::<Vec<_>>();
Ok(Value::String(String::from_utf8(bytes)?))
}
"Array" => {
let Value::I32(size) = read_value(&ty.children[0], e, data)? else {
unreachable!()
};
trace!("array of size {size}");
let mut elems = Vec::new();
for _ in 0..size {
elems.push(read_value(&ty.children[1], e, data)?);
}
Ok(Value::Array(elems))
}
_ => {
if ty.children.is_empty() && ty.byte_size != -1 {
todo!("need type {:?}", ty.type_string);
}
let mut fields = BTreeMap::new();
for c in &ty.children {
fields.insert(c.name_string.clone(), read_value(&c, e, data)?);
}
Ok(Value::Object {
fields,
class: ty.type_string.clone(),
})
}
};
if ty.post_align() {
data.align(4)?;
}
r
}
|
