File Manager

X7ROOT File Manager

Current Path: /opt/imunify360/venv/lib/python3.11/site-packages/imav/migrations

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

📁 ..
📄 000_noop_migration.py(947 B)
📄 001_fix_scan_unserialization.py(2.85 KB)
📄 002_convert_serialized_scans.py(2.48 KB)
📄 003_add_new_fields_to_malware_history.py(1.29 KB)
📄 003_whmapi1_set_importance_imav.py(1.64 KB)
📄 004_add_malwarehit_owner.py(1.1 KB)
📄 004_patchman_client.py(1.71 KB)
📄 004_plesk_configs.py(6.3 KB)
📄 004_plesk_eula.py(2.33 KB)
📄 005_plesk_cleanup_storage.py(4.01 KB)
📄 005_populate_malwarehit_owner.py(1.12 KB)
📄 006_add_malwarehit_snippet.py(1.11 KB)
📄 007_add_malwarehistory_fileuser.py(1.13 KB)
📄 007_revisium_ignore_list.py(2.6 KB)
📄 008_populate_malwarehistory_foleuser.py(1.16 KB)
📄 008_subscription_permissions.py(3.29 KB)
📄 009_revisium_scan_history.py(8.99 KB)
📄 010_add_malwarehistory_scan_id.py(1.13 KB)
📄 011_add_malwarehistory_index.py(1.49 KB)
📄 012_add_rescan_outdated_scan_type.py(1.4 KB)
📄 013_add_malware_ctime.py(1.2 KB)
📄 014_add_imunify_security.py(1.22 KB)
📄 015_manually_deleted_at_field_added.py(1.17 KB)
📄 016_add_vulnerability_hit_table.py(2.48 KB)
📄 017_add_imunify_patch_subscriptions_table.py(1.29 KB)
📄 __init__.py(0 B)
📁 __pycache__

Editing: 002_convert_serialized_scans.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>

Convert serialized scans classes to dicts.

In most cases it should be already done in 001_fix_scan_unserialization, but
it is still necessary for those who have already updated to 7.1.0
before running 001_fix_scan_unserialization migration.
See DEF-24115 for details.
"""
import collections
import logging
import pickle
from pathlib import Path

logger = logging.getLogger(__name__)

SCANS_PATH = Path("/var/imunify360/aibolit/scans.pickle")

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 b"imav.malwarelib" in SCANS_PATH.read_bytes():
        try:
            with SCANS_PATH.open("rb") as f:
                scans = pickle.load(f)
            # 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

X7ROOT File Manager

Editing: 002_convert_serialized_scans.py

Upload File

Create Folder