summaryrefslogtreecommitdiffstats
path: root/contrib/dracut/90zfs/zfs-generator.sh.in
blob: b57c64c688b1fe1859a46f9ed866669400f4ff0c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#!/bin/sh
# shellcheck disable=SC2016,SC1004

grep -wq debug /proc/cmdline && debug=1
[ -n "$debug" ] && echo "zfs-generator: starting" >> /dev/kmsg

GENERATOR_DIR="$1"
[ -n "$GENERATOR_DIR" ] || {
    echo "zfs-generator: no generator directory specified, exiting" >> /dev/kmsg
    exit 1
}

[ -f /lib/dracut-lib.sh ] && dracutlib=/lib/dracut-lib.sh
[ -f /usr/lib/dracut/modules.d/99base/dracut-lib.sh ] && dracutlib=/usr/lib/dracut/modules.d/99base/dracut-lib.sh
command -v getarg >/dev/null 2>&1 || {
    [ -n "$debug" ] && echo "zfs-generator: loading Dracut library from $dracutlib" >> /dev/kmsg
    . "$dracutlib"
}

. /lib/dracut-zfs-lib.sh

[ -z "$root" ]       && root=$(getarg root=)
[ -z "$rootfstype" ] && rootfstype=$(getarg rootfstype=)
[ -z "$rootflags" ]  && rootflags=$(getarg rootflags=)

# If root is not ZFS= or zfs: or rootfstype is not zfs
# then we are not supposed to handle it.
[ "${root##zfs:}" = "${root}" ] &&
	[ "${root##ZFS=}" = "${root}" ] &&
	[ "$rootfstype" != "zfs" ] &&
	exit 0

case ",${rootflags}," in
	*,zfsutil,*) ;;
	,,)	rootflags=zfsutil ;;
	*)	rootflags="zfsutil,${rootflags}" ;;
esac

if [ "${root}" != "zfs:AUTO" ]; then
  root="${root##zfs:}"
  root="${root##ZFS=}"
fi

[ -n "$debug" ] && echo "zfs-generator: writing extension for sysroot.mount to $GENERATOR_DIR/sysroot.mount.d/zfs-enhancement.conf" >> /dev/kmsg


mkdir -p "$GENERATOR_DIR"/sysroot.mount.d "$GENERATOR_DIR"/initrd-root-fs.target.requires "$GENERATOR_DIR"/dracut-pre-mount.service.d
{
    echo "[Unit]"
    echo "Before=initrd-root-fs.target"
    echo "After=zfs-import.target"
    echo
    echo "[Mount]"
    if [ "${root}" = "zfs:AUTO" ]; then
      echo "PassEnvironment=BOOTFS"
      echo 'What=${BOOTFS}'
    else
      echo "What=${root}"
    fi
    echo "Type=zfs"
    echo "Options=${rootflags}"
} > "$GENERATOR_DIR"/sysroot.mount.d/zfs-enhancement.conf
ln -fs ../sysroot.mount "$GENERATOR_DIR"/initrd-root-fs.target.requires/sysroot.mount


if [ "${root}" = "zfs:AUTO" ]; then
  {
      echo "[Unit]"
      echo "Before=initrd-root-fs.target"
      echo "After=sysroot.mount"
      echo "DefaultDependencies=no"
      echo
      echo "[Service]"
      echo "Type=oneshot"
      echo "PassEnvironment=BOOTFS"
      echo "ExecStart=/bin/sh -c '" '                                        \
        . /lib/dracut-zfs-lib.sh;                                            \
        _zfs_nonroot_necessities_cb() {                                      \
            zfs mount | grep -m1 -q "^$1 " && return 0;                      \
            echo "Mounting $1 on /sysroot$2";                                \
            mount -o zfsutil -t zfs "$1" "/sysroot$2";                       \
        };                                                                   \
        for_relevant_root_children "${BOOTFS}" _zfs_nonroot_necessities_cb;' \
      "'"
  } > "$GENERATOR_DIR"/zfs-nonroot-necessities.service
  ln -fs ../zfs-nonroot-necessities.service "$GENERATOR_DIR"/initrd-root-fs.target.requires/zfs-nonroot-necessities.service
else
  # We can solve this statically at generation time, so do!
  _zfs_generator_cb() {
      dset="${1}"
      mpnt="${2}"
      unit="sysroot$(echo "$mpnt" | sed 's;/;-;g').mount"

      {
          echo "[Unit]"
          echo "Before=initrd-root-fs.target"
          echo "After=sysroot.mount"
          echo
          echo "[Mount]"
          echo "Where=/sysroot${mpnt}"
          echo "What=${dset}"
          echo "Type=zfs"
          echo "Options=zfsutil"
      } > "$GENERATOR_DIR/${unit}"
      ln -fs ../"${unit}" "$GENERATOR_DIR"/initrd-root-fs.target.requires/"${unit}"
  }

  for_relevant_root_children "${root}" _zfs_generator_cb
fi


{
    echo "[Unit]"
    echo "After=zfs-import.target"
} > "$GENERATOR_DIR"/dracut-pre-mount.service.d/zfs-enhancement.conf

[ -n "$debug" ] && echo "zfs-generator: finished" >> /dev/kmsg

exit 0