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

refact: refactored code to use gopacket for mDNS related stuff

Browse source
This commit is contained in:
evilsocket 2018-08-30 19:09:22 +03:00
commit cf000fc5f7
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
308 changed files with 30 additions and 40636 deletions
Download patch file Download diff file Expand all files Collapse all files

29
modules/net_sniff_mdns.go
View file

@ -8,29 +8,22 @@ import (
"github.com/google/gopacket"
"github.com/google/gopacket/layers"
"github.com/miekg/dns"
)
func mdnsCollectHostname(m map[string][]string, hostname string, address string) {
if _, found := m[hostname]; found == false {
m[hostname] = make([]string, 0)
}
m[hostname] = append(m[hostname], address)
}
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 {
dns := layers.DNS{}
if err := dns.DecodeFromBytes(udp.Payload, gopacket.NilDecodeFeedback); err == nil && dns.OpCode == layers.DNSOpCodeQuery {
m := make(map[string][]string)
for _, answer := range append(msg.Answer, msg.Extra...) {
switch rr := answer.(type) {
case *dns.A:
mdnsCollectHostname(m, rr.Header().Name, answer.(*dns.A).A.String())
case *dns.AAAA:
mdnsCollectHostname(m, rr.Header().Name, answer.(*dns.AAAA).AAAA.String())
answers := append(dns.Answers, dns.Additionals...)
answers = append(answers, dns.Authorities...)
for _, answer := range answers {
if answer.Type == layers.DNSTypeA || answer.Type == layers.DNSTypeAAAA {
hostname := string(answer.Name)
if _, found := m[hostname]; found == false {
m[hostname] = make([]string, 0)
}
m[hostname] = append(m[hostname], answer.IP.String())
}
}