# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.
-RESULT=/dev/ttyS1
-FLOPPY_DEV=/dev/fd0
PROGNAME=$(basename $0)
PATH=/sbin:/bin:/usr/sbin:/usr/bin:/usr/local/sbin:/usr/local/bin
TUNE2FS=tune2fs
RESIZE2FS=resize2fs
PARTED=parted
+SFDISK=sfdisk
+MKSWAP=mkswap
+BLKID=blkid
+BLOCKDEV=blockdev
REGLOOKUP=reglookup
CHNTPW=chntpw
CLEANUP=( )
+ERRORS=( )
+WARNINGS=( )
+
+MSG_TYPE_TASK_START="TASK_START"
+MSG_TYPE_TASK_END="TASK_END"
+
+STDERR_LINE_SIZE=10
+
+
+# hypervisor param is passed in /proc/cmdline
+case "$hypervisor" in
+ kvm)
+ FLOPPY_DEV=/dev/fd0
+ ROOT_FSTAB_ENTRY=/dev/sda1
+ IMG_DEV=/dev/vda
+ RESULT=/dev/ttyS1
+ MONITOR=/dev/ttyS2
+ ;;
+ xen-hvm|xen-pvm)
+ FLOPPY_DEV=/dev/xvdc1
+ ROOT_FSTAB_ENTRY=/dev/xvda1
+ IMG_DEV=/dev/xvdb
+ RESULT=/dev/hvc0
+ MONITOR=/dev/hvc0
+ ;;
+esac
+
+to_monitor() {
+
+ echo "HELPER_MONITOR_$@" > "$MONITOR"
+
+}
+
+to_result() {
+
+ echo "HELPER_RESULT_$@" > "$RESULT"
+
+}
+
add_cleanup() {
local cmd=""
}
log_error() {
- echo "ERROR: $@" | tee $RESULT >&2
+ ERRORS+=("$@")
+ echo "ERROR: $@" >&2
+ to_result "ERROR: $@"
exit 1
}
warn() {
echo "Warning: $@" >&2
+ to_monitor "WARNING:$@"
+}
+
+report_task_start() {
+ to_monitor "$MSG_TYPE_TASK_START:${PROGNAME:2}"
+}
+
+report_task_end() {
+ to_monitor "$MSG_TYPE_TASK_END:${PROGNAME:2}"
+}
+
+report_error() {
+ if [ ${#ERRORS[*]} -eq 0 ]; then
+ # No error message. Print stderr
+ lines="$(tail --lines=${STDERR_LINE_SIZE} "$STDERR_FILE")"
+ cnt=$(echo $lines | wc -l)
+ for line in lines; do
+ to_monitor "$STDERR:$cnt: $line"
+ let cnt--
+ done
+ else
+ for line in "${ERRORS[@]}"; do
+ to_monitor "ERROR: $line"
+ done
+ fi
}
get_base_distro() {
echo "redhat"
elif [ -e "$root_dir/etc/slackware-version" ]; then
echo "slackware"
- elif [ -e "$root_dir/SuSE-release" ]; then
+ elif [ -e "$root_dir/etc/SuSE-release" ]; then
echo "suse"
- elif [ -e "$root_dir/gentoo-release" ]; then
+ elif [ -e "$root_dir/etc/gentoo-release" ]; then
echo "gentoo"
+ elif [ -e "$root_dir/etc/arch-release" ]; then
+ echo "arch"
+ else
+ warn "Unknown base distro."
fi
}
echo "redhat"
elif [ -e "$root_dir/etc/slackware-version" ]; then
echo "slackware"
- elif [ -e "$root_dir/SuSE-release" ]; then
+ elif [ -e "$root_dir/etc/SuSE-release" ]; then
echo "suse"
- elif [ -e "$root_dir/gentoo-release" ]; then
+ elif [ -e "$root_dir/etc/gentoo-release" ]; then
echo "gentoo"
+ elif [ -e "$root_dir/etc/arch-release" ]; then
+ echo "arch"
+ else
+ warn "Unknown distro."
fi
}
-get_last_partition() {
+
+get_partition_table() {
local dev="$1"
+ # If the partition table is gpt then parted will raise an error if the
+ # secondary gpt is not it the end of the disk, and a warning that has to
+ # do with the "Last Usable LBA" entry in gpt.
+ if ! output="$("$PARTED" -s -m "$dev" unit s print | grep -E -v "^(Warning|Error): ")"; then
+ log_error "Unable to read partition table for device \`${dev}'"
+ fi
- "$PARTED" -s -m "$dev" unit s print | tail -1
+ echo "$output"
}
-get_partition() {
- local dev="$1"
- local id="$2"
+get_partition_table_type() {
+ local ptable="$1"
+
+ local dev="$(sed -n 2p <<< "$ptable")"
+ declare -a field
+ IFS=':' read -ra field <<< "$dev"
- "$PARTED" -s -m "$dev" unit s print | grep "^$id"
+ echo "${field[5]}"
}
get_partition_count() {
+ local ptable="$1"
+
+ expr $(echo "$ptable" | wc -l) - 2
+}
+
+get_partition_by_num() {
+ local ptable="$1"
+ local id="$2"
+
+ grep "^$id:" <<< "$ptable"
+}
+
+get_last_partition() {
+ local ptable="$1"
+
+ echo "$ptable" | tail -1
+}
+
+is_extended_partition() {
local dev="$1"
+ local part_num="$2"
- expr $("$PARTED" -s -m "$dev" unit s print | wc -l) - 2
+ id=$($SFDISK --print-id "$dev" "$part_num")
+ if [ "$id" = "5" ]; then
+ echo "yes"
+ else
+ echo "no"
+ fi
+}
+
+get_extended_partition() {
+ local ptable="$1"
+ local dev="$(echo "$ptable" | sed -n 2p | cut -d':' -f1)"
+
+ tail -n +3 <<< "$ptable" | while read line; do
+ part_num=$(cut -d':' -f1 <<< "$line")
+ if [ $(is_extended_partition "$dev" "$part_num") == "yes" ]; then
+ echo "$line"
+ return 0
+ fi
+ done
+ echo ""
+}
+
+get_logical_partitions() {
+ local ptable="$1"
+
+ tail -n +3 <<< "$ptable" | while read line; do
+ part_num=$(cut -d':' -f1 <<< "$line")
+ if [ $part_num -ge 5 ]; then
+ echo "$line"
+ fi
+ done
+
+ return 0
+}
+
+get_last_primary_partition() {
+ local ptable="$1"
+ local dev=$(echo "ptable" | sed -n 2p | cut -d':' -f1)
+
+ for i in 4 3 2 1; do
+ if output=$(grep "^$i:" <<< "$ptable"); then
+ echo "$output"
+ return 0
+ fi
+ done
+ echo ""
+}
+
+get_partition_to_resize() {
+ local dev="$1"
+
+ table=$(get_partition_table "$dev")
+
+ if [ $(get_partition_count "$table") -eq 0 ]; then
+ return 0
+ fi
+
+ table_type=$(get_partition_table_type "$table")
+ last_part=$(get_last_partition "$table")
+ last_part_num=$(cut -d: -f1 <<< "$last_part")
+
+ if [ "$table_type" == "msdos" -a $last_part_num -gt 4 ]; then
+ extended=$(get_extended_partition "$table")
+ last_primary=$(get_last_primary_partition "$table")
+ ext_num=$(cut -d: -f1 <<< "$extended")
+ prim_num=$(cut -d: -f1 <<< "$last_primary")
+
+ if [ "$ext_num" != "$last_prim_num" ]; then
+ echo "$last_prim_num"
+ else
+ echo "$last_part_num"
+ fi
+ else
+ echo "$last_part_num"
+ fi
+}
+
+create_partition() {
+ local device="$1"
+ local part="$2"
+ local ptype="$3"
+
+ declare -a fields
+ IFS=":;" read -ra fields <<< "$part"
+ local id="${fields[0]}"
+ local start="${fields[1]}"
+ local end="${fields[2]}"
+ local size="${fields[3]}"
+ local fs="${fields[4]}"
+ local name="${fields[5]}"
+ local flags="${fields[6]//,/ }"
+
+ $PARTED -s -m -- $device mkpart "$ptype" $fs "$start" "$end"
+ for flag in $flags; do
+ $PARTED -s -m $device set "$id" "$flag" on
+ done
+}
+
+enlarge_partition() {
+ local device="$1"
+ local part="$2"
+ local ptype="$3"
+ local new_end="$4"
+
+ if [ -z "$new_end" ]; then
+ new_end=$(cut -d: -f 3 <<< "$(get_last_free_sector "$device")")
+ fi
+
+ declare -a fields
+ IFS=":;" read -ra fields <<< "$part"
+ fields[2]="$new_end"
+
+ local new_part=""
+ for ((i = 0; i < ${#fields[*]}; i = i + 1)); do
+ new_part="$new_part":"${fields[$i]}"
+ done
+ new_part=${new_part:1}
+
+ # If this is an extended partition, removing it will also remove the
+ # logical partitions it contains. We need to save them for later.
+ if [ "$ptype" = "extended" ]; then
+ local table="$(get_partition_table "$device")"
+ local logical="$(get_logical_partitions "$table")"
+ fi
+
+ id=${fields[0]}
+ $PARTED -s -m "$device" rm "$id"
+ create_partition "$device" "$new_part" "$ptype"
+
+ if [ "$ptype" = "extended" ]; then
+ # Recreate logical partitions
+ echo "$logical" | while read logical_part; do
+ create_partition "$device" "$logical_part" "logical"
+ done
+ fi
}
get_last_free_sector() {
local dev="$1"
- local last_line=$("$PARTED" -s -m "$dev" unit s print free | tail -1)
- local type=$(echo "$last_line" | cut -d: -f 5)
+ local unit="$2"
+
+ if [ -n "$unit" ]; then
+ unit="unit $unit"
+ fi
+
+ local last_line="$("$PARTED" -s -m "$dev" "$unit" print free | tail -1)"
+ local ptype="$(cut -d: -f 5 <<< "$last_line")"
- if [ "$type" = "free;" ]; then
- echo "$last_line" | cut -d: -f 3
+ if [ "$ptype" = "free;" ]; then
+ echo "$last_line"
fi
}
# before we give up with an error. This is needed for kpartx when
# dealing with ntfs partitions mounted through fuse. umount is not
# synchronous and may return while the partition is still busy. A
- # premature attempt to delete partition mappings through kpartx on a
- # device that hosts previously mounted ntfs partition may fail with
- # a `device-mapper: remove ioctl failed: Device or resource busy'
- # error. A sensible workaround for this is to wait for a while and
- # then try again.
+ # premature attempt to delete partition mappings through kpartx on
+ # a device that hosts previously mounted ntfs partition may fail
+ # with a `device-mapper: remove ioctl failed: Device or resource
+ # busy' error. A sensible workaround for this is to wait for a
+ # while and then try again.
local cmd=${CLEANUP[$i]}
$cmd || for interval in 0.25 0.5 1 2 4; do
echo "Command $cmd failed!"
fi
}
-check_if_excluded() {
+task_cleanup() {
+ rc=$?
+
+ if [ $rc -eq 0 ]; then
+ report_task_end
+ else
+ report_error
+ fi
- local exclude=SNF_IMAGE_PROPERTY_EXCLUDE_TASK_${PROGNAME:2}
+ cleanup
+}
+
+check_if_excluded() {
+ local name="$(tr [a-z] [A-Z] <<< ${PROGNAME:2})"
+ local exclude="SNF_IMAGE_PROPERTY_EXCLUDE_TASK_${name}"
if [ -n "${!exclude}" ]; then
- warn "Task $PROGNAME was excluded and will not run."
+ warn "Task ${PROGNAME:2} was excluded and will not run."
exit 0
fi
return 0
}
+
+return_success() {
+
+ case $hypervisor in
+ kvm)
+ echo "SUCCESS" > "$RESULT"
+ ;;
+ xen-pvm|xen-kvm)
+ domid=$(xenstore-read domid)
+ xenstore-write /local/domain/0/helper/$domid SUCCESS
+ ;;
+ esac
+
+}
+
trap cleanup EXIT
+set -o pipefail
+
+STDERR_FILE=$(mktemp)
+add_cleanup rm -f "$STDERR_FILE"
+exec 2> >(tee -a "$STDERR_FILE" >&2)
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :
projects / snf-image / blobdiff