mirror of
https://github.com/ehang-io/nps.git
synced 2025-07-02 04:00:42 +00:00
124 lines
2.8 KiB
Go
Executable File
124 lines
2.8 KiB
Go
Executable File
package server
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/cnlh/nps/lib/beego"
|
||
"github.com/cnlh/nps/bridge"
|
||
"github.com/cnlh/nps/lib/common"
|
||
"github.com/cnlh/nps/lib/conn"
|
||
"github.com/cnlh/nps/lib/file"
|
||
"github.com/cnlh/nps/lib/lg"
|
||
"net"
|
||
"path/filepath"
|
||
"strings"
|
||
)
|
||
|
||
type TunnelModeServer struct {
|
||
server
|
||
process process
|
||
listener *net.TCPListener
|
||
}
|
||
|
||
//tcp|http|host
|
||
func NewTunnelModeServer(process process, bridge *bridge.Bridge, task *file.Tunnel) *TunnelModeServer {
|
||
s := new(TunnelModeServer)
|
||
s.bridge = bridge
|
||
s.process = process
|
||
s.task = task
|
||
s.config = file.DeepCopyConfig(task.Config)
|
||
return s
|
||
}
|
||
|
||
//开始
|
||
func (s *TunnelModeServer) Start() error {
|
||
var err error
|
||
s.listener, err = net.ListenTCP("tcp", &net.TCPAddr{net.ParseIP("0.0.0.0"), s.task.TcpPort, ""})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for {
|
||
c, err := s.listener.AcceptTCP()
|
||
if err != nil {
|
||
if strings.Contains(err.Error(), "use of closed network connection") {
|
||
break
|
||
}
|
||
lg.Println(err)
|
||
continue
|
||
}
|
||
go s.process(conn.NewConn(c), s)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
//与客户端建立通道
|
||
func (s *TunnelModeServer) dealClient(c *conn.Conn, cnf *file.Config, addr string, method string, rb []byte) error {
|
||
link := conn.NewLink(s.task.Client.GetId(), common.CONN_TCP, addr, cnf.CompressEncode, cnf.CompressDecode, cnf.Crypt, c, s.task.Flow, nil, s.task.Client.Rate, nil)
|
||
|
||
if tunnel, err := s.bridge.SendLinkInfo(s.task.Client.Id, link); err != nil {
|
||
c.Close()
|
||
return err
|
||
} else {
|
||
s.linkCopy(link, c, rb, tunnel, s.task.Flow)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
//close
|
||
func (s *TunnelModeServer) Close() error {
|
||
return s.listener.Close()
|
||
}
|
||
|
||
//web管理方式
|
||
type WebServer struct {
|
||
server
|
||
}
|
||
|
||
//开始
|
||
func (s *WebServer) Start() error {
|
||
p, _ := beego.AppConfig.Int("httpport")
|
||
if !common.TestTcpPort(p) {
|
||
lg.Fatalln("web管理端口", p, "被占用!")
|
||
}
|
||
beego.BConfig.WebConfig.Session.SessionOn = true
|
||
lg.Println("web管理启动，访问端口为", p)
|
||
beego.SetStaticPath("/static", filepath.Join(common.GetRunPath(), "web", "static"))
|
||
beego.SetViewsPath(filepath.Join(common.GetRunPath(), "web", "views"))
|
||
beego.Run()
|
||
return errors.New("web管理启动失败")
|
||
}
|
||
|
||
//new
|
||
func NewWebServer(bridge *bridge.Bridge) *WebServer {
|
||
s := new(WebServer)
|
||
s.bridge = bridge
|
||
return s
|
||
}
|
||
|
||
type process func(c *conn.Conn, s *TunnelModeServer) error
|
||
|
||
//tcp隧道模式
|
||
func ProcessTunnel(c *conn.Conn, s *TunnelModeServer) error {
|
||
if !s.ResetConfig() {
|
||
c.Close()
|
||
return errors.New("流量超出")
|
||
}
|
||
return s.dealClient(c, s.config, s.task.Target, "", nil)
|
||
}
|
||
|
||
//http代理模式
|
||
func ProcessHttp(c *conn.Conn, s *TunnelModeServer) error {
|
||
if !s.ResetConfig() {
|
||
c.Close()
|
||
return errors.New("流量超出")
|
||
}
|
||
method, addr, rb, err, r := c.GetHost()
|
||
if err != nil {
|
||
c.Close()
|
||
return err
|
||
}
|
||
if err := s.auth(r, c, s.config.U, s.config.P); err != nil {
|
||
return err
|
||
}
|
||
return s.dealClient(c, s.config, addr, method, rb)
|
||
}
|