From 12176b6afc201a02be626cb8ee0222365964ede8 Mon Sep 17 00:00:00 2001
From: Matthew Hooker <mwhooker@gmail.com>
Date: Tue, 22 Nov 2016 15:59:38 -0800
Subject: [PATCH] provisioner/shell: treat disconnects as retryable.

---
 provisioner/shell/provisioner.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/provisioner/shell/provisioner.go b/provisioner/shell/provisioner.go
index 3575c70cd..00a81bf30 100644
--- a/provisioner/shell/provisioner.go
+++ b/provisioner/shell/provisioner.go
@@ -320,6 +320,10 @@ func (p *Provisioner) Provision(ui packer.Ui, comm packer.Communicator) error {
 						p.config.RemotePath, err)
 				}
 				cmd.Wait()
+				// treat disconnects as retryable by returning an error
+				if cmd.ExitStatus == packer.CmdDisconnect {
+					return fmt.Errorf("Disconnect while removing temporary script.")
+				}
 				return nil
 			})
 			if err != nil {