args = parser.parse_args()
pm = PluginManager() # Create plug-in manager
pm.loadPlugins() # Load all sources & parse data
cves = {} # Create empty dictionary to fill up
path = args.file if args.file else "VIA4CVE-feed.json" # Generate path based on user preferences
for _id in pm.getAllCVEIDs():
cves[_id] = pm.getCVERefs(_id) # Get data per CVE
if not args.no_update:
for _id in cves.keys():
pm.updateRefs(_id, cves[_id]) # Update data based on previous data
if not args.no_cleanup:
for _id in cves.keys():
pm.cleanUp(_id, cves[_id]) # Clean data
data = {
'cves': cves,
'metadata': {
'searchables': pm.getSearchables(),
'sources': pm.getPluginNames()
}
}
open(path, "w").write(json.dumps(data)) # Write data to path
if args.verify:
_cves = json.loads(open(path).read()) # Reload cves
test.testAll(_cves, test.tests, False) # test the data (unit test)
