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())
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())
