初次提交
This commit is contained in:
Mob2003
+837
@@ -0,0 +1,837 @@
|
||||
package server
|
||||
|
||||
/*
|
||||
*高级shell功能
|
||||
*node节点管理、remoteShell远程shell，config配置管理
|
||||
*/
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"os/exec"
|
||||
"rakshasa/aes"
|
||||
"rakshasa/cert"
|
||||
"rakshasa/common"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"github.com/abiosoft/readline"
|
||||
"github.com/creack/pty"
|
||||
"github.com/dlclark/regexp2"
|
||||
"github.com/luyu6056/ishell"
|
||||
"golang.org/x/text/encoding/simplifiedchinese"
|
||||
"golang.org/x/text/transform"
|
||||
)
|
||||
|
||||
var (
|
||||
shellMapLock sync.Mutex
|
||||
)
|
||||
|
||||
type StartCmdParam struct {
|
||||
Param string
|
||||
Size *pty.Winsize
|
||||
}
|
||||
type remoteCmd struct {
|
||||
cmdStatus int32
|
||||
cmd *exec.Cmd
|
||||
id uint32
|
||||
stdin io.WriteCloser
|
||||
inChan chan []byte
|
||||
translate func(in []byte) ([]byte, error)
|
||||
ping, pong int64
|
||||
}
|
||||
|
||||
func init() {
|
||||
|
||||
configShell := cliInit()
|
||||
configShell.SetPrompt("rakshasa\\config>")
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "info",
|
||||
Help: "打印当前配置",
|
||||
Func: func(c *ishell.Context) {
|
||||
c.Println("当前节点", currentNode.uuid)
|
||||
c.Println("上级节点地址", currentConfig.DstNode)
|
||||
c.Println("通讯密码", currentConfig.Password)
|
||||
c.Println("监听端口", currentConfig.Port)
|
||||
c.Println("监听IP", currentConfig.ListenIp)
|
||||
c.Println("禁止额外连接", currentConfig.Limit)
|
||||
c.Println("配置文件名", currentConfig.FileName)
|
||||
if currentConfig.FileSave {
|
||||
c.Println("当前配置：已写入文件")
|
||||
} else {
|
||||
c.Println("当前配置：未写入文件")
|
||||
}
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "save",
|
||||
Help: "保存文件",
|
||||
Func: func(c *ishell.Context) {
|
||||
if err := ConfigSave(); err == nil {
|
||||
c.Println("写入成功")
|
||||
} else {
|
||||
c.Println("保存失败", err.Error())
|
||||
}
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "d",
|
||||
Help: "修改上级节点地址，格式为 ip:端口 多个节点以,隔开 注意：不会立刻连接设置节点， 当发生 节点掉线重连 时候会连接该地址",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误，格式为 ip:端口 多个节点以,隔开 如 d 192.168.1.1:8883,192.168.1.2:8883")
|
||||
return
|
||||
}
|
||||
dstNode, err := common.ResolveTCPAddr(c.Args[0])
|
||||
if err != nil {
|
||||
c.Println("参数错误，格式为 ip:端口 多个节点以,隔开 如 d 192.168.1.1:8883,192.168.1.2:8883")
|
||||
return
|
||||
}
|
||||
currentConfig.DstNode = dstNode
|
||||
currentConfig.FileSave = false
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "password",
|
||||
Help: "修改通讯密码,立即生效",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误，格式为 password \"123456\"")
|
||||
return
|
||||
}
|
||||
c.Println(c.Args)
|
||||
currentConfig.Password = c.Args[0]
|
||||
currentConfig.FileSave = false
|
||||
aes.Key = aes.MD5_B(currentConfig.Password + string(cert.PublicKey[:16]))
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "port",
|
||||
Help: "修改监听端口,立即生效",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误，格式为 port 8883")
|
||||
return
|
||||
}
|
||||
port, _ := strconv.Atoi(c.Args[0])
|
||||
if port <= 0 || port > 65535 {
|
||||
c.Println("参数错误，端口范围是1-65535")
|
||||
return
|
||||
}
|
||||
c.Println("正在关闭server监听")
|
||||
if currentNode.listen != nil {
|
||||
currentNode.listen.Close()
|
||||
currentNode.listen = nil
|
||||
}
|
||||
currentConfig.Port = port
|
||||
currentNode.port = port
|
||||
currentConfig.FileSave = false
|
||||
currentNode.broadcastNode()
|
||||
StartServer(port)
|
||||
},
|
||||
})
|
||||
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "ip",
|
||||
Help: "修改本节点连接ip，当其他节点进行额外连接时候，优先使用此ip连接, 多个ip以,隔开",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
currentConfig.ListenIp = strings.Split(c.Args[0], ",")
|
||||
currentNode.mainIp = currentConfig.ListenIp
|
||||
currentConfig.FileSave = false
|
||||
currentNode.broadcastNode()
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "limit",
|
||||
Help: "修改本节点Limit设置，使用方法 limit true",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
currentConfig.Limit = c.Args[0] == "true"
|
||||
currentConfig.FileSave = false
|
||||
},
|
||||
})
|
||||
configShell.AddCmd(&ishell.Cmd{
|
||||
Name: "f",
|
||||
Help: "修改配置文件名，使用方法 f config.yaml",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
currentConfig.FileName = c.Args[0]
|
||||
currentConfig.FileSave = false
|
||||
},
|
||||
})
|
||||
|
||||
rootCli.AddCmd(&ishell.Cmd{
|
||||
Name: "config",
|
||||
Help: "配置管理",
|
||||
Func: func(c *ishell.Context) {
|
||||
configShell.Run()
|
||||
},
|
||||
})
|
||||
remoteShell := cliInit()
|
||||
|
||||
remoteShell.SetPrompt("rakshasa\\remoteshell>")
|
||||
|
||||
fileShell := ishell.New()
|
||||
remoteShell.AddCmd(&ishell.Cmd{
|
||||
Name: "file",
|
||||
Help: "连到节点进行文件管理，参数为id或者uuid",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
|
||||
workN, _ := getNode(c.Args[0])
|
||||
if workN == nil {
|
||||
c.Println("无法连接节点", c.Args[0])
|
||||
return
|
||||
}
|
||||
|
||||
if workN != nil {
|
||||
fileShell.Set("node", workN)
|
||||
result := make(chan interface{}, 1)
|
||||
id := workN.storeQuery(result)
|
||||
workN.Write(common.CMD_PWD, id, nil)
|
||||
select {
|
||||
case pwd := <-result:
|
||||
workN.deleteQuery(id)
|
||||
pwd = strings.ReplaceAll(pwd.(string), "\\", "/")
|
||||
fileShell.Set("pwd", pwd)
|
||||
fileShell.SetPrompt(workN.uuid + " " + pwd.(string) + ">")
|
||||
fileShell.Run()
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
workN.deleteQuery(id)
|
||||
c.Println("连接", c.Args[0], "超时")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
},
|
||||
})
|
||||
fileShell.AddCmd(&ishell.Cmd{
|
||||
Name: "dir",
|
||||
Help: "打印当前目录文件",
|
||||
Func: func(c *ishell.Context) {
|
||||
pwd := fileShell.Get("pwd")
|
||||
|
||||
n := c.Get("node").(*node)
|
||||
resChan := make(chan interface{}, 1)
|
||||
id := n.storeQuery(resChan)
|
||||
n.Write(common.CMD_DIR, id, []byte(pwd.(string)))
|
||||
select {
|
||||
case res := <-resChan:
|
||||
n.deleteQuery(id)
|
||||
c.Println(res)
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
n.deleteQuery(id)
|
||||
c.Println("dir time out")
|
||||
}
|
||||
},
|
||||
})
|
||||
fileShell.AddCmd(&ishell.Cmd{
|
||||
Name: "cd",
|
||||
Help: "切换工作目录",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
dir := c.Args[0]
|
||||
pwd := fileShell.Get("pwd").(string)
|
||||
n := c.Get("node").(*node)
|
||||
|
||||
if strings.Contains(dir, ":/") || dir[0] == '/' || dir == "~" {
|
||||
pwd = dir
|
||||
} else {
|
||||
pwd += "/" + dir
|
||||
pwd = strings.TrimRight(realpath(pwd), "/")
|
||||
}
|
||||
|
||||
resChan := make(chan interface{}, 1)
|
||||
id := n.storeQuery(resChan)
|
||||
n.Write(common.CMD_CD, id, []byte(pwd))
|
||||
|
||||
select {
|
||||
case res := <-resChan:
|
||||
n.deleteQuery(id)
|
||||
if err, ok := res.(error); ok {
|
||||
c.Println(err.Error())
|
||||
} else {
|
||||
pwd = res.(string)
|
||||
fileShell.Set("pwd", pwd)
|
||||
c.SetPrompt(n.uuid + " " + pwd + ">")
|
||||
}
|
||||
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
n.deleteQuery(id)
|
||||
c.Println("dir time out")
|
||||
}
|
||||
},
|
||||
})
|
||||
fileShell.AddCmd(&ishell.Cmd{
|
||||
Name: "upload",
|
||||
Help: "上传文件 ，upload 本地文件 远程目录(为空传到工作目录)",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 && len(c.Args) != 2 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
s, err := os.Stat(c.Args[0])
|
||||
if err != nil {
|
||||
c.Println("打开本地文件", c.Args[0], "错误 ", err)
|
||||
return
|
||||
}
|
||||
f, err := os.Open(c.Args[0])
|
||||
if err != nil {
|
||||
c.Println("打开本地文件", c.Args[0], "错误 ", err)
|
||||
return
|
||||
}
|
||||
defer f.Close()
|
||||
pwd := fileShell.Get("pwd").(string) + "/"
|
||||
n := c.Get("node").(*node)
|
||||
|
||||
if len(c.Args) == 2 {
|
||||
pwd = c.Args[1]
|
||||
}
|
||||
pwd = strings.ReplaceAll(pwd, "\\", "/")
|
||||
c.Args[0] = strings.ReplaceAll(c.Args[0], "\\", "/")
|
||||
i := strings.LastIndex(c.Args[0], "/")
|
||||
if i == -1 {
|
||||
i = 0
|
||||
}
|
||||
|
||||
if pwd[len(pwd)-1] == '/' {
|
||||
pwd += c.Args[0][i:]
|
||||
}
|
||||
i = strings.LastIndex(pwd, "/")
|
||||
if i == -1 {
|
||||
i = 0
|
||||
}
|
||||
filename := pwd[i+1:]
|
||||
dir := pwd[:i]
|
||||
dir = strings.TrimRight(realpath(dir), "/") + "/"
|
||||
pwd = dir + filename
|
||||
resChan := make(chan interface{}, 9999) //避免收消息阻塞
|
||||
|
||||
filereadChan := make(chan []byte, 10)
|
||||
|
||||
upload := func() {
|
||||
for i := 0; i < 10; i++ {
|
||||
buf := make([]byte, common.MAX_PACKAGE-len(pwd)-9)
|
||||
n, err := f.Read(buf)
|
||||
if err != nil {
|
||||
if err == io.EOF {
|
||||
|
||||
return
|
||||
}
|
||||
resChan <- err
|
||||
c.Println("读取文件", c.Args[0], "错误", err)
|
||||
return
|
||||
}
|
||||
|
||||
filereadChan <- buf[:n]
|
||||
}
|
||||
}
|
||||
|
||||
offset := 0
|
||||
be := len(pwd) + 1
|
||||
id := n.storeQuery(resChan)
|
||||
defer n.deleteQuery(id)
|
||||
b := []byte(pwd)
|
||||
b = append(b, 0, 0, 0, 0, 0, 0, 0, 0, 0)
|
||||
c.ProgressBar().Start()
|
||||
go upload()
|
||||
var resnum int
|
||||
for {
|
||||
select {
|
||||
case data := <-filereadChan:
|
||||
|
||||
b[be] = byte(offset)
|
||||
b[be+1] = byte(offset >> 8)
|
||||
b[be+2] = byte(offset >> 16)
|
||||
b[be+3] = byte(offset >> 24)
|
||||
b[be+4] = byte(offset >> 32)
|
||||
b[be+5] = byte(offset >> 40)
|
||||
b[be+6] = byte(offset >> 48)
|
||||
b[be+7] = byte(offset >> 56)
|
||||
offset += len(data)
|
||||
n.Write(common.CMD_UPLOAD, id, append(b, data...))
|
||||
case res := <-resChan:
|
||||
switch v := res.(type) {
|
||||
case error:
|
||||
c.ProgressBar().Stop()
|
||||
c.Println("上传失败", res)
|
||||
return
|
||||
case int64:
|
||||
resnum++
|
||||
i := v * 100 / s.Size()
|
||||
c.ProgressBar().Suffix(fmt.Sprint(" ", i, "%"))
|
||||
c.ProgressBar().Progress(int(i))
|
||||
if v == s.Size() {
|
||||
c.ProgressBar().Stop()
|
||||
c.Println(c.Args[0], "上传成功")
|
||||
return
|
||||
}
|
||||
if resnum >= 5 {
|
||||
go upload()
|
||||
resnum -= 10
|
||||
}
|
||||
default:
|
||||
c.Println("协议错误")
|
||||
return
|
||||
}
|
||||
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
c.ProgressBar().Stop()
|
||||
c.Println("upload time out")
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
},
|
||||
})
|
||||
fileShell.AddCmd(&ishell.Cmd{
|
||||
Name: "download",
|
||||
Help: "下载文件 ，download 远程文件 本地目录(为空本地执行目录)",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 && len(c.Args) != 2 {
|
||||
c.Println("参数错误")
|
||||
return
|
||||
}
|
||||
pwd := fileShell.Get("pwd").(string)
|
||||
n := c.Get("node").(*node)
|
||||
file := c.Args[0]
|
||||
file = strings.ReplaceAll(file, "\\", "/")
|
||||
|
||||
if strings.Contains(file, ":/") || file[0] == '/' {
|
||||
pwd = file
|
||||
} else {
|
||||
pwd += "/" + file
|
||||
|
||||
}
|
||||
i := strings.LastIndex(pwd, "/")
|
||||
if i == -1 {
|
||||
i = 0
|
||||
}
|
||||
filename := pwd[i+1:]
|
||||
dir := pwd[:i]
|
||||
dir = strings.TrimRight(realpath(dir), "/") + "/"
|
||||
mydir, err := os.Getwd()
|
||||
local := "./" + filename
|
||||
if err == nil {
|
||||
local = mydir + "/" + filename
|
||||
}
|
||||
|
||||
if len(c.Args) == 2 {
|
||||
s, err := os.Stat(c.Args[1])
|
||||
if err == nil {
|
||||
if s.IsDir() {
|
||||
local = strings.TrimRight(c.Args[1], "/") + "/" + filename
|
||||
} else {
|
||||
local = c.Args[1]
|
||||
}
|
||||
} else {
|
||||
local = c.Args[1]
|
||||
}
|
||||
}
|
||||
pwd = dir + filename
|
||||
|
||||
result := make(chan interface{}, 999)
|
||||
id := n.storeQuery(result)
|
||||
defer n.deleteQuery(id)
|
||||
b := []byte(pwd)
|
||||
b = append(b, []byte{0, 0, 0, 0, 0, 0, 0, 0, 0}...)
|
||||
total := int64(-1)
|
||||
be := len(pwd) + 1
|
||||
b[be] = byte(total)
|
||||
b[be+1] = byte(total >> 8)
|
||||
b[be+2] = byte(total >> 16)
|
||||
b[be+3] = byte(total >> 24)
|
||||
b[be+4] = byte(total >> 32)
|
||||
b[be+5] = byte(total >> 40)
|
||||
b[be+6] = byte(total >> 48)
|
||||
b[be+7] = byte(total >> 56)
|
||||
n.Write(common.CMD_DOWNLOAD, id, b)
|
||||
c.ProgressBar().Start()
|
||||
size := int64(0)
|
||||
resnum := 0
|
||||
total = 0
|
||||
var f *os.File
|
||||
for {
|
||||
select {
|
||||
case res := <-result:
|
||||
switch v := res.(type) {
|
||||
case error:
|
||||
c.ProgressBar().Stop()
|
||||
c.Println("下载失败", res)
|
||||
return
|
||||
case int64:
|
||||
var err error
|
||||
size = v
|
||||
f, err = os.OpenFile(local, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
|
||||
if err != nil {
|
||||
c.Println("本地文件 ", local, "写入失败", err.Error())
|
||||
return
|
||||
}
|
||||
defer f.Close()
|
||||
case []byte:
|
||||
if f == nil {
|
||||
c.Println("本地文件 ", local, "不可写入")
|
||||
return
|
||||
}
|
||||
resnum++
|
||||
num, err := f.Write(v)
|
||||
if err != nil {
|
||||
c.Println("本地文件 ", local, "写入失败", err.Error())
|
||||
return
|
||||
}
|
||||
if num != len(v) {
|
||||
c.Println("本地文件 ", local, "写入失败,写入量不符")
|
||||
return
|
||||
}
|
||||
total += int64(num)
|
||||
i := total * 100 / size
|
||||
c.ProgressBar().Suffix(fmt.Sprint(" ", i, "%"))
|
||||
c.ProgressBar().Progress(int(i))
|
||||
if total == size {
|
||||
c.ProgressBar().Stop()
|
||||
c.Println(c.Args[0], "下载成功 文件保存到", local)
|
||||
return
|
||||
}
|
||||
if resnum == 10 {
|
||||
resnum -= 10
|
||||
b[be] = byte(total)
|
||||
b[be+1] = byte(total >> 8)
|
||||
b[be+2] = byte(total >> 16)
|
||||
b[be+3] = byte(total >> 24)
|
||||
b[be+4] = byte(total >> 32)
|
||||
b[be+5] = byte(total >> 40)
|
||||
b[be+6] = byte(total >> 48)
|
||||
b[be+7] = byte(total >> 56)
|
||||
n.Write(common.CMD_DOWNLOAD, id, b)
|
||||
}
|
||||
default:
|
||||
c.Println("协议错误")
|
||||
return
|
||||
}
|
||||
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
c.ProgressBar().Stop()
|
||||
c.Println("upload time out")
|
||||
return
|
||||
}
|
||||
}
|
||||
},
|
||||
})
|
||||
|
||||
remoteShell.AddCmd(&ishell.Cmd{
|
||||
Name: "new",
|
||||
Help: "与一个或者多个节点连接，使用方法 new ip:端口 多个地址以,间隔 如1080 127.0.0.1:1081,127.0.0.1:1082",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) != 1 {
|
||||
c.Println("参数错误，使用方法 connect ip:端口")
|
||||
return
|
||||
}
|
||||
for _, addr := range strings.Split(c.Args[0], ",") {
|
||||
_, err := connectNew(addr)
|
||||
if err != nil {
|
||||
c.Println("连接", addr, "失败", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
},
|
||||
})
|
||||
remoteShell.AddCmd(&ishell.Cmd{
|
||||
Name: "shell",
|
||||
Help: "反弹shell 使用方法 shell id/uuid 启动参数 ，启动参数可为空，win默认启动cmd，linux默认启动bash， 如 shell 1 powershell 。 shell 1 zsh",
|
||||
Func: func(c *ishell.Context) {
|
||||
if len(c.Args) < 1 {
|
||||
c.Println("参数错误，例子 shell 1 powershell")
|
||||
return
|
||||
}
|
||||
param := ""
|
||||
if len(c.Args) == 2 {
|
||||
param = c.Args[1]
|
||||
}
|
||||
n, _ := getNode(c.Args[0])
|
||||
if n == nil {
|
||||
c.Println("无法连接节点", c.Args[0])
|
||||
return
|
||||
}
|
||||
res := make(chan interface{}, 999)
|
||||
id := n.storeQuery(res)
|
||||
|
||||
defer n.deleteQuery(id)
|
||||
p := StartCmdParam{
|
||||
Param: param,
|
||||
Size: common.GetSize(),
|
||||
}
|
||||
|
||||
b, _ := json.Marshal(p)
|
||||
n.Write(common.CMD_SHELL, id, b)
|
||||
s := &remoteCmd{
|
||||
cmd: nil,
|
||||
stdin: nil,
|
||||
inChan: make(chan []byte, 999),
|
||||
translate: func(in []byte) ([]byte, error) { return in, nil },
|
||||
pong: time.Now().Unix(),
|
||||
}
|
||||
|
||||
select {
|
||||
case i := <-res:
|
||||
switch v := i.(type) {
|
||||
case error:
|
||||
c.Println("启动shell失败，错误", v.Error())
|
||||
case []byte:
|
||||
data := v
|
||||
|
||||
s.id = uint32(data[0]) | uint32(data[1])<<8 | uint32(data[2])<<16 | uint32(data[3])<<24
|
||||
switch data[4] {
|
||||
case 0: //windows
|
||||
if string(data[len(data)-6:]) == string([]byte{32, 57, 51, 54, 13, 10}) { //活动代码页: 936
|
||||
//gbk转utf8
|
||||
s.translate = func(in []byte) ([]byte, error) {
|
||||
reader := transform.NewReader(bytes.NewReader(in), simplifiedchinese.GBK.NewDecoder())
|
||||
d, e := ioutil.ReadAll(reader)
|
||||
if e != nil {
|
||||
return nil, e
|
||||
}
|
||||
return d, nil
|
||||
}
|
||||
}
|
||||
case 1: //linux
|
||||
if runtime.GOOS == "windows" {
|
||||
if !common.EnableTermVt {
|
||||
s.translate = func(in []byte) ([]byte, error) {
|
||||
if in[0] == 27 {
|
||||
r, _ := regexp2.Compile(`\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])`, 0)
|
||||
res, _ := r.Replace(string(in), "", 0, -1)
|
||||
return []byte(res), nil
|
||||
}
|
||||
return in, nil
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
atomic.CompareAndSwapInt32(&s.cmdStatus, 0, 1)
|
||||
}
|
||||
case <-time.After(common.CMD_TIMEOUT):
|
||||
c.Println("启动shell失败，超时")
|
||||
return
|
||||
}
|
||||
|
||||
n.shellMap.Store(s.id, s)
|
||||
r, _ := readline.NewEx(&readline.Config{FuncIsTerminal: func() bool { return false }, ForcePrint: true})
|
||||
defer func() {
|
||||
n.shellMap.Delete(s.id)
|
||||
atomic.StoreInt32(&s.cmdStatus, -1)
|
||||
c.Println("请按回车键退出")
|
||||
r.Close()
|
||||
|
||||
}()
|
||||
|
||||
go func() {
|
||||
|
||||
for {
|
||||
|
||||
switch s.cmdStatus {
|
||||
case 1:
|
||||
|
||||
input, err := r.ReadlineEx()
|
||||
if err != nil {
|
||||
if err != readline.ErrInterrupt {
|
||||
res <- err
|
||||
return
|
||||
}
|
||||
if s.cmdStatus == 1 {
|
||||
|
||||
n.Write(common.CMD_SHELL_DATA, s.id, []byte{03})
|
||||
}
|
||||
}
|
||||
if s.cmdStatus == 1 {
|
||||
|
||||
n.Write(common.CMD_SHELL_DATA, s.id, []byte(input+"\n"))
|
||||
}
|
||||
|
||||
case 0:
|
||||
time.Sleep(time.Millisecond * 100)
|
||||
case -1:
|
||||
return
|
||||
}
|
||||
|
||||
}
|
||||
}()
|
||||
tick := time.NewTicker(common.CMD_TIMEOUT / 2)
|
||||
for {
|
||||
|
||||
select {
|
||||
case b := <-s.inChan:
|
||||
s.pong = time.Now().Unix()
|
||||
if len(b) > 0 {
|
||||
b, err := s.translate(b)
|
||||
if err != nil {
|
||||
c.Println("shell 运行失败", err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Print(string(b))
|
||||
}
|
||||
|
||||
case v := <-res:
|
||||
if err, ok := v.(error); ok {
|
||||
if err.Error() != "退出shell" {
|
||||
c.Println("运行shell", param, "失败", err)
|
||||
}
|
||||
|
||||
} else {
|
||||
c.Println("无法处理消息", v)
|
||||
}
|
||||
return
|
||||
case <-tick.C:
|
||||
s.ping = time.Now().Unix()
|
||||
if s.ping-s.pong > int64(common.CMD_TIMEOUT/time.Second) {
|
||||
c.Println("shell time out")
|
||||
return
|
||||
}
|
||||
n.Write(common.CMD_SHELL_DATA, s.id, nil)
|
||||
}
|
||||
}
|
||||
},
|
||||
})
|
||||
rootCli.AddCmd(&ishell.Cmd{
|
||||
Name: "remoteshell",
|
||||
Help: "远程shell",
|
||||
Func: func(c *ishell.Context) {
|
||||
remoteShell.Run()
|
||||
},
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// 打印节点
|
||||
func printNodes(c *ishell.Context) {
|
||||
l := clientLock.RLock()
|
||||
defer l.RUnlock()
|
||||
var list []*node
|
||||
for _, n := range nodeMap {
|
||||
list = append(list, n)
|
||||
}
|
||||
orderNode(list)
|
||||
c.Println("ID UUID HostName GOOS IP listenIP")
|
||||
c.Println("-----------------------------------------------------------------------------------------------------------------------------")
|
||||
for k, n := range list {
|
||||
n.id = k + 1
|
||||
hostname := bytes.Repeat([]byte(" "), 22)
|
||||
copy(hostname, n.hostName)
|
||||
ip := bytes.Repeat([]byte(" "), 23)
|
||||
if n.uuid == currentNode.uuid {
|
||||
|
||||
copy(ip, "(localhost)"+":"+strconv.Itoa(n.port))
|
||||
} else {
|
||||
copy(ip, n.addr+":"+strconv.Itoa(n.port))
|
||||
}
|
||||
|
||||
var s []string
|
||||
for _, ip := range n.mainIp {
|
||||
if ip != "" {
|
||||
s = append(s, ip+":"+strconv.Itoa(n.port))
|
||||
}
|
||||
}
|
||||
listenip := strings.Join(s, ",")
|
||||
goos := bytes.Repeat([]byte(" "), 11)
|
||||
copy(goos, n.goos)
|
||||
c.Printf("%2d %s %s %s %s %s\n", n.id, n.uuid, hostname, goos, ip, listenip)
|
||||
}
|
||||
}
|
||||
|
||||
func realpath(path string) string {
|
||||
|
||||
path_s := strings.Split(path, "/")
|
||||
realpath := []string{}
|
||||
if len(path_s) == 0 {
|
||||
return "error"
|
||||
}
|
||||
for _, value := range path_s {
|
||||
|
||||
if value == ".." {
|
||||
k := len(realpath)
|
||||
kk := k - 1
|
||||
realpath = append(realpath[:kk], realpath[k:]...)
|
||||
} else {
|
||||
realpath = append(realpath, value)
|
||||
}
|
||||
}
|
||||
|
||||
return strings.Join(realpath, "/")
|
||||
}
|
||||
func printConn() {
|
||||
connMap.Range(func(key, value interface{}) bool {
|
||||
fmt.Println(key)
|
||||
return true
|
||||
})
|
||||
}
|
||||
func getNode(arg string) (*node, error) {
|
||||
l := clientLock.RLock()
|
||||
|
||||
id, err := strconv.Atoi(arg)
|
||||
|
||||
if err == nil {
|
||||
for _, n := range nodeMap {
|
||||
if n.id == id && n.uuid != currentNode.uuid {
|
||||
l.RUnlock()
|
||||
return n, nil
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if v, ok := nodeMap[arg]; ok && v.uuid != currentNode.uuid {
|
||||
l.RUnlock()
|
||||
return v, nil
|
||||
}
|
||||
}
|
||||
l.RUnlock()
|
||||
|
||||
return connectNew(arg)
|
||||
}
|
||||
func getNodeWithCurrentNode(arg string) (*node, error) {
|
||||
l := clientLock.RLock()
|
||||
|
||||
id, err := strconv.Atoi(arg)
|
||||
|
||||
if err == nil {
|
||||
for _, n := range nodeMap {
|
||||
if n.id == id {
|
||||
l.RUnlock()
|
||||
return n, nil
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if v, ok := nodeMap[arg]; ok {
|
||||
l.RUnlock()
|
||||
return v, nil
|
||||
}
|
||||
}
|
||||
l.RUnlock()
|
||||
|
||||
return connectNew(arg)
|
||||
}
|
||||
Reference in New Issue
Block a user