mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 02:06:46 +00:00
33 lines
771 B
Go
33 lines
771 B
Go
|
package templates
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/segmentio/ksuid"
|
||
|
|
)
|
||
|
|
|
||
|
|
var preprocessorRegex = regexp.MustCompile(`\{\{([a-z0-9_]+)\}\}`)
|
||
|
|
|
||
|
|
// expandPreprocessors expands the pre-processors if any for a template data.
|
||
|
|
func (t *Template) expandPreprocessors(data []byte) []byte {
|
||
|
|
foundMap := make(map[string]struct{})
|
||
|
|
|
||
|
|
for _, expression := range preprocessorRegex.FindAllStringSubmatch(string(data), -1) {
|
||
|
|
if len(expression) != 2 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
value := expression[1]
|
||
|
|
|
||
|
|
if _, ok := foundMap[value]; ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
foundMap[value] = struct{}{}
|
||
|
|
if strings.EqualFold(value, "randstr") || strings.HasPrefix(value, "randstr_") {
|
||
|
|
data = bytes.ReplaceAll(data, []byte(expression[0]), []byte(ksuid.New().String()))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return data
|
||
|
|
}
|