forked from x/icebergs
92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
package ssh
|
|
|
|
import (
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/aaa"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
"github.com/shylinux/icebergs/base/tcp"
|
|
kit "github.com/shylinux/toolkits"
|
|
"golang.org/x/crypto/ssh"
|
|
|
|
"io"
|
|
"net"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
type Winsize struct{ Height, Width, x, y uint16 }
|
|
|
|
func _ssh_exec(m *ice.Message, cmd string, arg []string, env []string, tty io.ReadWriter, done func()) {
|
|
m.Log_IMPORT(CMD, cmd, ARG, arg, ENV, env)
|
|
c := exec.Command(cmd, arg...)
|
|
// c.Env = env
|
|
|
|
c.Stdin = tty
|
|
c.Stdout = tty
|
|
c.Stderr = tty
|
|
|
|
m.Assert(c.Start())
|
|
|
|
m.Go(func() {
|
|
defer done()
|
|
c.Process.Wait()
|
|
})
|
|
}
|
|
func _ssh_close(m *ice.Message, c net.Conn, channel ssh.Channel) {
|
|
defer channel.Close()
|
|
channel.Write([]byte(m.Conf(SERVICE, "meta.goodbye")))
|
|
}
|
|
func _ssh_watch(m *ice.Message, meta map[string]string, h string, input io.Reader, output io.Writer, display io.Writer) {
|
|
r, w := io.Pipe()
|
|
bio := io.TeeReader(input, w)
|
|
m.Go(func() { io.Copy(output, r) })
|
|
|
|
i, buf := 0, make([]byte, 4096)
|
|
m.Go(func() {
|
|
for {
|
|
n, e := bio.Read(buf[i:])
|
|
if e != nil {
|
|
break
|
|
}
|
|
|
|
switch buf[i] {
|
|
case '\r', '\n':
|
|
cmd := strings.TrimSpace(string(buf[:i]))
|
|
m.Log_IMPORT(aaa.HOSTNAME, meta[aaa.HOSTNAME], aaa.USERNAME, meta[aaa.USERNAME], CMD, cmd)
|
|
m.Cmdy(mdb.INSERT, CHANNEL, kit.Keys(kit.MDB_HASH, h), mdb.LIST, kit.MDB_TYPE, CMD, kit.MDB_TEXT, cmd)
|
|
i = 0
|
|
default:
|
|
if i += n; i >= 4096 {
|
|
i = 0
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
const CHANNEL = "channel"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
CHANNEL: {Name: "channel", Help: "通道", Value: kit.Data()},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
CHANNEL: {Name: "channel hash id auto 清理", Help: "通道", Action: map[string]*ice.Action{
|
|
mdb.PRUNES: {Name: "prunes", Help: "清理", Hand: func(m *ice.Message, arg ...string) {
|
|
m.Cmdy(mdb.PRUNES, CHANNEL, "", mdb.HASH, kit.MDB_STATUS, tcp.CLOSE)
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
m.Option(mdb.FIELDS, "time,hash,status,username,hostname,hostport,tty,count")
|
|
m.Cmdy(mdb.SELECT, CHANNEL, "", mdb.HASH)
|
|
return
|
|
}
|
|
|
|
m.Option(mdb.FIELDS, kit.Select("time,id,type,text", mdb.DETAIL, len(arg) > 1))
|
|
m.Cmdy(mdb.SELECT, CHANNEL, kit.Keys(kit.MDB_HASH, arg[0]), mdb.LIST, kit.MDB_ID, arg[1:])
|
|
}},
|
|
},
|
|
}, nil)
|
|
}
|