mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-30 05:23:50 +03:00
102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2013 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'doc/COPYING' for copying permission
|
|
"""
|
|
|
|
from lib.core.common import Backend
|
|
from lib.core.data import conf
|
|
from lib.core.data import logger
|
|
from lib.core.dicts import DBMS_DICT
|
|
from lib.core.enums import DBMS
|
|
from lib.core.exception import SqlmapConnectionException
|
|
from lib.core.settings import MSSQL_ALIASES
|
|
from lib.core.settings import MYSQL_ALIASES
|
|
from lib.core.settings import ORACLE_ALIASES
|
|
from lib.core.settings import PGSQL_ALIASES
|
|
from lib.core.settings import SQLITE_ALIASES
|
|
from lib.core.settings import ACCESS_ALIASES
|
|
from lib.core.settings import FIREBIRD_ALIASES
|
|
from lib.core.settings import MAXDB_ALIASES
|
|
from lib.core.settings import SYBASE_ALIASES
|
|
from lib.core.settings import DB2_ALIASES
|
|
from lib.utils.sqlalchemy import SQLAlchemy
|
|
|
|
from plugins.dbms.mssqlserver import MSSQLServerMap
|
|
from plugins.dbms.mssqlserver.connector import Connector as MSSQLServerConn
|
|
from plugins.dbms.mysql import MySQLMap
|
|
from plugins.dbms.mysql.connector import Connector as MySQLConn
|
|
from plugins.dbms.oracle import OracleMap
|
|
from plugins.dbms.oracle.connector import Connector as OracleConn
|
|
from plugins.dbms.postgresql import PostgreSQLMap
|
|
from plugins.dbms.postgresql.connector import Connector as PostgreSQLConn
|
|
from plugins.dbms.sqlite import SQLiteMap
|
|
from plugins.dbms.sqlite.connector import Connector as SQLiteConn
|
|
from plugins.dbms.access import AccessMap
|
|
from plugins.dbms.access.connector import Connector as AccessConn
|
|
from plugins.dbms.firebird import FirebirdMap
|
|
from plugins.dbms.firebird.connector import Connector as FirebirdConn
|
|
from plugins.dbms.maxdb import MaxDBMap
|
|
from plugins.dbms.maxdb.connector import Connector as MaxDBConn
|
|
from plugins.dbms.sybase import SybaseMap
|
|
from plugins.dbms.sybase.connector import Connector as SybaseConn
|
|
from plugins.dbms.db2 import DB2Map
|
|
from plugins.dbms.db2.connector import Connector as DB2Conn
|
|
|
|
def setHandler():
|
|
"""
|
|
Detect which is the target web application back-end database
|
|
management system.
|
|
"""
|
|
|
|
items = [
|
|
(DBMS.MYSQL, MYSQL_ALIASES, MySQLMap, MySQLConn),
|
|
(DBMS.ORACLE, ORACLE_ALIASES, OracleMap, OracleConn),
|
|
(DBMS.PGSQL, PGSQL_ALIASES, PostgreSQLMap, PostgreSQLConn),
|
|
(DBMS.MSSQL, MSSQL_ALIASES, MSSQLServerMap, MSSQLServerConn),
|
|
(DBMS.SQLITE, SQLITE_ALIASES, SQLiteMap, SQLiteConn),
|
|
(DBMS.ACCESS, ACCESS_ALIASES, AccessMap, AccessConn),
|
|
(DBMS.FIREBIRD, FIREBIRD_ALIASES, FirebirdMap, FirebirdConn),
|
|
(DBMS.MAXDB, MAXDB_ALIASES, MaxDBMap, MaxDBConn),
|
|
(DBMS.SYBASE, SYBASE_ALIASES, SybaseMap, SybaseConn),
|
|
(DBMS.DB2, DB2_ALIASES, DB2Map, DB2Conn),
|
|
]
|
|
|
|
_ = max(_ if (Backend.getIdentifiedDbms() or "").lower() in _[1] else None for _ in items)
|
|
if _:
|
|
items.remove(_)
|
|
items.insert(0, _)
|
|
|
|
for name, aliases, Handler, Connector in items:
|
|
if conf.dbms and conf.dbms not in aliases:
|
|
debugMsg = "skipping test for %s" % name
|
|
logger.debug(debugMsg)
|
|
continue
|
|
|
|
handler = Handler()
|
|
conf.dbmsConnector = Connector()
|
|
|
|
if conf.direct:
|
|
logger.debug("forcing timeout to 10 seconds")
|
|
conf.timeout = 10
|
|
|
|
dialect = DBMS_DICT[name][3]
|
|
sqlalchemy = SQLAlchemy(dialect=dialect)
|
|
sqlalchemy.connect()
|
|
|
|
if sqlalchemy.connector:
|
|
conf.dbmsConnector = sqlalchemy
|
|
else:
|
|
conf.dbmsConnector.connect()
|
|
|
|
if handler.checkDbms():
|
|
conf.dbmsHandler = handler
|
|
break
|
|
else:
|
|
conf.dbmsConnector = None
|
|
|
|
# At this point back-end DBMS is correctly fingerprinted, no need
|
|
# to enforce it anymore
|
|
Backend.flushForcedDbms()
|