use warnings;

use strict;

use 5.010;

use Getopt::Long;

use Time::HiRes qw(clock_gettime CLOCK_MONOTONIC);

our $VERSION = '1.2.1';

our $SHORTNAME = "SYSTEMD";

our %MP_CODES = (

OK => 0,

WARNING => 1,

CRITICAL => 2,

UNKNOWN => 3,

);

our %MP_TEXTS = reverse %MP_CODES;

my $ec = $MP_CODES{OK};

my $et;

my $pd;

sub print_help {

print("check_systemd.pl $VERSION

");

exit($MP_CODES{UNKNOWN});

}

my $opt = {};

Getopt::Long::Configure(qw( gnu_getopt ignore_case_always ));

Getopt::Long::GetOptions(

'unit|u=s' => \$opt->{'unit'},

'exclude|e=s@' => \$opt->{'exclude'},

'legacy|l' => \$opt->{'legacy'},

'warning|w=s' => \$opt->{'warning'},

'critical|c=s' => \$opt->{'critical'},

'verbose|v' => \$opt->{'verbose'},

'help|h' => \$opt->{'help'},

);

print_help() if (defined($opt->{'help'}));

plugin_die("Warning threshold must be an integer.") if (defined($opt->{'warning'}) && $opt->{'warning'} !~ /^\d+$/);

plugin_die("Critical threshold must be an integer.") if (defined($opt->{'critical'}) && $opt->{'critical'} !~ /^\d+$/);

sub set_ec {

my ($ec_new) = @_;

$ec = $ec_new if ($ec_new > $ec);

}

sub get_new_ec {

my ($ec_old, $ec_new) = @_;

return $ec_new if ($ec_new > $ec_old);

return $ec_old;

}

sub check_exclude {

my ($excludes, $unit) = @_;

my $out = 0;

foreach my $tmp_exclude (@$excludes) {

if ($unit =~ m/$tmp_exclude/gixms) {

$out = 1;

}

}

return $out;

}

sub get_systemd_units {

my $cmd = "systemctl list-units --all --full --no-pager";

my @out = `$cmd`;

my @units_all;

foreach my $line (@out) {

$line =~ s/[^[:ascii:]]//gxms; # Only return ascii-characters to $line

$line =~ s/\s+/\ /gxms; # Replace multiple whitespaces with one

$line =~ s/^\*+//gxms; # Remove leading star

$line =~ s/^\s+//gxms; # Remove leading whitespaces

if (! $line =~ m/\s*/xms) {

my %entry;

my ($unit, $load, $active, $sub, $desc) = ('','','','','');

my @line_split = split(/\s+/, $line);

$unit = $line_split[0]; shift (@line_split);

$load = $line_split[0]; shift (@line_split);

$active = $line_split[0]; shift (@line_split);

$sub = $line_split[0]; shift (@line_split);

$desc = join(' ', @line_split);

if (defined($unit) && defined($load) && defined($active) && defined($sub) &&

$unit ne '' && $load =~ /\S+/ixms && $active =~ /\S+/ixms && $sub ne '') {

$entry{'unit'} = $unit;

$entry{'active'} = $active;

$entry{'sub'} = $sub;

$entry{'description'} = $desc;

$entry{'load'} = $load;

#TODO: Validation

push(@units_all, \%entry);

}

}

}

return (\@units_all);

}

sub get_systemd_unit_parameters {

my ($unit_name) = @_;

my $cmd = "systemctl show $unit_name --property=ActiveState,ActiveEnterTimestamp,ActiveEnterTimestampMonotonic";

my @out = `$cmd`;

my %unit_params;

foreach my $line (@out) {

$line =~ s/[^[:ascii:]]//gxms; # Only return ascii-characters to $line

$line =~ s/\s+/\ /gxms; # Replace multiple whitespaces with one

$line =~ s/^\s+//gxms; # Remove leading whitespaces

if ($line =~ m/^(\S+)=(.+)/) {

$unit_params{$1} = $2;

}

}

return \%unit_params;

}

sub format_duration {

my $seconds = shift;

my $hours = int($seconds / 3600);

my $minutes = int(($seconds % 3600) / 60);

my $secs = $seconds % 60;

return sprintf("%02dh:%02dm:%02ds", $hours, $minutes, $secs);

}

sub plugin_die {

my ($et) = @_;

plugin_exit($MP_CODES{UNKNOWN}, $et);

}

sub plugin_exit {

my ($ec, $et, $pd) = @_;

$ec = $MP_CODES{UNKNOWN} unless defined $MP_CODES{$ec};

print("$SHORTNAME $MP_TEXTS{$ec}: ");

print($et);

print("\n|$pd") if($pd);

print("\n");

exit $ec;

}

sub verb {

my ($text) = @_;

print STDERR "[DEBUG] $text\n" if ($opt->{'verbose'});

}

my ($units) = get_systemd_units();

if ($opt->{'unit'}) {

if (my @units = grep { $_->{unit} eq $opt->{'unit'} } @$units) {

my $unit_active_state = $units[0]->{active};

$et = $opt->{'unit'}.' is '.$unit_active_state.'!';

if ($unit_active_state eq 'failed') {

set_ec($MP_CODES{CRITICAL});

}

if ($unit_active_state eq 'active' && (defined($opt->{'warning'}) || defined($opt->{'critical'}))) {

verb("evaluating ActiveEnterTimestamp");

my $unit_params = get_systemd_unit_parameters($opt->{'unit'});

my $active_enter_timestamp = $unit_params->{'ActiveEnterTimestampMonotonic'};

my $ec_tmp = $MP_CODES{OK};

my $time_since_boot = clock_gettime(CLOCK_MONOTONIC);

my $activeTime = ($time_since_boot * 1_000_000 - $active_enter_timestamp) / 1_000_000;

$activeTime = int($activeTime);

verb("activetime: ${activeTime}s");

my $activeTime_pretty = format_duration($activeTime);

if (defined($opt->{'critical'}) && $opt->{'critical'} > $activeTime){

set_ec($MP_CODES{CRITICAL});

$ec_tmp = get_new_ec($ec_tmp, $MP_CODES{CRITICAL});

}

if (defined($opt->{'warning'}) && $opt->{'warning'} > $activeTime){

set_ec($MP_CODES{WARNING});

$ec_tmp = get_new_ec($ec_tmp, $MP_CODES{WARNING});

}

$et .= " ActiveEnterTime $MP_TEXTS{$ec_tmp}, ${activeTime_pretty} ago.";

# $pd .= " 'activeTime'=${activeTime}s;;;0;";

$pd .= " 'activeTime'=${activeTime}s;";

$pd .= defined($opt->{'warning'}) ? "$opt->{'warning'};" : ";";

$pd .= defined($opt->{'critical'}) ? "$opt->{'critical'};" : ";";

$pd .= "0;"

}

} else {

plugin_die($opt->{'unit'}.' could not be found!');

}

} else {

my $units_excluded = [];

my $units_active = [];

my $units_inactive = [];

my $units_failed = [];

my $units_unknown = [];

foreach my $unit (@$units) {

if (check_exclude($opt->{'exclude'}, $unit->{unit})) {

push(@{$units_excluded}, $unit);

next;

}

if ($unit->{active} eq 'active') {

push(@{$units_active}, $unit);

} elsif ($unit->{active} eq 'inactive') {

push(@{$units_inactive}, $unit);

} elsif ($unit->{active} eq 'failed') {

push(@{$units_failed}, $unit);

} else {

push(@{$units_unknown}, $unit);

}

}

if (@$units_failed >= 1) {

set_ec($MP_CODES{CRITICAL});

}

$et .= @$units_failed.' failed units!';

foreach my $unit (@$units_failed) {

$et .= "\n".$unit->{unit}.': failed';

}

$pd .= " 'count_units'=".@$units;

$pd .= " 'units_active'=".@$units_active;

$pd .= " 'units_inactive'=".@$units_inactive;

$pd .= " 'units_failed'=".@$units_failed;

$pd .= " 'units_unknown'=".@$units_unknown;

$pd .= " 'units_excluded'=".@$units_excluded;

}

if ($ec == $MP_CODES{OK}) {

$et = 'SYSTEMD OK: '.$et;

} elsif ($ec == $MP_CODES{WARNING}) {

$et = 'SYSTEMD WARNING: '.$et;

} elsif ($ec == $MP_CODES{CRITICAL}) {

$et = 'SYSTEMD CRITICAL: '.$et;

} else {

$et = 'SYSTEMD UNKNOWN: '.$et;

}

$et = $et.' |'.$pd if ($pd);

print($et."\n");

exit($ec);