#!/usr/bin/env python

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

import os

from lib.core.data import conf
from lib.core.data import logger
from lib.core.exception import SqlmapFilePathException
from lib.core.exception import SqlmapUndefinedMethod

class Connector(object):
 """
 This class defines generic dbms protocol functionalities for plugins.
 """

 def __init__(self):
 self.connector = None
 self.cursor = None
 self.hostname = None

 def initConnection(self):
 self.user = conf.dbmsUser or ""
 self.password = conf.dbmsPass or ""
 self.hostname = conf.hostname
 self.port = conf.port
 self.db = conf.dbmsDb

 def printConnected(self):
 if self.hostname and self.port:
 infoMsg = "connection to %s server '%s:%d' established" % (conf.dbms, self.hostname, self.port)
 logger.info(infoMsg)

 def closed(self):
 if self.hostname and self.port:
 infoMsg = "connection to %s server '%s:%d' closed" % (conf.dbms, self.hostname, self.port)
 logger.info(infoMsg)

 self.connector = None
 self.cursor = None

 def initCursor(self):
 self.cursor = self.connector.cursor()

 def close(self):
 try:
 if self.cursor:
 self.cursor.close()
 if self.connector:
 self.connector.close()
 except Exception as ex:
 logger.debug(ex)
 finally:
 self.closed()

 def checkFileDb(self):
 if not os.path.exists(self.db):
 errMsg = "the provided database file '%s' does not exist" % self.db
 raise SqlmapFilePathException(errMsg)

 def connect(self):
 errMsg = "'connect' method must be defined "
 errMsg += "into the specific DBMS plugin"
 raise SqlmapUndefinedMethod(errMsg)

 def fetchall(self):
 errMsg = "'fetchall' method must be defined "
 errMsg += "into the specific DBMS plugin"
 raise SqlmapUndefinedMethod(errMsg)

 def execute(self, query):
 errMsg = "'execute' method must be defined "
 errMsg += "into the specific DBMS plugin"
 raise SqlmapUndefinedMethod(errMsg)

 def select(self, query):
 errMsg = "'select' method must be defined "
 errMsg += "into the specific DBMS plugin"
 raise SqlmapUndefinedMethod(errMsg)