Added SEAL library and FHE mode

.github/workflows/ci.yml

@@ -9,6 +9,7 @@ on:
 env:
   CARGO_TERM_COLOR: always
   ABY_SOURCE: "./../ABY"
+  SEAL_SOURCE: "./../SEAL"
 
 jobs:
   build:
@@ -31,7 +32,9 @@ jobs:
     - name: Cache third_party build
       uses: actions/cache@v2
       with:
-        path: ${ABY_SOURCE}/build
+        path: |
+          ${ABY_SOURCE}/build
+          ${SEAL_SOURCE}/build
         key: ${{ runner.os }}
     - name: Check
       run: python3 driver.py --check

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "cmake.configureOnOpen": false
+}

driver.py

@@ -23,6 +23,10 @@ def verify_path_empty(path) -> bool:
             if verify_path_empty(ABY_SOURCE):
                 subprocess.run(["git", "clone", "https://github.com/edwjchen/ABY.git", ABY_SOURCE])
                 subprocess.run(["./scripts/build_aby.zsh"])
+        if f == "seal":
+            if verify_path_empty(SEAL_SOURCE):
+                subprocess.run(["git", "clone", "https://github.com/Northrim/SEAL.git", SEAL_SOURCE])
+                subprocess.run(["./scripts/build_seal.zsh"])
 
     # install python requirements
     subprocess.run(["pip3", "install", "-r", "requirements.txt"])
@@ -76,6 +80,12 @@ def build(features):
         if "smt" in features and "zok" in features:
             subprocess.run(["./scripts/build_mpc_zokrates_test.zsh"], check=True)
         subprocess.run(["./scripts/build_aby.zsh"], check=True)
+    if "seal" in features:
+        if "c" in features:
+            subprocess.run(["./scripts/build_fhe_c_test.zsh"], check=True)
+        if "smt" in features and "zok" in features:
+            subprocess.run(["./scripts/build_fhe_zokrates_test.zsh"], check=True)
+        subprocess.run(["./scripts/build_seal.zsh"], check=True)
 
 def test(features):
     """
@@ -101,6 +111,8 @@ def test(features):
     if "zok" in features and "smt" in features:
         if "aby" in features:
             subprocess.run(["python3", "./scripts/aby_tests/zokrates_test_aby.py"], check=True)
+        if "seal" in features:
+            subprocess.run(["python3", "./scripts/seal_tests/zokrates_test_seal.py"], check=True)
         if "lp" in features:
             subprocess.run(["./scripts/test_zok_to_ilp.zsh"], check=True)
         if "r1cs" in features:
@@ -111,6 +123,8 @@ def test(features):
     if "c" in features:
         if "aby" in features:
             subprocess.run(["python3", "./scripts/aby_tests/c_test_aby.py"], check=True)
+        if "seal" in features:
+            subprocess.run(["python3", "./scripts/seal_tests/c_test_seal.py"], check=True)
 
 def benchmark(features):
     mode = load_mode()
@@ -165,6 +179,8 @@ def clean(features):
     print("cleaning!")
     if "aby" in features:
         subprocess.run(["./scripts/clean_aby.zsh"])
+    if "seal" in features:
+        subprocess.run(["./scripts/clean_seal.zsh"])
     subprocess.run(["rm", "-rf", "scripts/aby_tests/__pycache__"])
     subprocess.run(["rm", "-rf", "P", "V", "pi", "perf.data perf.data.old flamegraph.svg"])
 
@@ -210,7 +226,7 @@ def verify_feature(f):
     parser.add_argument("-m", "--mode", type=str, help="set `debug` or `release` mode")
     parser.add_argument("-A", "--all_features", action="store_true", help="set all features on")
     parser.add_argument("-L", "--list_features", action="store_true", help="print active features")
-    parser.add_argument("-F", "--features", nargs="+", help="set features on <aby, c, lp, r1cs, smt, zok>, reset features with -F none")
+    parser.add_argument("-F", "--features", nargs="+", help="set features on <aby, c, lp, r1cs, seal, smt, zok>, reset features with -F none")
     parser.add_argument("--benchmark", action="store_true", help="build benchmarks")
     parser.add_argument("extra", metavar="PASS_THROUGH_ARGS", nargs=argparse.REMAINDER, help="Extra arguments for --flamegraph. Prefix with --")
     args = parser.parse_args()

examples/C/fhe/unit_tests/arithmetic_tests/add.c

@@ -0,0 +1,3 @@
+int main(__attribute__((private(0))) int a, __attribute__((private(1))) int b) { 
+   return a + b;
+}

examples/C/fhe/unit_tests/arithmetic_tests/mult.c

@@ -0,0 +1,3 @@
+int main(__attribute__((private(0))) int a, __attribute__((private(1))) int b) { 
+   return a * b;
+}

examples/C/fhe/unit_tests/arithmetic_tests/mult_add_pub.c

@@ -0,0 +1,3 @@
+int main(__attribute__((private(0))) int a, __attribute__((private(1))) int b, __attribute__((public)) int v) { 
+   return a * b + v;
+}

examples/C/fhe/unit_tests/boolean_tests/boolean_and.c

@@ -0,0 +1,5 @@
+#include <stdbool.h>
+
+bool main(__attribute__((private(0))) bool a, __attribute__((private(1))) bool b) { 
+   return a && b;
+}

examples/C/fhe/unit_tests/boolean_tests/boolean_equals.c

@@ -0,0 +1,5 @@
+#include <stdbool.h>
+
+bool main(__attribute__((private(0))) bool a, __attribute__((private(1))) bool b) { 
+   return a == b;
+}

examples/C/fhe/unit_tests/boolean_tests/boolean_or.c

@@ -0,0 +1,5 @@
+#include <stdbool.h>
+
+bool main(__attribute__((private(0))) bool a, __attribute__((private(1))) bool b) { 
+   return a || b;
+}

examples/C/fhe/unit_tests/nary_arithmetic_tests/nary_arithmetic_add.c

@@ -0,0 +1,3 @@
+int main(__attribute__((private(0))) int a, __attribute__((private(1))) int b, __attribute__((private(1))) int c) { 
+  return a + b + c;
+}

examples/C/fhe/unit_tests/nary_boolean_tests/nary_boolean_and.c

@@ -0,0 +1,5 @@
+#include <stdbool.h>
+
+bool main(__attribute__((private(0))) bool a, __attribute__((private(1))) bool b, __attribute__((private(1))) bool c) { 
+  return a && b && c;
+}

examples/ZoKrates/fhe/unit_tests/arithmetic_tests/add.zok

@@ -0,0 +1,2 @@
+def main(private u32 a, private u32 b) -> u32:
+  return a + b

examples/ZoKrates/fhe/unit_tests/arithmetic_tests/mult.zok

@@ -0,0 +1,2 @@
+def main(private u32 a, private u32 b) -> u32:
+  return a * b

examples/ZoKrates/fhe/unit_tests/arithmetic_tests/mult_add_pub.zok

@@ -0,0 +1,3 @@
+
+def main(private u32 a, private u32 b, public u32 v) -> u32:
+  return a * b + v

examples/ZoKrates/fhe/unit_tests/boolean_tests/boolean_and.zok

@@ -0,0 +1,2 @@
+def main(private bool a, private bool b) -> bool:
+    return a && b

examples/ZoKrates/fhe/unit_tests/boolean_tests/boolean_equals.zok

@@ -0,0 +1,2 @@
+def main(private bool a, private bool b) -> bool:
+    return a == b

examples/ZoKrates/fhe/unit_tests/boolean_tests/boolean_or.zok

@@ -0,0 +1,2 @@
+def main(private bool a, private bool b) -> bool:
+    return a || b

examples/ZoKrates/fhe/unit_tests/nary_arithmetic_tests/nary_arithmetic_add.zok

@@ -0,0 +1,2 @@
+def main(private u32 a, private u32 b,  private u32 c) -> u32:
+  return a + b + c

examples/ZoKrates/fhe/unit_tests/nary_boolean_tests/nary_boolean_and.zok

@@ -0,0 +1,2 @@
+def main(private bool a, private bool b, private bool c) -> bool:
+  return a && b && c

examples/circ.rs

@@ -20,6 +20,7 @@ use circ::ir::{
     term::extras::Letified,
 };
 use circ::target::aby::trans::to_aby;
+use circ::target::fhe::trans::to_fhe;
 #[cfg(feature = "lp")]
 use circ::target::ilp::trans::to_ilp;
 #[cfg(feature = "r1cs")]
@@ -101,6 +102,7 @@ enum Backend {
     },
     Smt {},
     Ilp {},
+    Fhe {},
     Mpc {
         #[structopt(long, default_value = "hycc", name = "cost_model")]
         cost_model: String,
@@ -187,6 +189,7 @@ fn main() {
         Backend::Ilp { .. } => Mode::Opt,
         Backend::Mpc { .. } => Mode::Mpc(options.parties),
         Backend::Smt { .. } => Mode::Proof,
+        Backend::Fhe { .. } => Mode::Fhe,
     };
     let language = determine_language(&options.frontend.language, &options.path);
     let cs = match language {
@@ -246,6 +249,7 @@ fn main() {
             ],
             // vec![Opt::Sha, Opt::ConstantFold, Opt::Mem, Opt::ConstantFold],
         ),
+        Mode::Fhe => opt(cs, vec![Opt::ConstantFold]),
         Mode::Proof | Mode::ProofOfHighValue(_) => opt(
             cs,
             vec![
@@ -384,6 +388,15 @@ fn main() {
                 todo!()
             }
         }
+        Backend::Fhe { .. } => {
+            println!("Converting to fhe");
+            let lang_str = match language {
+                DeterminedLanguage::C => "c".to_string(),
+                DeterminedLanguage::Zsharp => "zok".to_string(),
+                _ => panic!("Language isn't supported by FHE backend: {:#?}", language),
+            };
+            to_fhe(cs, &path_buf, &lang_str);
+        }
         #[cfg(not(feature = "smt"))]
         Backend::Smt { .. } => {
             panic!("Missing feature: smt");

scripts/build_fhe_c_test.zsh

@@ -0,0 +1,40 @@
+#!/usr/bin/env zsh
+
+set -ex
+
+disable -r time
+
+# cargo build --release --features c --example circ 
+
+BIN=./target/release/examples/circ
+export CARGO_MANIFEST_DIR=$(pwd)
+
+case "$OSTYPE" in 
+    darwin*)
+        alias measure_time="gtime --format='%e seconds %M kB'"
+    ;;
+    linux*)
+        alias measure_time="time --format='%e seconds %M kB'"
+    ;;
+esac
+
+function fhe_test {
+    cpath=$1
+    RUST_BACKTRACE=1 measure_time $BIN $cpath fhe
+}
+
+# build boolean tests
+fhe_test ./examples/C/fhe/unit_tests/boolean_tests/boolean_and.c
+fhe_test ./examples/C/fhe/unit_tests/boolean_tests/boolean_or.c
+fhe_test ./examples/C/fhe/unit_tests/boolean_tests/boolean_equals.c
+
+# build nary boolean tests
+fhe_test ./examples/C/fhe/unit_tests/nary_boolean_tests/nary_boolean_and.c
+
+# build arithmetic tests
+fhe_test ./examples/C/fhe/unit_tests/arithmetic_tests/add.c
+fhe_test ./examples/C/fhe/unit_tests/arithmetic_tests/mult.c
+fhe_test ./examples/C/fhe/unit_tests/arithmetic_tests/mult_add_pub.c
+
+# build nary arithmetic tests
+fhe_test ./examples/C/fhe/unit_tests/nary_arithmetic_tests/nary_arithmetic_add.c