meta-netboot/recipes-core/initramfs-netboot/files/init.sh

   1 #!/bin/sh
   2
   3 ################################################################################
   4 #
   5 # Init script to boot over network through NBD
   6 #
   7 # Contact: Stéphane Desneux <stephane.desneux@iot.bzh>
   8 #
   9 ################################################################################
  10
  11 # banner generator: echo "AGL - Netboot" | figlet -f slant -w 80 -c
  12 cat <<'EOF' >&2
  13 ________________________________________________________________________________
  14          ___   ________                _   __     __  __                __
  15         /   | / ____/ /               / | / /__  / /_/ /_  ____  ____  / /_
  16        / /| |/ / __/ /      ______   /  |/ / _ \/ __/ __ \/ __ \/ __ \/ __/
  17       / ___ / /_/ / /___   /_____/  / /|  /  __/ /_/ /_/ / /_/ / /_/ / /_
  18      /_/  |_\____/_____/           /_/ |_/\___/\__/_.___/\____/\____/\__/
  19 ________________________________________________________________________________
  20 EOF
  21
  22 # global variables
  23
  24 SMACK=n
  25 NBD_SERVER=
  26 NBD_PORT=10809
  27 NBD_DEV=/dev/nbd0
  28 NBD_NAMEV3=
  29 DEBUG=n
  30
  31 # -------------------------------------------
  32
  33 log_info() { echo "$0[$$]: $@" >&2; }
  34 log_error() { echo "$0[$$]: ERROR $@" >&2; }
  35
  36 do_mount_fs() {
  37         log_info "mounting FS: $@"
  38         [[ -e /proc/filesystems ]] && { grep -q "$1" /proc/filesystems || { log_error "Unknown filesystem"; return 1; } }
  39         [[ -d "$2" ]] || mkdir -p "$2"
  40         [[ -e /proc/mounts ]] && { grep -q -e "^$1 $2 $1" /proc/mounts && { log_info "$2 ($1) already mounted"; return 0; } }
  41         mount -t "$1" "$1" "$2"
  42 }
  43
  44 bail_out() {
  45         log_error "$@"
  46         check_debug "Reboot will occur after exiting this shell."
  47         log_info "Rebooting..."
  48         exec reboot -f
  49 }
  50
  51 check_debug() {
  52         case $DEBUG in
  53                 Y|y|yes|1|true)
  54                         log_info "$@"
  55                         /bin/sh -i
  56                         ;;
  57         esac
  58 }
  59
  60 # -------------------------------------------
  61
  62 export PATH=/sbin:/usr/sbin:/bin:/usr/bin
  63
  64 log_info "starting initrd script"
  65
  66 do_mount_fs proc /proc
  67 do_mount_fs sysfs /sys
  68 do_mount_fs devtmpfs /dev
  69 do_mount_fs devpts /dev/pts
  70 do_mount_fs tmpfs /dev/shm
  71 do_mount_fs tmpfs /tmp
  72 do_mount_fs tmpfs /run
  73
  74 # parse kernel commandline to get NBD server
  75 for x in $(cat /proc/cmdline); do
  76         case $x in
  77                 nbd.server=*) NBD_SERVER=${x/*=/};;
  78                 nbd.port=*) NBD_PORT=${x/*=/};;
  79                 nbd.dev=*)  NBD_DEV=/dev/${x/*=/};;
  80                 nbd.namev3=*) NBD_NAMEV3=${x/*=/};;
  81                 nbd.debug=*) DEBUG=${x/*=/};;
  82         esac
  83 done
  84
  85 check_debug "Debug point 1. Exit to continue initrd script (mount NBD device)."
  86
  87 log_info "NBD parameters: device $NBD_DEV, server $NBD_SERVER:$NBD_PORT"
  88
  89 # check if smack is active (and if so, mount smackfs)
  90 grep -q smackfs /proc/filesystems && {
  91         SMACK=y
  92
  93         do_mount_fs smackfs /sys/fs/smackfs
  94
  95         # adjust current label and network label
  96         echo System >/proc/self/attr/current
  97         echo System >/sys/fs/smackfs/ambient
  98 }
  99
 100 # start nbd client
 101 try=5
 102 while :;do
 103         log_info "Starting NBD client"
 104         if [ -z "${NBD_NAMEV3}" ]; then
 105                 nbd-client $NBD_SERVER $NBD_PORT $NBD_DEV && { log_info "NBD client successfully started"; break; }
 106                 log_info "NBD client failed"
 107         else
 108                 nbd3-client $NBD_SERVER $NBD_DEV --name $NBD_NAMEV3 && { log_info "NBD3 client successfully started"; break; }
 109                 log_info "NBDv3 client failed"
 110         fi
 111         [[ $try -gt 0 ]] && { log_info "Retrying ($try trie(s) left)..."; sleep 3; try=$(( try - 1 )); continue; }
 112
 113         bail_out "Unable to mount NBD device $NBD_DEV using server $NBD_SERVER:$NBD_PORT"
 114 done
 115
 116 # mount NBD device
 117 mkdir -p /sysroot
 118 mount $NBD_DEV -o noatime /sysroot || bail_out "Unable to mount root NBD device"
 119
 120 # move mounted devices to new root
 121 cd /sysroot
 122 for x in dev proc sys tmp run; do
 123         log_info "Moving /$x to new rootfs"
 124         mount -o move /$x $x
 125 done
 126
 127 # switch to new rootfs
 128 log_info "Switching to new rootfs"
 129 mkdir -p boot/initramfs
 130 pivot_root . boot/initramfs || bail_out "pivot_root failed."
 131
 132 # workaround for connman (avoid bringing down the network interface used for booting, disable DNS proxy)
 133 if [[ -f /lib/systemd/system/connman.service ]]; then
 134         log_info "Adjusting Connman configuration"
 135         iface=$(ip -o link show up | tr ':' ' ' | awk '{print $2}' | grep -v -e "^lo$" | head -1)
 136         sed -i "s|connmand -n\$|connmand -r -n -I $iface|g" /lib/systemd/system/connman.service
 137 fi
 138
 139 # also use /proc/net/pnp to generate /etc/resolv.conf
 140 rm -f /etc/resolv.conf
 141 grep -v bootserver /proc/net/pnp | sed 's/^domain/search/g' >/etc/resolv.conf
 142
 143 # unmount tmp and run to let systemd remount them with correct smack labels (SPEC-2596)
 144 log_info "Unmounting /tmp and /run"
 145 umount /tmp
 146 umount /run
 147
 148 # finally, run systemd
 149 check_debug "Debug point 2. Exit to continue initrd script (run systemd)."
 150
 151 log_info "Exec'ing systemd"
 152 # banner generator: echo "AGL Booting . . ." | figlet -f slant -w 80 -c
 153 cat <<'EOF' >&2
 154 ________________________________________________________________________________
 155       ___   ________       ____              __  _
 156      /   | / ____/ /      / __ )____  ____  / /_(_)___  ____ _
 157     / /| |/ / __/ /      / __  / __ \/ __ \/ __/ / __ \/ __ `/
 158    / ___ / /_/ / /___   / /_/ / /_/ / /_/ / /_/ / / / / /_/ /   _     _     _
 159   /_/  |_\____/_____/  /_____/\____/\____/\__/_/_/ /_/\__, /   (_)   (_)   (_)
 160 _____________________________________________________/____/_____________________
 161 EOF
 162
 163 exec /lib/systemd/systemd </dev/console >/dev/console 2>&1
 164 bail_out
 165