midenc-dialect-ub 0.8.1

Miden IR Undefined Behavior Dialect
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

use midenc_hir::{
    AttrPrinter, Felt, Immediate, Type, attributes::AttrParser, derive::DialectAttribute,
    formatter, print::AsmPrinter,
};

use crate::UndefinedBehaviorDialect;

/// Represents the constant value of the 'hir.poison' operation
#[derive(DialectAttribute)]
#[attribute(
    dialect = UndefinedBehaviorDialect,
    remote = "Type",
    default = "poison_value",
    implements(AttrPrinter)
)]
#[allow(unused)]
struct Poison;

const fn poison_value() -> Type {
    Type::Unknown
}

impl PoisonAttr {
    pub fn as_immediate(&self) -> Option<Immediate> {
        Some(match &self.value {
            Type::I1 => Immediate::I1(false),
            Type::U8 => Immediate::U8(0xde),
            Type::I8 => Immediate::I8(0xdeu8 as i8),
            Type::U16 => Immediate::U16(0xdead),
            Type::I16 => Immediate::I16(0xdeadu16 as i16),
            Type::U32 => Immediate::U32(0xdeadc0de),
            Type::I32 => Immediate::I32(0xdeadc0deu32 as i32),
            Type::U64 => Immediate::U64(0xdeadc0dedeadc0de),
            Type::I64 => Immediate::I64(0xdeadc0dedeadc0deu64 as i64),
            Type::Felt => Immediate::Felt(Felt::new(0xdeadc0de)),
            Type::U128 => Immediate::U128(0xdeadc0dedeadc0dedeadc0dedeadc0de),
            Type::I128 => Immediate::I128(0xdeadc0dedeadc0dedeadc0dedeadc0deu128 as i128),
            // We emit a pointer that can never refer to a valid object in memory
            Type::Ptr(_) => Immediate::U32(u32::MAX),
            _ty => return None,
        })
    }
}

impl formatter::PrettyPrint for PoisonAttr {
    fn render(&self) -> formatter::Document {
        use formatter::*;

        display(&self.value)
    }
}

impl AttrPrinter for PoisonAttr {
    fn print(&self, printer: &mut AsmPrinter<'_>) {
        printer.print_type(&self.value);
    }
}

impl AttrParser for PoisonAttr {
    fn parse(
        parser: &mut dyn midenc_hir::parse::Parser<'_>,
    ) -> midenc_hir::parse::ParseResult<midenc_hir::AttributeRef> {
        let ty = parser.parse_type()?;
        Ok(parser.context_rc().create_attribute::<PoisonAttr, _>(ty.into_inner()))
    }
}