fix prefix,suffix, ext, bak not work bug

chainreactors · Jun 23, 2024 · 784776e · 784776e
1 parent 0d24501
commit 784776e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/internal/option.go b/internal/option.go
@@ -341,11 +341,11 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
 
 	if len(opt.Suffixes) != 0 {
 		mask.SpecialWords["suffix"] = opt.Suffixes
-		opt.Word += "{@suffix}"
+		opt.Word += "{?@suffix}"
 	}
 	if len(opt.Prefixes) != 0 {
 		mask.SpecialWords["prefix"] = opt.Prefixes
-		opt.Word = "{@prefix}" + opt.Word
+		opt.Word = "{?@prefix}" + opt.Word
 	}
 
 	if opt.ForceExtension && opt.Extensions != "" {
@@ -356,7 +356,7 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
 			}
 		}
 		mask.SpecialWords["ext"] = exts
-		opt.Word += "{@ext}"
+		opt.Word += "{?@ext}"
 	}
 
 	r.Wordlist, err = mask.Run(opt.Word, dicts, nil)

diff --git a/internal/pool/brutepool.go b/internal/pool/brutepool.go
@@ -710,7 +710,7 @@ func (pool *BrutePool) addFuzzyBaseline(bl *pkg.Baseline) {
 
 func (pool *BrutePool) doBak() {
 	defer pool.wg.Done()
-	worder, err := words.NewWorderWithDsl("{?0}.{@bak_ext}", [][]string{pkg.BakGenerator(pool.url.Host)}, nil)
+	worder, err := words.NewWorderWithDsl("{?0}.{?@bak_ext}", [][]string{pkg.BakGenerator(pool.url.Host)}, nil)
 	if err != nil {
 		return
 	}
@@ -722,7 +722,7 @@ func (pool *BrutePool) doBak() {
 		})
 	}
 
-	worder, err = words.NewWorderWithDsl("{@bak_name}.{@bak_ext}", nil, nil)
+	worder, err = words.NewWorderWithDsl("{?@bak_name}.{?@bak_ext}", nil, nil)
 	if err != nil {
 		return
 	}