PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-07 05:01:30 +00:00
Code Issues Projects Releases Wiki Activity

replaced third-party library python-mysql with python pymysql, http://code.google.com/p/pymysql/ (MIT license)

Browse Source
This commit is contained in:
Bernardo Damele
2011-06-22 13:31:07 +00:00
parent e76cb19e35
commit 1cb12ea659
5 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -8,7 +8,7 @@ See the file 'doc/COPYING' for copying permission
"""
try:
import MySQLdb
import pymysql
except ImportError, _:
pass
@@ -20,11 +20,11 @@ from plugins.generic.connector import Connector as GenericConnector
class Connector(GenericConnector):
"""
Homepage: http://mysql-python.sourceforge.net/
User guide: http://mysql-python.sourceforge.net/MySQLdb.html
API: http://mysql-python.sourceforge.net/MySQLdb-1.2.2/
Debian package: python-mysqldb
License: GPL
Homepage: http://code.google.com/p/pymysql/
User guide: http://code.google.com/p/pymysql/
API: http://code.google.com/p/pymysql/
Debian package: <none>
License: MIT
Possible connectors: http://wiki.python.org/moin/MySQL
"""
@@ -36,8 +36,8 @@ class Connector(GenericConnector):
self.initConnection()
try:
self.connector = MySQLdb.connect(host=self.hostname, user=self.user, passwd=self.password, db=self.db, port=self.port, connect_timeout=conf.timeout, use_unicode=True)
except MySQLdb.OperationalError, msg:
self.connector = pymysql.connect(host=self.hostname, user=self.user, passwd=self.password, db=self.db, port=self.port, connect_timeout=conf.timeout, use_unicode=True)
except pymysql.OperationalError, msg:
raise sqlmapConnectionException, msg[1]
self.setCursor()
@@ -46,16 +46,16 @@ class Connector(GenericConnector):
def fetchall(self):
try:
return self.cursor.fetchall()
except MySQLdb.ProgrammingError, msg:
except pymysql.ProgrammingError, msg:
logger.warn(msg[1])
return None
def execute(self, query):
try:
self.cursor.execute(query)
except (MySQLdb.OperationalError, MySQLdb.ProgrammingError), msg:
except (pymysql.OperationalError, pymysql.ProgrammingError), msg:
logger.warn(msg[1])
except MySQLdb.InternalError, msg:
except pymysql.InternalError, msg:
raise sqlmapConnectionException, msg[1]
self.connector.commit()