minor update

This commit is contained in:
Miroslav Stampar 2010-12-15 11:46:28 +00:00
parent 4dec24d056
commit 7cfeb5447b
2 changed files with 8 additions and 4 deletions

View File

@ -1754,3 +1754,6 @@ def getTechniqueData(technique=None):
retVal = kb.injection.data[technique] retVal = kb.injection.data[technique]
return retVal return retVal
def isTechniqueAvailable(technique=None):
return getTechniqueData(technique) is not None

View File

@ -27,6 +27,7 @@ from lib.controller.checks import checkConnection
from lib.core.common import getConsoleWidth from lib.core.common import getConsoleWidth
from lib.core.common import getFileItems from lib.core.common import getFileItems
from lib.core.common import getFileType from lib.core.common import getFileType
from lib.core.common import isTechniqueAvailable
from lib.core.common import normalizePath from lib.core.common import normalizePath
from lib.core.common import ntToPosixSlashes from lib.core.common import ntToPosixSlashes
from lib.core.common import parseTargetDirect from lib.core.common import parseTargetDirect
@ -1133,10 +1134,10 @@ def __setKnowledgeBaseAttributes():
kb.data = advancedDict() kb.data = advancedDict()
# Injection types # Injection types
kb.booleanTest = property(lambda self: getTechniqueData(PAYLOAD.TECHNIQUE.BOOLEAN) is not None) kb.booleanTest = property(lambda self: isTechniqueAvailable(PAYLOAD.TECHNIQUE.BOOLEAN))
kb.errorTest = property(lambda self: getTechniqueData(PAYLOAD.TECHNIQUE.ERROR) is not None) kb.errorTest = property(lambda self: isTechniqueAvailable(PAYLOAD.TECHNIQUE.ERROR))
kb.stackedTest = property(lambda self: getTechniqueData(PAYLOAD.TECHNIQUE.STACKED) is not None) kb.stackedTest = property(lambda self: isTechniqueAvailable(PAYLOAD.TECHNIQUE.STACKED))
kb.timeTest = property(lambda self: getTechniqueData(PAYLOAD.TECHNIQUE.TIME) is not None) kb.timeTest = property(lambda self: isTechniqueAvailable(PAYLOAD.TECHNIQUE.TIME))
kb.unionTest = None kb.unionTest = None
# Basic back-end DBMS fingerprint # Basic back-end DBMS fingerprint