From 0e76657248f422c3f88d55b5ea856e9c549eee59 Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Sat, 22 Feb 2025 14:01:10 +0800
Subject: [PATCH] fix: init failed bar not close and total bar not wait

---
 internal/pool/brutepool.go | 2 +-
 internal/runner.go         | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/internal/pool/brutepool.go b/internal/pool/brutepool.go
index ad64cc7..fff02b1 100644
--- a/internal/pool/brutepool.go
+++ b/internal/pool/brutepool.go
@@ -694,8 +694,8 @@ func (pool *BrutePool) doCheck() {
 		}
 		pool.isFallback.Store(true)
 		pool.fallback()
-		pool.Cancel()
 		pool.IsFailed = true
+		pool.Cancel()
 		return
 	}
 
diff --git a/internal/runner.go b/internal/runner.go
index 9879518..de35315 100644
--- a/internal/runner.go
+++ b/internal/runner.go
@@ -204,6 +204,7 @@ func (r *Runner) Prepare(ctx context.Context) error {
 				brutePool.Statistor.Error = err.Error()
 				if !r.Force {
 					// 如果没开启force, init失败将会关闭pool
+					brutePool.Bar.Close()
 					brutePool.Close()
 					r.PrintStat(brutePool)
 					r.Done()
@@ -253,6 +254,7 @@ Loop:
 		}
 	}
 
+	r.bar.Wait()
 	r.poolwg.Wait()
 	r.outwg.Wait()
 }