admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-05-06 10:41:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

优化过滤规则,现在可以自定义white, black,fuzzy的状态码

Browse Source
This commit is contained in:
M09Ic 2022-11-29 15:16:33 +08:00
parent 779c98487c
commit f0362b08b0
2 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
internal/option.go
View File

@ -62,8 +62,9 @@ type ModeOptions struct {
CheckPeriod int `long:"check-period" default:"100"`
ErrPeriod int `long:"error-period" default:"10"`
BreakThreshold int `long:"error-threshold" default:"20"`
BlackStatus string `long:"black-status" default:"default"`
WhiteStatus string `long:"white-status" `
BlackStatus string `long:"black-status" default:"404,400,410"`
WhiteStatus string `long:"white-status" default:"200"`
FuzzyStatus string `long:"fuzzy-status" default:"403,500,501,502,503"`
}
type MiscOptions struct {
@ -128,7 +129,7 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
r.ErrPeriod = max
}
if opt.BlackStatus != "default" {
if opt.BlackStatus != "" {
for _, s := range strings.Split(opt.BlackStatus, ",") {
si, err := strconv.Atoi(s)
if err != nil {
@ -136,8 +137,6 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
}
BlackStatus = append(BlackStatus, si)
}
} else {
BlackStatus = []int{400, 404, 410}
}
if opt.WhiteStatus != "" {
@ -150,6 +149,16 @@ func (opt *Option) PrepareRunner() (*Runner, error) {
}
}
if opt.FuzzyStatus != "" {
for _, s := range strings.Split(opt.FuzzyStatus, ",") {
si, err := strconv.Atoi(s)
if err != nil {
return nil, err
}
FuzzyStatus = append(FuzzyStatus, si)
}
}
// prepare url
var urls []string
var file *os.File

4
internal/runner.go
View File

@ -17,8 +17,8 @@ import (
)
var (
WhiteStatus []int
BlackStatus []int
WhiteStatus = []int{200}
BlackStatus = []int{400, 404, 410}
FuzzyStatus = []int{403, 500, 501, 502, 503}
WAFStatus = []int{493, 418}
)