diff --git a/install_fw.py b/install_fw.py index 72492df..045b076 100644 --- a/install_fw.py +++ b/install_fw.py @@ -1112,9 +1112,14 @@ class XqFlash(): gw.run_cmd("sync ; umount -a", timeout = 5) print("Please, reboot router!") else: + import ssh2 print('Send command "reboot" via SSH/Telnet ...') - gw.run_cmd("reboot -f", die_on_error = False) - print("Forced REBOOT activated!") + try: + gw.run_cmd("reboot -f", die_on_error = False) + print("Forced REBOOT activated!") + except ssh2.exceptions.SocketRecvError as e: + print("Forced REBOOT Activated!") + pass def flash_data_to_mtd(self, img_name, img: Image, timeout, check = True): print(f'Writing {img_name} image to addr 0x{img.addr:08X} ...')