macho-assembler 0.1.1

Mach-O assembler for Gaia project
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

use crate::{
    helpers::MachoWriter,
    types::{LoadCommand, MachoHeader, MachoProgram},
};
use gaia_binary::{BinaryWriter, Fixed, LittleEndian};
use gaia_types::GaiaError;
use std::io::Write;

/// Mach-O 目标文件写入器
#[derive(Debug)]
pub struct ObjectWriter<W: Write> {
    writer: BinaryWriter<W, Fixed<LittleEndian>>,
}

impl<W: Write> ObjectWriter<W> {
    /// 创建一个新的目标文件写入器
    pub fn new(writer: BinaryWriter<W, Fixed<LittleEndian>>) -> Self {
        Self { writer }
    }

    /// 写入目标文件头
    pub fn write_header(&mut self, header: &MachoHeader) -> Result<(), GaiaError> {
        self.writer.write_u32(header.magic)?;
        self.writer.write_u32(header.cpu_type)?;
        self.writer.write_u32(header.cpu_subtype)?;
        self.writer.write_u32(header.file_type)?;
        self.writer.write_u32(header.ncmds)?;
        self.writer.write_u32(header.sizeofcmds)?;
        self.writer.write_u32(header.flags)?;

        if let Some(reserved) = header.reserved {
            self.writer.write_u32(reserved)?;
        }

        Ok(())
    }

    /// 写入加载命令
    pub fn write_load_commands(&mut self, load_commands: &[LoadCommand]) -> Result<(), GaiaError> {
        for cmd in load_commands {
            self.writer.write_u32(cmd.cmd)?;
            self.writer.write_u32(cmd.cmdsize)?;
            self.writer.write_bytes(&cmd.data)?;
        }
        Ok(())
    }
}

impl<W: Write> MachoWriter<W> for ObjectWriter<W> {
    fn write_program(&mut self, program: &MachoProgram) -> Result<(), GaiaError> {
        self.write_header(&program.header)?;
        self.write_load_commands(&program.load_commands)?;
        Ok(())
    }

    fn writer(&mut self) -> &mut BinaryWriter<W, Fixed<LittleEndian>> {
        &mut self.writer
    }
}