sunscreen_runtime 0.8.1

This crate provides a runtime for performing various FHE operations within Sunscreen.
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

use crate::{
    Error, FheProgramInputTrait, InnerPlaintext, NumCiphertexts, Params, Plaintext, Result,
    TryFromPlaintext, TryIntoPlaintext, Type, TypeName, TypeNameInstance, WithContext,
};
use seal_fhe::Plaintext as SealPlaintext;

impl<T, const N: usize> TryIntoPlaintext for [T; N]
where
    T: TryIntoPlaintext,
    Self: TypeName,
{
    fn try_into_plaintext(&self, params: &Params) -> Result<Plaintext> {
        let element_plaintexts = self
            .iter()
            .map(|v| v.try_into_plaintext(params))
            .collect::<Result<Vec<Plaintext>>>()?
            .drain(0..)
            .flat_map(|p| match p.inner {
                InnerPlaintext::Seal(v) => v,
            })
            .collect::<Vec<WithContext<SealPlaintext>>>();

        Ok(Plaintext {
            inner: InnerPlaintext::Seal(element_plaintexts),
            data_type: Self::type_name(),
        })
    }
}

impl<T, const N: usize> TryFromPlaintext for [T; N]
where
    T: TryFromPlaintext + TypeName + NumCiphertexts,
    Self: TypeName + NumCiphertexts,
{
    fn try_from_plaintext(plaintext: &Plaintext, params: &Params) -> Result<Self> {
        let data = match &plaintext.inner {
            InnerPlaintext::Seal(p) => {
                if p.len() != Self::NUM_CIPHERTEXTS {
                    return Err(Error::MalformedPlaintext);
                }

                p.chunks(T::NUM_CIPHERTEXTS)
                    .map(|c| {
                        let p = Plaintext {
                            data_type: T::type_name(),
                            inner: InnerPlaintext::Seal(c.to_owned()),
                        };

                        T::try_from_plaintext(&p, params)
                    })
                    .collect::<Result<Vec<T>>>()?
            }
        };

        Ok(match data.try_into() {
            Ok(v) => v,
            _ => unreachable!(),
        })
    }
}

impl<T, const N: usize> TypeNameInstance for [T; N]
where
    T: TypeName,
{
    fn type_name_instance(&self) -> Type {
        Self::type_name()
    }
}

impl<T, const N: usize> FheProgramInputTrait for [T; N] where T: TypeName + TryIntoPlaintext {}

impl<T, const N: usize> NumCiphertexts for [T; N]
where
    T: NumCiphertexts,
{
    const NUM_CIPHERTEXTS: usize = T::NUM_CIPHERTEXTS * N;
}