Modded Tablspn Script

#!/bin/bash
## Adblocker.sh - Based on Adblocker.sh by Todd Stein ([email protected]), Saturday, October 25, 2014
##                Modifiyed by Mathew Evans ([email protected]), Monday, August 31, 2015
##                for use with Bind9 on *Nix setup
##
#
#  Requirements:
#  *nix & Bind9
#
#  Add to named.conf.local:
#  include "/etc/bind/blacklist.zones";
#
#
#  ToDo:
#  - Cronjob support.



readBlkList()
{
        while read p; do
                set -- "$p"
                IFS=" "; declare -a Array=($*)
                hostname=`echo "${Array[1]}" | sed 's/\\r//g'`
                appendBlackList $hostname
        done </tmp/adblocker_hostlist
}

appendBlackList()
{
        if [ $(hostExists $1) -le 0 ]; then
#               echo ">>New blocked host added: $1";
                echo "zone \"$1\" {type master; file \"/etc/bind/db.blocked\";};" >> /etc/bind/blacklist.zones
        fi
}

hostExists()
{
        res=$(find /etc/bind -type f -name "blacklist.zones" | xargs egrep -i '('$1')');
        if [ ${#res} -ge 1 ]; then echo 1  # Found
        else echo 0                        # Not Found
        fi
}

BuildBlockList()
{
HOST_LISTS="
        http://adaway.org/hosts.txt
        http://www.malwaredomainlist.com/hostslist/hosts.txt
        http://www.mvps.org/winhelp2002/hosts.txt
        http://pgl.yoyo.org/adservers/serverlist.php?hostformat=hosts&showintro=0&startdate%5Bday%5D=&startdate%5Bmonth%5D=&star
"
BLOCKLIST=/tmp/adblocker_hostlist
BLACKLIST=/etc/adblocker_blacklist
WHITELIST=/etc/adblocker_whitelist

# await internet connectivity before proceeding (in case rc.local executes this script before connectivity is achieved)
until ping -c1 -w3 google.com || ping -c1 -w3 yahoo.com; do
        sleep 5
done &>/dev/null

# initialize block list
>"$BLOCKLIST"

# grab blacklisted domains if any have been specified
[ -s "$BLACKLIST" ] && awk '/^[^#]/ { print "0.0.0.0",$1 }' "$BLACKLIST" >>"$BLOCKLIST"

# grab host lists from the internet
IP_REGEX='([0-9]{1,3}\.){3}[0-9]{1,3}'
wget -qO- $HOST_LISTS | sed -rn "s/^$IP_REGEX\W/0.0.0.0 /p" | awk '{ print $1,$2 }' | sort -uk2 >>"$BLOCKLIST"

if [ -s "$WHITELIST" ]; then
        white_listed_regex=`echo \`grep -o '^[^#]\+' "$WHITELIST"\` | tr ' ' '|'`
        sed -ri "/$white_listed_regex/d" "$BLOCKLIST"
fi
}

AddToCron()
{
echo "todo: CronJobs support";
#SCRIPT_NAME="/etc/bind/adblocker.sh";
#DELAY='$(head /dev/urandom | wc -c | /usr/bin/awk "{ print \$0 % 30 }")m'
#grep -Fq "$SCRIPT_NAME" /etc/crontabs/root 2>/dev/null || cat >>/etc/crontabs/root <<-:EOF:
        # Download updated ad and malware server lists every Tuesday between 3:00 and 3:30 AM
#       0 3 * * 2 /bin/sleep $DELAY && $SCRIPT_NAME
#:EOF:
}


## make sure added to cronjobs to autorun
AddToCron

## Download updated blocklist
BuildBlockList

## Read downloaded blocklist and add new to bind list
readBlkList

##Retart Bind to update to new blocked list
service bind9 restart