From 9ccbdb3fdfc095982b06d90b82d842a48c824d0e Mon Sep 17 00:00:00 2001 From: stamparm Date: Mon, 15 Apr 2013 15:36:10 +0200 Subject: [PATCH] Added a check for an Issue #361 --- lib/utils/sqlalchemy.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/lib/utils/sqlalchemy.py b/lib/utils/sqlalchemy.py index fd176e8b0..52d0e8f20 100644 --- a/lib/utils/sqlalchemy.py +++ b/lib/utils/sqlalchemy.py @@ -6,6 +6,7 @@ See the file 'doc/COPYING' for copying permission """ import imp +import os import sys _sqlalchemy = None @@ -18,6 +19,7 @@ except ImportError: 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 plugins.generic.connector import Connector as GenericConnector class SQLAlchemy(GenericConnector): @@ -32,12 +34,19 @@ class SQLAlchemy(GenericConnector): try: if not self.port and self.db: - if "///" not in conf.direct: - conf.direct = conf.direct.replace("//", "///", 1) + if not os.path.exists(self.db): + raise SqlmapFilePathException, "missing database file '%s'" % 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) + engine = _sqlalchemy.create_engine(conf.direct, connect_args={'check_same_thread':False} if self.dialect == "sqlite" else {}) self.connection = engine.connect() + except SqlmapFilePathException: + raise except Exception, msg: raise SqlmapConnectionException(msg[0])