--- /dev/null
+#!/bin/bash -
+# @configure_input@
+# libguestfs virt-sysprep tool
+# Copyright (C) 2011 Red Hat Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+unset CDPATH
+program="virt-sysprep"
+version="@PACKAGE_VERSION@"
+
+TEMP=`getopt \
+ -o a:c:d:vVx \
+ --long help,add:,connect:,domain:,enable:,format::,hostname:,list-operations,verbose,version \
+ -n $program -- "$@"`
+if [ $? != 0 ]; then
+ echo "$program: problem parsing the command line arguments"
+ exit 1
+fi
+eval set -- "$TEMP"
+
+# This array accumulates the arguments we pass through to guestfish.
+declare -a guestfish
+guestfish[0]="guestfish"
+guestfish[1]="--rw"
+guestfish[2]="--listen"
+guestfish[3]="-i"
+i=4
+
+verbose=
+add_params=0
+enable=
+hostname_param=localhost.localdomain
+
+usage ()
+{
+ echo "Usage:"
+ echo " $program [--options] -d domname"
+ echo " $program [--options] -a disk.img [-a disk.img ...]"
+ echo
+ echo "Read $program(1) man page for more information."
+ echo
+ echo "NOTE: $program modifies the guest or disk image *in place*."
+ exit $1
+}
+
+while true; do
+ case "$1" in
+ -a|--add)
+ guestfish[i++]="-a"
+ guestfish[i++]="$2"
+ ((add_params++))
+ shift 2;;
+ -c|--connect)
+ guestfish[i++]="-c"
+ guestfish[i++]="$2"
+ shift 2;;
+ -d|--domain)
+ guestfish[i++]="-d"
+ guestfish[i++]="$2"
+ ((add_params++))
+ shift 2;;
+ --enable)
+ if [ -n "$enable" ]; then
+ echo "error: --enable option can only be given once"
+ exit 1
+ fi
+ enable="$2"
+ shift 2;;
+ --format)
+ if [ -n "$2" ]; then
+ guestfish[i++]="--format=$2"
+ else
+ guestfish[i++]="--format"
+ fi
+ shift 2;;
+ --help)
+ usage 0;;
+ --hostname)
+ hostname_param="$2"
+ shift 2;;
+ --list-operations)
+ enable=list
+ shift;;
+ -v|--verbose)
+ guestfish[i++]="-v"
+ verbose=yes
+ shift;;
+ -V|--version)
+ echo "$program $version"
+ exit 0;;
+ -x)
+ guestfish[i++]="-x"
+ shift;;
+ --)
+ shift
+ break;;
+ *)
+ echo "Internal error!"
+ exit 1;;
+ esac
+done
+
+# Different sysprep operations that can be enabled. Default is to
+# enable all of these, although some of them are only done on certain
+# guest types (see details below).
+if [ -z "$enable" ]; then
+ hostname=yes
+ net_hwaddr=yes
+ ssh_hostkeys=yes
+ udev_persistent_net=yes
+elif [ "$enable" = "list" ]; then
+ echo "hostname"
+ echo "net-hwaddr"
+ echo "ssh-hostkeys"
+ echo "udev-persistent-net"
+ exit 0
+else
+ for opt in $(echo "$enable" | sed 's/,/ /g'); do
+ case "$opt" in
+ hostname) hostname=yes ;;
+ net-hwaddr) net_hwaddr=yes ;;
+ ssh-hostkeys) ssh_hostkeys=yes ;;
+ udev-persistent-net) udev_persistent_net=yes ;;
+ *)
+ echo "error: unknown --enable feature: $opt"
+ exit 1
+ esac
+ done
+fi
+
+# Make sure there were no extra parameters on the command line.
+if [ $# -gt 0 ]; then
+ echo "error: $program: extra parameters on the command line"
+ echo
+ usage 1
+fi
+
+# Did the user specify at least one -a or -d option?
+if [ $add_params -eq 0 ]; then
+ echo "error: $program: you need at least one -a or -d option"
+ echo
+ usage 1
+fi
+
+# end of command line parsing
+#----------------------------------------------------------------------
+
+set -e
+
+if [ "$verbose" = "yes" ]; then
+ echo command: "${guestfish[@]}"
+fi
+
+# Create a temporary directory for general purpose use during operations.
+tmpdir="$(mktemp -d)"
+
+# Call guestfish.
+GUESTFISH_PID=
+eval $("${guestfish[@]}")
+if [ -z "$GUESTFISH_PID" ]; then
+ echo "$program: guestfish didn't start up, see error messages above"
+ exit 1
+fi
+
+cleanup ()
+{
+ kill $GUESTFISH_PID >/dev/null 2>&1 ||:
+ rm -rf "$tmpdir" ||:
+}
+trap cleanup EXIT
+
+# Launch back-end, inspect for operating systems, and get the guest
+# root disk.
+root=$(guestfish --remote inspect-get-roots)
+
+if [ "$root" = "" ]; then
+ echo "$program: no operating system was found on this disk"
+ exit 1
+fi
+
+if [ "$verbose" = "yes" ]; then
+ echo root: "$root"
+fi
+
+# Get the guest type.
+type="$(guestfish --remote -- -inspect-get-type $root)"
+
+if [ "$type" = "linux" ]; then
+ distro="$(guestfish --remote -- -inspect-get-distro $root)"
+fi
+
+if [ "$type" = "windows" ]; then
+ systemroot="$(guestfish --remote -- -inspect-get-windows-systemroot $root)"
+fi
+
+#----------------------------------------------------------------------
+# Useful functions.
+
+# erase_line filename regex
+#
+# Erase line(s) in a file that match the given regex.
+erase_line ()
+{
+ guestfish --remote -- download "$1" "$tmpdir/file"
+ sed "/$2/d" < "$tmpdir/file" > "$tmpdir/file.1"
+ guestfish --remote -- upload "$tmpdir/file.1" "$1"
+}
+
+# rm_files wildcard
+#
+# Remove files. Doesn't fail if no files exist. Note the wildcard
+# parameter cannot contain spaces or characters that need special
+# quoting.
+rm_files ()
+{
+ files=$(guestfish --remote -- glob-expand "$1")
+ for f in $files; do
+ guestfish --remote -- rm "$f"
+ done
+}
+
+# rm_file filename
+#
+# Remove a single file. No error if the file doesn't exist or is not
+# a file.
+rm_file ()
+{
+ t=$(guestfish --remote -- is-file "$1")
+ if [ "$t" = "true" ]; then
+ guestfish --remote -- rm "$1"
+ fi
+}
+
+#----------------------------------------------------------------------
+# The sysprep operations.
+
+if [ "$hostname" = "yes" ]; then
+ case "$type/$distro" in
+ linux/fedora)
+ guestfish --remote -- \
+ download /etc/sysconfig/network "$tmpdir/network"
+ echo "HOSTNAME=$hostname_param" > "$tmpdir/network.1"
+ sed '/^HOSTNAME=/d' < "$tmpdir/network" >> "$tmpdir/network.1"
+ guestfish --remote -- \
+ upload "$tmpdir/network.1" /etc/sysconfig/network ;;
+ linux/debian|linux/ubuntu)
+ guestfish --remote -- write /etc/hostname "$hostname_param"
+ esac
+fi
+
+if [ "$net_hwaddr" = "yes" ]; then
+ case "$type/$distro" in
+ linux/fedora)
+ # XXX these filenames can have spaces and untrusted chars in them!
+ files=$(guestfish --remote -- glob-expand '/etc/sysconfig/network-scripts/ifcfg-*')
+ for f in $files; do
+ erase_line "$f" "^HWADDR="
+ done
+ esac
+fi
+
+if [ "$ssh_hostkeys" = "yes" -a "$type" != "windows" ]; then
+ rm_files "/etc/ssh/*_host_*"
+fi
+
+if [ "$udev_persistent_net" = "yes" -a "$type" = "linux" ]; then
+ rm_file /etc/udev/rules.d/70-persistent-net.rules
+fi
+
+# Clean up and close down.
+
+guestfish --remote umount-all
+guestfish --remote sync
+guestfish --remote exit
+
+exit 0
git.annexia.org / libguestfs.git / blobdiff