pfSense

#!/bin/sh

#

# rc.update_bogons.sh

#

# part of pfSense (https://www.pfsense.org)

# Copyright (c) 2004-2016 Electric Sheep Fencing, LLC

# All rights reserved.

#

# Based on src/etc/rc.d/savecore from FreeBSD

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions are met:

#

# 1. Redistributions of source code must retain the above copyright notice,

#    this list of conditions and the following disclaimer.

#

# 2. Redistributions in binary form must reproduce the above copyright

#    notice, this list of conditions and the following disclaimer in

#    the documentation and/or other materials provided with the

#    distribution.

#

# 3. All advertising materials mentioning features or use of this software

#    must display the following acknowledgment:

#    "This product includes software developed by the pfSense Project

#    for use in the pfSense® software distribution. (http://www.pfsense.org/).

#

# 4. The names "pfSense" and "pfSense Project" must not be used to

#    endorse or promote products derived from this software without

#    prior written permission. For written permission, please contact

#    coreteam@pfsense.org.

#

# 5. Products derived from this software may not be called "pfSense"

#    nor may "pfSense" appear in their names without prior written

#    permission of the Electric Sheep Fencing, LLC.

#

# 6. Redistributions of any form whatsoever must retain the following

#    acknowledgment:

#

# "This product includes software developed by the pfSense Project

# for use in the pfSense software distribution (http://www.pfsense.org/).

#

# THIS SOFTWARE IS PROVIDED BY THE pfSense PROJECT ``AS IS'' AND ANY

# EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE pfSense PROJECT OR

# ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)

# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,

# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED

# OF THE POSSIBILITY OF SUCH DAMAGE.

# Global variables

proc_error=""

# Download and extract if necessary

process_url() {

	local file=$1

	local url=$2

	local filename=${url##*/}

	local ext=${filename#*.}

	/usr/bin/fetch -a -w 600 -T 30 -q -o $file "${url}"

	if [ ! -f $file ]; then

		echo "Could not download ${url}" | logger

		proc_error="true"

	fi

	case "$ext" in

		tar)

			mv $file $file.tmp

			/usr/bin/tar -xf $file.tmp -O > $file 2> /dev/null

			;;

		tar.gz)

			mv $file $file.tmp

			/usr/bin/tar -xzf $file.tmp -O > $file 2> /dev/null

			;;

		tgz)

			mv $file $file.tmp

			/usr/bin/tar -xzf $file.tmp -O > $file 2> /dev/null

			;;

		tar.bz2)

			mv $file $file.tmp

			/usr/bin/tar -xjf $file.tmp -O > $file 2> /dev/null

			;;

		*)

			;;

	esac

	if [ -f $file.tmp ]; then

		rm $file.tmp

	fi

	if [ ! -f $file ]; then

		echo "Could not extract ${filename}" | logger

		proc_error="true"

	fi

}

echo "rc.update_bogons.sh is starting up." | logger

# Sleep for some time, unless an argument is specified.

if [ "$1" = "" ]; then

	# Grab a random value

	value=`od -A n -d -N2 /dev/random | awk '{ print $1 }'`

	echo "rc.update_bogons.sh is sleeping for $value" | logger

	sleep $value

fi

echo "rc.update_bogons.sh is beginning the update cycle." | logger

# Load custom bogon configuration

if [ -f /var/etc/bogon_custom ]; then

	. /var/etc/bogon_custom

fi

# Set default values if not overriden

v4url=${v4url:-"https://files.pfsense.org/lists/fullbogons-ipv4.txt"}

v6url=${v6url:-"https://files.pfsense.org/lists/fullbogons-ipv6.txt"}

v4urlcksum=${v4urlcksum:-"${v4url}.md5"}

v6urlcksum=${v6urlcksum:-"${v6url}.md5"}

process_url /tmp/bogons "${v4url}"

process_url /tmp/bogonsv6 "${v6url}"

if [ "$proc_error" != "" ]; then

	# Relaunch and sleep

	sh /etc/rc.update_bogons.sh &

	exit

fi

BOGON_V4_CKSUM=`/usr/bin/fetch -T 30 -q -o - "${v4urlcksum}" | awk '{ print $4 }'`

ON_DISK_V4_CKSUM=`md5 /tmp/bogons | awk '{ print $4 }'`

BOGON_V6_CKSUM=`/usr/bin/fetch -T 30 -q -o - "${v6urlcksum}" | awk '{ print $4 }'`

ON_DISK_V6_CKSUM=`md5 /tmp/bogonsv6 | awk '{ print $4 }'`

if [ "$BOGON_V4_CKSUM" = "$ON_DISK_V4_CKSUM" ] || [ "$BOGON_V6_CKSUM" = "$ON_DISK_V6_CKSUM" ]; then

	# At least one of the downloaded checksums matches, so mount RW

	/etc/rc.conf_mount_rw

	ENTRIES_MAX=`pfctl -s memory | awk '/table-entries/ { print $4 }'`

	if [ "$BOGON_V4_CKSUM" = "$ON_DISK_V4_CKSUM" ]; then

		ENTRIES_TOT=`pfctl -vvsTables | awk '/Addresses/ {s+=$2}; END {print s}'`

		ENTRIES_V4=`pfctl -vvsTables | awk '/-\tbogons$/ {getline; print $2}'`

		LINES_V4=`wc -l /tmp/bogons | awk '{ print $1 }'`

		if [ $ENTRIES_MAX -gt $((2*ENTRIES_TOT-${ENTRIES_V4:-0}+LINES_V4)) ]; then

			egrep -v "^192.168.0.0/16|^172.16.0.0/12|^10.0.0.0/8" /tmp/bogons > /etc/bogons

			RESULT=`/sbin/pfctl -t bogons -T replace -f /etc/bogons 2>&1`

			echo "$RESULT" | awk '{ print "Bogons V4 file downloaded: " $0 }' | logger

		else

			echo "Not updating IPv4 bogons (increase table-entries limit)" | logger

		fi

		rm /tmp/bogons

	else

		echo "Could not download ${v4url} (checksum mismatch)" | logger

		checksum_error="true"

	fi

	if [ "$BOGON_V6_CKSUM" = "$ON_DISK_V6_CKSUM" ]; then

		BOGONS_V6_TABLE_COUNT=`pfctl -sTables | grep ^bogonsv6$ | wc -l | awk '{ print $1 }'`

		ENTRIES_TOT=`pfctl -vvsTables | awk '/Addresses/ {s+=$2}; END {print s}'`

		LINES_V6=`wc -l /tmp/bogonsv6 | awk '{ print $1 }'`

		if [ $BOGONS_V6_TABLE_COUNT -gt 0 ]; then

			ENTRIES_V6=`pfctl -vvsTables | awk '/-\tbogonsv6$/ {getline; print $2}'`

			if [ $ENTRIES_MAX -gt $((2*ENTRIES_TOT-${ENTRIES_V6:-0}+LINES_V6)) ]; then

				egrep -iv "^fc00::/7" /tmp/bogonsv6 > /etc/bogonsv6

				RESULT=`/sbin/pfctl -t bogonsv6 -T replace -f /etc/bogonsv6 2>&1`

				echo "$RESULT" | awk '{ print "Bogons V6 file downloaded: " $0 }' | logger

			else

				echo "Not saving or updating IPv6 bogons (increase table-entries limit)" | logger

			fi

		else

			if [ $ENTRIES_MAX -gt $((2*ENTRIES_TOT+LINES_V6)) ]; then

				egrep -iv "^fc00::/7" /tmp/bogonsv6 > /etc/bogonsv6

				echo "Bogons V6 file downloaded but not updating IPv6 bogons table because it is not in use." | logger

			else

				echo "Not saving IPv6 bogons table (IPv6 Allow is off and table-entries limit is potentially too low)" | logger

			fi

		fi

		rm /tmp/bogonsv6

	else

		echo "Could not download ${v6url} (checksum mismatch)" | logger

		checksum_error="true"

	fi

	# We mounted RW, so switch back to RO

	/etc/rc.conf_mount_ro

fi

if [ "$checksum_error" != "" ]; then

	# Relaunch and sleep

	sh /etc/rc.update_bogons.sh &

	exit

fi

echo "rc.update_bogons.sh is ending the update cycle." | logger