builder/digitalocean: timeout TCP connections to SSH

13 years ago · 867e9d1c8f
parent b78fd17a0a
commit 867e9d1c8f
1 changed files with 4 additions and 1 deletions
--- a/builder/digitalocean/step_connect_ssh.go
+++ b/builder/digitalocean/step_connect_ssh.go
@ -61,7 +61,10 @@ func (s *stepConnectSSH) Run(state map[string]interface{}) multistep.StepAction
 			log.Printf(
 				"Opening TCP conn for SSH to %s:%d (attempt %d)",
 				ipAddress, config.SSHPort, attempts)
-			s.conn, err = net.Dial("tcp", fmt.Sprintf("%s:%d", ipAddress, config.SSHPort))
+			s.conn, err = net.DialTimeout(
+				"tcp",
+				fmt.Sprintf("%s:%d", ipAddress, config.SSHPort),
+				10*time.Second)
 			if err == nil {
 				break
 			}