From ab9dfbcb51d597b7fa05ab279df159a177efbe55 Mon Sep 17 00:00:00 2001
From: Jaime Melis <jmelis@opennebula.org>
Date: Thu, 6 Oct 2016 17:11:31 +0200
Subject: [PATCH] B #4781: Rollback for ssh, shared and qcow2

---
 src/mad/sh/scripts_common.sh | 6 +++---
 src/tm_mad/common/mkimage    | 9 ++++++++-
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/mad/sh/scripts_common.sh b/src/mad/sh/scripts_common.sh
index 91a7e126fd..1d60a017e3 100644
--- a/src/mad/sh/scripts_common.sh
+++ b/src/mad/sh/scripts_common.sh
@@ -359,10 +359,10 @@ EOF`
 function ssh_exec_and_log
 {
     ssh_exec_and_log_no_error "$@"
-    rc=$?
+    EXEC_RC=$?
 
-    if [ $rc !=0 ]; then
-        exit $rc
+    if [ "$EXEC_RC" != "0" ]; then
+        exit $EXEC_RC
     fi
 }
 
diff --git a/src/tm_mad/common/mkimage b/src/tm_mad/common/mkimage
index 43c296c88e..014bd188e5 100755
--- a/src/tm_mad/common/mkimage
+++ b/src/tm_mad/common/mkimage
@@ -54,9 +54,16 @@ ssh_make_path $DST_HOST $DST_DIR
 MKFS_CMD=`mkfs_command $DST_PATH $FSTYPE $SIZE`
 
 MKSCRIPT=$(cat <<EOF
-set -e -o pipefail
 export PATH=/usr/sbin:/sbin:\$PATH
 $MKFS_CMD
+
+rc=\$?
+
+if [ "\$rc" != "0" ]; then
+    rm -f $DST_PATH
+fi
+
+exit \$rc
 EOF
 )