Added SEAL library and FHE mode

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 == "fhe":
+            if verify_path_empty(SEAL_SOURCE):
+                subprocess.run(["git", "clone", "https://github.com/Northrim/SEAL.git", SEAL_SOURCE])
+                subprocess.run(["./scripts/build_aby.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 format():
     print("formatting!")
@@ -133,6 +147,8 @@ def clean(features):
     print("cleaning!")
     if "aby" in features:
         subprocess.run(["./scripts/clean_aby.zsh"])
+    if "fhe" 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"])
 
@@ -178,7 +194,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("extra", metavar="PASS_THROUGH_ARGS", nargs=argparse.REMAINDER, help="Extra arguments for --flamegraph. Prefix with --")
     args = parser.parse_args()
 

examples/C/fhe/unit_tests/boolean_tests/2pc_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/2pc_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/ZoKrates/fhe/inputs/fhe_and.txt

@@ -0,0 +1,3 @@
+a 1
+b 1
+res 1

examples/ZoKrates/fhe/unit_tests/boolean_tests/2pc_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/2pc_boolean_or.zok

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

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")]
@@ -96,6 +97,7 @@ enum Backend {
     },
     Smt {},
     Ilp {},
+    Fhe {},
     Mpc {
         #[structopt(long, default_value = "hycc", name = "cost_model")]
         cost_model: String,
@@ -180,6 +182,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 {
@@ -239,6 +242,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![
@@ -372,6 +376,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,28 @@
+#!/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/2pc_boolean_and.c
+fhe_test ./examples/C/fhe/unit_tests/boolean_tests/2pc_boolean_or.c

scripts/build_fhe_zokrates_test.zsh

@@ -0,0 +1,28 @@
+#!/usr/bin/env zsh
+
+set -ex
+
+disable -r time
+
+# cargo build --release --features smt,zok --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 {
+    zpath=$1
+    RUST_BACKTRACE=1 measure_time $BIN $zpath fhe
+}
+
+# build boolean tests
+fhe_test ./examples/ZoKrates/fhe/unit_tests/boolean_tests/2pc_boolean_and.zok
+fhe_test ./examples/ZoKrates/fhe/unit_tests/boolean_tests/2pc_boolean_or.zok 

scripts/build_seal.zsh

@@ -0,0 +1,9 @@
+#!/usr/bin/env zsh
+
+if [[ ! -z ${SEAL_SOURCE} ]]; then 
+    mkdir -p -- ${SEAL_SOURCE}/build
+    cd ${SEAL_SOURCE}
+    cmake --build build
+else
+    echo "Missing SEAL_SOURCE environment variable."
+fi

scripts/clean_seal.zsh

@@ -0,0 +1,3 @@
+#!/usr/bin/env zsh
+
+rm -rf ./scripts/seal_tests/tests

scripts/seal_tests/c_test_seal.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python
+
+from util import run_tests
+from test_suite import *
+
+if __name__ == "__main__":
+    tests = boolean_tests #+ \
+        # arithmetic_tests + \
+        # mod_tests + \
+        # arithmetic_boolean_tests + \
+        # nary_arithmetic_tests + \
+        # bitwise_tests + \
+        # nary_boolean_tests + \
+        # const_arith_tests + \
+        # const_bool_tests + \
+        # ite_tests + \
+        # div_tests + \
+        # array_tests + \
+        # c_array_tests + \
+        # misc_tests + \
+        # biomatch_tests + \
+        # kmeans_tests
+        # shift_tests
+
+    # TODO: add support for return value - int promotion
+    # unsigned_arithmetic_tests + \
+
+    run_tests('c', tests)

scripts/seal_tests/test_inputs/add.txt

@@ -0,0 +1,3 @@
+a 1 
+b 2
+res 3

scripts/seal_tests/test_inputs/add_2.txt

@@ -0,0 +1,3 @@
+a 1 
+b 2
+res 6

scripts/seal_tests/test_inputs/and_1.txt

@@ -0,0 +1,3 @@
+a 0
+b 0
+res 0

scripts/seal_tests/test_inputs/and_2.txt

@@ -0,0 +1,3 @@
+a 1
+b 0
+res 0

scripts/seal_tests/test_inputs/and_3.txt

@@ -0,0 +1,3 @@
+a 0
+b 1
+res 0

scripts/seal_tests/test_inputs/and_4.txt

@@ -0,0 +1,3 @@
+a 1
+b 1
+res 1

scripts/seal_tests/test_inputs/array.txt

@@ -0,0 +1,3 @@
+a 2
+b 2
+res 2

scripts/seal_tests/test_inputs/array_1.txt

@@ -0,0 +1,3 @@
+a 10
+b 3
+res 17

scripts/seal_tests/test_inputs/array_2.txt

@@ -0,0 +1,3 @@
+a 10
+b 3
+res 17

scripts/seal_tests/test_inputs/array_3.txt

@@ -0,0 +1,3 @@
+a 2
+b 3
+res 18

scripts/seal_tests/test_inputs/array_4.txt

@@ -0,0 +1,3 @@
+a 1 2 3 4 5
+b 1 2 3 4 5
+res 30

scripts/seal_tests/test_inputs/biomatch_1.txt

@@ -0,0 +1,3 @@
+db 0 1 2 3 1 2 3 4 2 3 4 5 3 4 5 6 4 5 6 7 5 6 7 8 6 7 8 9 7 8 9 10 8 9 10 11 9 10 11 12 10 11 12 13 11 12 13 14 12 13 14 15 13 14 15 16 14 15 16 17 15 16 17 18 16 17 18 19 17 18 19 20 18 19 20 21 19 20 21 22 20 21 22 23 21 22 23 24 22 23 24 25 23 24 25 26 24 25 26 27 25 26 27 28 26 27 28 29 27 28 29 30 28 29 30 31 29 30 31 32 30 31 32 33 31 32 33 34 32 33 34 35 33 34 35 36 34 35 36 37 35 36 37 38 36 37 38 39 37 38 39 40 38 39 40 41 39 40 41 42 40 41 42 43 41 42 43 44 42 43 44 45 43 44 45 46 44 45 46 47 45 46 47 48 46 47 48 49 47 48 49 50 48 49 50 51 49 50 51 52 50 51 52 53 51 52 53 54 52 53 54 55 53 54 55 56 54 55 56 57 55 56 57 58 56 57 58 59 57 58 59 60 58 59 60 61 59 60 61 62 60 61 62 63 61 62 63 64 62 63 64 65 63 64 65 66 64 65 66 67 65 66 67 68 66 67 68 69 67 68 69 70 68 69 70 71 69 70 71 72 70 71 72 73 71 72 73 74 72 73 74 75 73 74 75 76 74 75 76 77 75 76 77 78 76 77 78 79 77 78 79 80 78 79 80 81 79 80 81 82 80 81 82 83 81 82 83 84 82 83 84 85 83 84 85 86 84 85 86 87 85 86 87 88 86 87 88 89 87 88 89 90 88 89 90 91 89 90 91 92 90 91 92 93 91 92 93 94 92 93 94 95 93 94 95 96 94 95 96 97 95 96 97 98 96 97 98 99 97 98 99 100 98 99 100 101 99 100 101 102 100 101 102 103 101 102 103 104 102 103 104 105 103 104 105 106 104 105 106 107 105 106 107 108 106 107 108 109 107 108 109 110 108 109 110 111 109 110 111 112 110 111 112 113 111 112 113 114 112 113 114 115 113 114 115 116 114 115 116 117 115 116 117 118 116 117 118 119 117 118 119 120 118 119 120 121 119 120 121 122 120 121 122 123 121 122 123 124 122 123 124 125 123 124 125 126 124 125 126 127 125 126 127 128 126 127 128 129 127 128 129 130 128 129 130 131 129 130 131 132 130 131 132 133 131 132 133 134 132 133 134 135 133 134 135 136 134 135 136 137 135 136 137 138 136 137 138 139 137 138 139 140 138 139 140 141 139 140 141 142 140 141 142 143 141 142 143 144 142 143 144 145 143 144 145 146 144 145 146 147 145 146 147 148 146 147 148 149 147 148 149 150 148 149 150 151 149 150 151 152 150 151 152 153 151 152 153 154 152 153 154 155 153 154 155 156 154 155 156 157 155 156 157 158 156 157 158 159 157 158 159 160 158 159 160 161 159 160 161 162 160 161 162 163 161 162 163 164 162 163 164 165 163 164 165 166 164 165 166 167 165 166 167 168 166 167 168 169 167 168 169 170 168 169 170 171 169 170 171 172 170 171 172 173 171 172 173 174 172 173 174 175 173 174 175 176 174 175 176 177 175 176 177 178 176 177 178 179 177 178 179 180 178 179 180 181 179 180 181 182 180 181 182 183 181 182 183 184 182 183 184 185 183 184 185 186 184 185 186 187 185 186 187 188 186 187 188 189 187 188 189 190 188 189 190 191 189 190 191 192 190 191 192 193 191 192 193 194 192 193 194 195 193 194 195 196 194 195 196 197 195 196 197 198 196 197 198 199 197 198 199 200 198 199 200 201 199 200 201 202 200 201 202 203 201 202 203 204 202 203 204 205 203 204 205 206 204 205 206 207 205 206 207 208 206 207 208 209 207 208 209 210 208 209 210 211 209 210 211 212 210 211 212 213 211 212 213 214 212 213 214 215 213 214 215 216 214 215 216 217 215 216 217 218 216 217 218 219 217 218 219 220 218 219 220 221 219 220 221 222 220 221 222 223 221 222 223 224 222 223 224 225 223 224 225 226 224 225 226 227 225 226 227 228 226 227 228 229 227 228 229 230 228 229 230 231 229 230 231 232 230 231 232 233 231 232 233 234 232 233 234 235 233 234 235 236 234 235 236 237 235 236 237 238 236 237 238 239 237 238 239 240 238 239 240 241 239 240 241 242 240 241 242 243 241 242 243 244 242 243 244 245 243 244 245 246 244 245 246 247 245 246 247 248 246 247 248 249 247 248 249 250 248 249 250 251 249 250 251 252 250 251 252 253 251 252 253 254 252 253 254 255 253 254 255 256 254 255 256 257 255 256 257 258
+sample 0 0 0 0
+res 14

scripts/seal_tests/test_inputs/biomatch_2.txt

@@ -0,0 +1,3 @@
+db 0 1 2 3 1 2 3 4 2 3 4 5 3 4 5 6 4 5 6 7 5 6 7 8 6 7 8 9 7 8 9 10 8 9 10 11 9 10 11 12 10 11 12 13 11 12 13 14 12 13 14 15 13 14 15 16 14 15 16 17 15 16 17 18 16 17 18 19 17 18 19 20 18 19 20 21 19 20 21 22 20 21 22 23 21 22 23 24 22 23 24 25 23 24 25 26 24 25 26 27 25 26 27 28 26 27 28 29 27 28 29 30 28 29 30 31 29 30 31 32 30 31 32 33 31 32 33 34 32 33 34 35 33 34 35 36 34 35 36 37 35 36 37 38 36 37 38 39 37 38 39 40 38 39 40 41 39 40 41 42 40 41 42 43 41 42 43 44 42 43 44 45 43 44 45 46 44 45 46 47 45 46 47 48 46 47 48 49 47 48 49 50 48 49 50 51 49 50 51 52 50 51 52 53 51 52 53 54 52 53 54 55 53 54 55 56 54 55 56 57 55 56 57 58 56 57 58 59 57 58 59 60 58 59 60 61 59 60 61 62 60 61 62 63 61 62 63 64 62 63 64 65 63 64 65 66 64 65 66 67 65 66 67 68 66 67 68 69 67 68 69 70 68 69 70 71 69 70 71 72 70 71 72 73 71 72 73 74 72 73 74 75 73 74 75 76 74 75 76 77 75 76 77 78 76 77 78 79 77 78 79 80 78 79 80 81 79 80 81 82 80 81 82 83 81 82 83 84 82 83 84 85 83 84 85 86 84 85 86 87 85 86 87 88 86 87 88 89 87 88 89 90 88 89 90 91 89 90 91 92 90 91 92 93 91 92 93 94 92 93 94 95 93 94 95 96 94 95 96 97 95 96 97 98 96 97 98 99 97 98 99 100 98 99 100 101 99 100 101 102 100 101 102 103 101 102 103 104 102 103 104 105 103 104 105 106 104 105 106 107 105 106 107 108 106 107 108 109 107 108 109 110 108 109 110 111 109 110 111 112 110 111 112 113 111 112 113 114 112 113 114 115 113 114 115 116 114 115 116 117 115 116 117 118 116 117 118 119 117 118 119 120 118 119 120 121 119 120 121 122 120 121 122 123 121 122 123 124 122 123 124 125 123 124 125 126 124 125 126 127 125 126 127 128 126 127 128 129 127 128 129 130 128 129 130 131 129 130 131 132 130 131 132 133 131 132 133 134 132 133 134 135 133 134 135 136 134 135 136 137 135 136 137 138 136 137 138 139 137 138 139 140 138 139 140 141 139 140 141 142 140 141 142 143 141 142 143 144 142 143 144 145 143 144 145 146 144 145 146 147 145 146 147 148 146 147 148 149 147 148 149 150 148 149 150 151 149 150 151 152 150 151 152 153 151 152 153 154 152 153 154 155 153 154 155 156 154 155 156 157 155 156 157 158 156 157 158 159 157 158 159 160 158 159 160 161 159 160 161 162 160 161 162 163 161 162 163 164 162 163 164 165 163 164 165 166 164 165 166 167 165 166 167 168 166 167 168 169 167 168 169 170 168 169 170 171 169 170 171 172 170 171 172 173 171 172 173 174 172 173 174 175 173 174 175 176 174 175 176 177 175 176 177 178 176 177 178 179 177 178 179 180 178 179 180 181 179 180 181 182 180 181 182 183 181 182 183 184 182 183 184 185 183 184 185 186 184 185 186 187 185 186 187 188 186 187 188 189 187 188 189 190 188 189 190 191 189 190 191 192 190 191 192 193 191 192 193 194 192 193 194 195 193 194 195 196 194 195 196 197 195 196 197 198 196 197 198 199 197 198 199 200 198 199 200 201 199 200 201 202 200 201 202 203 201 202 203 204 202 203 204 205 203 204 205 206 204 205 206 207 205 206 207 208 206 207 208 209 207 208 209 210 208 209 210 211 209 210 211 212 210 211 212 213 211 212 213 214 212 213 214 215 213 214 215 216 214 215 216 217 215 216 217 218 216 217 218 219 217 218 219 220 218 219 220 221 219 220 221 222 220 221 222 223 221 222 223 224 222 223 224 225 223 224 225 226 224 225 226 227 225 226 227 228 226 227 228 229 227 228 229 230 228 229 230 231 229 230 231 232 230 231 232 233 231 232 233 234 232 233 234 235 233 234 235 236 234 235 236 237 235 236 237 238 236 237 238 239 237 238 239 240 238 239 240 241 239 240 241 242 240 241 242 243 241 242 243 244 242 243 244 245 243 244 245 246 244 245 246 247 245 246 247 248 246 247 248 249 247 248 249 250 248 249 250 251 249 250 251 252 250 251 252 253 251 252 253 254 252 253 254 255 253 254 255 256 254 255 256 257 255 256 257 258
+sample 1 2 3 4
+res 0

scripts/seal_tests/test_inputs/const_add.txt

@@ -0,0 +1,3 @@
+a 2 
+b 3
+res 6

scripts/seal_tests/test_inputs/const_eq_1.txt

@@ -0,0 +1,3 @@
+a 0 
+b 0
+res 0

scripts/seal_tests/test_inputs/const_eq_2.txt

@@ -0,0 +1,3 @@
+a 1
+b 0
+res 1

scripts/seal_tests/test_inputs/conv.txt

@@ -0,0 +1,3 @@
+a 0
+b 7
+res 7

scripts/seal_tests/test_inputs/div_1.txt

@@ -0,0 +1,3 @@
+a 10
+b 1
+res 10

scripts/seal_tests/test_inputs/div_2.txt

@@ -0,0 +1,3 @@
+a 10
+b 2
+res 5

scripts/seal_tests/test_inputs/div_3.txt

@@ -0,0 +1,3 @@
+a 11
+b 2
+res 5

scripts/seal_tests/test_inputs/div_4.txt

@@ -0,0 +1,3 @@
+a 1
+b 2
+res 0

scripts/seal_tests/test_inputs/eq_1.txt

@@ -0,0 +1,3 @@
+a 1
+b 0
+res 0

scripts/seal_tests/test_inputs/eq_2.txt

@@ -0,0 +1,3 @@
+a 1
+b 1
+res 1

scripts/seal_tests/test_inputs/ge_1.txt

@@ -0,0 +1,3 @@
+a 1
+b 2
+res 0

scripts/seal_tests/test_inputs/ge_2.txt

@@ -0,0 +1,3 @@
+a 1
+b 1
+res 1

scripts/seal_tests/test_inputs/ge_3.txt

@@ -0,0 +1,3 @@
+a 1
+b 0
+res 1

scripts/seal_tests/test_inputs/gt_1.txt

@@ -0,0 +1,3 @@
+a 1
+b 2
+res 0

scripts/seal_tests/test_inputs/gt_2.txt

@@ -0,0 +1,3 @@
+a 1
+b 1
+res 0

scripts/seal_tests/test_inputs/gt_3.txt

@@ -0,0 +1,3 @@
+a 1 
+b 0
+res 1

scripts/seal_tests/test_inputs/index.txt

@@ -0,0 +1,3 @@
+a 1
+b 2
+res 1

scripts/seal_tests/test_inputs/ite_1.txt

@@ -0,0 +1,4 @@
+a 0
+b 1
+sel 1
+res 0

scripts/seal_tests/test_inputs/ite_2.txt

@@ -0,0 +1,4 @@
+a 0
+b 1
+sel 0
+res 1