CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-05-14 01:16:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding integrity checks in case of unhandled exceptions

Browse source
This commit is contained in:
Miroslav Stampar 2016-07-17 00:04:30 +02:00
parent 7427b554e3
commit e485531b71
5 changed files with 33 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

16
sqlmap.py
View file

@ -34,6 +34,7 @@ from lib.core.data import logger
try:
from lib.controller.controller import start
from lib.core.common import banner
from lib.core.common import checkIntegrity
from lib.core.common import createGithubIssue
from lib.core.common import dataToStdout
from lib.core.common import getSafeExString
@ -196,7 +197,15 @@ def main():
excMsg = traceback.format_exc()
try:
if any(_ in excMsg for _ in ("No space left", "Disk quota exceeded")):
if not checkIntegrity():
errMsg = "code integrity check failed. "
errMsg += "You should retrieve the latest development version from official GitHub "
errMsg += "repository at '%s'" % GIT_PAGE
logger.critical(errMsg)
print
print excMsg.strip()
raise SystemExit
elif any(_ in excMsg for _ in ("No space left", "Disk quota exceeded")):
errMsg = "no space left on output device"
logger.error(errMsg)
raise SystemExit
@ -213,11 +222,6 @@ def main():
logger.error(errMsg)
raise SystemExit
elif "in _loadQueries" in excMsg:
errMsg = "invalid structure of 'xml/queries.xml' file"
logger.error(errMsg)
raise SystemExit
elif "_mkstemp_inner" in excMsg:
errMsg = "there has been a problem while accessing temporary files"
logger.error(errMsg)