mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-10 19:26:49 +00:00
effcbc15c9
Just like #15 and #19 but in a new file this time. Perhaps it is time to add some automated testing like Travis CI, Circle CI, Appveyor, etc. https://github.com/marketplace/category/continuous-integration
73 lines
2.9 KiB
Python
73 lines
2.9 KiB
Python
from __future__ import print_function
|
|
|
|
from prettytable import ALL, PrettyTable
|
|
|
|
from __main__ import config
|
|
from collector import services, vulnerabilities
|
|
|
|
EVIDENCE_PREVIEW = 40
|
|
MAX_TABLE_WIDTH = 20
|
|
|
|
|
|
class PlainReporter(object):
|
|
|
|
def get_report(self):
|
|
"""generates report tables"""
|
|
output = ""
|
|
if len(services):
|
|
output += self.nodes_table()
|
|
if not config.mapping:
|
|
output += self.services_table()
|
|
if len(vulnerabilities):
|
|
output += self.vulns_table()
|
|
else:
|
|
output += "\nNo vulnerabilities were found"
|
|
else:
|
|
print("\nKube Hunter couldn't find any clusters")
|
|
# print("\nKube Hunter couldn't find any clusters. {}".format("Maybe try with --active?" if not config.active else ""))
|
|
return output
|
|
|
|
def nodes_table(self):
|
|
nodes_table = PrettyTable(["Type", "Location"], hrules=ALL)
|
|
nodes_table.align = "l"
|
|
nodes_table.max_width = MAX_TABLE_WIDTH
|
|
nodes_table.padding_width = 1
|
|
nodes_table.sortby = "Type"
|
|
nodes_table.reversesort = True
|
|
nodes_table.header_style = "upper"
|
|
# TODO: replace with sets
|
|
id_memory = list()
|
|
for service in services:
|
|
if service.event_id not in id_memory:
|
|
nodes_table.add_row(["Node/Master", service.host])
|
|
id_memory.append(service.event_id)
|
|
return "\nNodes\n{}\n".format(nodes_table)
|
|
|
|
def services_table(self):
|
|
services_table = PrettyTable(["Service", "Location", "Description"], hrules=ALL)
|
|
services_table.align = "l"
|
|
services_table.max_width = MAX_TABLE_WIDTH
|
|
services_table.padding_width = 1
|
|
services_table.sortby = "Service"
|
|
services_table.reversesort = True
|
|
services_table.header_style = "upper"
|
|
for service in services:
|
|
services_table.add_row([service.get_name(), "{}:{}{}".format(service.host, service.port, service.get_path()), service.explain()])
|
|
return "\nDetected Services\n{}\n".format(services_table)
|
|
|
|
def vulns_table(self):
|
|
column_names = ["Location", "Category", "Vulnerability", "Description", "Evidence"]
|
|
vuln_table = PrettyTable(column_names, hrules=ALL)
|
|
vuln_table.align = "l"
|
|
vuln_table.max_width = MAX_TABLE_WIDTH
|
|
vuln_table.sortby = "Category"
|
|
vuln_table.reversesort = True
|
|
vuln_table.padding_width = 1
|
|
vuln_table.header_style = "upper"
|
|
for vuln in vulnerabilities:
|
|
row = ["{}:{}".format(vuln.host, vuln.port) if vuln.host else "", vuln.category.name, vuln.get_name(), vuln.explain()]
|
|
evidence = str(vuln.evidence)[:EVIDENCE_PREVIEW] + "..." if len(str(vuln.evidence)) > EVIDENCE_PREVIEW else str(vuln.evidence)
|
|
row.append(evidence)
|
|
vuln_table.add_row(row)
|
|
return "\nVulnerabilities\n{}\n".format(vuln_table)
|