mtkbootcmd.py

import sys
import time
import serial
import serial.tools.list_ports

if len(sys.argv) < 2:
    print("Usage: mtkbootcmd.py <BOOTMODE>")
    sys.exit(1)

PLBROM_HWID = "VID:PID=0E8D:2000"
BOOTMODE = bytes(sys.argv[1], "ascii")
READYCMD = b"READY"


def serial_port():
    ports = list(serial.tools.list_ports.comports())
    for port in ports:
        if PLBROM_HWID in port.hwid:
            print("Found {} with description: {}\nHWID: {}".format(port.device, port.description, port.hwid))
            return port.device
    return None


if __name__ == '__main__':
    print('Listening for ports!')

    abort = False
    while not abort:
        time.sleep(1)
        port = serial_port()
        if port is not None:
            print('PORT:', port)
            print('Initializing port:', port)
            ser = serial.Serial(port=port, baudrate=115200)

            try:
                resp = ser.read(5) # Expect 5 bytes, since the output will be “READY” if successful
                if resp == READYCMD:
                    ser.write(BOOTMODE)
                    print(f"{BOOTMODE} cmd sent")
                    abort = True
                    break
                else:
                    print(f"Error: {resp}")
                    sys.exit(2)

            except serial.SerialException as e:
                print(f"Serial error: {e}")
                sys.exit(3)
            except Exception as e:
                print(f"Unexpected error: {e}")
                sys.exit(4)

            finally:
                if ser.is_open:
                    ser.close()