From 2e483a7c518c19f1e93498f99838d118d1efadff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=BD=D0=B0=D0=B1?= <nabijaczleweli@nabijaczleweli.xyz>
Date: Sun, 23 May 2021 22:09:35 +0200
Subject: [PATCH] dracut: 90zfs: zfs-load-key: wait for key to appear
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Ahelenia Ziemiańska <nabijaczleweli@nabijaczleweli.xyz>
Closes #12065
---
 contrib/dracut/90zfs/zfs-load-key.sh.in | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/contrib/dracut/90zfs/zfs-load-key.sh.in b/contrib/dracut/90zfs/zfs-load-key.sh.in
index f15118ad02ca..708584e40646 100755
--- a/contrib/dracut/90zfs/zfs-load-key.sh.in
+++ b/contrib/dracut/90zfs/zfs-load-key.sh.in
@@ -43,9 +43,24 @@ if [ "$(zpool list -H -o feature@encryption "${BOOTFS%%/*}")" = 'active' ]; then
 
         KEYLOCATION="$(zfs get -H -o value keylocation "${ENCRYPTIONROOT}")"
         if ! [ "${KEYLOCATION}" = "prompt" ]; then
-            if ! [ "${KEYLOCATION#http}" = "${KEYLOCATION}" ]; then
-                systemctl start network-online.target
-            fi
+            case "${KEYLOCATION%%://*}" in
+                http*)
+                    systemctl start network-online.target
+                    ;;
+                file)
+                    KEYFILE="${KEYLOCATION#file://}"
+                    [ -r "${KEYFILE}" ] || udevadm settle
+                    [ -r "${KEYFILE}" ] || {
+                        info "Waiting for key ${KEYFILE} for ${ENCRYPTIONROOT}..."
+                        for _ in 1 2 3 4 5 6 7 8 9 10; do
+                            sleep 1
+                            [ -r "${KEYFILE}" ] && break
+                        done
+                    }
+                    [ -r "${KEYFILE}" ] || warn "Key ${KEYFILE} for ${ENCRYPTIONROOT} hasn't appeared. Trying anyway."
+                    ;;
+            esac
+
             zfs load-key "${ENCRYPTIONROOT}"
         else
             # decrypt them