ota: clean up images during lock acquired procedure (#881)

yodaos-project · Oct 21, 2019 · 2daa2fb · 2daa2fb
1 parent 279519a
commit 2daa2fb
Showing 1 changed file with 18 additions and 9 deletions.
diff --git a/runtime/services/otad/step.js b/runtime/services/otad/step.js
@@ -140,15 +140,24 @@ function runInCurrentContext (delegate, callback) {
       unlockProc = unlock
       cb()
     },
-    cb => delegate.prelude(cb),
-    (cb, result) => {
-      if (!result) {
-        return compose.Break(null)
-      }
-      cb()
-    },
-    /** actual procedure, shall skip if prepare failed */
-    cb => doRun(cb)
+    cb =>
+      compose([
+        cb => delegate.prelude(cb),
+        (cb, result) => {
+          if (!result) {
+            return compose.Break(null)
+          }
+          cb()
+        },
+        /** actual procedure, shall skip if prepare failed */
+        cb => doRun(cb)
+      ], (err, info) => {
+        if (err) {
+          common.cleanup(() => cb(err))
+          return
+        }
+        cb(null, info)
+      })
   ], (err, info) => {
     logger.info('ota unlocking proc lock.')
     if (typeof unlockProc !== 'function') {