Update scan.py

cve201811784/includes/scan.py

@@ -37,8 +37,11 @@ def cvescan(url, output):
                     response = session.get(
                         fullurl, verify=False, headers=const.Data.rheaders, allow_redirects=False, timeout=5)
                     print(f'Checking ===> {fullurl}')
-
-                    if response.status_code >= 301 and 'Location' in response.headers and f"cappriciosec.com" in response.headers['Location']:
+                    location = response.headers.get('location', None)
+                    parsed_url = urlparse(location)
+                    domain = parsed_url.netloc
+                    domain = domain.decode('utf-8') if isinstance(domain, bytes) else domain
+                    if response.status_code >= 301 and 'Location' in response.headers and f"cappriciosec.com" in response.headers['Location'] and domain is not None and 'cappriciosec.com' in domain:
                         outputprint = (
                             f"\n{const.Colors.RED}💸[Vulnerable]{const.Colors.RESET} ======> "
                             f"{const.Colors.BLUE}{url}{const.Colors.RESET} \n"