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

fix fuzzy output when not --fuzzy flag

Browse Source
This commit is contained in:
M09Ic 2024-08-28 12:39:00 +08:00
parent 15110ab895
commit d6e7e58b18
2 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/option.go
View File

@ -134,7 +134,7 @@ type ModeOptions struct {
UniqueStatus string `long:"unique-status" default:"403,200,404" description:"Strings (comma split), custom unique status" config:"unique-status"` UniqueStatus string `long:"unique-status" default:"403,200,404" description:"Strings (comma split), custom unique status" config:"unique-status"`
Unique bool `long:"unique" description:"Bool, unique response" config:"unique"` Unique bool `long:"unique" description:"Bool, unique response" config:"unique"`
RetryCount int `long:"retry" default:"0" description:"Int, retry count" config:"retry"` RetryCount int `long:"retry" default:"0" description:"Int, retry count" config:"retry"`
SimhashDistance int `long:"sim-distance" default:"5" config:"sim-distance"` SimhashDistance int `long:"sim-distance" default:"8" config:"sim-distance"`
} }
type MiscOptions struct { type MiscOptions struct {
@ -250,6 +250,9 @@ func (opt *Option) Prepare() error {
} else { } else {
pkg.UniqueStatus = pkg.ParseStatus(pkg.UniqueStatus, opt.UniqueStatus) pkg.UniqueStatus = pkg.ParseStatus(pkg.UniqueStatus, opt.UniqueStatus)
} }
logs.Log.Logf(pkg.LogVerbose, "Black Status: %v, WhiteStatus: %v, WAFStatus: %v", pkg.BlackStatus, pkg.WhiteStatus, pkg.WAFStatus)
logs.Log.Logf(pkg.LogVerbose, "Fuzzy Status: %v, Unique Status: %v", pkg.FuzzyStatus, pkg.UniqueStatus)
pool.MaxCrawl = opt.CrawlDepth pool.MaxCrawl = opt.CrawlDepth
return nil return nil
@ -403,18 +406,6 @@ func (opt *Option) NewRunner() (*Runner, error) {
} }
} }
//if opt.FuzzyFile != "" {
// r.FuzzyFile, err = files.NewFile(opt.FuzzyFile, false, false, true)
// if err != nil {
// return nil, err
// }
//} else if opt.AutoFile {
// r.FuzzyFile, err = files.NewFile("fuzzy.json", false, false, true)
// if err != nil {
// return nil, err
// }
//}
if opt.DumpFile != "" { if opt.DumpFile != "" {
r.DumpFile, err = files.NewFile(opt.DumpFile, false, false, true) r.DumpFile, err = files.NewFile(opt.DumpFile, false, false, true)
if err != nil { if err != nil {

6
internal/runner.go
View File

@ -412,8 +412,10 @@ func (r *Runner) OutputHandler() {
if !ok { if !ok {
return return
} }
r.Output(bl) if r.Fuzzy {
r.outwg.Done() r.Output(bl)
r.outwg.Done()
}
} }
} }
}() }()