D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
imunify360
/
venv
/
lib64
/
python3.11
/
site-packages
/
defence360agent
/
migrations
/
Filename :
109_dos_detector.py
back
Copy
import logging from defence360agent.contracts.config import ( _DOS_DETECTOR_MIN_LIMIT, ConfigFile, ) logger = logging.getLogger(__name__) def migrate(migrator, database, fake=False, **kwargs): if fake: return try: config_file = ConfigFile() config = config_file.config_to_dict(False) if "DOS" not in config: return if "max_connections" in config["DOS"] and isinstance( config["DOS"]["max_connections"], int ): config["DOS"]["default_limit"] = max( config["DOS"]["max_connections"], _DOS_DETECTOR_MIN_LIMIT ) del config["DOS"]["max_connections"] if "timeout" in config["DOS"]: config["DOS"]["interval"] = config["DOS"]["timeout"] del config["DOS"]["timeout"] config_file.dict_to_config(config, overwrite=True) except Exception: logger.exception("Failed to replace DOS settings") def rollback(migrator, database, fake=False, **kwargs): pass