7x31/sniffer-agent
1
0
Fork 0
mirror of https://github.com/zr-hebo/sniffer-agent.git synced 2025-09-19 10:16:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复 client端抓包，client的 IP显示不准确的 bug

Browse Source
This commit is contained in:
xiaobiao
2022-04-06 12:55:01 +08:00
parent b74b4fed20
commit 87a877750f
3 changed files with 14 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
capture/network.go
View File

@@ -165,7 +165,7 @@ func (nc *networkCard) parseTCPPackage(packet gopacket.Packet, authHeader *pp.He
}
// deal mysql server response
err = readToServerPackage(clientIP, clientPort, &srcIP, srcPort, &dstIP, tcpPkt, nc.receiver)
err = readToServerPackage(clientIP, clientPort, &dstIP, tcpPkt, nc.receiver)
if err != nil {
return
}
@@ -182,7 +182,7 @@ func (nc *networkCard) parseTCPPackage(packet gopacket.Packet, authHeader *pp.He
}
func readFromServerPackage(
srcIP *string, srcPort int, tcpPkt *layers.TCP) (err error) {
clientIP *string, clientPort int, tcpPkt *layers.TCP) (err error) {
defer func() {
if err != nil {
log.Error("read Mysql package send from mysql server to client failed <-- %s", err.Error())
@@ -190,7 +190,7 @@ func readFromServerPackage(
}()
if tcpPkt.FIN {
sessionKey := spliceSessionKey(srcIP, srcPort)
sessionKey := spliceSessionKey(clientIP, clientPort)
session := sessionPool[*sessionKey]
if session != nil {
session.Close()
@@ -204,7 +204,7 @@ func readFromServerPackage(
return
}
sessionKey := spliceSessionKey(srcIP, srcPort)
sessionKey := spliceSessionKey(clientIP, clientPort)
session := sessionPool[*sessionKey]
if session != nil {
pkt := model.NewTCPPacket(tcpPayload, int64(tcpPkt.Ack), false)
@@ -215,7 +215,7 @@ func readFromServerPackage(
}
func readToServerPackage(
clientIP *string, clientPort int, srcIP *string, srcPort int, destIP *string, tcpPkt *layers.TCP,
clientIP *string, clientPort int, destIP *string, tcpPkt *layers.TCP,
receiver chan model.QueryPiece) (err error) {
defer func() {
if err != nil {
@@ -225,7 +225,7 @@ func readToServerPackage(
// when client try close connection remove session from session pool
if tcpPkt.FIN {
sessionKey := spliceSessionKey(srcIP, srcPort)
sessionKey := spliceSessionKey(clientIP, clientPort)
session := sessionPool[*sessionKey]
if session != nil {
session.Close()
@@ -240,10 +240,10 @@ func readToServerPackage(
return
}
sessionKey := spliceSessionKey(srcIP, srcPort)
sessionKey := spliceSessionKey(clientIP, clientPort)
session := sessionPool[*sessionKey]
if session == nil {
session = sd.NewSession(sessionKey, clientIP, clientPort, srcIP, srcPort, destIP, snifferPort, receiver)
session = sd.NewSession(sessionKey, clientIP, clientPort, destIP, snifferPort, receiver)
sessionPool[*sessionKey] = session
}