Add fail-safe code for key migration

Change-Id: Ibed92f38d5dfd3de1ecc7faff5f1ffdb9ff68c02
Signed-off-by: Jaemin Ryu <jm77.ryu@samsung.com>

diff --git a/server/internal-encryption.cpp b/server/internal-encryption.cpp
index bcf0487..75c39b7 100644 (file)
--- a/server/internal-encryption.cpp
+++ b/server/internal-encryption.cpp
@@ -48,6 +48,7 @@
 #include "internal-encryption.h"
 #include "internal-encryption-common.h"
 #include "upgrade-support.h"
+#include "file-footer.h"
 
 namespace ode {
 
@@ -380,6 +381,14 @@ InternalEncryptionServer::InternalEncryptionServer(ServerContext& srv,
 
        std::string source = findDevPath();
 
+       if (getState() == State::Encrypted) {
+               //"error_partially_encrypted"
+               if (!FileFooter::exist(source) && !UpgradeSupport::checkUpgradeFlag()) {
+                       // Trigger key migration process
+                       UpgradeSupport::createUpgradeFlag();
+               }
+       }
+
        engine.reset(new INTERNAL_ENGINE(
                source, INTERNAL_PATH,
                ProgressBar([](int v) {