From 77eb81c6fe96354f3c24dd8e2ef528871305a379 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Sat, 15 Apr 2023 02:15:18 +0200
Subject: [PATCH] Check free diskspace

---
 disk.sh    | 5 ++++-
 install.sh | 6 +++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/disk.sh b/disk.sh
index f8d42ed..997e10d 100644
--- a/disk.sh
+++ b/disk.sh
@@ -14,7 +14,10 @@ DATA="$IMG/data$DISK_SIZE.img"
 
 if [ ! -f "$DATA" ]; then
     # Create an empty file
-    fallocate -l "${NEW_SIZE}" "${DATA}"
+    if ! fallocate -l "${NEW_SIZE}" "${DATA}"; then
+      rm -f "${DATA}"
+      echo "ERROR: Not enough free space to create virtual disk." && exit 88
+    fi
     # Format as BTRFS filesystem
     mkfs.btrfs -q -L data -d single -m dup "${DATA}" > /dev/null
 fi
diff --git a/install.sh b/install.sh
index 9aeec4c..0f0101a 100644
--- a/install.sh
+++ b/install.sh
@@ -101,7 +101,11 @@ echo "Install: Creating partition table..."
 
 SYSTEM="$TMP/sys.img"
 SYSTEM_SIZE="4954537983"
-fallocate -l "${SYSTEM_SIZE}" "${SYSTEM}"
+
+if ! fallocate -l "${SYSTEM_SIZE}" "${SYSTEM}"; then
+  rm -f "${SYSTEM}"
+  echo "ERROR: Not enough free space to create virtual disk." && exit 88
+fi
 
 PART="$TMP/partition.fdisk"