Patch for an Issue #212

This commit is contained in:
Miroslav Stampar
2012-10-23 15:34:59 +02:00
parent eb6f17b561
commit 6e2fce66aa
8 changed files with 34 additions and 16 deletions

View File

@@ -11,6 +11,8 @@ try:
except ImportError, _:
pass
import logging
from lib.core.convert import utf8encode
from lib.core.data import conf
from lib.core.data import logger
@@ -49,7 +51,7 @@ class Connector(GenericConnector):
try:
return self.cursor.fetchall()
except (pymssql.ProgrammingError, pymssql.OperationalError, _mssql.MssqlDatabaseException), msg:
logger.warn("(remote) %s" % str(msg).replace("\n", " "))
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % str(msg).replace("\n", " "))
return None
def execute(self, query):
@@ -59,7 +61,7 @@ class Connector(GenericConnector):
self.cursor.execute(utf8encode(query))
retVal = True
except (pymssql.OperationalError, pymssql.ProgrammingError), msg:
logger.warn("(remote) %s" % str(msg).replace("\n", " "))
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % str(msg).replace("\n", " "))
except pymssql.InternalError, msg:
raise sqlmapConnectionException, msg