#!/bin/sh /etc/rc.common # Copyright 2017-2020 Stan Grishin (stangri@melmac.net) # shellcheck disable=SC2039,SC1091,SC2016 PKG_VERSION='dev-test' # shellcheck disable=SC2034 START=94 # shellcheck disable=SC2034 USE_PROCD=1 LC_ALL=C extra_command 'check' 'Checks if specified domain is found in current block-list' extra_command 'dl' 'Force-downloads all enabled block-list' extra_command 'killcache' 'Delete all cached files' extra_command 'sizes' 'Displays the file-sizes of enabled block-listo' extra_command 'show' 'Shows the service last-run status' extra_command 'version' 'Show version information' readonly packageName='simple-adblock' readonly serviceName="$packageName $PKG_VERSION" readonly addnhostsFile="/var/run/${packageName}.addnhosts" readonly addnhostsCache="/var/run/${packageName}.addnhosts.cache" readonly addnhostsGzip="/etc/${packageName}.addnhosts.gz" readonly addnhostsOutputFilter='s|^|127.0.0.1 |;s|$||' readonly addnhostsOutputFilterIPv6='s|^|:: |;s|$||' readonly dnsmasqFile="/var/dnsmasq.d/${packageName}" readonly dnsmasqCache="/var/run/${packageName}.dnsmasq.cache" readonly dnsmasqGzip="/etc/${packageName}.dnsmasq.gz" readonly dnsmasqOutputFilter='s|^|local=/|;s|$|/|' readonly ipsetFile="/var/dnsmasq.d/${packageName}.ipset" readonly ipsetCache="/var/run/${packageName}.ipset.cache" readonly ipsetGzip="/etc/${packageName}.ipset.gz" readonly ipsetOutputFilter='s|^|ipset=/|;s|$|/adb|' readonly serversFile="/var/run/${packageName}.servers" readonly serversCache="/var/run/${packageName}.servers.cache" readonly serversGzip="/etc/${packageName}.servers.gz" readonly serversOutputFilter='s|^|server=/|;s|$|/|' readonly unboundFile="/var/lib/unbound/adb_list.${packageName}" readonly unboundCache="/var/run/${packageName}.unbound.cache" readonly unboundGzip="/etc/${packageName}.unbound.gz" readonly unboundOutputFilter='s|^|local-zone: "|;s|$|" static|' readonly A_TMP="/var/${packageName}.hosts.a.tmp" readonly B_TMP="/var/${packageName}.hosts.b.tmp" readonly jsonFile="/var/run/${packageName}.json" readonly sharedMemoryError="/dev/shm/$packageName-error" readonly sharedMemoryOutput="/dev/shm/$packageName-output" readonly hostsFilter='/localhost/d;/^#/d;/^[^0-9]/d;s/^0\.0\.0\.0.//;s/^127\.0\.0\.1.//;s/[[:space:]]*#.*$//;s/[[:cntrl:]]$//;s/[[:space:]]//g;/[`~!@#\$%\^&\*()=+;:"'\'',<>?/\|[{}]/d;/]/d;/\./!d;/^$/d;/[^[:alnum:]_.-]/d;' readonly domainsFilter='/^#/d;s/[[:space:]]*#.*$//;s/[[:space:]]*$//;s/[[:cntrl:]]$//;/[[:space:]]/d;/[`~!@#\$%\^&\*()=+;:"'\'',<>?/\|[{}]/d;/]/d;/\./!d;/^$/d;/[^[:alnum:]_.-]/d;' readonly _OK_='\033[0;32m\xe2\x9c\x93\033[0m' readonly _FAIL_='\033[0;31m\xe2\x9c\x97\033[0m' readonly __OK__='\033[0;32m[\xe2\x9c\x93]\033[0m' readonly __FAIL__='\033[0;31m[\xe2\x9c\x97]\033[0m' readonly _ERROR_='\033[0;31mERROR\033[0m' readonly _WARNING_='\033[0;33mWARNING\033[0m' version() { echo "$PKG_VERSION"; } getStatusText() { local _ret case "$1" in statusNoInstall) _ret="$serviceName is not installed or not found";; statusStopped) _ret="Stopped";; statusStarting) _ret="Starting";; statusRestarting) _ret="Restarting";; statusForceReloading) _ret="Force Reloading";; statusDownloading) _ret="Downloading";; statusProcessing) _ret="Processing";; statusError) _ret="Error";; statusWarning) _ret="Warning";; statusFail) _ret="Fail";; statusSuccess) _ret="Success";; esac printf "%b" "$_ret" } getErrorText() { local _ret case "$1" in errorOutputFileCreate) _ret="failed to create $outputFile file";; errorFailDNSReload) _ret="failed to restart/reload DNS resolver";; errorSharedMemory) _ret="failed to access shared memory";; errorSorting) _ret="failed to sort data file";; errorOptimization) _ret="failed to optimize data file";; errorAllowListProcessing) _ret="failed to process allow-list";; errorDataFileFormatting) _ret="failed to format data file";; errorMovingDataFile) _ret="failed to move data file '${A_TMP}' to '${outputFile}'";; errorCreatingCompressedCache) _ret="failed to create compressed cache";; errorRemovingTempFiles) _ret="failed to remove temporary files";; errorRestoreCompressedCache) _ret="failed to unpack compressed cache";; errorRestoreCache) _ret="failed to move '$outputCache' to '$outputFile'";; errorOhSnap) _ret="failed to create block-list or restart DNS resolver";; errorStopping) _ret="failed to stop $serviceName";; errorDNSReload) _ret="failed to reload/restart DNS resolver";; errorDownloadingConfigUpdate) _ret="failed to download Config Update file";; errorDownloadingList) _ret="failed to download";; errorParsingList) _ret="failed to parse";; errorParsingConfigUpdate) _ret="failed to parse Config Update file";; esac printf "%b" "$_ret" } output_ok() { output 1 "$_OK_"; output 2 "$__OK__\\n"; } output_okn() { output 1 "$_OK_\\n"; output 2 "$__OK__\\n"; } output_fail() { output 1 "$_FAIL_"; output 2 "$__FAIL__\\n"; } output_failn() { output 1 "$_FAIL_\\n"; output 2 "$__FAIL__\\n"; } str_replace() { printf "%b" "$1" | sed -e "s/$(printf "%b" "$2")/$(printf "%b" "$3")/g"; } str_contains() { test "$1" != "$(str_replace "$1" "$2" '')"; } compare_values() { test "$(printf '%s\n' "$@" | sort -V | head -n 1)" != "$1"; } is_chaos_calmer() { ubus -S call system board | grep -q 'Chaos Calmer'; } led_on(){ if [ -n "${1}" ] && [ -e "${1}/trigger" ]; then echo 'default-on' > "${1}/trigger" 2>&1; fi; } led_off(){ if [ -n "${1}" ] && [ -e "${1}/trigger" ]; then echo 'none' > "${1}/trigger" 2>&1; fi; } dnsmasq_hup() { killall -q -HUP dnsmasq; } dnsmasq_kill() { killall -q -KILL dnsmasq; } dnsmasq_restart() { /etc/init.d/dnsmasq restart >/dev/null 2>&1; } unbound_restart() { /etc/init.d/unbound restart >/dev/null 2>&1; } is_force_dns_active() { iptables-save | grep -q -w -- '--dport 53'; } is_present() { command -v "$1" >/dev/null 2>&1; } output() { # Can take a single parameter (text) to be output at any verbosity # Or target verbosity level and text to be output at specifc verbosity local msg memmsg logmsg if [ $# -ne 1 ]; then if [ $((verbosity & $1)) -gt 0 ] || [ "$verbosity" = "$1" ]; then shift; else return 0; fi fi [ -t 1 ] && printf "%b" "$1" msg="${1//$serviceName /service }"; if [ "$(printf "%b" "$msg" | wc -l)" -gt 0 ]; then [ -s "$sharedMemoryOutput" ] && memmsg="$(cat "$sharedMemoryOutput")" logmsg="$(printf "%b" "${memmsg}${msg}" | sed 's/\x1b\[[0-9;]*m//g')" logger -t "${packageName:-service} [$$]" "$(printf "%b" "$logmsg")" rm -f "$sharedMemoryOutput" else printf "%b" "$msg" >> "$sharedMemoryOutput" fi } serviceEnabled=''; forceDNS=''; forceDNSPorts=''; parallelDL=''; debug=''; compressedCache=''; ipv6Enabled=''; configUpdateEnabled=''; configUpdateURL=''; bootDelay=''; dlTimeout=''; curlRetry=''; verbosity=''; led=''; targetDNS=''; dnsInstance=''; allowed_domains=''; allowed_domains_urls=''; blocked_domains=''; blocked_domains_urls=''; blocked_hosts_urls=''; dl_command=''; dl_flag=''; isSSLSupported=''; allowIDN=''; outputFilter=''; outputFilterIPv6=''; outputFile=''; outputGzip=''; outputCache=''; awk='awk'; load_package_config() { config_load "$packageName" config_get_bool serviceEnabled 'config' 'enabled' 1 config_get_bool forceDNS 'config' 'force_dns' 1 config_get_bool parallelDL 'config' 'parallel_downloads' 1 config_get_bool debug 'config' 'debug' 0 config_get_bool compressedCache 'config' 'compressed_cache' 0 config_get_bool ipv6Enabled 'config' 'ipv6_enabled' 0 config_get_bool configUpdateEnabled 'config' 'config_update_enabled' 0 config_get forceDNSPorts 'config' 'force_dns_port' '53 853' config_get bootDelay 'config' 'boot_delay' '120' config_get dlTimeout 'config' 'download_timeout' '20' config_get curlRetry 'config' 'curl_retry' '3' config_get verbosity 'config' 'verbosity' '2' config_get led 'config' 'led' config_get targetDNS 'config' 'dns' 'dnsmasq.servers' config_get dnsInstance 'config' 'dns_instance' '0' config_get allowed_domains 'config' 'allowed_domain' config_get allowed_domains_urls 'config' 'allowed_domains_url' config_get blocked_domains 'config' 'blocked_domain' config_get blocked_domains_urls 'config' 'blocked_domains_url' config_get blocked_hosts_urls 'config' 'blocked_hosts_url' config_get configUpdateURL 'config' 'config_update_url' 'https://cdn.jsdelivr.net/gh/openwrt/packages/net/simple-adblock/files/simple-adblock.conf.update' if [ "$targetDNS" != 'dnsmasq.addnhosts' ] && [ "$targetDNS" != 'dnsmasq.conf' ] && \ [ "$targetDNS" != 'dnsmasq.servers' ] && [ "$targetDNS" != 'unbound.adb_list' ] && \ [ "$targetDNS" != 'dnsmasq.ipset' ] ; then targetDNS='dnsmasq.servers' fi case "$targetDNS" in dnsmasq.addnhosts) outputFilter="$addnhostsOutputFilter" outputFile="$addnhostsFile" outputCache="$addnhostsCache" outputGzip="$addnhostsGzip" [ "$ipv6Enabled" -gt 0 ] && outputFilterIPv6="$addnhostsOutputFilterIPv6" rm -f "$dnsmasqFile" "$dnsmasqCache" "$dnsmasqGzip" rm -f "$ipsetFile" "$ipsetCache" "$ipsetGzip" rm -f "$serversFile" "$serversCache" "$serversGzip" rm -f "$unboundFile" "$unboundCache" "$unboundGzip" ;; dnsmasq.conf) outputFilter="$dnsmasqOutputFilter" outputFile="$dnsmasqFile" outputCache="$dnsmasqCache" outputGzip="$dnsmasqGzip" rm -f "$addnhostsFile" "$addnhostsCache" "$addnhostsGzip" rm -f "$ipsetFile" "$ipsetCache" "$ipsetGzip" rm -f "$serversFile" "$serversCache" "$serversGzip" rm -f "$unboundFile" "$unboundCache" "$unboundGzip" ;; dnsmasq.ipset) outputFilter="$ipsetOutputFilter" outputFile="$ipsetFile" outputCache="$ipsetCache" outputGzip="$ipsetGzip" rm -f "$dnsmasqFile" "$dnsmasqCache" "$dnsmasqGzip" rm -f "$addnhostsFile" "$addnhostsCache" "$addnhostsGzip" rm -f "$serversFile" "$serversCache" "$serversGzip" rm -f "$unboundFile" "$unboundCache" "$unboundGzip" ;; dnsmasq.servers) outputFilter="$serversOutputFilter" outputFile="$serversFile" outputCache="$serversCache" outputGzip="$serversGzip" rm -f "$dnsmasqFile" "$dnsmasqCache" "$dnsmasqGzip" rm -f "$addnhostsFile" "$addnhostsCache" "$addnhostsGzip" rm -f "$ipsetFile" "$ipsetCache" "$ipsetGzip" rm -f "$unboundFile" "$unboundCache" "$unboundGzip" ;; unbound.adb_list) outputFilter="$unboundOutputFilter" outputFile="$unboundFile" outputCache="$unboundCache" outputGzip="$unboundGzip" rm -f "$addnhostsFile" "$addnhostsCache" "$addnhostsGzip" rm -f "$dnsmasqFile" "$dnsmasqCache" "$dnsmasqGzip" rm -f "$ipsetFile" "$ipsetCache" "$ipsetGzip" rm -f "$serversFile" "$serversCache" "$serversGzip" ;; esac if [ -z "${verbosity##*[!0-9]*}" ] || [ "$verbosity" -lt 0 ] || [ "$verbosity" -gt 2 ]; then verbosity=1 fi . /lib/functions/network.sh . /usr/share/libubox/jshn.sh is_present 'gawk' && awk='gawk' if ! is_present '/usr/libexec/grep-gnu' || ! is_present '/usr/libexec/sed-gnu' || \ ! is_present '/usr/libexec/sort-coreutils' || ! is_present 'gawk'; then local s="opkg update; opkg --force-overwrite install" is_present 'gawk' || s="$s gawk" is_present '/usr/libexec/grep-gnu' || s="$s grep" is_present '/usr/libexec/sed-gnu' || s="$s sed" is_present '/usr/libexec/sort-coreutils' || s="$s coreutils-sort" output "$_WARNING_: Some recommended packages are missing, install them by running:\\n" output "$s;\\n" fi # Prefer curl because it supports the file:// scheme. if is_present 'curl'; then dl_command="curl --insecure --retry $curlRetry --connect-timeout $dlTimeout --silent" dl_flag="-o" elif is_present '/usr/libexec/wget-ssl'; then dl_command="/usr/libexec/wget-ssl --no-check-certificate --timeout $dlTimeout -q" dl_flag="-O" elif is_present wget && wget --version 2>/dev/null | grep -q "+https"; then dl_command="wget --no-check-certificate --timeout $dlTimeout -q" dl_flag="-O" else dl_command="uclient-fetch --no-check-certificate --timeout $dlTimeout -q" dl_flag="-O" fi led="${led:+/sys/class/leds/$led}" if curl --version 2>/dev/null | grep -q "https" \ || wget --version 2>/dev/null | grep -q "+https" \ || grep -q "libustream-mbedtls" /usr/lib/opkg/status \ || grep -q "libustream-openssl" /usr/lib/opkg/status \ || grep -q "libustream-wolfssl" /usr/lib/opkg/status; then isSSLSupported=1 else unset isSSLSupported fi } is_enabled() { wan_if=''; wan_gw=''; load_package_config if [ "$debug" -ne 0 ]; then exec 1>>/tmp/simple-adblock.log exec 2>&1 set -x fi if [ "$serviceEnabled" -eq 0 ]; then case "$1" in on_start) output "$packageName is currently disabled.\\n" output "Run the following commands before starting service again:\\n" output "uci set ${packageName}.config.enabled='1'; uci commit $packageName;\\n" ;; esac return 1 fi case $targetDNS in dnsmasq.addnhosts | dnsmasq.conf | dnsmasq.ipset | dnsmasq.servers) if dnsmasq -v 2>/dev/null | grep -q 'no-IDN' || ! dnsmasq -v 2>/dev/null | grep -q -w 'IDN'; then allowIDN=0 else allowIDN=1 fi ;; unbound.adb_list) allowIDN=1;; esac case $targetDNS in dnsmasq.ipset) if dnsmasq -v 2>/dev/null | grep -q 'no-ipset' || ! dnsmasq -v 2>/dev/null | grep -q -w 'ipset'; then output "$_ERROR_: DNSMASQ ipset support is enabled in $packageName, but DNSMASQ is either not installed or installed DNSMASQ does not support ipsets!\\n" targetDNS='dnsmasq.servers' fi if ! ipset help hash:net >/dev/null 2>&1; then output "$_ERROR_: DNSMASQ ipset support is enabled in $packageName, but ipset is either not installed or installed ipset does not support 'hash:net' type!\\n" targetDNS='dnsmasq.servers' fi ;; esac [ ! -d "${outputFile%/*}" ] && mkdir -p "${outputFile%/*}" [ ! -d "${outputCache%/*}" ] && mkdir -p "${outputFile%/*}" [ ! -d "${outputGzip%/*}" ] && mkdir -p "${outputFile%/*}" cacheOps 'testGzip' && return 0 network_flush_cache; network_find_wan wan_if; network_get_gateway wan_gw "$wan_if"; [ -n "$wan_gw" ] && return 0 output "$_ERROR_: $serviceName failed to discover WAN gateway.\\n"; return 1; } dnsmasqOps() { local cfg="$1" param="$2" case "$param" in dnsmasq.addnhosts) if [ "$(uci -q get dhcp."$cfg".serversfile)" = "$serversFile" ]; then uci -q del dhcp."$cfg".serversfile fi if ! uci -q get dhcp."$cfg".addnhosts | grep -q "$addnhostsFile"; then uci add_list dhcp."$cfg".addnhosts="$addnhostsFile" fi ;; dnsmasq.conf|dnsmasq.ipset|unbound.adb_list|cleanup) uci -q del_list dhcp."$cfg".addnhosts="$addnhostsFile" if [ "$(uci -q get dhcp."$cfg".serversfile)" = "$serversFile" ]; then uci -q del dhcp."$cfg".serversfile fi ;; dnsmasq.servers) uci -q del_list dhcp."$cfg".addnhosts="$addnhostsFile" if [ "$(uci -q get dhcp."$cfg".serversfile)" != "$serversFile" ]; then uci set dhcp."$cfg".serversfile="$serversFile" fi ;; esac } dnsOps() { local param output_text i case $1 in on_start) if [ ! -s "$outputFile" ]; then jsonOps set status "statusFail" jsonOps add error "errorOutputFileCreate" output "$_ERROR_: $(getErrorText 'errorOutputFileCreate')!\\n" return 1 fi config_load 'dhcp' if [ "$dnsInstance" = "*" ]; then config_foreach dnsmasqOps 'dnsmasq' "$targetDNS" elif [ -n "$dnsInstance" ]; then for i in $dnsInstance; do dnsmasqOps "@dnsmasq[$i]" "$targetDNS" done fi case "$targetDNS" in dnsmasq.addnhosts|dnsmasq.servers) param=dnsmasq_hup output_text='Reloading DNSMASQ' ;; dnsmasq.conf|dnsmasq.ipset) param=dnsmasq_restart output_text='Restarting DNSMASQ' ;; unbound.adb_list) param=unbound_restart output_text='Restarting Unbound' ;; esac if [ -n "$(uci changes dhcp)" ]; then uci commit dhcp if [ "$param" = 'unbound_restart' ]; then param='dnsmasq_restart; unbound_restart;' output_text='Restarting Unbound/DNSMASQ' else param=dnsmasq_restart output_text='Restarting DNSMASQ' fi fi output 1 "$output_text " output 2 "$output_text " jsonOps set message "$output_text" if eval "$param"; then jsonOps set status "statusSuccess" led_on "$led" output_okn else output_fail jsonOps set status "statusFail" jsonOps add error "errorDNSReload" output "$_ERROR_: $(getErrorText 'errorDNSReload')!\\n" return 1 fi ;; on_stop) case "$targetDNS" in dnsmasq.addnhosts | dnsmasq.servers) param=dnsmasq_hup ;; dnsmasq.conf | dnsmasq.ipset) param=dnsmasq_restart ;; unbound.adb_list) param=unbound_restart ;; esac if [ -n "$(uci changes dhcp)" ]; then uci -q commit dhcp if [ "$param" = 'unbound_restart' ]; then param='dnsmasq_restart; unbound_restart;' else param=dnsmasq_restart fi fi eval "$param" return $? ;; quiet) case "$targetDNS" in dnsmasq.addnhosts | dnsmasq.conf | dnsmasq.ipset | dnsmasq.servers) param=dnsmasq_restart ;; unbound.adb_list) param=unbound_restart ;; esac eval "$param" return $? ;; esac } jsonOps() { # shellcheck disable=SC2034 local action="$1" param="$2" value="$3" local status message error stats local reload restart curReload curRestart ret i if [ -s "$jsonFile" ]; then json_load_file "$jsonFile" 2>/dev/null json_select 'data' 2>/dev/null for i in status message error stats reload restart; do json_get_var $i "$i" 2>/dev/null done fi case "$action" in get) case "$param" in triggers) curReload="$parallelDL $debug $dlTimeout $allowed_domains $blocked_domains $allowed_domains_urls $blocked_domains_urls $blocked_hosts_urls $targetDNS $configUpdateEnabled $configUpdateURL" curRestart="$compressedCache $forceDNS $led $forceDNSPorts" if [ ! -s "$jsonFile" ]; then ret='on_boot' elif [ "$curReload" != "$reload" ]; then ret='download' elif [ "$curRestart" != "$restart" ]; then ret='restart' fi printf "%b" "$ret" return;; *) printf "%b" "$(eval echo "\$$param")"; return;; esac ;; add) if [ -n "$(eval echo "\$$param")" ]; then value="$(eval echo "\$$param") ${value}" fi eval "$param"='${value}' ;; del) case "$param" in all) unset status message error stats;; triggers) unset reload restart;; *) unset "$param";; esac ;; set) case "$param" in triggers) reload="$parallelDL $debug $dlTimeout $allowed_domains $blocked_domains $allowed_domains_urls $blocked_domains_urls $blocked_hosts_urls $targetDNS $configUpdateEnabled $configUpdateURL" restart="$compressedCache $forceDNS $led $forceDNSPorts" ;; *) eval "$param"='$value';; esac ;; esac json_init json_add_object 'data' json_add_string version "$PKG_VERSION" json_add_string status "$status" json_add_string message "$message" json_add_string error "$error" json_add_string stats "$stats" json_add_string reload "$reload" json_add_string restart "$restart" json_close_object json_dump > "$jsonFile" sync } cacheOps() { local R_TMP case "$1" in create|backup) [ -s "$outputFile" ] && { mv -f "$outputFile" "$outputCache"; true > "$outputFile"; } >/dev/null 2>/dev/null return $? ;; restore|use) [ -s "$outputCache" ] && mv "$outputCache" "$outputFile" >/dev/null 2>/dev/null return $? ;; test) [ -s "$outputCache" ] return $? ;; testGzip) [ -s "$outputGzip" ] && gzip -t -c "$outputGzip" return $? ;; createGzip) R_TMP="$(mktemp -u -q -t ${packageName}_tmp.XXXXXXXX)" if gzip < "$outputFile" > "$R_TMP"; then if mv "$R_TMP" "$outputGzip"; then rm -f "$R_TMP" return 0 else rm -f "$R_TMP" return 1 fi else return 1 fi ;; expand|unpack|expandGzip|unpackGzip) [ -s "$outputGzip" ] && gzip -dc < "$outputGzip" > "$outputCache" return $? ;; esac } process_url() { local label type D_TMP R_TMP if [ -z "$1" ] || [ -z "$2" ] || [ -z "$3" ]; then return 1; fi label="${1##*//}"; label="${label%%/*}"; if [ "$2" = 'hosts' ]; then label="Hosts: $label"; filter="$hostsFilter"; else label="Domains: $label"; filter="$domainsFilter"; fi if [ "$3" = 'blocked' ]; then type='Blocked'; D_TMP="$B_TMP"; else type='Allowed'; D_TMP="$A_TMP"; fi if [ "${1:0:5}" == "https" ] && [ -z "$isSSLSupported" ]; then output 1 "$_FAIL_" output 2 "[DL] $type $label $__FAIL__\\n" echo "errorNoSSLSupport|${1}" >> "$sharedMemoryError" return 0 fi while [ -z "$R_TMP" ] || [ -e "$R_TMP" ]; do R_TMP="$(mktemp -u -q -t ${packageName}_tmp.XXXXXXXX)" done if ! $dl_command "$1" $dl_flag "$R_TMP" 2>/dev/null || [ ! -s "$R_TMP" ]; then output 1 "$_FAIL_" output 2 "[DL] $type $label $__FAIL__\\n" echo "errorDownloadingList|${1}" >> "$sharedMemoryError" else sed -i "$filter" "$R_TMP" if [ ! -s "$R_TMP" ]; then output 1 "$_FAIL_" output 2 "[DL] $type $label $__FAIL__\\n" echo "errorParsingList|${1}" >> "$sharedMemoryError" else cat "${R_TMP}" >> "$D_TMP" output 1 "$_OK_" output 2 "[DL] $type $label $__OK__\\n" fi fi rm -f "$R_TMP" return 0 } process_config_update() { local label R_TMP [ "$configUpdateEnabled" -eq 0 ] && return 0 label="${1##*//}"; label="${label%%/*}"; while [ -z "$R_TMP" ] || [ -e "$R_TMP" ]; do R_TMP="$(mktemp -u -q -t ${packageName}_tmp.XXXXXXXX)" done if ! $dl_command "$1" $dl_flag "$R_TMP" 2>/dev/null || [ ! -s "$R_TMP" ]; then output 1 "$_FAIL_" output 2 "[DL] Config Update: $label $__FAIL__\\n" jsonOps add error "errorDownloadingConfigUpdate" else if [ -s "$R_TMP" ] && sed -f "$R_TMP" -i /etc/config/simple-adblock 2>/dev/null; then output 1 "$_OK_" output 2 "[DL] Config Update: $label $__OK__\\n" else output 1 "$_FAIL_" output 2 "[DL] Config Update: $label $__FAIL__\\n" jsonOps add error "errorParsingConfigUpdate" fi fi rm -f "$R_TMP" return 0 } download_lists() { local hf w_filter j=0 R_TMP jsonOps set message "$(getStatusText "statusDownloading")..." jsonOps set status "statusDownloading" rm -f "$A_TMP" "$B_TMP" "$outputFile" "$outputCache" "$outputGzip" if [ "$($awk '/^MemFree/ {print int($2/1000)}' "/proc/meminfo")" -lt 32 ]; then output 3 'Low free memory, restarting resolver... ' if dnsOps 'quiet'; then output_okn else output_fail fi fi touch $A_TMP; touch $B_TMP; output 1 'Downloading lists ' process_config_update "$configUpdateURL" rm -f "$sharedMemoryError" if [ -n "$blocked_hosts_urls" ]; then for hf in ${blocked_hosts_urls}; do if [ "$parallelDL" -gt 0 ]; then process_url "$hf" 'hosts' 'blocked' & else process_url "$hf" 'hosts' 'blocked' fi done fi if [ -n "$blocked_domains_urls" ]; then for hf in ${blocked_domains_urls}; do if [ "$parallelDL" -gt 0 ]; then process_url "$hf" 'domains' 'blocked' & else process_url "$hf" 'domains' 'blocked' fi done fi if [ -n "$allowed_domains_urls" ]; then for hf in ${allowed_domains_urls}; do if [ "$parallelDL" -gt 0 ]; then process_url "$hf" 'domains' 'allowed' & else process_url "$hf" 'domains' 'allowed' fi done fi wait output 1 '\n' if [ -s "$sharedMemoryError" ]; then while IFS= read -r line; do jsonOps add error "$line" done < "$sharedMemoryError" rm -f "$sharedMemoryError" fi [ -n "$blocked_domains" ] && for hf in ${blocked_domains}; do echo "$hf" | sed "$domainsFilter" >> $B_TMP; done allowed_domains="${allowed_domains} $(cat $A_TMP)" [ -n "$allowed_domains" ] && for hf in ${allowed_domains}; do hf="$(echo "$hf" | sed 's/\./\\./g')"; w_filter="$w_filter/${hf}$/d;"; done [ ! -s "$B_TMP" ] && return 1 output 1 'Processing downloads ' output 2 'Sorting combined list ' jsonOps set message "$(getStatusText "statusProcessing"): sorting combined list" if [ "$allowIDN" -gt 0 ]; then if sort -u "$B_TMP" > "$A_TMP"; then output_ok else output_failn jsonOps add error "errorSorting" fi else if sort -u "$B_TMP" | grep -E -v '[^a-zA-Z0-9=/.-]' > "$A_TMP"; then output_ok else output_failn jsonOps add error "errorSorting" fi fi if [ "$targetDNS" = 'dnsmasq.conf' ] || \ [ "$targetDNS" = 'dnsmasq.ipset' ] || \ [ "$targetDNS" = 'dnsmasq.servers' ] || \ [ "$targetDNS" = 'unbound.adb_list' ]; then # TLD optimization written by Dirk Brenken (dev@brenken.org) output 2 'Optimizing combined list ' jsonOps set message "$(getStatusText "statusProcessing"): optimizing combined list" # sed -E 'G;:t;s/(.*)(\.)(.*)(\n)(.*)/\1\4\5\2\3/;tt;s/(.*)\n(\.)(.*)/\3\2\1/' is actually slower than command below if $awk -F "." '{for(f=NF;f>1;f--)printf "%s.",$f;print $1}' "$A_TMP" > "$B_TMP"; then if sort "$B_TMP" > "$A_TMP"; then if $awk '{if(NR=1){tld=$NF};while(getline){if($NF!~tld"\\."){print tld;tld=$NF}}print tld}' "$A_TMP" > "$B_TMP"; then if $awk -F "." '{for(f=NF;f>1;f--)printf "%s.",$f;print $1}' "$B_TMP" > "$A_TMP"; then if sort -u "$A_TMP" > "$B_TMP"; then output_ok else output_failn jsonOps add error "errorOptimization" mv "$A_TMP" "$B_TMP" fi else output_failn jsonOps add error "errorOptimization" fi else output_failn jsonOps add error "errorOptimization" mv "$A_TMP" "$B_TMP" fi else output_failn jsonOps add error "errorOptimization" fi else output_failn jsonOps add error "errorOptimization" mv "$A_TMP" "$B_TMP" fi else mv "$A_TMP" "$B_TMP" fi output 2 'Allowing domains ' jsonOps set message "$(getStatusText "statusProcessing"): allowing domains" if sed -i "$w_filter" "$B_TMP"; then output_ok else output_failn jsonOps add error "errorAllowListProcessing" fi output 2 'Formatting merged file ' jsonOps set message "$(getStatusText "statusProcessing"): formatting merged file" if [ -z "$outputFilterIPv6" ]; then if sed "$outputFilter" "$B_TMP" > "$A_TMP"; then output_ok else output_failn jsonOps add error "errorDataFileFormatting" fi else case "$targetDNS" in dnsmasq.addnhosts) if sed "$outputFilter" "$B_TMP" > "$A_TMP" && \ sed "$outputFilterIPv6" "$B_TMP" >> "$A_TMP"; then output_ok else output_failn jsonOps add error "errorDataFileFormatting" fi ;; esac fi case "$targetDNS" in dnsmasq.addnhosts) output 2 'Creating DNSMASQ addnhosts file ' jsonOps set message "$(getStatusText "statusProcessing"): creating DNSMASQ addnhosts file" ;; dnsmasq.conf) output 2 'Creating DNSMASQ config file ' jsonOps set message "$(getStatusText "statusProcessing"): creating DNSMASQ config file" ;; dnsmasq.ipset) output 2 'Creating DNSMASQ ipset file ' jsonOps set message "$(getStatusText "statusProcessing"): creating DNSMASQ ipset file" ;; dnsmasq.servers) output 2 'Creating DNSMASQ servers file ' jsonOps set message "$(getStatusText "statusProcessing"): creating DNSMASQ servers file" ;; unbound.adb_list) output 2 'Creating Unbound adb_list file ' jsonOps set message "$(getStatusText "statusProcessing"): creating Unbound adb_list file" ;; esac if mv "$A_TMP" "$outputFile"; then output_ok else output_failn jsonOps add error "errorMovingDataFile" fi if [ "$compressedCache" -gt 0 ]; then output 2 'Creating compressed cache ' jsonOps set message "$(getStatusText "statusProcessing"): creating compressed cache" if cacheOps 'createGzip'; then output_ok else output_failn jsonOps add error "errorCreatingCompressedCache" fi else rm -f "$outputGzip" fi output 2 'Removing temporary files ' jsonOps set message "$(getStatusText "statusProcessing"): removing temporary files" rm -f "/tmp/${packageName}_tmp.*" "$A_TMP" "$B_TMP" "$outputCache" || j=1 if [ $j -eq 0 ]; then output_ok else output_failn jsonOps add error "errorRemovingTempFiles" fi output 1 '\n' } boot() { load_package_config sleep "$bootDelay" rc_procd start_service 'on_boot' && rc_procd service_triggers } start_service() { is_enabled 'on_start' || return 1 local action status error message stats c status="$(jsonOps get status)" error="$(jsonOps get error)" message="$(jsonOps get message)" stats="$(jsonOps get stats)" action="$(jsonOps get triggers)" if [ "$action" = 'on_boot' ] || [ "$1" = 'on_boot' ]; then if cacheOps 'testGzip' || cacheOps 'test'; then action='restore' else action='download' fi elif [ "$action" = 'download' ] || [ "$1" = 'download' ] || [ -n "$error" ]; then action='download' elif [ ! -s "$outputFile" ]; then if cacheOps 'testGzip' || cacheOps 'test'; then action='restore' else action='download' fi elif [ "$action" = 'restart' ] || [ "$1" = 'restart' ]; then action='restart' elif [ -s "$outputFile" ] && [ "$status" = "statusSuccess" ] && [ -z "$error" ]; then showstatus exit 0 else action='download' fi jsonOps del all jsonOps set triggers procd_open_instance 'main' procd_set_param command /bin/true procd_set_param stdout 1 procd_set_param stderr 1 procd_open_data json_add_array firewall if [ "$forceDNS" -ne 0 ]; then for c in $forceDNSPorts; do if netstat -tuln | grep LISTEN | grep ":${c}" >/dev/null 2>&1; then json_add_object "" json_add_string type redirect json_add_string target DNAT json_add_string src lan json_add_string proto "tcp udp" json_add_string src_dport "$c" json_add_string dest_port "$c" json_add_boolean reflection 0 json_close_object else json_add_object "" json_add_string type rule json_add_string src lan json_add_string dest "*" json_add_string proto "tcp udp" json_add_string dest_port "$c" json_add_string target REJECT json_close_object fi done fi if [ "$targetDNS" = 'dnsmasq.ipset' ]; then json_add_object "" json_add_string type ipset json_add_string name adb json_add_string match dest_net json_add_string storage hash json_close_object json_add_object "" json_add_string type rule json_add_string ipset adb json_add_string src lan json_add_string dest "*" json_add_string proto "tcp udp" json_add_string target REJECT json_close_object fi json_close_array procd_close_data procd_close_instance if [ "$action" = 'restore' ]; then output 0 "Starting $serviceName... " output 3 "Starting $serviceName...\\n" jsonOps set status "statusStarting" if cacheOps 'testGzip' && ! cacheOps 'test' && [ ! -s "$outputFile" ]; then output 3 'Found compressed cache file, unpacking it ' jsonOps set message 'found compressed cache file, unpacking it.' if cacheOps 'unpackGzip'; then output_okn else output_fail jsonOps add error "errorRestoreCompressedCache" output "$_ERROR_: $(getErrorText 'errorRestoreCompressedCache')!\\n" action='download' fi fi if cacheOps 'test' && [ ! -s "$outputFile" ]; then output 3 'Found cache file, reusing it ' jsonOps set message 'found cache file, reusing it.' if cacheOps 'restore'; then output_okn dnsOps 'on_start' else output_fail jsonOps add error "errorRestoreCache" output "$_ERROR_: $(getErrorText 'errorRestoreCache')!\\n" action='download' fi fi fi if [ "$action" = 'download' ]; then if [ -s "$outputFile" ] || cacheOps 'test' || cacheOps 'testGzip'; then output 0 "Force-reloading $serviceName... " output 3 "Force-reloading $serviceName...\\n" jsonOps set status "statusForceReloading" else output 0 "Starting $serviceName... " output 3 "Starting $serviceName...\\n" jsonOps set status "statusStarting" fi download_lists dnsOps 'on_start' fi if [ "$action" = 'restart' ]; then output 0 "Restarting $serviceName... " output 3 "Restarting $serviceName...\\n" jsonOps set status "statusRestarting" dnsOps 'on_start' fi if [ "$action" = 'start' ]; then output 0 "Starting $serviceName... " output 3 "Starting $serviceName...\\n" jsonOps set status "statusStarting" dnsOps 'on_start' fi if [ -s "$outputFile" ] && [ "$(jsonOps get status)" != "statusFail" ]; then output 0 "$__OK__\\n"; jsonOps del message jsonOps set status "statusSuccess" jsonOps set stats "$serviceName is blocking $(wc -l < "$outputFile") domains (with ${targetDNS})" showstatus else output 0 "$__FAIL__\\n"; jsonOps set status "statusFail" jsonOps add error "errorOhSnap" showstatus fi } service_started() { procd_set_config_changed firewall; } service_stopped() { procd_set_config_changed firewall; } restart_service() { rc_procd start_service 'restart'; } reload_service() { rc_procd start_service 'restart'; } dl() { rc_procd start_service 'download'; } killcache() { rm -f "$addnhostsCache" "$addnhostsGzip" rm -f "$dnsmasqCache" "$dnsmasqGzip" rm -f "$ipsetCache" "$ipsetGzip" rm -f "$serversCache" "$serversGzip" rm -f "$unboundCache" "$unboundGzip" config_load 'dhcp' config_foreach dnsmasqOps 'dnsmasq' 'cleanup' uci -q commit 'dhcp' return 0 } show() { showstatus; } status_service() { showstatus; } showstatus() { local c url status message error stats status="$(jsonOps get status)" message="$(jsonOps get message)" error="$(jsonOps get error)" stats="$(jsonOps get stats)" if [ "$status" = "statusSuccess" ]; then output "$stats "; output_okn; else [ -n "$status" ] && status="$(getStatusText "$status")" if [ -n "$status" ] && [ -n "$message" ]; then status="${status}: $message" fi [ -n "$status" ] && output "$serviceName $status\\n" fi if [ -n "$error" ]; then for c in $error; do url="${c##*|}" c="${c%|*}" case "$c" in errorDownloadingList|errorParsingList) output "$_ERROR_: $(getErrorText "$c") $url!\\n";; *) output "$_ERROR_: $(getErrorText "$c")!\\n";; esac let n=n+1 done fi } stop_service() { load_package_config if [ -s "$outputFile" ]; then output "Stopping $serviceName... " cacheOps 'create' if dnsOps 'on_stop'; then led_off "$led" output 0 "$__OK__\\n"; output_okn; jsonOps set status "statusStopped" jsonOps del message else output 0 "$__FAIL__\\n"; output_fail; jsonOps set status "statusFail" jsonOps add error "errorStopping" output "$_ERROR_: $(getErrorText 'errorStopping')!\\n" fi fi } service_triggers() { procd_open_trigger procd_add_config_trigger 'config.change' "${packageName}" /etc/init.d/${packageName} reload procd_add_reload_interface_trigger 'wan' procd_add_interface_trigger 'interface.*.up' 'wan' /etc/init.d/${packageName} reload procd_close_trigger } check() { load_package_config local c string="$1" c="$(grep -c "$string" "$outputFile")" if [ ! -s "$outputFile" ]; then echo "No block-list ('$outputFile') found." elif [ -z "$string" ]; then echo "Usage: /etc/init.d/${packageName} check string" elif [ "$c" -gt 0 ]; then if [ "$c" -gt 1 ]; then echo "Found $c matches for '$string' in '$outputFile':" else echo "Found 1 match for '$string' in '$outputFile':" fi case "$targetDNS" in dnsmasq.addnhosts) grep "$string" "$outputFile" | sed 's|^127.0.0.1 ||;s|^:: ||;';; dnsmasq.conf) grep "$string" "$outputFile" | sed 's|local=/||;s|/$||;';; dnsmasq.ipset) grep "$string" "$outputFile" | sed 's|ipset=/||;s|/adb$||;';; dnsmasq.servers) grep "$string" "$outputFile" | sed 's|server=/||;s|/$||;';; unbound.adb_list) grep "$string" "$outputFile" | sed 's|^local-zone: "||;s|" static$||;';; esac else echo "The $string is not found in current block-list ('$outputFile')." fi } sizes() { local i load_package_config echo "# $(date)" for i in $blocked_domains_urls; do [ "${i//melmac}" != "$i" ] && continue if $dl_command "$i" $dl_flag /tmp/sast 2>/dev/null && [ -s /tmp/sast ]; then echo "# File size: $(du -sh /tmp/sast | $awk '{print $1}')" if compare_values "$(du -sk /tmp/sast)" "500"; then echo "# block-list too big for most routers" elif compare_values "$(du -sk /tmp/sast)" "100"; then echo "# block-list may be too big for some routers" fi rm -rf /tmp/sast echo " list blocked_domains_url '$i'" echo "" else echo "# site was down on last check" echo "# list blocked_domains_url '$i'" echo "" fi done for i in $blocked_hosts_urls; do if $dl_command "$i" $dl_flag /tmp/sast 2>/dev/null && [ -s /tmp/sast ]; then echo "# File size: $(du -sh /tmp/sast | $awk '{print $1}')" if compare_values "$(du -sk /tmp/sast)" "500"; then echo "# block-list too big for most routers" elif compare_values "$(du -sk /tmp/sast)" "100"; then echo "# block-list may be too big for some routers" fi rm -rf /tmp/sast echo " list blocked_hosts_url '$i'" echo "" else echo "# site was down on last check" echo "# list blocked_hosts_url '$i'" echo "" fi done }