mirror of https://github.com/k3s-io/k3s
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
// Copyright (c) 2017 Gorillalabs. All rights reserved.
|
|
|
|
package backend
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/juju/errors"
|
|
)
|
|
|
|
// sshSession exists so we don't create a hard dependency on crypto/ssh.
|
|
type sshSession interface {
|
|
Waiter
|
|
|
|
StdinPipe() (io.WriteCloser, error)
|
|
StdoutPipe() (io.Reader, error)
|
|
StderrPipe() (io.Reader, error)
|
|
Start(string) error
|
|
}
|
|
|
|
type SSH struct {
|
|
Session sshSession
|
|
}
|
|
|
|
func (b *SSH) StartProcess(cmd string, args ...string) (Waiter, io.Writer, io.Reader, io.Reader, error) {
|
|
stdin, err := b.Session.StdinPipe()
|
|
if err != nil {
|
|
return nil, nil, nil, nil, errors.Annotate(err, "Could not get hold of the SSH session's stdin stream")
|
|
}
|
|
|
|
stdout, err := b.Session.StdoutPipe()
|
|
if err != nil {
|
|
return nil, nil, nil, nil, errors.Annotate(err, "Could not get hold of the SSH session's stdout stream")
|
|
}
|
|
|
|
stderr, err := b.Session.StderrPipe()
|
|
if err != nil {
|
|
return nil, nil, nil, nil, errors.Annotate(err, "Could not get hold of the SSH session's stderr stream")
|
|
}
|
|
|
|
err = b.Session.Start(b.createCmd(cmd, args))
|
|
if err != nil {
|
|
return nil, nil, nil, nil, errors.Annotate(err, "Could not spawn process via SSH")
|
|
}
|
|
|
|
return b.Session, stdin, stdout, stderr, nil
|
|
}
|
|
|
|
func (b *SSH) createCmd(cmd string, args []string) string {
|
|
parts := []string{cmd}
|
|
simple := regexp.MustCompile(`^[a-z0-9_/.~+-]+$`)
|
|
|
|
for _, arg := range args {
|
|
if !simple.MatchString(arg) {
|
|
arg = b.quote(arg)
|
|
}
|
|
|
|
parts = append(parts, arg)
|
|
}
|
|
|
|
return strings.Join(parts, " ")
|
|
}
|
|
|
|
func (b *SSH) quote(s string) string {
|
|
return fmt.Sprintf(`"%s"`, s)
|
|
}
|