v9.10

- DietPi-Imager | Allow generating images from the host's root drive, by mounting its rootfs and in case bootfs R/O

.build/images/dietpi-imager

@@ -58,6 +58,7 @@
 	##########################################
 	# Process inputs
 	##########################################
+	HOST_ROOT_DRIVE=$(lsblk -npo PKNAME "$G_ROOTFS_DEV")
 	SOURCE_TYPE='Drive'
 	FP_SOURCE_IMG=
 	PART_TABLE_TYPE=
@@ -116,6 +117,12 @@
 		[[ -d $FP_MNT_TMP ]] && G_EXEC rmdir "$FP_MNT_TMP"
 		Delete_Loopback
 		[[ -e 'tmpiso' ]] && G_EXEC rm -R tmpiso
+		if [[ $FP_SOURCE == "$HOST_ROOT_DRIVE" ]]
+		then
+			findmnt -M /boot > /dev/null && G_EXEC mount -o remount,rw /boot
+			G_EXEC mount -o remount,rw /
+			/boot/dietpi/dietpi-services start
+		fi
 	}
 
 	Run_fsck()
@@ -162,37 +169,64 @@
 		if [[ $SOURCE_TYPE == 'Drive' ]]
 		then
 			# Detect drives with a partition table, containing a partition with and ext4, F2FS or Btrfs filesystem, excluding the hosts root filesystem drive
-			mapfile -t G_WHIP_MENU_ARRAY < <(mawk -v root="$(lsblk -npo PKNAME "$G_ROOTFS_DEV")" '$1!=root && $2~/^(ext4|f2fs|btrfs)$/ {print $1"\n"$1"_details"}' < <(lsblk -rnpo PKNAME,FSTYPE) | sort -u)
-
-			if [[ ! ${G_WHIP_MENU_ARRAY[0]} ]]
-			then
-				G_DIETPI-NOTIFY 1 'No drive with an ext4, F2FS or Btrfs formatted partition found, aborting...'
-				G_DIETPI-NOTIFY 2 'NB: This is the list of available block devices:'
-				lsblk -npo NAME,SIZE,MAJ:MIN,FSTYPE,MOUNTPOINT,MODEL
-				read -rp 'Press any key to return to menu...'
-				return 1
-			fi
+			mapfile -t G_WHIP_MENU_ARRAY < <(mawk -v root="$HOST_ROOT_DRIVE" '$1!=root && $2~/^(ext4|f2fs|btrfs)$/ {print $1"\n"$1"_details"}' < <(lsblk -rnpo PKNAME,FSTYPE) | sort -u)
 
 			# Visually separate dev name and size and add model and serial
 			for ((i=1;i<${#G_WHIP_MENU_ARRAY[@]};i+=2)); do G_WHIP_MENU_ARRAY[$i]=": $(lsblk -drno SIZE,MODEL,SERIAL "${G_WHIP_MENU_ARRAY[$i-1]}")"; done
 
+			# Add this hosts root filesystem
+			G_WHIP_MENU_ARRAY+=("$HOST_ROOT_DRIVE" ": $(lsblk -drno SIZE "$HOST_ROOT_DRIVE") This host's root drive!")
+
 			G_WHIP_DEFAULT_ITEM=$FP_SOURCE
 			G_WHIP_MENU 'Please select the drive you wish to create the image from:
 \nNB: All mounted partitions of the selected drive will be unmounted.' || return 0
 			FP_SOURCE=$G_WHIP_RETURNED_VALUE
 			FP_ROOT_DEV=
 
+			if [[ $FP_SOURCE == "$HOST_ROOT_DRIVE" ]]
+			then
+				G_WHIP_BUTTON_CANCEL_TEXT='Abort'
+				G_WHIP_YESNO '[WARNING] You selected this host'\''s root drive as source!
+\nThis is generally possible, but the root (and boot) filesystem(s) needs to be remounted read-only (R/O) to assure a consistent image. This may fail if certain system services are active, we hence stop services like during DietPi updates, dietpi-software installs and dietpi-backup runs. If remounting the root filesystem still fails, image generation is aborted.
+\nAdditionally, the image needs to be stored to an external drive. If you did not nativate to one before running this script, you are asked to select one.
+\nDo you want to continue?' || exit 0
+				/boot/dietpi/dietpi-services stop
+				[[ $(findmnt -M /boot) ]] && until mount -o remount,ro /boot
+				do
+					G_WHIP_BUTTON_OK_TEXT='Retry'
+					G_WHIP_BUTTON_CANCEL_TEXT='Abort'
+					G_WHIP_YESNO '[WARNING] Mounting the /boot filesystem R/O failed
+\nThis may be due to running system services or for other reasons. Please assure that no process is accessing the /boot filesystem in writing mode, then retry.' || Error_Exit 'Mounting the boot filesystem R/O failed'
+				done
+				until mount -o remount,ro /
+				do
+					G_WHIP_BUTTON_OK_TEXT='Retry'
+					G_WHIP_BUTTON_CANCEL_TEXT='Abort'
+					G_WHIP_YESNO '[WARNING] Mounting the root filesystem R/O failed
+\nThis may be due to running system services or for other reasons. Please assure that no process is accessing the root filesystem in writing mode, then retry.' || Error_Exit 'Mounting the root filesystem R/O failed'
+				done
+				until [[ $(findmnt -no TARGET -T .) != '/' ]]
+				do
+					/boot/dietpi/dietpi-drive_manager 1 || Error_Exit 'No target drive selected'
+					read -r FP_ORIGIN < /tmp/dietpi-drive_manager_selmnt
+					[[ $FP_ORIGIN == '/' ]] || break
+					G_WHIP_MSG '[FAILED] You cannot select this host'\''s root filesystem to store its own image. Please select a different mount'
+					G_EXEC cd "$FP_ORIGIN"
+					rm /tmp/dietpi-drive_manager_selmnt
+				done
+			fi
+
 			G_DIETPI-NOTIFY 2 "Unmounting all filesystems below selected $FP_SOURCE ..."
 			local mountpoint
 			for i in "$FP_SOURCE"?*
 			do
 				mountpoint=$(findmnt -no TARGET "$i")
-				[[ $mountpoint ]] && G_EXEC umount -R "$mountpoint"
+				[[ $mountpoint && $mountpoint != '/' && $mountpoint != '/boot' ]] && G_EXEC umount -R "$mountpoint"
 			done
 		else
 			# Open DietPi-Explorer for image file selection
 			/boot/dietpi/dietpi-explorer 1 || { G_DIETPI-NOTIFY 1 'No image file selected, aborting...'; read -rp 'Press any key to return to menu...'; return 1; }
-			FP_SOURCE_IMG=$(</tmp/.dietpi-explorer_selected_location)
+			read -r FP_SOURCE_IMG < /tmp/.dietpi-explorer_selected_location
 			FP_ROOT_DEV=
 			rm /tmp/.dietpi-explorer_selected_location
 			[[ -f $FP_SOURCE_IMG ]] || { G_DIETPI-NOTIFY 1 "Selected image file ($FP_SOURCE_IMG) does not exist, aborting..."; read -rp 'Press any key to return to menu...'; return 1; }