chore: fix sniffTcp

This commit is contained in:
mzz2017
2023-08-27 11:28:49 +08:00
parent 2af52bb8bd
commit 2dd5338d7c
2 changed files with 6 additions and 2 deletions

View File

@ -82,10 +82,14 @@ func (s *Sniffer) SniffTcp() (d string, err error) {
defer s.readMu.Unlock() defer s.readMu.Unlock()
if s.stream { if s.stream {
go func() { go func() {
_, err := s.buf.ReadFrom(s.r) // Read once.
s.buf.Reset()
s.buf.Grow(max(0, 4096-s.buf.Available()))
n, err := s.r.Read(s.buf.AvailableBuffer())
if err != nil { if err != nil {
s.dataError = err s.dataError = err
} }
s.buf.Truncate(n)
close(s.dataReady) close(s.dataReady)
}() }()

2
go.mod
View File

@ -1,6 +1,6 @@
module github.com/daeuniverse/dae module github.com/daeuniverse/dae
go 1.20 go 1.21
require ( require (
github.com/adrg/xdg v0.4.0 github.com/adrg/xdg v0.4.0