From 0ace1393ae29a56a5ad6aef407a4029d1cada445 Mon Sep 17 00:00:00 2001 From: "Rener Oliveira (Ubuntu WSL)" Date: Sat, 29 Jul 2023 23:05:18 -0300 Subject: [PATCH 1/2] enums for keygenmode --- src/lib/binfhe_bindings.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/lib/binfhe_bindings.cpp b/src/lib/binfhe_bindings.cpp index 79f2d8b..8c75ef2 100644 --- a/src/lib/binfhe_bindings.cpp +++ b/src/lib/binfhe_bindings.cpp @@ -97,6 +97,12 @@ void bind_binfhe_enums(py::module &m) m.attr("GINX") = py::cast(BINFHE_METHOD::GINX); m.attr("AP") = py::cast(BINFHE_METHOD::AP); + py::enum_(m, "KEYGEN_MODE") + .value("SYM_ENCRYPT", KEYGEN_MODE::SYM_ENCRYPT) + .value("PUB_ENCRYPT", KEYGEN_MODE::PUB_ENCRYPT); + m.attr("SYM_ENCRYPT") = py::cast(KEYGEN_MODE::SYM_ENCRYPT); + m.attr("PUB_ENCRYPT") = py::cast(KEYGEN_MODE::PUB_ENCRYPT); + py::enum_(m, "BINFHE_OUTPUT") .value("INVALID_OUTPUT", BINFHE_OUTPUT::INVALID_OUTPUT) .value("FRESH", BINFHE_OUTPUT::FRESH) From f4500c5ea0ff3095d2ac55aec87fd375c315a5a2 Mon Sep 17 00:00:00 2001 From: "Rener Oliveira (Ubuntu WSL)" Date: Sat, 29 Jul 2023 23:08:24 -0300 Subject: [PATCH 2/2] bug fix --- src/lib/binfhe_bindings.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lib/binfhe_bindings.cpp b/src/lib/binfhe_bindings.cpp index 8c75ef2..a26c331 100644 --- a/src/lib/binfhe_bindings.cpp +++ b/src/lib/binfhe_bindings.cpp @@ -159,7 +159,9 @@ void bind_binfhe_context(py::module &m) .def("KeyGen", &BinFHEContext::KeyGen, binfhe_KeyGen_docs) .def("BTKeyGen", &BinFHEContext::BTKeyGen, - binfhe_BTKeyGen_docs) + binfhe_BTKeyGen_docs, + py::arg("sk"), + py::arg("keygenMode") = SYM_ENCRYPT) .def("Encrypt", &binfhe_EncryptWrapper, binfhe_Encrypt_docs, py::arg("sk"), @@ -172,8 +174,6 @@ void bind_binfhe_context(py::module &m) py::arg("sk"), py::arg("ct"), py::arg("p") = 4) - // LWECiphertext EvalBinGate(BINGATE gate, ConstLWECiphertext& ct1, ConstLWECiphertext& ct2) const - // bind using static_cast .def("EvalBinGate", static_cast(&BinFHEContext::EvalBinGate), binfhe_EvalBinGate_docs,