FM / plugins

/opt/imunify360/venv/lib/python3.11/site-packages/imav/plugins/

/opt/imunify360/venv/lib/python3.11/site-packages/imav/plugins

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
aibolit_resident_ff_watcher.py	10300	0644	edit dl rm
check_license.py	9492	0644	edit dl rm
conflicts.py	3255	0644	edit dl rm
detect_admin_tools_watcher.py	2708	0644	edit dl rm
event_hooks.py	3549	0644	edit dl rm
event_hook_executor.py	3806	0644	edit dl rm
generic_sensor.py	6940	0644	edit dl rm
im360_register.py	2890	0644	edit dl rm
imunify_patch_id.py	2381	0644	edit dl rm
inotify.py	1787	0644	edit dl rm
malware_filters.py	3681	0644	edit dl rm
mr_proper.py	2736	0644	edit dl rm
plesk_notifications.py	4704	0644	edit dl rm
post_action.py	1758	0644	edit dl rm
restore_from_backup.py	3451	0644	edit dl rm
send_malware_infection_state.py	11787	0644	edit dl rm
server_pull.py	2621	0644	edit dl rm
service_manager.py	2692	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imunify360/venv/lib/python3.11/site-packages/imav/plugins/imunify_patch_id.py (2381B)

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import logging
from typing import Any
from defence360agent.contracts.messages import MessageType, ServerConfig
from defence360agent.contracts.plugins import (
    MessageSink,
    expect,
)
from defence360agent.utils import Scope
from imav.contracts.imunify_patch_id import (
    get_imunify_patch_id,
    get_imunify_patch_purchase_url,
)
from defence360agent.subsys.panels.hosting_panel import HostingPanel

UserData = dict[str, Any]

logger = logging.getLogger(__name__)

class EnrichServerConfigWithImunifyPatchId(MessageSink):
    SCOPE = Scope.AV_IM360
    PROCESSING_ORDER = MessageSink.ProcessingOrder.PRE_PROCESS_MESSAGE

async def create_sink(self, loop):
        pass

@expect(MessageType.ServerConfig)
    async def enrich_server_config(
        self, message: ServerConfig
    ) -> ServerConfig:
        if "users" in message:
            users = message["users"]
        else:
            users = [
                {"username": user} for user in await HostingPanel().get_users()
            ]

message["users"] = [
            await self._enrich_user_data(user) for user in users
        ]

return message

async def _enrich_user_data(self, user: UserData) -> UserData:
        if "username" not in user:
            logger.warning(f"User data missing 'username' field: {user=}")
            return user
        user["imunify_patch_user_id"] = await get_imunify_patch_id(
            user["username"]
        )
        user[
            "imunify_patch_purchase_url"
        ] = await get_imunify_patch_purchase_url(user["username"])
        return user

""" This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see """ import logging from typing import Any from defence360agent.contracts.messages import MessageType, ServerConfig from defence360agent.contracts.plugins import ( MessageSink, expect, ) from defence360agent.utils import Scope from imav.contracts.imunify_patch_id import ( get_imunify_patch_id, get_imunify_patch_purchase_url, ) from defence360agent.subsys.panels.hosting_panel import HostingPanel UserData = dict[str, Any] logger = logging.getLogger(__name__) class EnrichServerConfigWithImunifyPatchId(MessageSink): SCOPE = Scope.AV_IM360 PROCESSING_ORDER = MessageSink.ProcessingOrder.PRE_PROCESS_MESSAGE async def create_sink(self, loop): pass @expect(MessageType.ServerConfig) async def enrich_server_config( self, message: ServerConfig ) -> ServerConfig: if "users" in message: users = message["users"] else: users = [ {"username": user} for user in await HostingPanel().get_users() ] message["users"] = [ await self._enrich_user_data(user) for user in users ] return message async def _enrich_user_data(self, user: UserData) -> UserData: if "username" not in user: logger.warning(f"User data missing 'username' field: {user=}") return user user["imunify_patch_user_id"] = await get_imunify_patch_id( user["username"] ) user[ "imunify_patch_purchase_url" ] = await get_imunify_patch_purchase_url(user["username"]) return user