zishuo/opennotr
1
0
Fork 0
mirror of https://github.com/ICKelin/opennotr.git synced 2025-09-27 12:12:07 +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

4
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)
}()
}
func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
@@ -177,7 +175,6 @@ func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
}
}()
go func() {
defer remoteConn.Close()
defer stream.Close()
buf := make([]byte, 64*1024)
@@ -195,7 +192,6 @@ func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
break
}
}
}()
}
func encode(raw []byte) []byte {