mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-27 03:53:48 +03:00
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
try:
|
|
import cx_Oracle
|
|
except:
|
|
pass
|
|
|
|
import logging
|
|
import os
|
|
import re
|
|
|
|
from lib.core.convert import utf8encode
|
|
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
|
|
|
|
os.environ["NLS_LANG"] = ".AL32UTF8"
|
|
|
|
class Connector(GenericConnector):
|
|
"""
|
|
Homepage: https://oracle.github.io/python-cx_Oracle/
|
|
User https://cx-oracle.readthedocs.io/en/latest/
|
|
API: https://wiki.python.org/moin/DatabaseProgramming
|
|
License: https://cx-oracle.readthedocs.io/en/latest/license.html#license
|
|
"""
|
|
|
|
def __init__(self):
|
|
GenericConnector.__init__(self)
|
|
|
|
def connect(self):
|
|
self.initConnection()
|
|
self.__dsn = cx_Oracle.makedsn(self.hostname, self.port, self.db)
|
|
self.__dsn = utf8encode(self.__dsn)
|
|
self.user = utf8encode(self.user)
|
|
self.password = utf8encode(self.password)
|
|
|
|
try:
|
|
self.connector = cx_Oracle.connect(dsn=self.__dsn, user=self.user, password=self.password, mode=cx_Oracle.SYSDBA)
|
|
logger.info("successfully connected as SYSDBA")
|
|
except (cx_Oracle.OperationalError, cx_Oracle.DatabaseError, cx_Oracle.InterfaceError), ex:
|
|
if "Oracle Client library" in str(ex):
|
|
msg = re.sub(r"DPI-\d+:\s+", "", str(ex))
|
|
msg = re.sub(r': ("[^"]+")', r" (\g<1>)", msg)
|
|
msg = re.sub(r". See (http[^ ]+)", r'. See "\g<1>"', msg)
|
|
raise SqlmapConnectionException(msg)
|
|
|
|
try:
|
|
self.connector = cx_Oracle.connect(dsn=self.__dsn, user=self.user, password=self.password)
|
|
except (cx_Oracle.OperationalError, cx_Oracle.DatabaseError, cx_Oracle.InterfaceError), msg:
|
|
raise SqlmapConnectionException(msg)
|
|
|
|
self.initCursor()
|
|
self.printConnected()
|
|
|
|
def fetchall(self):
|
|
try:
|
|
return self.cursor.fetchall()
|
|
except cx_Oracle.InterfaceError, msg:
|
|
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
|
|
return None
|
|
|
|
def execute(self, query):
|
|
retVal = False
|
|
|
|
try:
|
|
self.cursor.execute(utf8encode(query))
|
|
retVal = True
|
|
except cx_Oracle.DatabaseError, msg:
|
|
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
|
|
|
|
self.connector.commit()
|
|
|
|
return retVal
|
|
|
|
def select(self, query):
|
|
retVal = None
|
|
|
|
if self.execute(query):
|
|
retVal = self.fetchall()
|
|
|
|
return retVal
|