reworked template signer

projectdiscovery · Oct 3, 2023 · bf7b5af · bf7b5af
1 parent d8cb12e
commit bf7b5af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/v2/cmd/integration-test/integration-test.go b/v2/cmd/integration-test/integration-test.go
@@ -132,8 +132,10 @@ func runTests(customTemplatePaths []string) []string {
 	var failedTestTemplatePaths []string
 
 	for proto, testCaseInfos := range protocolTests {
-		if proto != "" && !strings.EqualFold(proto, protocol) {
-			continue
+		if protocol != "" {
+			if !strings.EqualFold(proto, protocol) {
+				continue
+			}
 		}
 		if len(customTemplatePaths) == 0 {
 			fmt.Printf("Running test cases for %q protocol\n", aurora.Blue(proto))

diff --git a/v2/pkg/templates/template_sign.go b/v2/pkg/templates/template_sign.go
@@ -60,11 +60,8 @@ func SignTemplate(templateSigner *signer.TemplateSigner, templatePath string) er
 		// if template not verified then sign it
 
 		if len(template.RequestsCode) > 0 {
-			//TODO: if template contains code protocol and digest then check if it belongs to user
-			// digestData := signer.GetSignatureFromData(bin)
-			// if len(digestData) != 0 {
-			// 	return fmt.Errorf("re-signing of code protocol templates is not supported")
-			// }
+			// verify using current template signer
+			// if verified then 
 		}
 
 		signatureData, err := templateSigner.Sign(bin, template)