Swap TcpID without overwriting it

This commit is contained in:
M. Mert Yildiran 2021-08-25 16:05:01 +03:00
parent d0f0e187cb
commit 3ca6437320
No known key found for this signature in database
GPG Key ID: D42ADB236521BF7A
2 changed files with 8 additions and 7 deletions

View File

@ -41,16 +41,18 @@ type TcpID struct {
Ident string
}
func (t *TcpID) Swap() {
func (t *TcpID) Swap() *TcpID {
srcIP := t.SrcIP
dstIP := t.DstIP
srcPort := t.SrcPort
dstPort := t.DstPort
t.SrcIP = dstIP
t.SrcPort = dstPort
t.DstIP = srcIP
t.DstPort = srcPort
return &TcpID{
SrcIP: dstIP,
SrcPort: dstPort,
DstIP: srcIP,
DstPort: srcPort,
}
}
type GenericMessage struct {

View File

@ -89,8 +89,7 @@ func (d dissecting) Dissect(b *bufio.Reader, isClient bool, tcpID *api.TcpID, em
}
success = true
} else if isClient {
tcpID.Swap()
err = handleHTTP1ClientStream(b, tcpID, emitter)
err = handleHTTP1ClientStream(b, tcpID.Swap(), emitter)
if err == io.EOF || err == io.ErrUnexpectedEOF {
break
} else if err != nil {