From d1b78ef911f8565b36049677ae41fcb342167e6a Mon Sep 17 00:00:00 2001 From: Shini31 Date: Thu, 23 Jul 2015 14:41:43 +0200 Subject: [PATCH] add traffic mode --- apps/docker/mode/traffic.pm | 305 ++++++++++++++++++++++++++++++++++++ 1 file changed, 305 insertions(+) create mode 100644 apps/docker/mode/traffic.pm diff --git a/apps/docker/mode/traffic.pm b/apps/docker/mode/traffic.pm new file mode 100644 index 000000000..d75bc8fa8 --- /dev/null +++ b/apps/docker/mode/traffic.pm @@ -0,0 +1,305 @@ +############################################################################### +# Copyright 2005-2015 CENTREON +# 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 . +# +# 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 CENTREON +# permission to link this program with independent modules to produce an timeelapsedutable, +# regardless of the license terms of these independent modules, and to copy and +# distribute the resulting timeelapsedutable under terms of CENTREON choice, provided that +# CENTREON 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 : Mathieu Cinquin +# +#################################################################################### + +package apps::docker::mode::traffic; + +use base qw(centreon::plugins::mode); + +use strict; +use warnings; +use centreon::plugins::httplib; +use centreon::plugins::statefile; +use JSON; + +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 => + { + "hostname:s" => { name => 'hostname' }, + "port:s" => { name => 'port', default => '2376'}, + "proto:s" => { name => 'proto', default => 'https' }, + "urlpath:s" => { name => 'url_path', default => '/' }, + "name:s" => { name => 'name' }, + "id:s" => { name => 'id' }, + "warning-in:s" => { name => 'warning_in' }, + "critical-in:s" => { name => 'critical_in' }, + "warning-out:s" => { name => 'warning_out' }, + "critical-out:s" => { name => 'critical_out' }, + "credentials" => { name => 'credentials' }, + "username:s" => { name => 'username' }, + "password:s" => { name => 'password' }, + "ssl:s" => { name => 'ssl', }, + "cert-file:s" => { name => 'cert_file' }, + "key-file:s" => { name => 'key_file' }, + "cacert-file:s" => { name => 'cacert_file' }, + "timeout:s" => { name => 'timeout', default => '3' }, + }); + + $self->{statefile_value} = centreon::plugins::statefile->new(%options); + + return $self; +} + +sub check_options { + my ($self, %options) = @_; + $self->SUPER::init(%options); + + if (!defined($self->{option_results}->{hostname})) { + $self->{output}->add_option_msg(short_msg => "Please set the hostname option"); + $self->{output}->option_exit(); + } + if ((defined($self->{option_results}->{credentials})) && (!defined($self->{option_results}->{username}) || !defined($self->{option_results}->{password}))) { + $self->{output}->add_option_msg(short_msg => "You need to set --username= and --password= options when --credentials is used"); + $self->{output}->option_exit(); + } + if ((defined($self->{option_results}->{name})) && (defined($self->{option_results}->{id}))) { + $self->{output}->add_option_msg(short_msg => "Please set the name or id option"); + $self->{output}->option_exit(); + } + if ((!defined($self->{option_results}->{name})) && (!defined($self->{option_results}->{id}))) { + $self->{output}->add_option_msg(short_msg => "Please set the name or id option"); + $self->{output}->option_exit(); + } + if (($self->{perfdata}->threshold_validate(label => 'warning-in', value => $self->{option_results}->{warning_in})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong warning 'in' threshold '" . $self->{option_results}->{warning_in} . "'."); + $self->{output}->option_exit(); + } + if (($self->{perfdata}->threshold_validate(label => 'critical-in', value => $self->{option_results}->{critical_in})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong critical 'in' threshold '" . $self->{option_results}->{critical_in} . "'."); + $self->{output}->option_exit(); + } + if (($self->{perfdata}->threshold_validate(label => 'warning-out', value => $self->{option_results}->{warning_out})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong warning 'out' threshold '" . $self->{option_results}->{warning_out} . "'."); + $self->{output}->option_exit(); + } + if (($self->{perfdata}->threshold_validate(label => 'critical-out', value => $self->{option_results}->{critical_out})) == 0) { + $self->{output}->add_option_msg(short_msg => "Wrong critical 'out' threshold '" . $self->{option_results}->{critical_out} . "'."); + $self->{output}->option_exit(); + } + + $self->{statefile_value}->check_options(%options); +} + +sub run { + my ($self, %options) = @_; + + my $new_datas = {}; + + if (defined($self->{option_results}->{id})) { + $self->{statefile_value}->read(statefile => 'docker_' . $self->{option_results}->{id} . '_' . centreon::plugins::httplib::get_port($self) . '_' . $self->{mode}); + } elsif (defined($self->{option_results}->{name})) { + $self->{statefile_value}->read(statefile => 'docker_' . $self->{option_results}->{name} . '_' . centreon::plugins::httplib::get_port($self) . '_' . $self->{mode}); + } + + my $jsoncontent; + my $query_form_get = { stream => 'false' }; + + if (defined($self->{option_results}->{id})) { + $self->{option_results}->{url_path} = "/containers/".$self->{option_results}->{id}."/stats"; + $jsoncontent = centreon::plugins::httplib::connect($self, query_form_get => $query_form_get, connection_exit => 'critical'); + } elsif (defined($self->{option_results}->{name})) { + $self->{option_results}->{url_path} = "/containers/".$self->{option_results}->{name}."/stats"; + $jsoncontent = centreon::plugins::httplib::connect($self, query_form_get => $query_form_get, connection_exit => 'critical'); + } + + my $json = JSON->new; + + my $webcontent; + + eval { + $webcontent = $json->decode($jsoncontent); + }; + + if ($@) { + $self->{output}->add_option_msg(short_msg => "Cannot decode json response"); + $self->{output}->option_exit(); + } + + my $rx_bytes = $webcontent->{network}->{rx_bytes}; + my $tx_bytes = $webcontent->{network}->{tx_bytes}; + $new_datas->{rx_bytes} = $rx_bytes; + $new_datas->{tx_bytes} = $tx_bytes; + $new_datas->{last_timestamp} = time(); + my $old_timestamp = $self->{statefile_value}->get(name => 'last_timestamp'); + + if (!defined($old_timestamp)) { + $self->{output}->output_add(severity => 'OK', + short_msg => "Buffer creation..."); + $self->{statefile_value}->write(data => $new_datas); + $self->{output}->display(); + $self->{output}->exit(); + } + + my $time_delta = $new_datas->{last_timestamp} - $old_timestamp; + if ($time_delta <= 0) { + # At least one second. two fast calls ;) + $time_delta = 1; + } + + my $old_rx_bytes = $self->{statefile_value}->get(name => 'rx_bytes'); + my $old_tx_bytes = $self->{statefile_value}->get(name => 'tx_bytes'); + + if ($new_datas->{rx_bytes} < $old_rx_bytes) { + # We set 0. Has reboot. + $old_rx_bytes = 0; + } + if ($new_datas->{tx_bytes} < $old_tx_bytes) { + # We set 0. Has reboot. + $old_tx_bytes = 0; + } + + my $delta_rx_bits = ($rx_bytes - $old_rx_bytes) * 8; + my $delta_tx_bits = ($tx_bytes - $old_tx_bytes) * 8; + my $rx_absolute_per_sec = $delta_rx_bits / $time_delta; + my $tx_absolute_per_sec = $delta_tx_bits / $time_delta; + + my $exit1 = $self->{perfdata}->threshold_check(value => $rx_absolute_per_sec, threshold => [ { label => 'critical-in', 'exit_litteral' => 'critical' }, { label => 'warning-in', exit_litteral => 'warning' } ]); + my $exit2 = $self->{perfdata}->threshold_check(value => $tx_absolute_per_sec, threshold => [ { label => 'critical-out', 'exit_litteral' => 'critical' }, { label => 'warning-out', exit_litteral => 'warning' } ]); + + my ($rx_value, $rx_unit) = $self->{perfdata}->change_bytes(value => $rx_absolute_per_sec, network => 1); + my ($tx_value, $tx_unit) = $self->{perfdata}->change_bytes(value => $tx_absolute_per_sec, network => 1); + my $exit = $self->{output}->get_most_critical(status => [ $exit1, $exit2 ]); + $self->{output}->output_add(severity => $exit, + short_msg => sprintf("Traffic In : %s/s, Out : %s/s", + $rx_value . $rx_unit, + $tx_value . $tx_unit)); + + $self->{output}->perfdata_add(label => 'traffic_in', unit => 'b/s', + value => sprintf("%.2f", $rx_absolute_per_sec), + warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning-in'), + critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical-in'), + min => 0); + $self->{output}->perfdata_add(label => 'traffic_out', unit => 'b/s', + value => sprintf("%.2f", $tx_absolute_per_sec), + warning => $self->{perfdata}->get_perfdata_for_output(label => 'warning-out'), + critical => $self->{perfdata}->get_perfdata_for_output(label => 'critical-out'), + min => 0); + + $self->{statefile_value}->write(data => $new_datas); + + $self->{output}->display(); + $self->{output}->exit(); + +} + +1; + +__END__ + +=head1 MODE + +Check Container's Network traffic usage + +=over 8 + +=item B<--hostname> + +IP Addr/FQDN of Docker's API + +=item B<--port> + +Port used by Docker's API (Default: '2576') + +=item B<--proto> + +Specify https if needed (Default: 'https') + +=item B<--urlpath> + +Set path to get Docker's container information (Default: '/') + +=item B<--id> + +Specify one container's id + +=item B<--name> + +Specify one container's name + +=item B<--warning-in> + +Threshold warning in b/s for 'in' traffic. + +=item B<--critical-in> + +Threshold critical in b/s for 'in' traffic. + +=item B<--warning-out> + +Threshold warning in b/s for 'out' traffic. + +=item B<--critical-out> + +Threshold critical in b/s for 'out' traffic. + +=item B<--credentials> + +Specify this option if you access webpage over basic authentification + +=item B<--username> + +Specify username + +=item B<--password> + +Specify password + +=item B<--ssl> + +Specify SSL version (example : 'sslv3', 'tlsv1'...) + +=item B<--cert-file> + +Specify certificate to send to the webserver + +=item B<--key-file> + +Specify key to send to the webserver + +=item B<--cacert-file> + +Specify root certificate to send to the webserver + +=item B<--timeout> + +Threshold for HTTP timeout (Default: 3) + +=back + +=cut