iSharkFly-Docs
/
packer-cn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

builder/vmware: Handle interrupts while waiting for SSH

This commit is contained in:
Mitchell Hashimoto 2013-06-08 18:55:34 -07:00
parent be1e60aa26
commit 174ae65a12
1 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
builder/vmware/step_wait_for_ssh.go
View File

@ -44,6 +44,10 @@ func (s *stepWaitForSSH) Run(state map[string]interface{}) multistep.StepAction
log.Printf("Waiting for SSH, up to timeout: %s", config.SSHWaitTimeout.String())
timeout := time.After(config.SSHWaitTimeout)
for {
// Wait for either SSH to become available, a timeout to occur,
// or an interrupt to come through.
select {
case <-waitDone:
if err != nil {
@ -52,10 +56,17 @@ func (s *stepWaitForSSH) Run(state map[string]interface{}) multistep.StepAction
}
state["communicator"] = comm
case <-time.After(config.SSHWaitTimeout):
break
case <-timeout:
ui.Error("Timeout waiting for SSH.")
s.cancel = true
return multistep.ActionHalt
case <-time.After(1 * time.Second):
if _, ok := state[multistep.StateCancelled]; ok {
log.Println("Interrupt detected, quitting waiting for SSH.")
return multistep.ActionHalt
}
}
}
return multistep.ActionContinue