import os
import subprocess

def check_kernel_version():
    """
    Coleta informações sobre a versão do Kernel e arquitetura.
    """
    print("[*] Verificando versão do Kernel...")
    try:
        kernel = subprocess.check_output("uname -a", shell=True).decode().strip()
        print(f"[+] Kernel Info: {kernel}")
    except Exception as e:
        print(f"[!] Não foi possível obter o kernel: {e}")

def find_suid_files():
    """
    Busca por arquivos com o bit SUID definido.
    """
    print("\n[*] Buscando por arquivos com permissão SUID...")
    print("[*] Isso pode levar alguns segundos...")

    # Comando find para buscar arquivos com permissão 4000 (SUID)
    # 2>/dev/null descarta erros de permissão negada durante a busca
    command = "find / -perm -4000 -type f 2>/dev/null | head -n 20"

    try:
        suid_files = subprocess.check_output(command, shell=True).decode().strip()
        if suid_files:
            print("[!] Binários SUID encontrados (Top 20):")
            print(suid_files)
        else:
            print("[-] Nenhum binário SUID incomum detectado.")
    except Exception:
        print("[!] Erro ao executar a busca por SUID.")

def check_writable_files():
    """
    Verifica se arquivos críticos são editáveis pelo usuário atual.
    """
    critical_files = ["/etc/passwd", "/etc/shadow", "/etc/sudoers"]
    print("\n[*] Verificando permissões em arquivos críticos...")

    for f in critical_files:
        if os.access(f, os.W_OK):
            print(f"[!!!] PERIGO: O arquivo {f} é EDITÁVEL por você!")
        else:
            print(f"[-] {f:12}: Apenas leitura ou sem acesso.")

def main():
    print("=" * 60)
    print("Módulo: Privilege Escalation Checker | Projeto: 30 Lessons")
    print("=" * 60)

    # Verifica se estamos em um sistema Linux
    if os.name != 'posix':
        print("[!] Este módulo foi desenhado para ambientes Linux/Unix.")
        return

    check_kernel_version()
    check_writable_files()
    find_suid_files()

    print("\n" + "=" * 60)
    print("[*] Auditoria de privilégios finalizada.")

if __name__ == "__main__":
    main()