check if nil before the interface conversion

This commit is contained in:
Murilo Santana 2018-02-01 14:09:21 -02:00
commit b03b7c8b56

View file

@ -234,14 +234,18 @@ func (s *DNSSpoofer) shouldSpoof(domain string) bool {
} }
func (s *DNSSpoofer) onPacket(pkt gopacket.Packet) { func (s *DNSSpoofer) onPacket(pkt gopacket.Packet) {
eth := pkt.Layer(layers.LayerTypeEthernet).(*layers.Ethernet) typeEth := pkt.Layer(layers.LayerTypeEthernet)
udp := pkt.Layer(layers.LayerTypeUDP).(*layers.UDP) typeUDP := pkt.Layer(layers.LayerTypeUDP)
if typeEth == nil || typeUDP == nil {
if eth == nil || udp == nil {
return return
} else if s.All == true || bytes.Compare(eth.DstMAC, s.Session.Interface.HW) == 0 { }
eth := typeEth.(*layers.Ethernet)
if s.All || bytes.Compare(eth.DstMAC, s.Session.Interface.HW) == 0 {
dns, parsed := pkt.Layer(layers.LayerTypeDNS).(*layers.DNS) dns, parsed := pkt.Layer(layers.LayerTypeDNS).(*layers.DNS)
if parsed == true && dns.OpCode == layers.DNSOpCodeQuery && len(dns.Questions) > 0 && len(dns.Answers) == 0 { if parsed && dns.OpCode == layers.DNSOpCodeQuery && len(dns.Questions) > 0 && len(dns.Answers) == 0 {
udp := typeUDP.(*layers.UDP)
for _, q := range dns.Questions { for _, q := range dns.Questions {
qName := string(q.Name) qName := string(q.Name)
if s.shouldSpoof(qName) == true { if s.shouldSpoof(qName) == true {