ds-decomp 0.11.0

Library for ds-decomp, a DS decompilation toolkit.
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

use std::borrow::Cow;

use ds_rom::rom::{Arm9AutoloadError, Rom, raw::RawBuildInfoError};
use snafu::Snafu;

use crate::config::module::ModuleKind;

#[derive(Debug, Snafu)]
pub enum RomGetCodeError {
    #[snafu(display("Module {} not found", module_kind))]
    ModuleNotFound { module_kind: ModuleKind },
    #[snafu(transparent)]
    RawBuildInfo { source: RawBuildInfoError },
    #[snafu(transparent)]
    Arm9Autoload { source: Arm9AutoloadError },
}

pub trait RomExt {
    /// Returns the full binary code of the given module.
    ///
    /// # Errors
    ///
    /// This function will return an error if the given module does not exist in the ROM or if
    /// [`ds_rom::rom::Arm9::code`] or [`ds_rom::rom::Arm9::autoloads`] fail.
    fn get_code(&self, kind: ModuleKind) -> Result<Cow<'_, [u8]>, RomGetCodeError>;
}

impl RomExt for Rom<'_> {
    fn get_code(&self, kind: ModuleKind) -> Result<Cow<'_, [u8]>, RomGetCodeError> {
        match kind {
            ModuleKind::Arm9 => Ok(self.arm9().code()?.into()),
            ModuleKind::Overlay(id) => Ok(self
                .arm9_overlays()
                .get(id as usize)
                .ok_or_else(|| RomGetCodeError::ModuleNotFound { module_kind: kind })?
                .code()
                .into()),
            ModuleKind::Autoload(autoload_kind) => {
                let autoloads = self.arm9().autoloads()?;
                let autoload = autoloads
                    .iter()
                    .find(|a| a.kind() == autoload_kind)
                    .ok_or_else(|| RomGetCodeError::ModuleNotFound { module_kind: kind })?;
                Ok(autoload.code().to_owned().into())
            }
        }
    }
}