use crate::{
metadata::{
tables::{CodedIndex, CodedIndexType, ImplMapRaw, RowReadable, TableId, TableInfoRef},
token::Token,
},
utils::{read_le_at, read_le_at_dyn},
Result,
};
impl RowReadable for ImplMapRaw {
const TABLE_ID: TableId = TableId::ImplMap;
fn row_read(data: &[u8], offset: &mut usize, rid: u32, sizes: &TableInfoRef) -> Result<Self> {
Ok(ImplMapRaw {
rid,
token: Token::new(0x1C00_0000u32.saturating_add(rid)),
offset: *offset,
mapping_flags: u32::from(read_le_at::<u16>(data, offset)?),
member_forwarded: CodedIndex::read(
data,
offset,
sizes,
CodedIndexType::MemberForwarded,
)?,
import_name: read_le_at_dyn(data, offset, sizes.is_large_str())?,
import_scope: read_le_at_dyn(data, offset, sizes.is_large(TableId::ModuleRef))?,
})
}
}
#[cfg(test)]
mod tests {
use std::sync::Arc;
use super::*;
use crate::metadata::tables::{MetadataTable, TableId, TableInfo};
#[test]
fn crafted_short() {
let data = vec![
0x01, 0x01, 0x02, 0x00, 0x03, 0x03, 0x04, 0x04, ];
let sizes = Arc::new(TableInfo::new_test(
&[
(TableId::ImplMap, 1),
(TableId::Field, 10),
(TableId::MethodDef, 10),
(TableId::ModuleRef, 10),
],
false,
false,
false,
));
let table = MetadataTable::<ImplMapRaw>::new(&data, 1, sizes).unwrap();
let eval = |row: ImplMapRaw| {
assert_eq!(row.rid, 1);
assert_eq!(row.token.value(), 0x1C000001);
assert_eq!(row.mapping_flags, 0x0101);
assert_eq!(
row.member_forwarded,
CodedIndex::new(TableId::Field, 1, CodedIndexType::MemberForwarded)
);
assert_eq!(row.import_name, 0x0303);
assert_eq!(row.import_scope, 0x0404);
};
{
for row in table.iter() {
let row = row.expect("row parses");
eval(row);
}
}
{
let row = table.get(1).unwrap().unwrap();
eval(row);
}
}
#[test]
fn crafted_long() {
let data = vec![
0x01, 0x01, 0x02, 0x00, 0x00, 0x00, 0x03, 0x03, 0x03, 0x03, 0x04, 0x04, 0x04, 0x04, ];
let sizes = Arc::new(TableInfo::new_test(
&[
(TableId::ImplMap, u16::MAX as u32 + 3),
(TableId::Field, u16::MAX as u32 + 3),
(TableId::MethodDef, u16::MAX as u32 + 3),
(TableId::ModuleRef, u16::MAX as u32 + 3),
],
true,
true,
true,
));
let table = MetadataTable::<ImplMapRaw>::new(&data, 1, sizes).unwrap();
let eval = |row: ImplMapRaw| {
assert_eq!(row.rid, 1);
assert_eq!(row.token.value(), 0x1C000001);
assert_eq!(row.mapping_flags, 0x0101);
assert_eq!(
row.member_forwarded,
CodedIndex::new(TableId::Field, 1, CodedIndexType::MemberForwarded)
);
assert_eq!(row.import_name, 0x03030303);
assert_eq!(row.import_scope, 0x04040404);
};
{
for row in table.iter() {
let row = row.expect("row parses");
eval(row);
}
}
{
let row = table.get(1).unwrap().unwrap();
eval(row);
}
}
}