007_revisium_ignore_list.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>
"""
import logging
import os

from peewee import IntegerField, Model, SqliteDatabase, TextField

from imav.migration_utils.other import skip_for_im360
from imav.migration_utils.revisium import find_revisium_db, get_all_domains

logger = logging.getLogger(__name__)

ra = SqliteDatabase(None)

class HashException(ra.Model):
    domain_id = IntegerField()
    file_sub_path = TextField()

class Meta:
        db_table = "hash_exception"

# noinspection PyPep8Naming
class db:
    MalwareIgnorePath: Model

@classmethod
    def init(cls, migrator):
        cls.MalwareIgnorePath = migrator.orm["malware_ignore_path"]

def get_ignore_list(database) -> set[str]:
    ra.init(database)

try:
        domains = get_all_domains()
    except Exception as e:
        logger.warning("Failed to get Plesk domains: %r", e)
        return set()

ignore_list = set()
    for hash_exception in HashException:
        try:
            doc_root = domains[str(hash_exception.domain_id)]["document_root"]
        except KeyError:
            continue

file_sub_path = hash_exception.file_sub_path.lstrip(os.path.sep)
        path = os.path.join(doc_root, file_sub_path)
        ignore_list.add(path)

return ignore_list

@skip_for_im360
def migrate(migrator, database, fake=False, **kwargs):
    if fake:
        return

revisium_db_path = find_revisium_db()
    if revisium_db_path is None:
        logger.info("No legacy ImunifyAV database found. Skipping...")
        return

ignore_list = get_ignore_list(revisium_db_path)

db.init(migrator)
    for path in ignore_list:
        try:
            db.MalwareIgnorePath.create(path=path, resource_type="file")
        except Exception as e:
            logger.warning("Failed to add %r to the ignore list: %r", path, e)

@skip_for_im360
def rollback(migrator, database, fake=False, **kwargs):
    pass

docroot: /home/induvedam/public_html

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

	.	4096
	..	4096
X	000_noop_migration.py	947
X	001_fix_scan_unserialization.py	2923
X	002_convert_serialized_scans.py	2544
X	003_add_new_fields_to_malware_history.py	1323
X	003_whmapi1_set_importance_imav.py	1681
X	004_add_malwarehit_owner.py	1127
X	004_patchman_client.py	1544
X	004_plesk_configs.py	6451
X	004_plesk_eula.py	2386
X	005_plesk_cleanup_storage.py	4111
X	005_populate_malwarehit_owner.py	1142
X	006_add_malwarehit_snippet.py	1137
X	007_add_malwarehistory_fileuser.py	1157
X	007_revisium_ignore_list.py	2660
X	008_populate_malwarehistory_foleuser.py	1185
X	008_subscription_permissions.py	3371
X	009_revisium_scan_history.py	9208
X	010_add_malwarehistory_scan_id.py	1159
X	011_add_malwarehistory_index.py	1527
X	012_add_rescan_outdated_scan_type.py	1432
X	013_add_malware_ctime.py	1224
X	014_add_imunify_security.py	986
X	015_manually_deleted_at_field_added.py	989
X	016_add_vulnerability_hit_table.py	2539
X	017_add_imunify_patch_subscriptions_table.py	1325
X	017_add_version_field.py	988
X	018_delete_suspicious_hits.py	977
X	019_add_malware_history_signature_id_field.py	1169
X	020_add_default_plesk_settings_av.py	1784
X	021_remove_cron_scan_file.py	1091
X	022_scan_queue_to_json.py	1612
X	023_create_mrs_upload_queue.py	1832
X	024_create_mrs_queue_counters_and_meta.py	1332
X	025_cluster_tenant_prefix_paths.py	3274
X	026_add_malware_history_path_index.py	1935
X	027_purge_unnormalized_ignore_paths.py	2808
X	028_add_malware_hits_user_index.py	1216
X	__init__.py	0
	__pycache__	8192