mirror of https://github.com/daffainfo/nuclei.git
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package matchers
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"regexp"
|
|
|
|
"github.com/Knetic/govaluate"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators/common/dsl"
|
|
)
|
|
|
|
// CompileMatchers performs the initial setup operation on a matcher
|
|
func (m *Matcher) CompileMatchers() error {
|
|
var ok bool
|
|
|
|
// Support hexadecimal encoding for matchers too.
|
|
if m.Encoding == "hex" {
|
|
for i, word := range m.Words {
|
|
if decoded, err := hex.DecodeString(word); err == nil && len(decoded) > 0 {
|
|
m.Words[i] = string(decoded)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Setup the matcher type
|
|
m.matcherType, ok = MatcherTypes[m.Type]
|
|
if !ok {
|
|
return fmt.Errorf("unknown matcher type specified: %s", m.Type)
|
|
}
|
|
// By default, match on body if user hasn't provided any specific items
|
|
if m.Part == "" {
|
|
m.Part = "body"
|
|
}
|
|
|
|
// Compile the regexes
|
|
for _, regex := range m.Regex {
|
|
compiled, err := regexp.Compile(regex)
|
|
if err != nil {
|
|
return fmt.Errorf("could not compile regex: %s", regex)
|
|
}
|
|
m.regexCompiled = append(m.regexCompiled, compiled)
|
|
}
|
|
|
|
// Compile the dsl expressions
|
|
for _, expr := range m.DSL {
|
|
compiled, err := govaluate.NewEvaluableExpressionWithFunctions(expr, dsl.HelperFunctions())
|
|
if err != nil {
|
|
return fmt.Errorf("could not compile dsl: %s", expr)
|
|
}
|
|
m.dslCompiled = append(m.dslCompiled, compiled)
|
|
}
|
|
|
|
// Setup the condition type, if any.
|
|
if m.Condition != "" {
|
|
m.condition, ok = ConditionTypes[m.Condition]
|
|
if !ok {
|
|
return fmt.Errorf("unknown condition specified: %s", m.Condition)
|
|
}
|
|
} else {
|
|
m.condition = ORCondition
|
|
}
|
|
return nil
|
|
}
|