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 {
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)

View File

@ -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
}