Set unpacked_dir to recovery file at StepMountUnpacked

The unpacked dir path should be set for recovery mode.

Change-Id: I4fafbb4ffb9d48cbb3ce753041b1fde11c118e55
Signed-off-by: Sangyoon Jang <jeremy.jang@samsung.com>

diff --git a/src/common/step/mount/step_mount_unpacked.cc b/src/common/step/mount/step_mount_unpacked.cc
index c4dfb2c..445a42b 100644 (file)
--- a/src/common/step/mount/step_mount_unpacked.cc
+++ b/src/common/step/mount/step_mount_unpacked.cc
@@ -30,6 +30,12 @@ namespace mount {
 Step::Status StepMountUnpacked::process() {
   bf::path tmp_dir = GenerateTmpDir(kPackageUnpackDirPath);
   context_->unpacked_dir_path.set(tmp_dir);
+  // write unpacked directory for recovery file
+  // to remove unpacked dir properly when recovery mode
+  if (context_->recovery_info.get().recovery_file) {
+    context_->recovery_info.get().recovery_file->set_unpacked_dir(tmp_dir);
+    context_->recovery_info.get().recovery_file->WriteAndCommitFileContent();
+  }
 
   auto zip = CreateZipInterface(context_->unpacked_dir_path.get());
   if (!zip->MountZip(context_->file_path.get())) {