Add jenkins public key to instance

Author: davidz627

hi

@@ -44,11 +44,11 @@ func (n *E2ERemote) SetupTestPackage(tardir string) error {
 
 	// TODO(dyzz): build the gce driver tests instead.
 
-	cmd := exec.Command("ginkgo", "build", "test/e2e")
+	cmd := exec.Command("go", "test", "-c", "sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/test/e2e", "-o", "test/e2e/e2e.test")
 	err := cmd.Run()
 
 	if err != nil {
-		return fmt.Errorf("Failed to ginkgo build test: %v", err)
+		return fmt.Errorf("Failed to build test: %v", err)
 	}
 
 	cmd = exec.Command("cp", "test/e2e/e2e.test", "bin")

test/remote/remote/e2e.go

@@ -19,6 +19,7 @@ package remote
 import (
 	"flag"
 	"fmt"
+	"os"
 	"os/exec"
 	"os/user"
 	"strings"
@@ -69,6 +70,8 @@ func GetHostnameOrIP(hostname string) string {
 	}
 	if *sshUser != "" {
 		host = fmt.Sprintf("%s@%s", *sshUser, host)
+	} else if _, ok := os.LookupEnv("JENKINS_GCE_SSH_PRIVATE_KEY_FILE"); ok {
+		host = fmt.Sprintf("%s@%s", "prow", host)
 	}
 	return host
 }
@@ -92,7 +95,10 @@ func SSHNoSudo(host string, cmd ...string) (string, error) {
 
 // runSSHCommand executes the ssh or scp command, adding the flag provided --ssh-options
 func runSSHCommand(cmd string, args ...string) (string, error) {
-	if *sshKey != "" {
+	if pk, ok := os.LookupEnv("JENKINS_GCE_SSH_PRIVATE_KEY_FILE"); ok {
+		glog.Infof("Running on Jenkins, using special private key file at %v", pk)
+		args = append([]string{"-i", pk}, args...)
+	} else if *sshKey != "" {
 		args = append([]string{"-i", *sshKey}, args...)
 	} else if key, found := sshDefaultKeyMap[*sshEnv]; found {
 		args = append([]string{"-i", key}, args...)

test/remote/remote/ssh.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
 	"net/http"
 	"os"
@@ -320,8 +321,20 @@ func createInstance(serviceAccount string) (string, error) {
 		glog.Infof("Compute service GOT instance %v: %#v", i.Name, gotInstance)
 	}
 
+	pubkey, ok := os.LookupEnv("JENKINS_GCE_SSH_PUBLIC_KEY_FILE")
+	if ok {
+		glog.Infof("Running on Jenkins and JENKINS_GCE_SSH_PUBLIC_KEY_FILE set")
+		// If we're on CI add ServiceAccount Keys to the instance
+		err = addPubKeyToInstance(*project, *zone, i.Name, pubkey)
+		if err != nil {
+			return "", fmt.Errorf("could not add Jenkins Public Key %v to instance %v: %v", pubkey, i.Name, err)
+		}
+	} else {
+		glog.Infof("JENKINS_GCE_SSH_PUBLIC_KEY_FILE not set, not adding SSH Public Key to Instance")
+	}
+
 	then := time.Now()
-	err = wait.Poll(15*time.Second, 10*time.Minute, func() (bool, error) {
+	err = wait.Poll(10*time.Second, 5*time.Minute, func() (bool, error) {
 		glog.V(2).Infof("Waiting for instance %v to come up. %v elapsed", name, time.Since(then))
 		var instance *compute.Instance
 		instance, err = computeService.Instances.Get(*project, *zone, name).Do()
@@ -341,9 +354,10 @@ func createInstance(serviceAccount string) (string, error) {
 			remote.AddHostnameIP(name, externalIP)
 		}
 
-		if _, err = remote.SSHNoSudo(name, "echo"); err != nil {
+		if sshOut, err := remote.SSHNoSudo(name, "echo"); err != nil {
 			err = fmt.Errorf("Instance %v in state RUNNING but not available by SSH: %v", name, err)
 			glog.Error(err)
+			glog.Errorf("SSH output: %v", sshOut)
 			return false, nil
 		}
 
@@ -359,6 +373,58 @@ func createInstance(serviceAccount string) (string, error) {
 	return name, nil
 }
 
+func addPubKeyToInstance(project, zone, name, pubKeyFile string) error {
+	found := false
+	i, err := computeService.Instances.Get(project, zone, name).Do()
+	if err != nil {
+		return err
+	}
+	fingerprint := i.Metadata.Fingerprint
+	items := i.Metadata.Items
+	for _, item := range items {
+		if item.Key == "ssh-keys" {
+			found = true
+		}
+	}
+	newKeys := ""
+	if found {
+		// Append these to newKeys first
+		glog.Infof("Found existing public keys on instance %v", name)
+	}
+	glog.Infof("Public key file: %v", pubKeyFile)
+	publicKeyByte, err := ioutil.ReadFile(pubKeyFile)
+	if err != nil {
+		return err
+	}
+
+	publicKey := string(publicKeyByte)
+
+	// Take username and prepend it to the public key
+	tokens := strings.Split(publicKey, " ")
+	if len(tokens) != 3 {
+		return fmt.Errorf("Public key not comprised of 3 parts, instead was: %v", publicKey)
+	}
+	publicKey = strings.TrimSpace(tokens[2]) + ":" + publicKey
+
+	newKeys = newKeys + publicKey
+	glog.Infof("New Keys: %v", newKeys)
+	newMeta := &compute.Metadata{
+		Fingerprint: fingerprint,
+		Items: []*compute.MetadataItems{
+			&compute.MetadataItems{
+				Key:   "ssh-keys",
+				Value: &newKeys,
+			},
+		},
+	}
+	_, err = computeService.Instances.SetMetadata(project, zone, name, newMeta).Do()
+	if err != nil {
+		return err
+	}
+	return nil
+
+}
+
 func getexternalIP(instance *compute.Instance) string {
 	for i := range instance.NetworkInterfaces {
 		ni := instance.NetworkInterfaces[i]

test/remote/run_remote/run_remote.go

@@ -5,6 +5,7 @@ set -x
 
 readonly PKGDIR=sigs.k8s.io/gcp-compute-persistent-disk-csi-driver
 
+echo ${USER}
 echo "Testing CSI-Sanity"
 go test -timeout 30s "${PKGDIR}/test/sanity/" -run ^TestSanity$
 echo "Running E2E Tests"