tfhe 1.6.0

TFHE-rs is a fully homomorphic encryption (FHE) library that implements Zama's variant of TFHE.
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
116
117
118
119
120
121
122
123
124
125
126
127
128

use crate::high_level_api::integers::signed::FheIntId;
use crate::high_level_api::integers::unsigned::FheUintId;
use crate::high_level_api::re_randomization::ReRandomizationMetadata;
use crate::integer::ciphertext::{DataKind, Expandable};
use crate::integer::BooleanBlock;
use crate::shortint::Ciphertext;
use crate::{FheBool, FheInt, FheUint, Tag};

fn num_bits_of_blocks(blocks: &[Ciphertext]) -> u32 {
    blocks
        .iter()
        .map(|block| block.message_modulus.0.ilog2())
        .sum::<u32>()
}

impl<Id: FheUintId> Expandable for FheUint<Id> {
    fn from_expanded_blocks(blocks: Vec<Ciphertext>, kind: DataKind) -> crate::Result<Self> {
        match kind {
            DataKind::Unsigned(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                if stored_num_bits == Id::num_bits() {
                    // The expander will be responsible for setting the correct tag and metadata
                    Ok(Self::new(
                        crate::integer::RadixCiphertext::from(blocks),
                        Tag::default(),
                        ReRandomizationMetadata::default(),
                    ))
                } else {
                    Err(crate::error!(
                        "Tried to expand a FheUint{} while a FheUint{} is stored in this slot",
                        Id::num_bits(),
                        stored_num_bits
                    ))
                }
            }
            DataKind::Signed(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                Err(crate::error!(
                    "Tried to expand a FheUint{} while a FheInt{} is stored in this slot",
                    Id::num_bits(),
                    stored_num_bits
                ))
            }
            DataKind::Boolean => Err(crate::error!(
                "Tried to expand a FheUint{} while a FheBool is stored in this slot",
                Id::num_bits(),
            )),
            DataKind::String { .. } => Err(crate::error!(
                "Tried to expand a FheUint{}  while a string is stored in this slot",
                Id::num_bits()
            )),
        }
    }
}

impl<Id: FheIntId> Expandable for FheInt<Id> {
    fn from_expanded_blocks(blocks: Vec<Ciphertext>, kind: DataKind) -> crate::Result<Self> {
        match kind {
            DataKind::Unsigned(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                Err(crate::error!(
                    "Tried to expand a FheInt{} while a FheUint{} is stored in this slot",
                    Id::num_bits(),
                    stored_num_bits
                ))
            }
            DataKind::Signed(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                if stored_num_bits == Id::num_bits() {
                    // The expander will be responsible for setting the correct tag
                    Ok(Self::new(
                        crate::integer::SignedRadixCiphertext::from(blocks),
                        Tag::default(),
                        ReRandomizationMetadata::default(),
                    ))
                } else {
                    Err(crate::error!(
                        "Tried to expand a FheInt{} while a FheInt{} is stored in this slot",
                        Id::num_bits(),
                        stored_num_bits
                    ))
                }
            }
            DataKind::Boolean => Err(crate::error!(
                "Tried to expand a FheInt{} while a FheBool is stored in this slot",
                Id::num_bits(),
            )),
            DataKind::String { .. } => Err(crate::error!(
                "Tried to expand a FheInt{}  while a string is stored in this slot",
                Id::num_bits()
            )),
        }
    }
}

impl Expandable for FheBool {
    fn from_expanded_blocks(blocks: Vec<Ciphertext>, kind: DataKind) -> crate::Result<Self> {
        match kind {
            DataKind::Unsigned(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                Err(crate::error!(
                    "Tried to expand a FheBool while a FheUint{stored_num_bits} is stored in this slot",
                ))
            }
            DataKind::Signed(_) => {
                let stored_num_bits = num_bits_of_blocks(&blocks) as usize;
                Err(crate::error!(
                    "Tried to expand a FheBool while a FheInt{stored_num_bits} is stored in this slot",
                ))
            }
            DataKind::Boolean => {
                let mut boolean_block = BooleanBlock::new_unchecked(blocks[0].clone());
                // We know the value is a boolean one (via the data kind)
                boolean_block.0.degree = crate::shortint::ciphertext::Degree::new(1);

                // The expander will be responsible for setting the correct tag and metadata
                Ok(Self::new(
                    boolean_block,
                    Tag::default(),
                    ReRandomizationMetadata::default(),
                ))
            }
            DataKind::String { .. } => Err(crate::Error::new(
                "Tried to expand a FheBool while a string is stored in this slot".to_string(),
            )),
        }
    }
}