From dbcb0f85b5b6ef4cc2d198c9216bd51c024f0386 Mon Sep 17 00:00:00 2001
From: "Mariusz B. / mgeeky" <mb@binary-offensive.com>
Date: Tue, 7 Mar 2023 14:14:37 +0100
Subject: [PATCH] next

---
 clouds/aws/assume-role-helper.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/clouds/aws/assume-role-helper.sh b/clouds/aws/assume-role-helper.sh
index 5b81325..b4f5c37 100644
--- a/clouds/aws/assume-role-helper.sh
+++ b/clouds/aws/assume-role-helper.sh
@@ -72,6 +72,12 @@ else
     out=$(aws --profile $PROFILE_NAME sts assume-role --serial-number $SERIAL_MFA --role-arn $ROLE_ARN --role-session-name $ROLE_NAME --duration-seconds $DURATION --token-code $code 2>&1)
 fi
 
+if echo "$out" | grep -q -i 'error occurred' ; then
+    echo -e "[!] Assume role failed:\n"
+    echo "$out"
+    exit 1
+fi
+
 rolename=$PROFILE_NAME-$SESSION_NAME
 
 if [[ "$OUTPUT_ROLE_NAME" != "" ]]; then