poky/scripts/contrib/ddimage
Richard Purdie 7350e82ce1 scripts: Add copyright statements to files without one
Where there isn't a copyright statement, add one to make it explicit.
Also drop editor config lines where they were present and add license
identifiers as MIT if there isn't one.

(From OE-Core rev: deb3ccec53e0bd63bc4235cf2b0d3fc781687361)

Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
2022-08-12 11:58:01 +01:00

3.7 KiB
Executable File

#!/bin/sh

Copyright OpenEmbedded Contributors

SPDX-License-Identifier: GPL-2.0-only

1MB blocksize

BLOCKSIZE=1048576

usage() { echo "Usage: $(basename $0) IMAGE DEVICE" }

image_details() { IMG=$1 echo "Image details" echo "=============" echo " image: $(basename $IMG)" # stat format is different on Mac OS and Linux if [ "$(uname)" = "Darwin" ]; then echo " size: $(stat -L -f '%z bytes' $IMG)" echo " modified: $(stat -L -f '%Sm' $IMG)" else echo " size: $(stat -L -c '%s bytes' $IMG)" echo " modified: $(stat -L -c '%y' $IMG)" fi echo " type: $(file -L -b $IMG)" echo "" }

device_details() { BLOCK_SIZE=512

echo "Device details"
echo "=============="

# Collect disk info using diskutil on Mac OS
if [ "$(uname)" = "Darwin" ]; then
	diskutil info $DEVICE | egrep "(Device Node|Media Name|Total Size)"
	return
fi

# Default / Linux information collection
ACTUAL_DEVICE=`readlink -f $DEVICE`
DEV=`basename $ACTUAL_DEVICE`
if [ "$ACTUAL_DEVICE" != "$DEVICE" ] ; then
	echo "  device: $DEVICE -> $ACTUAL_DEVICE"
else
	echo "  device: $DEVICE"
fi
if [ -f "/sys/class/block/$DEV/device/vendor" ]; then
	echo "  vendor: $(cat /sys/class/block/$DEV/device/vendor)"
else
	echo "  vendor: UNKNOWN"
fi
if [ -f "/sys/class/block/$DEV/device/model" ]; then
	echo "   model: $(cat /sys/class/block/$DEV/device/model)"
else
	echo "   model: UNKNOWN"
fi
if [ -f "/sys/class/block/$DEV/size" ]; then
	echo "    size: $(($(cat /sys/class/block/$DEV/size) * $BLOCK_SIZE)) bytes"
else
	echo "    size: UNKNOWN"
fi
echo ""

}

check_mount_device() { if cat /proc/self/mounts | awk '{ print $1 }' | grep /dev/ | grep -q -E "^$1$" ; then return 0 fi return 1 }

is_mounted() { if [ "$(uname)" = "Darwin" ]; then if df | awk '{ print $1 }' | grep /dev/ | grep -q -E "^$1(s[0-9]+)?$" ; then return 0 fi else if check_mount_device $1 ; then return 0 fi DEV=basename $1 if [ -d /sys/class/block/$DEV/ ] ; then PARENT_BLKDEV=basename $(readlink -f "/sys/class/block/$DEV/..") if [ "$PARENT_BLKDEV" != "block" ] ; then if check_mount_device $PARENT_BLKDEV ; then return 0 fi fi for CHILD_BLKDEV in find /sys/class/block/$DEV/ -mindepth 1 -maxdepth 1 -name "$DEV*" -type d do if check_mount_device /dev/basename $CHILD_BLKDEV ; then return 0 fi done fi fi return 1 }

is_inuse() { HOLDERS_DIR="/sys/class/block/basename $1/holders" if [ -d $HOLDERS_DIR ] && [ ls -A $HOLDERS_DIR ] ; then return 0 fi return 1 }

if [ $# -ne 2 ]; then usage exit 1 fi

IMAGE=$1 DEVICE=$2

if [ ! -e "$IMAGE" ]; then echo "ERROR: Image $IMAGE does not exist" usage exit 1 fi

if [ ! -e "$DEVICE" ]; then echo "ERROR: Device $DEVICE does not exist" usage exit 1 fi

if [ "$(uname)" = "Darwin" ]; then # readlink doesn't support -f on MacOS, just assume it isn't a symlink ACTUAL_DEVICE=$DEVICE else ACTUAL_DEVICE=readlink -f $DEVICE fi if is_mounted $ACTUAL_DEVICE ; then echo "ERROR: Device $DEVICE is currently mounted - check if this is the right device, and unmount it first if so" device_details exit 1 fi if is_inuse $ACTUAL_DEVICE ; then echo "ERROR: Device $DEVICE is currently in use (possibly part of LVM) - check if this is the right device!" device_details exit 1 fi

if [ ! -w "$DEVICE" ]; then echo "ERROR: Device $DEVICE is not writable - possibly use sudo?" usage exit 1 fi

image_details $IMAGE device_details

printf "Write $IMAGE to $DEVICE [y/N]? " read RESPONSE if [ "$RESPONSE" != "y" ]; then echo "Write aborted" exit 0 fi

echo "Writing image..." if which pv >/dev/null 2>&1; then pv "$IMAGE" | dd of="$DEVICE" bs="$BLOCKSIZE" else dd if="$IMAGE" of="$DEVICE" bs="$BLOCKSIZE" fi sync