run.py

import subprocess
import sys
import time
import signal

import userinteraction
import config

# Get configuration variables
ip = config.ip
package_name = config.package_name

# Store subprocesses to kill them all on Ctrl+C
processes = []  


def kill_handler(sig, frame):
    """Handler for Ctrl+C: terminates all subprocesses."""
    
    print("\n[CTRL+C detected] Closing all subprocesses and reversing global proxy...")

    # Logout
    userinteraction.logout()

    # Close app
    subprocess.run(f"adb shell am force-stop {package_name}")

    # Reverse global proxy
    subprocess.run("adb shell settings put global http_proxy :0")

    for p in processes:
        try:
            p.terminate()  # Try graceful shutdown
        except:
            pass

    time.sleep(1)

    for p in processes:
        try:
            p.kill()  # Force kill if needed
        except:
            pass

    print("All subprocesses closed.")
    sys.exit(0)


def main():
    """Main execution: starts all tools and handles user interaction."""
    
    # Register Ctrl+C handler
    signal.signal(signal.SIGINT, kill_handler)  

    # Start global proxy to capture traffic
    print("[INFO] Changing proxy to mitm address...")
    subprocess.run(f"adb shell settings put global http_proxy {ip}:8080")
    
    # Start screen mirroring
    print("[INFO] Starting screen mirroring...")
    screen = subprocess.Popen("programs\scrcpy-win64-v3.3.3\scrcpy.exe", creationflags=subprocess.CREATE_NEW_CONSOLE)
    processes.append(screen)

    # Start mitmproxy web interface
    print("[INFO] Starting mitmproxy web interface...")
    mitmproxy = subprocess.Popen("mitmweb",creationflags=subprocess.CREATE_NEW_CONSOLE)
    processes.append(mitmproxy)

    # # Start Frida certificate unpinning
    # print("[INFO] Starting Frida certificate unpinning script...")
    # fridaunpin = subprocess.Popen(f"frida --codeshare akabe1/frida-multiple-unpinning -U -f {package_name}",creationflags=subprocess.CREATE_NEW_CONSOLE)
    # processes.append(fridaunpin)

    # Start app
    subprocess.run(f"frida -U -f {package_name} -l .\\fridascript.js")


    # Wait for the app to load
    print("[INFO] Waiting for the app to start...")
    time.sleep(15)

    print("[INFO] All services started. Press Ctrl+C to exit.")

    # Run user interaction logic
    print("[INFO] Running user interaction logic...")
    userinteraction.login()

    # Do some user interactions
    time.sleep(1)
    print("[INFO] Guided app traverse...")
    userinteraction.traverseApp()
    time.sleep(1)
    print("[INFO] Random app traverse...")
    userinteraction.randomthings(123, 50)

    # Keep the script alive to catch Ctrl+C
    while True:
        time.sleep(1)


if __name__ == "__main__":
    main()