sqlmap/lib/parse/html.py

66 lines
1.7 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2008-10-15 19:38:22 +04:00
"""
2015-01-06 17:02:16 +03:00
Copyright (c) 2006-2015 sqlmap developers (http://sqlmap.org/)
2010-10-15 03:18:29 +04:00
See the file 'doc/COPYING' for copying permission
2008-10-15 19:38:22 +04:00
"""
import re
from xml.sax.handler import ContentHandler
2010-04-16 23:57:00 +04:00
from lib.core.common import parseXmlFile
from lib.core.data import kb
from lib.core.data import paths
from lib.core.threads import getCurrentThreadData
2008-10-15 19:38:22 +04:00
class HTMLHandler(ContentHandler):
2008-10-15 19:38:22 +04:00
"""
This class defines methods to parse the input HTML page to
fingerprint the back-end database management system
"""
def __init__(self, page):
2011-01-15 15:53:40 +03:00
ContentHandler.__init__(self)
self._dbms = None
self._page = page
2008-10-15 19:38:22 +04:00
2011-04-30 17:20:05 +04:00
self.dbms = None
2008-10-15 19:38:22 +04:00
def _markAsErrorPage(self):
threadData = getCurrentThreadData()
threadData.lastErrorPage = (threadData.lastRequestUID, self._page)
2008-10-15 19:38:22 +04:00
def startElement(self, name, attrs):
if name == "dbms":
self._dbms = attrs.get("value")
2008-10-15 19:38:22 +04:00
2011-11-16 20:06:21 +04:00
elif name == "error":
if re.search(attrs.get("regexp"), self._page, re.I):
self.dbms = self._dbms
self._markAsErrorPage()
2008-10-15 19:38:22 +04:00
2008-11-17 03:13:49 +03:00
def htmlParser(page):
2008-10-15 19:38:22 +04:00
"""
This function calls a class that parses the input HTML page to
fingerprint the back-end database management system
"""
2008-11-17 03:13:49 +03:00
xmlfile = paths.ERRORS_XML
handler = HTMLHandler(page)
2011-01-23 14:21:27 +03:00
parseXmlFile(xmlfile, handler)
2008-10-15 19:38:22 +04:00
if handler.dbms and handler.dbms not in kb.htmlFp:
kb.lastParserStatus = handler.dbms
kb.htmlFp.append(handler.dbms)
else:
kb.lastParserStatus = None
# generic SQL warning/error messages
if re.search(r"SQL (warning|error|syntax)", page, re.I):
handler._markAsErrorPage()
2008-10-15 19:38:22 +04:00
return handler.dbms