118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
|
package digitalocean
|
||
|
|
|
||
|
|
import (
|
||
|
|
gossh "code.google.com/p/go.crypto/ssh"
|
||
|
|
"fmt"
|
||
|
|
"github.com/mitchellh/multistep"
|
||
|
|
"github.com/mitchellh/packer/communicator/ssh"
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type stepConnectSSH struct {
|
||
|
|
conn net.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepConnectSSH) Run(state map[string]interface{}) multistep.StepAction {
|
||
|
|
config := state["config"].(config)
|
||
|
|
privateKey := state["privateKey"].(string)
|
||
|
|
ui := state["ui"].(packer.Ui)
|
||
|
|
ipAddress := state["droplet_ip"]
|
||
|
|
|
||
|
|
// Build the keyring for authentication. This stores the private key
|
||
|
|
// we'll use to authenticate.
|
||
|
|
keyring := &ssh.SimpleKeychain{}
|
||
|
|
err := keyring.AddPEMKey(privateKey)
|
||
|
|
if err != nil {
|
||
|
|
ui.Say(fmt.Sprintf("Error setting up SSH config: %s", err))
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
// Build the actual SSH client configuration
|
||
|
|
sshConfig := &gossh.ClientConfig{
|
||
|
|
User: config.SSHUsername,
|
||
|
|
Auth: []gossh.ClientAuth{
|
||
|
|
gossh.ClientAuthKeyring(keyring),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
// Start trying to connect to SSH
|
||
|
|
connected := make(chan bool, 1)
|
||
|
|
connectQuit := make(chan bool, 1)
|
||
|
|
defer func() {
|
||
|
|
connectQuit <- true
|
||
|
|
}()
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
var err error
|
||
|
|
|
||
|
|
ui.Say("Connecting to the droplet via SSH...")
|
||
|
|
attempts := 0
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-connectQuit:
|
||
|
|
return
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
|
||
|
|
attempts += 1
|
||
|
|
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))
|
||
|
|
if err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
// A brief sleep so we're not being overly zealous attempting
|
||
|
|
// to connect to the instance.
|
||
|
|
time.Sleep(500 * time.Millisecond)
|
||
|
|
}
|
||
|
|
|
||
|
|
connected <- true
|
||
|
|
}()
|
||
|
|
|
||
|
|
log.Printf("Waiting up to %s for SSH connection", config.SSHTimeout)
|
||
|
|
timeout := time.After(config.SSHTimeout)
|
||
|
|
|
||
|
|
ConnectWaitLoop:
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-connected:
|
||
|
|
// We connected. Just break the loop.
|
||
|
|
break ConnectWaitLoop
|
||
|
|
case <-timeout:
|
||
|
|
ui.Error("Timeout while waiting to connect to SSH.")
|
||
|
|
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
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var comm packer.Communicator
|
||
|
|
if err == nil {
|
||
|
|
comm, err = ssh.New(s.conn, sshConfig)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
ui.Error(fmt.Sprintf("Error connecting to SSH: %s", err))
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set the communicator on the state bag so it can be used later
|
||
|
|
state["communicator"] = comm
|
||
|
|
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepConnectSSH) Cleanup(map[string]interface{}) {
|
||
|
|
if s.conn != nil {
|
||
|
|
s.conn.Close()
|
||
|
|
}
|
||
|
|
}
|