File Coverage

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

Criterion	Covered	Total	%
statement	15	33	45.4
branch	0	12	0.0
condition	0	3	0.0
subroutine	5	6	83.3
pod	0	1	0.0
total	20	55	36.3

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

package Rex::Virtualization::CBSD::bcheckpoint_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 ) = @_;

	if ( !defined( $opts{vm} ) ) {
		die 'The required variable "vm" is not set';
	}

	# set the defaults
	if ( !defined( $opts{suspend} ) ) {
		$opts{suspend} = 0;
	}
	if ( !defined( $opts{name} ) ) {
		$opts{name} = 'checkpoint';
	}

	# make sure all the keys are sane
	if (   ( $opts{vm} =~ /[\t\ \=\\\/\'\"\n\;\&]/ )
		|| ( $opts{name} =~ /[\t\ \=\\\/\'\"\n\;\&]/ ) )
	{
		die 'The value either for "vm", "'
			. $opts{vm}
			. '" or "name", "'
			. $opts{name}
			. '", matched /[\t\ \=\/\\\'\"\n\;\&]/, meaning it is not a valid value';
	}
	if ( $opts{suspend} =~ /^[01]$/ ) {
		die 'The value for "suspend", "'
			. $opts{suspend}
			. '", matched /[\t\'\"\n\;\&]/, meaning it is not a valid value';
	}

	# put together the command
	my $command
		= 'cbsd bcheckpoint mode=create jname=' . $opts{vm} . " name='" . $opts{name} . "' suspend=" . $opts{suspend};

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

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

	# the output is colorized, if there is an error
	$returned = colorstrip($returned);

	# check for this second as no VM will also exit non-zero
	if ( $? != 0 ) {
		die( "Error running '" . $command . "'" );
	}

	return 1;
}

1;