This commit is contained in:
刘河
2019-01-12 01:22:53 +08:00
parent a67cdff02a
commit 0ce6ab1e15
4 changed files with 20 additions and 25 deletions

View File

@@ -141,6 +141,11 @@ func DelTask(vKey string) error {
if err := StopServer(vKey); err != nil {
return err
}
for _, v := range CsvDb.Hosts {
if v.Vkey == vKey {
CsvDb.DelHost(v.Host)
}
}
return CsvDb.DelTask(vKey)
}

View File

@@ -104,13 +104,20 @@ func (s *TunnelModeServer) Close() error {
//tcp隧道模式
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代理模式
func ProcessHttp(c *utils.Conn, s *TunnelModeServer) error {
method, addr, rb, err, r := c.GetHost()
if err != nil {
log.Println(err)
c.Close()
return err
}