fritzflash.py

#! /usr/bin/env python3
import argparse
import platform
import random
import socket
import sys
import time
from contextlib import contextmanager
from ftplib import FTP
from ipaddress import (
    AddressValueError,
    IPv4Address,
    IPv4Interface,
    IPv6Address,
    IPv6Interface,
    ip_address,
    ip_interface,
)
from pathlib import Path
from subprocess import run, CalledProcessError
from typing import Generator, List, Literal, Tuple, Union

UDP_IP = "0.0.0.0"
UDP_PORT = 69  # TFTP Protocol Port (69)

SOCK_TIMEOUT = 5
MAX_TIMEOUT_RETRIES = 5
SESSIONS: dict = dict()
FILE_DIR = Path(__file__)

# header opcode is 2 bytes
TFTP_OPCODES = {1: "RRQ", 2: "WRQ", 3: "DATA", 4: "ACK", 5: "ERROR"}

TRANSFER_MODES = ["netascii", "octet"]  # "mail" is deprecated
TRANSFER_MODES_T = Literal["netascii", "octet", "mail"]

TFTP_ERRORS = {
    0: "Not Defined",
    1: "File Not Found",
    2: "Access Violation",
    3: "Disk Full or Allocation Exceeded",
    4: "Illegal TFTP operation",
    5: "Unknown Transfer TID",
    6: "File Already Exists",
    7: "No Such User",
}
TFTP_ERRORS_T = Literal[0, 1, 2, 3, 4, 5, 6, 7]


def create_data_packet(block, file: Path, mode: TRANSFER_MODES_T) -> bytes:
    data = bytearray()
    # append data opcode (03)
    data.append(0)
    data.append(3)

    # append block number (2 bytes)
    data.append(int(block / 256))
    data.append(int(block % 256))

    # append data (512 bytes max)
    content = read_file(block, file)
    data += content

    return bytes(data)


def create_ack_packet(block) -> bytes:
    ack = bytearray()
    # append acknowledgement opcode (04)
    ack.append(0)
    ack.append(4)

    # append block number (2 bytes)
    ack.append(int(block / 256))
    ack.append(int(block % 256))

    return bytes(ack)


def create_error_packet(error_code: TFTP_ERRORS_T) -> bytes:
    err = bytearray()
    # append error opcode (05)
    err.append(0)
    err.append(5)

    # append error code
    ec = f"{error_code:02}"
    err.append(int(ec[0]))
    err.append(int(ec[1]))

    # append error message followed by null terminator
    msg = bytearray(TFTP_ERRORS[error_code].encode("utf-8"))
    err += msg
    err.append(0)

    return bytes(err)


def read_file(block: int, file: Path) -> bytes:
    offset = (block - 1) * 512
    with file.open("rb") as f:
        f.seek(offset, 0)
        content = f.read(512)
    return content


# Get opcode from TFTP header
def get_opcode(bytes):
    opcode = int.from_bytes(bytes[0:2], byteorder="big")
    if opcode not in TFTP_OPCODES.keys():
        return False
    return TFTP_OPCODES[opcode]


# Return filename and mode from decoded RRQ/WRQ header
def decode_request_header(data) -> Tuple[Path, TRANSFER_MODES_T]:
    header = data[2:].split(b"\x00")
    file = Path(header[0].decode("utf-8"))
    mode = header[1].decode("utf-8").lower()
    return file, mode


# Find a random port between 1025 and 65535 that is not in use
# by this service
def get_random_port() -> int:
    while True:
        port = random.randint(1025, 65536)
        if port not in SESSIONS.keys():
            return port


@contextmanager
def create_udp_socket(ip=UDP_IP, port=UDP_PORT) -> Generator[socket.socket, None, None]:
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # Internet  # UDP
    sock.bind((ip, port))
    try:
        yield sock
    finally:
        _, port = sock.getsockname()
        if port in SESSIONS:
            del SESSIONS[port]
        sock.close()


def serve(port: int, file: Path, mode: TRANSFER_MODES_T) -> Tuple[bool, str]:
    session = SESSIONS[port]

    with create_udp_socket(port=port) as sock:
        block = 1
        packet = create_data_packet(block, file, mode)
        size = file.stat().st_size // 512

        sock.sendto(packet, session["addr"])

        SESSIONS[port]["packet"] = packet
        while True:
            try:
                sock.settimeout(SOCK_TIMEOUT)
                data, addr = sock.recvfrom(1024)  # buffer size is 1024 bytes
                session["consec_timeouts"] = 0

                # Check address (IP, port) matches initial connection address
                if addr != session["addr"]:
                    sock.sendto(create_error_packet(5), addr)

                opcode = get_opcode(data)
                if opcode == "ACK":
                    len_block_ack = int.from_bytes(data[2:4], byteorder="big")
                    len_last_sent = int.from_bytes(
                        session["packet"][2:4], byteorder="big"
                    )
                    # Check length of the last DATA packet sent, if the Datagram length is
                    # less than 516 it is the last packet. Upon receiving the final ACK packet
                    # from the client we can terminate the connection.
                    if len(session["packet"]) < 516 and len_block_ack == len_last_sent:
                        return True, addr
                    block += 1
                    if block % 256 == 0:
                        print(f"\r{block} / {size}", end="")

                    packet = create_data_packet(block, file, mode)
                    sock.sendto(packet, addr)
                    session["packet"] = packet
                else:
                    # Threads only handle incoming packets with ACK opcodes, send
                    # 'Illegal TFTP Operation' ERROR packet for any other opcode.
                    sock.sendto(create_error_packet(4), addr)
            except socket.timeout:
                if session["consec_timeouts"] < MAX_TIMEOUT_RETRIES:
                    session["consec_timeouts"] += 1
                    sock.sendto(session["packet"], session["addr"])
                else:
                    return False, addr


def serve_file(
    file: Path, serve_name: str = "FRITZ7530.bin"
) -> Generator[Tuple[bool, str], None, None]:
    file = Path(file)
    with create_udp_socket(port=69) as server_sock:
        print(
            f"TFTP Server started listening on Port 69, serving {file} under name {serve_name}"
        )

        # yield file transfers until
        while True:
            data, addr = server_sock.recvfrom(1024)
            opcode = get_opcode(data)
            # Only serving read operations
            if opcode != "RRQ":
                server_sock.sendto(create_error_packet(4), addr)
                print(f"wrong opcode: {opcode} was requested")
                continue
            rfile, mode = decode_request_header(data)
            # Mail is deprecated
            if mode not in TRANSFER_MODES:
                server_sock.sendto(create_error_packet(0), addr)
                print(f"unsupported mode: {mode} was requested")
                continue
            # Only serving a specific file
            print(f"{rfile} was requested")
            if rfile.name.strip() != serve_name:
                server_sock.sendto(create_error_packet(1), addr)
                continue

            port = get_random_port()
            SESSIONS[port] = {"addr": addr, "packet": None, "consec_timeouts": 0}
            yield serve(port, file, mode)


def main_tftp():
    for success, host in serve_file(sys.argv[1]):
        if success:
            print(f"Successfully served file {sys.argv[1]} to Host {host}")
            try:
                input(
                    "Press enter key to serve to another Host, CTRL-c or CTRL-d to stop TFTP server."
                )
            except (KeyboardInterrupt, EOFError):
                break
        else:
            print(f"Timeout serving file to Host {host}")


IPInterface = Union[IPv4Interface, IPv6Interface]
IPAddress = Union[IPv4Address, IPv6Address]

AUTODISCOVER_TIMEOUT = 1
FTP_TIMEOUT = 2
FTP_MAX_RETRY = 10

INITRAMFS_BOOT_TIMEOUT = 60  # in seconds
IS_POSIX = platform.system() in ["Linux", "Darwin", "FreeBSD"]


class FritzFTP(FTP):
    class ConnectionTimeout(Exception):
        pass

    def __init__(
        self,
        ip,
        username="adam2",
        password="adam2",
        timeout=1,
        max_retry=0,
        retry_cb=None,
    ):
        i = 1
        while i <= max_retry:
            try:
                retry_cb(i, max_retry)
                super().__init__(ip, user=username, passwd=password, timeout=timeout)
                break
            except socket.timeout:
                i += 1
            except OSError:
                time.sleep(1)
                i += 1
        if i > max_retry:
            raise FritzFTP.ConnectionTimeout()
        self.set_pasv(True)

    def getenv(self):
        env = [b""]
        fritzenv = {}

        def storeenv(x):
            env[0] += x

        self.voidcmd("MEDIA SDRAM")
        try:
            self.retrbinary("RETR env", storeenv)
        except socket.timeout:
            pass

        for line in env[0].decode("ascii").splitlines():
            l = line.split()
            if len(l) < 2:
                print(f"'{l}' is not a tuple, ignoring")
                # after using the fritz recovery tool, my FB7530 had
                # a ['ptest'] entry, without a value..
                continue
            fritzenv[l[0]] = l[1]

        return fritzenv

    def set_flash_timeout(self):
        self.sock.settimeout(60 * 5)

    def upload_image(self, image):
        self.set_flash_timeout()
        self.voidcmd("MEDIA FLSH")
        with image.open("rb") as file:
            self.storbinary("STOR mtd1", file)

    def reboot(self):
        self.voidcmd("REBOOT")
        self.close()


@contextmanager
def set_ip(
    ipinterface: IPInterface, network_device: str
) -> Generator[bool, None, None]:
    if IS_POSIX:
        start_command = [
            "ip",
            "addr",
            "add",
            ipinterface.with_prefixlen,
            "dev",
            network_device,
        ]
        shutdown_command = [
            "ip",
            "addr",
            "delete",
            ipinterface.with_prefixlen,
            "dev",
            network_device,
        ]
    else:
        start_command = [
            "netsh",
            "interface",
            "ipv4",
            "add",
            "address",
            f"{network_device}",
            f"{ipinterface.ip}",
            f"{ipinterface.netmask}",
        ]
        shutdown_command = [
            "netsh",
            "interface",
            "ipv4",
            "delete",
            "address",
            f"{network_device}",
            f"{ipinterface.ip}",
        ]

    output = run(
        start_command,
        capture_output=True,
    )
    try:
        if IS_POSIX:
            yield output.returncode in [0, 2]
        else:
            # sleep is sometimes required on windows
            # just give the network adapter some time
            time.sleep(5)
            yield output.returncode == 0
    finally:
        run(shutdown_command)


def await_online(host: IPAddress):
    returncode = -1
    while returncode != 0:
        response = run(
            ["ping", "-c", "1", "-w", str(INITRAMFS_BOOT_TIMEOUT), str(host)],
            capture_output=True,
        )
        returncode = response.returncode
        print(".", end="")
    return returncode


def scp_legacy_check() -> bool:
    """Since OpenSSH release 9.0, scp uses the sftp protocol by default which
    is known to be incompatible with the uboot present on stock 7530/7520"""
    response = run(["ssh", "-V"], capture_output=True)
    version_string = response.stderr.decode().strip()
    ssh_ver, ssl_ver = version_string.split(",")
    ssh_ver = ssh_ver.strip("OpenSSH_for_Windows_")
    ssh_ver = ssh_ver.strip("OpenSSH_")
    major = int(ssh_ver.split(".")[0])
    return major >= 9


def ssh(host: IPAddress, cmd: List[str], user: str = "root"):
    null_file = "/dev/null" if IS_POSIX else "NUL"
    args = [
        "-o",
        "StrictHostKeyChecking=no",
        "-o",
        f"UserKnownHostsFile={null_file}",
        "-o",
        "HostKeyAlgorithms=+ssh-rsa",
    ]
    run(["ssh", *args, f"{user}@{host}", *cmd], capture_output=True).check_returncode()


def scp(host: IPAddress, file: Path, user: str = "root", target: Path = Path("/tmp/")):
    null_file = "/dev/null" if IS_POSIX else "NUL"
    args = ["-o", "StrictHostKeyChecking=no", "-o", f"UserKnownHostsFile={null_file}"]
    if scp_legacy_check():
        args.append("-O")
    run(["scp", *args, file.name, f"{user}@{host}:{target}"]).check_returncode()


def connection_refused_message():
    print(
        "\nIt seems you have a booted-up AVM device running in your Network.\n"
        "This might be because you missed the 10 second window after powering on your AVM device.\n"
        "In this case: Powercycle your device and retry.\n"
        "If this problem persits, check if you might have connections to another AVM device, e.g. via WiFi/WLAN.\n\n"
    )


def start_message():
    print(
        "This program will help you installing OpenWRT or Gluon, a widely used Firmware for Freifunk networks, onto your AVM device.\n"
        "You can always find the most current version of this script at https://www.github.com/freifunk-darmstadt/fritz-tools\n\n"
        "It is strongly recommended to only connect your computer to the device you want to flash.\n"
        "Try to disable all other connections (Ethernet, WiFi/WLAN, VMs) if detection fails.\n"
        "Sometimes an unmanaged switch between your AVM device and your computer is helpful."
    )


def connect_message():
    print("We will now connect to your devices bootloader.")


def flash_message():
    print(
        "\nWriting Gluon image to your AVM device...\n"
        "This process may take a lot of time.\n\n"
        "First, the device will erase its current Operating System.\n"
        "Next, the device will write the Gluon image to its memory.\n"
        "The red Info LED will illuminate in this step. Don't worry, this is expected behavior.\n\n"
        "Do *not* turn off the device!\n\n"
        "We will tell you when your device has finished installing Gluon (this may take a while)."
    )


def finish_message():
    print(
        "\n== Congratulations! ==\n\n"
        "Your device is now running Gluon.\n"
        "It will restart and in 2-5 minutes you will be able to visit its config-mode.\n"
        "Remember to reconfigure your interface to automatically obtain an IP-address!\n"
        "You can reach config-mode by typing in http://192.168.1.1/ in your preferred Webbrowser.\n"
    )
    print("Press enter to exit.")
    input()


def retry_status(current_try, max_try):
    print("--> Try %d of %d" % (current_try, max_try))


def autodiscover_avm_ip():
    sender = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    try:
        sender.bind(("192.168.178.2", 0))
    except OSError as e:
        if e.errno == 99:
            print("\rIP address 192.168.178.2 is not configured on any interface.")
            exit(1)
        else:
            raise e from None
    sender.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    sender.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
    sender.settimeout(0.5)

    receiver = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    receiver.settimeout(AUTODISCOVER_TIMEOUT)
    receiver.bind(("192.168.178.2", 5035))

    i = 1
    while True:
        try:
            print("\rTry %d..." % i, end="")
            receiver.sendto(
                b"aa", ("192.168.178.1", 5035)
            )  # Dirty hack to add conntrack entry
            sender.sendto(
                bytearray.fromhex("0000120101000000c0a8b20100000000"),
                ("255.255.255.255", 5035),
            )
            while True:
                data, addr = receiver.recvfrom(64)
                if addr[0] == "192.168.178.1":
                    print("\rFritzBox found at %s" % addr[0])
                    return addr[0]
        except socket.timeout:
            i += 1
        except OSError:
            i += 1
            time.sleep(0.1)
        except KeyboardInterrupt:
            print("\rAborting...\n")
            return None

    return None


def determine_image_name(env_string):
    models = {
        "172": {
            "gluon": ["avm-fritz-box-7320-sysupgrade.bin"],
            "openwrt": ["openwrt-lantiq-xway-avm_fritz7320-squashfs-sysupgrade.bin"],
        },
        "173": {
            "gluon": ["avm-fritz-wlan-repeater-300e-sysupgrade.bin"],
            "openwrt": ["openwrt-ath79-generic-avm_fritz300e-squashfs-sysupgrade.bin"],
        },
        "179": {
            "gluon": ["avm-fritz-box-7330-sysupgrade.bin"],
            "openwrt": ["openwrt-lantiq-xway-avm_fritz7330-squashfs-sysupgrade.bin"],
        },
        "181": {
            "gluon": ["avm-fritz-box-7360-sl-sysupgrade.bin"],
            "openwrt": [
                "openwrt-lantiq-xrx200-avm_fritz7360sl-squashfs-sysupgrade.bin"
            ],
        },
        "183": {
            "gluon": ["avm-fritz-box-7360-v1-sysupgrade.bin"],
            "openwrt": [
                "openwrt-lantiq-xrx200-avm_fritz7360-v1-squashfs-sysupgrade.bin"
            ],
        },
        "188": {
            "gluon": ["avm-fritz-box-7330-sl-sysupgrade.bin"],
            "openwrt": ["openwrt-lantiq-xway-avm_fritz7320-squashfs-sysupgrade.bin"],
        },
        "189": {
            "gluon": ["avm-fritz-box-7312-sysupgrade.bin"],
            "openwrt": ["openwrt-lantiq-xway-avm_fritz7312-squashfs-sysupgrade.bin"],
        },
        "193": {
            "openwrt": ["openwrt-lantiq-xrx200-avm_fritz3390-initramfs-kernel.bin"],
        },
        "196": {
            "gluon": ["avm-fritz-box-7360-v2-sysupgrade.bin"],
            "openwrt": [
                "openwrt-lantiq-xrx200-avm_fritz7360-v2-squashfs-sysupgrade.bin"
            ],
        },
        "200": {
            "gluon": ["avm-fritz-wlan-repeater-450e-sysupgrade.bin"],
            "openwrt": ["openwrt-ath79-generic-avm_fritz450e-squashfs-sysupgrade.bin"],
        },
        "203": {
            "openwrt": ["openwrt-lantiq-xrx200-avm_fritz7362sl-initramfs-kernel.bin"]
        },
        "206": {
            "gluon": ["avm-fritz-wlan-repeater-1750e-sysupgrade.bin"],
            "openwrt": ["openwrt-ath79-generic-avm_fritz1750e-squashfs-sysupgrade.bin"],
        },
        "209": {
            "openwrt": ["openwrt-lantiq-xrx200-avm_fritz7412-initramfs-kernel.bin"]
        },
        "218": {
            "openwrt": ["openwrt-lantiq-xrx200-avm_fritz7430-initramfs-kernel.bin"]
        },
        "219": {
            "gluon": ["avm-fritz-box-4020-sysupgrade.bin"],
            "openwrt": [
                "fritz4020-squashfs-sysupgrade.bin",
                "openwrt-ath79-generic-avm_fritz4020-squashfs-sysupgrade.bin",
            ],
        },
        "227": {
            "gluon": ["avm-fritz-box-4040-bootloader.bin"],
            "openwrt": ["openwrt-ipq40xx-generic-avm_fritzbox-4040-squashfs-eva.bin"],
        },
        "236": {"openwrt": ["uboot-fritz7530.bin"]},
        "244": {"openwrt": ["uboot-fritz1200.bin"]},
        "246": {"openwrt": ["uboot-fritz3000.bin"]},
        "247": {
            # fritzbox 7520
            "openwrt": ["uboot-fritz7530.bin"]
        },
    }
    for model in models.keys():
        if model == env_string:
            image_names = []
            if "gluon" in models[model]:
                image_names += models[model]["gluon"]
            if "openwrt" in models[model]:
                image_names += models[model]["openwrt"]
            return image_names
    return None


def autoload_image(ip):
    print("\nStarting automatic image-selection!")
    print("-> Establishing connection to device!")

    try:
        ftp = FritzFTP(
            ip, timeout=FTP_TIMEOUT, max_retry=FTP_MAX_RETRY, retry_cb=retry_status
        )
    except FritzFTP.ConnectionTimeout:
        print("-> Max retrys exceeded! Check connection and try again.")
        exit(1)
    except ConnectionRefusedError:
        connection_refused_message()
        exit(1)

    env = ftp.getenv()
    ftp.close()

    if "HWRevision" not in env:
        print("\nAutomatic image-selection unsuccessful!")
        print("-> No model saved on device!")
        exit(1)

    image_names = determine_image_name(env["HWRevision"])

    if image_names is None:
        print("\nAutomatic image-selection unsuccessful!")
        print("-> Unknown Model %s!" % env["HWRevision"])
        print("Press enter to exit.")
        input()
        exit(1)

    cwd = Path()
    files = []
    for file in cwd.iterdir():
        if not file.is_file():
            continue
        for image_name in image_names:
            if image_name in file.name:
                files.append(file)

    if len(files) > 1:
        print("\nAutomatic image-selection unsuccessful!")
        print("-> Multiple potential image files found!")
        for file in files:
            print("--> %s" % file)
        print("\nPlease specify the image via `--image` parameter.")
        print("Press enter to exit.")
        input()
        exit(1)

    if not files:
        print("\nAutomatic image-selection unsuccessful!")
        print("--> No potential image file found!")
        print(f"Did not find any file of {image_names}")
        print("\nPlease download and specify the image via `--image` parameter.")
        print("Press enter to exit.")
        input()
        exit(1)

    print("-> Automatic image-selection successful!")
    print("--> Will flash %s" % files[0])

    return files[0], env["HWRevision"]


def perform_flash(ip, file):
    print("-> Establishing connection to device!")

    try:
        ftp = FritzFTP(
            ip, timeout=FTP_TIMEOUT, max_retry=FTP_MAX_RETRY, retry_cb=retry_status
        )
    except FritzFTP.ConnectionTimeout:
        print("-> Max retries exceeded! Check connection and try again.")
        print("Press enter to exit.")
        input()
        exit(1)
    except ConnectionRefusedError:
        connection_refused_message()
        print("Press enter to exit.")
        input()
        exit(1)

    print("-> Flash image")

    # if uboot or initramfs in file.name?
    if file.name in [
        "uboot-fritz7520.bin",
        "uboot-fritz7530.bin",
        "uboot-fritz1200.bin",
        "uboot-fritz3000.bin",
        "openwrt-lantiq-xrx200-avm_fritz7412-initramfs-kernel.bin",
        "openwrt-lantiq-xrx200-avm_fritz7362sl-initramfs-kernel.bin",
        "openwrt-lantiq-xrx200-avm_fritz7430-initramfs-kernel.bin",
        "openwrt-lantiq-xrx200-avm_fritz3390-initramfs-kernel.bin",
    ]:
        size = file.stat().st_size
        assert size < 0x2000000

        if file.name in [
            "uboot-fritz7520.bin",
            "uboot-fritz7530.bin",
            "uboot-fritz1200.bin",
            "uboot-fritz3000.bin",
        ]:
            addr = size
            haddr = 0x85000000
        else:
            addr = (0x8000000 - size) & ~0xFFF
            haddr = 0x80000000 + addr

        with file.open("rb") as img:
            # The following parameters allow booting the avm recovery system with this
            # script.
            if file.name in [
                "openwrt-lantiq-xrx200-avm_fritz7412-initramfs-kernel.bin",
                "openwrt-lantiq-xrx200-avm_fritz7430-initramfs-kernel.bin",
                "openwrt-lantiq-xrx200-avm_fritz3390-initramfs-kernel.bin",
            ]:
                ftp.voidcmd("SETENV linux_fs_start 0")
            ftp.voidcmd("SETENV memsize 0x%08x" % (addr))
            ftp.voidcmd("SETENV kernel_args_tmp mtdram1=0x%08x,0x88000000" % (haddr))
            ftp.voidcmd("MEDIA SDRAM")
            try:
                ftp.storbinary("STOR 0x%08x 0x88000000" % (haddr), img)
            except TimeoutError:
                print("storbinary timeout - expected on initramfs-kernel flash")
        print("-> Image write successful")
    else:
        flash_message()
        ftp.upload_image(file)
        print("-> Image write successful")
        print("-> Performing reboot")
        ftp.reboot()


def perform_bootloader_flash(
    sysupgradefile: Path,
    imagefile: Path = None,
    flash_tftp: bool = None,
):
    with set_ip(ip_interface("192.168.1.70/24"), args.device) as can_set_ip:
        if not can_set_ip:
            print("could not set ip to 192.168.1.70/24")
            print(
                "please set the IP of your network adapter to this adress and press enter to continue or press CTRL+C to abort and re-run with admin rights"
            )
            input()
        target_host = ip_address("192.168.1.1")
        print(f"Waiting for Host to come up with IP Adress {target_host} ...")
        await_online(target_host)
        print("-> Host online.\nWait to finish boot")
        time.sleep(15)
        print("Transfering sysupgrade target firmware")
        scp(target_host, sysupgradefile)
        if flash_tftp:
            print("-> Transfering bootloader")
            scp(target_host, imagefile)
            print("Writing Bootloader")
            ssh(
                target_host,
                [
                    "mtd",
                    "write",
                    f"/tmp/{imagefile}",
                    "uboot0",
                    "&&",
                    "mtd",
                    "write",
                    f"/tmp/{imagefile}",
                    "uboot1",
                ],
            )
            ssh(
                target_host,
                [
                    "ubirmvol",
                    "/dev/ubi0",
                    "--name=avm_filesys_0",
                    "&&",
                    "ubirmvol",
                    "/dev/ubi0",
                    "--name=avm_filesys_1",
                ],
            )
        print("Executing Sysupgrade")
        try:
            ssh(target_host, ["sysupgrade", "-n", f"/tmp/{sysupgradefile.name}"])
        except CalledProcessError as e:
            if e.returncode == 246:
                # ssh disconnects
                print("Sysupgrade successful, device reboots")
            else:
                print(f"Error: {e}")


def perform_tftp_flash(initramfsfile: Path, sysupgradefile: Path, serve_name: str):
    with set_ip(ip_interface("192.168.1.70/24"), args.device) as can_set_ip:
        if not can_set_ip:
            print("could not set ip to 192.168.1.70/24")
            print(
                "please set the IP of your network adapter to this adress and press enter to continue or press CTRL+C to abort and re-run with admin rights"
            )
            input()
        success = False
        while not success:
            success, host = next(serve_file(initramfsfile, serve_name))
        print(f"-> Transfered initramfs image to {host}.")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Flash Gluon image to AVM devices using EVA or/and TFTP."
    )
    parser.add_argument(
        "--image",
        type=str,
        help="(uboot) image file to transfer. Autodiscovery if not specified.",
    )
    parser.add_argument(
        "--initramfs",
        type=str,
        help="Compatible openwrt initramfs image file for uboot+TFTP flash.",
    )
    parser.add_argument(
        "--sysupgrade",
        type=str,
        help="Target system image file, the operating system for TFTP flash.",
    )
    parser.add_argument(
        "--device",
        "-dev",
        type=str,
        help="Name of the Ethernet adapter (look it up ie by 'ip link')",
        default="",
    )
    args = parser.parse_args()

    # target_host = ip_address("192.168.1.1")
    # print(f"Waiting for Host to come up with IP Adress {target_host} ...")
    # await_online(target_host)
    # scp(target_host, Path(args.sysupgrade))

    tftp_serve_name = ""
    ramfsfile = None
    sysupgradefile = None
    imagefile = None
    if args.image:
        imagefile = Path(args.image)
        if not imagefile.is_file():
            print(f'Image file "{imagefile.absolute()}" does not exist!')
            exit(1)
        print(
            "If this device needs TFTP flashing, enter the serve name (e.g. FRITZ7520.bin) - else leave empty"
        )
        tftp_serve_name = input().strip()

    start_message()

    client_intf = ip_interface("192.168.178.2/24")
    with set_ip(client_intf, args.device) as can_set_ip:
        if can_set_ip:
            print(f"did set ip to {client_intf}")
        else:
            print(f"could not set ip to {client_intf}")
            print(
                "If you run this program with according permission, it will configure IPs on your host automatically.\n"
                "Otherwise, make sure you have assigned your PC a static IP Address in the Subnet of the device you want to flash.\n"
                "The following example would be a completely fine option:\n"
            )
            print(f"IP-Address: {client_intf.ip}")
            print(f"Subnet: {client_intf.netmask}")
            print("Gateway: 192.168.178.1")
            print("DNS Servers: Leave blank\n")
            print(
                "Once you're ready to flash, press enter, disconnect power from your AVM device and reconnect the power-supply."
            )
            input()

        if "ip" in args:
            try:
                ip = ip_address(args.ip)
            except AddressValueError:
                print(f"{args.ip} is not a valid IPv4 address!")
                exit(1)
        else:
            print("Trying to autodiscover! Abort via Ctrl-c.")
            ip = autodiscover_avm_ip()

            if ip is None:
                print("\nAutodiscovery failed!")
                print("Press enter to exit.")
                input()
                exit(1)

            print("\nAutodiscovery succesful!")
            print(f"-> Device detected at {ip}.")

        if args.image is None:
            # Try to automatically locate an image to use
            imagefile, hwrevision = autoload_image(ip)

            tftp_serve_file = {
                "236": "FRITZ7530.bin",
                "244": "FRITZ1200.bin",
                "246": "FRITZ3000.bin",
                "247": "FRITZ7530.bin",
            }
            tftp_serve_name = tftp_serve_file.get(hwrevision, "")

        if tftp_serve_name:
            if not (args.initramfs and args.sysupgrade):
                print("Providing initramfs and sysupgrade is required for this device")
                exit(1)
            ramfsfile = Path(args.initramfs)
            sysupgradefile = Path(args.sysupgrade)
            if not ramfsfile.is_file():
                print(f'File "{ramfsfile.absolute()}" does not exist!')
                print("Please check file name and Path.")
                exit(1)
            if not sysupgradefile.is_file():
                print(f'File "{sysupgradefile.absolute()}" does not exist!')
                print("Please check file name and Path.")
                exit(1)

        perform_flash(ip, imagefile)

        print("")

    if tftp_serve_name:
        print("Starting TFTP flash process for FB 7530/7520, FR 1200 or FR 3000")
        perform_tftp_flash(args.initramfs, args.sysupgrade, tftp_serve_name)
        print("Sleep 90s - let system boot")
        time.sleep(60)
        print("Device will come up for ~5s about now.")
    if args.sysupgrade:
        print("we still need to wait 30s for boot completion")
        time.sleep(30)
        sysupgradefile = Path(args.sysupgrade)
        print(f"flashing sysupgrade {sysupgradefile} through ssh")
        perform_bootloader_flash(sysupgradefile, imagefile, tftp_serve_name)
    elif "initramfs" in imagefile.name:
        print("WARNING: Openwrt is now installed in RAM")
        print(
            "as no additional sysupgrade was provided, you should connect per ssh and run sysupgrade to persistently install firmware"
        )
    print("Finished flash procedure")
    finish_message()