fixed #916 by addressing the failure to terminate

owasp-amass · May 9, 2023 · 2c8fe69 · 2c8fe69
1 parent fb8cbc3
commit 2c8fe69
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/enum/input.go b/enum/input.go
@@ -11,12 +11,12 @@ import (
 	"sync"
 	"time"
 
-	amassnet "github.com/owasp-amass/amass/v3/net"
-	"github.com/owasp-amass/amass/v3/net/dns"
-	"github.com/owasp-amass/amass/v3/requests"
 	"github.com/caffix/pipeline"
 	"github.com/caffix/queue"
 	"github.com/caffix/service"
+	amassnet "github.com/owasp-amass/amass/v3/net"
+	"github.com/owasp-amass/amass/v3/net/dns"
+	"github.com/owasp-amass/amass/v3/requests"
 	bf "github.com/tylertreat/BoomFilters"
 )
 
@@ -189,8 +189,7 @@ func (r *enumSource) Next(ctx context.Context) bool {
 			r.markDone()
 			return false
 		case <-t.C:
-			if r.pipeline.DataItemCount() <= 0 &&
-				!r.enum.requestsPending() && r.queue.Len() == 0 {
+			if !r.enum.requestsPending() && r.queue.Len() == 0 {
 				r.markDone()
 				return false
 			}