Minor fix for --forms

This commit is contained in:
Miroslav Stampar 2014-12-10 12:13:37 +01:00
parent d700e50b36
commit ee20d98bca

View File

@ -3472,18 +3472,18 @@ def findPageForms(content, url, raise_=False, addToTargets=False):
except UnicodeError: except UnicodeError:
pass pass
except ParseError: except ParseError:
warnMsg = "badly formed HTML at the given URL ('%s'). Going to filter it" % url if "<html" in (content or ""):
logger.warning(warnMsg) warnMsg = "badly formed HTML at the given URL ('%s'). Going to filter it" % url
response.seek(0) logger.warning(warnMsg)
filtered = _("".join(re.findall(FORM_SEARCH_REGEX, response.read())), response.geturl()) filtered = _("".join(re.findall(FORM_SEARCH_REGEX, content)), url)
try: try:
forms = ParseResponse(filtered, backwards_compat=False) forms = ParseResponse(filtered, backwards_compat=False)
except ParseError: except ParseError:
errMsg = "no success" errMsg = "no success"
if raise_: if raise_:
raise SqlmapGenericException(errMsg) raise SqlmapGenericException(errMsg)
else: else:
logger.debug(errMsg) logger.debug(errMsg)
if forms: if forms:
for form in forms: for form in forms: