Adding private key support

fairmath · Oct 9, 2024 · 0fd0a56 · 0fd0a56
1 parent 445bc39
commit 0fd0a56
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 0 deletions.
diff --git a/src/PrivateKey.cc b/src/PrivateKey.cc
@@ -17,4 +17,9 @@ std::shared_ptr<PrivateKeyImpl>& PrivateKeyDCRTPoly::GetRef() noexcept
     return m_privateKey;
 }
 
+std::unique_ptr<PrivateKeyDCRTPoly> DCRTPolyGenNullPrivateKey()
+{
+    return std::make_unique<PrivateKeyDCRTPoly>();
+}
+
 } // openfhe
diff --git a/src/PrivateKey.h b/src/PrivateKey.h
@@ -12,6 +12,7 @@ class PrivateKeyDCRTPoly final
 {
     std::shared_ptr<PrivateKeyImpl> m_privateKey;
 public:
+    PrivateKeyDCRTPoly() = default;
     PrivateKeyDCRTPoly(const std::shared_ptr<PrivateKeyImpl>& privateKey) noexcept;
     PrivateKeyDCRTPoly(const PrivateKeyDCRTPoly&) = delete;
     PrivateKeyDCRTPoly(PrivateKeyDCRTPoly&&) = delete;
@@ -22,4 +23,6 @@ class PrivateKeyDCRTPoly final
     [[nodiscard]] std::shared_ptr<PrivateKeyImpl>& GetRef() noexcept;
 };
 
+[[nodiscard]] std::unique_ptr<PrivateKeyDCRTPoly> DCRTPolyGenNullPrivateKey();
+
 } // openfhe
diff --git a/src/SerialDeserial.cc b/src/SerialDeserial.cc
@@ -4,6 +4,7 @@
 
 #include "Ciphertext.h"
 #include "CryptoContext.h"
+#include "PrivateKey.h"
 #include "PublicKey.h"
 
 namespace openfhe
@@ -237,4 +238,16 @@ bool DCRTPolySerializePublicKeyToFile(const std::string& publicKeyLocation,
     return Serial(publicKeyLocation, publicKey, serialMode);
 }
 
+bool DCRTPolyDeserializePrivateKeyFromFile(const std::string& privateKeyLocation,
+    PrivateKeyDCRTPoly& privateKey, const SerialMode serialMode)
+{
+    return Deserial(privateKeyLocation, privateKey, serialMode);
+}
+
+bool DCRTPolySerializePrivateKeyToFile(const std::string& privateKeyLocation,
+    const PrivateKeyDCRTPoly& privateKey, const SerialMode serialMode)
+{
+    return Serial(privateKeyLocation, privateKey, serialMode);
+}
+
 } // openfhe
diff --git a/src/SerialDeserial.h b/src/SerialDeserial.h
@@ -9,6 +9,7 @@ namespace openfhe
 
 class CiphertextDCRTPoly;
 class CryptoContextDCRTPoly;
+class PrivateKeyDCRTPoly;
 class PublicKeyDCRTPoly;
 
 // Ciphertext
@@ -55,4 +56,9 @@ class PublicKeyDCRTPoly;
 [[nodiscard]] bool DCRTPolySerializePublicKeyToFile(const std::string& publicKeyLocation,
     const PublicKeyDCRTPoly& publicKey, const SerialMode serialMode);
 
+[[nodiscard]] bool DCRTPolyDeserializePrivateKeyFromFile(const std::string& privateKeyLocation,
+    PrivateKeyDCRTPoly& privateKey, const SerialMode serialMode);
+[[nodiscard]] bool DCRTPolySerializePrivateKeyToFile(const std::string& privateKeyLocation,
+    const PrivateKeyDCRTPoly& cryptoContext, const SerialMode serialMode);
+
 } // openfhe
diff --git a/src/lib.rs b/src/lib.rs
@@ -1056,6 +1056,13 @@ pub mod ffi
         fn DCRTPolyGenNullPublicKey() -> UniquePtr<PublicKeyDCRTPoly>;
     }
 
+    // PrivateKeyDCRTPoly
+    unsafe extern "C++"
+    {
+        // Generator functions
+        fn DCRTPolyGenNullPrivateKey() -> UniquePtr<PrivateKeyDCRTPoly>;
+    }
+
     // Serialize / Deserialize
     unsafe extern "C++"
     {
@@ -1110,6 +1117,14 @@ pub mod ffi
         fn DCRTPolySerializePublicKeyToFile(publicKeyLocation: &CxxString,
                                             publicKey: &PublicKeyDCRTPoly,
                                             serialMode: SerialMode) -> bool;
+
+        // PrivateKey
+        fn DCRTPolyDeserializePrivateKeyFromFile(privateKeyLocation: &CxxString,
+                                                 privateKey: Pin<&mut PrivateKeyDCRTPoly>,
+                                                 serialMode: SerialMode) -> bool;
+        fn DCRTPolySerializePrivateKeyToFile(privateKeyLocation: &CxxString,
+                                             privateKey: &PrivateKeyDCRTPoly,
+                                             serialMode: SerialMode) -> bool;
     }
 }