Add server disk space check for mounted share when capturing an image…

…, thanks to rluzuriaga

Resolves #60

Buildroot/board/FOG/FOS/rootfs_overlay/bin/fog.checkmountdrivesize

@@ -0,0 +1,23 @@
+#!/bin/bash
+dots "Checking Server Disk Space"
+getServerDiskSpaceUsage=$(df -h | grep "/images" | sed -n '/dev/{s/  */ /gp}')
+echo "Done"
+
+dots "Server Disk Size Total"
+serverDiskSizeTotal=$(echo "$getServerDiskSpaceUsage" | cut -d ' ' -f2)
+echo "$serverDiskSizeTotal"
+
+dots "Server Disk Size Used"
+serverDiskSizeUsed=$(echo "$getServerDiskSpaceUsage" | cut -d ' ' -f3)
+echo "$serverDiskSizeUsed"
+
+dots "Server Disk Size Used Percentage"
+serverDiskSizeUsedPercent=$(echo "$getServerDiskSpaceUsage" | cut -d ' ' -f5)
+echo "$serverDiskSizeUsedPercent"
+
+dots "Server Disk Size Available"
+serverDiskSizeAvailable=$(echo "$getServerDiskSpaceUsage" | cut -d ' ' -f4)
+[[ $serverDiskSizeAvailable == "0" ]] && serverDiskSizeAvailable="0M"
+echo "$serverDiskSizeAvailable"
+
+debugPause

Buildroot/board/FOG/FOS/rootfs_overlay/bin/fog.upload

@@ -3,6 +3,7 @@
 . /bin/fog.checkin
 . /bin/fog.mount
 . /bin/fog.checkmount
+. /bin/fog.checkmountdrivesize
 . /bin/fog.checkimgvar
 imagePath="/images/$macWinSafe"
 parts=""

Buildroot/board/FOG/FOS/rootfs_overlay/usr/share/fog/lib/funcs.sh

@@ -422,6 +422,13 @@ getPartBlockSize() {
     [[ -z $varVar ]] && handleError "No variable to set passed (${FUNCNAME[0]})\n   Args Passed: $*"
     printf -v "$varVar" $(blockdev --getpbsz $part)
 }
+# Retrieve available space from NFS share
+# Should only be used when the share is mounted to `/images`
+getServerDiskSpaceSvailable() {
+    local space=$(df -h | grep "/images" | sed -n '/dev/{s/  */ /gp}' | cut -d ' ' -f4)
+    [[ $space == "0" ]] && local space="0M"
+    echo $space
+}
 # Prepares location info for uploads
 #
 # $1 is the image path
@@ -437,7 +444,8 @@ prepareUploadLocation() {
             *)
                 echo "Failed"
                 debugPause
-                handleError "Failed to create image capture path (${FUNCNAME[0]})\n   Args Passed: $*"
+                local spaceAvailable=$(getServerDiskSpaceSvailable)
+                handleError "Failed to create image capture path (${FUNCNAME[0]})\nServer Disk Space Available: $spaceAvailable\n   Args Passed: $*"
                 ;;
         esac
     fi
@@ -2231,7 +2239,8 @@ savePartition() {
                     debugPause
                     ;;
                 *)
-                    handleError "Failed to complete capture (${FUNCNAME[0]})\n    Args Passed: $*\n    CMD: partclone.$fstype -n \"Storage Location $storage, Image name $img\" -s -O $fifoname -Nf 1\n    Exit code: $exitcode\n    Server Disk Space Available: $(df -h /images | awk '{print $4}')"
+                    local spaceAvailable=$(getServerDiskSpaceSvailable)
+                    handleError "Failed to complete capture (${FUNCNAME[0]})\n    Args Passed: $*\n    CMD: partclone.$fstype -n \"Storage Location $storage, Image name $img\" -s -O $fifoname -Nf 1\n    Exit code: $exitcode\n    Server Disk Space Available: $spaceAvailable"
                     ;;
             esac
             ;;
@@ -2257,7 +2266,8 @@ savePartition() {
                             debugPause
                             ;;
                         *)
-                            handleError "Failed to complete capture (${FUNCNAME[0]})\n    Args Passed: $*\n    CMD: partclone.$fstype -n \"Storage Location $storage, Image name $img\" -cs -O $fifoname -Nf 1 -a0\n    Exit code: $exitcode\n    Server Disk Space Available: $(df -h /images | awk '{print $4}')"
+                            local spaceAvailable=$(getServerDiskSpaceSvailable)
+                            handleError "Failed to complete capture (${FUNCNAME[0]})\n    Args Passed: $*\n    CMD: partclone.$fstype -n \"Storage Location $storage, Image name $img\" -cs -O $fifoname -Nf 1 -a0\n    Exit code: $exitcode\n    Server Disk Space Available: $spaceAvailable"
                             ;;
                     esac
                     ;;