minor beautification

stamparm · stamparm · commit c57941c1029e · 2011-12-15T23:33:44.000Z
diff --git a/lib/controller/checks.py b/lib/controller/checks.py
@@ -7,6 +7,7 @@
 See the file 'doc/COPYING' for copying permission
 """
 
+import httplib
 import logging
 import re
 import socket
@@ -972,7 +973,7 @@ def checkConnection(suppressOutput=False):
         errMsg = getUnicode(errMsg)
         logger.critical(errMsg)
 
-        if any(code in kb.httpErrorCodes for code in (404, )):
+        if any(code in kb.httpErrorCodes for code in (httplib.NOT_FOUND, )):
             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
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -425,19 +425,19 @@ def getPage(**kwargs):
 
             logger.log(7, responseMsg)
 
-            if e.code == 401:
+            if e.code == httplib.UNAUTHORIZED:
                 errMsg = "not authorized, try to provide right HTTP "
                 errMsg += "authentication type and valid credentials (%d)" % code
                 raise sqlmapConnectionException, errMsg
-            elif e.code == 404:
+            elif e.code == httplib.NOT_FOUND:
                 if raise404:
                     errMsg = "page not found (%d)" % code
                     raise sqlmapConnectionException, errMsg
                 else:
                     debugMsg = "page not found (%d)" % code
                     logger.debug(debugMsg)
                     processResponse(page, responseHeaders)
-            elif e.code == 504:
+            elif e.code == httplib.GATEWAY_TIMEOUT:
                 if ignoreTimeout:
                     return None, None, None
                 else:
