#!/bin/bash -e
# vim: set tabstop=8 shiftwidth=4 softtabstop=4 expandtab smarttab colorcolumn=80:
#
# Copyright (c) 2016 Red Hat, Inc.
# Author: Harald Hoyer <harald@redhat.com>
# Author: Nathaniel McCallum <npmccallum@redhat.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

UUID=cb6e8904-81ff-40da-a84a-07ab9ab5715e

shopt -s nullglob

path=/run/systemd/ask-password
while getopts ":lp:" o; do
    case "$o" in
    l) loop=true;;
    p) path=$OPTARG;;
    esac
done

while true; do
    todo=0

    for question in $path/ask.*; do
        metadata=false
        unlocked=false
        d=
        s=

        while read line; do
            case "$line" in
                Id=cryptsetup:*) d="${line##Id=cryptsetup:}";;
                Socket=*) s="${line##Socket=}";;
            esac
        done < "$question"

        [ -z "$d" -o -z "$s" ] && continue

        if cryptsetup isLuks --type luks1 "$d"; then
            # If the device is not initialized, sliently skip it.
            luksmeta test -d "$d" || continue

            while read -r slot state uuid; do
                [ "$state" != "active" ] && continue
                [ "$uuid" != "$UUID" ] && continue
                metadata=true

                if pt="`luksmeta load -d $d -s $slot -u $UUID | clevis decrypt`"; then
                    echo -n "+$pt" | nc -U -u --send-only "$s"
                    unlocked=true
                    break
                fi
            done < <(luksmeta show -d "$d")
        elif cryptsetup isLuks --type luks2 "$d"; then
            ids=`cryptsetup luksDump "$d" | sed -rn 's|^\s+([0-9]+): clevis|\1|p'`
            for id in $ids; do
                tok=`cryptsetup token export --token-id "$id" "$d"`
                jwe=`jose fmt -j- -Og jwe -o- <<<"$tok" | jose jwe fmt -i- -c`
                metadata=true

                if pt=`echo -n "$jwe" | clevis decrypt`; then
                    echo -n "+$pt" | nc -U -u --send-only "$s"
                    unlocked=true
                    break
                fi
            done
        fi

        [ $metadata == true ] || continue
        [ $unlocked == true ] && continue
        todo=$((todo + 1))
    done

    if [ $todo -eq 0 ] || [ "$loop" != "true" ]; then
        break;
    fi

    sleep 0.5
done