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/server_pull.py (2621B)

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 hashlib
import logging
import os
from pathlib import Path
import tempfile
from contextlib import ExitStack

from defence360agent import utils
from defence360agent.contracts.config import Core
from defence360agent.contracts.messages import MessageType
from defence360agent.contracts.plugins import MessageSink, expect
from defence360agent.utils import Scope
from imav.malwarelib.utils import malware_response

logger = logging.getLogger(__name__)

class ServerPullFileSender(MessageSink):
    SCOPE = Scope.IM360
    loop = None

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

@expect(MessageType.MalwareSendFiles)
    async def upload_files(self, message):
        reason = message["reason"]
        for file in message["files"]:
            try:
                with ExitStack() as stack:
                    # Save directory contents into tempfile and upload it to MRS
                    if os.path.isdir(file):
                        tf = stack.enter_context(
                            tempfile.NamedTemporaryFile(dir=Core.TMPDIR)
                        )
                        tf.write(await utils.check_run(["ls", "-al", file]))
                        tf.seek(0)
                        file = malware_response.MalwareHitPath(tf.name, file)
                        hash_ = hashlib.sha256(
                            Path(tf.name).read_bytes()
                        ).hexdigest()
                        logger.info(
                            "Uploading temporary file %s with hash: %s",
                            tf.name,
                            hash_,
                        )
                    await malware_response.upload_file(
                        file, upload_reason=reason
                    )
            except Exception as e:
                logger.warning(str(e))

""" 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 hashlib import logging import os from pathlib import Path import tempfile from contextlib import ExitStack from defence360agent import utils from defence360agent.contracts.config import Core from defence360agent.contracts.messages import MessageType from defence360agent.contracts.plugins import MessageSink, expect from defence360agent.utils import Scope from imav.malwarelib.utils import malware_response logger = logging.getLogger(__name__) class ServerPullFileSender(MessageSink): SCOPE = Scope.IM360 loop = None async def create_sink(self, loop): self.loop = loop @expect(MessageType.MalwareSendFiles) async def upload_files(self, message): reason = message["reason"] for file in message["files"]: try: with ExitStack() as stack: # Save directory contents into tempfile and upload it to MRS if os.path.isdir(file): tf = stack.enter_context( tempfile.NamedTemporaryFile(dir=Core.TMPDIR) ) tf.write(await utils.check_run(["ls", "-al", file])) tf.seek(0) file = malware_response.MalwareHitPath(tf.name, file) hash_ = hashlib.sha256( Path(tf.name).read_bytes() ).hexdigest() logger.info( "Uploading temporary file %s with hash: %s", tf.name, hash_, ) await malware_response.upload_file( file, upload_reason=reason ) except Exception as e: logger.warning(str(e))