File Coverage

blib/lib/Umlmgr/Uml.pm

Criterion	Covered	Total	%
statement	43	118	36.4
branch	6	54	11.1
condition	0	14	0.0
subroutine	9	15	60.0
pod	0	8	0.0
total	58	209	27.7

package Umlmgr::Uml;

use 5.010000;
use strict;
use warnings;
use Config::IniFiles;
use IPC::Open3;
use Sys::Syslog;
use Umlmgr::Utils;

sub new {
    my ($class, $config, %options) = @_;

    my $self = {};
    eval {
    $self->{config} = Config::IniFiles->new(
        -file => $config
    );
    };
    return if (! $self->{config});
    $self->{$_} = $options{$_} foreach (keys %options);
    ($self->{umid}) = $config =~ m:([^/]+)\.uml$:;
    Sys::Syslog::openlog('umlmgr', 'pid', 'daemon');
    bless($self, $class);
}

sub build_uml_cmd {
    my ($self) = @_;

    my @cmd;

    push(@cmd, $self->{config}->val('env', 'kernel', 'linux'));

    foreach my $param ($self->{config}->Parameters('uml')) {
        $self->{config}->val('uml', $param) or next;
        push(@cmd, $param . '=' . $self->{config}->val('uml', $param));
    }
    if (!$self->{config}->val('uml', 'umid')) {
        push(@cmd, "umid=$self->{umid}");
    }
    if (!$self->{config}->val('uml', 'con0')) {
        push(@cmd, 'con0=fd:0,fd:1');
    }
    if (!$self->{config}->val('uml', 'con')) {
        push(@cmd, 'con=pts');
    }

    return @cmd;
}

sub onboot {
    my ($self) = @_;
    $self->{config}->val('env', 'onboot')
}

sub id {
    my ($self) = @_;
    $self->{config}->val('uml', 'umid', $self->{umid});
}

sub start {
    my ($self) = @_;

    if ($> == 0 && !$self->{user}) {
        warn "No user defined, running vm as root is a bad idea\n";
        return;
    }

    my $pid = fork;
    return if (!defined($pid));

    if ($pid) {
        return 1; # we hope that's ok
    } else {
        if ($self->{user}) {
            Umlmgr::Utils::become_user($self->{user}) or return;
        }
        my @cmd = $self->build_uml_cmd;
        my $id = $self->id;
        $self->{config} = undef;
        Sys::Syslog::syslog('info', $id . ': Starting as ' . join(' ', @cmd));
        my $upid = open3(my $in, my $out, my $err, @cmd);
        while (my $line = <$out>) {
            chomp($line);
            Sys::Syslog::syslog('info', $id . " $line");
        }
        waitpid($upid, 0);
        Sys::Syslog::syslog('info', $id . " terminated with exit status $?");
        exit(0);
    }
}

sub stop {
    my ($self) = @_;

    if ($> == 0 && !$self->{user}) {
        warn "No user defined, running vm as root is a bad idea\n";
        return;
    }

    my $pid = fork;
    if (!defined($pid)) { return }
    if ($pid) {
        waitpid($pid, 0);
    } else {
    if ($self->{user}) {
        Umlmgr::Utils::become_user($self->{user}) or return;
    }
    Sys::Syslog::syslog('info', $self->id . ": sending halt");
    system('uml_mconsole', $self->id, 'halt');
    exit(0);
    }
}

sub status {
    my ($self) = @_;

    if ($> == 0 && !$self->{user}) {
        warn "No user defined, running vm as root is a bad idea\n";
        return;
    }

    my $pid = fork;
    return if (!defined($pid));
    if ($pid) {
        waitpid($pid, 0);
        return(($? >> 8) == 0);
    } else {
        if ($self->{user}) {
            Umlmgr::Utils::become_user($self->{user}) or return;
        }
        my $opid = open3(my $in, my $out, my $err, 'uml_mconsole', $self->id, 'version');
        waitpid($opid, 0);
        exit($? >> 8);
    }
}

sub get_console {
    my ($self, $console) = @_;

    if ($> == 0 && !$self->{user}) {
        warn "No user defined, running vm as root is a bad idea\n";
        return;
    }
    my $pid = fork;
    return if (!defined($pid));
    if ($pid) {
        waitpid($pid, 0);
        return(($? >> 8) == 0);
    } else {
        if ($self->{user}) {
            Umlmgr::Utils::become_user($self->{user}) or return;
        }
        my $pts;

        if (open(my $hd,
            join(' ', map { quotemeta($_) }
            ('uml_mconsole', $self->id, 'config', $console)) . ' |')) {
            my $line = <$hd>;
            $line ||= '';
            chomp $line;
            ($pts) = $line =~ /^OK pts:(.*)/;
            close($hd);
        }
        if ($pts) {
            print "$pts\n";
        } else {
            warn "Can't find a valid console\n";
        }
    }
}

# Preloaded methods go here.

# Autoload methods go after =cut, and are processed by the autosplit program.

1;
__END__