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
use crate::{ BinaryReader, BinaryReaderError, Range, Result, SectionIteratorLimited, SectionReader, SectionWithLimitedItems, }; pub struct ModuleCodeSectionReader<'a> { reader: BinaryReader<'a>, count: u32, } #[derive(Debug)] pub struct ModuleCode<'a> { reader: BinaryReader<'a>, } impl<'a> ModuleCodeSectionReader<'a> { pub fn new(data: &'a [u8], offset: usize) -> Result<ModuleCodeSectionReader<'a>> { let mut reader = BinaryReader::new_with_offset(data, offset); let count = reader.read_var_u32()?; Ok(ModuleCodeSectionReader { reader, count }) } pub fn original_position(&self) -> usize { self.reader.original_position() } pub fn get_count(&self) -> u32 { self.count } fn verify_module_end(&self, end: usize) -> Result<()> { if self.reader.buffer.len() < end { return Err(BinaryReaderError::new( "module body extends past end of the module code section", self.reader.original_offset + self.reader.buffer.len(), )); } Ok(()) } pub fn read(&mut self) -> Result<ModuleCode<'a>> { let size = self.reader.read_var_u32()? as usize; let module_start = self.reader.position; let module_end = module_start + size; self.verify_module_end(module_end)?; self.reader.skip_to(module_end); Ok(ModuleCode { reader: BinaryReader::new_with_offset( &self.reader.buffer[module_start..module_end], self.reader.original_offset + module_start, ), }) } } impl<'a> SectionReader for ModuleCodeSectionReader<'a> { type Item = ModuleCode<'a>; fn read(&mut self) -> Result<Self::Item> { ModuleCodeSectionReader::read(self) } fn eof(&self) -> bool { self.reader.eof() } fn original_position(&self) -> usize { ModuleCodeSectionReader::original_position(self) } fn range(&self) -> Range { self.reader.range() } } impl<'a> SectionWithLimitedItems for ModuleCodeSectionReader<'a> { fn get_count(&self) -> u32 { ModuleCodeSectionReader::get_count(self) } } impl<'a> IntoIterator for ModuleCodeSectionReader<'a> { type Item = Result<ModuleCode<'a>>; type IntoIter = SectionIteratorLimited<ModuleCodeSectionReader<'a>>; fn into_iter(self) -> Self::IntoIter { SectionIteratorLimited::new(self) } } impl<'a> ModuleCode<'a> { pub fn raw_bytes(&self) -> (usize, &[u8]) { (self.reader.original_position(), self.reader.buffer) } pub fn original_position(&self) -> usize { self.reader.original_position() } }