bettercap/modules/net_sniff_mdns.go

60 lines
1.4 KiB
Go

package modules
import (
"strings"
"github.com/bettercap/bettercap/core"
"github.com/bettercap/bettercap/packets"
"github.com/google/gopacket"
"github.com/google/gopacket/layers"
"github.com/miekg/dns"
)
func mdnsParser(ip *layers.IPv4, pkt gopacket.Packet, udp *layers.UDP) bool {
if udp.SrcPort == packets.MDNSPort && udp.DstPort == packets.MDNSPort {
var msg dns.Msg
if err := msg.Unpack(udp.Payload); err == nil && msg.Opcode == dns.OpcodeQuery && len(msg.Answer) > 0 {
m := make(map[string][]string)
for _, answer := range append(msg.Answer, msg.Extra...) {
switch rr := answer.(type) {
case *dns.A:
name := rr.Header().Name
if _, found := m[name]; found == false {
m[name] = make([]string, 0)
}
m[name] = append(m[name], answer.(*dns.A).A.String())
case *dns.AAAA:
name := rr.Header().Name
if _, found := m[name]; found == false {
m[name] = make([]string, 0)
}
m[name] = append(m[name], answer.(*dns.AAAA).AAAA.String())
}
}
for hostname, ips := range m {
NewSnifferEvent(
pkt.Metadata().Timestamp,
"mdns",
ip.SrcIP.String(),
ip.DstIP.String(),
nil,
"%s %s : %s is %s",
core.W(core.BG_DGRAY+core.FG_WHITE, "mdns"),
vIP(ip.SrcIP),
core.Yellow(hostname),
core.Dim(strings.Join(ips, ", ")),
).Push()
}
return true
}
}
return false
}