stoffelcrypto 0.1.0

Asynchronous HoneyBadgerMPC protocols, preprocessing, and arithmetic for Stoffel.
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

use crate::ffi::c_bindings::{
    network::{GenericNetwork, NetworkOpaque},
    ByteSlice, UsizeSlice,
};
use std::{collections::HashMap, mem::ManuallyDrop, sync::Arc};
use stoffelmpc_network::fake_network::{FakeInnerNetwork, FakeNetwork, FakeNetworkConfig};
use stoffelnet::network_utils::ClientId;
use tokio::sync::mpsc::Receiver;

// struct that includes receivers of the FakeNetwork
pub struct FakeNetworkReceivers {
    pub node_receivers: Vec<Vec<Receiver<Vec<u8>>>>,
    pub client_receivers: HashMap<ClientId, Vec<Receiver<Vec<u8>>>>,
}

// opaque pointer for FakeNetworkReceivers
#[repr(C)]
pub struct FakeNetworkReceiversOpaque {
    _data: (),
    _marker: core::marker::PhantomData<(*mut u8, core::marker::PhantomPinned)>,
}

// create FakeNetwork
#[no_mangle]
pub extern "C" fn new_fake_network(
    n_nodes: usize,
    n_clients: Option<&UsizeSlice>,
    channel_buff_size: usize,
    returned_receivers: *mut *mut FakeNetworkReceiversOpaque,
) -> *mut NetworkOpaque {
    let config = FakeNetworkConfig::new(channel_buff_size);
    let n_clients = match n_clients {
        None => None,
        Some(u) => {
            let c_vec = unsafe { Vec::from_raw_parts(u.pointer, u.len, u.len) };
            let r_vec = c_vec.clone();
            // prevent rust from dropping the pointer from C
            std::mem::forget(c_vec);
            Some(r_vec)
        }
    };

    // build inner network once
    let (inner, node_receivers, client_receivers) =
        FakeInnerNetwork::new(n_nodes, n_clients, config);

    //  return receivers (now 2-D)
    let receivers = FakeNetworkReceivers {
        node_receivers,
        client_receivers,
    };
    unsafe {
        *returned_receivers = Box::into_raw(Box::new(receivers)) as *mut FakeNetworkReceiversOpaque;
    }
    // create network handle for node 0 by default
    // (C side can clone / create others later if needed)
    let network = FakeNetwork::new(0, inner);

    let network = GenericNetwork::FakeNetwork(Arc::new(network));

    Box::into_raw(Box::new(network)) as *mut NetworkOpaque
}

#[no_mangle]
pub extern "C" fn node_receiver_recv_sync(
    receivers: *mut FakeNetworkReceiversOpaque,
    to_node: usize,
    from_node: usize,
) -> ByteSlice {
    let receivers = unsafe { &mut *(receivers as *mut FakeNetworkReceivers) };
    let receiver = &mut receivers.node_receivers[to_node][from_node];
    let msg = tokio::runtime::Runtime::new()
        .unwrap()
        .block_on(receiver.recv());
    match msg {
        None => ByteSlice {
            pointer: std::ptr::null_mut(),
            len: 0,
        },
        Some(m) => {
            let mut m = ManuallyDrop::new(m);
            let slice = ByteSlice {
                pointer: m.as_mut_ptr(),
                len: m.len(),
            };
            slice
        }
    }
}

#[no_mangle]
pub extern "C" fn free_fake_network_receivers(receivers: *mut FakeNetworkReceiversOpaque) {
    if !receivers.is_null() {
        unsafe {
            // Reconstruct the Box and drop it to free memory
            let receivers = receivers as *mut FakeNetworkReceivers;
            drop(Box::from_raw(receivers));
        }
    }
}