sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-22 17:46:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

some fixes

Browse Source
This commit is contained in:
Miroslav Stampar 2010-04-13 08:29:15 +00:00
parent 9e29120603
commit 939fa5d2c4
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,17 +48,30 @@ class Connector(GenericConnector):
def connect(self):
self.initConnection()
try:
self.connector = pyodbc.connect('DSN=%s' % self.db)
except (pyodbc.Error, pyodbc.OperationalError), msg:
raise sqlmapConnectionException, msg[1]
lastMsg = None
for connString in ('Driver={Microsoft Access Driver (*.mdb)};Dbq=%s;Uid=Admin;Pwd=;' % self.db, 'DSN=%s' % self.db):
try:
self.connector = pyodbc.connect(connString)
break
except pyodbc.Error, msg:
lastMsg = msg
except pyodbc.OperationalError, msg:
raise sqlmapConnectionException, msg[1]
if not self.connector:
raise sqlmapConnectionException, lastMsg[1]
self.setCursor()
self.connected()
def fetchall(self):
try:
return self.cursor.fetchall()
output = self.cursor.fetchall()
for i in xrange(len(output)):
for j in xrange(len(output[i])):
if type(output[i][j]) == str and output[i][j].find('\0') != -1:
output[i][j] = output[i][j][:output[i][j].find('\0')].rstrip()
return output
except pyodbc.ProgrammingError, msg:
logger.log(8, msg[1])
return None