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

misc: small fix or general refactoring i did not bother commenting

Browse source
This commit is contained in:
evilsocket 2018-08-30 14:05:56 +03:00
commit 6df190a715
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
1 changed files with 9 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

22
modules/net_sniff_mdns.go
View file

@ -12,6 +12,13 @@ import (
"github.com/miekg/dns" "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 { func mdnsParser(ip *layers.IPv4, pkt gopacket.Packet, udp *layers.UDP) bool {
if udp.SrcPort == packets.MDNSPort && udp.DstPort == packets.MDNSPort { if udp.SrcPort == packets.MDNSPort && udp.DstPort == packets.MDNSPort {
var msg dns.Msg var msg dns.Msg
@ -20,22 +27,11 @@ func mdnsParser(ip *layers.IPv4, pkt gopacket.Packet, udp *layers.UDP) bool {
for _, answer := range append(msg.Answer, msg.Extra...) { for _, answer := range append(msg.Answer, msg.Extra...) {
switch rr := answer.(type) { switch rr := answer.(type) {
case *dns.A: case *dns.A:
name := rr.Header().Name mdnsCollectHostname(m, rr.Header().Name, answer.(*dns.A).A.String())
if _, found := m[name]; found == false {
m[name] = make([]string, 0)
}
m[name] = append(m[name], answer.(*dns.A).A.String())
case *dns.AAAA: case *dns.AAAA:
name := rr.Header().Name mdnsCollectHostname(m, rr.Header().Name, answer.(*dns.AAAA).AAAA.String())
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 { for hostname, ips := range m {