sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-05-16 21:53:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

added support for reusing connections

Browse Source
This commit is contained in:
Miroslav Stampar 2010-03-30 13:52:47 +00:00
parent f0729565a9
commit 1973024ebf
7 changed files with 37 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
plugins/dbms/access/connector.py
View File

@ -45,7 +45,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
try: try:
@ -89,4 +92,5 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

6
plugins/dbms/firebird/connector.py
View File

@ -43,7 +43,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
try: try:
@ -83,3 +86,4 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

6
plugins/dbms/mysql/connector.py
View File

@ -47,7 +47,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
try: try:
@ -87,3 +90,4 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

6
plugins/dbms/oracle/connector.py
View File

@ -46,7 +46,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
self.__dsn = cx_Oracle.makedsn(self.hostname, self.port, self.db) self.__dsn = cx_Oracle.makedsn(self.hostname, self.port, self.db)
@ -91,3 +94,4 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

6
plugins/dbms/postgresql/connector.py
View File

@ -46,7 +46,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
try: try:
@ -86,3 +89,4 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

6
plugins/dbms/sqlite/connector.py
View File

@ -47,7 +47,10 @@ class Connector(GenericConnector):
def __init__(self): def __init__(self):
GenericConnector.__init__(self) GenericConnector.__init__(self)
def connect(self): def connect(self, reuse=True):
if reuse and self.connector:
return
self.initConnection() self.initConnection()
try: try:
@ -87,3 +90,4 @@ class Connector(GenericConnector):
def close(self): def close(self):
self.cursor.close() self.cursor.close()
self.connector.close() self.connector.close()
self.closed()

7
plugins/generic/connector.py
View File

@ -47,6 +47,13 @@ class Connector:
infoMsg += ":%d established" % self.port infoMsg += ":%d established" % self.port
logger.info(infoMsg) logger.info(infoMsg)
def closed(self):
self.connector = None
self.cursor = None
infoMsg = "connection to %s server %s" % (conf.dbms, self.hostname)
infoMsg += ":%d closed" % self.port
logger.info(infoMsg)
def connect(self): def connect(self):
errMsg = "'connect' method must be defined " errMsg = "'connect' method must be defined "
errMsg += "into the specific DBMS plugin" errMsg += "into the specific DBMS plugin"