lib/ssh: add parameter context to Execute method

This changes require the fork of our golang.org/x/crypto.

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/ssh/client.go b/lib/ssh/client.go
index 06243db0..d1ead996 100644
--- a/lib/ssh/client.go
+++ b/lib/ssh/client.go
@@ -6,6 +6,7 @@ package ssh
 
 import (
 	"bytes"
+	"context"
 	"fmt"
 	"io"
 	"log"
@@ -291,7 +292,7 @@ func (cl *Client) Close() (err error) {
 }
 
 // Execute a command on remote server.
-func (cl *Client) Execute(cmd string) (err error) {
+func (cl *Client) Execute(ctx context.Context, cmd string) (err error) {
 	sess, err := cl.Client.NewSession()
 	if err != nil {
 		return fmt.Errorf("ssh: NewSession: " + err.Error())
@@ -307,7 +308,7 @@ func (cl *Client) Execute(cmd string) (err error) {
 		}
 	}
 
-	err = sess.Run(cmd)
+	err = sess.RunWithContext(ctx, cmd)
 	if err != nil {
 		err = fmt.Errorf("ssh: Run %q: %s", cmd, err.Error())
 	}