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

fix: fix udp mem pool

Browse Source
This commit is contained in:
ICKelin
2021-05-14 20:53:19 +08:00
parent f915cb12a6
commit f846329384
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
opennotr/client.go
View File

@@ -122,7 +122,6 @@ func (c *Client) handleStream(stream *yamux.Stream) {
log.Println("unmarshal fail: ", err)
return
}
switch proxyProtocol.Protocol {
case "tcp":
c.tcpProxy(stream, &proxyProtocol)
@@ -177,23 +176,22 @@ func (c *Client) udpProxy(stream *yamux.Stream, p *proto.ProxyProtocol) {
defer remoteConn.Close()
defer stream.Close()
hdr := make([]byte, 2)
obj := c.udppool.Get()
buf := obj.([]byte)
defer c.udppool.Put(buf)
for {
_, err := io.ReadFull(stream, hdr)
if err != nil {
log.Println("read stream fail: ", err)
break
}
nlen := binary.BigEndian.Uint16(hdr)
_, err = io.ReadFull(stream, buf[:nlen])
buf := make([]byte, nlen)
nr, err := io.ReadFull(stream, buf)
if err != nil {
log.Println("read stream body fail: ", err)
break
}
remoteConn.Write(buf)
remoteConn.Write(buf[:nr])
}
}()