github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 04:59:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

fix: counting udp probe packets in sent stats

Browse source
This commit is contained in:
evilsocket 2018-01-17 17:31:51 +01:00
commit 97c97a668a
2 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
modules/net_probe.go
View file

@ -75,7 +75,13 @@ func (p *Prober) sendProbe(from net.IP, from_hw net.HardwareAddr, ip net.IP) {
} else { } else {
// log.Debug("UDP connection to %s enstablished.", name) // log.Debug("UDP connection to %s enstablished.", name)
defer con.Close() defer con.Close()
con.Write([]byte{0xde, 0xad, 0xbe, 0xef}) wrote, _ := con.Write([]byte{0xde, 0xad, 0xbe, 0xef})
if wrote > 0 {
p.Session.Queue.Lock()
p.Session.Queue.Sent += uint64(wrote)
p.Session.Queue.Unlock()
}
} }
} }

4
packets/queue.go
View file

@ -74,6 +74,8 @@ func (q *Queue) worker() {
if q.active == false { if q.active == false {
return return
} }
q.Lock()
pktSize := uint64(len(pkt.Data())) pktSize := uint64(len(pkt.Data()))
q.PktReceived++ q.PktReceived++
@ -131,6 +133,8 @@ func (q *Queue) worker() {
q.Traffic[addr].Received += pktSize q.Traffic[addr].Received += pktSize
} }
} }
q.Unlock()
} }
} }