Class CherryTree

autorecon/default-plugins/reporting-cherrytree.py:6–91

import os, glob

class CherryTree(Report):

	def __init__(self):
		super().__init__()
		self.name = 'CherryTree'
		self.tags = []

	async def run(self, targets):
		if len(targets) > 1:
			report = os.path.join(config['output'], 'report.xml.ctd')
		elif len(targets) == 1:
			report = os.path.join(targets[0].reportdir, 'report.xml.ctd')
		else:
			return

		with open(report, 'w') as output:
			output.writelines('<?xml version="1.0" encoding="UTF-8"?>\n<cherrytree>\n')
			for target in targets:
				output.writelines('<node name="' + escape(target.address) + '" is_bold="1" custom_icon_id="1">\n')

				files = [os.path.abspath(filename) for filename in glob.iglob(os.path.join(target.scandir, '**/*'), recursive=True) if os.path.isfile(filename) and filename.endswith(('.txt', '.html'))]

				if target.scans['ports']:
					output.writelines('<node name="Port Scans" custom_icon_id="2">\n')
					for scan in target.scans['ports'].keys():
						if len(target.scans['ports'][scan]['commands']) > 0:
							output.writelines('<node name="PortScan: ' + escape(target.scans['ports'][scan]['plugin'].name) + '" custom_icon_id="21">\n')
							for command in target.scans['ports'][scan]['commands']:
								output.writelines('<rich_text>' + escape(command[0]))
								for filename in files:
									if filename in command[0] or (command[1] is not None and filename == command[1]) or (command[2] is not None and filename == command[2]):
										output.writelines('\n\n' + escape(filename) + ':\n\n')
										with open(filename, 'r') as file:
											output.writelines(escape(file.read()) + '\n')
								output.writelines('</rich_text>\n')
							output.writelines('</node>\n')
					output.writelines('</node>\n')
				if target.scans['services']:
					output.writelines('<node name="Services" custom_icon_id="2">\n')
					for service in target.scans['services'].keys():
						output.writelines('<node name="Service: ' + escape(service.tag()) + '" custom_icon_id="3">\n')
						for plugin in target.scans['services'][service].keys():
							if len(target.scans['services'][service][plugin]['commands']) > 0:
								output.writelines('<node name="' + escape(target.scans['services'][service][plugin]['plugin'].name) + '" custom_icon_id="21">\n')
								for command in target.scans['services'][service][plugin]['commands']:
									output.writelines('<rich_text>' + escape(command[0]))
									for filename in files:
										if filename in command[0] or (command[1] is not None and filename == command[1]) or (command[2] is not None and filename == command[2]):
											output.writelines('\n\n' + escape(filename) + ':\n\n')
											with open(filename, 'r') as file:
												output.writelines(escape(file.read()) + '\n')
									output.writelines('</rich_text>\n')
								output.writelines('</node>\n')
						output.writelines('</node>\n')
					output.writelines('</node>\n')

				manual_commands = os.path.join(target.scandir, '_manual_commands.txt')
				if os.path.isfile(manual_commands):