You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

958 lines
27 KiB

  1. #!/bin/sh
  2. # banIP - ban incoming and outgoing ip adresses/subnets via ipset
  3. # written by Dirk Brenken (dev@brenken.org)
  4. # This is free software, licensed under the GNU General Public License v3.
  5. # You should have received a copy of the GNU General Public License
  6. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  7. # (s)hellcheck exceptions
  8. # shellcheck disable=1091 disable=2039 disable=2143 disable=2181 disable=2188
  9. # set initial defaults
  10. #
  11. LC_ALL=C
  12. PATH="/usr/sbin:/usr/bin:/sbin:/bin"
  13. ban_ver="0.3.10"
  14. ban_basever=""
  15. ban_enabled=0
  16. ban_automatic="1"
  17. ban_sources=""
  18. ban_iface=""
  19. ban_debug=0
  20. ban_backupdir="/mnt"
  21. ban_maxqueue=4
  22. ban_autoblacklist=1
  23. ban_autowhitelist=1
  24. ban_realtime="false"
  25. ban_fetchutil=""
  26. ban_ipt="$(command -v iptables)"
  27. ban_ipt_save="$(command -v iptables-save)"
  28. ban_ipt_restore="$(command -v iptables-restore)"
  29. ban_ipt6="$(command -v ip6tables)"
  30. ban_ipt6_save="$(command -v ip6tables-save)"
  31. ban_ipt6_restore="$(command -v ip6tables-restore)"
  32. ban_ipset="$(command -v ipset)"
  33. ban_logger="$(command -v logger)"
  34. ban_chain="banIP"
  35. ban_action="${1:-"start"}"
  36. ban_pidfile="/var/run/banip.pid"
  37. ban_rtfile="/tmp/ban_runtime.json"
  38. ban_logservice="/etc/banip/banip.service"
  39. ban_sshdaemon=""
  40. ban_setcnt=0
  41. ban_cnt=0
  42. # load environment
  43. #
  44. f_envload()
  45. {
  46. # get system information
  47. #
  48. ban_sysver="$(ubus -S call system board 2>/dev/null | jsonfilter -e '@.model' -e '@.release.description' | \
  49. awk 'BEGIN{ORS=", "}{print $0}' | awk '{print substr($0,1,length($0)-2)}')"
  50. # parse 'global' and 'extra' section by callback
  51. #
  52. config_cb()
  53. {
  54. local type="${1}"
  55. if [ "${type}" = "banip" ]
  56. then
  57. option_cb()
  58. {
  59. local option="${1}"
  60. local value="${2}"
  61. eval "${option}=\"${value}\""
  62. }
  63. else
  64. reset_cb
  65. fi
  66. }
  67. # parse 'source' typed sections
  68. #
  69. parse_config()
  70. {
  71. local value opt section="${1}" options="ban_src ban_src_6 ban_src_rset ban_src_rset_6 ban_src_settype ban_src_ruletype ban_src_on ban_src_on_6 ban_src_cat"
  72. for opt in ${options}
  73. do
  74. config_get value "${section}" "${opt}"
  75. if [ -n "${value}" ]
  76. then
  77. eval "${opt}_${section}=\"${value}\""
  78. if [ "${opt}" = "ban_src" ]
  79. then
  80. eval "ban_sources=\"${ban_sources} ${section}\""
  81. elif [ "${opt}" = "ban_src_6" ]
  82. then
  83. eval "ban_sources=\"${ban_sources} ${section}_6\""
  84. fi
  85. fi
  86. done
  87. }
  88. # load config
  89. #
  90. config_load banip
  91. config_foreach parse_config source
  92. # log daemon check
  93. #
  94. if [ "$(/etc/init.d/log running; printf "%u" "${?}")" -eq 1 ]
  95. then
  96. unset ban_logger
  97. f_log "info" "your log daemon 'logd' is not running, please enable 'logd' to use this service"
  98. f_rmtemp
  99. exit 1
  100. fi
  101. # version check
  102. #
  103. if [ -z "${ban_basever}" ] || [ "${ban_ver%.*}" != "${ban_basever}" ]
  104. then
  105. f_log "info" "your banIP config seems to be too old, please update your config with the '--force-maintainer' opkg option"
  106. f_rmtemp
  107. exit 0
  108. fi
  109. # create temp directory & files
  110. #
  111. f_temp
  112. # check status
  113. #
  114. if [ "${ban_enabled}" -eq 0 ]
  115. then
  116. f_bgserv "stop"
  117. f_jsnup disabled
  118. f_ipset destroy
  119. f_rmbackup
  120. f_rmtemp
  121. f_log "info" "banIP is currently disabled, please set ban_enabled to '1' to use this service"
  122. exit 0
  123. fi
  124. }
  125. # check environment
  126. #
  127. f_envcheck()
  128. {
  129. local util utils packages iface tmp cnt=0 cnt_max=0
  130. f_jsnup "running"
  131. f_log "info" "start banIP processing (${ban_action})"
  132. # check backup directory
  133. #
  134. if [ ! -d "${ban_backupdir}" ]
  135. then
  136. f_log "err" "the backup directory '${ban_backupdir}' does not exist or has not been mounted yet, please create the directory or raise the 'ban_triggerdelay' to defer the banIP start"
  137. fi
  138. # get wan devices and wan subnets
  139. #
  140. if [ "${ban_automatic}" = "1" ]
  141. then
  142. while [ "${cnt}" -le 30 ]
  143. do
  144. network_find_wan iface
  145. if [ -n "${iface}" ] && [ -z "$(printf "%s\\n" "${ban_iface}" | grep -F "${iface}")" ]
  146. then
  147. ban_iface="${ban_iface} ${iface}"
  148. if [ "${cnt_max}" -eq 0 ]
  149. then
  150. cnt_max=$((cnt+5))
  151. fi
  152. fi
  153. network_find_wan6 iface
  154. if [ -n "${iface}" ] && [ -z "$(printf "%s\\n" "${ban_iface}" | grep -F "${iface}")" ]
  155. then
  156. ban_iface="${ban_iface} ${iface}"
  157. if [ "${cnt_max}" -eq 0 ]
  158. then
  159. cnt_max=$((cnt+5))
  160. fi
  161. fi
  162. if [ -z "${ban_iface}" ] || [ "${cnt}" -le "${cnt_max}" ]
  163. then
  164. network_flush_cache
  165. cnt=$((cnt+1))
  166. sleep 1
  167. else
  168. break
  169. fi
  170. done
  171. fi
  172. for iface in ${ban_iface}
  173. do
  174. network_get_device tmp "${iface}"
  175. if [ -n "${tmp}" ] && [ -z "$(printf "%s\\n" "${ban_dev}" | grep -F "${tmp}")" ]
  176. then
  177. ban_dev="${ban_dev} ${tmp}"
  178. else
  179. network_get_physdev tmp "${iface}"
  180. if [ -n "${tmp}" ] && [ -z "$(printf "%s\\n" "${ban_dev}" | grep -F "${tmp}")" ]
  181. then
  182. ban_dev="${ban_dev} ${tmp}"
  183. fi
  184. fi
  185. network_get_subnets tmp "${iface}"
  186. if [ -n "${tmp}" ] && [ -z "$(printf "%s\\n" "${ban_subnets}" | grep -F "${tmp}")" ]
  187. then
  188. ban_subnets="${ban_subnets} ${tmp}"
  189. fi
  190. network_get_subnets6 tmp "${iface}"
  191. if [ -n "${tmp}" ] && [ -z "$(printf "%s\\n" "${ban_subnets6}" | grep -F "${tmp}")" ]
  192. then
  193. ban_subnets6="${ban_subnets6} ${tmp}"
  194. fi
  195. done
  196. ban_dev_all="$(ip link show 2>/dev/null | awk 'BEGIN{FS="[@: ]"}/^[0-9:]/{if($3!="lo"){print $3}}')"
  197. if [ -z "${ban_iface}" ] || [ -z "${ban_dev}" ] || [ -z "${ban_dev_all}" ]
  198. then
  199. f_log "err" "wan interface(s)/device(s) (${ban_iface:-"-"}/${ban_dev:-"-"}) not found, please please check your configuration"
  200. fi
  201. # check fetch utility
  202. #
  203. if [ -z "${ban_fetchutil}" ]
  204. then
  205. cnt_max=$((cnt+5))
  206. while [ -z "${packages}" ]
  207. do
  208. packages="$(opkg list-installed 2>/dev/null)"
  209. if [ "${cnt}" -gt "${cnt_max}" ]
  210. then
  211. break
  212. fi
  213. cnt=$((cnt+1))
  214. sleep 1
  215. done
  216. if [ -n "${packages}" ]
  217. then
  218. utils="aria2c curl wget uclient-fetch"
  219. for util in ${utils}
  220. do
  221. if { [ "${util}" = "uclient-fetch" ] && [ -n "$(printf "%s\\n" "${packages}" | grep "^libustream-")" ]; } || \
  222. { [ "${util}" = "wget" ] && [ -n "$(printf "%s\\n" "${packages}" | grep "^wget -")" ]; } || \
  223. { [ "${util}" != "uclient-fetch" ] && [ "${util}" != "wget" ]; }
  224. then
  225. ban_fetchutil="$(command -v "${util}")"
  226. if [ -x "${ban_fetchutil}" ]
  227. then
  228. break
  229. fi
  230. fi
  231. unset ban_fetchutil util
  232. done
  233. fi
  234. else
  235. util="${ban_fetchutil}"
  236. ban_fetchutil="$(command -v "${util}")"
  237. if [ ! -x "${ban_fetchutil}" ]
  238. then
  239. unset ban_fetchutil util
  240. fi
  241. fi
  242. case "${util}" in
  243. "aria2c")
  244. ban_fetchparm="${ban_fetchparm:-"--timeout=20 --allow-overwrite=true --auto-file-renaming=false --check-certificate=true --dir=" " -o"}"
  245. ;;
  246. "curl")
  247. ban_fetchparm="${ban_fetchparm:-"--connect-timeout 20 -o"}"
  248. ;;
  249. "uclient-fetch")
  250. ban_fetchparm="${ban_fetchparm:-"--timeout=20 -O"}"
  251. ;;
  252. "wget")
  253. ban_fetchparm="${ban_fetchparm:-"--no-cache --no-cookies --max-redirect=0 --timeout=20 -O"}"
  254. ;;
  255. esac
  256. if [ -z "${ban_fetchutil}" ] || [ -z "${ban_fetchparm}" ]
  257. then
  258. f_log "err" "download utility with SSL support not found, please install 'uclient-fetch' with a 'libustream-*' variant or another download utility like 'wget', 'curl' or 'aria2'"
  259. fi
  260. # check ssh daemon
  261. #
  262. if [ -z "${ban_sshdaemon}" ]
  263. then
  264. utils="dropbear sshd"
  265. for util in ${utils}
  266. do
  267. if [ -x "$(command -v "${util}")" ]
  268. then
  269. ban_sshdaemon="${util}"
  270. break
  271. fi
  272. done
  273. fi
  274. if [ -z "${ban_sshdaemon}" ]
  275. then
  276. f_log "err" "ssh daemon not found, please install 'dropbear' or 'sshd'"
  277. fi
  278. }
  279. # create temporary files and directories
  280. #
  281. f_temp()
  282. {
  283. if [ -d "/tmp" ] && [ -z "${ban_tmpdir}" ]
  284. then
  285. ban_tmpdir="$(mktemp -p /tmp -d)"
  286. ban_tmpfile="$(mktemp -p "${ban_tmpdir}" -tu)"
  287. elif [ ! -d "/tmp" ]
  288. then
  289. f_log "err" "the temp directory '/tmp' does not exist or has not been mounted yet, please create the directory or raise the 'ban_triggerdelay' to defer the banIP start"
  290. fi
  291. if [ ! -f "${ban_pidfile}" ] || [ ! -s "${ban_pidfile}" ]
  292. then
  293. printf "%s" "${$}" > "${ban_pidfile}"
  294. fi
  295. }
  296. # remove temporary files and directories
  297. #
  298. f_rmtemp()
  299. {
  300. if [ -d "${ban_tmpdir}" ]
  301. then
  302. rm -rf "${ban_tmpdir}"
  303. fi
  304. > "${ban_pidfile}"
  305. }
  306. # remove backup files
  307. #
  308. f_rmbackup()
  309. {
  310. if [ -d "${ban_backupdir}" ]
  311. then
  312. rm -f "${ban_backupdir}"/banIP.*.gz
  313. fi
  314. }
  315. # iptables rules engine
  316. #
  317. f_iptrule()
  318. {
  319. local rc timeout="-w 5" action="${1}" rule="${2}"
  320. if [ "${src_name##*_}" = "6" ]
  321. then
  322. if [ -x "${ban_ipt6}" ]
  323. then
  324. rc="$("${ban_ipt6}" "${timeout}" -C ${rule} 2>/dev/null; printf "%u" ${?})"
  325. if { [ "${rc}" -ne 0 ] && { [ "${action}" = "-A" ] || [ "${action}" = "-I" ]; } } || \
  326. { [ "${rc}" -eq 0 ] && [ "${action}" = "-D" ]; }
  327. then
  328. "${ban_ipt6}" "${timeout}" "${action}" ${rule} 2>/dev/null
  329. fi
  330. fi
  331. else
  332. if [ -x "${ban_ipt}" ]
  333. then
  334. rc="$("${ban_ipt}" "${timeout}" -C ${rule} 2>/dev/null; printf "%u" ${?})"
  335. if { [ "${rc}" -ne 0 ] && { [ "${action}" = "-A" ] || [ "${action}" = "-I" ]; } } || \
  336. { [ "${rc}" -eq 0 ] && [ "${action}" = "-D" ]; }
  337. then
  338. "${ban_ipt}" "${timeout}" "${action}" ${rule} 2>/dev/null
  339. fi
  340. fi
  341. fi
  342. if [ "${?}" -ne 0 ]
  343. then
  344. > "${tmp_err}"
  345. f_log "info" "can't create iptables rule: action: '${action:-"-"}', rule: '${rule:-"-"}'"
  346. fi
  347. }
  348. # remove/add iptables rules
  349. #
  350. f_iptadd()
  351. {
  352. local rm="${1}" dev
  353. for dev in ${ban_dev_all}
  354. do
  355. f_iptrule "-D" "${ban_chain} -i ${dev} -m conntrack --ctstate NEW -m set --match-set ${src_name} src -j ${target_src}"
  356. f_iptrule "-D" "${ban_chain} -o ${dev} -m conntrack --ctstate NEW -m set --match-set ${src_name} dst -j ${target_dst}"
  357. done
  358. if [ -z "${rm}" ] && [ "${cnt}" -gt 0 ]
  359. then
  360. if [ "${src_ruletype}" != "dst" ]
  361. then
  362. if [ "${src_name##*_}" = "6" ]
  363. then
  364. # dummy, special IPv6 rules
  365. /bin/true
  366. else
  367. f_iptrule "-I" "${wan_input} -p udp --dport 67:68 --sport 67:68 -j RETURN"
  368. fi
  369. f_iptrule "-A" "${wan_input} -j ${ban_chain}"
  370. f_iptrule "-A" "${wan_forward} -j ${ban_chain}"
  371. for dev in ${ban_dev}
  372. do
  373. f_iptrule "${action:-"-A"}" "${ban_chain} -i ${dev} -m conntrack --ctstate NEW -m set --match-set ${src_name} src -j ${target_src}"
  374. done
  375. fi
  376. if [ "${src_ruletype}" != "src" ]
  377. then
  378. if [ "${src_name##*_}" = "6" ]
  379. then
  380. # dummy, special IPv6 rules
  381. /bin/true
  382. else
  383. f_iptrule "-I" "${lan_input} -p udp --dport 67:68 --sport 67:68 -j RETURN"
  384. fi
  385. f_iptrule "-A" "${lan_input} -j ${ban_chain}"
  386. f_iptrule "-A" "${lan_forward} -j ${ban_chain}"
  387. for dev in ${ban_dev}
  388. do
  389. f_iptrule "${action:-"-A"}" "${ban_chain} -o ${dev} -m conntrack --ctstate NEW -m set --match-set ${src_name} dst -j ${target_dst}"
  390. done
  391. fi
  392. else
  393. if [ -x "${ban_ipset}" ] && [ -n "$("${ban_ipset}" -q -n list "${src_name}")" ]
  394. then
  395. "${ban_ipset}" -q destroy "${src_name}"
  396. fi
  397. fi
  398. }
  399. # ipset/iptables actions
  400. #
  401. f_ipset()
  402. {
  403. local out_rc source action ruleset ruleset_6 rule cnt=0 cnt_ip=0 cnt_cidr=0 timeout="-w 5" mode="${1}" in_rc="${src_rc:-0}"
  404. if [ "${src_name%_6*}" = "whitelist" ]
  405. then
  406. target_src="RETURN"
  407. target_dst="RETURN"
  408. action="-I"
  409. fi
  410. case "${mode}" in
  411. "backup")
  412. gzip -cf "${tmp_load}" 2>/dev/null > "${ban_backupdir}/banIP.${src_name}.gz"
  413. out_rc="${?:-"${in_rc}"}"
  414. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}, out_rc: ${out_rc}"
  415. return "${out_rc}"
  416. ;;
  417. "restore")
  418. if [ -f "${ban_backupdir}/banIP.${src_name}.gz" ]
  419. then
  420. zcat "${ban_backupdir}/banIP.${src_name}.gz" 2>/dev/null > "${tmp_load}"
  421. out_rc="${?}"
  422. fi
  423. out_rc="${out_rc:-"${in_rc}"}"
  424. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}, out_rc: ${out_rc}"
  425. return "${out_rc}"
  426. ;;
  427. "remove")
  428. if [ -f "${ban_backupdir}/banIP.${src_name}.gz" ]
  429. then
  430. rm -f "${ban_backupdir}/banIP.${src_name}.gz"
  431. out_rc="${?}"
  432. fi
  433. out_rc="${out_rc:-"${in_rc}"}"
  434. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}, out_rc: ${out_rc}"
  435. return "${out_rc}"
  436. ;;
  437. "initial")
  438. if [ -x "${ban_ipt}" ] && [ -z "$("${ban_ipt}" "${timeout}" -nL "${ban_chain}" 2>/dev/null)" ]
  439. then
  440. "${ban_ipt}" "${timeout}" -N "${ban_chain}" 2>/dev/null
  441. out_rc="${?}"
  442. elif [ -x "${ban_ipt}" ]
  443. then
  444. src_name="ruleset"
  445. ruleset="${ban_wan_input_chain:-"input_wan_rule"} ${ban_wan_forward_chain:-"forwarding_wan_rule"} ${ban_lan_input_chain:-"input_lan_rule"} ${ban_lan_forward_chain:-"forwarding_lan_rule"}"
  446. for rule in ${ruleset}
  447. do
  448. f_iptrule "-D" "${rule} -j ${ban_chain}"
  449. done
  450. fi
  451. if [ -x "${ban_ipt6}" ] && [ -z "$("${ban_ipt6}" "${timeout}" -nL "${ban_chain}" 2>/dev/null)" ]
  452. then
  453. "${ban_ipt6}" "${timeout}" -N "${ban_chain}" 2>/dev/null
  454. out_rc="${?}"
  455. elif [ -x "${ban_ipt6}" ]
  456. then
  457. src_name="ruleset_6"
  458. ruleset_6="${ban_wan_input_chain_6:-"input_wan_rule"} ${ban_wan_forward_chain_6:-"forwarding_wan_rule"} ${ban_lan_input_chain_6:-"input_lan_rule"} ${ban_lan_forward_chain_6:-"forwarding_lan_rule"}"
  459. for rule in ${ruleset_6}
  460. do
  461. f_iptrule "-D" "${rule} -j ${ban_chain}"
  462. done
  463. fi
  464. out_rc="${out_rc:-"${in_rc}"}"
  465. f_log "debug" "f_ipset ::: name: -, mode: ${mode:-"-"}, chain: ${ban_chain:-"-"}, ruleset: ${ruleset:-"-"}, ruleset_6: ${ruleset_6:-"-"}, out_rc: ${out_rc}"
  466. return "${out_rc}"
  467. ;;
  468. "create")
  469. if [ -x "${ban_ipset}" ]
  470. then
  471. if [ -s "${tmp_file}" ] && [ -z "$("${ban_ipset}" -q -n list "${src_name}")" ]
  472. then
  473. "${ban_ipset}" -q create "${src_name}" hash:"${src_settype}" hashsize 64 maxelem 262144 family "${src_setipv}" counters
  474. out_rc="${?}"
  475. else
  476. "${ban_ipset}" -q flush "${src_name}"
  477. out_rc="${?}"
  478. fi
  479. if [ -s "${tmp_file}" ] && [ "${out_rc}" -eq 0 ]
  480. then
  481. "${ban_ipset}" -q -! restore < "${tmp_file}"
  482. out_rc="${?}"
  483. if [ "${out_rc}" -eq 0 ]
  484. then
  485. "${ban_ipset}" -q save "${src_name}" > "${tmp_file}"
  486. cnt="$(($(wc -l 2>/dev/null < "${tmp_file}")-1))"
  487. cnt_cidr="$(grep -cF "/" "${tmp_file}")"
  488. cnt_ip="$((cnt-cnt_cidr))"
  489. printf "%s\\n" "${cnt}" > "${tmp_cnt}"
  490. fi
  491. fi
  492. f_iptadd
  493. fi
  494. end_ts="$(date +%s)"
  495. out_rc="${out_rc:-"${in_rc}"}"
  496. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}, settype: ${src_settype:-"-"}, setipv: ${src_setipv:-"-"}, ruletype: ${src_ruletype:-"-"}, count(sum/ip/cidr): ${cnt}/${cnt_ip}/${cnt_cidr}, time: $((end_ts-start_ts)), out_rc: ${out_rc}"
  497. return "${out_rc}"
  498. ;;
  499. "refresh")
  500. if [ -x "${ban_ipset}" ] && [ -n "$("${ban_ipset}" -q -n list "${src_name}")" ]
  501. then
  502. "${ban_ipset}" -q save "${src_name}" > "${tmp_file}"
  503. out_rc="${?}"
  504. if [ -s "${tmp_file}" ] && [ "${out_rc}" -eq 0 ]
  505. then
  506. cnt="$(($(wc -l 2>/dev/null < "${tmp_file}")-1))"
  507. cnt_cidr="$(grep -cF "/" "${tmp_file}")"
  508. cnt_ip="$((cnt-cnt_cidr))"
  509. printf "%s\\n" "${cnt}" > "${tmp_cnt}"
  510. fi
  511. f_iptadd
  512. fi
  513. end_ts="$(date +%s)"
  514. out_rc="${out_rc:-"${in_rc}"}"
  515. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}, count: ${cnt}/${cnt_ip}/${cnt_cidr}, time: $((end_ts-start_ts)), out_rc: ${out_rc}"
  516. return "${out_rc}"
  517. ;;
  518. "flush")
  519. f_iptadd "remove"
  520. if [ -x "${ban_ipset}" ] && [ -n "$("${ban_ipset}" -q -n list "${src_name}")" ]
  521. then
  522. "${ban_ipset}" -q flush "${src_name}"
  523. "${ban_ipset}" -q destroy "${src_name}"
  524. fi
  525. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}"
  526. ;;
  527. "destroy")
  528. if [ -x "${ban_ipt}" ] && [ -x "${ban_ipt_save}" ] && [ -x "${ban_ipt_restore}" ] && \
  529. [ -n "$("${ban_ipt}" "${timeout}" -nL "${ban_chain}" 2>/dev/null)" ]
  530. then
  531. "${ban_ipt_save}" | grep -v -- "-j ${ban_chain}" | "${ban_ipt_restore}"
  532. "${ban_ipt}" "${timeout}" -F "${ban_chain}" 2>/dev/null
  533. "${ban_ipt}" "${timeout}" -X "${ban_chain}" 2>/dev/null
  534. fi
  535. if [ -x "${ban_ipt6}" ] && [ -x "${ban_ipt6_save}" ] && [ -x "${ban_ipt6_restore}" ] && \
  536. [ -n "$("${ban_ipt6}" "${timeout}" -nL "${ban_chain}" 2>/dev/null)" ]
  537. then
  538. "${ban_ipt6_save}" | grep -v -- "-j ${ban_chain}" | "${ban_ipt6_restore}"
  539. "${ban_ipt6}" "${timeout}" -F "${ban_chain}" 2>/dev/null
  540. "${ban_ipt6}" "${timeout}" -X "${ban_chain}" 2>/dev/null
  541. fi
  542. for source in ${ban_sources}
  543. do
  544. if [ -x "${ban_ipset}" ] && [ -n "$("${ban_ipset}" -q -n list "${source}")" ]
  545. then
  546. "${ban_ipset}" -q destroy "${source}"
  547. fi
  548. done
  549. f_log "debug" "f_ipset ::: name: ${src_name:-"-"}, mode: ${mode:-"-"}"
  550. ;;
  551. esac
  552. }
  553. # write to syslog
  554. #
  555. f_log()
  556. {
  557. local class="${1}" log_msg="${2}"
  558. if [ -n "${log_msg}" ] && { [ "${class}" != "debug" ] || [ "${ban_debug}" -eq 1 ]; }
  559. then
  560. if [ -x "${ban_logger}" ]
  561. then
  562. "${ban_logger}" -p "${class}" -t "banIP-${ban_ver}[${$}]" "${log_msg}"
  563. else
  564. printf "%s %s %s\\n" "${class}" "banIP-${ban_ver}[${$}]" "${log_msg}"
  565. fi
  566. if [ "${class}" = "err" ]
  567. then
  568. f_jsnup error
  569. f_ipset destroy
  570. f_rmbackup
  571. f_rmtemp
  572. exit 1
  573. fi
  574. fi
  575. }
  576. # start log service to trace failed ssh/luci logins
  577. #
  578. f_bgserv()
  579. {
  580. local bg_pid status="${1}"
  581. bg_pid="$(pgrep -f "^/bin/sh ${ban_logservice}.*|^logread -f -e ${ban_sshdaemon}\|luci: failed login|^grep -qE Exit before auth|luci: failed login|[0-9]+ \[preauth\]$" | awk '{ORS=" "; print $1}')"
  582. if [ -z "${bg_pid}" ] && [ "${status}" = "start" ] \
  583. && [ -x "${ban_logservice}" ] && [ "${ban_realtime}" = "true" ]
  584. then
  585. ( "${ban_logservice}" "${ban_ver}" "${ban_sshdaemon}" & )
  586. elif [ -n "${bg_pid}" ] && [ "${status}" = "stop" ]
  587. then
  588. kill -HUP "${bg_pid}" 2>/dev/null
  589. fi
  590. f_log "debug" "f_bgserv ::: status: ${status:-"-"}, bg_pid: ${bg_pid:-"-"}, ban_realtime: ${ban_realtime:-"-"}, log_service: ${ban_logservice:-"-"}"
  591. }
  592. # main function for banIP processing
  593. #
  594. f_main()
  595. {
  596. local pid pid_list start_ts end_ts ip tmp_raw tmp_cnt tmp_load tmp_file mem_total mem_free cnt=1
  597. local src_name src_on src_url src_rset src_setipv src_settype src_ruletype src_cat src_log src_addon src_ts src_rc
  598. local wan_input wan_forward lan_input lan_forward target_src target_dst ssh_log luci_log
  599. ssh_log="$(logread -e "${ban_sshdaemon}" | grep -o "${ban_sshdaemon}.*" | sed 's/:[0-9]*$//g')"
  600. luci_log="$(logread -e "luci: failed login" | grep -o "luci:.*")"
  601. mem_total="$(awk '/^MemTotal/ {print int($2/1000)}' "/proc/meminfo" 2>/dev/null)"
  602. mem_free="$(awk '/^MemFree/ {print int($2/1000)}' "/proc/meminfo" 2>/dev/null)"
  603. f_log "debug" "f_main ::: fetch_util: ${ban_fetchutil:-"-"}, fetch_parm: ${ban_fetchparm:-"-"}, ssh_daemon: ${ban_sshdaemon}, interface(s): ${ban_iface:-"-"}, device(s): ${ban_dev:-"-"}, all_devices: ${ban_dev_all:-"-"}, backup_dir: ${ban_backupdir:-"-"}, mem_total: ${mem_total:-0}, mem_free: ${mem_free:-0}, max_queue: ${ban_maxqueue}"
  604. # chain creation
  605. #
  606. f_ipset initial
  607. if [ "${?}" -ne 0 ]
  608. then
  609. f_log "err" "banIP processing failed, fatal error during iptables chain creation (${ban_sysver})"
  610. fi
  611. # main loop
  612. #
  613. for src_name in ${ban_sources}
  614. do
  615. unset src_on
  616. if [ "${src_name##*_}" = "6" ]
  617. then
  618. if [ -x "${ban_ipt6}" ]
  619. then
  620. src_on="$(eval printf "%s" \"\$\{ban_src_on_6_${src_name%_6*}\}\")"
  621. src_url="$(eval printf "%s" \"\$\{ban_src_6_${src_name%_6*}\}\")"
  622. src_rset="$(eval printf "%s" \"\$\{ban_src_rset_6_${src_name%_6*}\}\")"
  623. src_setipv="inet6"
  624. wan_input="${ban_wan_input_chain_6:-"input_wan_rule"}"
  625. wan_forward="${ban_wan_forward_chain_6:-"forwarding_wan_rule"}"
  626. lan_input="${ban_lan_input_chain_6:-"input_lan_rule"}"
  627. lan_forward="${ban_lan_forward_chain_6:-"forwarding_lan_rule"}"
  628. target_src="${ban_target_src_6:-"DROP"}"
  629. target_dst="${ban_target_dst_6:-"REJECT"}"
  630. fi
  631. else
  632. if [ -x "${ban_ipt}" ]
  633. then
  634. src_on="$(eval printf "%s" \"\$\{ban_src_on_${src_name}\}\")"
  635. src_url="$(eval printf "%s" \"\$\{ban_src_${src_name}\}\")"
  636. src_rset="$(eval printf "%s" \"\$\{ban_src_rset_${src_name}\}\")"
  637. src_setipv="inet"
  638. wan_input="${ban_wan_input_chain:-"input_wan_rule"}"
  639. wan_forward="${ban_wan_forward_chain:-"forwarding_wan_rule"}"
  640. lan_input="${ban_lan_input_chain:-"input_lan_rule"}"
  641. lan_forward="${ban_lan_forward_chain:-"forwarding_lan_rule"}"
  642. target_src="${ban_target_src:-"DROP"}"
  643. target_dst="${ban_target_dst:-"REJECT"}"
  644. fi
  645. fi
  646. src_settype="$(eval printf "%s" \"\$\{ban_src_settype_${src_name%_6*}\}\")"
  647. src_ruletype="$(eval printf "%s" \"\$\{ban_src_ruletype_${src_name%_6*}\}\")"
  648. src_cat="$(eval printf "%s" \"\$\{ban_src_cat_${src_name%_6*}\}\")"
  649. src_addon=""
  650. src_rc=4
  651. tmp_load="${ban_tmpfile}.${src_name}.load"
  652. tmp_file="${ban_tmpfile}.${src_name}.file"
  653. tmp_raw="${tmp_file}.raw"
  654. tmp_cnt="${tmp_file}.cnt"
  655. tmp_err="${tmp_file}.err"
  656. # basic pre-checks
  657. #
  658. f_log "debug" "f_main ::: name: ${src_name}, src_on: ${src_on:-"-"}"
  659. if [ -z "${src_on}" ] || [ "${src_on}" != "1" ] || [ -z "${src_url}" ] || \
  660. [ -z "${src_rset}" ] || [ -z "${src_settype}" ] || [ -z "${src_ruletype}" ]
  661. then
  662. f_ipset flush
  663. f_ipset remove
  664. continue
  665. elif [ "${ban_action}" = "refresh" ] && [ ! -f "${src_url}" ]
  666. then
  667. start_ts="$(date +%s)"
  668. f_ipset refresh
  669. if [ "${?}" -eq 0 ]
  670. then
  671. continue
  672. fi
  673. fi
  674. # download queue processing
  675. #
  676. (
  677. start_ts="$(date +%s)"
  678. if [ "${ban_action}" = "start" ] && [ ! -f "${src_url}" ]
  679. then
  680. f_ipset restore
  681. fi
  682. src_rc="${?}"
  683. if [ "${src_rc}" -ne 0 ] || [ ! -s "${tmp_load}" ]
  684. then
  685. if [ -f "${src_url}" ]
  686. then
  687. src_log="$(cat "${src_url}" 2>/dev/null > "${tmp_load}")"
  688. src_rc="${?}"
  689. case "${src_name}" in
  690. "whitelist")
  691. src_addon="${ban_subnets}"
  692. ;;
  693. "whitelist_6")
  694. src_addon="${ban_subnets6}"
  695. ;;
  696. "blacklist")
  697. if [ "${ban_sshdaemon}" = "dropbear" ]
  698. then
  699. pid_list="$(printf "%s\\n" "${ssh_log}" | grep -F "Exit before auth" | awk 'match($0,/(\[[0-9]+\])/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  700. for pid in ${pid_list}
  701. do
  702. src_addon="${src_addon} $(printf "%s\\n" "${ssh_log}" | grep -F "${pid}" | awk 'match($0,/([0-9]{1,3}\.){3}[0-9]{1,3}$/){ORS=" ";print substr($0,RSTART,RLENGTH);exit}')"
  703. done
  704. elif [ "${ban_sshdaemon}" = "sshd" ]
  705. then
  706. src_addon="$(printf "%s\\n" "${ssh_log}" | grep -E "[0-9]+ \[preauth\]$" | awk 'match($0,/([0-9]{1,3}\.){3}[0-9]{1,3}$/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  707. fi
  708. src_addon="${src_addon} $(printf "%s\\n" "${luci_log}" | awk 'match($0,/([0-9]{1,3}\.){3}[0-9]{1,3}$/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  709. ;;
  710. "blacklist_6")
  711. if [ "${ban_sshdaemon}" = "dropbear" ]
  712. then
  713. pid_list="$(printf "%s\\n" "${ssh_log}" | grep -F "Exit before auth" | awk 'match($0,/(\[[0-9]+\])/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  714. for pid in ${pid_list}
  715. do
  716. src_addon="${src_addon} $(printf "%s\\n" "${ssh_log}" | grep -F "${pid}" | awk 'match($0,/(([0-9A-f]{0,4}::?){1,7}[0-9A-f]{0,4}$)/){ORS=" ";print substr($0,RSTART,RLENGTH);exit}')"
  717. done
  718. elif [ "${ban_sshdaemon}" = "sshd" ]
  719. then
  720. src_addon="$(printf "%s\\n" "${ssh_log}" | grep -E "[0-9]+ \[preauth\]$" | awk 'match($0,/(([0-9A-f]{0,4}::?){1,7}[0-9A-f]{0,4}$)/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  721. fi
  722. src_addon="${src_addon} $(printf "%s\\n" "${luci_log}" | awk 'match($0,/(([0-9A-f]{0,4}::?){1,7}[0-9A-f]{0,4}$)/){ORS=" ";print substr($0,RSTART,RLENGTH)}')"
  723. ;;
  724. esac
  725. for ip in ${src_addon}
  726. do
  727. if [ -z "$(grep -F "${ip}" "${src_url}")" ]
  728. then
  729. printf "%s\\n" "${ip}" >> "${tmp_load}"
  730. if { [ "${src_name//_*/}" = "blacklist" ] && [ "${ban_autoblacklist}" -eq 1 ]; } || \
  731. { [ "${src_name//_*/}" = "whitelist" ] && [ "${ban_autowhitelist}" -eq 1 ]; }
  732. then
  733. src_ts="# auto-added $(date "+%d.%m.%Y %H:%M:%S")"
  734. printf "%s %s\\n" "${ip}" "${src_ts}" >> "${src_url}"
  735. fi
  736. fi
  737. done
  738. elif [ -n "${src_cat}" ]
  739. then
  740. if [ "${src_cat//[0-9]/}" != "${src_cat}" ]
  741. then
  742. for as in ${src_cat}
  743. do
  744. src_log="$("${ban_fetchutil}" ${ban_fetchparm} "${tmp_raw}" "${src_url}AS${as}" 2>&1)"
  745. src_rc="${?}"
  746. if [ "${src_rc}" -eq 0 ]
  747. then
  748. jsonfilter -i "${tmp_raw}" -e '@.data.prefixes.*.prefix' 2>/dev/null >> "${tmp_load}"
  749. else
  750. break
  751. fi
  752. done
  753. if [ "${src_rc}" -eq 0 ]
  754. then
  755. f_ipset backup
  756. elif [ "${ban_action}" != "start" ]
  757. then
  758. f_ipset restore
  759. fi
  760. else
  761. for co in ${src_cat}
  762. do
  763. src_log="$("${ban_fetchutil}" ${ban_fetchparm} "${tmp_raw}" "${src_url}${co}&v4_format=prefix" 2>&1)"
  764. src_rc="${?}"
  765. if [ "${src_rc}" -eq 0 ]
  766. then
  767. if [ "${src_name##*_}" = "6" ]
  768. then
  769. jsonfilter -i "${tmp_raw}" -e '@.data.resources.ipv6.*' 2>/dev/null >> "${tmp_load}"
  770. else
  771. jsonfilter -i "${tmp_raw}" -e '@.data.resources.ipv4.*' 2>/dev/null >> "${tmp_load}"
  772. fi
  773. else
  774. break
  775. fi
  776. done
  777. if [ "${src_rc}" -eq 0 ]
  778. then
  779. f_ipset backup
  780. elif [ "${ban_action}" != "start" ]
  781. then
  782. f_ipset restore
  783. fi
  784. fi
  785. else
  786. src_log="$("${ban_fetchutil}" ${ban_fetchparm} "${tmp_raw}" "${src_url}" 2>&1)"
  787. src_rc="${?}"
  788. if [ "${src_rc}" -eq 0 ]
  789. then
  790. zcat "${tmp_raw}" 2>/dev/null > "${tmp_load}"
  791. src_rc="${?}"
  792. if [ "${src_rc}" -ne 0 ]
  793. then
  794. mv -f "${tmp_raw}" "${tmp_load}"
  795. src_rc="${?}"
  796. fi
  797. if [ "${src_rc}" -eq 0 ]
  798. then
  799. f_ipset backup
  800. src_rc="${?}"
  801. fi
  802. elif [ "${ban_action}" != "start" ]
  803. then
  804. f_ipset restore
  805. src_rc="${?}"
  806. fi
  807. fi
  808. fi
  809. if [ "${src_rc}" -eq 0 ]
  810. then
  811. awk "${src_rset}" "${tmp_load}" 2>/dev/null > "${tmp_file}"
  812. src_rc="${?}"
  813. if [ "${src_rc}" -eq 0 ]
  814. then
  815. f_ipset create
  816. src_rc="${?}"
  817. elif [ "${ban_action}" != "refresh" ]
  818. then
  819. f_ipset refresh
  820. src_rc="${?}"
  821. fi
  822. else
  823. src_log="$(printf "%s" "${src_log}" | awk '{ORS=" ";print $0}')"
  824. if [ "${ban_action}" != "refresh" ]
  825. then
  826. f_ipset refresh
  827. src_rc="${?}"
  828. fi
  829. f_log "debug" "f_main ::: name: ${src_name}, url: ${src_url}, rc: ${src_rc}, log: ${src_log:-"-"}"
  830. fi
  831. )&
  832. hold="$((cnt%ban_maxqueue))"
  833. if [ "${hold}" -eq 0 ]
  834. then
  835. wait
  836. fi
  837. cnt="$((cnt+1))"
  838. done
  839. wait
  840. if [ -z "$(ls "${ban_tmpfile}".*.err 2>/dev/null)" ]
  841. then
  842. for cnt in $(cat "${ban_tmpfile}".*.cnt 2>/dev/null)
  843. do
  844. ban_cnt="$((ban_cnt+cnt))"
  845. done
  846. if [ "${ban_cnt}" -gt 0 ]
  847. then
  848. ban_setcnt="$(ls "${ban_tmpfile}".*.cnt 2>/dev/null | wc -l)"
  849. fi
  850. f_log "info" "${ban_setcnt} IPSets with overall ${ban_cnt} IPs/Prefixes loaded successfully (${ban_sysver})"
  851. f_bgserv "start"
  852. f_jsnup
  853. f_rmtemp
  854. else
  855. f_log "err" "banIP processing failed, fatal iptables error(s) during subshell processing (${ban_sysver})"
  856. fi
  857. }
  858. # update runtime information
  859. #
  860. f_jsnup()
  861. {
  862. local rundate status="${1:-"enabled"}"
  863. rundate="$(date "+%d.%m.%Y %H:%M:%S")"
  864. ban_cntinfo="${ban_setcnt} IPSets with overall ${ban_cnt} IPs/Prefixes"
  865. > "${ban_rtfile}"
  866. json_load_file "${ban_rtfile}" >/dev/null 2>&1
  867. json_init
  868. json_add_object "data"
  869. json_add_string "status" "${status}"
  870. json_add_string "version" "${ban_ver}"
  871. json_add_string "util_info" "${ban_fetchutil:-"-"}, ${ban_realtime:-"-"}"
  872. json_add_string "ipset_info" "${ban_cntinfo:-"-"}"
  873. json_add_string "backup_dir" "${ban_backupdir}"
  874. json_add_string "last_run" "${rundate:-"-"}"
  875. json_add_string "system" "${ban_sysver}"
  876. json_close_object
  877. json_dump > "${ban_rtfile}"
  878. f_log "debug" "f_jsnup ::: status: ${status}, setcnt: ${ban_setcnt}, cnt: ${ban_cnt}"
  879. }
  880. # source required system libraries
  881. #
  882. if [ -r "/lib/functions.sh" ] && [ -r "/lib/functions/network.sh" ] && [ -r "/usr/share/libubox/jshn.sh" ]
  883. then
  884. . "/lib/functions.sh"
  885. . "/lib/functions/network.sh"
  886. . "/usr/share/libubox/jshn.sh"
  887. else
  888. f_log "err" "system libraries not found"
  889. fi
  890. # handle different banIP actions
  891. #
  892. f_envload
  893. case "${ban_action}" in
  894. "stop")
  895. f_bgserv "stop"
  896. f_jsnup stopped
  897. f_ipset destroy
  898. f_rmbackup
  899. f_rmtemp
  900. ;;
  901. "start"|"restart"|"reload"|"refresh")
  902. f_bgserv "stop"
  903. f_envcheck
  904. f_main
  905. ;;
  906. esac