From 374cb3b1ba192ce0cf46c69e9ef02ad0dffba6cc Mon Sep 17 00:00:00 2001 From: kazet Date: Mon, 30 Sep 2024 18:37:35 +0200 Subject: [PATCH] Using static extensions from core, not a statically-defined list (#124) * Using static extensions from core, not a statically-defined list * . --- karton_sqlmap/karton_sqlmap.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/karton_sqlmap/karton_sqlmap.py b/karton_sqlmap/karton_sqlmap.py index 03d1fa0..e63da04 100644 --- 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