admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-09-15 19:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

优化options的细节, 递归深度的选择

Browse Source
This commit is contained in:
M09Ic 2023-01-04 00:15:34 +08:00
parent 5142012ceb
commit ffd387a28e
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/option.go
View File

@ -30,7 +30,7 @@ type Option struct {
} }
type InputOptions struct { type InputOptions struct {
ResumeFrom string `long:"resume-from"` ResumeFrom string `long:"resume"`
URL string `short:"u" long:"url" description:"String, input baseurl (separated by commas), e.g.: http://google.com, http://baidu.com"` URL string `short:"u" long:"url" description:"String, input baseurl (separated by commas), e.g.: http://google.com, http://baidu.com"`
URLFile string `short:"l" long:"list" description:"File, input filename"` URLFile string `short:"l" long:"list" description:"File, input filename"`
Raw string `long:"raw" description:"File, input raw request filename"` Raw string `long:"raw" description:"File, input raw request filename"`
@ -82,7 +82,7 @@ type ModeOptions struct {
Depth int `long:"depth" default:"0" description:"Int, recursive depth"` Depth int `long:"depth" default:"0" description:"Int, recursive depth"`
Active bool `long:"active" description:"Bool, enable active finger detect"` Active bool `long:"active" description:"Bool, enable active finger detect"`
Crawl bool `long:"crawl" description:"Bool, enable crawl"` Crawl bool `long:"crawl" description:"Bool, enable crawl"`
CrawlDepth int `long:"spider-depth" default:"3" description:"Int, crawl depth"` CrawlDepth int `long:"crawl-depth" default:"3" description:"Int, crawl depth"`
CheckPeriod int `long:"check-period" default:"200" description:"Int, check period when request"` CheckPeriod int `long:"check-period" default:"200" description:"Int, check period when request"`
ErrPeriod int `long:"error-period" default:"10" description:"Int, check period when error"` ErrPeriod int `long:"error-period" default:"10" description:"Int, check period when error"`
BreakThreshold int `long:"error-threshold" default:"20" description:"Int, break when the error exceeds the threshold "` BreakThreshold int `long:"error-threshold" default:"20" description:"Int, break when the error exceeds the threshold "`
@ -400,13 +400,17 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
r.FilterExpr = exp r.FilterExpr = exp
} }
if opt.Depth > 0 { if opt.Recursive != "current.IsDir()" {
maxRecursion = 1
exp, err := expr.Compile(opt.Recursive) exp, err := expr.Compile(opt.Recursive)
if err != nil { if err != nil {
return nil, err return nil, err
} }
r.RecursiveExpr = exp r.RecursiveExpr = exp
} }
if opt.Depth != 0 {
maxRecursion = opt.Depth
}
// prepare header // prepare header
for _, h := range opt.Headers { for _, h := range opt.Headers {