first commit

inc/bash-colors.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+BG_RED="\e[41m\e[97m"
+FG_BLUE="\e[34m"
+FG_YELLOW="\e[33m"
+C_RESET="\e[0m"

inc/main-inc.sh

@@ -0,0 +1,38 @@
+#!/bin/bash
+pushd $(dirname ${BASH_SOURCE[0]}) >/dev/null
+
+# Include configuration
+source ../.env
+export INFLUX_TOKEN
+
+# Include other functions
+source output.sh
+
+# Include all sensor scripts
+if [ -d "sensor-scripts" ]; then
+    for script in $(ls sensor-scripts/*.sh 2>/dev/null); do
+        log "Loading sensor function $(basename ${script} .sh)"
+        source "${script}"
+    done
+fi
+
+# Test if influxdb2 CLI is installed
+if ! which influx >/dev/null; then
+    logError "Influx CLI not found!"
+    exit 1
+fi
+
+# Write data to influx DB
+function writeData {
+    MEASURE="$1"
+    TAGS="$2"
+    FIELDS="$3"
+
+    influx write \
+        -b ${INFLUX_BUCKET} \
+        -o ${INFLUX_ORG} \
+        -p s \
+        "${MEASURE},${TAGS} ${FIELDS}"
+}
+
+popd >/dev/null

inc/output.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+source "bash-colors.sh"
+
+# Echo formatted time right now
+function _now {
+    date +"%y/%m/%d %H:%M:%S"
+}
+
+# Write log to stdout
+function log {
+    echo "[ $(_now) :: INFO ] $@"
+}
+
+# Log with custom tag
+function logAdv {
+    TAG="${1}"
+    shift
+
+    echo -e "${FG_BLUE}[ $(_now) :: ${TAG} ] ${@}${C_RESET}"
+}
+
+# Log tagged for sensor scripts, can be muted with main.sh -qs
+function logScript {
+    if ! [ "${SCRIPTS_QUIET}" = "yes" ]; then
+        echo -e "[ $(_now) :: SENS ] ${@}"
+    fi
+}
+
+# Write error to stderr
+function logError {
+    echo -e "${BG_RED}[ $(_now) :: FAIL ] ${@}${C_RESET}" >&2
+}
+
+# Write pipe input to stdout
+function logPipe {
+    PREFIX="PIPE"
+    [ "${1}" = "" ] || PREFIX="${@}"
+
+    while read -r LINE; do
+        echo -e "${FG_YELLOW}[ $(_now) :: ${PREFIX} ] ${LINE}${C_RESET}"
+    done
+}

inc/sensor-scripts/iperf3.sh

@@ -0,0 +1,55 @@
+#!/bin/bash
+SCRIPT_NAME=$(basename ${BASH_SOURCE[0]})
+
+if ! which jq >/dev/null; then
+    logError "${SCRIPT_NAME}: Did not find jq in path!"
+    return
+elif ! which iperf3 >/dev/null; then
+    logError "${SCRIPT_NAME}: Did not find iperf3 in path!"
+    return
+fi
+
+# Usage: doIperfTest <target-host> [<time>] [<additional-iperf-args>] 
+function doIperfTest {
+    HOST="${1}"
+    shift
+
+    TIME=10
+    if ! [ "${1}" = "" ]; then
+        TIME=${1}
+        LOG_SUFFIX=" for ${TIME} seconds"
+        shift
+    fi
+
+    logScript "Doing IPerf3 test to ${HOST}${LOG_SUFFIX}"
+
+    OUT=$(iperf3 -c "${HOST}" -t ${TIME} -i0 -J ${@})
+
+    SUCCESS=$( [ "$(echo ${OUT} | jq -r .error)" = null ] && echo true || echo false )
+    PROTO=$(echo ${OUT} | jq -r .start.test_start.protocol)
+    REMOTE_HOST=$(echo ${OUT} | jq -r .start.connected[0].remote_host)
+    REMOTE_PORT=$(echo ${OUT} | jq -r .start.connected[0].remote_port)
+    
+    NUM_STREAMS=$(echo ${OUT} | jq -r .start.test_start.num_streams)
+    REVERSE=$(echo ${OUT} | jq -r .start.test_start.reverse)
+    SENT_SECONDS=$(echo ${OUT} | jq -r .end.sum_sent.seconds)
+    SENT_BPS=$(echo ${OUT} | jq -r .end.sum_sent.bits_per_second)
+    SENT_RETRANSMITS=$(echo ${OUT} | jq -r .end.sum_sent.retransmits)
+    RECEIVED_SECONDS=$(echo ${OUT} | jq -r .end.sum_received.seconds)
+    RECEIVED_BPS=$(echo ${OUT} | jq -r .end.sum_received.bits_per_second)
+
+    if ! [ "${SUCCESS}" = "true" ]; then
+        NUM_STREAMS=0
+        REVERSE=0
+        SENT_SECONDS=0
+        SENT_BPS=0
+        SENT_RETRANSMITS=0
+        RECEIVED_SECONDS=0
+        RECEIVED_BPS=0
+    fi
+
+    writeData \
+        iperf3 \
+        "sourceHost=$(hostname -f),targetHost=${HOST}" \
+        "success=${SUCCESS},connected_host=\"${REMOTE_HOST}:${REMOTE_PORT}\",proto=\"${PROTO}\",num_streams=${NUM_STREAMS},reverse=${REVERSE},retransmits=${SENT_RETRANSMITS},sent_seconds=${SENT_SECONDS},sent_bps=${SENT_BPS},received_seconds=${RECEIVED_SECONDS},received_bps=${RECEIVED_BPS}"
+}

inc/sensor-scripts/ping.sh

@@ -0,0 +1,37 @@
+#!/bin/bash
+
+# Usage: doPingTest <target-host> [<count>]
+function doPingTest {
+    HOST=${1}
+
+    logScript "Doing ping to ${HOST}"
+
+    COUNT=1
+    [ "${2}" = "" ] || COUNT=${2}
+
+    ping "${HOST}" -W 10 -c ${COUNT} | {
+        read -r HEADER
+        RESOLVED_NAME=$(echo ${HEADER} | sed 's/.* (\(.*\)) .*/\1/')
+        
+        while read -r PINGLINE; do
+            if [[ "${PINGLINE}" == *" bytes from "* ]]; then
+                SUCCESS=true
+                BYTES=$(echo ${PINGLINE} | awk '{ print $1 }')
+                TTL=$(echo ${PINGLINE} | sed 's/.*ttl=\([0-9]\+\) .*/\1/')
+                LATENCY=$(echo ${PINGLINE} | sed 's/.*time=\([0-9\.]\+\) .*/\1/')
+            elif [ "${PINGLINE}" = "" ]; then
+                SUCCESS=false
+                BYTES=0
+                TTL=0
+                LATENCY=-1
+            else
+                continue
+            fi
+
+            writeData \
+                ping \
+                "sourceHost=$(hostname -f),targetHost=${HOST}" \
+                "success=${SUCCESS},resolved=\"${RESOLVED_NAME}\",latency=${LATENCY},ttl=${TTL},bytes=${BYTES}"
+        done
+    }
+}

inc/sensor-scripts/tcp-port.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+if ! which nc >/dev/null; then
+    logError "${SCRIPT_NAME}: Did not find nc in path!"
+    return
+fi
+
+# Usage: doTcpPortTest <target-host> <target-port> [<timeout>]
+function doTcpPortTest {
+    HOST=${1}
+    PORT=${2}
+    TIMEOUT=5
+    [ "${3}" = "" ] || TIMEOUT=${3}
+
+    logScript "Doing TCP port test to ${HOST}:${PORT}"
+
+    SUCCESS=false
+    if nc -z -w ${TIMEOUT} ${HOST} ${PORT}; then
+        SUCCESS=true
+    fi
+
+    writeData \
+        tcp-port \
+        "sourceHost=$(hostname -f),targetHost=${HOST},targetPort=${PORT}" \
+        "open=${SUCCESS}"
+}