Fix Helper bootsSymEncrypt

virtualsecureplatform · Oct 23, 2024 · cef10bf · cef10bf
1 parent 260ec8d
commit cef10bf
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/include/tlwe.hpp b/include/tlwe.hpp
@@ -112,7 +112,7 @@ typename P::T tlweSymIntDecrypt(const TLWE<P> &c, const Key<P> &key)
     return tlweSymIntDecrypt<P, P::plain_modulus>(c, key);
 }
 
-template <class P, typename P::T μ>
+template <class P, std::make_signed_t<typename P::T> μ>
 std::vector<TLWE<P>> bootsSymEncrypt(const std::vector<uint8_t> &p,
                                      const Key<P> &key)
 {
@@ -127,7 +127,7 @@ template <class P>
 std::vector<TLWE<P>> bootsSymEncrypt(const std::vector<uint8_t> &p,
                                      const Key<P> &key)
 {
-    return bootsSymEncrypt<P, typename P::T>(p, key);
+    return bootsSymEncrypt<P, P::μ>(p, key);
 }
 
 template <class P = lvl1param>
@@ -137,6 +137,13 @@ std::vector<TLWE<P>> bootsSymEncrypt(const std::vector<uint8_t> &p,
     return bootsSymEncrypt<P>(p, sk.key.get<P>());
 }
 
+template <class P = lvl1param, std::make_signed_t<typename P::T> μ>
+std::vector<TLWE<P>> bootsSymEncrypt(const std::vector<uint8_t> &p,
+                                     const SecretKey &sk)
+{
+    return bootsSymEncrypt<P, μ>(p, sk.key.get<P>());
+}
+
 template <class P>
 std::vector<uint8_t> bootsSymDecrypt(const std::vector<TLWE<P>> &c,
                                      const Key<P> &key)