From ea5247d72e45c74940b218be6a7fa00289f34e9d Mon Sep 17 00:00:00 2001
From: ghe0 <gheorghe@gheo.tech>
Date: Thu, 7 Nov 2024 21:21:47 +0200
Subject: [PATCH] it boots

---
 create.sh            | 5 +++++
 creator_functions.sh | 8 +++++++-
 init.sh              | 1 +
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/create.sh b/create.sh
index 912b4d1..6249625 100755
--- a/create.sh
+++ b/create.sh
@@ -21,9 +21,14 @@ install_init_script
 echo_cyan "Installing kernel modules..."
 install_module "$(df -T / | awk '{ print $2 }' | tail -1)"
 install_module btrfs
+install_module virtio_blk
 install_module msr
 install_module sev-guest
 install_module dm_crypt
+disk_driver=$(lshw -c disk 2>/dev/null | grep -oE 'driver=[a-z\-\_]+' | cut -d '=' -f2)
+[[ -n $disk_driver ]] && install_module "$disk_driver"
+
+echo_cyan "Building module dependency tree..."
 cp /lib/modules/${KERNEL}/modules.{order,builtin,builtin.modinfo} "${ROOT}/lib/modules/${KERNEL}/"
 depmod -b "$ROOT" "$KERNEL"
 
diff --git a/creator_functions.sh b/creator_functions.sh
index 4a1a1a1..247d5fc 100644
--- a/creator_functions.sh
+++ b/creator_functions.sh
@@ -21,12 +21,18 @@ create_dirs() {
   mkdir -p "$ROOT" "${ROOT}/usr/bin/"
   mkdir -p ${ROOT}/usr/bin
   mkdir -p ${ROOT}/usr/lib
+  mkdir -p ${ROOT}/dev
+  mkdir -p ${ROOT}/proc
+  mkdir -p ${ROOT}/run
+  mkdir -p ${ROOT}/tmp
+  mkdir -p ${ROOT}/var
   ln -s usr/bin ${ROOT}/bin
   ln -s usr/bin ${ROOT}/sbin
   ln -s usr/lib ${ROOT}/lib
   ln -s usr/lib ${ROOT}/lib64
   ln -s lib ${ROOT}/usr/lib64
   ln -s bin ${ROOT}/usr/sbin
+  ln -s ../run ${ROOT}/var/run
 }
 
 # Installs a library. Expects absolute path.
@@ -98,7 +104,7 @@ install_init_script() {
 
 install_module() {
   local module="$1" 
-  echo "modprobe $module" >> "${ROOT}/load_modules.sh.tmp"
+  echo "modprobe $module" >> "${ROOT}/load_modules.sh"
   _install_module "$module"
 }
 
diff --git a/init.sh b/init.sh
index 06a02cc..a5745ef 100755
--- a/init.sh
+++ b/init.sh
@@ -1,5 +1,6 @@
 #!/bin/bash
 source /init_functions.sh
+create_mounts
 load_modules
 mount_root
 exec switch_root /mnt /sbin/init "$@"