Store database storage to files, also loading

1 files changed, 51 insertions, 0 deletions

diff --git a/include/odhtdb/LocalUserEncrypted.hpp b/include/odhtdb/LocalUserEncrypted.hpp
new file mode 100644
index 0000000..c250d13
--- /dev/null
+++ b/include/odhtdb/LocalUserEncrypted.hpp
@@ -0,0 +1,51 @@
+#pragma once
+
+#include "User.hpp"
+#include "types.hpp"
+#include "Encryption.hpp"
+
+namespace odhtdb
+{
+    struct EncryptedPrivateKey
+    {
+        u8 nonce[ENCRYPTION_NONCE_BYTE_SIZE];
+        u8 encryptedPrivateKey[16 + PRIVATE_KEY_NUM_BYTES];
+        
+        EncryptedPrivateKey();
+        EncryptedPrivateKey(const EncryptedPrivateKey &other);
+        
+        // Throws DecryptionException if password (or salt) is wrong
+        Signature::PrivateKey decrypt(const DataView &plainPassword, const DataView &salt) const;
+    };
+    
+    // Local user with encrypted private key
+    class LocalUserEncrypted : public User
+    {
+    public:
+        static LocalUserEncrypted* create(const Signature::PublicKey &publicKey, const EncryptedPrivateKey &encryptedPrivateKey, const std::string &name, Group *group)
+        {
+            return new LocalUserEncrypted(publicKey, encryptedPrivateKey, name, group);
+        }
+        
+        const Signature::PublicKey& getPublicKey() const override
+        {
+            return publicKey;
+        }
+        
+        const EncryptedPrivateKey& getPrivateKey() const
+        {
+            return encryptedPrivateKey;
+        }
+    private:
+        LocalUserEncrypted(const Signature::PublicKey &_publicKey, const EncryptedPrivateKey &_encryptedPrivateKey, const std::string &name, Group *group) : 
+            User(User::Type::LOCAL_ENCRYPTED, name, group),
+            publicKey(_publicKey),
+            encryptedPrivateKey(_encryptedPrivateKey)
+        {
+            
+        }
+    private:
+        Signature::PublicKey publicKey;
+        EncryptedPrivateKey encryptedPrivateKey;
+    };
+}