vzt-kernel-install

   1 #!/bin/bash
   2 # Install kernel to test host
   3 CONFIG=.vzt/kernel-install.cfg
   4 function show_usage
   5 {
   6         echo "$0: <host> [OPTS]"
   7         echo "Usage:"
   8         echo "Install kernele to test host via ssh or pxe"
   9         echo "Examples:"
  10         echo "Install kernel, modules and initrd to host via \
  11 ssh and setup local boot loader"
  12         echo "  $0 my_host.org --all"
  13         echo "Install kernel only via ssh and setup local boot loader"
  14         echo "  $0 my_host.org "
  15         echo "Install kernel and setup pxe boot loader"
  16         echo "  $0 my_host --pxe"
  17         echo "Disable pxe boot  for host"
  18         echo "  $0 my_host --pxe --disable"
  19
  20 }
  21
  22 if [ ! -f "$CONFIG" ]; then
  23         echo "Cant find .vzt/kernel-install.cfg"
  24         echo "you may use example config:"
  25         echo "  mkdir .vzt; cp /etc/vzt/kernel-install.cfg.example \
  26 .vzt/kernel-install.cfg "
  27         exit
  28 fi
  29 . $CONFIG
  30 krel=`make kernelrelease`
  31
  32 if [ ! $? -eq 0 ]; then
  33         echo "Can't get kernel release version"
  34         echo "This is probably happend CWD is wrong."
  35         echo "do: cd /your/path/to/kernel/linux-tree/"
  36         show_usage
  37         exit 1
  38 fi
  39 function SSH
  40 {
  41         ssh -n -oStrictHostKeyChecking=no -oBatchMode=yes $@
  42         return $?
  43 }
  44
  45 host="${1}"
  46 shift
  47 if [ -z "$host" ] ;then
  48         echo "Err: Bad arguments"
  49         show_usage
  50         exit 1
  51 fi
  52 boot_proto=local
  53 modules=n
  54 mkinitrd=n
  55 if [ "$1" = "--pxe" ]; then
  56         boot_proto="pxe"
  57         shift
  58 fi
  59
  60 if [ "$1" = "--all" ]; then
  61         modules=y
  62         mkinitrd=y
  63         shift
  64 fi
  65
  66 if [ "$1" = "--suffix" ]; then
  67         shift
  68         krel=${krel}-${1}
  69         shift
  70 fi
  71
  72 if [ "$1" = "--initrd" ]; then
  73         mkinitrd=y
  74         shift
  75 fi
  76
  77 if [ "$1" = "--disable" ]; then
  78         if [ "$boot_proto" = "pxe" ]; then
  79                 SSH root@$PXE_SERVER " vzt-pxe-boot $host del"
  80                 exit $?
  81         else
  82                 echo "ERR: --disable option valid only for pxe boot mode"
  83                 exit 1
  84         fi
  85 fi
  86
  87 title="kernel-$krel"
  88 echo "Install kernel to $server via protocol: $boot_proto"
  89
  90 if [ $modules == "y" ]; then
  91         make modules_install || exit 1;
  92         tar zcf modules-$krel.tar.gz /lib/modules/$krel \
  93 --exclude /lib/modules/$krel/build \
  94 --exclude /lib/modules/$krel/source || exit 1
  95         scp modules-$krel.tar.gz root@$host:/tmp || exit 1
  96         SSH root@$host "rm \-rf /lib/modules/$krel " || exit 1
  97         SSH root@$host " tar mzxf /tmp/modules-$krel.tar.gz -C / "|| exit 1
  98         SSH root@$host "mkinitrd \-f /boot/initrd-$krel.img $krel $MKINITRD_OPT" || exit 1
  99         SSH root@$host " wall 'modules was intalled to /lib/modules/$krel'"
 100         rm -f modules-$krel.tar.gz
 101         mkinitrd=n
 102 fi
 103
 104 ##### pxe boot #####
 105 if [ $boot_proto == "pxe" ] ;then
 106         echo "$BOOT_OPTS" | egrep "[^[:space:]]+"
 107         if [ $? -ne 0 ];then
 108                 echo "Err: 'BOOT_OPTS' is empty. You must specify correct boot options"
 109                 exit 1
 110         fi
 111         cp arch/$ARCH/boot/bzImage vmlinuz-$krel || exit 1
 112         vmlinuz-$krel root@$PXE_SERVER:/$PXE_ROOT/$PXE_KERNEL_DIR/vmlinuz-$krel \
 113                 || exit 1
 114         rm -f vmlinuz-$krel || exit 1
 115         SSH root@$PXE_SERVER " chmod 0444 /$PXE_ROOT/$PXE_KERNEL_DIR/vmlinuz-$krel"
 116         if [ $mkinitrd == "y" ]; then
 117                 echo "Using initrd from different host is realy errorprone way."
 118                 echo "kernel probably can't boot with this initrd image"
 119                 scp initrd-$krel.img root@$PXE_SERVER:/$PXE_ROOT/$PXE_KERNEL_DIR || exit 1
 120                 SSH root@$PXE_SERVER " chmod 0444 /$PXE_ROOT/$PXE_KERNEL_DIR/initrd-$krel.img "
 121         fi
 122         SSH root@$PXE_SERVER " vzt-pxe-boot $host add vmlinuz-$krel initrd-$krel.img '$BOOT_OPTS' $title"
 123         exit $?
 124 fi
 125
 126 function grub_setup
 127 {
 128         host=$1
 129         shift
 130         kernel_img=$1
 131         shift
 132         initrd_img=$1
 133         shift
 134         title_text=$1
 135         shift
 136         boot_opt=$@
 137
 138         ssh root@${host} test  /boot/${kernel_img} || {
 139                 cho "Kernel image : /boot/${kernel_img} not found";
 140                 exit 1;
 141         }
 142
 143         ssh root@${host} test /boot/${initrd_img} || {
 144                 echo "initrd image :/boot/${initrd_img} not found" ;
 145                 exit 1;
 146         }
 147
 148         if [ -z "$boot_opt" ]; then
 149                 args=--copy-default
 150         else
 151                 args='--args=$boot_opt'
 152         fi
 153         ssh root@${host} grubby --remove-kernel=/boot/${kernel_img} \
 154                                 --add-kernel=/boot/${kernel_img} \
 155                                 --initrd=/boot/${initrd_img} \
 156                                 $args --title="${title_text}" \
 157                                 --make-default
 158         ret=$?
 159         if [ $ret -ne 0 ] ;then
 160                 echo "Error while grub setup"
 161         else
 162                 SSH root@$host " wall 'grub default was changed to /boot/\
 163 ${kernel_img}'"
 164                 echo "grub setup OK"
 165         fi
 166         return $ret
 167 }
 168
 169
 170 ##### local boot  #####
 171 if [ $boot_proto == "local" ]; then
 172         cp arch/$ARCH/boot/bzImage vmlinuz-$krel || exit
 173         scp vmlinuz-$krel root@$host:/boot/vmlinuz-$krel || exit 1
 174         rm -f vmlinuz-$krel || exit 1
 175         SSH root@$host "wall 'new kernel was intalled to /boot/vmlinuz-$krel'"
 176         if [ $mkinitrd == "y" ]; then
 177                 SSH root@$host "mkinitrd \-f /boot/initrd-$krel.img $krel $MKINITRD_OPT" || exit 1
 178         fi
 179         grub_setup $host vmlinuz-$krel initrd-$krel.img $title $BOOT_OPTS \
 180         #SSH root@$host " ls -l  "
 181         exit $?
 182 fi
 183 #################
 184 echo "Bad boot protocol" ;
 185 exit 1
 186