From 36809a99dcd86d3233cfbe7da54a4f77eaa1511f Mon Sep 17 00:00:00 2001
From: Michel Roegl-Brunner <michel.roegl-brunner@htl-braunau.at>
Date: Fri, 16 May 2025 10:40:19 +0200
Subject: [PATCH] Enhance config_file function to implement robust password
 validation and user feedback for root password setup

---
 misc/build.func | 32 ++++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

diff --git a/misc/build.func b/misc/build.func
index 3d59382..3ec9080 100644
--- a/misc/build.func
+++ b/misc/build.func
@@ -870,8 +870,36 @@ config_file() {
     fi
     PW="-password $PW"
   else
-    PW=""
-    echo -e "${VERIFYPW}${BOLD}${DGN}Root Password: ${BGN}Automatic Login${CL}"
+    while true; do
+    if PW1=$(whiptail --backtitle "Proxmox VE Helper Scripts" --passwordbox "\nSet Root Password (needed for root ssh access)" 9 58 --title "PASSWORD (leave blank for automatic login)" 3>&1 1>&2 2>&3); then
+      if [[ -n "$PW1" ]]; then
+        if [[ "$PW1" == *" "* ]]; then
+          whiptail --msgbox "Password cannot contain spaces. Please try again." 8 58
+        elif [ ${#PW1} -lt 5 ]; then
+          whiptail --msgbox "Password must be at least 5 characters long. Please try again." 8 58
+        else
+          if PW2=$(whiptail --backtitle "Proxmox VE Helper Scripts" --passwordbox "\nVerify Root Password" 9 58 --title "PASSWORD VERIFICATION" 3>&1 1>&2 2>&3); then
+            if [[ "$PW1" == "$PW2" ]]; then
+              PW="-password $PW1"
+              echo -e "${VERIFYPW}${BOLD}${DGN}Root Password: ${BGN}********${CL}"
+              break
+            else
+              whiptail --msgbox "Passwords do not match. Please try again." 8 58
+            fi
+          else
+            exit_script
+          fi
+        fi
+      else
+        PW1="Automatic Login"
+        PW=""
+        echo -e "${VERIFYPW}${BOLD}${DGN}Root Password: ${BGN}$PW1${CL}"
+        break
+      fi
+    else
+      exit_script
+    fi
+  done
   fi
 
   if [[ -n "$HN" ]]; then