simple-adblock.init (8693B)
1 #!/bin/sh /etc/rc.common 2 PKG_VERSION= 3 4 export START=94 5 export USE_PROCD=1 6 #PROCD_DEBUG=1 7 8 readonly A_TMP='/var/hosts.allowed.tmp' 9 readonly B_TMP='/var/hosts.blocked.tmp' 10 readonly T_TMP='/var/simple-adblock.hosts' 11 readonly dl='wget --no-check-certificate -qO-' 12 readonly h_filter='/localhost/d;/^#/d;/*/d;/^[^0-9]/d;s/^0\.0\.0\.0.//;s/^127\.0\.0\.1.//;s/[[:space:]]*#.*$//;s/[[:cntrl:]]$//;s/[[:space:]]//g;/^$/d;' 13 readonly d_filter='/localhost/d;/^#/d;/*/d;s/[[:space:]]*#.*$//;s/[[:space:]]*$//;s/[[:cntrl:]]$//;/[[:space:]]/d;/^</d;/^$/d;' 14 readonly f_filter='s|^|local=/|;s|$|/|' 15 readonly _ok_='\033[0;32m\xe2\x9c\x93\033[0m' 16 readonly _fail_='\033[0;31m\xe2\x9c\x97\033[0m' 17 readonly __ok__='\033[0;32m[\xe2\x9c\x93]\033[0m' 18 readonly __fail__='\033[0;31m[\xe2\x9c\x97]\033[0m' 19 readonly _error_='\033[0;31mERROR\033[0m' 20 21 export verbosity=2 force_dns=1 bgrun=0 hosts_file='/var/dnsmasq.d/simple-adblock' led wan_if wan_gw wanphysdev 22 23 ok() { case $verbosity in 1) output "$_ok_";; 2) output "$__ok__\n";; esac; } 24 okn() { case $verbosity in 1) output "$_ok_\n";; 2) output "$__ok__\n";; esac; } 25 fail() { case $verbosity in 1) output "$_fail_";; 2) output "$__fail__\n";; esac; } 26 failn() { case $verbosity in 1) output "$_fail_\n";; 2) output "$__fail__\n";; esac; } 27 output() { [[ $# -ne 1 ]] && { [[ ! $((verbosity & $1)) -gt 0 ]] && return 0 || shift; }; local msg; msg=$(echo -n "${1/$p_name /service }" | sed 's|\\033\[[0-9]\?;\?[0-9]\?[0-9]\?m||g'); [[ -t 1 ]] && echo -e -n "$1"; [[ $(echo -e -n "$msg" | wc -l) -gt 0 ]] && logger -t "${PKG_NAME:-service} [$$]" "$(echo -e -n ${logmsg}${msg})" && logmsg='' || logmsg=${logmsg}${msg}; } 28 PKG_NAME="${PKG_NAME:-simple-adblock}"; p_name="${PKG_NAME} ${PKG_VERSION}" 29 30 led_on(){ [[ -n "$led" && -e "$led/trigger" ]] && echo "default-on" > "$led/trigger"; } 31 led_off(){ [[ -n "$led" && -e "$led/trigger" ]] && echo "none" > "$led/trigger"; } 32 33 is_enabled () { 34 local c=1 enabled 35 config_load $PKG_NAME 36 config_get_bool enabled 'config' 'enabled' 1 37 config_get_bool bgrun 'config' 'run_in_background' 0 38 config_get_bool force_dns 'config' 'force_dns' 1 39 config_get verbosity 'config' 'verbosity' '2' 40 config_get hosts_file 'config' 'hosts_file' '/var/dnsmasq.d/simple-adblock' 41 config_get led 'config' 'led' 42 led="${led:+/sys/class/leds/$led}" 43 [[ $enabled -gt 0 ]] || { output "$_error_: $p_name is not enabled.\n"; return 1; } 44 source /lib/functions/network.sh 45 while : ; do 46 network_find_wan wan_if; [ -n "$wan_if" ] && network_get_gateway wan_gw $wan_if; 47 [[ $c -ge 25 || -n "$wan_gw" ]] && break 48 output "$p_name waiting for wan gateway...\n"; sleep 2; network_flush_cache; let "c+=1"; 49 done 50 [ -n "$wan_gw" ] && return 0 || { output "$_error_: $p_name failed to discover WAN gateway.\n"; return 1; } 51 } 52 53 reset_iptables() { 54 [[ $force_dns -eq 0 ]] && return 0 55 [ -z "$PKG_NAME" ] && return 1 56 iptables-save | grep -Fv -- "$PKG_NAME" | iptables-restore 57 lsmod | grep -q ip6table_nat && ip6tables-save | grep -Fv -- "$PKG_NAME" | ip6tables-restore 58 [ ! "$1" == "quiet" ] && output 'No longer forcing local DNS server.\n' 59 } 60 61 set_iptables() { 62 local ip ipv6 label ipv6wan brname 63 network_get_ipaddr ip lan; network_get_ipaddr6 ipv6 lan; network_get_device brname lan; network_get_physdev wanphysdev wan; 64 ipv6wan=$(ifconfig $wanphysdev | grep inet6 | awk '{print $3}') 65 66 if [[ $force_dns -ne 0 ]]; then 67 [ -n "$ip" ] && iptables -t nat -A prerouting_rule -i $brname -p tcp --dport 53 -j DNAT --to $ip -m comment --comment "$PKG_NAME" 68 [ -n "$ip" ] && iptables -t nat -A prerouting_rule -i $brname -p udp --dport 53 -j DNAT --to $ip -m comment --comment "$PKG_NAME" 69 if [[ -n "$ipv6" && -n "$ipv6wan" ]] && lsmod | grep -q ip6table_nat; then 70 ip6tables -t nat -A PREROUTING -i $brname -p tcp --dport 53 -j DNAT --to-destination [$ipv6] -m comment --comment "$PKG_NAME" 71 ip6tables -t nat -A PREROUTING -i $brname -p udp --dport 53 -j DNAT --to-destination [$ipv6] -m comment --comment "$PKG_NAME" 72 label="$ip/$ipv6" 73 else 74 label="$ip" 75 fi 76 [ -n "$label" ] && output "Forcing local DNS server: $label.\n" || output "$_error_: $p_name failed to obtain LAN IP address for DNS forcing!\n" 77 fi 78 } 79 80 stop_adblocking () { 81 [ -f $hosts_file ] && mv $hosts_file $T_TMP 82 output 3 "Restarting dnsmasq " 83 led_off 84 /etc/init.d/dnsmasq restart >/dev/null 2>&1 85 [[ $? -eq 0 ]] && { okn; output "$p_name stopped.\n"; } || { failn; output "$_error_: $p_name failed to reload dnsmasq!\n"; } 86 } 87 88 process_url() { 89 local label type D_TMP R_TMP 90 [[ -n "$1" && -n "$2" && -n "$3" ]] || return 1 91 local url=$1 92 [ "$2" == "hosts" ] && label="Hosts: $(echo $1 | cut -d'/' -f3)" filter="$h_filter" || label="Domains: $(echo $1 | cut -d'/' -f3)" filter="$d_filter" 93 [ "$3" == "blocked" ] && { type='Blocked'; D_TMP="$B_TMP"; } || { type='Allowed'; D_TMP="$A_TMP"; } 94 R_TMP="/var/simple-adblock_$(head /dev/urandom | tr -dc 'A-Za-z0-9' | head -c10)" 95 while [ -e "$R_TMP" ]; do R_TMP="/var/simple-adblock_$(head /dev/urandom | tr -dc 'A-Za-z0-9' | head -c10)"; done 96 touch "$R_TMP" 97 output 2 "[DL] $type $label " 98 $dl "${url}" > "$R_TMP" && ok || fail 99 { sed -i "$filter" "$R_TMP"; cat "$R_TMP" >> "$D_TMP"; rm -f "$R_TMP"; } & 100 } 101 102 start_adblocking () { 103 local whitelist_domains blacklist_domains whitelist_domains_urls blacklist_domains_urls blacklist_hosts_urls 104 config_get whitelist_domains 'config' 'whitelist_domain' 105 config_get blacklist_domains 'config' 'blacklist_domain' 106 config_get whitelist_domains_urls 'config' 'whitelist_domains_url' 107 config_get blacklist_domains_urls 'config' 'blacklist_domains_url' 108 config_get blacklist_hosts_urls 'config' 'blacklist_hosts_url' 109 110 local hf w_filter 111 112 [ ! -d ${hosts_file%/*} ] && mkdir -p ${hosts_file%/*} 113 if [[ -s $T_TMP && ! "$1" == "reload" ]]; then 114 output 3 'Found existing data file, reusing it ' 115 mv $T_TMP $hosts_file && okn || failn 116 else 117 [ -f $A_TMP ] && rm -f $A_TMP; [ -f $B_TMP ] && rm -f $B_TMP; [ -f $T_TMP ] && rm -f $T_TMP; [ -f $hosts_file ] && rm -f $hosts_file 118 touch $A_TMP; touch $B_TMP; touch $T_TMP; 119 120 if [ -n "$blacklist_hosts_urls" ]; then 121 output 1 '[DL] Blocked Hosts ' 122 for hf in ${blacklist_hosts_urls}; do process_url "$hf" 'hosts' 'blocked'; done 123 output 1 '\n' 124 fi 125 126 if [ -n "$blacklist_domains_urls" ]; then 127 output 1 '[DL] Blocked Domains ' 128 for hf in ${blacklist_domains_urls}; do process_url "$hf" 'domains' 'blocked'; done 129 output 1 '\n' 130 fi 131 132 if [ -n "$whitelist_domains_urls" ]; then 133 output 1 '[DL] Allowed Domains ' 134 for hf in ${whitelist_domains_urls}; do process_url "$hf" 'domains' 'allowed'; done 135 output 1 '\n' 136 fi 137 output 3 'Waiting for background processes ' 138 wait && okn 139 140 [ -n "$blacklist_domains" ] && for hf in ${blacklist_domains}; do echo "$hf" | sed "$d_filter" >> $B_TMP; done 141 whitelist_domains="${whitelist_domains}"$'\n'"$(cat $A_TMP)" 142 [ -n "$whitelist_domains" ] && for hf in ${whitelist_domains}; do hf=$(echo $hf | sed 's/\./\\./g'); w_filter="$w_filter/^${hf}$/d;/\\.${hf}$/d;"; done 143 144 if [ -s $B_TMP ]; then 145 output 1 'Processing downloads ' 146 output 2 'Sorting merged file '; sort $B_TMP | uniq > $T_TMP && ok || fail 147 output 2 'Whitelisting domains '; sed -i "$w_filter" $T_TMP && ok || fail 148 output 2 'Formatting merged file '; sed "$f_filter" $T_TMP > $hosts_file && ok || fail 149 output 1 '\n' 150 151 output 3 'Removing temporary files ' 152 [ -f $A_TMP ] && rm -f $A_TMP; [ -f $B_TMP ] && rm -f $B_TMP; [ -f $T_TMP ] && rm -f $T_TMP; 153 okn 154 fi 155 fi 156 157 if [ -s $hosts_file ]; then 158 output 3 'Restarting dnsmasq ' 159 /etc/init.d/dnsmasq restart >/dev/null 2>&1 160 if [[ $? -eq 0 ]]; then 161 led_on; okn; 162 output "$p_name blocking $(wc -l < $hosts_file) domains $_ok_\n" 163 else 164 failn; output "$_error_: $p_name failed to reload dnsmasq!\n"; 165 exit 1 166 fi 167 else 168 output "$_error_: $p_name failed to create its data file!\n" 169 exit 1 170 fi 171 } 172 173 boot() { sleep 10; rc_procd start_service; rc_procd service_triggers; } 174 175 reload_service () { 176 is_enabled || return 1 177 [[ -t 1 && $bgrun -eq 1 ]] && (start_adblocking 'reload' | cat &) || start_adblocking 'reload' 178 } 179 180 start_service () { 181 is_enabled || return 1 182 183 procd_open_instance 184 procd_set_param stdout 1 185 procd_set_param stderr 1 186 procd_close_instance 187 188 reset_iptables quiet 189 set_iptables 190 [[ -t 1 && $bgrun -eq 1 ]] && (start_adblocking $1 | cat &) || start_adblocking $1 191 } 192 193 stop_service () { 194 is_enabled || return 1 195 reset_iptables 196 [[ -t 1 && $bgrun -eq 1 ]] && (stop_adblocking | cat &) || stop_adblocking 197 } 198 199 service_triggers () { 200 procd_add_reload_trigger 'simple-adblock' 201 }