????

Your IP : 18.223.99.202

Current Path : /opt/imunify360/venv/lib/python3.11/site-packages/im360/plugins/resident/
Upload File :
Current File : //opt/imunify360/venv/lib/python3.11/site-packages/im360/plugins/resident/repeater.py

from logging import getLogger

from defence360agent.contracts.messages import MessageType
from defence360agent.contracts.plugins import MessageSink, expect
from defence360agent.utils import Scope
from im360.plugins.sensor.generic import send_to_agent_socket

logger = getLogger(__name__)


class Repeater(MessageSink):
    SCOPE = Scope.IM360_RESIDENT

    async def create_sink(self, loop):
        self._loop = loop

    @expect(MessageType.MalwareRescanFiles)
    async def resend_to_simple_socket(self, msg):
        send_to_agent_socket(
            ["malware", "rescan"], params={"files": msg.files}
        )

    @expect(MessageType.MalwareSendFiles)
    async def resend_malware_send_files(self, msg):
        send_to_agent_socket(
            ["malware", "send", "files"],
            params={"reason": msg.reason, "files": msg.files},
        )