mirror of https://github.com/ehang-io/nps
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package proxy
|
|
|
|
import (
|
|
"errors"
|
|
"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/vender/github.com/astaxie/beego/logs"
|
|
"net"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
type Service interface {
|
|
Start() error
|
|
Close() error
|
|
}
|
|
|
|
//Server BaseServer struct
|
|
type BaseServer struct {
|
|
id int
|
|
bridge *bridge.Bridge
|
|
task *file.Tunnel
|
|
errorContent []byte
|
|
sync.Mutex
|
|
}
|
|
|
|
func NewBaseServer(bridge *bridge.Bridge, task *file.Tunnel) *BaseServer {
|
|
return &BaseServer{
|
|
bridge: bridge,
|
|
task: task,
|
|
errorContent: nil,
|
|
Mutex: sync.Mutex{},
|
|
}
|
|
}
|
|
|
|
func (s *BaseServer) FlowAdd(in, out int64) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
s.task.Flow.ExportFlow += out
|
|
s.task.Flow.InletFlow += in
|
|
}
|
|
|
|
func (s *BaseServer) FlowAddHost(host *file.Host, in, out int64) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
host.Flow.ExportFlow += out
|
|
host.Flow.InletFlow += in
|
|
}
|
|
|
|
func (s *BaseServer) writeConnFail(c net.Conn) {
|
|
c.Write([]byte(common.ConnectionFailBytes))
|
|
c.Write(s.errorContent)
|
|
}
|
|
|
|
//权限认证
|
|
func (s *BaseServer) auth(r *http.Request, c *conn.Conn, u, p string) error {
|
|
if u != "" && p != "" && !common.CheckAuth(r, u, p) {
|
|
c.Write([]byte(common.UnauthorizedBytes))
|
|
c.Close()
|
|
return errors.New("401 Unauthorized")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *BaseServer) CheckFlowAndConnNum(client *file.Client) error {
|
|
if client.Flow.FlowLimit > 0 && (client.Flow.FlowLimit<<20) < (client.Flow.ExportFlow+client.Flow.InletFlow) {
|
|
return errors.New("Traffic exceeded")
|
|
}
|
|
if !client.GetConn() {
|
|
return errors.New("Connections exceed the current client limit")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//与客户端建立通道
|
|
func (s *BaseServer) DealClient(c *conn.Conn, client *file.Client, addr string, rb []byte, tp string, f func()) error {
|
|
link := conn.NewLink(tp, addr, client.Cnf.Crypt, client.Cnf.Compress, c.Conn.RemoteAddr().String())
|
|
if target, err := s.bridge.SendLinkInfo(client.Id, link, c.Conn.RemoteAddr().String(), s.task); err != nil {
|
|
logs.Warn("task id %d get connection from client id %d error %s", s.task.Id, client.Id, err.Error())
|
|
c.Close()
|
|
return err
|
|
} else {
|
|
if rb != nil {
|
|
//HTTP proxy crypt or compress
|
|
conn.GetConn(target, link.Crypt, link.Compress, client.Rate, true).Write(rb)
|
|
}
|
|
if f != nil {
|
|
f()
|
|
}
|
|
conn.CopyWaitGroup(target, c.Conn, link.Crypt, link.Compress, client.Rate, s.task.Flow, true)
|
|
}
|
|
client.AddConn()
|
|
return nil
|
|
}
|