github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-27 08:39:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove exhaustBuffer field and add rewind function

Browse Source
This commit is contained in:
M. Mert Yildiran 2022-05-12 13:23:50 +03:00
parent ce840940b3
commit 8c8eb20ad6
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tap/tcp_reader.go
View File

@ -22,7 +22,6 @@ type tcpReader struct {
isOutgoing bool
msgQueue chan api.TcpReaderDataMsg // Channel of captured reassembled tcp payload
buffer []byte
exhaustBuffer bool
data []byte
progress *api.ReadProgress
captureTime time.Time
@ -58,7 +57,7 @@ func (reader *tcpReader) run(options *api.TrafficFilteringOptions, wg *sync.Wait
if err == nil || reader.isProtocolIdentified() {
break
}
reader.exhaustBuffer = true
reader.rewind()
}
}
@ -83,12 +82,11 @@ func (reader *tcpReader) isProtocolIdentified() bool {
return reader.parent.protocol != nil
}
func (reader *tcpReader) Read(p []byte) (int, error) {
if reader.exhaustBuffer {
reader.data = reader.buffer
reader.exhaustBuffer = false
}
func (reader *tcpReader) rewind() {
reader.data = reader.buffer
}
func (reader *tcpReader) Read(p []byte) (int, error) {
var msg api.TcpReaderDataMsg
ok := true