minor refactoring

This commit is contained in:
Miroslav Stampar 2010-12-06 07:48:14 +00:00
parent a43d252ae9
commit e8be14e00a

View File

@ -11,6 +11,7 @@ import re
import time
from lib.core.agent import agent
from lib.core.common import extractRegexResult
from lib.core.common import getUnicode
from lib.core.common import randomInt
from lib.core.common import replaceNewlineTabs
@ -55,10 +56,7 @@ def errorUse(expression):
payload = agent.payload(newValue=expression)
reqBody, _ = Request.queryPage(payload, content=True)
match = re.search(check, reqBody, re.DOTALL | re.IGNORECASE)
if match:
output = match.group('result')
output = extractRegexResult(check, reqBody, re.DOTALL | re.IGNORECASE)
if output:
output = output.replace(ERROR_SPACE, " ").replace(ERROR_EMPTY_CHAR, "")