Fix for an Issue #889

stamparm · stamparm · commit baf9ada28dad · 2014-11-01T17:13:33.000+01:00
diff --git a/lib/parse/sitemap.py b/lib/parse/sitemap.py
@@ -5,11 +5,13 @@
 See the file 'doc/COPYING' for copying permission
 """
 
+import httplib
 import re
 
 from lib.core.common import readInput
 from lib.core.data import kb
 from lib.core.data import logger
+from lib.core.exception import SqlmapSyntaxException
 from lib.request.connect import Connect as Request
 from thirdparty.oset.pyoset import oset
 
@@ -26,8 +28,13 @@ def parseSitemap(url, retVal=None):
             abortedFlag = False
             retVal = oset()
 
-        content = Request.getPage(url=url, raise404=True)[0] if not abortedFlag else ""
-        for match in re.finditer(r"<loc>\s*([^<]+)", content):
+        try:
+            content = Request.getPage(url=url, raise404=True)[0] if not abortedFlag else ""
+        except httplib.InvalidURL:
+            errMsg = "invalid URL given for sitemap ('%s')" % url
+            raise SqlmapSyntaxException, errMsg
+
+        for match in re.finditer(r"<loc>\s*([^<]+)", content or ""):
             if abortedFlag:
                 break
             url = match.group(1).strip()
