From 27d244b32682264e1774a397c42ca79c2ae0352b Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Thu, 15 Dec 2011 23:29:11 +0000 Subject: [PATCH] minor update --- lib/controller/checks.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/lib/controller/checks.py b/lib/controller/checks.py index 24d57fa7b..3f570f369 100644 --- a/lib/controller/checks.py +++ b/lib/controller/checks.py @@ -972,10 +972,13 @@ def checkConnection(suppressOutput=False): errMsg = getUnicode(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 + if any(code in kb.httpErrorCodes for code in (404, )): + 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 + else: + kb.ignoreNotFound = True else: - kb.ignoreNotFound = True + raise return True