From 2c8fe6928be86415811ad26bd7510b1d9e8dfa0b Mon Sep 17 00:00:00 2001 From: caffix Date: Sat, 1 Apr 2023 23:14:34 -0400 Subject: [PATCH] fixed #916 by addressing the failure to terminate --- enum/input.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/enum/input.go b/enum/input.go index 42ee4160d..c75ca5e86 100644 --- 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 }