FM / utils

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

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

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
chattr.py	2446	0644	edit dl rm
check_file.py	2486	0644	edit dl rm
cloudways.py	2565	0644	edit dl rm
crontab.py	20802	0644	edit dl rm
endpoints.py	1028	0644	edit dl rm
malware_response.py	16817	0644	edit dl rm
persistent_mrs_queue.py	12028	0644	edit dl rm
revisium.py	4270	0644	edit dl rm
sqlite_limits.py	1689	0644	edit dl rm
submit.py	2152	0644	edit dl rm
user_list.py	13737	0644	edit dl rm
__init__.py	2074	0644	edit dl rm

Edit: /opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/utils/cloudways.py (2565B)

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 pwd
from dataclasses import dataclass
from pathlib import Path
from typing import List, Set

from defence360agent.utils import is_cloudways

@dataclass
class CloudwaysUser:
    """
    Helper to override a `user` for Cloudways environment where a file (db)
    in a customer's directory may belong to another user but should be
    visible for the customer and at the same time it must be cleanable
    (the true `owner` of a file is used for cleaning up)
    """

name: str = None
    uid: int = None

@classmethod
    def get_user_from_path(
        cls,
        path: Path,
        fallback: "CloudwaysUser",
        users_from_panel: Set[str],
        pw_all: List[pwd.struct_passwd],
    ) -> "CloudwaysUser":
        if is_cloudways():
            return next(
                (
                    cls(name=pw.pw_name, uid=pw.pw_uid)
                    for pw in pw_all
                    if pw.pw_name in users_from_panel
                    and Path(pw.pw_dir) in path.parents
                ),
                fallback,
            )
        return fallback

@classmethod
    def override_name_by_path(
        cls,
        path: Path,
        fallback_name: str,
        users_from_panel: Set[str],
        pw_all: List[pwd.struct_passwd],
    ) -> str:
        fallback = cls(name=fallback_name)
        user = cls.get_user_from_path(path, fallback, users_from_panel, pw_all)
        return user.name

@classmethod
    def override_uid_by_path(
        cls,
        path: Path,
        fallback_uid: int,
        users_from_panel: Set[str],
        pw_all: List[pwd.struct_passwd],
    ) -> int:
        fallback = cls(uid=fallback_uid)
        user = cls.get_user_from_path(path, fallback, users_from_panel, pw_all)
        return user.uid

""" 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 pwd from dataclasses import dataclass from pathlib import Path from typing import List, Set from defence360agent.utils import is_cloudways @dataclass class CloudwaysUser: """ Helper to override a `user` for Cloudways environment where a file (db) in a customer's directory may belong to another user but should be visible for the customer and at the same time it must be cleanable (the true `owner` of a file is used for cleaning up) """ name: str = None uid: int = None @classmethod def get_user_from_path( cls, path: Path, fallback: "CloudwaysUser", users_from_panel: Set[str], pw_all: List[pwd.struct_passwd], ) -> "CloudwaysUser": if is_cloudways(): return next( ( cls(name=pw.pw_name, uid=pw.pw_uid) for pw in pw_all if pw.pw_name in users_from_panel and Path(pw.pw_dir) in path.parents ), fallback, ) return fallback @classmethod def override_name_by_path( cls, path: Path, fallback_name: str, users_from_panel: Set[str], pw_all: List[pwd.struct_passwd], ) -> str: fallback = cls(name=fallback_name) user = cls.get_user_from_path(path, fallback, users_from_panel, pw_all) return user.name @classmethod def override_uid_by_path( cls, path: Path, fallback_uid: int, users_from_panel: Set[str], pw_all: List[pwd.struct_passwd], ) -> int: fallback = cls(uid=fallback_uid) user = cls.get_user_from_path(path, fallback, users_from_panel, pw_all) return user.uid