minor refactoring

This commit is contained in:
Miroslav Stampar 2010-10-19 23:05:24 +00:00
parent 1b376c99a6
commit 8776db872c
2 changed files with 7 additions and 1 deletions

View File

@ -33,6 +33,10 @@ LOGGER_HANDLER.setFormatter(FORMATTER)
LOGGER.addHandler(LOGGER_HANDLER) LOGGER.addHandler(LOGGER_HANDLER)
LOGGER.setLevel(logging.WARN) LOGGER.setLevel(logging.WARN)
# error based injection
ERROR_SPACE = "%c%c%c" % (58, 95, 58)
ERROR_EMPTY_CHAR = "%c%c%c" % (58, 120, 58)
# System variables # System variables
IS_WIN = subprocess.mswindows IS_WIN = subprocess.mswindows
# The name of the operating system dependent module imported. The following # The name of the operating system dependent module imported. The following

View File

@ -27,6 +27,8 @@ from lib.core.data import conf
from lib.core.data import kb from lib.core.data import kb
from lib.core.data import logger from lib.core.data import logger
from lib.core.data import queries from lib.core.data import queries
from lib.core.settings import ERROR_SPACE
from lib.core.settings import ERROR_EMPTY_CHAR
from lib.core.unescaper import unescaper from lib.core.unescaper import unescaper
from lib.request.connect import Connect as Request from lib.request.connect import Connect as Request
from lib.request.direct import direct from lib.request.direct import direct
@ -359,7 +361,7 @@ def __goError(expression, resumeValue=True):
if match: if match:
output = match.group('result') output = match.group('result')
if output: if output:
output = output.replace("%c%c%c" % (58, 95, 58), " ").replace("%c%c%c" % (58, 120, 58), "") #':_:' -> EMPTY CHAR, ':x:' -> SPACE CHAR output = output.replace(ERROR_SPACE, " ").replace(ERROR_EMPTY_CHAR, "")
if kb.misc.testedDbms == 'MySQL': if kb.misc.testedDbms == 'MySQL':
output = output[:-1] output = output[:-1]