diff --git a/lib/core/agent.py b/lib/core/agent.py index e267a110c..f9d6f3f7f 100644 --- a/lib/core/agent.py +++ b/lib/core/agent.py @@ -13,6 +13,7 @@ from xml.etree import ElementTree as ET from lib.core.common import getCompiledRegex from lib.core.common import isDBMSVersionAtLeast +from lib.core.common import isTechniqueAvailable from lib.core.common import randomInt from lib.core.common import randomStr from lib.core.convert import urlencode @@ -77,7 +78,7 @@ class Agent: paramDict = conf.paramDict[kb.injection.place] origValue = paramDict[kb.injection.parameter] - if kb.technique and kb.technique in kb.injection.data: + if isTechniqueAvailable(kb.technique): where = kb.injection.data[kb.technique].where if where == 1: diff --git a/lib/core/option.py b/lib/core/option.py index 0a58853c9..abd799ab2 100644 --- a/lib/core/option.py +++ b/lib/core/option.py @@ -27,7 +27,6 @@ from lib.controller.checks import checkConnection from lib.core.common import getConsoleWidth from lib.core.common import getFileItems from lib.core.common import getFileType -from lib.core.common import isTechniqueAvailable from lib.core.common import normalizePath from lib.core.common import ntToPosixSlashes from lib.core.common import parseTargetDirect diff --git a/lib/takeover/web.py b/lib/takeover/web.py index 03350a196..0d19493fa 100644 --- a/lib/takeover/web.py +++ b/lib/takeover/web.py @@ -19,6 +19,7 @@ from lib.core.common import extractRegexResult from lib.core.common import getDirs from lib.core.common import getDocRoot from lib.core.common import ntToPosixSlashes +from lib.core.common import isTechniqueAvailable from lib.core.common import isWindowsDriveLetterPath from lib.core.common import normalizePath from lib.core.common import posixToNtSlashes @@ -104,7 +105,7 @@ class Web: uplQuery = fileContent.replace("WRITABLE_DIR", directory.replace('/', '\\\\') if kb.os == "Windows" else directory) query = "" - if kb.technique and kb.technique in kb.injection.data: + if isTechniqueAvailable(kb.technique): where = kb.injection.data[kb.technique].where if where == 2: