From de8cf13102e6ba3fefb36c16ea4ae81f29ff8742 Mon Sep 17 00:00:00 2001
From: Simon Bomm <sbomm@merethis.com>
Date: Fri, 6 Feb 2015 17:11:22 +0100
Subject: [PATCH] Add partitioning and broker retention checks

---
 apps/centreon/local/mode/retentionbroker.pm | 122 ++++++++++++++++++++
 apps/centreon/local/plugin.pm               |   3 +-
 2 files changed, 124 insertions(+), 1 deletion(-)
 create mode 100644 apps/centreon/local/mode/retentionbroker.pm

diff --git a/apps/centreon/local/mode/retentionbroker.pm b/apps/centreon/local/mode/retentionbroker.pm
new file mode 100644
index 000000000..bc57ec1f3
--- /dev/null
+++ b/apps/centreon/local/mode/retentionbroker.pm
@@ -0,0 +1,122 @@
+################################################################################
+# Copyright 2005-2013 MERETHIS
+# Centreon is developped by : Julien Mathis and Romain Le Merlus under
+# GPL Licence 2.0.
+# 
+# This program is free software; you can redistribute it and/or modify it under 
+# the terms of the GNU General Public License as published by the Free Software 
+# Foundation ; either version 2 of the License.
+# 
+# This program is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A 
+# PARTICULAR PURPOSE. See the GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License along with 
+# this program; if not, see <http://www.gnu.org/licenses>.
+# 
+# Linking this program statically or dynamically with other modules is making a 
+# combined work based on this program. Thus, the terms and conditions of the GNU 
+# General Public License cover the whole combination.
+# 
+# As a special exception, the copyright holders of this program give MERETHIS 
+# permission to link this program with independent modules to produce an executable, 
+# regardless of the license terms of these independent modules, and to copy and 
+# distribute the resulting executable under terms of MERETHIS choice, provided that 
+# MERETHIS also meet, for each linked independent module, the terms  and conditions 
+# of the license of that module. An independent module is a module which is not 
+# derived from this program. If you modify this program, you may extend this 
+# exception to your version of the program, but you are not obliged to do so. If you
+# do not wish to do so, delete this exception statement from your version.
+# 
+# For more information : contact@centreon.com
+# Authors : Florian Asche <info@florian-asche.de>
+#
+####################################################################################
+
+package apps::centreon::local::mode::retentionbroker;
+
+use base qw(centreon::plugins::mode);
+
+use strict;
+use warnings;
+use XML::LibXML;
+
+sub new {
+    my ($class, %options) = @_;
+    my $self = $class->SUPER::new(package => __PACKAGE__, %options);
+    bless $self, $class;
+    
+    $self->{version} = '1.0';
+    $options{options}->add_options(arguments =>
+                                {
+                                 "rrd-config-file:s"    => { name => 'rrd_config_file', default => 'central-rrd.xml' },
+                                 "sql-config-file:s"    => { name => 'sql_config_file', default => 'central-broker.xml' },
+                                 "config-path"          => { name => 'config_path', default => '/etc/centreon-broker/' },
+                                });
+    return $self;
+}
+
+sub check_options {
+    my ($self, %options) = @_;
+    $self->SUPER::init(%options);
+    if ($self->{option_results}->{config_path} !~ /\/$/) {
+        $self->{output}->add_option_msg(short_msg => "Please set the last / the path to your config-path option");
+        $self->{output}->option_exit();
+    }
+}
+
+sub run {
+    my ($self, %options) = @_;
+    
+    my $parser = XML::LibXML->new();
+    
+    my @configFiles;
+    push @configFiles, $self->{option_results}->{config_path}.$self->{option_results}->{rrd_config_file}, $self->{option_results}->{config_path}.$self->{option_results}->{sql_config_file};
+    
+    $self->{output}->output_add(severity => 'OK',
+                                short_msg => 'No retention files, centreon-broker outputs are OK');
+
+     
+    foreach my $file (@configFiles) {
+        my $config = $parser->parse_file($file);
+        foreach my $data ($config->findnodes('/centreonBroker/output/path')) {
+            next if ($data->to_literal =~ m/rrdcached/);
+            if (-f $data->to_literal) {
+                $self->{output}->output_add(severity => 'CRITICAL',
+                                            short_msg => sprintf("Retention on : %s", $data->to_literal)
+                                            );
+            }
+
+        }
+    }
+
+    $self->{output}->display();
+    $self->{output}->exit();
+    
+}
+
+1;
+
+__END__
+
+=head1 MODE
+
+Check if one of centreon-broker output is dead and failover file is present. CRITICAL STATE ONLY
+
+=over 8
+
+=item B<--rrd-config-file>
+
+Specify the name of your master rrd config-file (default: central-rrd.xml)
+
+=item B<--sql-config-file>
+
+Specify the name of your master sql config file (default: central-broker.xml)
+
+=item B<--config-path>
+
+Specify the path to your broker config files (defaut: /etc/centreon-broker/)
+
+=back
+
+=cut
diff --git a/apps/centreon/local/plugin.pm b/apps/centreon/local/plugin.pm
index 60b30c8f9..399be9702 100644
--- a/apps/centreon/local/plugin.pm
+++ b/apps/centreon/local/plugin.pm
@@ -48,7 +48,8 @@ sub new {
     $self->{version} = '0.1';
     %{$self->{modes}} = (
                          'metaservice'  => 'apps::centreon::local::mode::metaservice',
-                         'partitioning' => 'apps::centreon::local::mode::partitioning'
+                         'partitioning' => 'apps::centreon::mysql::mode::partitioning',
+                         'retention-broker' => 'apps::centreon::local::mode::retentionbroker',
                          );
 
     return $self;