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

use cache reduce malloc

Browse Source
This commit is contained in:
hebo
2019-09-09 21:36:18 +08:00
parent 3294a0fc5f
commit 4086028217
12 changed files with 280 additions and 169 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
capture/network.go
View File

@@ -117,7 +117,7 @@ func (nc *networkCard) listenNormal() {
}
// throw packets according to a certain probability
throwPacketRate := communicator.GetThrowPacketRate()
throwPacketRate := communicator.GetTCPThrowPacketRate()
if throwPacketRate >= 1.0 {
time.Sleep(time.Second*3)
continue
@@ -255,7 +255,11 @@ func readFromServerPackage(
if tcpPkt.FIN {
sessionKey := spliceSessionKey(srcIP, srcPort)
delete(sessionPool, *sessionKey)
session := sessionPool[*sessionKey]
if session != nil {
session.Close()
delete(sessionPool, *sessionKey)
}
return
}
@@ -285,7 +289,11 @@ func readToServerPackage(
// when client try close connection remove session from session pool
if tcpPkt.FIN {
sessionKey := spliceSessionKey(srcIP, srcPort)
delete(sessionPool, *sessionKey)
session := sessionPool[*sessionKey]
if session != nil {
session.Close()
delete(sessionPool, *sessionKey)
}
log.Debugf("close connection from %s", *sessionKey)
return
}