mirror of
https://github.com/chainreactors/spray.git
synced 2025-09-15 19:50:18 +00:00
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
|
package pool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/chainreactors/logs"
|
||
|
|
"github.com/chainreactors/spray/pkg"
|
||
|
|
"github.com/chainreactors/words"
|
||
|
|
"github.com/chainreactors/words/rule"
|
||
|
|
"github.com/expr-lang/expr/vm"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
BaseURL string
|
||
|
|
ProxyAddr string
|
||
|
|
Thread int
|
||
|
|
Wordlist []string
|
||
|
|
Timeout time.Duration
|
||
|
|
ProcessCh chan *pkg.Baseline
|
||
|
|
OutputCh chan *pkg.Baseline
|
||
|
|
FuzzyCh chan *pkg.Baseline
|
||
|
|
Outwg *sync.WaitGroup
|
||
|
|
RateLimit int
|
||
|
|
CheckPeriod int
|
||
|
|
ErrPeriod int32
|
||
|
|
BreakThreshold int32
|
||
|
|
Method string
|
||
|
|
Mod SprayMod
|
||
|
|
Headers map[string]string
|
||
|
|
ClientType int
|
||
|
|
MatchExpr *vm.Program
|
||
|
|
FilterExpr *vm.Program
|
||
|
|
RecuExpr *vm.Program
|
||
|
|
AppendRule *rule.Program
|
||
|
|
Fns []words.WordFunc
|
||
|
|
AppendWords []string
|
||
|
|
Fuzzy bool
|
||
|
|
IgnoreWaf bool
|
||
|
|
Crawl bool
|
||
|
|
Scope []string
|
||
|
|
Active bool
|
||
|
|
Bak bool
|
||
|
|
Common bool
|
||
|
|
RetryLimit int
|
||
|
|
RandomUserAgent bool
|
||
|
|
Random string
|
||
|
|
Index string
|
||
|
|
MaxRedirect int
|
||
|
|
MaxCrawlDepth int
|
||
|
|
MaxRecursionDepth int
|
||
|
|
MaxAppendDepth int
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewBruteWords(config *Config, list []string) *words.Worder {
|
||
|
|
word := words.NewWorderWithList(list)
|
||
|
|
word.Fns = config.Fns
|
||
|
|
word.Run()
|
||
|
|
return word
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewBruteDSL(config *Config, dsl string, params [][]string) *words.Worder {
|
||
|
|
word, err := words.NewWorderWithDsl(dsl, params, nil)
|
||
|
|
if err != nil {
|
||
|
|
logs.Log.Error(err.Error())
|
||
|
|
}
|
||
|
|
word.Fns = config.Fns
|
||
|
|
word.Run()
|
||
|
|
return word
|
||
|
|
}
|