zfsbootmenu: refactor installation, support system and user runtime hooks

- All core components, docs and runtime hooks are handled by a common
  install_helper that is aware of both dracut and mkinitcpio methods

- Installation now adds "system" runtime hooks in addition to those
  configured by the user; user hooks with names that conflict with
  system hooks will "mask" the system version

- The hooks `10-console-init.sh` and `xhci-teardown.sh` have been
  elevated to "system" status and are always installed in ZBM images

- The `zfsbootmenu/hook` path has been renamed to `zfsbootmenu/pre-init`
  to avoid confusion with the new `zfsbootmenu/hooks` directory of
  system runtime hooks, and the dracut-only `hook` scripts have been
  moved to the dracut module

- A new `zbm.skip_hooks` command-line argument allows a comma-separated
  list of hooks to skip (matched on basename only) to provide an easy
  mechanism for skipping default hooks without `zbm.hookroot`

- Create the `zfsbootmenu-run-hooks` libexec helper, allowing easy
  addition of future hook points.

Co-authored-by: Zach Dykstra <[email protected]>
Co-authored-by: Andrew J. Hesford <[email protected]>

Author: 3 people

docs/man/zfsbootmenu.7.rst

@@ -133,6 +133,12 @@ These options are set on the kernel command line when booting the initramfs or U
 
     zbm.kcl_override="some alternate set='of arguments'"
 
+**zbm.skip_hooks=<hooklist>**
+
+  Skip execution of any early-setup, setup or teardown hooks with file names matching any entry in the comma-separated list *hooklist*. Only base names of hooks (*i.e.*, with any other path component removed) are matched against the *hooklist*.
+
+  **NOTE**: The *hooklist* argument **MUST NOT** contain spaces and **MUST NOT** be enclosed in quotes.
+
 Deprecated Command-Line Parameters
 ==================================
 

dracut/module-setup.sh

@@ -40,8 +40,9 @@ install() {
   # BUILDROOT is an initcpio-ism
   # shellcheck disable=SC2154,2034
   BUILDROOT="${initdir}"
+
   # shellcheck disable=SC2034
-  BUILDSTYLE="dracut"
+  ZBM_BUILDSTYLE="dracut"
 
   local _rule _exec _ret
 
@@ -84,90 +85,43 @@ install() {
     fi
   done <<< "${_libgcc_s}"
 
-  # shellcheck disable=SC2154
-  while read -r doc ; do
-    relative="${doc//${zfsbootmenu_module_root}\//}"
-    inst_simple "${doc}" "/usr/share/docs/${relative}"
-  done <<<"$( find "${zfsbootmenu_module_root}/help-files" -type f )"
-
   compat_dirs=( "/etc/zfs/compatibility.d" "/usr/share/zfs/compatibility.d/" )
   for compat_dir in "${compat_dirs[@]}"; do
     # shellcheck disable=2164
     [ -d "${compat_dir}" ] && tar -cf - "${compat_dir}" | ( cd "${initdir}" ; tar xfp - )
   done
-  _ret=0
 
-  # Core ZFSBootMenu functionality
-  # shellcheck disable=SC2154
-  for _lib in "${zfsbootmenu_module_root}"/lib/*; do
-    inst_simple "${_lib}" "/lib/$( basename "${_lib}" )" || _ret=$?
-  done
+  _ret=0
 
-  # Helper tools not intended for direct human consumption
-  for _libexec in "${zfsbootmenu_module_root}"/libexec/*; do
-    inst_simple "${_libexec}" "/libexec/$( basename "${_libexec}" )" || _ret=$?
-  done
+  # Install core ZFSBootMenu functionality
+  install_zbm_core || _ret=$?
 
-  # User-facing utilities, useful for running in a recovery shell
-  for _bin in "${zfsbootmenu_module_root}"/bin/*; do
-    inst_simple "${_bin}" "/bin/$( basename "${_bin}" )" || _ret=$?
-  done
+  # Install runtime hooks
+  install_zbm_hooks || _ret=$?
 
   # Hooks necessary to initialize ZBM
-  inst_hook cmdline 95 "${zfsbootmenu_module_root}/hook/zfsbootmenu-parse-commandline.sh" || _ret=$?
-  inst_hook pre-mount 90 "${zfsbootmenu_module_root}/hook/zfsbootmenu-preinit.sh" || _ret=$?
+  inst_hook cmdline 95 "${zfsbootmenu_module_root}/pre-init/zfsbootmenu-parse-commandline.sh" || _ret=$?
+  inst_hook pre-mount 90 "${zfsbootmenu_module_root}/pre-init/zfsbootmenu-preinit.sh" || _ret=$?
 
   # Hooks to force the dracut event loop to fire at least once
   # Things like console configuration are done in optional event-loop hooks
-  inst_hook initqueue/settled 99 "${zfsbootmenu_module_root}/hook/zfsbootmenu-ready-set.sh" || _ret=$?
-  inst_hook initqueue/finished 99 "${zfsbootmenu_module_root}/hook/zfsbootmenu-ready-chk.sh" || _ret=$?
-
-  # optionally enable early Dracut profiling
-  if [ -n "${dracut_trace_enable}" ]; then
-    inst_hook cmdline 00 "${zfsbootmenu_module_root}/profiling/profiling-lib.sh"
-  fi
-
-  # Install "early setup" hooks
   # shellcheck disable=SC2154
-  if [ -n "${zfsbootmenu_early_setup}" ]; then
-    for _exec in ${zfsbootmenu_early_setup}; do
-      if [ -x "${_exec}" ]; then
-        inst_simple "${_exec}" "/libexec/early-setup.d/$(basename "${_exec}")" || _ret=$?
-      else
-        dwarning "setup script (${_exec}) missing or not executable; cannot install"
-      fi
-    done
-  fi
-
-  # Install "setup" hooks
-  # shellcheck disable=SC2154
-  if [ -n "${zfsbootmenu_setup}" ]; then
-    for _exec in ${zfsbootmenu_setup}; do
-      if [ -x "${_exec}" ]; then
-        inst_simple "${_exec}" "/libexec/setup.d/$(basename "${_exec}")" || _ret=$?
-      else
-        dwarning "setup script (${_exec}) missing or not executable; cannot install"
-      fi
-    done
-  fi
-
-  # Install "teardown" hooks
-  # shellcheck disable=SC2154
-  if [ -n "${zfsbootmenu_teardown}" ]; then
-    for _exec in ${zfsbootmenu_teardown}; do
-      if [ -x "${_exec}" ]; then
-        inst_simple "${_exec}" "/libexec/teardown.d/$(basename "${_exec}")" || _ret=$?
-      else
-        dwarning "teardown script (${_exec}) missing or not executable; cannot install"
-      fi
-    done
-  fi
+  inst_hook initqueue/settled 99 "${moddir}/zfsbootmenu-ready-set.sh" || _ret=$?
+  inst_hook initqueue/finished 99 "${moddir}/zfsbootmenu-ready-chk.sh" || _ret=$?
 
   if [ ${_ret} -ne 0 ]; then
     dfatal "Unable to install core ZFSBootMenu functions"
     exit 1
   fi
 
+  # Install online documentation if possible
+  install_zbm_docs
+
+  # optionally enable early Dracut profiling
+  if [ -n "${dracut_trace_enable}" ]; then
+    inst_hook cmdline 00 "${zfsbootmenu_module_root}/profiling/profiling-lib.sh"
+  fi
+
   # vdev_id.conf and hostid files are host-specific
   # and do not belong in public release images
   if [ -z "${release_build}" ]; then

zfsbootmenu/hook/zfsbootmenu-ready-chk.sh dracut/zfsbootmenu-ready-chk.shzfsbootmenu/hook/zfsbootmenu-ready-chk.sh renamed to dracut/zfsbootmenu-ready-chk.sh

@@ -1,5 +1,4 @@
-zfsbootmenu_teardown+=" /zbm/contrib/xhci-teardown.sh "
-zfsbootmenu_early_setup+=" /zbm/contrib/10-console-init.sh /zbm/contrib/20-console-autosize.sh "
+zfsbootmenu_early_setup+=" /zbm/contrib/20-console-autosize.sh "
 
 # zbm-kcl
 install_optional_items+=" /zbm/bin/zbm-kcl "

zfsbootmenu/hook/zfsbootmenu-ready-set.sh dracut/zfsbootmenu-ready-set.shzfsbootmenu/hook/zfsbootmenu-ready-set.sh renamed to dracut/zfsbootmenu-ready-set.sh

@@ -122,7 +122,7 @@ build() {
     source "${zfsbootmenu_module_root}/install-helpers.sh" || exit 1
 
     # shellcheck disable=SC2034
-    BUILDSTYLE="initcpio"
+    ZBM_BUILDSTYLE="mkinitcpio"
 
     # Modules (required and optional) used by ZBM
     # shellcheck disable=SC2154
@@ -154,52 +154,32 @@ build() {
         fi
     done <<< "${_libgcc_s}"
 
-    # On-line documentation
-    while read -r doc; do
-        relative="${doc#"${zfsbootmenu_module_root}/"}"
-        [ "${relative}" = "${doc}" ] && continue
-        add_file "${doc}" "/usr/share/docs/${relative}"
-    done <<< "$( find "${zfsbootmenu_module_root}/help-files" -type f )"
-
     # Install core ZBM functionality
-    for _file in "${zfsbootmenu_module_root}"/lib/*; do
-        add_file "${_file}" "/lib/${_file##*/}"
-    done
+    if ! install_zbm_core; then
+        error "Failed to install ZFSBootMenu core"
+        exit 1
+    fi
 
-    for _file in "${zfsbootmenu_module_root}"/libexec/*; do
-        add_file "${_file}" "/libexec/${_file##*/}"
-    done
+    # Install runtime hooks
+    if ! install_zbm_hooks; then
+        error "Failed to install runtime hooks"
+        exit 1
+    fi
 
-    for _file in "${zfsbootmenu_module_root}"/bin/*; do
-        add_file "${_file}" "/bin/${_file##*/}"
-    done
+    # Install online documentation if possible
+    install_zbm_docs
 
-    hooks=( zfsbootmenu-{parse-commandline,preinit}.sh )
-    for _file in "${hooks[@]}"; do
-        add_file "${zfsbootmenu_module_root}/hook/${_file}" "/lib/${_file}"
+    # Install pre-init scripts
+    for _file in "${zfsbootmenu_module_root}"/pre-init/*; do
+        add_file "${_file}" "/lib/${_file##*/}" && continue;
+
+        error "Failed to install ZFSBootMenu component ${_file}"
+        exit 1
     done
 
     # allow mount(8) to "autodetect" ZFS
     echo 'zfs' >>"${BUILDROOT}/etc/filesystems"
 
-    # shellcheck disable=SC2154
-    for _file in "${zfsbootmenu_early_setup[@]}"; do
-        [ -x "${_file}" ] || continue
-        add_file "${_file}" "/libexec/early-setup.d/${_file##*/}"
-    done
-
-    # shellcheck disable=SC2154
-    for _file in "${zfsbootmenu_setup[@]}"; do
-        [ -x "${_file}" ] || continue
-        add_file "${_file}" "/libexec/setup.d/${_file##*/}"
-    done
-
-    # shellcheck disable=SC2154
-    for _file in "${zfsbootmenu_teardown[@]}"; do
-        [ -x "${_file}" ] || continue
-        add_file "${_file}" "/libexec/teardown.d/${_file##*/}"
-    done
-
     compat_dirs=( "/etc/zfs/compatibility.d" "/usr/share/zfs/compatibility.d/" )
     for compat_dir in "${compat_dirs[@]}"; do
         [ -d "${compat_dir}" ] && add_full_dir "${compat_dir}"

etc/zfsbootmenu/release.conf.d/common.conf

@@ -68,14 +68,8 @@ if [ -r "${BASE}/bootfs" ]; then
 fi
 
 # Run setup hooks, if they exist
-if [ -d /libexec/setup.d ]; then
-  tput clear
-  for _hook in /libexec/setup.d/*; do
-    zinfo "Processing hook: ${_hook}"
-    [ -x "${_hook}" ] && "${_hook}"
-  done
-  unset _hook
-fi
+tput clear
+/libexec/zfsbootmenu-run-hooks "setup.d"
 
 # Clear screen before a possible password prompt
 tput clear

initcpio/install/zfsbootmenu

@@ -6,6 +6,10 @@
 ## which causes the normal dracut initqueue to be thrown out; the ZFSBootMenu
 ## initqueue hooks that force the dracut event loop to run at least once are
 ## purged, so dracut terminates the event loop before console initialization.
+##
+
+# There is nothing to do if we're not in dracut
+[ "${ZBM_BUILDSTYLE,,}" = "dracut" ] || exit 0
 
 # There is nothing to do if the console initializer is not executable
 [ -x /lib/udev/console_init ] || exit 0