aDriv4 website!

Linux premium8.web-hosting.com 4.18.0-553.44.1.lve.el8.x86_64 #1 SMP Thu Mar 13 14:29:12 UTC 2025 x86_64
LiteSpeed
198.187.31.46

File Name : 001_fix_scan_unserialization.py

"""
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/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>

Used to fix issue with inability to unserialize stored scans.
See DEF-23158 for details.
"""
import collections
import importlib
import logging
import pickle
from pathlib import Path

logger = logging.getLogger(__name__)

SCANS_PATH = Path("/var/imunify360/aibolit/scans.pickle")
OLD_MALWARELIB = "defence360agent.malwarelib"
NEW_MALWARELIB = "imav.malwarelib"

class AVUnpickler(pickle.Unpickler):
    def find_class(self, module, name):
        try:
            return super().find_class(module, name)
        except (ImportError, ModuleNotFoundError):
            if module.startswith(OLD_MALWARELIB):
                av_module = importlib.import_module(
                    module.replace(OLD_MALWARELIB, NEW_MALWARELIB)
                )
                return getattr(av_module, name)
            raise

def dump(obj, path):
    temp_path = path.with_name(path.name + ".temp")
    with temp_path.open("wb") as f:
        pickle.dump(obj, f)
    # to avoid the possibility of leaving a broken file,
    # if any errors occurred above
    temp_path.replace(path)

def migrate(migrator, *_, fake=False, **__):
    if fake or not SCANS_PATH.exists():
        return

if OLD_MALWARELIB.encode() in SCANS_PATH.read_bytes():
        try:
            with SCANS_PATH.open("rb") as f:
                scans = AVUnpickler(f).load()
            # convert class instance to a simple dict
            updated = collections.deque(
                {
                    "path": item.path,
                    "scanid": item.scanid,
                    "scan_type": item.scan_type,
                    "created": item.created,
                    "started": item.started,
                    "args": item.args,
                    "resource_type": item.resource_type.value,
                    "state": item.state.value,
                }
                for item in scans
            )
        except Exception as exc:
            logger.exception(
                "Failed to load pickle scans %s: %s", SCANS_PATH, exc
            )
        else:
            dump(updated, SCANS_PATH)

def rollback(migrator, *_, fake=False, **__):
    pass

Name	Type	Size	Owner/Group	Permission	Last Modified	Actions
.	dir	-	root/root	0755	2025-08-05 07:02:08
..	dir	-	root/root	0755	2025-08-05 07:02:08
__pycache__	dir	-	root/root	0755	2025-08-05 07:02:08
000_noop_migration.py	text/plain	947 B	root/root	0644	2025-07-23 06:44:12
001_fix_scan_unserialization.py	text/plain	2.85 KB	root/root	0644	2025-07-23 06:44:12
002_convert_serialized_scans.py	text/plain	2.48 KB	root/root	0644	2025-07-23 06:44:12
003_add_new_fields_to_malware_history.py	text/plain	1.29 KB	root/root	0644	2025-07-23 06:44:12
003_whmapi1_set_importance_imav.py	text/plain	1.64 KB	root/root	0644	2025-07-23 06:44:12
004_add_malwarehit_owner.py	text/plain	1.1 KB	root/root	0644	2025-07-23 06:44:12
004_patchman_client.py	text/plain	1.71 KB	root/root	0644	2025-07-23 06:44:12
004_plesk_configs.py	text/plain	6.3 KB	root/root	0644	2025-07-23 06:44:12
004_plesk_eula.py	text/plain	2.33 KB	root/root	0644	2025-07-23 06:44:12
005_plesk_cleanup_storage.py	text/plain	4.01 KB	root/root	0644	2025-07-23 06:44:12
005_populate_malwarehit_owner.py	text/plain	1.12 KB	root/root	0644	2025-07-23 06:44:12
006_add_malwarehit_snippet.py	text/plain	1.11 KB	root/root	0644	2025-07-23 06:44:12
007_add_malwarehistory_fileuser.py	text/plain	1.13 KB	root/root	0644	2025-07-23 06:44:12
007_revisium_ignore_list.py	text/plain	2.6 KB	root/root	0644	2025-07-23 06:44:12
008_populate_malwarehistory_foleuser.py	text/plain	1.16 KB	root/root	0644	2025-07-23 06:44:12
008_subscription_permissions.py	text/plain	3.29 KB	root/root	0644	2025-07-23 06:44:12
009_revisium_scan_history.py	text/plain	8.99 KB	root/root	0644	2025-07-23 06:44:12
010_add_malwarehistory_scan_id.py	text/plain	1.13 KB	root/root	0644	2025-07-23 06:44:12
011_add_malwarehistory_index.py	text/plain	1.49 KB	root/root	0644	2025-07-23 06:44:12
012_add_rescan_outdated_scan_type.py	text/plain	1.4 KB	root/root	0644	2025-07-23 06:44:12
013_add_malware_ctime.py	text/plain	1.2 KB	root/root	0644	2025-07-23 06:44:12
014_add_imunify_security.py	text/plain	1.22 KB	root/root	0644	2025-07-23 06:44:12
015_manually_deleted_at_field_added.py	text/plain	1.17 KB	root/root	0644	2025-07-23 06:44:12
016_add_vulnerability_hit_table.py	text/plain	2.48 KB	root/root	0644	2025-07-23 06:44:12
017_add_imunify_patch_subscriptions_table.py	text/plain	1.29 KB	root/root	0644	2025-07-23 06:44:12
017_add_version_field.py	text/plain	1.16 KB	root/root	0644	2025-07-23 06:44:12
018_delete_suspicious_hits.py	text/plain	977 B	root/root	0644	2025-07-23 06:44:12
__init__.py	application/x-empty	0 B	root/root	0644	2025-07-23 06:44:12

For more Private tool come to me!

https://t.me/aDriv4vz