From e94e2b11686d5c1e07edb84d5b1dca174dee6d82 Mon Sep 17 00:00:00 2001
From: ading2210 <ading2019@gmail.com>
Date: Tue, 3 Oct 2023 17:29:55 -0700
Subject: [PATCH] list disks in the bootloader

---
 bootloader/bin/bootstrap.sh | 9 ++++++++-
 bootloader/bin/init         | 4 ----
 build.sh                    | 2 +-
 build_image.sh              | 5 +++++
 4 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/bootloader/bin/bootstrap.sh b/bootloader/bin/bootstrap.sh
index 8766bf5..e7bf684 100644
--- a/bootloader/bin/bootstrap.sh
+++ b/bootloader/bin/bootstrap.sh
@@ -27,12 +27,19 @@ enable_debug_console() {
   invoke_terminal "${tty}" "[Bootstrap Debug Console]" "/bin/busybox sh"
 }
 
+find_rootfs_partitions() {
+  disks=$(fdisk -l | sed -n "s/Disk \(\/dev\/.*\):.*/\1/p")
+  for disk in $disks; do
+    echo $disk
+  done
+}
 
 main() {
   echo "...:::||| Bootstrapping ChromeOS Factory Shim |||:::..."
   echo "TTY: ${TTY}, LOG: ${LOG_TTY}, echo: ${echo_TTY}, DEBUG: ${DEBUG_TTY}"
   echo "idk please work"
-
+  
+  find_rootfs_partitions
   enable_debug_console "/dev/pts/0"
 }
 
diff --git a/bootloader/bin/init b/bootloader/bin/init
index b8ae447..893a8a1 100644
--- a/bootloader/bin/init
+++ b/bootloader/bin/init
@@ -10,11 +10,7 @@
 
 set -x
 
-. /lib/init.sh
-
 setup_environment() {
-  #initialize
-
   # Install additional utility programs.
   /bin/busybox --install /bin || true
 }
diff --git a/build.sh b/build.sh
index 692cdd3..4ce519e 100755
--- a/build.sh
+++ b/build.sh
@@ -15,7 +15,7 @@ print_help() {
 }
 
 check_deps() {
-  local needed_commands="cpio binwalk pcregrep realpath cgpt"
+  local needed_commands="cpio binwalk pcregrep realpath cgpt mkfs.ext4 mkfs.ext2 fdisk"
   for command in $needed_commands; do
     if ! command -v $command &> /dev/null; then
       echo $command
diff --git a/build_image.sh b/build_image.sh
index 14c5968..72fde69 100755
--- a/build_image.sh
+++ b/build_image.sh
@@ -59,6 +59,11 @@ partition_disk() {
     echo #accept default parition number
     echo #accept default first sector
     echo #accept default size to fill rest of image
+    echo x #enter expert mode
+    echo n #change the partition name
+    echo #accept default partition number
+    echo "shimboot_rootfs:default" #set partition name
+    echo r #return to normal more
 
     #write changes
     echo w