D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
imunify360
/
venv
/
lib
/
python3.11
/
site-packages
/
imav
/
plugins
/
Filename :
server_pull.py
back
Copy
""" 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 <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> """ import logging from defence360agent.contracts.messages import MessageType from defence360agent.contracts.plugins import MessageSink, expect from imav.malwarelib.utils import malware_response from defence360agent.utils import Scope 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: await malware_response.upload_file(file, upload_reason=reason) except Exception as e: logger.warning(str(e))