From aa19c5b9ae38658e6c141c0ea466616c19e6520a Mon Sep 17 00:00:00 2001 From: tedteng Date: Mon, 2 Nov 2020 14:32:24 +0800 Subject: [PATCH] add passing custom arguments --- pkg/cmd/ssh_aws.go | 3 +++ pkg/cmd/ssh_azure.go | 3 +++ pkg/cmd/ssh_gcp.go | 3 +++ 3 files changed, 9 insertions(+) diff --git a/pkg/cmd/ssh_aws.go b/pkg/cmd/ssh_aws.go index d97c4a5d6..b12a5f275 100644 --- a/pkg/cmd/ssh_aws.go +++ b/pkg/cmd/ssh_aws.go @@ -90,6 +90,9 @@ func sshToAWSNode(nodeName, path, user, pathSSKeypair string, sshPublicKey []byt args = append([]string{"-vvv"}, args...) } + command := os.Args[3:] + args = append(args, command...) + cmd := exec.Command("ssh", args...) cmd.Stdout = os.Stdout cmd.Stdin = os.Stdin diff --git a/pkg/cmd/ssh_azure.go b/pkg/cmd/ssh_azure.go index 33673a007..7772dd48a 100644 --- a/pkg/cmd/ssh_azure.go +++ b/pkg/cmd/ssh_azure.go @@ -76,6 +76,9 @@ func sshToAZNode(nodeName, path, user, pathSSKeypair string, sshPublicKey []byte args = append([]string{"-vvv"}, args...) } + command := os.Args[3:] + args = append(args, command...) + cmd := exec.Command("ssh", args...) cmd.Stdout = os.Stdout cmd.Stdin = os.Stdin diff --git a/pkg/cmd/ssh_gcp.go b/pkg/cmd/ssh_gcp.go index d67226fbf..ba2472f34 100644 --- a/pkg/cmd/ssh_gcp.go +++ b/pkg/cmd/ssh_gcp.go @@ -76,6 +76,9 @@ func sshToGCPNode(nodeName, path, user, pathSSKeypair string, sshPublicKey []byt args = append([]string{"-vvv"}, args...) } + command := os.Args[3:] + args = append(args, command...) + cmd := exec.Command("ssh", args...) cmd.Stdout = os.Stdout cmd.Stdin = os.Stdin