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

new: WSD discovery agent for net.probe

Browse source
This commit is contained in:
evilsocket 2018-09-09 14:10:24 +03:00
commit b6adb2b65f
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
3 changed files with 65 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
modules/net_probe.go
View file

@ -16,6 +16,7 @@ type Probes struct {
NBNS bool
MDNS bool
UPNP bool
WSD bool
}
type Prober struct {
@ -43,6 +44,10 @@ func NewProber(s *session.Session) *Prober {
"true",
"Enable UPNP discovery probes."))
p.AddParam(session.NewBoolParameter("net.probe.wsd",
"true",
"Enable WSD discovery probes."))
p.AddParam(session.NewIntParameter("net.probe.throttle",
"10",
"If greater than 0, probe packets will be throttled by this value in milliseconds."))
@ -96,6 +101,8 @@ func (p *Prober) Configure() error {
return err
} else if err, p.probes.UPNP = p.BoolParam("net.probe.upnp"); err != nil {
return err
} else if err, p.probes.WSD = p.BoolParam("net.probe.wsd"); err != nil {
return err
} else {
log.Debug("Throttling packets of %d ms.", p.throttle)
}
@ -135,6 +142,10 @@ func (p *Prober) Start() error {
p.sendProbeUPNP(from, from_hw)
}
if p.probes.WSD {
p.sendProbeWSD(from, from_hw)
}
for _, ip := range addresses {
if !p.Running() {
return