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

use common::*;
use dbi::Module;
use msf::Stream;
use std::mem;
use symbol::SymbolIter;

/// The signature at the start of a module information stream.
const MODI_SIGNATURE: u32 = 4;

#[allow(dead_code)]
enum Lines {
    C11(usize),
    C13(usize),
}

/// This struct contains data about a single module from its module info stream.
///
/// The module info stream is where private symbols and line info is stored.
pub struct ModuleInfo<'m> {
    stream: Stream<'m>,
    symbols_size: usize,
    _lines: Lines,
}

impl<'m> ModuleInfo<'m> {
    /// Get an iterator over the private symbols of this module.
    pub fn symbols(&self) -> Result<SymbolIter> {
        let mut buf = self.stream.parse_buffer();
        buf.parse_u32()?;
        let symbols = buf.take(self.symbols_size - mem::size_of::<u32>())?;
        Ok(SymbolIter::new(symbols.into()))
    }
}

pub fn new_module_info<'s, 'm>(stream: Stream<'s>, module: &Module<'m>) -> Result<ModuleInfo<'s>> {
    let info = module.info();
    {
        let mut buf = stream.parse_buffer();
        if buf.parse_u32()? != MODI_SIGNATURE {
            return Err(Error::UnimplementedFeature("Unsupported module info format"));
        }
    }
    let lines = if info.lines_size > 0 {
        Lines::C11(info.lines_size as usize)
    } else {
        Lines::C13(info.c13_lines_size as usize)
    };
    let symbols_size = info.symbols_size as usize;
    Ok(ModuleInfo {
        stream,
        symbols_size,
        _lines: lines,
    })
}