mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 02:45:26 +00:00
8f313629b8
* Replaced strings.Replaced with fasttemplate reducing allocations Custom template parsing logic was replaced with fasttemplate package for reducing allocations in the replacer.Replace hotpath leading to allocation reduction which accounted for 30% of total nuclei allocations. $ go test -bench=. -benchmem goos: darwin goarch: arm64 pkg: github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/replacer BenchmarkReplacer-8 837232 1422 ns/op 2112 B/op 31 allocs/op BenchmarkReplacerNew-8 3672765 320.3 ns/op 48 B/op 4 allocs/op * Fixed tests failing * Use pre-compiled map of DSL expressions * Reworked expression parsing logic to reduce memory allocations $ go test -bench=. -benchmem goos: darwin goarch: arm64 pkg: github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/expressions BenchmarkEvaluate-8 31560 37769 ns/op 31731 B/op 265 allocs/op BenchmarkEvaluateNew-8 109144 9621 ns/op 6253 B/op 116 allocs/op
92 lines
3.2 KiB
Go
92 lines
3.2 KiB
Go
package matchers
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Knetic/govaluate"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators/common/dsl"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWordANDCondition(t *testing.T) {
|
|
m := &Matcher{condition: ANDCondition, Words: []string{"a", "b"}}
|
|
|
|
isMatched, matched := m.MatchWords("a b", nil)
|
|
require.True(t, isMatched, "Could not match words with valid AND condition")
|
|
require.Equal(t, m.Words, matched)
|
|
|
|
isMatched, matched = m.MatchWords("b", nil)
|
|
require.False(t, isMatched, "Could match words with invalid AND condition")
|
|
require.Equal(t, []string{}, matched)
|
|
}
|
|
|
|
func TestRegexANDCondition(t *testing.T) {
|
|
m := &Matcher{Type: MatcherTypeHolder{MatcherType: RegexMatcher}, Condition: "and", Regex: []string{"[a-z]{3}", "\\d{2}"}}
|
|
err := m.CompileMatchers()
|
|
require.Nil(t, err)
|
|
|
|
isMatched, matched := m.MatchRegex("abc abcd 123")
|
|
require.True(t, isMatched, "Could not match regex with valid AND condition")
|
|
require.Equal(t, []string{"abc", "abc", "12"}, matched)
|
|
|
|
isMatched, matched = m.MatchRegex("bc 1")
|
|
require.False(t, isMatched, "Could match regex with invalid AND condition")
|
|
require.Equal(t, []string{}, matched)
|
|
}
|
|
|
|
func TestORCondition(t *testing.T) {
|
|
m := &Matcher{condition: ORCondition, Words: []string{"a", "b"}}
|
|
|
|
isMatched, matched := m.MatchWords("a b", nil)
|
|
require.True(t, isMatched, "Could not match valid word OR condition")
|
|
require.Equal(t, []string{"a"}, matched)
|
|
|
|
isMatched, matched = m.MatchWords("b", nil)
|
|
require.True(t, isMatched, "Could not match valid word OR condition")
|
|
require.Equal(t, []string{"b"}, matched)
|
|
|
|
isMatched, matched = m.MatchWords("c", nil)
|
|
require.False(t, isMatched, "Could match invalid word OR condition")
|
|
require.Equal(t, []string{}, matched)
|
|
}
|
|
|
|
func TestRegexOrCondition(t *testing.T) {
|
|
m := &Matcher{Type: MatcherTypeHolder{MatcherType: RegexMatcher}, Condition: "or", Regex: []string{"[a-z]{3}", "\\d{2}"}}
|
|
err := m.CompileMatchers()
|
|
require.Nil(t, err)
|
|
|
|
isMatched, matched := m.MatchRegex("ab 123")
|
|
require.True(t, isMatched, "Could not match valid regex OR condition")
|
|
require.Equal(t, []string{"12"}, matched)
|
|
|
|
isMatched, matched = m.MatchRegex("bc 1")
|
|
require.False(t, isMatched, "Could match invalid regex OR condition")
|
|
require.Equal(t, []string{}, matched)
|
|
}
|
|
|
|
func TestHexEncoding(t *testing.T) {
|
|
m := &Matcher{Encoding: "hex", Type: MatcherTypeHolder{MatcherType: WordsMatcher}, Part: "body", Words: []string{"50494e47"}}
|
|
err := m.CompileMatchers()
|
|
require.Nil(t, err, "could not compile matcher")
|
|
|
|
isMatched, matched := m.MatchWords("PING", nil)
|
|
require.True(t, isMatched, "Could not match valid Hex condition")
|
|
require.Equal(t, m.Words, matched)
|
|
}
|
|
|
|
func TestMatcher_MatchDSL(t *testing.T) {
|
|
compiled, err := govaluate.NewEvaluableExpressionWithFunctions("contains(body, \"{{VARIABLE}}\")", dsl.HelperFunctions)
|
|
require.Nil(t, err, "couldn't compile expression")
|
|
|
|
m := &Matcher{Type: MatcherTypeHolder{MatcherType: DSLMatcher}, dslCompiled: []*govaluate.EvaluableExpression{compiled}}
|
|
err = m.CompileMatchers()
|
|
require.Nil(t, err, "could not compile matcher")
|
|
|
|
values := []string{"PING", "pong"}
|
|
|
|
for value := range values {
|
|
isMatched := m.MatchDSL(map[string]interface{}{"body": value, "VARIABLE": value})
|
|
require.True(t, isMatched)
|
|
}
|
|
}
|