zishuo/natpass
1
0
Fork 0
mirror of https://github.com/lwch/natpass synced 2025-10-06 13:57:16 +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_forward:
case network.Msg_shell_create:
case network.Msg_shell_created:
case network.Msg_shell_resize:
case network.Msg_shell_data:
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)
}
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
func (h *Handler) msgHook(msg *network.Msg, from, to *client) {
switch msg.GetXType() {
@@ -184,16 +196,13 @@ func (h *Handler) msgHook(msg *network.Msg, from, to *client) {
case network.Msg_disconnect,
network.Msg_shell_close:
h.removeLink(msg.GetLinkId(), from, to)
// connect response
// response link
case network.Msg_connect_rep:
rep := msg.GetCrep()
if rep.GetOk() {
logging.Info("link %s from %s-%d to %s-%d connect successed",
msg.GetLinkId(), from.parent.id, from.idx, to.parent.id, to.idx)
} else {
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())
}
h.responseLink(msg.GetLinkId(), rep.GetOk(), rep.GetMsg(), from, to)
case network.Msg_shell_created:
rep := msg.GetScreated()
h.responseLink(msg.GetLinkId(), rep.GetOk(), rep.GetMsg(), from, to)
// forward data
case network.Msg_forward:
data := msg.GetXData()