cy/goproxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

no message

Browse Source
This commit is contained in:
arraykeys
2018-07-05 00:03:46 +08:00
parent c20e19d74f
commit e92375f6a9
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
services/socks/socks.go
View File

@ -602,6 +602,7 @@ func (s *Socks) proxyUDP(inConn *net.Conn, methodReq socks.MethodsRequest, reque
} }
}() }()
if *s.cfg.Parent != "" { if *s.cfg.Parent != "" {
//parent proxy
outconn, err := s.getOutConn(nil, nil, "", false) outconn, err := s.getOutConn(nil, nil, "", false)
if err != nil { if err != nil {
clean("connnect fail", fmt.Sprintf("%s", err)) clean("connnect fail", fmt.Sprintf("%s", err))
@ -625,6 +626,7 @@ func (s *Socks) proxyUDP(inConn *net.Conn, methodReq socks.MethodsRequest, reque
s.log.Printf("parent udp address %s", client.UDPAddr) s.log.Printf("parent udp address %s", client.UDPAddr)
} else { } else {
//local proxy
for { for {
buf := utils.LeakyBuffer.Get() buf := utils.LeakyBuffer.Get()
defer utils.LeakyBuffer.Put(buf) defer utils.LeakyBuffer.Put(buf)
@ -652,9 +654,10 @@ func (s *Socks) proxyUDP(inConn *net.Conn, methodReq socks.MethodsRequest, reque
continue continue
} }
s.udpRelatedPacketConns.Set(srcAddr.String(), outUDPConn) s.udpRelatedPacketConns.Set(srcAddr.String(), outUDPConn)
go func() { go func() {
defer s.udpRelatedPacketConns.Remove(srcAddr.String()) defer s.udpRelatedPacketConns.Remove(srcAddr.String())
//bind //out->local io copy
buf := utils.LeakyBuffer.Get() buf := utils.LeakyBuffer.Get()
defer utils.LeakyBuffer.Put(buf) defer utils.LeakyBuffer.Put(buf)
for { for {
@ -685,6 +688,7 @@ func (s *Socks) proxyUDP(inConn *net.Conn, methodReq socks.MethodsRequest, reque
} else { } else {
outUDPConn = v.(*net.UDPConn) outUDPConn = v.(*net.UDPConn)
} }
//local->out io copy
_, err = outUDPConn.Write(p.Data()) _, err = outUDPConn.Write(p.Data())
if err != nil { if err != nil {
if isClosedErr(err) { if isClosedErr(err) {