midenc-codegen-masm 0.7.2

Miden Assembly backend for the Miden compiler
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

use alloc::sync::Arc;

use miden_assembly::{
    PathBuf as LibraryPath,
    ast::{Module, ModuleKind},
};
use midenc_session::diagnostics::{PrintDiagnostic, SourceLanguage, SourceManager, Uri};

pub const I32_INTRINSICS_MODULE_NAME: &str = "::intrinsics::i32";
pub const I64_INTRINSICS_MODULE_NAME: &str = "::intrinsics::i64";
pub const I128_INTRINSICS_MODULE_NAME: &str = "::intrinsics::i128";
pub const MEM_INTRINSICS_MODULE_NAME: &str = "::intrinsics::mem";
pub const CRYPTO_INTRINSICS_MODULE_NAME: &str = "::intrinsics::crypto";
pub const ADVICE_INTRINSICS_MODULE_NAME: &str = "::intrinsics::advice";

pub const INTRINSICS_MODULE_NAMES: [&str; 6] = [
    I32_INTRINSICS_MODULE_NAME,
    I64_INTRINSICS_MODULE_NAME,
    I128_INTRINSICS_MODULE_NAME,
    MEM_INTRINSICS_MODULE_NAME,
    CRYPTO_INTRINSICS_MODULE_NAME,
    ADVICE_INTRINSICS_MODULE_NAME,
];

const I32_INTRINSICS: &str =
    include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/i32.masm"));
const I64_INTRINSICS: &str =
    include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/i64.masm"));
const MEM_INTRINSICS: &str =
    include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/mem.masm"));
const ADVICE_INTRINSICS: &str =
    include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/advice.masm"));

/// This is a mapping of intrinsics module name to the raw MASM source for that module
const INTRINSICS: [(&str, &str, &str); 4] = [
    (
        I32_INTRINSICS_MODULE_NAME,
        I32_INTRINSICS,
        concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/i32.masm"),
    ),
    (
        I64_INTRINSICS_MODULE_NAME,
        I64_INTRINSICS,
        concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/i64.masm"),
    ),
    (
        MEM_INTRINSICS_MODULE_NAME,
        MEM_INTRINSICS,
        concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/mem.masm"),
    ),
    (
        ADVICE_INTRINSICS_MODULE_NAME,
        ADVICE_INTRINSICS,
        concat!(env!("CARGO_MANIFEST_DIR"), "/intrinsics/advice.masm"),
    ),
];

/// This helper loads the named module from the set of intrinsics modules defined in this crate.
///
/// Expects the fully-qualified name to be given, e.g. `::intrinsics::mem`
pub fn load<N: AsRef<str>>(name: N, source_manager: Arc<dyn SourceManager>) -> Option<Box<Module>> {
    let name = name.as_ref();
    let (name, source, filename) = INTRINSICS.iter().copied().find(|(n, ..)| *n == name)?;
    let filename = Uri::new(filename);
    let source_file = source_manager.load(SourceLanguage::Masm, filename, source.to_string());
    let path = LibraryPath::new(name).expect("invalid module name");
    match Module::parse(path, ModuleKind::Library, source_file.clone(), source_manager) {
        Ok(module) => Some(module),
        Err(err) => {
            let err = PrintDiagnostic::new(err);
            panic!("failed to parse intrinsic module: {err}");
        }
    }
}