use crate::{
metadata::{
tables::{PropertyMapRaw, RowReadable, TableId, TableInfoRef},
token::Token,
},
utils::read_le_at_dyn,
Result,
};
impl RowReadable for PropertyMapRaw {
const TABLE_ID: TableId = TableId::PropertyMap;
fn row_read(data: &[u8], offset: &mut usize, rid: u32, sizes: &TableInfoRef) -> Result<Self> {
let offset_org = *offset;
let parent = read_le_at_dyn(data, offset, sizes.is_large(TableId::TypeDef))?;
let property_list = read_le_at_dyn(data, offset, sizes.is_large(TableId::Property))?;
Ok(PropertyMapRaw {
rid,
token: Token::new(0x1500_0000 + rid),
offset: offset_org,
parent,
property_list,
})
}
}
#[cfg(test)]
mod tests {
use std::sync::Arc;
use crate::metadata::tables::{MetadataTable, TableId, TableInfo};
use super::*;
#[test]
fn crafted_short() {
let data = vec![
0x01, 0x01, 0x02, 0x02, ];
let sizes = Arc::new(TableInfo::new_test(
&[(TableId::TypeDef, 1), (TableId::Property, 1)],
false,
false,
false,
));
let table = MetadataTable::<PropertyMapRaw>::new(&data, 1, sizes).unwrap();
let eval = |row: PropertyMapRaw| {
assert_eq!(row.rid, 1);
assert_eq!(row.token.value(), 0x15000001);
assert_eq!(row.parent, 0x0101);
assert_eq!(row.property_list, 0x0202);
};
{
for row in table.iter() {
eval(row);
}
}
{
let row = table.get(1).unwrap();
eval(row);
}
}
#[test]
fn crafted_long() {
let data = vec![
0x01, 0x01, 0x01, 0x01, 0x02, 0x02, 0x02, 0x02, ];
let sizes = Arc::new(TableInfo::new_test(
&[
(TableId::TypeDef, u16::MAX as u32 + 3),
(TableId::Property, u16::MAX as u32 + 3),
],
true,
true,
true,
));
let table =
MetadataTable::<PropertyMapRaw>::new(&data, u16::MAX as u32 + 3, sizes).unwrap();
let eval = |row: PropertyMapRaw| {
assert_eq!(row.rid, 1);
assert_eq!(row.token.value(), 0x15000001);
assert_eq!(row.parent, 0x01010101);
assert_eq!(row.property_list, 0x02020202);
};
{
for row in table.iter() {
eval(row);
}
}
{
let row = table.get(1).unwrap();
eval(row);
}
}
}