From be0652270edb64067791612b62bfe1777ed90846 Mon Sep 17 00:00:00 2001
From: vvitkov <zeridon@users.noreply.github.com>
Date: Wed, 31 Oct 2018 16:42:57 +0200
Subject: [PATCH] rtmp check

---
 .../check_mk-agent/check_rtmp.cfg             |  1 +
 .../check_mk-agent/check_rtmp.sh              | 91 +++++++++++++++++++
 .../check_mk-agent/count-streams.sh           |  2 +-
 3 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 monitoring/custom-scripts/check_mk-agent/check_rtmp.cfg
 create mode 100644 monitoring/custom-scripts/check_mk-agent/check_rtmp.sh

diff --git a/monitoring/custom-scripts/check_mk-agent/check_rtmp.cfg b/monitoring/custom-scripts/check_mk-agent/check_rtmp.cfg
new file mode 100644
index 0000000..4457c26
--- /dev/null
+++ b/monitoring/custom-scripts/check_mk-agent/check_rtmp.cfg
@@ -0,0 +1 @@
+URIS="rtmp://127.0.0.1/st/hall-a rtmp://127.0.0.1/st/hall-b rtmp://127.0.0.1/st/hall-c"
\ No newline at end of file
diff --git a/monitoring/custom-scripts/check_mk-agent/check_rtmp.sh b/monitoring/custom-scripts/check_mk-agent/check_rtmp.sh
new file mode 100644
index 0000000..797b9c4
--- /dev/null
+++ b/monitoring/custom-scripts/check_mk-agent/check_rtmp.sh
@@ -0,0 +1,91 @@
+#!/bin/sh
+# FILE: "check_rtmp"
+# DESCRIPTION:check_mk plugin for checking rtmp streams.
+# REQUIRES: rtmpdump (http://rtmpdump.mplayerhq.hu/)
+# AUTHOR: vladimir vitkov
+# DATE: oct-2018
+# $Id:$
+#
+
+PROGNAME=`readlink -f $0`
+PROGPATH=`echo $PROGNAME | sed -e 's,[\\/][^\\/][^\\/]*$,,'`
+REVISION=`echo '$Revision: .2 $' | sed -e 's/[^0-9.]//g'`
+
+RTMPDUMP=`which rtmpdump`
+
+print_usage() {
+  echo "Usage:"
+  echo "  $PROGNAME -u <url> -t <timeout> "
+  echo "  $PROGNAME -h "
+}
+
+echo "<<<mrpe>>>"
+print_help() {
+    print_revision $PROGNAME $REVISION
+    echo ""
+    print_usage
+
+    echo "Comprova l'estat d'un stream RTMP"
+    echo ""
+    echo "Opcions:"
+    echo "  -u URL a testejar Exemple: rtmp://server/app/streamName"
+    echo "  -t Temps a monitoritzar"
+    echo ""
+    exit $STATE_UNKNOWN
+}
+
+STATE_OK=0
+STATE_WARNING=1
+STATE_CRITICAL=2
+STATE_UNKNOWN=3
+
+URLS=""
+TIMEOUT=2
+
+if [ -e $MK_CONFDIR/check_rtmp.cfg ] ; then
+    . $MK_CONFDIR/check_rtmp.cfg
+fi
+
+# Proces de parametres
+while getopts ":u:t:h" Option
+do
+        case $Option in
+                u ) URLS=$OPTARG;;
+                t ) TIMEOUT=$OPTARG;;
+                h ) print_help;;
+                * ) echo "unimplemented option";;
+                esac
+done
+
+if [ ! $URLS ] ; then
+    echo "Stream_UNKNOWN ${STATE_UNKNOWN} No URL's defined"
+
+fi
+
+for _uri in ${URLS} ; do
+    # Construir noms de fitxers temporals
+    NAME=`echo $URL | sed -e s/[^A-Za-z0-9.]/_/g`
+    ERR=/tmp/check_rtmp_err_$NAME.tmp
+
+    # Testejant
+    ( $RTMPDUMP -m 4 --live -r $URL --stop $TIMEOUT > /dev/null 2> $ERR ) & sleep 5; kill $! 2> /dev/null
+    _data=$?
+
+    # Retorn de resultats
+    CONNECTA=`grep "INFO: Connected" $ERR`
+
+    if [ -z "$CONNECTA" ]
+    then
+        echo "echo Stream_${_uri} ${STATE_CRITICAL} Cannot Connect"
+    else
+       ERROR=`grep "INFO: Metadata:" $ERR`
+       if [ ! -z "$ERROR" ]
+       then
+            echo "Stream_${_uri} ${STATE_OK} Stream is normal"
+        fi
+        echo "Stream_${_uri} ${STATE_CRITICAL} Stream is not broadcasting"
+    fi
+
+    echo "Stream_${_uri} ${STATE_UNKNOWN} Unknown output from stream check. Manual check is advised"
+    exit $STATE_UNKNOWN
+done
\ No newline at end of file
diff --git a/monitoring/custom-scripts/check_mk-agent/count-streams.sh b/monitoring/custom-scripts/check_mk-agent/count-streams.sh
index ebb6817..586604f 100644
--- a/monitoring/custom-scripts/check_mk-agent/count-streams.sh
+++ b/monitoring/custom-scripts/check_mk-agent/count-streams.sh
@@ -35,5 +35,5 @@ for STREAMER in ${STREAMERS} ; do
             _res=2
     fi
 
-    echo Streamcount_${STREAMER} ${_res} ${_data}
+    echo "Streamcount_${STREAMER} ${_res} ${_data}"
 done
\ No newline at end of file