From 2c1dd008fab63977c086559be1b6a3509bacd218 Mon Sep 17 00:00:00 2001
From: Johan Gunnarsson <johan.gunnarsson@gmail.com>
Date: Tue, 25 Jul 2023 19:47:59 +0200
Subject: [PATCH] zz-update-uimg: Configure initrd size

---
 .../etc/initramfs/post-update.d/zz-update-uimg               | 3 ++-
 .../etc/initramfs/post-update.d/zz-update-uimg               | 5 +++--
 .../etc/initramfs/post-update.d/zz-update-uimg               | 5 +++--
 .../etc/initramfs/post-update.d/zz-update-uimg               | 3 ++-
 4 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/2nd-stage-files/pre-2nd-stage-files-amd64/etc/initramfs/post-update.d/zz-update-uimg b/2nd-stage-files/pre-2nd-stage-files-amd64/etc/initramfs/post-update.d/zz-update-uimg
index 0dd2b5f..3070a04 100755
--- a/2nd-stage-files/pre-2nd-stage-files-amd64/etc/initramfs/post-update.d/zz-update-uimg
+++ b/2nd-stage-files/pre-2nd-stage-files-amd64/etc/initramfs/post-update.d/zz-update-uimg
@@ -8,9 +8,10 @@ ABI="${1}"
 cat << EOF > /boot/boot.cmd
 load \${devtype} \${devnum}:\${distro_bootpart} \${kernel_addr_r} /boot/vmlinuz-${ABI}
 load \${devtype} \${devnum}:\${distro_bootpart} \${ramdisk_addr_r} /boot/initrd.img-${ABI}
+setenv ramdisk_size \${filesize}
 part uuid \${devtype} \${devnum}:\${distro_bootpart} partuuid
 setenv bootargs root=PARTUUID=\${partuuid} rw rootwait
-zboot \${kernel_addr_r} - \${ramdisk_addr_r} 0x2000000
+zboot \${kernel_addr_r} - \${ramdisk_addr_r} \${ramdisk_size}
 EOF
 
 # Create U-Boot boot script
diff --git a/2nd-stage-files/pre-2nd-stage-files-arm64/etc/initramfs/post-update.d/zz-update-uimg b/2nd-stage-files/pre-2nd-stage-files-arm64/etc/initramfs/post-update.d/zz-update-uimg
index 06d605b..0642954 100755
--- a/2nd-stage-files/pre-2nd-stage-files-arm64/etc/initramfs/post-update.d/zz-update-uimg
+++ b/2nd-stage-files/pre-2nd-stage-files-arm64/etc/initramfs/post-update.d/zz-update-uimg
@@ -22,15 +22,16 @@ load \${devtype} \${devnum}:\${distro_bootpart} \${kernel_addr_r} /boot/vmlinuz-
 load \${devtype} \${devnum}:\${distro_bootpart} \${ramdisk_addr_r} /boot/initrd.uimg-${ABI}
 setexpr kernel_comp_addr_r \${ramdisk_addr_r} + \${filesize}
 setexpr kernel_comp_size 0x4000000
+setenv ramdisk_size \${filesize}
 part uuid \${devtype} \${devnum}:\${distro_bootpart} partuuid
 setenv bootargs root=PARTUUID=\${partuuid} rw rootwait
 for fdtpath in ${FDT}; do
   if test -e \${devtype} \${devnum}:\${distro_bootpart} \${fdtpath}/\${fdtfile}; then
     load \${devtype} \${devnum}:\${distro_bootpart} \${fdt_addr_r} \${fdtpath}/\${fdtfile}
-    booti \${kernel_addr_r} \${ramdisk_addr_r} \${fdt_addr_r}
+    booti \${kernel_addr_r} \${ramdisk_addr_r}:\${ramdisk_size} \${fdt_addr_r}
   fi
 done
-booti \${kernel_addr_r} \${ramdisk_addr_r} \${fdtcontroladdr}
+booti \${kernel_addr_r} \${ramdisk_addr_r}:\${ramdisk_size} \${fdtcontroladdr}
 EOF
 
 # Create U-Boot boot script
diff --git a/2nd-stage-files/pre-2nd-stage-files-armhf/etc/initramfs/post-update.d/zz-update-uimg b/2nd-stage-files/pre-2nd-stage-files-armhf/etc/initramfs/post-update.d/zz-update-uimg
index e6f4e71..2682083 100755
--- a/2nd-stage-files/pre-2nd-stage-files-armhf/etc/initramfs/post-update.d/zz-update-uimg
+++ b/2nd-stage-files/pre-2nd-stage-files-armhf/etc/initramfs/post-update.d/zz-update-uimg
@@ -20,15 +20,16 @@ FDT="\
 cat << EOF > /boot/boot.cmd
 load \${devtype} \${devnum}:\${distro_bootpart} \${kernel_addr_r} /boot/vmlinuz-${ABI}
 load \${devtype} \${devnum}:\${distro_bootpart} \${ramdisk_addr_r} /boot/initrd.uimg-${ABI}
+setenv ramdisk_size \${filesize}
 part uuid \${devtype} \${devnum}:\${distro_bootpart} partuuid
 setenv bootargs root=PARTUUID=\${partuuid} rw rootwait
 for fdtpath in ${FDT}; do
   if test -e \${devtype} \${devnum}:\${distro_bootpart} \${fdtpath}/\${fdtfile}; then
     load \${devtype} \${devnum}:\${distro_bootpart} \${fdt_addr_r} \${fdtpath}/\${fdtfile}
-    bootz \${kernel_addr_r} \${ramdisk_addr_r} \${fdt_addr_r}
+    bootz \${kernel_addr_r} \${ramdisk_addr_r}:\${ramdisk_size} \${fdt_addr_r}
   fi
 done
-bootz \${kernel_addr_r} \${ramdisk_addr_r} \${fdtcontroladdr}
+bootz \${kernel_addr_r} \${ramdisk_addr_r}:\${ramdisk_size} \${fdtcontroladdr}
 EOF
 
 # Create U-Boot boot script
diff --git a/2nd-stage-files/pre-2nd-stage-files-i386/etc/initramfs/post-update.d/zz-update-uimg b/2nd-stage-files/pre-2nd-stage-files-i386/etc/initramfs/post-update.d/zz-update-uimg
index 62277b4..159ef4e 100755
--- a/2nd-stage-files/pre-2nd-stage-files-i386/etc/initramfs/post-update.d/zz-update-uimg
+++ b/2nd-stage-files/pre-2nd-stage-files-i386/etc/initramfs/post-update.d/zz-update-uimg
@@ -8,9 +8,10 @@ ABI="${1}"
 cat << EOF > /boot/boot.cmd
 load \${devtype} \${devnum}:\${distro_bootpart} \${kernel_addr_r} /boot/vmlinuz-${ABI}
 load \${devtype} \${devnum}:\${distro_bootpart} \${ramdisk_addr_r} /boot/initrd.img-${ABI}
+setenv ramdisk_size \${filesize}
 part uuid \${devtype} \${devnum}:\${distro_bootpart} partuuid
 setenv bootargs root=PARTUUID=\${partuuid} rw rootwait
-zboot \${kernel_addr_r} - \${ramdisk_addr_r} 0x2000000
+zboot \${kernel_addr_r} - \${ramdisk_addr_r} \${ramdisk_size}
 EOF
 
 # Create U-Boot boot script