From 347309153d4ea59f5a2ee8bf0159e144964b5ebc Mon Sep 17 00:00:00 2001
From: remittor <remittor@gmail.com>
Date: Fri, 20 Oct 2023 07:51:43 +0300
Subject: [PATCH] [install_fw] Fix install OpenWRT image for RB01/RB03

---
 install_fw.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/install_fw.py b/install_fw.py
index 891af0d..c8632ba 100644
--- a/install_fw.py
+++ b/install_fw.py
@@ -947,9 +947,12 @@ class XqFlash():
                 rootfs.cmd = 'mtd -e "{part}" write "{bin}" "{part}"'.format(part=rootfs.partname, bin=rootfs.fn_remote)
 
         if self.install_method in [ 200, 300, 400 ]:
-            if dev.rootfs.num is None or dev.rootfs.num < 0:
-                die("Cannot detect current booted rootfs! (X)")
-            self.install_fw_num = 1 - dev.rootfs.num
+            if not self.img_stock and self.install_method == 300:
+                self.install_fw_num = 0
+            else:
+                if dev.rootfs.num is None or dev.rootfs.num < 0:
+                    die("Cannot detect current booted rootfs! (X)")
+                self.install_fw_num = 1 - dev.rootfs.num
             fw_img.partname = self.install_parts[self.install_fw_num]
             fw_part = dev.get_part(fw_img.partname)
             fw_img.addr = fw_part['addr']