def importDirectoryFile(importFile,isOverwrite=1): parser = xml.sax.make_parser() handler = DirectoryContentHandler() parser.setContentHandler(handler) parser.setEntityResolver(handler) parser.parse(importFile) vulDir,threatDir = handler.directories() vdSize = len(vulDir) tdSize = len(threatDir) b = Borg() if (vdSize > 0): b.dbProxy.addVulnerabilityDirectory(vulDir,isOverwrite) if (tdSize > 0): b.dbProxy.addThreatDirectory(threatDir,isOverwrite) msgStr = 'Imported ' + str(vdSize) + ' template vulnerabilities and ' + str(tdSize) + ' template threats.' return msgStr
def importDirectoryFile(importFile, isOverwrite=1): parser = xml.sax.make_parser() handler = DirectoryContentHandler() parser.setContentHandler(handler) parser.setEntityResolver(handler) parser.parse(importFile) vulDir, threatDir = handler.directories() vdSize = len(vulDir) tdSize = len(threatDir) b = Borg() if (vdSize > 0): b.dbProxy.addVulnerabilityDirectory(vulDir, isOverwrite) if (tdSize > 0): b.dbProxy.addThreatDirectory(threatDir, isOverwrite) msgStr = 'Imported ' + str( vdSize) + ' template vulnerabilities and ' + str( tdSize) + ' template threats.' return msgStr
def importDirectoryFile(importFile,isOverwrite=1,session_id = None): try: parser = xml.sax.make_parser() handler = DirectoryContentHandler() parser.setContentHandler(handler) parser.setEntityResolver(handler) parser.parse(importFile) vulDir,threatDir = handler.directories() vdSize = len(vulDir) tdSize = len(threatDir) b = Borg() db_proxy = b.get_dbproxy(session_id) if (vdSize > 0): db_proxy.addVulnerabilityDirectory(vulDir,isOverwrite) if (tdSize > 0): db_proxy.addThreatDirectory(threatDir,isOverwrite) msgStr = 'Imported ' + str(vdSize) + ' template vulnerabilities and ' + str(tdSize) + ' template threats.' return msgStr except xml.sax.SAXException, e: raise ARMException("Error parsing" + importFile + ": " + e.getMessage())