mrubyedge 1.0.0

mruby/edge is yet another mruby that is specialized for running on WASM
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use std::rc::Rc;

use crate::{yamrb::{helpers::mrb_define_cmethod, value::RObject, vm::VM}, Error};

use super::array::mrb_array_push;

pub(crate) fn initialize_string(vm: &mut VM) {
    let string_class = vm.define_standard_class("String");

    mrb_define_cmethod(vm, string_class.clone(), "unpack", Box::new(mrb_string_unpack));
}

fn mrb_string_unpack(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let this = vm.getself();
    let value: Vec<u8> = this.as_ref().try_into()?;
    let format: Vec<u8> = args[0].as_ref().try_into()?;
    let mut cursor: usize = 0;
    let result = Rc::new(RObject::array(Vec::new()));

    for c in format.iter() {
        // We just support Ruby#pack's format of:
        //   - Q: 64-bit unsigned (unsigned long long)
        //   - q: 64-bit signed (signed long long)
        //   - L: 32-bit unsigned (unsigned long)
        //   - l: 32-bit signed (signed long)
        //   - I: 32-bit unsigned (unsigned int)
        //   - i: 32-bit signed (signed int)
        //   - S: 16-bit unsigned (unsigned short)
        //   - s: 16-bit signed (signed short)
        //   - C: 8-bit unsigned (unsigned char)
        //   - c: 8-bit signed (signed char)
        let value = match c {
            b'Q' => {
                let value = u64::from_le_bytes(value[cursor..cursor+8].try_into().unwrap());
                cursor += 8;
                value as i64
            }
            b'q' => {
                let value = i64::from_le_bytes(value[cursor..cursor+8].try_into().unwrap());
                cursor += 8;
                value as i64
            }
            b'L' | b'I' => {
                let value = u32::from_le_bytes(value[cursor..cursor+4].try_into().unwrap());
                cursor += 4;
                value as i64
            }
            b'l' | b'i' => {
                let value = i32::from_le_bytes(value[cursor..cursor+4].try_into().unwrap());
                cursor += 4;
                value as i64
            }
            b'S' => {
                let value = u16::from_le_bytes(value[cursor..cursor+2].try_into().unwrap());
                cursor += 2;
                value as i64
            }
            b's' => {
                let value = i16::from_le_bytes(value[cursor..cursor+2].try_into().unwrap());
                cursor += 2;
                value as i64
            }
            b'C' => {
                let value = i8::from_le_bytes(value[cursor..cursor+1].try_into().unwrap());
                cursor += 1;
                value as i64
            }
            b'c' => {
                let value = u8::from_le_bytes(value[cursor..cursor+1].try_into().unwrap());
                cursor += 1;
                value as i64
            }
            b' ' => {
                // ignore space
                continue;
            }
            _ => {
                return Err(Error::RuntimeError("Unsupported format".to_string()));
            }
        };
        mrb_array_push(result.clone(), &[Rc::new(RObject::integer(value as i64))])?;
    }

    Ok(result)
}

#[test]
fn test_mrb_string_unpack() {
    use crate::yamrb::*;

    let mut vm = VM::empty();
    prelude::prelude(&mut vm);

    let data = Rc::new(RObject::string_from_vec(
        vec![
            0x01,
            0x02, 0x03,
            0x04, 0x05, 0x06, 0x07,
            0x04, 0x04, 0x03, 0x03, 0x02, 0x02, 0x00, 0x00,
        ],
    ));
    let format = Rc::new(RObject::string(
        "c s l q".to_string(),
    ));
    let arg = vec![format];

    let ret = helpers::mrb_funcall(&mut vm, Some(data), "unpack", &arg).expect("unpack failed");
    
    let answers = vec![
        0x01,
        0x02 | 0x03 << 8,
        0x04 | 0x05 << 8 | 0x06 << 16 | 0x07 << 24,
        0x04 | 0x04 << 8 | 0x03 << 16 | 0x03 << 24 | 0x02 << 32 | 0x02 << 40 | 0x00 << 48 | 0x00 << 56,
    ];

    for (i, expected) in answers.iter().enumerate() {
        let args = vec![Rc::new(RObject::integer(i as i64))];
        let value = prelude::array::mrb_array_get_index(ret.clone(), &args).expect("getting index failed");
        let value: i64 = value.as_ref().try_into().expect("value is not integer");
        assert_eq!(value, *expected);
    }
}