From dcba6d56347ec8004404b3b39e77b8b027225ee1 Mon Sep 17 00:00:00 2001
From: Victor LABORIE <vlaborie@evolix.fr>
Date: Wed, 30 May 2018 16:37:13 +0200
Subject: [PATCH] Merge check_nrpe into bkctld check subcommand

---
 Vagrantfile     |  1 -
 bash_completion |  2 +-
 bkctld          | 29 ++++++++++++++++++++++++++++-
 check_nrpe      | 38 --------------------------------------
 test/nrpe.bats  |  6 +++---
 5 files changed, 32 insertions(+), 44 deletions(-)
 delete mode 100755 check_nrpe

diff --git a/Vagrantfile b/Vagrantfile
index e15dd93..060c15d 100644
--- a/Vagrantfile
+++ b/Vagrantfile
@@ -20,7 +20,6 @@ ln -fs /vagrant/tpl /usr/share/bkctld
 ln -fs /vagrant/bash_completion /usr/share/bash-completion/completions/bkctld
 ln -fs /vagrant/bkctld.conf /etc/default/bkctld
 mkdir -p /usr/lib/nagios/plugins/
-ln -fs /vagrant/check_nrpe /usr/lib/nagios/plugins/check_bkctld
 SCRIPT
 
   $deps = <<SCRIPT
diff --git a/bash_completion b/bash_completion
index 3284b5e..581546c 100644
--- a/bash_completion
+++ b/bash_completion
@@ -9,7 +9,7 @@ function _bkctld()
 
 	cur=${COMP_WORDS[COMP_CWORD]};
 	prev=${COMP_WORDS[COMP_CWORD-1]};
-	commands="init update remove start stop reload restart sync status key port ip inc rm"
+	commands="init update remove start stop reload restart sync status key port ip inc rm check"
 
 	if [ $COMP_CWORD -eq 1 ]; then
 		COMPREPLY=($(compgen -W '${commands}' -- ${cur}))
diff --git a/bkctld b/bkctld
index 813699e..b30cddc 100755
--- a/bkctld
+++ b/bkctld
@@ -28,6 +28,7 @@ Subcommands:
     ip          <jailname>      [<ip>|all]      Set or get allowed(s) ip(s) of <jailname>
     inc                                         Make incremental inc of all jails
     rm                                          Remove old incremtal inc of all jails
+    check                                       Run check on jails (NRPE output)
 
 EOF
 }
@@ -470,6 +471,30 @@ sub_rm() {
     rm "${pidfile}"
 }
 
+sub_check() {
+    cur_time=$(date "+%s")
+    return=0
+    jails=$(ls "${JAILDIR}")
+    for jail in ${jails}; do
+        if [ -f "${JAILDIR}/${jail}/var/log/lastlog" ]; then
+            last_conn=$(stat --format=%Y "${JAILDIR}/${jail}/var/log/lastlog")
+            date_diff=$(( (cur_time - last_conn) / (60*60) ))
+            if [ "${date_diff}" -gt "${CRITICAL}" ]; then
+                echo "CRITICAL - ${jail} - ${date_diff} hours"
+                return=2
+            elif [ "${date_diff}" -gt "${WARNING}" ]; then
+                echo "WARNING - ${jail} - ${date_diff} hours"
+                [ "${return}" -ne 2 ] && return=1
+            fi
+        else
+            echo "CRITICAL - ${jail} doesn't have lastlog !"
+            return=2
+        fi
+    done
+    [ "${return}" -eq 0 ] && echo "OK - Nothing to signal"
+    exit "${return}"
+}
+
 ## main function : check usage and valid params
 
 main() {
@@ -486,6 +511,8 @@ main() {
     AUTHORIZED_KEYS="${AUTHORIZED_KEYS:-/root/.ssh/authorized_keys}"
     FIREWALL_RULES="${FIREWALL_RULES:-}"
     LOGLEVEL="${LOGLEVEL:-6}"
+    CRITICAL="${CRITICAL:-48}"
+    WARNING="${WARNING:-24}"
     BTRFS=$(command -v btrfs)
     mkdir -p "${CONFDIR}" "${JAILDIR}" "${INCDIR}"
     subcommand="${1:-}"
@@ -495,7 +522,7 @@ main() {
         "" | "-h" | "--help")
             usage
             ;;
-        "inc" | "rm")
+        "inc" | "rm" | "check")
             "sub_${subcommand}"
         ;;
             "init")
diff --git a/check_nrpe b/check_nrpe
deleted file mode 100755
index c449b51..0000000
--- a/check_nrpe
+++ /dev/null
@@ -1,38 +0,0 @@
-#!/bin/sh
-#
-# bkctld(8) - NRPE check
-#
-# Copyright (c) 2017 Victor Laborie <vlaborie@evolix.fr>
-#
-
-[ -f /etc/default/bkctld ] && . /etc/default/bkctld
-
-[ -z "$JAILDIR" ] && JAILDIR="/backup/jails"
-
-[ -z "$CRITICAL" ] && CRITICAL=48
-[ -z "$WARNING" ] && WARNING=24
-
-cur_time=$(date "+%s")
-return=0
-
-jails=$(ls "$JAILDIR")
-for jail in $jails; do
-    if [ -f "$JAILDIR/$jail/var/log/lastlog" ]; then
-        last_conn=$(stat --format=%Y "$JAILDIR/$jail/var/log/lastlog")
-        date_diff=$(( ( $cur_time - $last_conn ) / (60*60) ))
-        if [ "$date_diff" -gt "$CRITICAL" ]; then
-            echo "CRITICAL - $jail - $date_diff hours"
-            return=2
-        elif [ "$date_diff" -gt "$WARNING" ]; then
-            echo "WARNING - $jail - $date_diff hours"
-            [ "$return" -ne 2 ] && return=1
-        fi
-    else
-        echo "CRITICAL - $jail doesn't have lastlog !"
-        return=2
-    fi
-done
-
-[ "$return" -eq 0 ] && echo "OK - Nothing to signal"
-
-exit "$return"
diff --git a/test/nrpe.bats b/test/nrpe.bats
index 6caeddf..6b72133 100755
--- a/test/nrpe.bats
+++ b/test/nrpe.bats
@@ -9,18 +9,18 @@ teardown() {
 }
 
 @test "ok" {
-    run /usr/lib/nagios/plugins/check_bkctld
+    run bkctld check
     [ "$status" -eq 0 ]
 }
 
 @test "warning" {
     touch --date="$(date -d -2days)" /backup/jails/*/var/log/lastlog
-    run /usr/lib/nagios/plugins/check_bkctld
+    run bkctld check
     [ "$status" -eq 1 ]
 }
 
 @test "critical" {
     touch --date="$(date -d -3days)" /backup/jails/*/var/log/lastlog
-    run /usr/lib/nagios/plugins/check_bkctld
+    run bkctld check
     [ "$status" -eq 2 ]
 }