added test for site existance

stamparm · stamparm · commit 3110bb10fc8e · 2010-05-21T13:36:49.000Z
diff --git a/lib/controller/checks.py b/lib/controller/checks.py
@@ -23,6 +23,7 @@
 """
 
 import re
+import socket
 import time
 
 from lib.core.agent import agent
@@ -380,6 +381,14 @@ def checkRegexp():
         return False
 
 def checkConnection():
+    infoMsg = "testing if site exists"
+    logger.info(infoMsg)
+    try:
+        socket.gethostbyname(conf.hostname)
+    except socket.gaierror:
+        errMsg = "site '%s' does not exist" % conf.hostname
+        raise sqlmapConnectionException, errMsg
+
     infoMsg = "testing connection to the target url"
     logger.info(infoMsg)
 
