43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
|
package sysdig
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/mic92/clusterssh"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Start(hostnames []string) (*clusterssh.Command, error) {
|
||
|
|
hosts := make([]clusterssh.Host, len(hostnames))
|
||
|
|
for i, arg := range hostnames {
|
||
|
|
host, err := clusterssh.ParseHost(arg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("invalid host '%s': %v", arg, err)
|
||
|
|
}
|
||
|
|
hosts[i] = *host
|
||
|
|
}
|
||
|
|
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
||
|
|
f1, err := os.Open(filepath.Join(dir, "capture-connections.lua"))
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to open sysdig plugin")
|
||
|
|
}
|
||
|
|
sysdigPlugin, err := ioutil.ReadAll(f1)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to read sysdig plugin: %v", err)
|
||
|
|
}
|
||
|
|
f2, err := os.Open(filepath.Join(dir, "sysdig.sh"))
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to open sysdig plugin: %v", err)
|
||
|
|
}
|
||
|
|
sysdigCommand, err := ioutil.ReadAll(f2)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to read sysdig command: %v", err)
|
||
|
|
}
|
||
|
|
cluster := clusterssh.Cluster{hosts}
|
||
|
|
cmd := cluster.Run(string(sysdigCommand), sysdigPlugin)
|
||
|
|
return &cmd, nil
|
||
|
|
}
|