forked from Ponysearch/Ponysearch
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||
|
|
# lint: pylint
|
||
|
|
""".. _botdetection.ip_lists:
|
||
|
|
|
||
|
|
Method ``ip_lists``
|
||
|
|
-------------------
|
||
|
|
|
||
|
|
The ``ip_lists`` method implements IP :py:obj:`block- <block_ip>` and
|
||
|
|
:py:obj:`pass-lists <pass_ip>`.
|
||
|
|
|
||
|
|
.. code:: toml
|
||
|
|
|
||
|
|
[botdetection.ip_lists]
|
||
|
|
|
||
|
|
pass_ip = [
|
||
|
|
'140.238.172.132', # IPv4 of check.searx.space
|
||
|
|
'192.168.0.0/16', # IPv4 private network
|
||
|
|
'fe80::/10' # IPv6 linklocal
|
||
|
|
]
|
||
|
|
block_ip = [
|
||
|
|
'93.184.216.34', # IPv4 of example.org
|
||
|
|
'257.1.1.1', # invalid IP --> will be ignored, logged in ERROR class
|
||
|
|
]
|
||
|
|
|
||
|
|
"""
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
from typing import Tuple
|
||
|
|
from ipaddress import (
|
||
|
|
ip_network,
|
||
|
|
IPv4Address,
|
||
|
|
IPv6Address,
|
||
|
|
)
|
||
|
|
|
||
|
|
from searx.tools import config
|
||
|
|
from ._helpers import logger
|
||
|
|
|
||
|
|
logger = logger.getChild('ip_limit')
|
||
|
|
|
||
|
|
SEARXNG_ORG = [
|
||
|
|
# https://github.com/searxng/searxng/pull/2484#issuecomment-1576639195
|
||
|
|
'140.238.172.132', # IPv4 check.searx.space
|
||
|
|
'2603:c022:0:4900::/56', # IPv6 check.searx.space
|
||
|
|
]
|
||
|
|
"""Passlist of IPs from the SearXNG organization, e.g. `check.searx.space`."""
|
||
|
|
|
||
|
|
|
||
|
|
def pass_ip(real_ip: IPv4Address | IPv6Address, cfg: config.Config) -> Tuple[bool, str]:
|
||
|
|
"""Checks if the IP on the subnet is in one of the members of the
|
||
|
|
``botdetection.ip_lists.pass_ip`` list.
|
||
|
|
"""
|
||
|
|
|
||
|
|
if cfg.get('botdetection.ip_lists.pass_searxng_org', default=True):
|
||
|
|
for net in SEARXNG_ORG:
|
||
|
|
net = ip_network(net, strict=False)
|
||
|
|
if real_ip.version == net.version and real_ip in net:
|
||
|
|
return True, f"IP matches {net.compressed} in SEARXNG_ORG list."
|
||
|
|
return ip_is_subnet_of_member_in_list(real_ip, 'botdetection.ip_lists.pass_ip', cfg)
|
||
|
|
|
||
|
|
|
||
|
|
def block_ip(real_ip: IPv4Address | IPv6Address, cfg: config.Config) -> Tuple[bool, str]:
|
||
|
|
"""Checks if the IP on the subnet is in one of the members of the
|
||
|
|
``botdetection.ip_lists.block_ip`` list.
|
||
|
|
"""
|
||
|
|
|
||
|
|
block, msg = ip_is_subnet_of_member_in_list(real_ip, 'botdetection.ip_lists.block_ip', cfg)
|
||
|
|
if block:
|
||
|
|
msg += " To remove IP from list, please contact the maintainer of the service."
|
||
|
|
return block, msg
|
||
|
|
|
||
|
|
|
||
|
|
def ip_is_subnet_of_member_in_list(
|
||
|
|
real_ip: IPv4Address | IPv6Address, list_name: str, cfg: config.Config
|
||
|
|
) -> Tuple[bool, str]:
|
||
|
|
|
||
|
|
for net in cfg.get(list_name, default=[]):
|
||
|
|
try:
|
||
|
|
net = ip_network(net, strict=False)
|
||
|
|
except ValueError:
|
||
|
|
logger.error("invalid IP %s in %s", net, list_name)
|
||
|
|
continue
|
||
|
|
if real_ip.version == net.version and real_ip in net:
|
||
|
|
return True, f"IP matches {net.compressed} in {list_name}."
|
||
|
|
return False, f"IP is not a member of an item in the f{list_name} list"
|