sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-07-15 02:32:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Force pymssql to version >= 1.0.2

Browse Source
This commit is contained in:
Bernardo Damele 2010-03-31 15:31:11 +00:00
parent b19de015c5
commit cad8f61d55
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lib/core/common.py
View File

@ -651,6 +651,13 @@ def parseTargetDirect():
if dbmsName == "Microsoft SQL Server": if dbmsName == "Microsoft SQL Server":
import _mssql import _mssql
import pymssql import pymssql
if not hasattr(pymssql, "__version__") or pymssql.__version__ < "1.0.2":
errMsg = "pymssql library on your system must be "
errMsg += "version 1.0.2 to work, get it from "
errMsg += "http://sourceforge.net/projects/pymssql/files/pymssql/1.0.2/"
raise sqlmapMissingDependence, errMsg
elif dbmsName == "MySQL": elif dbmsName == "MySQL":
import MySQLdb import MySQLdb
elif dbmsName == "PostgreSQL": elif dbmsName == "PostgreSQL":

9
plugins/dbms/mssqlserver/connector.py
View File

@ -43,6 +43,9 @@ class Connector(GenericConnector):
License: LGPL License: LGPL
Possible connectors: http://wiki.python.org/moin/SQL%20Server Possible connectors: http://wiki.python.org/moin/SQL%20Server
Important note: pymssql library on your system MUST be version 1.0.2
to work, get it from http://sourceforge.net/projects/pymssql/files/pymssql/1.0.2/
""" """
def __init__(self): def __init__(self):
@ -79,6 +82,10 @@ class Connector(GenericConnector):
def select(self, query): def select(self, query):
self.execute(query) self.execute(query)
value = self.fetchall() value = self.fetchall()
self.connector.commit()
try:
self.connector.commit()
except pymssql.OperationalError:
pass
return value return value