github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 21:13:18 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fixed a bug of the probe and recon algos which prevented targets from being removed once offline

Browse source
This commit is contained in:
evilsocket 2018-02-01 19:45:09 +01:00
commit 73f887087a
4 changed files with 12 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

28
modules/net_recon.go
View file

@ -102,27 +102,21 @@ func (d *Discovery) checkShared(new net.ArpTable) {
}
func (d *Discovery) runDiff() {
var new net.ArpTable = make(net.ArpTable)
// check for endpoints who disappeared
var rem net.ArpTable = make(net.ArpTable)
if d.before != nil {
new = net.ArpDiff(d.current, d.before)
rem = net.ArpDiff(d.before, d.current)
} else {
new = d.current
for mac, t := range d.Session.Targets.Targets {
if _, found := d.current[mac]; found == false {
rem[mac] = t.IpAddress
}
}
if len(new) > 0 || len(rem) > 0 {
d.checkShared(new)
for mac, ip := range rem {
d.Session.Targets.Remove(ip, mac)
}
// refresh target pool
for ip, mac := range rem {
d.Session.Targets.Remove(ip, mac)
}
for ip, mac := range new {
d.Session.Targets.AddIfNotExist(ip, mac)
}
// now check for new friends ^_^
for ip, mac := range d.current {
d.Session.Targets.AddIfNotExist(ip, mac)
}
}