iankelling.org Git - automated-distro-installer/blob - fai/config/files/root/fai-check/VOL_STABLE_BOOTSTRAP

   1 #!/bin/bash
   2
   3 set -eE -o pipefail
   4 trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR
   5
   6
   7
   8 usage() {
   9     cat <<EOF
  10 Usage: ${0##*/} [OPTION]
  11 If grub var set, act like pxe rom and pxe-kexec to faiserver
  12
  13 -f|--force  do kexec if we can reach faiserver
  14 -h|--help  Print help and exit.
  15
  16 Note: Uses GNU getopt options parsing style
  17 EOF
  18     exit $1
  19 }
  20
  21
  22
  23 # Keep it short so we don't delay too much wnen we don't have networking.
  24 # In practice, on my home network, on an x200, it took 15 seconds, so
  25 # give it an extra 10 seconds, which seems fairly short as I write this.
  26 NETWORK_TIMOUT_SECS=25
  27 did_fai_check=false
  28
  29 m() { printf "%s\n" "$*";  "$@"; }
  30
  31 try-kexec() {
  32     deadline=$(( `date +%s` + NETWORK_TIMOUT_SECS ))
  33     while ! nc -zu faiserver 69; do
  34         if (( `date +%s` > deadline )); then
  35             echo "fai-check: hit $NETWORK_TIMOUT_SECS s tftp server timeout"
  36             return 0
  37         fi
  38         sleep 1
  39     done
  40     m pxe-kexec -n --ignore-whitelist -l fai-generated faiserver ||:
  41 }
  42
  43 case $1 in
  44     -f|--force)
  45         try-kexec
  46         exit
  47         ;;
  48 esac
  49
  50 first=true
  51 for dev in $(btrfs fi show / | sed -rn 's#^\s*devid\s.*\s([^0-9 ]+)\S+$#\1#p' \
  52                  |sort); do
  53     dev+=4
  54     mount $dev /mnt
  55     if $first; then
  56         if [[ -e /mnt/grubenv ]]; then
  57             set -x
  58             source <(grub-editenv /mnt/grubenv list)
  59             set +x
  60         fi
  61         first=false
  62         # we could just as well check if last_boot != /debianstable_boostrap
  63         # the intent with this one is just a little clearer.
  64         if [[ $did_fai_check == true ]]; then
  65             grub-editenv /mnt/grubenv set did_fai_check=os_true
  66             # our service does not wait for network-online.target,
  67             # because it will wait for too long when we don't have a network
  68             # connection. So, we wait for 10 seconds.
  69             # ref: https://www.freedesktop.org/wiki/Software/systemd/NetworkTarget/
  70             try-kexec ||:
  71         else
  72             return 0
  73         fi
  74     else
  75         # we make sure there is only 1 grubenv,
  76         # so grub can just find the first one, in whatever order
  77         # if looks at them, which may not be the same as us.
  78         # If the disk dies, we just lose the default boot option,
  79         # we will have to do manual steps to replace it anyways.
  80         rm -f /mnt/gruvenv
  81     fi
  82     umount /mnt
  83 done
  84
  85 # the check for last_boot is not needed afaik, just sanity check.
  86 if [[ $did_fai_check == true && $last_boot != /debianstable_boostrap ]]; then
  87     # no need to reboot if we actually want to boot into this os.
  88     reboot
  89 fi