mirror of
https://github.com/lwch/natpass.git
synced 2024-04-21 12:41:54 +00:00
100 lines
3.2 KiB
Go
100 lines
3.2 KiB
Go
package app
|
|
|
|
import (
|
|
"github.com/lwch/logging"
|
|
"github.com/lwch/natpass/code/client/conn"
|
|
"github.com/lwch/natpass/code/client/global"
|
|
"github.com/lwch/natpass/code/client/rule"
|
|
"github.com/lwch/natpass/code/client/rule/code"
|
|
"github.com/lwch/natpass/code/client/rule/shell"
|
|
"github.com/lwch/natpass/code/client/rule/vnc"
|
|
"github.com/lwch/natpass/code/network"
|
|
)
|
|
|
|
func (p *program) shellCreate(mgr *rule.Mgr, conn *conn.Conn, msg *network.Msg) {
|
|
create := msg.GetCreq()
|
|
tn := mgr.GetLinked(create.GetName(), msg.GetFrom())
|
|
if tn == nil {
|
|
tn = shell.New(global.Rule{
|
|
Name: create.GetName(),
|
|
Target: msg.GetFrom(),
|
|
Type: "shell",
|
|
Exec: create.GetCshell().GetExec(),
|
|
Env: create.GetCshell().GetEnv(),
|
|
}, p.cfg.ReadTimeout, p.cfg.WriteTimeout)
|
|
mgr.Add(tn.(rule.Rule))
|
|
}
|
|
lk := tn.NewLink(msg.GetLinkId(), msg.GetFrom(), nil, conn).(*shell.Link)
|
|
logging.Info("create link %s for shell rule [%s] from %s to %s",
|
|
msg.GetLinkId(), create.GetName(),
|
|
msg.GetFrom(), p.cfg.ID)
|
|
err := lk.Exec()
|
|
if err != nil {
|
|
logging.Error("create shell failed: %v", err)
|
|
conn.SendConnectError(msg.GetFrom(), msg.GetLinkId(), err.Error())
|
|
return
|
|
}
|
|
conn.SendConnectOK(msg.GetFrom(), msg.GetLinkId())
|
|
lk.Forward()
|
|
}
|
|
|
|
func (p *program) vncCreate(confDir string, mgr *rule.Mgr, conn *conn.Conn, msg *network.Msg) {
|
|
create := msg.GetCreq()
|
|
tn := mgr.GetLinked(create.GetName(), msg.GetFrom())
|
|
if tn == nil {
|
|
tn = vnc.New(global.Rule{
|
|
Name: create.GetName(),
|
|
Target: msg.GetFrom(),
|
|
Type: "vnc",
|
|
Fps: create.GetCvnc().GetFps(),
|
|
}, p.cfg.ReadTimeout, p.cfg.WriteTimeout)
|
|
mgr.Add(tn.(rule.Rule))
|
|
}
|
|
lk := tn.NewLink(msg.GetLinkId(), msg.GetFrom(), nil, conn).(*vnc.Link)
|
|
logging.Info("create link %s for vnc rule [%s] from %s to %s",
|
|
msg.GetLinkId(), create.GetName(),
|
|
msg.GetFrom(), p.cfg.ID)
|
|
lk.SetQuality(create.GetCvnc().GetQuality())
|
|
err := lk.Fork(confDir)
|
|
if err != nil {
|
|
logging.Error("create vnc failed: %v", err)
|
|
conn.SendConnectError(msg.GetFrom(), msg.GetLinkId(), err.Error())
|
|
return
|
|
}
|
|
conn.SendConnectOK(msg.GetFrom(), msg.GetLinkId())
|
|
lk.Forward()
|
|
}
|
|
|
|
func (p *program) benchCreate(confDir string, mgr *rule.Mgr, conn *conn.Conn, msg *network.Msg) {
|
|
create := msg.GetCreq()
|
|
logging.Info("create link %s for bench rule [%s] from %s to %s",
|
|
msg.GetLinkId(), create.GetName(),
|
|
msg.GetFrom(), p.cfg.ID)
|
|
conn.SendConnectOK(msg.GetFrom(), msg.GetLinkId())
|
|
}
|
|
|
|
func (p *program) codeCreate(confDir string, mgr *rule.Mgr, conn *conn.Conn, msg *network.Msg) {
|
|
create := msg.GetCreq()
|
|
tn := mgr.GetLinked(create.GetName(), msg.GetFrom())
|
|
if tn == nil {
|
|
tn = code.New(global.Rule{
|
|
Name: create.GetName(),
|
|
Target: msg.GetFrom(),
|
|
Type: "code-server",
|
|
}, p.cfg.ReadTimeout, p.cfg.WriteTimeout)
|
|
mgr.Add(tn.(rule.Rule))
|
|
}
|
|
workspace := tn.NewLink(msg.GetLinkId(), msg.GetFrom(), nil, conn).(*code.Workspace)
|
|
logging.Info("create link %s for code-server rule [%s] from %s to %s",
|
|
msg.GetLinkId(), create.GetName(),
|
|
msg.GetFrom(), p.cfg.ID)
|
|
err := workspace.Exec(p.cfg.CodeDir)
|
|
if err != nil {
|
|
logging.Error("create vnc failed: %v", err)
|
|
conn.SendConnectError(msg.GetFrom(), msg.GetLinkId(), err.Error())
|
|
return
|
|
}
|
|
conn.SendConnectOK(msg.GetFrom(), msg.GetLinkId())
|
|
workspace.Forward()
|
|
}
|