|
| 1 | +#!/usr/bin/env python |
| 2 | + |
| 3 | +""" |
| 4 | +Copyright (c) 2006-2016 sqlmap developers (http://sqlmap.org/) |
| 5 | +See the file 'doc/COPYING' for copying permission |
| 6 | +""" |
| 7 | + |
| 8 | +try: |
| 9 | + import ibm_db_dbi |
| 10 | +except ImportError: |
| 11 | + pass |
| 12 | + |
| 13 | +import logging |
| 14 | + |
| 15 | +from lib.core.data import conf |
| 16 | +from lib.core.data import logger |
| 17 | +from lib.core.exception import SqlmapConnectionException |
| 18 | +from plugins.generic.connector import Connector as GenericConnector |
| 19 | + |
| 20 | +class Connector(GenericConnector): |
| 21 | + """ |
| 22 | + Homepage: http://code.google.com/p/ibm-db/ |
| 23 | + User guide: http://code.google.com/p/ibm-db/wiki/README |
| 24 | + API: http://www.python.org/dev/peps/pep-0249/ |
| 25 | + License: Apache License 2.0 |
| 26 | + """ |
| 27 | + |
| 28 | + def __init__(self): |
| 29 | + GenericConnector.__init__(self) |
| 30 | + |
| 31 | + def connect(self): |
| 32 | + self.initConnection() |
| 33 | + |
| 34 | + try: |
| 35 | + database = "DATABASE=%s;HOSTNAME=%s;PORT=%s;PROTOCOL=TCPIP;" % (self.db, self.hostname, self.port) |
| 36 | + self.connector = ibm_db_dbi.connect(database, self.user, self.password) |
| 37 | + except ibm_db_dbi.OperationalError, msg: |
| 38 | + raise SqlmapConnectionException(msg) |
| 39 | + |
| 40 | + |
| 41 | + self.initCursor() |
| 42 | + self.printConnected() |
| 43 | + |
| 44 | + def fetchall(self): |
| 45 | + try: |
| 46 | + return self.cursor.fetchall() |
| 47 | + except ibm_db_dbi.ProgrammingError, msg: |
| 48 | + logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg[1]) |
| 49 | + return None |
| 50 | + |
| 51 | + def execute(self, query): |
| 52 | + try: |
| 53 | + self.cursor.execute(query) |
| 54 | + except (ibm_db_dbi.OperationalError, ibm_db_dbi.ProgrammingError), msg: |
| 55 | + logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg[1]) |
| 56 | + except ibm_db_dbi.InternalError, msg: |
| 57 | + raise SqlmapConnectionException(msg[1]) |
| 58 | + |
| 59 | + self.connector.commit() |
| 60 | + |
| 61 | + def select(self, query): |
| 62 | + self.execute(query) |
| 63 | + return self.fetchall() |
0 commit comments