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,14 +51,14 @@ class Connector(GenericConnector):
try:
return self.cursor.fetchall()
except (pymssql.ProgrammingError, pymssql.OperationalError, _mssql.MssqlDatabaseException), msg:
logger.warn("(remote) %s" % msg)
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
return None
def execute(self, query):
try:
self.cursor.execute(utf8encode(query))
except (pymssql.OperationalError, pymssql.ProgrammingError), msg:
logger.warn("(remote) %s" % msg)
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
except pymssql.InternalError, msg:
raise sqlmapConnectionException, msg