mitm_proxy.py

import subprocess
import threading

from networking_functions import get_local_ip
from packet_analysis import print_creds_from_packet
from colorama import init, Fore

init()
GREEN = Fore.GREEN
BLUE = Fore.BLUE
RED = Fore.RED
RESET = Fore.RESET

def start_mitm_proxy(printing_queue, verbosity, cancel_token):

    command = f"./assets/binaries/mitmdump --listen-host {get_local_ip()} --listen-port 8080 --flow-detail 4"
    proc = subprocess.Popen(command.split(), stdout=subprocess.PIPE)

    printing_queue.put(f"{GREEN}[+] Mitm proxy thread started{RESET}")
    while not cancel_token.is_set():
        if verbosity>0:
            if proc.stdout.readline() != b'':
                print_creds_from_packet(proc.stdout.readline().decode("UTF-8"), printing_queue)
                printing_queue.put(proc.stdout.readline().decode("UTF-8"))
    proc.kill()
    printing_queue.put(f"{GREEN}[+] Stopping http server")



def start_mitm_proxy_thread(printing_queue, verbosity):
    cancel_token = threading.Event()
    mitm_proxy_thread = threading.Thread(target=start_mitm_proxy, args=(printing_queue, verbosity, cancel_token))
    mitm_proxy_thread.start()
    return cancel_token