xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Chore: adjust tun_wireguard cache buffer

Browse source
This commit is contained in:
yaling888 2022-03-29 07:06:41 +08:00
parent b3b7a393f8
commit 56e2c172e1
1 changed files with 10 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
listener/tun/device/tun/tun_wireguard.go
View file

@ -8,7 +8,6 @@ import (
"os"
"runtime"
"github.com/Dreamacro/clash/common/pool"
"github.com/Dreamacro/clash/listener/tun/device"
"github.com/Dreamacro/clash/listener/tun/device/iobased"
@ -22,6 +21,8 @@ type TUN struct {
mtu uint32
name string
offset int
cache []byte
}
func Open(name string, mtu uint32) (_ device.Device, err error) {
@ -70,6 +71,10 @@ func Open(name string, mtu uint32) (_ device.Device, err error) {
}
t.mtu = uint32(tunMTU)
if t.offset > 0 {
t.cache = make([]byte, 65535)
}
return t, nil
}
@ -78,19 +83,9 @@ func (t *TUN) Read(packet []byte) (int, error) {
return t.nt.Read(packet, t.offset)
}
buff := pool.Get(t.offset + cap(packet))
defer func() {
_ = pool.Put(buff)
}()
n, err := t.nt.Read(t.cache, t.offset)
n, err := t.nt.Read(buff, t.offset)
if err != nil {
return 0, err
}
_ = buff[:t.offset]
copy(packet, buff[t.offset:t.offset+n])
copy(packet, t.cache[t.offset:t.offset+n])
return n, err
}
@ -100,7 +95,8 @@ func (t *TUN) Write(packet []byte) (int, error) {
return t.nt.Write(packet, t.offset)
}
packet = append(make([]byte, t.offset), packet...)
_ = t.cache[:t.offset]
packet = append(t.cache[:t.offset], packet...)
return t.nt.Write(packet, t.offset)
}