CERBERUS/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2026-08-04 14:49:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use XMLWriter for output.

Browse source 
This avoids having to build a whole DOM tree in memory.
This commit is contained in:
david 2011-12-21 04:54:26 +00:00
parent d6f867cdb5
commit e4beb1ab48
1 changed files with 23 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

42
ndiff/ndiff
View file

@ -530,45 +530,49 @@ class ScanDiff(object):
def print_xml(self, f = sys.stdout):
impl = xml.dom.minidom.getDOMImplementation()
document = impl.createDocument(None, u"nmapdiff", None)
root = document.documentElement
root.setAttribute(u"version", NDIFF_XML_VERSION)
scandiff_elem = document.createElement(u"scandiff")
root.appendChild(scandiff_elem)
document = impl.createDocument(None, None, None)
writer = XMLWriter(f)
writer.startDocument()
writer.startElement(u"nmapdiff", {u"version": NDIFF_XML_VERSION})
writer.startElement(u"scandiff", {})
if self.nmaprun_differs():
a_elem = document.createElement(u"a")
a_elem.appendChild(self.scan_a.nmaprun_to_dom_fragment(document))
scandiff_elem.appendChild(a_elem)
b_elem = document.createElement(u"b")
b_elem.appendChild(self.scan_b.nmaprun_to_dom_fragment(document))
scandiff_elem.appendChild(b_elem)
writer.frag_a(self.scan_a.nmaprun_to_dom_fragment(document))
writer.frag_b(self.scan_b.nmaprun_to_dom_fragment(document))
elif verbose:
scandiff_elem.appendChild(self.scan_a.nmaprun_to_dom_fragment(document))
writer.frag(self.scan_a.nmaprun_to_dom_fragment(document))
# prerule script changes.
if len(self.pre_script_result_diffs) > 0 or verbose:
prescript_elem = document.createElement(u"prescript")
scandiff_elem.appendChild(script_result_diffs_to_dom_fragment(
frag = script_result_diffs_to_dom_fragment(
prescript_elem, self.scan_a.pre_script_results,
self.scan_b.pre_script_results, self.pre_script_result_diffs,
document))
document)
writer.frag(frag)
frag.unlink()
for host in self.hosts:
h_diff = self.host_diffs[host]
frag = h_diff.to_dom_fragment(document)
scandiff_elem.appendChild(frag)
writer.frag(frag)
frag.unlink()
# postrule script changes.
if len(self.post_script_result_diffs) > 0 or verbose:
postscript_elem = document.createElement(u"postscript")
scandiff_elem.appendChild(script_result_diffs_to_dom_fragment(
frag = script_result_diffs_to_dom_fragment(
postscript_elem, self.scan_a.post_script_results,
self.scan_b.post_script_results, self.post_script_result_diffs,
document))
document)
writer.frag(frag)
frag.unlink()
document.writexml(f, addindent = u" ", newl = u"\n", encoding = "UTF-8")
document.unlink()
writer.endElement(u"scandiff")
writer.endElement(u"nmapdiff")
writer.endDocument()
cost = property(lambda self: sum([hd.cost for hd in self.host_diffs.values()]))