From 49fddaf668b0ca40ddd44abca2f3d053ebd3a477 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Sun, 20 Nov 2011 23:11:18 +0000 Subject: [PATCH] minor update (for cases with 404 original page - e.g. time based injections in some cases) --- lib/controller/checks.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lib/controller/checks.py b/lib/controller/checks.py index a7d4a5dea..be8a328d8 100644 --- a/lib/controller/checks.py +++ b/lib/controller/checks.py @@ -51,6 +51,7 @@ from lib.core.enums import PLACE from lib.core.exception import sqlmapConnectionException from lib.core.exception import sqlmapGenericException from lib.core.exception import sqlmapNoneDataException +from lib.core.exception import sqlmapSilentQuitException from lib.core.exception import sqlmapUserQuitException from lib.core.session import setDynamicMarkings from lib.core.settings import CONSTANT_RATIO @@ -966,6 +967,10 @@ def checkConnection(suppressOutput=False): kb.errorIsNone = True except sqlmapConnectionException, errMsg: errMsg = getUnicode(errMsg) - raise sqlmapConnectionException, errMsg + logger.critical(errMsg) + + msg = "it is not recommended to continue in this kind of cases. Do you want to quit and make sure that everything is set up properly? [Y/n] " + if readInput(msg, default="Y") not in ("n", "N"): + raise sqlmapSilentQuitException return True