github/Sunscreen
1
1
Fork 0
mirror of https://github.com/Sunscreen-tech/Sunscreen.git synced 2026-01-10 06:08:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

Allow insecure params for tests (#344)

Browse Source
This commit is contained in:
Sam Tay
2024-02-11 16:03:00 -05:00
committed by GitHub
parent ae42ab7199
commit e9125df70d
11 changed files with 85 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
logproof/Cargo.toml
View File

@@ -24,6 +24,7 @@ thiserror = { workspace = true }
bincode = { workspace = true }
criterion = { workspace = true }
once_cell = { workspace = true }
seal_fhe = { workspace = true, features = ["insecure-params"] }
[features]
default = []

9
logproof/src/bfv_statement.rs
View File

@@ -768,16 +768,16 @@ mod tests {
impl BFVTestContext {
fn new() -> Self {
let plain_modulus = PlainModulus::raw(1153).unwrap();
let plain_modulus = PlainModulus::raw(32).unwrap();
let coeff_modulus =
CoefficientModulus::bfv_default(1024, SecurityLevel::TC128).unwrap();
let params = BfvEncryptionParametersBuilder::new()
.set_poly_modulus_degree(1024)
.set_poly_modulus_degree(64)
.set_coefficient_modulus(coeff_modulus)
.set_plain_modulus(plain_modulus)
.build()
.unwrap();
let ctx = Context::new(&params, false, SecurityLevel::TC128).unwrap();
let ctx = Context::new_insecure(&params, false).unwrap();
let gen = KeyGenerator::new(&ctx).unwrap();
let public_key = gen.create_public_key();
let secret_key = gen.secret_key();
@@ -902,8 +902,9 @@ mod tests {
let mut rng = rand::thread_rng();
let mut pt = Plaintext::new().unwrap();
let modulus = self.params.plain_modulus();
let len = self.params.get_poly_modulus_degree() as usize;
let size = rng.gen_range(0..100);
let size = rng.gen_range(0..len);
pt.resize(size);
for i in 0..size {

22
logproof/src/rings.rs
View File

@@ -47,6 +47,16 @@ pub type ZqRistretto = Zq<4, BarrettBackend<4, RistrettoConfig>>;
num_limbs = 3
)]
pub struct SealQ128_8192 {}
impl SealQ128_8192 {
/// The SEAL modulus chain
pub const Q: &'static [u64] = &[
0x7fffffd8001,
0x7fffffc8001,
0xfffffffc001,
0xffffff6c001,
0xfffffebc001,
];
}
/**
* The configuration type for q modulus SEAL BFV uses with 128-bit security
@@ -65,6 +75,10 @@ pub struct SealQ128_8192 {}
#[derive(BarrettConfig)]
#[barrett_config(modulus = "4722344527977019809793", num_limbs = 2)]
pub struct SealQ128_4096 {}
impl SealQ128_4096 {
/// The SEAL modulus chain
pub const Q: &'static [u64] = &[0xffffee001, 0xffffc4001, 0x1ffffe0001];
}
/**
* The configuration type for q modulus SEAL BFV uses with 128-bit security
@@ -83,6 +97,10 @@ pub struct SealQ128_4096 {}
#[derive(BarrettConfig)]
#[barrett_config(modulus = "18014398492704769", num_limbs = 1)]
pub struct SealQ128_2048 {}
impl SealQ128_2048 {
/// The SEAL modulus chain
pub const Q: &'static [u64] = &[0x3fffffff000001];
}
/**
* The configuration type for q modulus SEAL BFV uses with 128-bit security
@@ -101,6 +119,10 @@ pub struct SealQ128_2048 {}
#[derive(BarrettConfig)]
#[barrett_config(modulus = "132120577", num_limbs = 1)]
pub struct SealQ128_1024 {}
impl SealQ128_1024 {
/// The SEAL modulus chain
pub const Q: &'static [u64] = &[0x7e00001];
}
#[allow(unused)]
/**

1
seal_fhe/Cargo.toml
View File

@@ -37,3 +37,4 @@ serde_json = { workspace = true }
hexl = []
transparent-ciphertexts = []
deterministic = []
insecure-params = []

20
seal_fhe/src/context.rs
View File

@@ -79,6 +79,26 @@ impl Context {
Ok(Context { handle })
}
/**
* Creates an instance of SEALContext and performs several pre-computations
* on the given EncryptionParameters. This function explicitly allows insecure parameters,
* and is only for testing!
*
* * `params` - The encryption parameters.
* * `expand_mod_chain` - Determines whether the modulus switching chain
* should be created.
*/
#[cfg(feature = "insecure-params")]
pub fn new_insecure(params: &EncryptionParameters, expand_mod_chain: bool) -> Result<Self> {
let mut handle: *mut c_void = null_mut();
convert_seal_error(unsafe {
bindgen::SEALContext_Create(params.get_handle(), expand_mod_chain, 0, &mut handle)
})?;
Ok(Context { handle })
}
/**
* Returns handle to the underlying SEAL object.
*/

1
sunscreen/Cargo.toml
View File

@@ -58,6 +58,7 @@ proptest = { workspace = true }
rand = { workspace = true }
sunscreen_zkp_backend = { workspace = true, features = ["bulletproofs"] }
sunscreen_compiler_common = { workspace = true }
sunscreen_runtime = { workspace = true, features = ["insecure-params"] }
serde_json = { workspace = true }
[features]

29
sunscreen/tests/linked.rs
View File

@@ -1,6 +1,7 @@
#[cfg(feature = "linkedproofs")]
mod linked_tests {
use lazy_static::lazy_static;
use logproof::rings::SealQ128_1024;
use num::Rational64;
use sunscreen::types::bfv::{Rational, Signed, Unsigned64};
use sunscreen::types::zkp::{AsFieldElement, BfvRational, BfvSigned, BulletproofsField};
@@ -15,10 +16,10 @@ mod linked_tests {
use sunscreen_zkp_backend::bulletproofs::BulletproofsBackend;
lazy_static! {
static ref SMALL_PARAMS: Params = Params {
lattice_dimension: 1024,
coeff_modulus: vec![0x7e00001],
plain_modulus: 4_096,
static ref TEST_PARAMS: Params = Params {
lattice_dimension: 128,
coeff_modulus: SealQ128_1024::Q.to_vec(),
plain_modulus: 32,
scheme_type: SchemeType::Bfv,
security_level: sunscreen::SecurityLevel::TC128,
};
@@ -45,7 +46,7 @@ mod linked_tests {
fn test_valid_transaction_example() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(valid_transaction)
.compile()
@@ -90,7 +91,7 @@ mod linked_tests {
fn test_invalid_transaction_example() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(valid_transaction)
.compile()
@@ -127,7 +128,7 @@ mod linked_tests {
fn test_signed_encoding() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(is_eq_signed)
.compile()
@@ -178,7 +179,7 @@ mod linked_tests {
fn test_rational_encoding() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(is_eq_rational)
.compile()
@@ -221,7 +222,7 @@ mod linked_tests {
fn can_compare_signed() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(compare_signed)
.compile()
@@ -270,7 +271,7 @@ mod linked_tests {
fn can_compare_rationals() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(compare_rational)
.compile()
@@ -326,7 +327,7 @@ mod linked_tests {
// proves equivalence of pt x, pt y, and field elem z within ZKP
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(is_eq_3)
.compile()
@@ -385,7 +386,7 @@ mod linked_tests {
let is_eq_zkp = app.get_zkp_program(is_eq_signed).unwrap();
// but use runtime with modulus 4096
let rt = FheZkpRuntime::new(&SMALL_PARAMS, &BulletproofsBackend::new()).unwrap();
let rt = FheZkpRuntime::new(&TEST_PARAMS, &BulletproofsBackend::new()).unwrap();
let mut proof_builder = LogProofBuilder::new(&rt);
let res = proof_builder.zkp_program(is_eq_zkp);
@@ -400,7 +401,7 @@ mod linked_tests {
fn test_case(num_linked_inputs: usize, num_private_inputs: usize) {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(is_eq_3)
.compile()
@@ -436,7 +437,7 @@ mod linked_tests {
fn throws_linked_arg_type_mismatch() {
let app = Compiler::new()
.fhe_program(doggie)
.with_params(&SMALL_PARAMS)
.with_params(&TEST_PARAMS)
.zkp_backend::<BulletproofsBackend>()
.zkp_program(compare_signed)
.compile()

19
sunscreen/tests/sdlp.rs
View File

@@ -1,16 +1,17 @@
#[cfg(feature = "linkedproofs")]
mod sdlp_tests {
use lazy_static::lazy_static;
use sunscreen::types::bfv::Signed;
use logproof::rings::SealQ128_1024;
use sunscreen::types::bfv::{Signed, Unsigned64};
use sunscreen_fhe_program::SchemeType;
use sunscreen_runtime::{FheRuntime, LogProofBuilder, Params};
lazy_static! {
static ref SMALL_PARAMS: Params = Params {
lattice_dimension: 1024,
coeff_modulus: vec![0x7e00001],
plain_modulus: 4_096,
static ref TEST_PARAMS: Params = Params {
lattice_dimension: 128,
coeff_modulus: SealQ128_1024::Q.to_vec(),
plain_modulus: 32,
scheme_type: SchemeType::Bfv,
security_level: sunscreen::SecurityLevel::TC128,
};
@@ -18,7 +19,7 @@ mod sdlp_tests {
#[test]
fn prove_one_asymmetric_statement() {
let rt = FheRuntime::new(&SMALL_PARAMS).unwrap();
let rt = FheRuntime::new(&TEST_PARAMS).unwrap();
let (public_key, _secret_key) = rt.generate_keys().unwrap();
let mut logproof_builder = LogProofBuilder::new(&rt);
@@ -32,12 +33,12 @@ mod sdlp_tests {
#[test]
fn prove_one_symmetric_statement() {
let rt = FheRuntime::new(&SMALL_PARAMS).unwrap();
let rt = FheRuntime::new(&TEST_PARAMS).unwrap();
let (_public_key, private_key) = rt.generate_keys().unwrap();
let mut logproof_builder = LogProofBuilder::new(&rt);
let _ct = logproof_builder
.encrypt_symmetric(&Signed::from(3), &private_key)
.encrypt_symmetric(&Unsigned64::from(3), &private_key)
.unwrap();
let sdlp = logproof_builder.build_logproof().unwrap();
@@ -46,7 +47,7 @@ mod sdlp_tests {
#[test]
fn prove_linked_statements() {
let rt = FheRuntime::new(&SMALL_PARAMS).unwrap();
let rt = FheRuntime::new(&TEST_PARAMS).unwrap();
let (public_key, private_key) = rt.generate_keys().unwrap();
let mut logproof_builder = LogProofBuilder::new(&rt);

1
sunscreen_runtime/Cargo.toml
View File

@@ -52,3 +52,4 @@ linkedproofs = [
"dep:paste",
]
deterministic = ["seal_fhe/deterministic"]
insecure-params = ["seal_fhe/insecure-params"]

11
sunscreen_runtime/src/builder.rs
View File

@@ -219,7 +219,6 @@ mod linked {
rings::{SealQ128_1024, SealQ128_2048, SealQ128_4096, SealQ128_8192},
Bounds, LogProofProverKnowledge,
};
use seal_fhe::SecurityLevel;
use sunscreen_compiler_common::{Type, TypeName};
use sunscreen_math::ring::{BarrettBackend, BarrettConfig, Zq};
use sunscreen_zkp_backend::{
@@ -594,17 +593,17 @@ mod linked {
fn build_sdlp_pk(&self) -> Result<SealSdlpProverKnowledge> {
let params = self.runtime.params();
match (params.lattice_dimension, params.security_level) {
(1024, SecurityLevel::TC128) => Ok(SealSdlpProverKnowledge::from(
match &params.coeff_modulus[..] {
SealQ128_1024::Q => Ok(SealSdlpProverKnowledge::from(
self.build_sdlp_pk_generic::<1, SealQ128_1024>()?,
)),
(2048, SecurityLevel::TC128) => Ok(SealSdlpProverKnowledge::from(
SealQ128_2048::Q => Ok(SealSdlpProverKnowledge::from(
self.build_sdlp_pk_generic::<1, SealQ128_2048>()?,
)),
(4096, SecurityLevel::TC128) => Ok(SealSdlpProverKnowledge::from(
SealQ128_4096::Q => Ok(SealSdlpProverKnowledge::from(
self.build_sdlp_pk_generic::<2, SealQ128_4096>()?,
)),
(8192, SecurityLevel::TC128) => Ok(SealSdlpProverKnowledge::from(
SealQ128_8192::Q => Ok(SealSdlpProverKnowledge::from(
self.build_sdlp_pk_generic::<3, SealQ128_8192>()?,
)),
_ => Err(BuilderError::UnsupportedParameters(Box::new(params.clone())).into()),

4
sunscreen_runtime/src/runtime.rs
View File

@@ -840,6 +840,10 @@ impl GenericRuntime<(), ()> {
)
.build()?;
#[cfg(feature = "insecure-params")]
let context = SealContext::new_insecure(&bfv_params, true)?;
#[cfg(not(feature = "insecure-params"))]
let context = SealContext::new(&bfv_params, true, params.security_level)?;
Ok(FheRuntimeData {