code refactoring

This commit is contained in:
Miroslav Stampar 2010-12-15 12:50:56 +00:00
parent cda00c7501
commit de54219571
3 changed files with 4 additions and 3 deletions

View File

@ -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:

View File

@ -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

View File

@ -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: