#!/bin/bash

MY1USBDEF="disk.img"
MY1USBDRV=""
MY1BOOTLD=""

# parse command parameters
while [ "$1" != "" ]; do
	case $1 in
		--image|-i)
			shift
			MY1USBDRV=$1
			;;
		-*)
			echo "Unknown option '$1'"
			exit 1
			;;
		*)
			[ "$MY1BOOTLD" != "" ] &&
				echo "Unknown parameter ($1)!" && exit 1
			MY1BOOTLD=$1
			;;
	esac
	shift
done

[ "$MY1USBDRV" = "" ] && MY1USBDRV="$MY1USBDEF"
[ ! -f "$MY1USBDRV" ] &&
	echo "Cannot find disk image '$MY1USBDRV'! Aborting!" && exit 1

[ "$MY1BOOTLD" = "" ] && MY1BOOTLD="boot.bin"
[ ! -f "$MY1BOOTLD" ] &&
	echo "Cannot find boot image '$MY1BOOTLD'! Aborting!" && exit 1

DD_OPT_MBR1="bs=1 count=446 conv=notrunc"
DD_OPT_MBR2="bs=1 count=2 conv=notrunc skip=510 seek=510"

echo -n "Copying boot code '$MY1BOOTLD' to disk image '$MY1USBDRV'... "
dd if=$MY1BOOTLD of=$MY1USBDRV $DD_OPT_MBR1 2>/dev/null
[ $? -ne 0 ] && echo "Failed! Aborting!" && exit 1
dd if=$MY1BOOTLD of=$MY1USBDRV $DD_OPT_MBR2 2>/dev/null
[ $? -ne 0 ] && echo "Failed! Aborting!" && exit 1
echo "done!"