mirrors/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2025-07-14 13:22:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
影舞者 2022-05-12 11:56:01 +08:00
parent 55825f3b7c
commit 4915539fb3
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
WebScan/lib/check.go
View File

@ -53,7 +53,6 @@ func CheckMultiPoc(req *http.Request, pocs []*Poc, workers int) {
} }
func executePoc(oReq *http.Request, p *Poc) (bool, error, string) { func executePoc(oReq *http.Request, p *Poc) (bool, error, string) {
var lock sync.Mutex
c := NewEnvOption() c := NewEnvOption()
c.UpdateCompileOptions(p.Set) c.UpdateCompileOptions(p.Set)
if len(p.Sets) > 0 { if len(p.Sets) > 0 {
@ -164,6 +163,7 @@ func executePoc(oReq *http.Request, p *Poc) (bool, error, string) {
} }
DealWithRule := func(rule Rules) (bool, error) { DealWithRule := func(rule Rules) (bool, error) {
rule.Headers = cloneMap(rule.Headers)
var ( var (
flag, ok bool flag, ok bool
) )
@ -174,9 +174,7 @@ func executePoc(oReq *http.Request, p *Poc) (bool, error, string) {
} }
value := fmt.Sprintf("%v", v1) value := fmt.Sprintf("%v", v1)
for k2, v2 := range rule.Headers { for k2, v2 := range rule.Headers {
lock.Lock()
rule.Headers[k2] = strings.ReplaceAll(v2, "{{"+k1+"}}", value) rule.Headers[k2] = strings.ReplaceAll(v2, "{{"+k1+"}}", value)
lock.Unlock()
} }
rule.Path = strings.ReplaceAll(strings.TrimSpace(rule.Path), "{{"+k1+"}}", value) rule.Path = strings.ReplaceAll(strings.TrimSpace(rule.Path), "{{"+k1+"}}", value)
rule.Body = strings.ReplaceAll(strings.TrimSpace(rule.Body), "{{"+k1+"}}", value) rule.Body = strings.ReplaceAll(strings.TrimSpace(rule.Body), "{{"+k1+"}}", value)
@ -246,7 +244,7 @@ func executePoc(oReq *http.Request, p *Poc) (bool, error, string) {
if len(p.Rules) > 0 { if len(p.Rules) > 0 {
success = DealWithRules(p.Rules) success = DealWithRules(p.Rules)
} else { // Groups } else {
for name, rules := range p.Groups { for name, rules := range p.Groups {
success = DealWithRules(rules) success = DealWithRules(rules)
if success { if success {