fix prefix,suffix, ext, bak not work bug

This commit is contained in:
M09Ic 2024-06-23 16:59:44 +08:00
parent 0d24501a8a
commit 784776edb3
2 changed files with 5 additions and 5 deletions

View File

@ -341,11 +341,11 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
if len(opt.Suffixes) != 0 { if len(opt.Suffixes) != 0 {
mask.SpecialWords["suffix"] = opt.Suffixes mask.SpecialWords["suffix"] = opt.Suffixes
opt.Word += "{@suffix}" opt.Word += "{?@suffix}"
} }
if len(opt.Prefixes) != 0 { if len(opt.Prefixes) != 0 {
mask.SpecialWords["prefix"] = opt.Prefixes mask.SpecialWords["prefix"] = opt.Prefixes
opt.Word = "{@prefix}" + opt.Word opt.Word = "{?@prefix}" + opt.Word
} }
if opt.ForceExtension && opt.Extensions != "" { if opt.ForceExtension && opt.Extensions != "" {
@ -356,7 +356,7 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
} }
} }
mask.SpecialWords["ext"] = exts mask.SpecialWords["ext"] = exts
opt.Word += "{@ext}" opt.Word += "{?@ext}"
} }
r.Wordlist, err = mask.Run(opt.Word, dicts, nil) r.Wordlist, err = mask.Run(opt.Word, dicts, nil)

View File

@ -710,7 +710,7 @@ func (pool *BrutePool) addFuzzyBaseline(bl *pkg.Baseline) {
func (pool *BrutePool) doBak() { func (pool *BrutePool) doBak() {
defer pool.wg.Done() 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 { if err != nil {
return 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 { if err != nil {
return return
} }