From 3db01c4dcf74954ec04fddf7489a27e7874280e1 Mon Sep 17 00:00:00 2001 From: ekultek Date: Fri, 10 Nov 2017 11:00:21 -0600 Subject: [PATCH] will only display the headers that are equipped for protection and will no longer prompt you if you want to continue, a warning should be enough, fixed a new line issue in an inof message --- lib/header_check/__init__.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/lib/header_check/__init__.py b/lib/header_check/__init__.py index 224bbc0..5f8bc58 100644 --- a/lib/header_check/__init__.py +++ b/lib/header_check/__init__.py @@ -89,7 +89,7 @@ def main_header_check(url, **kwargs): )) comparable_headers = load_xml_data(HEADER_XML_DATA) logger.info(set_color( - "attempting to get request headers for '{}'...".format(url) + "attempting to get request headers for '{}'...".format(url.strip()) )) found_headers = load_headers(url, proxy=proxy, agent=agent, xforward=xforward) if verbose: @@ -102,10 +102,6 @@ def main_header_check(url, **kwargs): logger.warning(set_color( "provided target has {}...".format(definition[key][0]), level=30 )) - else: - logger.info(set_color( - "provided target does not have {}...".format(definition[key][0]) - )) for key in found_headers.iterkeys(): protection[key] = found_headers[key] return write_to_log_file(protection, HEADER_RESULT_PATH, "{}-headers.json".format(replace_http(url)))