41 lines
1.1 KiB
Ruby
Executable file
41 lines
1.1 KiB
Ruby
Executable file
#!/usr/bin/env ruby
|
|
|
|
require 'optparse'
|
|
require './asset_generator'
|
|
require './cve_report'
|
|
require './kdi_format'
|
|
|
|
ARGV << '-h' if ARGV.empty?
|
|
|
|
params = {}
|
|
|
|
OptionParser.new do |opts|
|
|
opts.banner = "Usage: kenna_kdi_importer.rb [OPTIONS]"
|
|
|
|
opts.on('-a NUM', '--assets', Integer, "Number of assets to create") { |a| params[:assets] = a }
|
|
opts.on('-v NUM', '--vulns', Integer, "Number of vulns that may be created per asset") {|v| params[:vulns] = v }
|
|
# opts.on('-g BOOLEAN', '--generate', Boolean, "Generate a new json of cve ids in an array") {|g| params[:generate] = g }
|
|
|
|
opts.on_tail("-h", "--help", "Show this message") do
|
|
puts opts
|
|
exit
|
|
end
|
|
end.parse!(into: params)
|
|
|
|
raise OptionParser::MissingArgument if params[:assets].nil?
|
|
raise OptionParser::MissingArgument if params[:vulns].nil?
|
|
|
|
num_of_assets = params[:assets]
|
|
num_of_vulns = params[:vulns]
|
|
|
|
asset_generator = KennaKdi::AssetGenerator.new('./data/cve_ids.json')
|
|
|
|
while true do
|
|
hash = asset_generator.create_assets(num_of_assets, num_of_vulns)
|
|
if hash
|
|
break
|
|
end
|
|
end
|
|
|
|
json = hash.to_json
|
|
puts JSON.pretty_generate(JSON.parse(json))
|