mirror of
https://github.com/pi-hole/pi-hole.git
synced 2024-12-01 02:23:21 +00:00
146 lines
4.5 KiB
Bash
Executable file
146 lines
4.5 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# Pi-hole: A black hole for Internet advertisements
|
|
# (c) 2015, 2016 by Jacob Salmela
|
|
# Network-wide ad blocking via your Raspberry Pi
|
|
# http://pi-hole.net
|
|
# Calculates stats and displays to an LCD
|
|
#
|
|
# Pi-hole 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, either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
source /usr/local/include/pihole/piholeInclude
|
|
|
|
rerun_pihole "$0" "$@"
|
|
|
|
#Functions##############################################################################################################
|
|
piLog="/var/log/pihole.log"
|
|
gravity="/etc/pihole/gravity.list"
|
|
|
|
today=$(date "+%b %e")
|
|
|
|
function CalcBlockedDomains(){
|
|
CheckIPv6
|
|
if [ -e "$gravity" ]; then
|
|
#Are we IPV6 or IPV4?
|
|
if [[ -n $piholeIPv6 ]];then
|
|
#We are IPV6
|
|
blockedDomainsTotal=$(wc -l /etc/pihole/gravity.list | awk '{print $1/2}')
|
|
else
|
|
#We are IPV4
|
|
blockedDomainsTotal=$(wc -l /etc/pihole/gravity.list | awk '{print $1}')
|
|
fi
|
|
else
|
|
blockedDomainsTotal="Err."
|
|
fi
|
|
}
|
|
|
|
function CalcQueriesToday(){
|
|
if [ -e "$piLog" ];then
|
|
queriesToday=$(cat "$piLog" | grep "$today" | awk '/query/ {print $6}' | wc -l)
|
|
else
|
|
queriesToday="Err."
|
|
fi
|
|
}
|
|
|
|
function CalcblockedToday(){
|
|
if [ -e "$piLog" ] && [ -e "$gravity" ];then
|
|
blockedToday=$(cat $piLog | awk '/\/etc\/pihole\/gravity.list/ && !/address/ {print $6}' | wc -l)
|
|
else
|
|
blockedToday="Err."
|
|
fi
|
|
}
|
|
|
|
function CalcPercentBlockedToday(){
|
|
if [ "$queriesToday" != "Err." ] && [ "$blockedToday" != "Err." ]; then
|
|
if [ "$queriesToday" != 0 ]; then #Fixes divide by zero error :)
|
|
#scale 2 rounds the number down, so we'll do scale 4 and then trim the last 2 zeros
|
|
percentBlockedToday=$(echo "scale=4; $blockedToday/$queriesToday*100" | bc)
|
|
percentBlockedToday=$(sed 's/.\{2\}$//' <<< "$percentBlockedToday")
|
|
else
|
|
percentBlockedToday=0
|
|
fi
|
|
fi
|
|
}
|
|
|
|
function CheckIPv6(){
|
|
piholeIPv6file="/etc/pihole/.useIPv6"
|
|
if [[ -f $piholeIPv6file ]];then
|
|
# If the file exists, then the user previously chose to use IPv6 in the automated installer
|
|
piholeIPv6=$(ip -6 route get 2001:4860:4860::8888 | awk -F " " '{ for(i=1;i<=NF;i++) if ($i == "src") print $(i+1) }')
|
|
fi
|
|
}
|
|
|
|
function outputJSON(){
|
|
CalcQueriesToday
|
|
CalcblockedToday
|
|
CalcPercentBlockedToday
|
|
|
|
CalcBlockedDomains
|
|
|
|
printf '{"domains_being_blocked":"%s","dns_queries_today":"%s","ads_blocked_today":"%s","ads_percentage_today":"%s"}\n' "$blockedDomainsTotal" "$queriesToday" "$blockedToday" "$percentBlockedToday"
|
|
}
|
|
|
|
function normalChrono(){
|
|
for (( ; ; ))
|
|
do
|
|
clear
|
|
# Displays a colorful Pi-hole logo
|
|
toilet -f small -F gay Pi-hole
|
|
echo " $(ifconfig eth0 | awk '/inet addr/ {print $2}' | cut -d':' -f2)"
|
|
echo ""
|
|
uptime | cut -d' ' -f11-
|
|
uptime -p
|
|
echo "-------------------------------"
|
|
# Uncomment to continually read the log file and display the current domain being blocked
|
|
#tail -f /var/log/pihole.log | awk '/\/etc\/pihole\/gravity.list/ {if ($7 != "address" && $7 != "name" && $7 != "/etc/pihole/gravity.list") print $7; else;}'
|
|
|
|
#uncomment next 4 lines to use original query count calculation
|
|
#today=$(date "+%b %e")
|
|
#todaysQueryCount=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ {print $7}' | wc -l)
|
|
#todaysQueryCountV4=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ && /\[A\]/ {print $7}' | wc -l)
|
|
#todaysQueryCountV6=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ && /\[AAAA\]/ {print $7}' | wc -l)
|
|
|
|
|
|
CalcQueriesToday
|
|
CalcblockedToday
|
|
CalcPercentBlockedToday
|
|
|
|
CalcBlockedDomains
|
|
|
|
echo "Blocking: $blockedDomainsTotal"
|
|
#below commented line does not add up to todaysQueryCount
|
|
#echo "Queries: $todaysQueryCountV4 / $todaysQueryCountV6"
|
|
echo "Queries: $queriesToday" #same total calculation as dashboard
|
|
echo "Pi-holed: $blockedToday ($percentBlockedToday%)"
|
|
|
|
sleep 5
|
|
done
|
|
}
|
|
|
|
function displayHelp(){
|
|
echo "Displays stats about your piHole!"
|
|
echo " "
|
|
echo "Usage: chronometer.sh [optional:-j]"
|
|
echo "Note: If no option is passed, then stats are displayed on screen, updated every 5 seconds"
|
|
echo " "
|
|
echo "Options:"
|
|
echo " -j, --json output stats as JSON formatted string"
|
|
echo " -h, --help display this help text"
|
|
|
|
exit 1
|
|
}
|
|
|
|
if [[ $# = 0 ]]; then
|
|
normalChrono
|
|
fi
|
|
|
|
for var in "$@"
|
|
do
|
|
case "$var" in
|
|
"-j" | "--json" ) outputJSON;;
|
|
"-h" | "--help" ) displayHelp;;
|
|
* ) exit 1;;
|
|
esac
|
|
done
|