cleanup function will return the code which ssh returns

pkg/cmd/ssh_openstack.go

@@ -40,6 +40,7 @@ type OpenstackInstanceAttribute struct {
 
 //sshToOpenstackNode ssh to openstack node
 func sshToOpenstackNode(nodeName, path, user, pathSSKeypair string, sshPublicKey []byte, myPublicIP string) {
+	sshStatus := 0
 	a := &OpenstackInstanceAttribute{}
 	a.InstanceID = nodeName
 	var err error
@@ -68,7 +69,13 @@ func sshToOpenstackNode(nodeName, path, user, pathSSKeypair string, sshPublicKey
 	operate("openstack", "server add floating ip "+a.InstanceID+" "+a.FIP)
 	time.Sleep(5000)
 
-	defer a.cleanUpOpenstack()
+	defer a.cleanUpOpenstack(&sshStatus)
+
+	err = CheckIPPortReachable(a.FIP, "22")
+
+	if err != nil {
+		sshStatus = 1
+	}
 
 	node := user + "@" + a.FIP
 	fmt.Println("(4/5) Establishing SSH connection")
@@ -89,11 +96,14 @@ func sshToOpenstackNode(nodeName, path, user, pathSSKeypair string, sshPublicKey
 	cmd.Stderr = os.Stderr
 	if err := cmd.Run(); err != nil {
 		fmt.Println(err)
+		if exitError, ok := err.(*exec.ExitError); ok {
+			sshStatus = exitError.ExitCode()
+		}
 	}
 }
 
 //cleanUpOpenstack clean the resource added to ssh to openstack node
-func (a *OpenstackInstanceAttribute) cleanUpOpenstack() {
+func (a *OpenstackInstanceAttribute) cleanUpOpenstack(exitStatus *int) {
 	fmt.Println("")
 	fmt.Println("(5/5) Cleanup")
 
@@ -102,5 +112,5 @@ func (a *OpenstackInstanceAttribute) cleanUpOpenstack() {
 
 	fmt.Println("Delete the floating IP")
 	operate("openstack", "floating ip delete "+a.FIP)
-
+	os.Exit(*exitStatus)
 }

pkg/cmd/utils.go

@@ -21,11 +21,13 @@ import (
 	"io"
 	"io/ioutil"
 	"log"
+	"net"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"runtime"
 	"strings"
+	"time"
 
 	gardencorev1beta1 "github.com/gardener/gardener/pkg/apis/core/v1beta1"
 	gardenerlogger "github.com/gardener/gardener/pkg/logger"
@@ -348,3 +350,18 @@ func PrintoutObject(objectToPrint interface{}, writer io.Writer, outputFormat st
 	}
 	return nil
 }
+
+//CheckIPPortReachable check whether IP with port is reachable with 1 min
+func CheckIPPortReachable(ip string, port string) error {
+	timeout := time.Second * 60
+	conn, err := net.DialTimeout("tcp", net.JoinHostPort(ip, port), timeout)
+	if err != nil {
+		fmt.Println("Connecting error:", err)
+	}
+	if conn != nil {
+		defer conn.Close()
+		fmt.Printf("IP %s port %s is reachable\n", ip, port)
+		return nil
+	}
+	return fmt.Errorf("IP %s port %s is not reachable", ip, port)
+}