pe-assembler 0.1.1

PE/COFF assembler for Windows instruction sets - strongly typed, object-oriented, zero-dependency core
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use crate::{
    helpers::{read_coff_header, read_coff_object, read_section_headers, CoffReader},
    types::coff::{CoffHeader, CoffInfo, CoffObject, SectionHeader},
};
use gaia_types::{GaiaDiagnostics, GaiaError};
use std::io::{Read, Seek};

/// COFF object file (.obj) structure, lazy reader
#[derive(Debug)]
pub struct ObjReader<R> {
    /// Binary reader (positioned at the start of the object file)
    reader: R,
    lazy_header: Option<CoffHeader>,
    lazy_section_headers: Option<Vec<SectionHeader>>,
    lazy_object: Option<CoffObject>,
    lazy_info: Option<CoffInfo>,
    diagnostics: Vec<GaiaError>,
}

impl<R> ObjReader<R> {
    pub fn new(reader: R) -> Self {
        Self { reader, lazy_header: None, lazy_section_headers: None, lazy_object: None, lazy_info: None, diagnostics: vec![] }
    }
}

impl<W: Read + Seek> CoffReader<W> for ObjReader<W> {
    fn get_viewer(&mut self) -> &mut W {
        &mut self.reader
    }

    fn add_diagnostics(&mut self, error: impl Into<GaiaError>) {
        self.diagnostics.push(error.into());
    }

    fn get_coff_header(&mut self) -> Result<&CoffHeader, GaiaError> {
        if self.lazy_header.is_none() {
            let header = read_coff_header(self)?;
            self.lazy_header = Some(header);
        }
        Ok(self.lazy_header.as_ref().unwrap())
    }

    fn set_coff_header(&mut self, head: CoffHeader) -> Option<CoffHeader> {
        self.lazy_header.replace(head)
    }

    fn get_section_headers(&mut self) -> Result<&[SectionHeader], GaiaError> {
        if self.lazy_section_headers.is_none() {
            let headers = read_section_headers(self)?;
            self.lazy_section_headers = Some(headers);
        }
        Ok(self.lazy_section_headers.as_ref().unwrap())
    }

    fn set_section_headers(&mut self, headers: Vec<SectionHeader>) -> Vec<SectionHeader> {
        self.lazy_section_headers.replace(headers).unwrap_or_default()
    }

    fn get_coff_object(&mut self) -> Result<&CoffObject, GaiaError> {
        if self.lazy_object.is_none() {
            let object = read_coff_object(self)?;
            self.lazy_object = Some(object);
        }
        Ok(self.lazy_object.as_ref().unwrap())
    }

    fn set_coff_object(&mut self, object: CoffObject) -> Option<CoffObject> {
        self.lazy_object.replace(object)
    }

    fn get_coff_info(&mut self) -> Result<&CoffInfo, GaiaError> {
        if self.lazy_info.is_none() {
            let info = self.create_coff_info()?;
            self.lazy_info = Some(info);
        }
        Ok(self.lazy_info.as_ref().unwrap())
    }

    fn set_coff_info(&mut self, info: CoffInfo) -> Option<CoffInfo> {
        self.lazy_info.replace(info)
    }
}

impl<W: Read + Seek> ObjReader<W> {
    /// Read full COFF object (lazy read, will cache the result)
    pub fn read_object(mut self) -> GaiaDiagnostics<CoffObject> {
        match self.get_coff_object() {
            Ok(_) => match self.lazy_object {
                Some(object) => GaiaDiagnostics { result: Ok(object), diagnostics: self.diagnostics },
                None => unreachable!(),
            },
            Err(error) => GaiaDiagnostics { result: Err(error), diagnostics: self.diagnostics },
        }
    }

    /// View COFF object file info
    pub fn view(&mut self) -> Result<CoffInfo, GaiaError> {
        Ok(self.get_coff_info()?.clone())
    }
}