hao 0.0.3

.net binary reader
Documentation 
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

use std::path::Path;
use std::{fmt::Debug, path::PathBuf, rc::Rc};

use crate::error::HaoError;
use crate::{error::Result, Module};

// Hard code for testing
const TRUSTED_ASSEMBLY_LOCATION: &'static str =
    r#"C:\Program Files\dotnet\shared\Microsoft.NETCore.App\6.0.16\"#;

pub enum AssemblyLoadResult {
    Loaded(Rc<Module>),
    Ignore,
}

pub trait AssemblyResolver: Debug {
    fn load(&mut self, assembly_name: &str) -> Result<AssemblyLoadResult>;
}

#[derive(Debug, Default)]
pub struct PathAssemblyResolver {
    base_path: std::path::PathBuf,
    pub(crate) assembly_list: Vec<(String, Rc<Module>)>,
}

impl PathAssemblyResolver {
    pub fn new(path: &std::path::Path) -> Self {
        let base_path = path.to_owned();
        Self {
            base_path: base_path,
            assembly_list: Vec::new(),
        }
    }
}

impl PathAssemblyResolver {
    fn find_path_for(&self, assembly_name: &str) -> Option<PathBuf> {
        const KNOWN_EXTENTIONS: [&'static str; 2] = ["dll", "CoreLib.dll"];
        let asm_locations = [Path::new(TRUSTED_ASSEMBLY_LOCATION), &self.base_path];

        for asm_root_path in asm_locations.into_iter() {
            let mut path = asm_root_path.join(format!("{}.ext", assembly_name));

            for ext in KNOWN_EXTENTIONS {
                path.set_extension(ext);
                if path.exists() {
                    return Some(path);
                }
            }
        }
        None
    }
}

impl AssemblyResolver for PathAssemblyResolver {
    fn load(&mut self, assembly_name: &str) -> Result<AssemblyLoadResult> {
        let cached_module = self
            .assembly_list
            .iter()
            .filter(|(saved_asm, _)| saved_asm == assembly_name)
            .map(|asm| asm.1.clone())
            .next();

        if let Some(asm) = cached_module {
            return Ok(AssemblyLoadResult::Loaded(asm));
        }

        if let Some(found_path) = self.find_path_for(assembly_name) {
            let loaded_asm = Rc::new(Module::from_path_no_resolve(found_path)?);
            self.assembly_list
                .push((assembly_name.into(), loaded_asm.clone()));
            loaded_asm.load_dependancies(self)?;
            return Ok(AssemblyLoadResult::Loaded(loaded_asm));
        } else {
            Err(HaoError::IoError(std::io::Error::new(
                std::io::ErrorKind::NotFound,
                format!("Could not resolve assembly {:?}", assembly_name),
            )))
        }
    }
}