rms/amneziawg-go
1
0
Fork 0
mirror of https://github.com/amnezia-vpn/amneziawg-go.git synced 2026-05-17 08:15:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

chore: stream read cleanup

Browse Source
This commit is contained in:
Yaroslav Gurov
2026-01-07 20:32:50 +00:00
parent 5f04c25146
commit 861b341f54
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
conn/bind_stream.go
View File

@@ -73,12 +73,9 @@ func (b *BindStream) readStream(ep *streamEndpoint) {
defer b.wg.Done()
for {
var err error
sp := b.streamPacketPool.Get().(*streamPacketQueue)
sp.ep = ep
sp.n, err = ep.conn.Read(sp.buf[:])
sp.err = err
n, err := ep.conn.Read(sp.buf[:])
sp.ep, sp.err, sp.n = ep, err, n
b.queue <- sp
if err != nil {
@@ -255,7 +252,7 @@ func (b *BindStream) BatchSize() int {
}
func (b *BindStream) SetObfsIn(obfs conceal.Obfs) {
b.obfConnOpts.ObfsOut = obfs
b.obfConnOpts.ObfsIn = obfs
}
func (b *BindStream) SetObfsOut(obfs conceal.Obfs) {
@@ -265,9 +262,10 @@ func (b *BindStream) SetObfsOut(obfs conceal.Obfs) {
var _ Endpoint = (*streamEndpoint)(nil)
type streamEndpoint struct {
conn net.Conn
conn net.Conn
dst netip.AddrPort
mutex sync.Mutex
mutex sync.RWMutex
}
func streamEndpointFromConn(conn net.Conn) (*streamEndpoint, error) {