nuclei/v2/pkg/protocols/http/request_condition.go
Mzack9999 9493dfdb20
Adding automatic request condition detection (#2707)
* Adding automatic request condition detection

* adding missing checks on part

* test update as per latest change

Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
2022-10-15 15:19:04 +05:30

42 lines
919 B
Go

package http
import (
"regexp"
)
var (
// Determines if request condition are needed by detecting the pattern _xxx
reRequestCondition = regexp.MustCompile(`(?m)_\d+`)
)
// NeedsRequestCondition determines if request condition should be enabled
func (request *Request) NeedsRequestCondition() bool {
for _, matcher := range request.Matchers {
if checkRequestConditionExpressions(matcher.DSL...) {
return true
}
if checkRequestConditionExpressions(matcher.Part) {
return true
}
}
for _, extractor := range request.Extractors {
if checkRequestConditionExpressions(extractor.DSL...) {
return true
}
if checkRequestConditionExpressions(extractor.Part) {
return true
}
}
return false
}
func checkRequestConditionExpressions(expressions ...string) bool {
for _, expression := range expressions {
if reRequestCondition.MatchString(expression) {
return true
}
}
return false
}