mirror of https://github.com/ehang-io/nps
bug
parent
a67cdff02a
commit
0ce6ab1e15
20
config.json
20
config.json
|
@ -1,20 +0,0 @@
|
||||||
{
|
|
||||||
"Server": {
|
|
||||||
"ip": "123.206.77.88",
|
|
||||||
"tcp": 8224,
|
|
||||||
"vkey": "DKibZF5TXvic1g3kY",
|
|
||||||
"num": 10
|
|
||||||
},
|
|
||||||
"SiteList": [
|
|
||||||
{
|
|
||||||
"host": "a.server.ourcauc.com",
|
|
||||||
"url": "10.1.50.203",
|
|
||||||
"port": 80
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"host": "b.server.ourcauc.com",
|
|
||||||
"url": "10.1.50.196",
|
|
||||||
"port": 4000
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
|
@ -141,6 +141,11 @@ func DelTask(vKey string) error {
|
||||||
if err := StopServer(vKey); err != nil {
|
if err := StopServer(vKey); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
for _, v := range CsvDb.Hosts {
|
||||||
|
if v.Vkey == vKey {
|
||||||
|
CsvDb.DelHost(v.Host)
|
||||||
|
}
|
||||||
|
}
|
||||||
return CsvDb.DelTask(vKey)
|
return CsvDb.DelTask(vKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -104,13 +104,20 @@ func (s *TunnelModeServer) Close() error {
|
||||||
|
|
||||||
//tcp隧道模式
|
//tcp隧道模式
|
||||||
func ProcessTunnel(c *utils.Conn, s *TunnelModeServer) error {
|
func ProcessTunnel(c *utils.Conn, s *TunnelModeServer) error {
|
||||||
return s.dealClient(c, s.config, s.config.Target, "", nil)
|
_, _, rb, err, r := c.GetHost()
|
||||||
|
if err == nil {
|
||||||
|
if err := s.auth(r, c, s.config.U, s.config.P); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return s.dealClient(c, s.config, s.config.Target, "", rb)
|
||||||
}
|
}
|
||||||
|
|
||||||
//http代理模式
|
//http代理模式
|
||||||
func ProcessHttp(c *utils.Conn, s *TunnelModeServer) error {
|
func ProcessHttp(c *utils.Conn, s *TunnelModeServer) error {
|
||||||
method, addr, rb, err, r := c.GetHost()
|
method, addr, rb, err, r := c.GetHost()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
c.Close()
|
c.Close()
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,7 +10,6 @@ import (
|
||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httputil"
|
|
||||||
"net/url"
|
"net/url"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
@ -245,16 +244,20 @@ func (s *Conn) SetAlive() {
|
||||||
|
|
||||||
//从tcp报文中解析出host,连接类型等 TODO 多种情况
|
//从tcp报文中解析出host,连接类型等 TODO 多种情况
|
||||||
func (s *Conn) GetHost() (method, address string, rb []byte, err error, r *http.Request) {
|
func (s *Conn) GetHost() (method, address string, rb []byte, err error, r *http.Request) {
|
||||||
r, err = http.ReadRequest(bufio.NewReader(s))
|
var b [32 * 1024]byte
|
||||||
if err != nil {
|
var n int
|
||||||
|
if n, err = s.Read(b[:]); err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
rb, err = httputil.DumpRequest(r, true)
|
rb = b[:n]
|
||||||
|
r, err = http.ReadRequest(bufio.NewReader(bytes.NewReader(rb)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
hostPortURL, err := url.Parse(r.Host)
|
hostPortURL, err := url.Parse(r.Host)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
address = r.Host
|
||||||
|
err = nil
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if hostPortURL.Opaque == "443" { //https访问
|
if hostPortURL.Opaque == "443" { //https访问
|
||||||
|
|
Loading…
Reference in New Issue