mirror of https://github.com/ehang-io/nps
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/cnlh/easyProxy/bridge"
|
|
"github.com/cnlh/easyProxy/utils"
|
|
"sync"
|
|
)
|
|
|
|
//server base struct
|
|
type server struct {
|
|
bridge *bridge.Bridge
|
|
task *utils.Tunnel
|
|
config *utils.Config
|
|
sync.Mutex
|
|
}
|
|
|
|
func (s *server) GetTunnelAndWriteHost(connType string, clientId int, cnf *utils.Config, addr string) (link *utils.Conn, err error) {
|
|
if link, err = s.bridge.GetTunnel(clientId, cnf.CompressEncode, cnf.CompressDecode, cnf.Crypt, cnf.Mux); err != nil {
|
|
return
|
|
}
|
|
if _, err = link.WriteHost(connType, addr); err != nil {
|
|
link.Close()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *server) FlowAdd(in, out int64) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
s.task.Flow.ExportFlow += out
|
|
s.task.Flow.InletFlow += in
|
|
}
|
|
|
|
func (s *server) FlowAddHost(host *utils.Host, in, out int64) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
host.Flow.ExportFlow += out
|
|
host.Flow.InletFlow += in
|
|
}
|
|
|
|
//热更新配置
|
|
func (s *server) ResetConfig() {
|
|
//获取最新数据
|
|
task, err := CsvDb.GetTask(s.task.Id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
s.task.UseClientCnf = task.UseClientCnf
|
|
//使用客户端配置
|
|
if s.task.UseClientCnf {
|
|
client, err := CsvDb.GetClient(s.task.Client.Id)
|
|
if err == nil {
|
|
s.config.U = client.Cnf.U
|
|
s.config.P = client.Cnf.P
|
|
s.config.Compress = client.Cnf.Compress
|
|
s.config.Mux = client.Cnf.Mux
|
|
s.config.Crypt = client.Cnf.Crypt
|
|
}
|
|
} else {
|
|
if err == nil {
|
|
s.config.U = task.Config.U
|
|
s.config.P = task.Config.P
|
|
s.config.Compress = task.Config.Compress
|
|
s.config.Mux = task.Config.Mux
|
|
s.config.Crypt = task.Config.Crypt
|
|
}
|
|
}
|
|
s.config.CompressDecode, s.config.CompressEncode = utils.GetCompressType(s.config.Compress)
|
|
}
|