60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
|
package ssh
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"golang.org/x/crypto/ssh"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/freifunkmanager/lib/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SSHRunResultHandler func([]byte, error)
|
||
|
|
|
||
|
|
func (m *Manager) RunEverywhere(cmd string, handler SSHRunResultHandler) {
|
||
|
|
for host, client := range m.clients {
|
||
|
|
result, err := m.run(host, client, cmd)
|
||
|
|
handler(result, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Manager) RunOn(host net.IP, cmd string) ([]byte, error) {
|
||
|
|
client := m.ConnectTo(host)
|
||
|
|
return m.run(host.String(), client, cmd)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Manager) run(host string, client *ssh.Client, cmd string) ([]byte, error) {
|
||
|
|
session, err := client.NewSession()
|
||
|
|
defer session.Close()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Log.Warnf("can not create session on %s: %s", host, err)
|
||
|
|
delete(m.clients, host)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
stdout, err := session.StdoutPipe()
|
||
|
|
buffer := &bytes.Buffer{}
|
||
|
|
go io.Copy(buffer, stdout)
|
||
|
|
if err != nil {
|
||
|
|
log.Log.Warnf("can not create pipe for run on %s: %s", host, err)
|
||
|
|
delete(m.clients, host)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
err = session.Run(cmd)
|
||
|
|
if err != nil {
|
||
|
|
log.Log.Warnf("could not run %s on %s: %s", cmd, host, err)
|
||
|
|
delete(m.clients, host)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var result []byte
|
||
|
|
for {
|
||
|
|
b, err := buffer.ReadByte()
|
||
|
|
if err != nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
result = append(result, b)
|
||
|
|
}
|
||
|
|
return result, nil
|
||
|
|
}
|