fix bug single host report PDF

Author: Rev3rseSecurity

pdf.py

@@ -22,7 +22,13 @@ def reportPDFView(request):
 	counters = {'po':0,'pc':0,'pf':0,'hostsup':0,'ostype':{},'pi':{},'ss':{}}
 
 	scanmd5 = hashlib.md5(str(request.session['scanfile']).encode('utf-8')).hexdigest()
-	for i in o['host']:
+	for ik in o['host']:
+
+		# this fix single host report
+		if type(ik) is dict:
+			i = ik
+		else:
+			i = o['host']
 
 		hostcounters = {'po':0,'pc':0,'pf':0,'ostype':{},'pi':{},'ss':{}}
 		hostdetails_html = ''
@@ -168,6 +174,9 @@ def reportPDFView(request):
 
 				portsfound = True
 
+			# this fix single host report
+			if type(ik) is not dict:
+				break;
 
 
 		notesout,notesb64 = '',''

views.py

@@ -178,9 +178,9 @@ def details(request, address):
 					'<td colspan="2" style="color:#999;font-size:12px;">State: '+p['state']['@state']+'<br>Reason: '+p['state']['@reason']+'</td>'+\
 					'<td><button onclick="javascript:apiPortDetails(\''+html.escape(address)+'\',\''+html.escape(p['@portid'])+'\');" class="btn blue right"><i class="material-icons">receipt</i></button></td></tr>'
 
-				# this fix single host report
-				if type(ik) is not dict:
-					break;
+		# this fix single host report
+		if type(ik) is not dict:
+			break;
 
 	notesout,notesb64,removenotes = '','',''
 	if scanmd5 in noteshost: