From 632f7700bfcdaa2b3c2d9d0d893f44f23433e522 Mon Sep 17 00:00:00 2001 From: evilsocket Date: Sun, 9 Sep 2018 14:02:36 +0300 Subject: [PATCH] misc: small fix or general refactoring i did not bother commenting --- packets/mdns.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/packets/mdns.go b/packets/mdns.go index e43cab7f..85246247 100644 --- a/packets/mdns.go +++ b/packets/mdns.go @@ -20,6 +20,12 @@ var ( func MDNSGetMeta(pkt gopacket.Packet) map[string]string { meta := make(map[string]string) + defer func() { + if r := recover(); r != nil { + meta = nil + } + }() + if ludp := pkt.Layer(layers.LayerTypeUDP); ludp != nil { if udp := ludp.(*layers.UDP); udp != nil && udp.SrcPort == MDNSPort && udp.DstPort == MDNSPort { dns := layers.DNS{} @@ -47,7 +53,7 @@ func MDNSGetMeta(pkt gopacket.Packet) map[string]string { } } - if len(meta) > 0 { + if meta != nil && len(meta) > 0 { return meta } return nil