zishuo/natpass
1
0
Fork 0
mirror of https://github.com/lwch/natpass synced 2025-10-28 14:21:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改server处理逻辑，支持shell_created消息

Browse Source
This commit is contained in:
lwch
2021-09-28 11:48:14 +08:00
parent dd7fc5ff07
commit 69a7dad72e
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
code/server/handler/handler.go
View File

@@ -126,6 +126,7 @@ func (h *Handler) onMessage(from *client, conn *network.Conn, msg *network.Msg)
case network.Msg_disconnect: case network.Msg_disconnect:
case network.Msg_forward: case network.Msg_forward:
case network.Msg_shell_create: case network.Msg_shell_create:
case network.Msg_shell_created:
case network.Msg_shell_resize: case network.Msg_shell_resize:
case network.Msg_shell_data: case network.Msg_shell_data:
case network.Msg_shell_close: case network.Msg_shell_close:
@@ -172,6 +173,17 @@ func (h *Handler) removeLink(id string, from, to *client) {
id, from.parent.id, from.idx, to.parent.id, to.idx) id, from.parent.id, from.idx, to.parent.id, to.idx)
} }
func (h *Handler) responseLink(id string, ok bool, msg string, from, to *client) {
if ok {
logging.Info("link %s from %s-%d to %s-%d connect successed",
id, from.parent.id, from.idx, to.parent.id, to.idx)
} else {
logging.Info("link %s from %s-%d to %s-%d connect failed, %s",
id, from.parent.id, from.idx, to.parent.id, to.idx, msg)
// TODO: remove link?
}
}
// msgHook hook from on message // msgHook hook from on message
func (h *Handler) msgHook(msg *network.Msg, from, to *client) { func (h *Handler) msgHook(msg *network.Msg, from, to *client) {
switch msg.GetXType() { switch msg.GetXType() {
@@ -184,16 +196,13 @@ func (h *Handler) msgHook(msg *network.Msg, from, to *client) {
case network.Msg_disconnect, case network.Msg_disconnect,
network.Msg_shell_close: network.Msg_shell_close:
h.removeLink(msg.GetLinkId(), from, to) h.removeLink(msg.GetLinkId(), from, to)
// connect response // response link
case network.Msg_connect_rep: case network.Msg_connect_rep:
rep := msg.GetCrep() rep := msg.GetCrep()
if rep.GetOk() { h.responseLink(msg.GetLinkId(), rep.GetOk(), rep.GetMsg(), from, to)
logging.Info("link %s from %s-%d to %s-%d connect successed", case network.Msg_shell_created:
msg.GetLinkId(), from.parent.id, from.idx, to.parent.id, to.idx) rep := msg.GetScreated()
} else { h.responseLink(msg.GetLinkId(), rep.GetOk(), rep.GetMsg(), from, to)
logging.Info("link %s from %s-%d to %s-%d connect failed, %s",
msg.GetLinkId(), from.parent.id, from.idx, to.parent.id, to.idx, rep.GetMsg())
}
// forward data // forward data
case network.Msg_forward: case network.Msg_forward:
data := msg.GetXData() data := msg.GetXData()