lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-02-20 23:47:21 -05:00
Code Issues Releases Wiki Activity

release buffer

Browse Source
This commit is contained in:
Darien Raymond 2017-01-28 16:27:46 +01:00
parent 0bf5f3e531
commit 1ee9450dfc
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/proxyman/inbound/worker.go
View File

@ -82,7 +82,7 @@ func (w *tcpWorker) Port() v2net.Port {
type udpConn struct {
lastActivityTime int64 // in seconds
input chan []byte
input chan *buf.Buffer
output func([]byte) (int, error)
remote net.Addr
local net.Addr
@ -98,8 +98,9 @@ func (c *udpConn) Read(buf []byte) (int, error) {
if !open {
return 0, io.EOF
}
defer in.Release()
c.updateActivity()
return copy(buf, in), nil
return copy(buf, in.Bytes()), nil
}
func (c *udpConn) Write(buf []byte) (int, error) {
@ -164,7 +165,7 @@ func (w *udpWorker) getConnection(src v2net.Destination) (*udpConn, bool) {
}
conn := &udpConn{
input: make(chan []byte, 32),
input: make(chan *buf.Buffer, 32),
output: func(b []byte) (int, error) {
return w.hub.WriteTo(b, src)
},
@ -185,7 +186,11 @@ func (w *udpWorker) getConnection(src v2net.Destination) (*udpConn, bool) {
func (w *udpWorker) callback(b *buf.Buffer, source v2net.Destination, originalDest v2net.Destination) {
conn, existing := w.getConnection(source)
conn.input <- b.Bytes()
select {
case conn.input <- b:
default:
b.Release()
}
if !existing {
go func() {