tweak to the cumulative comparison

owasp-amass · Feb 25, 2019 · 2282893 · 2282893
1 parent 507fa68
commit 2282893
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/amass/resolvers.go b/amass/resolvers.go
@@ -256,7 +256,9 @@ func (r *resolver) processMessage(msg *dns.Msg) {
 	// Check that the query was successful
 	if msg.Rcode != dns.RcodeSuccess {
 		var again bool
-		if msg.Rcode == dns.RcodeRefused || msg.Rcode == dns.RcodeServerFailure {
+		if msg.Rcode == dns.RcodeRefused ||
+			msg.Rcode == dns.RcodeServerFailure ||
+			msg.Rcode == dns.RcodeNotImplemented {
 			again = true
 		}
 		r.returnRequest(req, &resolveResult{

diff --git a/cmd/amass.tracker/main.go b/cmd/amass.tracker/main.go
@@ -168,8 +168,8 @@ func cumulativeOutput(domain string, enums []string, ea, la []time.Time, h handl
 	filter := utils.NewStringFilter()
 
 	var cum []*core.Output
-	for _, enum := range enums[:idx] {
-		for _, out := range getEnumDataInScope(domain, enum, h) {
+	for i := idx - 1; i >= 0; i-- {
+		for _, out := range getEnumDataInScope(domain, enums[i], h) {
 			if !filter.Duplicate(out.Name) {
 				cum = append(cum, out)
 			}