minor update

stamparm · stamparm · commit 27d244b32682 · 2011-12-15T23:29:11.000Z
diff --git 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
