sunscreen 0.8.1

A Fully Homomorphic Encryption (FHE) compiler supporting the Brakerski/Fan-Vercauteren (BFV) scheme.
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

use sunscreen::{
    types::{bfv::Signed, Cipher},
    *,
};

#[test]
fn compiling_multiple_programs_yields_same_params() {
    #[fhe_program(scheme = "bfv")]
    fn add(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a + b
    }

    #[fhe_program(scheme = "bfv")]
    fn mul(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a * b
    }

    let app = Compiler::new()
        .fhe_program(add)
        .fhe_program(mul)
        .compile()
        .unwrap();

    assert_eq!(
        *app.params(),
        app.get_fhe_program(add).unwrap().metadata.params
    );
    assert_eq!(
        *app.params(),
        app.get_fhe_program(mul).unwrap().metadata.params
    );
}

#[test]
fn can_reference_program_strongly_or_stringly() {
    #[fhe_program(scheme = "bfv")]
    fn add(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a + b
    }

    #[fhe_program(scheme = "bfv")]
    fn mul(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a * b
    }

    let app = Compiler::new()
        .fhe_program(add)
        .fhe_program(mul)
        .compile()
        .unwrap();

    assert_eq!(mul.name(), "mul");
    assert_eq!(add.name(), "add");

    assert!(app.get_fhe_program(mul).is_some());
    assert!(app.get_fhe_program("mul").is_some());
    assert!(app.get_fhe_program(add).is_some());
    assert!(app.get_fhe_program("add").is_some());
}

#[test]
fn get_programs_iterates_every_program() {
    #[fhe_program(scheme = "bfv")]
    fn add(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a + b
    }

    #[fhe_program(scheme = "bfv")]
    fn mul(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
        a * b
    }

    let app = Compiler::new()
        .fhe_program(add)
        .fhe_program(mul)
        .compile()
        .unwrap();

    assert_eq!(app.get_fhe_programs().count(), 2);
    assert!(app.get_fhe_programs().any(|(k, _)| k == "mul"));
    assert!(app.get_fhe_programs().any(|(k, _)| k == "add"));
}

#[test]
fn cant_compile_multiple_programs_with_same_name() {
    mod foo {
        use super::*;

        #[fhe_program(scheme = "bfv")]
        pub fn add(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
            a + b
        }
    }

    mod bar {
        use super::*;

        #[fhe_program(scheme = "bfv")]
        pub fn add(a: Cipher<Signed>, b: Cipher<Signed>) -> Cipher<Signed> {
            a + b
        }
    }

    let result = Compiler::new()
        .fhe_program(foo::add)
        .fhe_program(bar::add)
        .compile();

    match result {
        Err(Error::NameCollision) => {}
        _ => {
            panic!("Expected Error::NameCollision")
        }
    };
}