digital-forensics-labs/SecGen
1
0
Fork 0
mirror of https://github.com/cliffe/SecGen.git synced 2026-02-22 19:58:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

oVirt provider spinup with multiple options - ovirturl, ovirtpass, ovirtuser, ovirt-vmname, ovirtcluster, ovirttemplate etc

Browse Source
This commit is contained in:
Gajendra
2017-08-23 23:14:14 +01:00
committed by thomashaw
parent 24ce91af38
commit 4ac2cd3ddc
2 changed files with 98 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
lib/templates/Vagrantfile.erb
View File

@@ -6,12 +6,39 @@
# Based on <%= @scenario %>
<% require 'json'
require 'base64' -%>
<% vmnum = 0 %>
VAGRANTFILE_API_VERSION = "2"
Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
<% @systems.each do |system| %>
config.vm.define "<%= system.name %>" do |<%= system.name %>|
<% if (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
#oVirt provider begin
config.vm.provider :ovirt4 do |ovirt|
<%=
" ovirt.username = '#{@options[:ovirtuser]}'
ovirt.password = '#{@options[:ovirtpass]}'" %>
<%= if(@options.has_key? :ovirturl)
" ovirt.url = '#{@options[:ovirturl]}'"
else
" ovirt.url = 'DEFAULT_API_URL'"
end %>
<%= if(@options.has_key? :ovirtcluster)
" ovirt.cluster = '#{@options[:ovirtcluster]}'"
else
" ovirt.cluster = 'DEFAULT_OVIRT_CLUSTER'"
end %>
<%= if(@options.has_key? :ovirttemplate)
" ovirt.template = '#{@options[:ovirttemplate]}'"
else
" ovirt.template = 'DEFAULT_OVIRT_TEMPLATE'"
end %>
ovirt.insecure = true
ovirt.debug = true
end
# oVirt provider end
<%
else %>
config.vm.provider :virtualbox do |vb|
<% system.module_selections.each do |selected_module|
if selected_module.module_type == 'base'
@@ -45,7 +72,7 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
" vb.customize ['modifyvm', :id, '--cpuexecutioncap', '#{@options[:max_cpu_usage]}']\n"
end -%>
end
<% end %>
# SecGen datastore
# <%= JSON.generate($datastore) %>
@@ -55,8 +82,19 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
<%= selected_module.to_s_comment -%>
<% case selected_module.module_type
when 'base' -%>
<% if (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
<% if (@options.has_key? :ovirtvmname) %>
<% vmnum = vmnum + 1 %>
<%= system.name %>.vm.hostname = '<%= "#{@options[:ovirtvmname]}#{vmnum}" %>'
<% else %>
<%= system.name %>.vm.hostname = '<%= "SecGen-oVirt-VM-#{Time.new.strftime("%Y%m%d-%H%M")}" %>'
<% end %>
<%= system.name %>.vm.box = 'ovirt4'
<%= system.name %>.vm.box_url = 'https://github.com/myoung34/vagrant-ovirt4/blob/master/example_box/dummy.box?raw=true'
<% else %>
<%= system.name %>.vm.box = "<%= selected_module.module_path_name %>"
<%= system.name %>.vm.box_url = "<%= selected_module.attributes['url'].first %>"
<% end %>
<% if selected_module.attributes['platform'].first.downcase == 'windows' %>
<%= system.name %>.vm.communicator = 'winrm'
<%= system.name %>.vm.guest = :windows
@@ -65,9 +103,21 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
<% end %>
<% when 'network' -%>
<% if selected_module.attributes['range'].first.nil? || selected_module.attributes['range'].first == "dhcp" -%>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, type: "dhcp"
<% if (@options.has_key? :ovirtnetwork) && (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, :ovirt__network_name => '<%= "#{@options[:ovirtnetwork]}" %>'
<% elsif (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, :ovirt__network_name => 'DEFAULT_NETWORK_NAME'
<% else %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, type: "dhcp"
<% end %>
<% else -%>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, ip: "<%= resolve_network(selected_module.attributes['range'].first)%>"
<% if (@options.has_key? :ovirtip) && (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, :ovirt__ip => '<%= "#{@options[:ovirtip]}" %>'
<% elsif (@options.has_key? :ovirtuser) && (@options.has_key? :ovirtpass) %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, :ovirt__ip => "<%= resolve_network(selected_module.attributes['range'].first)%>"
<% else %>
<%= system.name %>.vm.network :<%= selected_module.attributes['type'].first %>, ip: "<%= resolve_network(selected_module.attributes['range'].first)%>"
<% end %>
<% end -%>
<% when 'vulnerability', 'service', 'utility', 'build' -%>
<% module_name = selected_module.module_path_name -%>

44
secgen.rb
View File

@@ -26,6 +26,16 @@ def usage
--vtxvpid: enable VTX support
--forensic-image-type [image type]: Forensic image format of generated image (raw, ewf)
OVIRT OPTIONS:
--ovirtuser [ovirt_username] (REQUIRED)
--ovirtpass [ovirt_password] (REQUIRED)
--ovirt-vmname [ovirt_vm_name] (OPTIONAL)
--ovirt-url [ovirt_api_url] (REQUIRED)
--ovirt-cluster [ovirt_cluster] (REQUIRED)
--ovirt-template [ovirt_template] (REQUIRED)
--ovirt-ip [ovirt_static_ip] (OPTIONAL)
--ovirt-network [ovirt_network_name] (OPTIONAL)
COMMANDS:
run, r: Builds project and then builds the VMs
build-project, p: Builds project (vagrant and puppet config), but does not build VMs
@@ -235,6 +245,14 @@ opts = GetoptLong.new(
[ '--max-cpu-usage', GetoptLong::REQUIRED_ARGUMENT],
[ '--shutdown', GetoptLong::NO_ARGUMENT],
[ '--forensic-image-type', GetoptLong::REQUIRED_ARGUMENT],
[ '--ovirt-vmname', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirtuser', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirtpass', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirt-url', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirt-cluster', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirt-template', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirt-ip', GetoptLong::REQUIRED_ARGUMENT ],
[ '--ovirt-network', GetoptLong::REQUIRED_ARGUMENT ],
)
scenario = SCENARIO_XML
@@ -295,6 +313,32 @@ opts.each do |opt, arg|
Print.info "Image output type set to #{arg}"
options[:forensic_image_type] = arg
when '--ovirt-vmname'
Print.info "Ovirt VM Name : #{arg}"
options[:ovirtvmname] = arg
filename = arg;
when '--ovirtuser'
Print.info "Ovirt Username : #{arg}"
options[:ovirtuser] = arg
when '--ovirtpass'
Print.info "Ovirt Password : ********"
options[:ovirtpass] = arg
when '--ovirt-url'
Print.info "Ovirt API url : #{arg}"
options[:ovirturl] = arg
when '--ovirt-cluster'
Print.info "Ovirt Cluster : #{arg}"
options[:ovirtcluster] = arg
when '--ovirt-template'
Print.info "Ovirt Template : #{arg}"
options[:ovirttemplate] = arg
when '--ovirt-ip'
Print.info "Ovirt Static IP : #{arg}"
options[:ovirtip] = arg
when '--ovirt-network'
Print.info "Ovirt Network Name : #{arg}"
options[:ovirtnetwork] = arg
else
Print.err "Argument not valid: #{arg}"
usage