From b75438ff3225e976ac4884205cb4387efebeb160 Mon Sep 17 00:00:00 2001
From: Trent Nelson <trent@solana.com>
Date: Mon, 14 Oct 2019 10:36:20 -0600
Subject: [PATCH] gce.sh: Unwind allocation upon failure (#6343)

automerge
---
 net/gce.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/net/gce.sh b/net/gce.sh
index 4003e69a12..a051b02359 100755
--- a/net/gce.sh
+++ b/net/gce.sh
@@ -575,6 +575,14 @@ delete() {
   $metricsWriteDatapoint "testnet-deploy net-delete-complete=1"
 }
 
+create_error_cleanup() {
+  declare RC=$?
+  if [[ "$RC" -ne 0 ]]; then
+    delete
+  fi
+  exit $RC
+}
+
 case $command in
 delete)
   delete
@@ -587,6 +595,10 @@ create)
 
   $metricsWriteDatapoint "testnet-deploy net-create-begin=1"
 
+  if $failOnValidatorBootupFailure; then
+    trap create_error_cleanup EXIT
+  fi
+
   rm -rf "$sshPrivateKey"{,.pub}
 
   # Note: using rsa because |aws ec2 import-key-pair| seems to fail for ecdsa