mirror of
https://github.com/pi-hole/pi-hole
synced 2024-12-22 23:08:07 +00:00
Merge pull request #61 from nathanpjones/improve-whitelist-script
Improve whitelisting script
This commit is contained in:
commit
271163ed66
@ -1,13 +1,44 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
|
if [ $# = 0 ]; then
|
||||||
|
echo "Immediately whitelists one or more domains."
|
||||||
|
echo "Usage: whitelist.sh domain1 [domain2 ...]"
|
||||||
|
fi
|
||||||
|
|
||||||
|
combopattern=""
|
||||||
|
|
||||||
# For each argument passed to this script
|
# For each argument passed to this script
|
||||||
for var in "$@"
|
for var in "$@"
|
||||||
do
|
do
|
||||||
echo "Whitelisting $var..."
|
echo "Whitelisting $var..."
|
||||||
# Use sed to search for the domain in /etc/pihole/gravity.list and remove it using an in-place edit
|
|
||||||
sed -i "/$var/d" /etc/pihole/gravity.list
|
# Construct basic pattern to match domain name.
|
||||||
# Also add the domain to the whitelist.txt in /etc/pihole
|
basicpattern=$(echo $var | awk -F '[# \t]' 'NF>0&&$1!="" {print ""$1""}' | sed 's/\./\\./g')
|
||||||
echo "$var" >> /etc/pihole/whitelist.txt
|
|
||||||
|
if [ "$basicpattern" != "" ]; then
|
||||||
|
# Add to the combination pattern that will be used below
|
||||||
|
if [ "$combopattern" != "" ]; then combopattern="$combopattern|"; fi
|
||||||
|
combopattern="$combopattern$basicpattern"
|
||||||
|
|
||||||
|
# Also add the domain to the whitelist but only if it's not already present
|
||||||
|
grep -E -q "^$basicpattern$" /etc/pihole/whitelist.txt \
|
||||||
|
|| echo "$var" >> /etc/pihole/whitelist.txt
|
||||||
|
fi
|
||||||
done
|
done
|
||||||
echo "** $# domain(s) whitelisted."
|
|
||||||
# Force dnsmasq to reload /etc/pihole/gravity.list
|
# Now report on and remove matched domains
|
||||||
kill -HUP $(pidof dnsmasq)
|
if [ "$combopattern" != "" ]; then
|
||||||
|
echo "Modifying hosts file..."
|
||||||
|
|
||||||
|
# Construct pattern to match entry in hosts file.
|
||||||
|
# This consists of one or more IP addresses followed by the domain name.
|
||||||
|
pattern=$(echo $combopattern | awk -F '[# \t]' '{printf "%s", "^(([0-9]+\.){3}[0-9]+ +)+("$1")$"}')
|
||||||
|
|
||||||
|
# Output what will be removed and then actually remove
|
||||||
|
sed -r -n 's/'"$pattern"'/ Removed: \3/p' /etc/pihole/gravity.list
|
||||||
|
sed -r -i '/'"$pattern"'/d' /etc/pihole/gravity.list
|
||||||
|
|
||||||
|
echo "** $# domain(s) whitelisted."
|
||||||
|
# Force dnsmasq to reload /etc/pihole/gravity.list
|
||||||
|
kill -HUP $(pidof dnsmasq)
|
||||||
|
fi
|
||||||
|
Loading…
Reference in New Issue
Block a user