Minor improvement

This commit is contained in:
Bernardo Damele 2011-01-15 23:54:03 +00:00
parent 29ea0950b6
commit 02b333e30b
2 changed files with 27 additions and 2 deletions

View File

@ -280,7 +280,10 @@ class Fingerprint(GenericFingerprint):
elif result is False: elif result is False:
kb.os = "Windows" kb.os = "Windows"
infoMsg = "the back-end DBMS operating system is %s" % (kb.os or "Unknown") if kb.os:
logger.info(infoMsg) infoMsg = "the back-end DBMS operating system is %s" % kb.os
logger.info(infoMsg)
else:
self.userChooseDbmsOs()
self.cleanup(onlyFileTbl=True) self.cleanup(onlyFileTbl=True)

View File

@ -7,7 +7,9 @@ Copyright (c) 2006-2010 sqlmap developers (http://sqlmap.sourceforge.net/)
See the file 'doc/COPYING' for copying permission See the file 'doc/COPYING' for copying permission
""" """
from lib.core.common import readInput
from lib.core.data import kb from lib.core.data import kb
from lib.core.data import logger
from lib.core.exception import sqlmapUndefinedMethod from lib.core.exception import sqlmapUndefinedMethod
class Fingerprint: class Fingerprint:
@ -35,3 +37,23 @@ class Fingerprint:
def forceDbmsEnum(self): def forceDbmsEnum(self):
pass pass
def userChooseDbmsOs(self):
warnMsg = "for some reason sqlmap was unable to fingerprint "
warnMsg += "the back-end DBMS operating system"
logger.warn(warnMsg)
msg = "do you want to provide the OS? [(W)indows/(l)inux]"
while True:
os = readInput(msg, default="W")
if os[0].lower() == "w":
kb.os = "Windows"
break
elif os[0].lower() == "l":
kb.os = "Linux"
break
else:
warnMsg = "invalid value"
logger.warn(warnMsg)