56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package ssh
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
type SSHResultHandler func(string, string, error)
|
|
|
|
func SSHResultToString(result string) string {
|
|
if len(result) > 0 {
|
|
result = result[:len(result)-1]
|
|
}
|
|
return result
|
|
}
|
|
|
|
func SSHResultToStringHandler(handler SSHResultHandler) SSHResultHandler {
|
|
return func(addr string, result string, err error) {
|
|
handler(addr, SSHResultToString(result), err)
|
|
}
|
|
}
|
|
|
|
func (m *Manager) RunOn(addr net.TCPAddr, cmd string) (string, error) {
|
|
client, err := m.ConnectTo(addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer client.Close()
|
|
return m.run(addr.IP.String(), client, cmd)
|
|
}
|
|
|
|
func (m *Manager) run(host string, client *ssh.Client, cmd string) (string, error) {
|
|
session, err := client.NewSession()
|
|
defer session.Close()
|
|
|
|
if err != nil {
|
|
log.Warnf("can not create session on %s: %s", host, err)
|
|
return "", err
|
|
}
|
|
buffer := &bytes.Buffer{}
|
|
session.Stdout = buffer
|
|
if err != nil {
|
|
log.Warnf("can not create pipe for run on %s: %s", host, err)
|
|
return "", err
|
|
}
|
|
err = session.Run(cmd)
|
|
if err != nil {
|
|
log.Debugf("could not run %s on %s: %s", cmd, host, err)
|
|
return "", err
|
|
}
|
|
return buffer.String(), nil
|
|
}
|