from weevely.core.loggers import log
from weevely.core.module import Module
from weevely.core.vectors import ModuleExec


class Filesystem(Module):
    """Audit the file system for weak permissions."""

    def init(self):
        self.register_info({"author": ["Emilio Pinna"], "license": "GPLv3"})

        self.check_functions = [f for f in dir(self) if f.startswith("check_")]

        self.register_arguments([{"name": "check", "choices": self.check_functions, "nargs": "?"}])

    def check_writable_binaries(self):
        """Search writable files in binary folders"""

        results = []

        for path in [
            "/bin/",
            "/usr/bin/",
            "/usr/sbin",
            "/sbin",
            "/usr/local/bin",
            "/usr/local/sbin",
            "/lib/",
            "/usr/lib/",
            "/usr/local/lib",
        ]:
            result = ModuleExec("file_find", ["-writable", path]).run()

            if result and any(r for r in result if r):
                results += result

        return results

    def check_writable_etc(self):
        """Search writable files in etc folder"""

        result = ModuleExec("file_find", ["-writable", "/etc/"]).run()

        if result and any(r for r in result if r):
            return result

    def check_writable_root(self):
        """Search writable files in / folder"""

        result = ModuleExec("file_find", ["-no-recursion", "-writable", "/"]).run()

        if result and any(r for r in result if r):
            return result

    def check_home_writable(self):
        """Search writable files in /home/ folder"""

        result = ModuleExec("file_find", ["-no-recursion", "-writable", "/home/"]).run()

        if result and any(r for r in result if r):
            return result

    def check_spool_crons(self):
        """Search writable files in /var/spool/cron/ folder"""

        result = ModuleExec("file_find", ["-writable", "/var/spool/cron/"]).run()

        if result and any(r for r in result if r):
            return result

    def check_home_executable(self):
        """Search executable files in /home/ folder"""

        result = ModuleExec("file_find", ["-no-recursion", "-executable", "/home/"]).run()

        if result and any(r for r in result if r):
            return result

    def check_readable_etc(self):
        """Search certain readable files in etc folder"""

        readable_files = ModuleExec("file_find", ["-readable", "/etc/"]).run()

        files_paths = ["shadow", "ap-secrets", "mysql/debian.cnf", "sa_key$", "keys", r"\.gpg", "sudoers"]

        readable_files = ModuleExec("file_find", ["-readable", "/etc/"]).run()

        files_paths = ["shadow", "ap-secrets", "mysql/debian.cnf", "sa_key$", "keys", r"\.gpg", "sudoers"]

        return [f for f in readable_files if f and any(p for p in files_paths if p and p in f)]

    def check_readable_logs(self):
        """Search certain readable log files"""

        readable_files = ModuleExec("file_find", ["-readable", "/var/log/"]).run()

        files_paths = ["lastlog", "dpkg", "Xorg", "wtmp", "pm", "alternatives", "udev", "boot"]

        return [
            f
            for f in readable_files
            if f and not f.endswith("gz") and not f.endswith("old") and any(p for p in files_paths if p and p in f)
        ]

    def run(self, **kwargs):
        results = {}

        for func_name in [
            # Execute every function starting with check_*
            fn
            for fn in self.check_functions
            # if the user does not specify any name
            if not self.args.get("check")
            # of if specify the current function name
            or self.args.get("check") == fn
        ]:
            function = getattr(self, func_name)
            log.warn(function.__doc__)

            result = function()

            if result:
                log.info("\n".join(result))
                results.update({func_name: result})

        return results

    def print_result(self, result):
        pass