Introducing lib/utils/sqlalchemy.py (Issue #361)

lib/utils/sqlalchemy.py

@@ -0,0 +1,58 @@
+#!/usr/bin/env python
+
+"""
+Copyright (c) 2006-2013 sqlmap developers (http://sqlmap.org/)
+See the file 'doc/COPYING' for copying permission
+"""
+
+import imp
+import sys
+
+_sqlalchemy = None
+try:
+    f, pathname, desc = imp.find_module("sqlalchemy", sys.path[1:])
+    _sqlalchemy = imp.load_module("sqlalchemy", f, pathname, desc)
+except ImportError:
+    pass
+
+from lib.core.data import conf
+from lib.core.data import logger
+from lib.core.exception import SqlmapConnectionException
+from plugins.generic.connector import Connector as GenericConnector
+
+class SQLAlchemy(GenericConnector):
+    def __init__(self):
+        GenericConnector.__init__(self)
+
+    def connect(self):
+        self.initConnection()
+        try:
+            #_sqlalchemy.dialects.__all__
+            if not self.port and self.db:
+                if "///" not in conf.direct:
+                    conf.direct = conf.direct.replace("//", "///")
+            engine = _sqlalchemy.create_engine(conf.direct, connect_args={'check_same_thread':False})
+            self.connection = engine.connect()
+        except _sqlalchemy.exc.OperationalError, msg:
+            raise SqlmapConnectionException(msg[0])
+
+        self.connected()
+
+    def fetchall(self):
+        try:
+            return self.cursor.fetchall()
+        except _sqlalchemy.exc.ProgrammingError, msg:
+            logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg[1])
+            return None
+
+    def execute(self, query):
+        try:
+            self.cursor = self.connection.execute(query)
+        except (_sqlalchemy.exc.OperationalError, _sqlalchemy.exc.ProgrammingError), msg:
+            logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg[1])
+        except _sqlalchemy.exc.InternalError, msg:
+            raise SqlmapConnectionException(msg[1])
+
+    def select(self, query):
+        self.execute(query)
+        return self.fetchall()