1#!/bin/sh
2
3echo "zfs-generator: starting" >> /dev/kmsg
4
5GENERATOR_DIR="$1"
6[ -n "$GENERATOR_DIR" ] || {
7    echo "zfs-generator: no generator directory specified, exiting" >> /dev/kmsg
8    exit 1
9}
10
11[ -f /lib/dracut-lib.sh ] && dracutlib=/lib/dracut-lib.sh
12[ -f /usr/lib/dracut/modules.d/99base/dracut-lib.sh ] && dracutlib=/usr/lib/dracut/modules.d/99base/dracut-lib.sh
13
14command -v getarg >/dev/null 2>&1 || {
15    echo "zfs-generator: loading Dracut library from $dracutlib" >> /dev/kmsg
16    . "$dracutlib"
17}
18
19[ -z "$root" ]       && root=$(getarg root=)
20[ -z "$rootfstype" ] && rootfstype=$(getarg rootfstype=)
21[ -z "$rootflags" ]  && rootflags=$(getarg rootflags=)
22
23# If root is not ZFS= or zfs: or rootfstype is not zfs
24# then we are not supposed to handle it.
25[ "${root##zfs:}" = "${root}" ] &&
26	[ "${root##ZFS=}" = "${root}" ] &&
27	[ "$rootfstype" != "zfs" ] &&
28	exit 0
29
30rootfstype=zfs
31case ",${rootflags}," in
32	*,zfsutil,*) ;;
33	,,)	rootflags=zfsutil ;;
34	*)	rootflags="zfsutil,${rootflags}" ;;
35esac
36
37echo "zfs-generator: writing extension for sysroot.mount to $GENERATOR_DIR"/sysroot.mount.d/zfs-enhancement.conf >> /dev/kmsg
38
39[ -d "$GENERATOR_DIR" ] || mkdir "$GENERATOR_DIR"
40[ -d "$GENERATOR_DIR"/sysroot.mount.d ] || mkdir "$GENERATOR_DIR"/sysroot.mount.d
41
42{
43    echo "[Unit]"
44    echo "Before=initrd-root-fs.target"
45    echo "After=zfs-import.target"
46    echo "[Mount]"
47    if [ "${root}" = "zfs:AUTO" ] ; then
48      echo "PassEnvironment=BOOTFS"
49      echo 'What=${BOOTFS}'
50    else
51      root="${root##zfs:}"
52      root="${root##ZFS=}"
53      echo "What=${root}"
54    fi
55    echo "Type=${rootfstype}"
56    echo "Options=${rootflags}"
57} > "$GENERATOR_DIR"/sysroot.mount.d/zfs-enhancement.conf
58
59[ -d "$GENERATOR_DIR"/initrd-root-fs.target.requires ] || mkdir -p "$GENERATOR_DIR"/initrd-root-fs.target.requires
60ln -s ../sysroot.mount "$GENERATOR_DIR"/initrd-root-fs.target.requires/sysroot.mount
61
62echo "zfs-generator: finished" >> /dev/kmsg
63