| Server IP : 188.40.95.74 / Your IP : 216.73.216.142 Web Server : Apache System : Linux cp01.striminghost.net 3.10.0-1160.119.1.el7.tuxcare.els13.x86_64 #1 SMP Fri Nov 22 06:29:45 UTC 2024 x86_64 User : vlasotin ( 1054) PHP Version : 5.6.40 Disable Function : NONE MySQL : ON | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/simple_rpc/ |
Upload File : |
from collections import defaultdict
from defence360agent.contracts.config import ConfigFile
from defence360agent.rpc_tools.lookup import RootEndpoints
from defence360agent.utils.config import update_config
from defence360agent.rpc_tools import lookup
from defence360agent.api.server.events import EventsAPI
from defence360agent.feature_management.checkers import config_cleanup
class AdvisorEndpoints(RootEndpoints):
@lookup.bind("advisor", "apply")
async def advisor_apply(self, advices):
return await self._apply(advices)
@lookup.bind("advisor", "apply-all")
async def apply_all(self):
advices = await EventsAPI.advices()
return await self._apply(advices)
async def _apply(self, advices):
target_conf = defaultdict(dict)
current_conf = ConfigFile().config_to_dict()
for advise in advices:
self._extract_conf_from_advise(advise, current_conf, target_conf)
await update_config(self._sink, target_conf)
return {"items": config_cleanup(ConfigFile().config_to_dict())}
@staticmethod
def _extract_conf_from_advise(advise, current_conf, target_conf):
for section_key, section_value in advise["ignore"].items():
for value_key, ignored_values in section_value.items():
if current_conf[section_key][value_key] in ignored_values:
return
for section_key, section_value in advise["config_action"].items():
target_conf[section_key].update(section_value)