github/Sunscreen
1
1
Fork 0
mirror of https://github.com/Sunscreen-tech/Sunscreen.git synced 2026-04-19 03:00:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

fmt

Browse Source
This commit is contained in:
Matthew Liu
2023-07-21 16:18:21 -07:00
parent ac068c6a59
commit 93aa993929
4 changed files with 95 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sunscreen_compiler_common/src/context.rs
View File

@@ -11,7 +11,7 @@ use serde::{Deserialize, Serialize};
use crate::{Operation, Render};
#[cfg(feature = "debugger")]
use crate::lookup::{StackFrameLookup, GroupLookup};
use crate::lookup::{GroupLookup, StackFrameLookup};
/**
* Stores debug information about groups and stack traces.
@@ -49,7 +49,7 @@ impl DebugData {
stack_lookup: StackFrameLookup::new(),
group_lookup: GroupLookup::new(),
group_counter: 0,
stack_counter: 0
stack_counter: 0,
}
}
}
@@ -345,7 +345,6 @@ where
let group_id = self.graph.metadata.group_counter;
let stack_id = self.graph.metadata.stack_counter;
// TOOD: figure out updates to group and stack id?
self.graph.add_node(NodeInfo {

18
sunscreen_runtime/src/debugger/server.rs
View File

@@ -1,8 +1,6 @@
use actix_web::{get, http::header, web, App, HttpResponse, HttpServer, Responder};
use seal_fhe::{
BfvEncryptionParametersBuilder, CoefficientModulus, Context, Decryptor
};
use seal_fhe::{BfvEncryptionParametersBuilder, CoefficientModulus, Context, Decryptor};
use semver::Version;
use std::sync::OnceLock;
@@ -174,7 +172,7 @@ pub async fn get_fhe_node_data(
.coeff_modulus
.iter()
.map(|&num| num.leading_zeros() as i32)
.collect();
.collect();
// Decrypt inner ciphertext
let encryption_params_builder =
BfvEncryptionParametersBuilder::new()
@@ -191,8 +189,14 @@ pub async fn get_fhe_node_data(
);
let encryption_params = encryption_params_builder.build().unwrap();
println!("encryption params");
println!("poly mod degree: {:?}", encryption_params.get_poly_modulus_degree());
println!("coeff mod degree: {:?}", encryption_params.get_coefficient_modulus());
println!(
"poly mod degree: {:?}",
encryption_params.get_poly_modulus_degree()
);
println!(
"coeff mod degree: {:?}",
encryption_params.get_coefficient_modulus()
);
println!("plain mod: {:?}", encryption_params.get_plain_modulus());
println!("scheme: {:?}", encryption_params.get_scheme());
@@ -281,4 +285,4 @@ pub async fn get_fhe_node_data(
} else {
Ok(HttpResponse::NotFound().body(format!("Session {} not found", session)))
}
}
}

4
sunscreen_runtime/src/debugger/sessions.rs
View File

@@ -29,13 +29,13 @@ impl From<BfvSession> for Session {
impl Session {
pub fn unwrap_bfv_session(&self) -> &BfvSession {
match self {
Self::BfvSession(s) => s
Self::BfvSession(s) => s,
}
}
pub fn unwrap_bfv_session_mut(&mut self) -> &mut BfvSession {
match self {
Self::BfvSession(s) => s
Self::BfvSession(s) => s,
}
}
}

98
sunscreen_runtime/src/run.rs
View File

@@ -4,14 +4,14 @@ use sunscreen_compiler_common::{GraphQuery, GraphQueryError};
use sunscreen_fhe_program::Operation;
use sunscreen_fhe_program::{FheProgram, FheProgramTrait, Literal, Operation::*};
#[cfg(feature = "debugger")]
use sunscreen_fhe_program::SchemeType::Bfv;
#[cfg(feature = "debugger")]
use sunscreen_fhe_program::SecurityLevel::TC128;
#[cfg(feature = "debugger")]
use crate::debugger::sessions::{get_sessions, BfvSession};
#[cfg(feature = "debugger")]
use crate::WithContext;
#[cfg(feature = "debugger")]
use sunscreen_fhe_program::SchemeType::Bfv;
#[cfg(feature = "debugger")]
use sunscreen_fhe_program::SecurityLevel::TC128;
use crossbeam::atomic::AtomicCell;
use petgraph::{stable_graph::NodeIndex, Direction};
@@ -233,10 +233,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
match &node.operation {
InputCiphertext { id } => {
set_data(&data, index, &inputs[*id], &session_name).unwrap_or_else(|_| panic!("Failed to set data for InputCiphertext {:?}", id));
set_data(&data, index, &inputs[*id], &session_name).unwrap_or_else(|_| {
panic!("Failed to set data for InputCiphertext {:?}", id)
});
}
InputPlaintext { id } => {
set_data(&data, index, &inputs[*id], &session_name).unwrap_or_else(|_| panic!("Failed to set data for InputPlaintext {:?}", id));
set_data(&data, index, &inputs[*id], &session_name).unwrap_or_else(|_| {
panic!("Failed to set data for InputPlaintext {:?}", id)
});
}
ShiftLeft => {
let (left, right) = query.get_binary_operands(index)?;
@@ -259,7 +263,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
.as_ref()
.ok_or(FheProgramRunFailure::MissingGaloisKeys)?,
)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for ShiftLeft, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for ShiftLeft, (left: {:?}, right: {:?}",
left, right
)
},
);
}
ShiftRight => {
let (left, right) = query.get_binary_operands(index)?;
@@ -282,7 +293,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
.as_ref()
.ok_or(FheProgramRunFailure::MissingGaloisKeys)?,
)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for ShiftRight, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for ShiftRight, (left: {:?}, right: {:?}",
left, right
)
},
);
}
Add => {
let (left, right) = query.get_binary_operands(index)?;
@@ -292,7 +310,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.add(a, b)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for Add, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for Add, (left: {:?}, right: {:?}",
left, right
)
},
);
}
AddPlaintext => {
let (left, right) = query.get_binary_operands(index)?;
@@ -302,7 +327,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.add_plain(a, b)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for AddPlaintext, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for AddPlaintext, (left: {:?}, right: {:?}",
left, right
)
},
);
}
Multiply => {
let (left, right) = query.get_binary_operands(index)?;
@@ -312,7 +344,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.multiply(a, b)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for Multiply, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for Multiply, (left: {:?}, right: {:?}",
left, right
)
},
);
}
MultiplyPlaintext => {
let (left, right) = query.get_binary_operands(index)?;
@@ -335,7 +374,8 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let y = evaluator.rotate_columns(x, galois_keys)?;
set_data(&data, index, &Arc::new(y.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for SwapRows {:?}", input));
set_data(&data, index, &Arc::new(y.into()), &session_name)
.unwrap_or_else(|_| panic!("Failed to set data for SwapRows {:?}", input));
}
Relinearize => {
let relin_keys = relin_keys
@@ -348,7 +388,9 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.relinearize(a, relin_keys)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for Relinearize {:?}", input));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| panic!("Failed to set data for Relinearize {:?}", input),
);
}
Negate => {
let x_id = query.get_unary_operand(index)?;
@@ -357,7 +399,8 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let y = evaluator.negate(x)?;
set_data(&data, index, &Arc::new(y.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for Negate {:?}", x_id));
set_data(&data, index, &Arc::new(y.into()), &session_name)
.unwrap_or_else(|_| panic!("Failed to set data for Negate {:?}", x_id));
}
Sub => {
let (left, right) = query.get_binary_operands(index)?;
@@ -367,7 +410,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.sub(a, b)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for Sub, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for Sub, (left: {:?}, right: {:?}",
left, right
)
},
);
}
SubPlaintext => {
let (left, right) = query.get_binary_operands(index)?;
@@ -377,7 +427,14 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let c = evaluator.sub_plain(a, b)?;
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(|_| panic!("Failed to set data for SubPlaintext, (left: {:?}, right: {:?}", left, right));
set_data(&data, index, &Arc::new(c.into()), &session_name).unwrap_or_else(
|_| {
panic!(
"Failed to set data for SubPlaintext, (left: {:?}, right: {:?}",
left, right
)
},
);
}
Operation::Literal { val: x } => {
if let Literal::Plaintext(p) = x {
@@ -396,7 +453,10 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
index,
&Arc::new(p[0].data.clone().into()),
&session_name,
).unwrap_or_else(|_| panic!("Failed to set data for Literal, plaintext {:?}", p));
)
.unwrap_or_else(|_| {
panic!("Failed to set data for Literal, plaintext {:?}", p)
});
}
};
}
@@ -406,7 +466,9 @@ pub unsafe fn run_program_unchecked<E: Evaluator + Sync + Send>(
let a = get_data(&data, input.index())?;
set_data(&data, index, &a.clone(), &session_name).unwrap_or_else(|_| panic!("Failed to set data for OutputCiphertext, input {:?}", input));
set_data(&data, index, &a.clone(), &session_name).unwrap_or_else(|_| {
panic!("Failed to set data for OutputCiphertext, input {:?}", input)
});
}
};