add cpus setting

[code/vinst.git] / vinst

diff --git a/vinst b/vinst
index bf0a14d99a0ac1f22282dba22008a6fbbb178b40..a07629a2e431e309ea7614f1fd6a84760e5be9c2 100755 (executable)
--- a/vinst
+++ b/vinst
@@ -5,6 +5,7 @@ set -e
 # defaults
 RAM=512
 DISK=20
+CPUS=1
 STORAGE_POOL=default
 QEMU_URI=qemu:///system
 SUITE=wheezy
@@ -15,6 +16,17 @@ RESTART=0
 
 while [ -n "$1" ]; do
   case "$1" in
+    (--debug)
+      DEBUG=1
+      ;;
+    (-S|--suite)
+      shift
+      if [ -z "$1" ]; then
+        echo >&2 'E: --suite/-S needs suite argument'
+        exit 1
+      fi
+      SUITE="$1"
+      ;;
     (-n|--name)
       shift
       if [ -z "$1" ]; then
@@ -86,6 +98,14 @@ while [ -n "$1" ]; do
       fi
       MAC="$1"
       ;;
+    (-c|--cpus)
+      shift
+      if [ -z "$1" ]; then
+        echo >&2 'E: --cpus/-c needs integer argument'
+        exit 1
+      fi
+      CPUS="$1"
+      ;;
     (--debug)
       DEBUG=1
       set -vx
@@ -93,7 +113,7 @@ while [ -n "$1" ]; do
     (--wipe)
       WIPE=1
       ;;
-    (--console)
+    (--restart)
       RESTART=1
       ;;
   esac
@@ -110,10 +130,15 @@ BASEDIR=$(cd ${0%/*}; pwd)
 tmpdir=$(mktemp -d)
 tar -C $BASEDIR/preseed/$SUITE -cf $tmpdir/commands.tar commands
 
-extra_args="auto $CONSOLE \
+extra_kargs="auto $CONSOLE \
 hostname=${HOSTNAME%%.*} \
 domain=${HOSTNAME#*.}"
 
+extra_opts=
+if [ -n "$DEBUG" ]; then
+  extra_opts='--debug'
+fi
+
 if [ -n "$IPADDRESS" ]; then
   if [ -z "$GATEWAY" ] && [ -x $(command -v ipcalc) ]; then
     # default to the first IP in the network
@@ -123,7 +148,7 @@ if [ -n "$IPADDRESS" ]; then
   IPADDRESS="${IPADDRESS%/*}"
   : ${NAMESERVER:=$GATEWAY}
 
-  extra_args="$extra_args netcfg/disable_dhcp=true \
+  extra_kargs="$extra_kargs netcfg/disable_dhcp=true \
   netcfg/get_ipaddress=$IPADDRESS \
   netcfg/get_netmask=$NETMASK \
   netcfg/get_gateway=$GATEWAY \
@@ -132,7 +157,7 @@ if [ -n "$IPADDRESS" ]; then
   BRIDGE=wan
 
 else
-  extra_args="$extra_args netcfg/disable_dhcp=false"
+  extra_kargs="$extra_kargs netcfg/disable_dhcp=false"
   BRIDGE=virt-br
 fi
 
@@ -141,15 +166,17 @@ if [ -z "$ARCH" ]; then
 fi
 
 if [ -z "$MAC" ]; then
-  bytes=$(dd status=none if=/dev/urandom bs=3 count=1 | hexdump -v -e '/1 ":%02X"')
+  bytes=$(dd status=noxfer if=/dev/urandom bs=3 count=1 | hexdump -v -e '/1 ":%02X"')
   MAC=52:54:00$bytes
 fi
 
 cat >&2 <<_eof
   hostname: $HOSTNAME
       arch: $ARCH
+     suite: $SUITE
        ram: $RAM MiB
       disk: $DISK GiB
+      cpus: $CPUS
       wipe: $WIPE
    console: $CONSOLE
    restart: $RESTART
@@ -173,6 +200,7 @@ echo >&2
 
 virsh --connect=$QEMU_URI vol-create-as $STORAGE_POOL virt-$HOSTNAME ${DISK}G
 if [ $WIPE = 1 ]; then
+  echo >&2 "Wiping disk…"
   virsh --connect=$QEMU_URI vol-wipe --pool $STORAGE_POOL --algorithm zero virt-$HOSTNAME
 fi
 
@@ -184,22 +212,23 @@ fi
 virt-install --connect=$QEMU_URI \
   ${DEBUGFLAG:-} \
   -n $HOSTNAME \
+  --vcpus=$CPUS \
   -r $RAM \
   --disk vol=$STORAGE_POOL/virt-$HOSTNAME \
   --network bridge=$BRIDGE,mac=$MAC \
   --graphics=vnc \
   --serial=pty \
   --watchdog i6300esb,action=reset \
-  --os-variant=debian$SUITE \
-  -l $BASEDIR/installer-amd64 \
+  --os-variant=debianwheezy \
   --initrd-inject=$BASEDIR/preseed/$SUITE/preseed.cfg \
   --initrd-inject=$tmpdir/commands.tar \
-  --noautoconsole --autostart \
-  --extra-args="$extra_args" \
+  --extra-args="$extra_kargs" \
+  --prompt --noautoconsole --autostart $extra_opts \
+  -l $BASEDIR/installer-amd64/$SUITE \
   "$@"
 
 virsh --connect=$QEMU_URI console $HOSTNAME
 
-if [ $RESTART = 1]; then
+if [ $RESTART = 1 ]; then
   virsh --connect=$QEMU_URI "start $HOSTNAME; console $HOSTNAME"
 fi