wait for SSH when starting a VM

cmd/ignite/run/start.go

@@ -2,6 +2,10 @@ package run
 
 import (
 	"fmt"
+	"net"
+	"time"
+
+	"github.com/weaveworks/ignite/pkg/apis/ignite"
 
 	"github.com/weaveworks/ignite/pkg/operations"
 	"github.com/weaveworks/ignite/pkg/preflight/checkers"
@@ -47,9 +51,42 @@ func Start(so *startOptions) error {
 		return err
 	}
 
+	if err := waitForSSH(so.vm, 10); err != nil {
+		return err
+	}
+
 	// If starting interactively, attach after starting
 	if so.Interactive {
 		return Attach(so.attachOptions)
 	}
 	return nil
 }
+
+func waitForSSH(vm *ignite.VM, seconds int) error {
+	// When --ssh is enabled, wait until SSH service started on port 22 at most N seconds
+	ssh := vm.Spec.SSH
+	if ssh != nil && ssh.Generate && len(vm.Status.IPAddresses) > 0 {
+		addr := vm.Status.IPAddresses[0].String() + ":22"
+		perSecond := 10
+		delay := time.Second / time.Duration(perSecond)
+		var err error
+		for i := 0; i < seconds*perSecond; i++ {
+			conn, dialErr := net.DialTimeout("tcp", addr, delay)
+			if conn != nil {
+				defer conn.Close()
+				err = nil
+				break
+			}
+			err = dialErr
+			time.Sleep(delay)
+		}
+		if err != nil {
+			if err, ok := err.(*net.OpError); ok && err.Timeout() {
+				return fmt.Errorf("Tried connecting to SSH but timed out %s", err)
+			}
+			return err
+		}
+	}
+
+	return nil
+}