From b9aea3470893d73de06459e14c84bc96d758bdff Mon Sep 17 00:00:00 2001
From: Ing <wjz304@qq.com>
Date: Sat, 9 Mar 2024 15:50:30 +0800
Subject: [PATCH] update img-gen.sh

---
 img-gen.sh | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/img-gen.sh b/img-gen.sh
index e44f86d6..920a2152 100755
--- a/img-gen.sh
+++ b/img-gen.sh
@@ -23,14 +23,13 @@ sudo mount ${LOOPX}p1 "/tmp/files/p1"
 sudo mount ${LOOPX}p3 "/tmp/files/p3"
 
 echo "Get Buildroot"
-[ ! -f "br/bzImage-rr" -o ! -f "br/initrd-rr" ] && getBuildroot "br" "latest"
-[ ! -f "br/bzImage-rr" -o ! -f "br/initrd-rr" ] && return 1
+[ ! -f "/tmp/files/p3/bzImage-rr" -o ! -f "/tmp/files/p3/initrd-rr" ] && getBuildroot "/tmp/files/p3" true
+[ ! -f "/tmp/files/p3/bzImage-rr" -o ! -f "/tmp/files/p3/initrd-rr" ] && return 1
 
 read -p "Press enter to continue"
 
 echo "Repack initrd"
-sudo cp -f "br/bzImage-rr" "/tmp/files/p3/bzImage-rr"
-repackInitrd "br/initrd-rr" "files/initrd" "/tmp/files/p3/initrd-rr"
+repackInitrd "/tmp/files/p3/initrd-rr" "files/initrd"
 
 echo "Copying files"
 sudo cp -Rf "files/p1/"* "/tmp/files/p1"