#! /bin/bash

# Copyright (C) 2004,2012 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>

#

# 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; version 2 and/or 3 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, write to the Free Software

# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

# Usage:  clamav-notify-servers

CFGFILE=/etc/sysconfig/clamav-servers

OP=nc

NC=nc

TIMEOUT_TERM=30

TIMEOUT_KILL=60

K_OPT=1

LANG=C timeout --help | grep -q -- '--kill-after' || unset K_OPT

get_sockets() {

    (

	cd /etc/clamd.d

	for i in *.conf; do

	    test -f "$i" || continue

	    test -r "$i" || continue

	    base=${i%%.conf}

	    S=/var/run/clamd.$base/clamd.sock

	    test -S "$S" || continue

	    test -w "$S" || continue

	    echo "$S"

	done

    )

}

op_nc() {

    cd /var/run

    for i in ${CLAMD_SOCKETS}; do

	case $i in

	  (/*)

		test -S "$i" || { 

		  echo "socket '$i' does not exist" >&2

		  continue

		}

		CMD=( $NC -U $i )

		;;

	  (ip4:*)

		addr=${i##ip4:}

		CMD=( $NC -4 ${addr%%:*} ${addr##*:} )

		;;

	  (ip6:*)

		addr=${i##ip6:}

		CMD=( $NC -6 ${addr%%:*} ${addr##*:} )

		;;

	  (*)

		echo "unsupported socket name '$i'" >&2

		continue

	esac

	resp=`printf 'zRELOAD\0' | timeout ${K_OPT:+-k $TIMEOUT_KILL} $TIMEOUT_TERM "${CMD[@]}"`

	case $resp in

	  (RELOADING)

		;;

	  (*)

		echo "clamd server '$i' gave '$resp' response" >&2

		let ++fail

		;;

	esac

    done

    cd - &>/dev/null

}

CLAMD_SOCKETS=`get_sockets`

f=$CFGFILE

test ! -e "$f" || . "$f"

fail=0

op_$OP "$@"

exit $fail