Force pymssql to version >= 1.0.2

bdamele · bdamele · commit cad8f61d5521 · 2010-03-31T15:31:11.000Z
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -651,6 +651,13 @@ def parseTargetDirect():
                 if dbmsName == "Microsoft SQL Server":
                     import _mssql
                     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":
                     import MySQLdb
                 elif dbmsName == "PostgreSQL":
diff --git a/plugins/dbms/mssqlserver/connector.py b/plugins/dbms/mssqlserver/connector.py
@@ -43,6 +43,9 @@ class Connector(GenericConnector):
     License: LGPL
 
     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):
@@ -79,6 +82,10 @@ def execute(self, query):
     def select(self, query):
         self.execute(query)
         value = self.fetchall()
-        self.connector.commit()
+
+        try:
+            self.connector.commit()
+        except pymssql.OperationalError:
+            pass
 
         return value
