#!/usr/bin/env python

"""
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
See the file 'LICENSE' for copying permission
"""

import imp
import logging
import os
import sys
import traceback
import warnings

_sqlalchemy = None
try:
 f, pathname, desc = imp.find_module("sqlalchemy", sys.path[1:])
 _ = imp.load_module("sqlalchemy", f, pathname, desc)
 if hasattr(_, "dialects"):
 _sqlalchemy = _
 warnings.simplefilter(action="ignore", category=_sqlalchemy.exc.SAWarning)
except ImportError:
 pass

try:
 import MySQLdb # used by SQLAlchemy in case of MySQL
 warnings.filterwarnings("error", category=MySQLdb.Warning)
except ImportError:
 pass

from lib.core.data import conf
from lib.core.data import logger
from lib.core.exception import SqlmapConnectionException
from lib.core.exception import SqlmapFilePathException
from lib.core.exception import SqlmapMissingDependence
from plugins.generic.connector import Connector as GenericConnector

def getSafeExString(ex, encoding=None): # Cross-referenced function
 raise NotImplementedError

class SQLAlchemy(GenericConnector):
 def __init__(self, dialect=None):
 GenericConnector.__init__(self)
 self.dialect = dialect

 def connect(self):
 if _sqlalchemy:
 self.initConnection()

 try:
 if not self.port and self.db:
 if not os.path.exists(self.db):
 raise SqlmapFilePathException("the provided database file '%s' does not exist" % self.db)

 _ = conf.direct.split("//", 1)
 conf.direct = "%s////%s" % (_[0], os.path.abspath(self.db))

 if self.dialect:
 conf.direct = conf.direct.replace(conf.dbms, self.dialect, 1)

 if self.dialect == "sqlite":
 engine = _sqlalchemy.create_engine(conf.direct, connect_args={"check_same_thread": False})
 elif self.dialect == "oracle":
 engine = _sqlalchemy.create_engine(conf.direct)
 else:
 engine = _sqlalchemy.create_engine(conf.direct, connect_args={})

 self.connector = engine.connect()
 except (TypeError, ValueError):
 if "_get_server_version_info" in traceback.format_exc():
 try:
 import pymssql
 if int(pymssql.__version__[0]) < 2:
 raise SqlmapConnectionException("SQLAlchemy connection issue (obsolete version of pymssql ('%s') is causing problems)" % pymssql.__version__)
 except ImportError:
 pass
 elif "invalid literal for int() with base 10: '0b" in traceback.format_exc():
 raise SqlmapConnectionException("SQLAlchemy connection issue ('https://bitbucket.org/zzzeek/sqlalchemy/issues/3975')")
 raise
 except SqlmapFilePathException:
 raise
 except Exception as ex:
 raise SqlmapConnectionException("SQLAlchemy connection issue ('%s')" % getSafeExString(ex))

 self.printConnected()
 else:
 raise SqlmapMissingDependence("SQLAlchemy not available")

 def fetchall(self):
 try:
 retVal = []
 for row in self.cursor.fetchall():
 retVal.append(tuple(row))
 return retVal
 except _sqlalchemy.exc.ProgrammingError as ex:
 logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % getSafeExString(ex))
 return None

 def execute(self, query):
 try:
 self.cursor = self.connector.execute(query)
 except (_sqlalchemy.exc.OperationalError, _sqlalchemy.exc.ProgrammingError) as ex:
 logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % getSafeExString(ex))
 except _sqlalchemy.exc.InternalError as ex:
 raise SqlmapConnectionException(getSafeExString(ex))

 def select(self, query):
 self.execute(query)
 return self.fetchall()