Fixed panic with bad dns records

projectdiscovery · Feb 5, 2021 · 1415f43 · 1415f43
1 parent 82e8b62
commit 1415f43
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/fastdialer/dialer.go b/fastdialer/dialer.go
@@ -2,6 +2,7 @@ package fastdialer
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"net"
 	"strings"
@@ -124,14 +125,15 @@ func (d *Dialer) GetDNSData(hostname string) (*retryabledns.DNSData, error) {
 		if err != nil {
 			return nil, err
 		}
+		if data == nil {
+			return nil, errors.New("could not resolve host")
+		}
 		b, _ := data.Marshal()
 		err = d.hm.Set(hostname, b)
 		if err != nil {
 			return nil, err
 		}
-
 		return data, nil
 	}
-
 	return data, nil
 }