#!/bin/bash

# Author: Steve Barnes (steve@echo.id.au)

if [ "$CONTAINERTEST" -eq "1" ]; then

    t_Log "Running in container -> SKIP"

    exit 0

fi

t_Log "Running $0 - checking iptraf runs and returns non-zero exit status."

TMP=/tmp/iptraf.log

# clean up after ourselves

trap "[ -e ${TMP} ] && { /bin/rm ${TMP}; }" EXIT

# iptraf only be run by root

[ ${EUID} -eq 0 ] || { t_Log "Not running as root, skipping this test. Non-fatal."; exit $PASS; }

if [ "$centos_ver" -ge 7 ] ; then

 IPTRAF=`which iptraf-ng`

else

 IPTRAF=`which iptraf`

fi

PING=`which ping`

STAT=`which stat`

KILL=`which kill`

[ "$IPTRAF" ] || { t_Log "Failed to find iptraf binary. That ain't good..."; exit $FAIL; }

[ "$PING" ] || { t_Log "Failed to find the ping binary. That ain't good..."; exit $FAIL; }

[ "$STAT" ] || { t_Log "Failed to find the stat binary. That ain't good..."; exit $FAIL; }

[ "$KILL" ] || { t_Log "Failed to find the kill binary. That ain't good..."; exit $FAIL; }

# start iptraf running in the background on all interfaces, logging to a file.

${IPTRAF} -i all -t 1 -B -L ${TMP} &>/dev/null

# give iptraf something to chew on

${PING} -c 5 127.0.0.1 &>/dev/null

# check the our log file actually has some data in it, which it should, given that we just pinged ourselves...

LOGSIZE=`stat -c '%s' ${TMP}`

# kill iptraf

${KILL} -USR2 `pidof $IPTRAF`

# confirm our iptraf log file has something in it

if [ ${LOGSIZE} -gt 0 ] ; then

  t_CheckExitStatus 0

else

  t_Log "iptraf failed to log any traffic?!. That ain't good..."

  t_CheckExitStatus 1

fi