CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-04 06:50:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

some code refactoring

Browse source
This commit is contained in:
Miroslav Stampar 2010-04-16 19:57:00 +00:00
parent 1bdf94f236
commit 915d3441e9
5 changed files with 23 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

11
lib/parse/banner.py
View file

@ -24,11 +24,10 @@ Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
import re
from StringIO import StringIO
from xml.sax import parse
from xml.sax.handler import ContentHandler
from lib.core.common import checkFile
from lib.core.common import parseXmlFile
from lib.core.common import sanitizeStr
from lib.core.data import kb
from lib.core.data import paths
@ -122,11 +121,11 @@ def bannerParser(banner):
if kb.dbms == "Microsoft SQL Server":
handler = MSSQLBannerHandler(banner, kb.bannerFp)
parse(StringIO(open(xmlfile).read()), handler)
parseXmlFile(xmlfile, handler)
handler = FingerprintHandler(banner, kb.bannerFp)
parse(StringIO(open(paths.GENERIC_XML).read()), handler)
parseXmlFile(paths.GENERIC_XML, handler)
else:
handler = FingerprintHandler(banner, kb.bannerFp)
parse(StringIO(open(xmlfile).read()), handler)
parse(StringIO(open(paths.GENERIC_XML).read()), handler)
parseXmlFile(xmlfile, handler)
parseXmlFile(paths.GENERIC_XML, handler)