This commit is contained in:
garnier-quentin 2020-09-22 10:28:34 +02:00
parent 3cc69a2150
commit 83cfe2a246
2 changed files with 154 additions and 127 deletions

View File

@ -26,7 +26,7 @@ use strict;
use warnings; use warnings;
use centreon::plugins::misc; use centreon::plugins::misc;
use File::Basename; use File::Basename;
use XML::LibXML; use XML::Simple;
use Win32; use Win32;
sub new { sub new {
@ -35,32 +35,25 @@ sub new {
bless $self, $class; bless $self, $class;
$options{options}->add_options(arguments => { $options{options}->add_options(arguments => {
'config:s' => { name => 'config' }, 'config:s' => { name => 'config' },
'language:s' => { name => 'language', default => 'en' }, 'language:s' => { name => 'language', default => 'en' },
'dfsr' => { name => 'dfsr' }, 'dfsr' => { name => 'dfsr' },
'noeventlog' => { name => 'noeventlog' }, 'noeventlog' => { name => 'noeventlog' },
'nomachineaccount' => { name => 'nomachineaccount' }, 'nomachineaccount' => { name => 'nomachineaccount' },
'timeout:s' => { name => 'timeout', default => 30 }, 'timeout:s' => { name => 'timeout', default => 30 }
}); });
$self->{os_is2003} = 0; $self->{os_is2003} = 0;
$self->{os_is2008} = 0; $self->{os_is2008} = 0;
$self->{os_is2012} = 0; $self->{os_is2012} = 0;
$self->{os_is2016} = 0; $self->{os_is2016} = 0;
$self->{msg} = { global => undef, ok => undef, warning => undef, critical => undef };
return $self; return $self;
} }
sub check_options { sub check_options {
my ($self, %options) = @_; my ($self, %options) = @_;
$self->SUPER::init(%options); $self->SUPER::init(%options);
if (defined($self->{option_results}->{config})) {
$self->{config_file} = $self->{option_results}->{config};
} else {
$self->{output}->add_option_msg(short_msg => "Need to specify config file option.");
$self->{output}->option_exit();;
}
} }
sub check_version { sub check_version {
@ -82,55 +75,67 @@ sub check_version {
} elsif ($ver_major == 10 && $ver_minor == 0) { } elsif ($ver_major == 10 && $ver_minor == 0) {
$self->{os_is2016} = 1; $self->{os_is2016} = 1;
} else { } else {
$self->{output}->output_add(severity => 'UNKNOWN', $self->{output}->output_add(
short_msg => 'OS version ' . $ver_major . '.' . $ver_minor . ' not managed.'); severity => 'UNKNOWN',
short_msg => 'OS version ' . $ver_major . '.' . $ver_minor . ' not managed.'
);
return 1; return 1;
} }
return 0; return 0;
} }
sub load_xml { sub read_config {
my ($self, %options) = @_; my ($self, %options) = @_;
# Load XML File my $content_file = <<END_FILE
my $parser = XML::LibXML->new(); <?xml version="1.0" encoding="UTF-8"?>
my $doc = $parser->parse_file($self->{config_file}); <root>
my $nodes = $doc->getElementsByTagName('dcdiag'); <dcdiag language="en">
if ($nodes->size() == 0) { <messages>
$self->{output}->output_add(severity => 'UNKNOWN', <global>Starting test.*?:\s+(.*?)\n.*?(passed|warning|failed)</global>
short_msg => 'Cannot find a <dcdiag> node in XML Config file'); <ok>passed</ok>
return 1; <warning>warning</warning>
} <critical>failed</critical>
</messages>
my $language_found = 0; </dcdiag>
foreach my $node ($nodes->get_nodelist()) { <dcdiag language="fr">
if ($node->getAttribute('language') eq $self->{option_results}->{language}) { <messages>
$language_found = 1; <global>D.*?marrage du test.*?:\s+(.*?)\n.*?(a r.*?ussi|a .*?chou.|warning)</global>
my $messages_node = $node->getElementsByTagName('messages'); <ok>a r.*?ussi</ok>
foreach my $node2 ($messages_node->get_nodelist()) { <warning>warning</warning>
foreach my $element_msg ($node2->getChildrenByTagName('*')) { <critical>a .*?chou.</critical>
$self->{msg}->{$element_msg->nodeName} = $element_msg->textContent if (exists($self->{msg}->{$element_msg->nodeName})); </messages>
} </dcdiag>
</root>
END_FILE
if (defined($self->{option_results}->{config}) && $self->{option_results}->{config} ne '') {
$content_file = do {
local $/ = undef;
if (!open my $fh, "<", $self->{option_results}->{config}) {
$self->{output}->add_option_msg(short_msg => "Could not open file $self->{option_results}->{config} : $!");
$self->{output}->option_exit();
} }
last; <$fh>;
} };
} }
if ($language_found == 0) { my $content;
$self->{output}->output_add(severity => 'UNKNOWN', eval {
short_msg => "Cannot found language '" . $self->{option_results}->{language} . "' in XML Config file"); $content = XMLin($content_file, ForceArray => ['dcdiag'], KeyAttr => ['language']);
return 1; };
if ($@) {
$self->{output}->add_option_msg(short_msg => "Cannot decode xml response: $@");
$self->{output}->option_exit();
} }
foreach my $label (keys %{$self->{msg}}) { if (!defined($content->{dcdiag}->{$self->{option_results}->{language}})) {
if (!defined($self->{msg}->{$label})) { $self->{output}->add_option_msg(short_msg => "Cannot find language '$self->{option_results}->{language}' in config file");
$self->{output}->output_add(severity => 'UNKNOWN', $self->{output}->option_exit();
short_msg => "Message '" . $label . "' for language '" . $self->{option_results}->{language} . "' not defined in XML Config file");
return 1;
}
} }
return 0; return $content->{dcdiag}->{ $self->{option_results}->{language} };
} }
sub dcdiag { sub dcdiag {
@ -140,13 +145,13 @@ sub dcdiag {
$dcdiag_cmd .= ' /test:machineaccount' if (!defined($self->{option_results}->{nomachineaccount})); $dcdiag_cmd .= ' /test:machineaccount' if (!defined($self->{option_results}->{nomachineaccount}));
$dcdiag_cmd .= ' /test:frssysvol' if ($self->{os_is2003} == 1); $dcdiag_cmd .= ' /test:frssysvol' if ($self->{os_is2003} == 1);
$dcdiag_cmd .= ' /test:sysvolcheck' if ($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1); $dcdiag_cmd .= ' /test:sysvolcheck' if ($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1);
if (!defined($self->{option_results}->{noeventlog})) { if (!defined($self->{option_results}->{noeventlog})) {
$dcdiag_cmd .= ' /test:frsevent /test:kccevent' if ($self->{os_is2003} == 1); $dcdiag_cmd .= ' /test:frsevent /test:kccevent' if ($self->{os_is2003} == 1);
$dcdiag_cmd .= ' /test:frsevent /test:kccevent' if (($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1) && !defined($self->{option_results}->{dfsr})); $dcdiag_cmd .= ' /test:frsevent /test:kccevent' if (($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1) && !defined($self->{option_results}->{dfsr}));
$dcdiag_cmd .= ' /test:dfsrevent /test:kccevent' if (($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1) && defined($self->{option_results}->{dfsr})); $dcdiag_cmd .= ' /test:dfsrevent /test:kccevent' if (($self->{os_is2008} == 1 || $self->{os_is2012} == 1 || $self->{os_is2016} == 1) && defined($self->{option_results}->{dfsr}));
} }
my ($stdout) = centreon::plugins::misc::windows_execute( my ($stdout) = centreon::plugins::misc::windows_execute(
output => $self->{output}, output => $self->{output},
timeout => $self->{option_results}->{timeout}, timeout => $self->{option_results}->{timeout},
@ -154,42 +159,51 @@ sub dcdiag {
command_path => undef, command_path => undef,
command_options => undef command_options => undef
); );
my $match = 0; my $match = 0;
while ($stdout =~ /$self->{msg}->{global}/imsg) { while ($stdout =~ /$options{config}->{messages}->{global}/imsg) {
my ($test_name, $pattern) = ($1, lc($2)); my ($test_name, $pattern) = ($1, lc($2));
if ($pattern =~ /$self->{msg}->{ok}/i) { if ($pattern =~ /$options{config}->{messages}->{ok}/i) {
$match = 1; $match = 1;
$self->{output}->output_add(severity => 'OK', $self->{output}->output_add(
short_msg => $test_name); severity => 'OK',
} elsif ($pattern =~ /$self->{msg}->{critical}/i) { short_msg => $test_name
);
} elsif ($pattern =~ /$options{config}->{messages}->{critical}/i) {
$match = 1; $match = 1;
$self->{output}->output_add(severity => 'CRITICAL', $self->{output}->output_add(
short_msg => 'test ' . $test_name); severity => 'CRITICAL',
} elsif ($pattern =~ /$self->{msg}->{warning}/i) { short_msg => 'test ' . $test_name
);
} elsif ($pattern =~ /$options{config}->{messages}->{warning}/i) {
$match = 1; $match = 1;
$self->{output}->output_add(severity => 'WARNING', $self->{output}->output_add(
short_msg => 'test ' . $test_name); severity => 'WARNING',
short_msg => 'test ' . $test_name
);
} }
} }
if ($match == 0) { if ($match == 0) {
$self->{output}->output_add(severity => 'UNKNOWN', $self->{output}->output_add(
short_msg => 'Cannot match output test (maybe you need to set the good language)'); severity => 'UNKNOWN',
short_msg => 'Cannot match output test (maybe you need to set the good language)'
);
return 1; return 1;
} }
return 0; return 0;
} }
sub run { sub run {
my ($self, %options) = @_; my ($self, %options) = @_;
if ($self->load_xml() == 0 && $self->check_version() == 0) { my $config = $self->read_config();
$self->dcdiag(); if ($self->check_version() == 0) {
$self->dcdiag(config => $config);
} }
$self->{output}->display(); $self->{output}->display();
$self->{output}->exit(); $self->{output}->exit();
} }

View File

@ -37,45 +37,44 @@ sub set_counters {
$self->{maps_counters}->{global} = [ $self->{maps_counters}->{global} = [
{ label => 'sessions-created', nlabel => 'sessions.created.total.count', set => { { label => 'sessions-created', nlabel => 'sessions.created.total.count', set => {
key_values => [ { name => 'sessions_created', diff => 1 } ], key_values => [ { name => 'sessions_created', diff => 1 } ],
output_template => 'created : %s', output_template => 'created: %s',
perfdatas => [ perfdatas => [
{ label => 'sessions_created', value => 'sessions_created', template => '%s', min => 0 }, { label => 'sessions_created', template => '%s', min => 0 }
], ]
} }
}, },
{ label => 'sessions-disconnected', nlabel => 'sessions.disconnected.total.count', set => { { label => 'sessions-disconnected', nlabel => 'sessions.disconnected.total.count', set => {
key_values => [ { name => 'sessions_disconnected', diff => 1 } ], key_values => [ { name => 'sessions_disconnected', diff => 1 } ],
output_template => 'disconnected : %s', output_template => 'disconnected: %s',
perfdatas => [ perfdatas => [
{ label => 'sessions_disconnected', value => 'sessions_disconnected', template => '%s', min => 0 }, { label => 'sessions_disconnected', template => '%s', min => 0 }
], ]
} }
}, },
{ label => 'sessions-reconnected', nlabel => 'sessions.reconnected.total.count', set => { { label => 'sessions-reconnected', nlabel => 'sessions.reconnected.total.count', set => {
key_values => [ { name => 'sessions_reconnected', diff => 1 } ], key_values => [ { name => 'sessions_reconnected', diff => 1 } ],
output_template => 'reconnected : %s', output_template => 'reconnected : %s',
perfdatas => [ perfdatas => [
{ label => 'sessions_reconnected', value => 'sessions_reconnected', template => '%s', min => 0 }, { label => 'sessions_reconnected', template => '%s', min => 0 }
], ]
} }
}, },
{ label => 'sessions-active', nlabel => 'sessions.active.current.count', set => { { label => 'sessions-active', nlabel => 'sessions.active.current.count', set => {
key_values => [ { name => 'sessions_active' } ], key_values => [ { name => 'sessions_active' } ],
output_template => 'current active : %s', output_template => 'current active : %s',
perfdatas => [ perfdatas => [
{ label => 'sessions_active', value => 'sessions_active', template => '%s', min => 0 }, { label => 'sessions_active', template => '%s', min => 0 }
], ]
} }
}, },
{ label => 'sessions-disconnected-current', nlabel => 'sessions.disconnected.current.count', set => { { label => 'sessions-disconnected-current', nlabel => 'sessions.disconnected.current.count', set => {
key_values => [ { name => 'sessions_disconnected_current' } ], key_values => [ { name => 'sessions_disconnected_current' } ],
output_template => 'current disconnected : %s', output_template => 'current disconnected : %s',
perfdatas => [ perfdatas => [
{ label => 'sessions_disconnected_current', value => 'sessions_disconnected_current', { label => 'sessions_disconnected_current', template => '%s', min => 0 }
template => '%s', min => 0 }, ]
],
} }
}, }
]; ];
} }
@ -91,42 +90,56 @@ sub new {
bless $self, $class; bless $self, $class;
$options{options}->add_options(arguments => { $options{options}->add_options(arguments => {
'command:s' => { name => 'command', default => 'qwinsta' }, 'command:s' => { name => 'command', default => 'qwinsta' },
'command-path:s' => { name => 'command_path' }, 'command-path:s' => { name => 'command_path' },
'command-options:s' => { name => 'command_options', default => '/COUNTER' }, 'command-options:s' => { name => 'command_options', default => '/COUNTER' },
'timeout:s' => { name => 'timeout', default => 30 }, 'timeout:s' => { name => 'timeout', default => 30 },
'filter-sessionname:s' => { name => 'filter_sessionname' }, 'filter-sessionname:s' => { name => 'filter_sessionname' },
'config:s' => { name => 'config' }, 'config:s' => { name => 'config' },
'language:s' => { name => 'language', default => 'en' }, 'language:s' => { name => 'language', default => 'en' }
}); });
return $self; return $self;
} }
sub check_options {
my ($self, %options) = @_;
$self->SUPER::check_options(%options);
if (defined($self->{option_results}->{config})) {
$self->{config_file} = $self->{option_results}->{config};
} else {
$self->{output}->add_option_msg(short_msg => "Need to specify config file option.");
$self->{output}->option_exit();;
}
}
sub read_config { sub read_config {
my ($self, %options) = @_; my ($self, %options) = @_;
my $content_file = do { my $content_file = <<END_FILE
local $/ = undef; <?xml version="1.0" encoding="UTF-8"?>
if (!open my $fh, "<", $self->{option_results}->{config}) { <root>
$self->{output}->add_option_msg(short_msg => "Could not open file $self->{option_results}->{config} : $!"); <qwinsta language="en">
$self->{output}->option_exit(); <created>Total sessions created</created>
} <disconnected>Total sessions disconnected</disconnected>
<$fh>; <reconnected>Total sessions reconnected</reconnected>
}; <activestate>Active</activestate>
<disconnectedstate>Disc</disconnectedstate>
<header_sessionname>SESSIONNAME</header_sessionname>
<header_state>STATE</header_state>
</qwinsta>
<qwinsta language="fr">
<created>Nombre total de sessions c.*?s</created>
<disconnected>Nombre total de sessions d.*?connect.*?es</disconnected>
<reconnected>Nombre total de sessions reconnect.*?es</reconnected>
<activestate>Actif</activestate>
<disconnectedstate>D.*?co</disconnectedstate>
<header_sessionname>SESSION</header_sessionname>
<header_state>^.*?TAT</header_state>
</qwinsta>
</root>
END_FILE
if (defined($self->{option_results}->{config}) && $self->{option_results}->{config} ne '') {
$content_file = do {
local $/ = undef;
if (!open my $fh, "<", $self->{option_results}->{config}) {
$self->{output}->add_option_msg(short_msg => "Could not open file $self->{option_results}->{config} : $!");
$self->{output}->option_exit();
}
<$fh>;
};
}
my $content; my $content;
eval { eval {
$content = XMLin($content_file, ForceArray => ['qwinsta'], KeyAttr => ['language']); $content = XMLin($content_file, ForceArray => ['qwinsta'], KeyAttr => ['language']);
@ -135,7 +148,7 @@ sub read_config {
$self->{output}->add_option_msg(short_msg => "Cannot decode xml response: $@"); $self->{output}->add_option_msg(short_msg => "Cannot decode xml response: $@");
$self->{output}->option_exit(); $self->{output}->option_exit();
} }
if (!defined($content->{qwinsta}->{$self->{option_results}->{language}})) { if (!defined($content->{qwinsta}->{$self->{option_results}->{language}})) {
$self->{output}->add_option_msg(short_msg => "Cannot find language '$self->{option_results}->{language}' in config file"); $self->{output}->add_option_msg(short_msg => "Cannot find language '$self->{option_results}->{language}' in config file");
$self->{output}->option_exit(); $self->{output}->option_exit();
@ -146,7 +159,7 @@ sub read_config {
sub read_qwinsta { sub read_qwinsta {
my ($self, %options) = @_; my ($self, %options) = @_;
$self->{output}->output_add(long_msg => $options{stdout}, debug => 1); $self->{output}->output_add(long_msg => $options{stdout}, debug => 1);
if ($options{stdout} !~ /^(.*?)$options{config}->{created}/si) { if ($options{stdout} !~ /^(.*?)$options{config}->{created}/si) {
$self->{output}->add_option_msg(short_msg => "Cannot find information in command output"); $self->{output}->add_option_msg(short_msg => "Cannot find information in command output");
@ -166,7 +179,7 @@ sub read_qwinsta {
my $data = {}; my $data = {};
for (my $pos = 0; $pos <= $#position_wrap; $pos++) { for (my $pos = 0; $pos <= $#position_wrap; $pos++) {
my $area; my $area;
if (length($line) < $position_wrap[$pos]->{begin}) { if (length($line) < $position_wrap[$pos]->{begin}) {
$area = ''; $area = '';
} else { } else {
@ -176,7 +189,7 @@ sub read_qwinsta {
$area = substr($line, $position_wrap[$pos]->{begin}); $area = substr($line, $position_wrap[$pos]->{begin});
} }
} }
$data->{$position_wrap[$pos]->{label}} = '-'; $data->{$position_wrap[$pos]->{label}} = '-';
while ($area =~ /([^\s]+)/g) { while ($area =~ /([^\s]+)/g) {
if (($-[1] >= $position_wrap[$pos]->{word_begin} - $position_wrap[$pos]->{begin} && $-[1] <= $position_wrap[$pos]->{end} - $position_wrap[$pos]->{begin}) if (($-[1] >= $position_wrap[$pos]->{word_begin} - $position_wrap[$pos]->{begin} && $-[1] <= $position_wrap[$pos]->{end} - $position_wrap[$pos]->{begin})
@ -189,7 +202,7 @@ sub read_qwinsta {
} }
push @$session_data, $data; push @$session_data, $data;
} }
return $session_data; return $session_data;
} }
@ -203,7 +216,7 @@ sub read_qwinsta_counters {
if ($options{stdout} =~ /$options{config}->{disconnected}.*?(\d+)/si); if ($options{stdout} =~ /$options{config}->{disconnected}.*?(\d+)/si);
$counters->{sessions_reconnected} = $1 $counters->{sessions_reconnected} = $1
if ($options{stdout} =~ /$options{config}->{reconnected}.*?(\d+)/si); if ($options{stdout} =~ /$options{config}->{reconnected}.*?(\d+)/si);
return $counters; return $counters;
} }
@ -218,7 +231,7 @@ sub manage_selection {
command_path => $self->{option_results}->{command_path}, command_path => $self->{option_results}->{command_path},
command_options => $self->{option_results}->{command_options} command_options => $self->{option_results}->{command_options}
); );
my $datas = $self->read_qwinsta(stdout => $stdout, config => $config); my $datas = $self->read_qwinsta(stdout => $stdout, config => $config);
my $counters = $self->read_qwinsta_counters(stdout => $stdout, config => $config); my $counters = $self->read_qwinsta_counters(stdout => $stdout, config => $config);
@ -229,7 +242,7 @@ sub manage_selection {
$self->{output}->output_add(long_msg => "skipping '" . $session->{$config->{header_sessionname}} . "': no matching filter.", debug => 1); $self->{output}->output_add(long_msg => "skipping '" . $session->{$config->{header_sessionname}} . "': no matching filter.", debug => 1);
next; next;
} }
my ($matching_active, $matching_discon) = (0, 0); my ($matching_active, $matching_discon) = (0, 0);
foreach my $label (keys %$session) { foreach my $label (keys %$session) {
$matching_active = 1 if ($label =~ /$config->{header_state}/ && $matching_active = 1 if ($label =~ /$config->{header_state}/ &&
@ -237,7 +250,7 @@ sub manage_selection {
$matching_discon = 1 if ($label =~ /$config->{header_state}/ && $matching_discon = 1 if ($label =~ /$config->{header_state}/ &&
$session->{$label} =~ /$config->{disconnectedstate}/); $session->{$label} =~ /$config->{disconnectedstate}/);
} }
if ($matching_active == 1 || $matching_discon == 1) { if ($matching_active == 1 || $matching_discon == 1) {
$active++ if ($matching_active == 1); $active++ if ($matching_active == 1);
$disconnected++ if ($matching_discon == 1); $disconnected++ if ($matching_discon == 1);
@ -249,7 +262,7 @@ sub manage_selection {
$self->{global} = { %$counters, sessions_active => $active, sessions_disconnected_current => $disconnected }; $self->{global} = { %$counters, sessions_active => $active, sessions_disconnected_current => $disconnected };
$self->{cache_name} = "windows_" . $self->{mode} . '_' . $self->{cache_name} = 'windows_' . $self->{mode} . '_' .
(defined($self->{option_results}->{filter_counters}) ? md5_hex($self->{option_results}->{filter_counters}) : md5_hex('all')) . '_' . (defined($self->{option_results}->{filter_counters}) ? md5_hex($self->{option_results}->{filter_counters}) : md5_hex('all')) . '_' .
(defined($self->{option_results}->{filter_name}) ? md5_hex($self->{option_results}->{filter_name}) : md5_hex('all')); (defined($self->{option_results}->{filter_name}) ? md5_hex($self->{option_results}->{filter_name}) : md5_hex('all'));
} }