File Coverage

blib/lib/Rex/Virtualization/CBSD/create.pm

Criterion	Covered	Total	%
statement	15	36	41.6
branch	0	12	0.0
condition			n/a
subroutine	5	6	83.3
pod	0	1	0.0
total	20	55	36.3

#
# (c) Zane C. Bowers-Hadley
#

package Rex::Virtualization::CBSD::create;

use strict;
use warnings;

our $VERSION = '0.0.1';    # VERSION

use Rex::Logger;
use Rex::Helper::Run;
use Term::ANSIColor qw(colorstrip);

sub execute {
	my ( $class, %opts ) = @_;

	#make sure we have the very minimum needed
	# also that each value is sane
	my @required_keys = ( 'name', 'vm_os_type', 'vm_os_profile' );
	foreach my $key (@required_keys) {
		if ( !defined( $opts{$key} ) ) {
			die 'Required key "' . $key . '" not defined';
		}

		# make sure it does not contain any tabs, spaces, =, \. /, ', ", or new lines.
		if ( $opts{$key} =~ /[\t\ \=\\\/\'\"\n]/ ) {
			die 'The value "'
				. $opts{$key}
				. '" for key "'
				. $key
				. '" matched /[\t\ \=\/\\\'\"\n]/, meaning it is not a valid value';
		}
	}

	my $command
		= 'env NOINTER=1 cbsd bcreate jname="'
		. $opts{name}
		. '" vm_os_type="'
		. $opts{vm_os_type}
		. '" vm_os_profile="'
		. $opts{vm_os_profile}
		. '" inter=0';

	# the variables to check for.
	my @variables = (
		'bhyve_vnc_tcp_bind', 'imgsize',          'interface2',        'nic_flags',
		'nic_flags2',         'quiet',            'removejconf',       'runasap',
		'vm_cpus',            'vm_ram',           'zfs_snapsrc',       'ci_gw4',
		'ci_interface2',      'ci_interface_mtu', 'ci_interface_mtu2', 'ci_ip4_addr',
		'ci_ip4_addr2',       'ci_user_pubkey',   'ci_user_pw_user',   'ci_user_pw_root'
	);

	# add each found variable to the command
	foreach my $key (@variables) {

		if ( defined( $opts{$key} ) ) {

			# make sure it does not contain any tabs, single/double quotes, and new lines
			if ( $opts{$key} =~ /[\t\'\"\n]/ ) {
				die 'The value "'
					. $opts{$key}
					. '" for key "'
					. $key
					. '" matched /[\t\'\"\n]/, meaning it is not a valid value';
			}

			if ( defined( $opts{$key} ) ) {
				$command = $command.' ' . $key . '="' . $opts{$key} . '"';
			}
		}
	}

	Rex::Logger::debug( "Creating a new CBSD VM via... " . $command );

	my $returned = i_run( $command, fail_ok => 1 );

	# the output is colorized
	$returned = colorstrip($returned);

	# test after no such as that will also exit non-zero
	if ( $? != 0 ) {
		die( "Error running '" . $command . "' returned... " . $returned );
	}

	return $returned;
}

1;