Update testssl_csv_parser.py

Author: nallamuthu

Parser/testssl_csv_parser.py

@@ -2,20 +2,25 @@
 #Input   - testssl out csv file
 #Output  - Dictonary contains ssl vulnerability details
 import csv
+import os
 
 #Function call to parser the CSV File
 def parse_testssl_csv_file(input_file):
 	result_dict={}
 	search_list=['heartbleed','ccs','ticketbleed','robot','crime_tls','breach','poodle_ssl','freak','beast','lucky13','sweet32','logjam','drown','secure_renego','secure_client_renego','fallback_scsv']
 	for item in search_list:
-		result_dict[item]="N/A"	 #Pre Populate the dictonary with all the vulnerabilities set to N/A
-	heartbleed=ccs=ticketbleed=robot=crime=breach=poodle=freak=beast=lucky13=sweet32=logjam=drown=sr_server=sr_client=fallback_scsv="N/A"
-	with open(input_file, 'r') as file:
-		reader = csv.reader(file)
-		for csv_row in reader:
-			for item in search_list:
-				if item == csv_row[0].lower(): #csv_row[0] contains the vulnerability name
-					result_dict[item]=csv_row[3] #csv_row[3] contains the result vulnerable or not
+		result_dict[item]="N/A"	 #Populate the dictonary with all the vulnerabilities set to N/A
+	try:
+		if os.path.isfile(input_file):  #Check if the file exist
+			with open(input_file, 'r') as file:
+				reader = csv.reader(file)
+				for csv_row in reader:
+					for item in search_list:
+						if item == csv_row[0].lower(): #csv_row[0] contains the vulnerability name
+							result_dict[item]=csv_row[3] #csv_row[3] contains the result vulnerable or not
+	except Exception as e:
+		#print(e)
+		result_dict = dict.fromkeys( result_dict, "ERROR" ) #if any error occurs set all the value to ERROR
 	return result_dict #Result dict
 
 #Call to function by passing the testssl out file