Fix tempalte-id validation

projectdiscovery · Jun 4, 2024 · 49ad3c4 · 49ad3c4
1 parent 5428d05
commit 49ad3c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/templates/templates.go b/pkg/templates/templates.go
@@ -326,6 +326,10 @@ func (template *Template) UnmarshalYAML(unmarshal func(interface{}) error) error
 	}
 	*template = Template(*alias)
 
+	if !ReTemplateID.MatchString(template.ID) {
+		return errorutil.New("template id must match expression %v", ReTemplateID).WithTag("invalid template")
+	}
+
 	if len(template.RequestsHTTP) > 0 || len(template.RequestsNetwork) > 0 {
 		_ = deprecatedProtocolNameTemplates.Set(template.ID, true)
 	}