some update

This commit is contained in:
Miroslav Stampar
2010-04-06 14:59:31 +00:00
parent 60f04f0a41
commit c24f1cc07c
8 changed files with 45 additions and 39 deletions

View File

@@ -31,6 +31,7 @@ except ImportError, _:
from lib.core.data import conf
from lib.core.data import logger
from lib.core.exception import sqlmapConnectionException
from lib.utils.timeout import timeout
from plugins.generic.connector import Connector as GenericConnector
@@ -63,11 +64,10 @@ class Connector(GenericConnector):
self.connected()
def fetchall(self):
try:
return self.cursor.fetchall()
except (pymssql.ProgrammingError, pymssql.OperationalError, _mssql.MssqlDatabaseException), msg:
logger.log(8, msg)
return None
retVal = timeout(func=self.cursor.fetchall, duration=conf.timeout, default=None)
if self.exceptionMsg:
logger.log(8, self.exceptionMsg)
return retVal
def execute(self, query):
logger.debug(query)