diff --git a/tools/get_uec_image.sh b/tools/get_uec_image.sh index 28af71a..935feba 100755 --- a/tools/get_uec_image.sh +++ b/tools/get_uec_image.sh @@ -51,6 +51,7 @@ fi # Default args DIST_NAME=$1 IMG_FILE=$2 +IMG_FILE_TMP=`mktemp $IMG_FILE.XXXXXX` case $FORMAT in kvm|qcow2) FORMAT=qcow2 @@ -88,11 +89,6 @@ case $DIST_NAME in ;; esac -# Set up nbd -modprobe nbd max_part=63 -NBD=${NBD:-/dev/nbd9} -NBD_DEV=`basename $NBD` - # Prepare the base image # Get the UEC image @@ -103,24 +99,37 @@ fi if [ "$FORMAT" = "qcow2" ]; then # Just copy image - cp -p $CACHEDIR/$UEC_NAME-disk1.img $IMG_FILE + cp -p $CACHEDIR/$UEC_NAME-disk1.img $IMG_FILE_TMP else # Convert image - qemu-img convert -O $QFORMAT $CACHEDIR/$UEC_NAME-disk1.img $IMG_FILE + qemu-img convert -O $QFORMAT $CACHEDIR/$UEC_NAME-disk1.img $IMG_FILE_TMP fi # Resize the image if necessary if [ $ROOTSIZE -gt 2000 ]; then # Resize the container - qemu-img resize $IMG_FILE +$((ROOTSIZE - 2000))M + qemu-img resize $IMG_FILE_TMP +$((ROOTSIZE - 2000))M fi -# Connect to nbd and wait till it is ready -qemu-nbd -c $NBD $IMG_FILE -if ! timeout 60 sh -c "while ! [ -e /sys/block/$NBD_DEV/pid ]; do sleep 1; done"; then -echo "Couldn't connect $NBD" +# Set up nbd +modprobe nbd max_part=63 +for i in `seq 1 15`; do + if [ ! -e /sys/block/nbd$i/pid ]; then + NBD=/dev/nbd$i + # Connect to nbd and wait till it is ready + qemu-nbd -c $NBD $IMG_FILE_TMP + if ! timeout 60 sh -c "while ! [ -e ${NBD}p1 ]; do sleep 1; done"; then + echo "Couldn't connect $NBD" + exit 1 + fi + break + fi +done +if [ -z "$NBD" ]; then + echo "No free NBD slots" exit 1 fi +NBD_DEV=`basename $NBD` # Resize partition 1 to full size of the disk image echo "d @@ -153,3 +162,5 @@ rm -f $MNTDIR/etc/resolv.conf umount $MNTDIR rmdir $MNTDIR qemu-nbd -d $NBD + +mv $IMG_FILE_TMP $IMG_FILE