From 0ed9bc94fb83089b144c7dc584d33b1afd789235 Mon Sep 17 00:00:00 2001
From: jrwimmer <67992074+jrwimmer@users.noreply.github.com>
Date: Mon, 20 Dec 2021 20:32:06 -0800
Subject: [PATCH] Added rootfs-size sanity check

* Module will now check the total size of the active root filesystem
* - This is done to catch and prevent instances of double-extroot migration
---
 modules/extroot | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/modules/extroot b/modules/extroot
index c782f96..cc18080 100644
--- a/modules/extroot
+++ b/modules/extroot
@@ -32,12 +32,37 @@ function configure {
       $DIALOG_OK)
         perform_migration;;
       $DIALOG_CANCEL)
-        dialog --title "Extroot" --msgbox "Operation cancelled" 14 72;;
+        dialog --title "Extroot" --msgbox "Operation cancelled" 14 72
+        exit 0;;
       $DIALOG_ESC)
-        dialog --title "Extroot" --msgbox "Operation cancelled" 14 72;;
+        dialog --title "Extroot" --msgbox "Operation cancelled" 14 72
+        exit 0;;
     esac
 }
 
+function check_space {
+    # Get block size for the root partition
+    rootfs_blocksize=$(df -P / | awk '{print $2}' | tail -n+2)
+    rootfs_humansize=$(df -h / | awk '{print $2}' | tail -n+2)
+
+    # Check if the root partition is larger than 256 MB
+    if (( $rootfs_blocksize > 256000 )); then
+        dialog --title "Extroot: WARNING" --defaultno --yesno "::: WARNING :::\n\n\nThe root partition appears to be larger than 256MB in size (Seen: $rootfs_humansize)\n\nThis may indicate that the overlay configuration process has already been performed on this installation.\n\nAre you sure you wish to continue?" 14 72
+        
+        return=$?
+        case $return in
+          $DIALOG_OK)
+            return 0;;
+          $DIALOG_CANCEL)
+            dialog --title "Extroot" --msgbox "Operation cancelled" 14 72
+            exit 0;;
+          $DIALOG_ESC)
+            dialog --title "Extroot" --msgbox "Operation cancelled" 14 72
+            exit 0;;
+        esac
+    fi
+}
+
 function check_network {
     # Check for Internet connection by attempting to contact an opkg repository server
 
@@ -68,8 +93,6 @@ function install_dependencies {
 }
 
 function remount_rootfsdata {
-    #TODO: Add sanity check for the overlay having already been migrated
-
     # Identify flash memory device containing current overlay
     rootfsdata_dev="$(sed -n -e "/\s\/overlay\s.*$/s///p" /etc/mtab)"
 
@@ -183,6 +206,7 @@ function perform_step {
 }
 
 function perform_migration {
+    check_space
 
     perform_step "check_network" "Checking network connectivity"