stoffel-vm 0.1.0

Register-based virtual machine runtime and MPC execution engine for Stoffel programs
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

//! AVSS Keygen Program Generator
//!
//! Generates a `.stflb` bytecode program that performs distributed key generation
//! using the AVSS backend. The program:
//! 1. Calls `Share.random()` to run cooperative DKG (no party knows the secret)
//! 2. Calls `Share.get_commitment(share, 0)` to extract the public key
//! 3. Returns the public key bytes (all parties produce the same result)

use stoffel_vm_types::{
    compiled_binary::utils::{
        from_vm_functions, load_from_file, save_to_file, try_to_vm_functions,
    },
    core_types::Value,
    functions::VMFunction,
    instructions::Instruction,
};

/// Build the AVSS keygen program as bytecode instructions.
pub fn build_avss_keygen_program() -> Vec<Instruction> {
    vec![
        // r0 = Share.random()  → jointly-random share (DKG)
        Instruction::CALL("Share.random".to_string()),
        // r1 = share object
        Instruction::MOV(1, 0),
        // r2 = 0 (commitment index 0 = public key)
        Instruction::LDI(2, Value::I64(0)),
        // Share.get_commitment(r1, r2) → byte array
        Instruction::PUSHARG(1),
        Instruction::PUSHARG(2),
        Instruction::CALL("Share.get_commitment".to_string()),
        // return the public key byte array
        Instruction::RET(0),
    ]
}

#[test]
fn avss_keygen_bytecode_round_trips_without_mutating_fixtures() {
    let func = VMFunction::new(
        "main".to_string(),
        vec![],
        vec![],
        None,
        4,
        build_avss_keygen_program(),
        Default::default(),
    );

    let binary = from_vm_functions(&[func]);

    let dir = tempfile::tempdir().expect("temp dir");
    let path = dir.path().join("avss_keygen.stflb");
    save_to_file(&binary, &path).expect("failed to save avss_keygen bytecode");

    let loaded = load_from_file(&path).expect("failed to reload");
    let functions =
        try_to_vm_functions(&loaded).expect("generated AVSS keygen bytecode should be executable");
    assert_eq!(functions.len(), 1);
    assert_eq!(functions[0].name(), "main");
    assert_eq!(functions[0].instructions().len(), 7);
}