Using static extensions from core, not a statically-defined list (#124)

* Using static extensions from core, not a statically-defined list * .
CERT-Polska · Sep 30, 2024 · 374cb3b · 374cb3b
1 parent 37cab43
commit 374cb3b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/karton_sqlmap/karton_sqlmap.py b/karton_sqlmap/karton_sqlmap.py
@@ -14,6 +14,7 @@
 from artemis.binds import Service, TaskStatus, TaskType
 from artemis.config import Config
 from artemis.module_base import ArtemisBase
+from artemis.modules.data.static_extensions import STATIC_EXTENSIONS
 from artemis.task_utils import get_target_url
 from bs4 import BeautifulSoup
 from karton.core import Task
@@ -301,10 +302,7 @@ def run(self, current_task: Task) -> None:
 
                 new_url = new_url.split("#")[0]
 
-                if any(
-                    new_url.split("?")[0].lower().endswith(extension)
-                    for extension in [".png", ".jpg", ".svg", ".jpeg", ".css", ".js", ".ico", ".woff2"]
-                ):
+                if any(new_url.split("?")[0].lower().endswith(extension) for extension in STATIC_EXTENSIONS):
                     # Let's not inject image/style paths
                     continue