Add command line arguments for disk_image

Add some simple command line arguments and some dependency checking that
makes running the script outside of a container much easier.

Signed-off-by: Brian Woods <brian.woods@xilinx.com>
Signed-off-by: Stefano Stabellini <stefano.stabellini@xilinx.com>
Reviewed-by: Stefano Stabellini <sstabellini@kernel.org>

1 files changed, 83 insertions, 7 deletions

diff --git a/scripts/disk_image b/scripts/disk_image
index 949e323..88a1663 100755
--- a/scripts/disk_image
+++ b/scripts/disk_image
@@ -3,18 +3,89 @@
 # This helps to see what is going on
 set -e -x
 
-# Used to maintain intermediary state. (Not currently used)
-DESTDIR="$(mktemp -d /tmp/imagebuilder-zynqmp.XXXXXX)"
-UBOOT_OUT="/tmp/output"
-OUTDIR="/tmp/img"
-mkdir ${OUTDIR}
+PROG_REQ=( kpartx mkfs.ext4 losetup sgdisk readlink )
+
+function check_depends()
+{
+    for ((i=0; i<${#PROG_REQ[@]}; i++))
+    do
+        if ! command -v ${PROG_REQ[i]} > /dev/null
+        then
+            echo "Please install the needed program: ${PROG_REQ[i]}."
+            exit 1
+        fi
+    done
+}
+
+function print_help
+{
+    echo "usage:"
+    echo "	$0 -c CONFIG_FILE -d UBOOT_DIRECTORY <-w WORK_DIRECTORY> -o IMG_FILE"
+    echo "	$0 -h"
+    echo "where:"
+    echo "	-c CONFIG_FILE - configuration file"
+    echo "	-d UBOOT_DIRECTORY - root directory for the paths specified in CONFIG_FILE"
+    echo "	-w WORK_DIRECTORY - work directory used when building the image"
+    echo "	-o IMG_FILE - the output img file "
+    echo "Example:"
+    echo "	$0 -c ../config -d ./build42 -w tmp -o disk.img"
+}
+
+# before anything else, check if we have root privilege
+if ! [ $(id -u) = 0 ]
+then
+    echo "This script needs root privilege to run, exiting."
+    exit 1
+fi
 
-IMG="${OUTDIR}/zynqmp.img"
+while getopts ":w:d:c:o:h" opt
+do
+    case ${opt} in
+    w )
+        DESTDIR=$OPTARG
+        ;;
+    d )
+        UBOOT_OUT=$OPTARG
+        ;;
+    c )
+        CFG_FILE=$OPTARG
+        ;;
+    o )
+        IMG=$OPTARG
+        ;;
+    h )
+        print_help
+        exit 0
+        ;;
+    * )
+        echo "Unknown option, see \"$0 -h\""
+        exit 1
+        ;;
+    esac
+done
+shift $((OPTIND -1))
+
+if [ -z "$UBOOT_OUT" ] || [ -z "$CFG_FILE" ] || [ -z "$IMG" ]
+then
+    echo "Undefined arguments, see \"$0 -h\""
+    exit 1
+fi
 
-. config
+# if the user hasn't specified a working directing, create it
+if [ -z "$DESTDIR" ]
+then
+    DESTDIR="$(mktemp -d /tmp/imagebuilder.XXXXXX)"
+    DESTDIR_DEL=true
+else
+    DESTDIR_DEL=false
+fi
 
 UBOOT_OUT_ABS="$(readlink -f $UBOOT_OUT)"
 
+check_depends
+
+source "$CFG_FILE"
+
 offset=$((2*1024*1024))
 _part1_size=`stat --printf="%s" $UBOOT_OUT/$XEN`
 _part1_size=$(( $_part1_size + `stat --printf="%s" $UBOOT_OUT/$DOM0_KERNEL` ))
@@ -155,3 +226,8 @@ umount $DESTDIR/part/disk1
 kpartx -d /dev/mapper/diskimage
 dmsetup remove diskimage
 losetup -d $_loop_dev
+
+if [ "$DESTDIR_DEL" = true ]
+then
+    rm -rf "$DESTDIR"
+fi