detecting simple script tag inside the markup

bgalek · Sep 16, 2024 · 17ef711 · 17ef711
1 parent ceb202a
commit 17ef711
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/main/java/com/github/bgalek/security/svg/SvgSecurityValidator.java b/src/main/java/com/github/bgalek/security/svg/SvgSecurityValidator.java
@@ -7,11 +7,7 @@
 import javax.xml.parsers.DocumentBuilder;
 import java.io.ByteArrayInputStream;
 import java.nio.charset.StandardCharsets;
-import java.util.Arrays;
-import java.util.Collections;
-import java.util.HashSet;
-import java.util.Objects;
-import java.util.Set;
+import java.util.*;
 import java.util.regex.Pattern;
 
 /**
@@ -25,6 +21,7 @@
 public class SvgSecurityValidator implements XssDetector {
 
     private static final Pattern JAVASCRIPT_PROTOCOL_IN_CSS_URL = Pattern.compile("url\\(.?javascript");
+    private static final Pattern SCRIPT_TAG = Pattern.compile("</?\\s*(?)script\\s*[a-zA-Z=/\"]*\\s*>", Pattern.CASE_INSENSITIVE);
 
     private final String[] svgElements;
     private final String[] svgAttributes;
@@ -81,6 +78,7 @@ private void validateXMLSchema(String input) {
 
     private Set<String> getOffendingElements(String xml) {
         if (JAVASCRIPT_PROTOCOL_IN_CSS_URL.matcher(xml).find()) return Collections.singleton("style");
+        if (SCRIPT_TAG.matcher(xml).find()) return Collections.singleton("script");
         PolicyFactory policy = new HtmlPolicyBuilder()
                 .allowElements(this.svgElements)
                 .allowAttributes(this.svgAttributes).globally()

diff --git a/src/main/java/com/github/bgalek/security/svg/ValidationResult.java b/src/main/java/com/github/bgalek/security/svg/ValidationResult.java
@@ -12,7 +12,7 @@ public interface ValidationResult {
     boolean hasViolations();
 
     /**
-     * @return list of invalid elements or attributes found in SVG content
+     * @return set of invalid elements or attributes found in SVG content
      */
     Set<String> getOffendingElements();
 }