smp-tee-runtime 0.1.0

Hardened minimal runtime for TEE-based federated aggregation
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

use smp_tee_runtime::{AggregationAlgorithm, ComputationParams, InMemoryTee, TeeGuard};

fn to_bytes(values: &[f32]) -> Vec<u8> {
    values.iter().flat_map(|v| v.to_le_bytes()).collect()
}

fn to_f32(bytes: &[u8]) -> Vec<f32> {
    bytes
        .chunks_exact(4)
        .map(|chunk| f32::from_le_bytes([chunk[0], chunk[1], chunk[2], chunk[3]]))
        .collect()
}

fn main() {
    let mut tee = InMemoryTee::default();
    tee.initialize().expect("TEE init failed");

    let p1 = tee.allocate_memory(8).expect("allocation failed");
    let p2 = tee.allocate_memory(8).expect("allocation failed");

    tee.write_data(p1, &to_bytes(&[1.0, 3.0]))
        .expect("write failed");
    tee.write_data(p2, &to_bytes(&[3.0, 5.0]))
        .expect("write failed");

    let result = tee
        .execute_computation(
            &[p1.cast_const(), p2.cast_const()],
            &ComputationParams {
                algorithm: AggregationAlgorithm::FederatedAveraging,
            },
        )
        .expect("computation failed");

    println!("federated average = {:?}", to_f32(&result));
}