zishuo/opennotr
1
0
Fork 0
mirror of https://github.com/ICKelin/opennotr.git synced 2025-09-26 20:01:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplify client goroutine

Browse Source
This commit is contained in:
ICKelin
2021-05-13 19:06:18 +08:00
parent ea0e7b24ff
commit b05294abf5
1 changed files with 19 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
opennotr/client.go
View File

@@ -133,11 +133,9 @@ func (c *Client) tcpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
io.Copy(remoteConn, stream)
}()
go func() {
defer remoteConn.Close()
defer stream.Close()
io.Copy(stream, remoteConn)
}()
defer remoteConn.Close()
defer stream.Close()
io.Copy(stream, remoteConn)
}
func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
@@ -177,25 +175,23 @@ func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
}
}()
go func() {
defer remoteConn.Close()
defer stream.Close()
buf := make([]byte, 64*1024)
for {
nr, err := remoteConn.Read(buf)
if err != nil {
log.Println(err)
break
}
bytes := encode(buf[:nr])
_, err = stream.Write(bytes)
if err != nil {
log.Println(err)
break
}
defer remoteConn.Close()
defer stream.Close()
buf := make([]byte, 64*1024)
for {
nr, err := remoteConn.Read(buf)
if err != nil {
log.Println(err)
break
}
}()
bytes := encode(buf[:nr])
_, err = stream.Write(bytes)
if err != nil {
log.Println(err)
break
}
}
}
func encode(raw []byte) []byte {