mirror of
				https://github.com/searxng/searxng.git
				synced 2025-10-31 10:37:06 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			149 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			149 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # SPDX-License-Identifier: AGPL-3.0-or-later
 | |
| """.. _botdetection.ip_limit:
 | |
| 
 | |
| Method ``ip_limit``
 | |
| -------------------
 | |
| 
 | |
| The ``ip_limit`` method counts request from an IP in *sliding windows*.  If
 | |
| there are to many requests in a sliding window, the request is evaluated as a
 | |
| bot request.  This method requires a valkey DB and needs a HTTP X-Forwarded-For_
 | |
| header.  To take privacy only the hash value of an IP is stored in the valkey DB
 | |
| and at least for a maximum of 10 minutes.
 | |
| 
 | |
| The :py:obj:`.link_token` method can be used to investigate whether a request is
 | |
| *suspicious*.  To activate the :py:obj:`.link_token` method in the
 | |
| :py:obj:`.ip_limit` method add the following configuration:
 | |
| 
 | |
| .. code:: toml
 | |
| 
 | |
|    [botdetection.ip_limit]
 | |
|    link_token = true
 | |
| 
 | |
| If the :py:obj:`.link_token` method is activated and a request is *suspicious*
 | |
| the request rates are reduced:
 | |
| 
 | |
| - :py:obj:`BURST_MAX` -> :py:obj:`BURST_MAX_SUSPICIOUS`
 | |
| - :py:obj:`LONG_MAX` -> :py:obj:`LONG_MAX_SUSPICIOUS`
 | |
| 
 | |
| To intercept bots that get their IPs from a range of IPs, there is a
 | |
| :py:obj:`SUSPICIOUS_IP_WINDOW`.  In this window the suspicious IPs are stored
 | |
| for a longer time.  IPs stored in this sliding window have a maximum of
 | |
| :py:obj:`SUSPICIOUS_IP_MAX` accesses before they are blocked.  As soon as the IP
 | |
| makes a request that is not suspicious, the sliding window for this IP is
 | |
| dropped.
 | |
| 
 | |
| .. _X-Forwarded-For:
 | |
|    https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/X-Forwarded-For
 | |
| 
 | |
| """
 | |
| 
 | |
| from ipaddress import (
 | |
|     IPv4Network,
 | |
|     IPv6Network,
 | |
| )
 | |
| 
 | |
| import flask
 | |
| import werkzeug
 | |
| 
 | |
| from searx.valkeylib import incr_sliding_window, drop_counter
 | |
| 
 | |
| from . import link_token
 | |
| from . import config
 | |
| from . import valkeydb
 | |
| from ._helpers import (
 | |
|     too_many_requests,
 | |
|     logger,
 | |
| )
 | |
| 
 | |
| 
 | |
| logger = logger.getChild('ip_limit')
 | |
| 
 | |
| BURST_WINDOW = 20
 | |
| """Time (sec) before sliding window for *burst* requests expires."""
 | |
| 
 | |
| BURST_MAX = 15
 | |
| """Maximum requests from one IP in the :py:obj:`BURST_WINDOW`"""
 | |
| 
 | |
| BURST_MAX_SUSPICIOUS = 2
 | |
| """Maximum of suspicious requests from one IP in the :py:obj:`BURST_WINDOW`"""
 | |
| 
 | |
| LONG_WINDOW = 600
 | |
| """Time (sec) before the longer sliding window expires."""
 | |
| 
 | |
| LONG_MAX = 150
 | |
| """Maximum requests from one IP in the :py:obj:`LONG_WINDOW`"""
 | |
| 
 | |
| LONG_MAX_SUSPICIOUS = 10
 | |
| """Maximum suspicious requests from one IP in the :py:obj:`LONG_WINDOW`"""
 | |
| 
 | |
| API_WINDOW = 3600
 | |
| """Time (sec) before sliding window for API requests (format != html) expires."""
 | |
| 
 | |
| API_MAX = 4
 | |
| """Maximum requests from one IP in the :py:obj:`API_WINDOW`"""
 | |
| 
 | |
| SUSPICIOUS_IP_WINDOW = 3600 * 24 * 30
 | |
| """Time (sec) before sliding window for one suspicious IP expires."""
 | |
| 
 | |
| SUSPICIOUS_IP_MAX = 3
 | |
| """Maximum requests from one suspicious IP in the :py:obj:`SUSPICIOUS_IP_WINDOW`."""
 | |
| 
 | |
| 
 | |
| def filter_request(
 | |
|     network: IPv4Network | IPv6Network,
 | |
|     request: flask.Request,
 | |
|     cfg: config.Config,
 | |
| ) -> werkzeug.Response | None:
 | |
| 
 | |
|     # pylint: disable=too-many-return-statements
 | |
|     valkey_client = valkeydb.get_valkey_client()
 | |
| 
 | |
|     if network.is_link_local and not cfg['botdetection.ip_limit.filter_link_local']:
 | |
|         logger.debug("network %s is link-local -> not monitored by ip_limit method", network.compressed)
 | |
|         return None
 | |
| 
 | |
|     if request.args.get('format', 'html') != 'html':
 | |
|         c = incr_sliding_window(valkey_client, 'ip_limit.API_WINDOW:' + network.compressed, API_WINDOW)
 | |
|         if c > API_MAX:
 | |
|             return too_many_requests(network, "too many request in API_WINDOW")
 | |
| 
 | |
|     if cfg['botdetection.ip_limit.link_token']:
 | |
| 
 | |
|         suspicious = link_token.is_suspicious(network, request, True)
 | |
| 
 | |
|         if not suspicious:
 | |
|             # this IP is no longer suspicious: release ip again / delete the counter of this IP
 | |
|             drop_counter(valkey_client, 'ip_limit.SUSPICIOUS_IP_WINDOW' + network.compressed)
 | |
|             return None
 | |
| 
 | |
|         # this IP is suspicious: count requests from this IP
 | |
|         c = incr_sliding_window(
 | |
|             valkey_client, 'ip_limit.SUSPICIOUS_IP_WINDOW' + network.compressed, SUSPICIOUS_IP_WINDOW
 | |
|         )
 | |
|         if c > SUSPICIOUS_IP_MAX:
 | |
|             logger.error("BLOCK: too many request from %s in SUSPICIOUS_IP_WINDOW (redirect to /)", network)
 | |
|             response = flask.redirect(flask.url_for('index'), code=302)
 | |
|             response.headers["Cache-Control"] = "no-store, max-age=0"
 | |
|             return response
 | |
| 
 | |
|         c = incr_sliding_window(valkey_client, 'ip_limit.BURST_WINDOW' + network.compressed, BURST_WINDOW)
 | |
|         if c > BURST_MAX_SUSPICIOUS:
 | |
|             return too_many_requests(network, "too many request in BURST_WINDOW (BURST_MAX_SUSPICIOUS)")
 | |
| 
 | |
|         c = incr_sliding_window(valkey_client, 'ip_limit.LONG_WINDOW' + network.compressed, LONG_WINDOW)
 | |
|         if c > LONG_MAX_SUSPICIOUS:
 | |
|             return too_many_requests(network, "too many request in LONG_WINDOW (LONG_MAX_SUSPICIOUS)")
 | |
| 
 | |
|         return None
 | |
| 
 | |
|     # vanilla limiter without extensions counts BURST_MAX and LONG_MAX
 | |
|     c = incr_sliding_window(valkey_client, 'ip_limit.BURST_WINDOW' + network.compressed, BURST_WINDOW)
 | |
|     if c > BURST_MAX:
 | |
|         return too_many_requests(network, "too many request in BURST_WINDOW (BURST_MAX)")
 | |
| 
 | |
|     c = incr_sliding_window(valkey_client, 'ip_limit.LONG_WINDOW' + network.compressed, LONG_WINDOW)
 | |
|     if c > LONG_MAX:
 | |
|         return too_many_requests(network, "too many request in LONG_WINDOW (LONG_MAX)")
 | |
| 
 | |
|     return None
 |