Cosmetics

This commit is contained in:
Bernardo Damele 2010-11-03 22:07:13 +00:00
parent 71d0b1bcd7
commit b152b1a04d
2 changed files with 13 additions and 13 deletions

View File

@ -15,6 +15,13 @@ class Replication:
replication purposes.
"""
# sqlite data types
NULL = DataType('NULL')
INTEGER = DataType('INTEGER')
REAL = DataType('REAL')
TEXT = DataType('TEXT')
BLOB = DataType('BLOB')
def __init__(self, dbpath):
self.dbpath = dbpath
self.connection = sqlite3.connect(dbpath)
@ -67,13 +74,6 @@ class Replication:
stmt += 'WHERE %s' % condition
return self.parent.cursor.execute(stmt)
# sqlite data types
NULL = DataType('NULL')
INTEGER = DataType('INTEGER')
REAL = DataType('REAL')
TEXT = DataType('TEXT')
BLOB = DataType('BLOB')
def createTable(self, tblname, columns=None, typeless=False):
"""
This function creates Table instance with current connection settings.

View File

@ -212,12 +212,15 @@ class Connect:
page = decodePage(page, responseHeaders.get("Content-Encoding"), responseHeaders.get("Content-Type"))
except urllib2.HTTPError, e:
code = e.code
status = e.msg
if e.code == 401:
errMsg = "not authorized, try to provide right HTTP "
errMsg += "authentication type and valid credentials (%d)" % e.code
errMsg += "authentication type and valid credentials (%d)" % code
raise sqlmapConnectionException, errMsg
elif e.code == 404 and raise404:
errMsg = "page not found (%d)" % e.code
errMsg = "page not found (%d)" % code
raise sqlmapConnectionException, errMsg
else:
try:
@ -225,13 +228,10 @@ class Connect:
responseHeaders = e.info()
except socket.timeout:
warnMsg = "connection timed out while trying "
warnMsg += "to get error page information (%d)" % e.code
warnMsg += "to get error page information (%d)" % code
logger.warn(warnMsg)
return None, None
code = e.code
status = e.msg
debugMsg = "got HTTP error code: %d (%s)" % (code, status)
logger.debug(debugMsg)