github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 10:46:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

progress

Browse source
This commit is contained in:
Simone Margaritelli 2024-09-18 23:21:30 +02:00
commit 67cc9680ed
17 changed files with 1833 additions and 274 deletions
Download patch file Download diff file Expand all files Collapse all files

9
modules/net_probe/net_probe.go
View file

@ -118,7 +118,7 @@ func (mod *Prober) Start() error {
}
if mod.probes.MDNS {
go mod.mdnsProber()
mod.Session.Run("mdns.discovery on")
}
fromIP := mod.Session.Interface.IP
@ -129,9 +129,6 @@ func (mod *Prober) Start() error {
mod.Info("probing %d addresses on %s", len(addresses), cidr)
for mod.Running() {
if mod.probes.MDNS {
mod.sendProbeMDNS(fromIP, fromHW)
}
if mod.probes.UPNP {
mod.sendProbeUPNP(fromIP, fromHW)
@ -160,6 +157,10 @@ func (mod *Prober) Start() error {
func (mod *Prober) Stop() error {
return mod.SetRunning(false, func() {
if mod.probes.MDNS {
mod.Session.Run("mdns.discovery off")
}
mod.waitGroup.Wait()
})
}

100
modules/net_probe/net_probe_mdns.go
View file

@ -1,100 +0,0 @@
package net_probe
import (
"fmt"
"io/ioutil"
"log"
"net"
"github.com/bettercap/bettercap/v2/packets"
"github.com/hashicorp/mdns"
)
var services = []string{
"_hap._tcp.local",
"_homekit._tcp.local",
"_airplay._tcp.local",
"_raop._tcp.local",
"_sleep-proxy._udp.local",
"_companion-link._tcp.local",
"_googlezone._tcp.local",
"_googlerpc._tcp.local",
"_googlecast._tcp.local",
"local",
}
func (mod *Prober) sendProbeMDNS(from net.IP, from_hw net.HardwareAddr) {
err, raw := packets.NewMDNSProbe(from, from_hw)
if err != nil {
mod.Error("error while sending mdns probe: %v", err)
return
} else if err := mod.Session.Queue.Send(raw); err != nil {
mod.Error("error sending mdns packet: %s", err)
} else {
mod.Debug("sent %d bytes of MDNS probe", len(raw))
}
}
func (mod *Prober) mdnsListener(c chan *mdns.ServiceEntry) {
mod.Debug("mdns listener started")
defer mod.Debug("mdns listener stopped")
for entry := range c {
addrs := []string{}
if entry.AddrV4 != nil {
addrs = append(addrs, entry.AddrV4.String())
}
if entry.AddrV6 != nil {
addrs = append(addrs, entry.AddrV6.String())
}
for _, addr := range addrs {
if host := mod.Session.Lan.GetByIp(addr); host != nil {
meta := make(map[string]string)
meta["mdns:name"] = entry.Name
meta["mdns:hostname"] = entry.Host
if entry.AddrV4 != nil {
meta["mdns:ipv4"] = entry.AddrV4.String()
}
if entry.AddrV6 != nil {
meta["mdns:ipv6"] = entry.AddrV6.String()
}
meta["mdns:port"] = fmt.Sprintf("%d", entry.Port)
mod.Debug("meta for %s: %v", addr, meta)
host.OnMeta(meta)
} else {
mod.Debug("got mdns entry for unknown ip %s", entry.AddrV4)
}
}
}
}
func (mod *Prober) mdnsProber() {
mod.Debug("mdns prober started")
defer mod.Debug("mdns.prober stopped")
mod.waitGroup.Add(1)
defer mod.waitGroup.Done()
log.SetOutput(ioutil.Discard)
ch := make(chan *mdns.ServiceEntry)
defer close(ch)
go mod.mdnsListener(ch)
for mod.Running() {
for _, svc := range services {
if mod.Running() {
mdns.Lookup(svc, ch)
}
}
}
}