1 files changed, 67 insertions, 2 deletions

diff --git a/sys/boot/tpm.nix b/sys/boot/tpm.nix
index 4932b7c..1e446ab 100644
--- a/sys/boot/tpm.nix
+++ b/sys/boot/tpm.nix
@@ -1,10 +1,57 @@
 { config, lib, pkgs, ... }:
 with lib; let
   cfg = config.local.boot.tpm;
+
+  pcrList = concatStringsSep "," (map toString cfg.initrd.pcrs);
+
+  # Crear signing-key con:
+  # $ openssl genrsa -out ~/vtmp/signing-key.priv 2048
+  # $ openssl rsa -in ~/vtmp/signing-key.priv -out ~/vtmp/signing-key.pub -pubout
+  # Y copiar signing-key.pub a /boot/tpm-boot. Guardar signing-key.priv en lugar seguro.
+  #
+  # Crear llave con:
+  # $ tpm2_loadexternal -G rsa -C owner -u signing-key.pub -c signing-key.ctx -n signing-key.name
+  # $ tpm2_startauthsession -S session.ctx
+  # $ tpm2_policyauthorize -S session.ctx -L require-signed.policy -n signing-key.name
+  # $ tpm2_flushcontext session.ctx
+  # $ tpm2_createprimary -C owner -g sha256 -G ecc -c prim.ctx
+  # $ head -c128 /dev/urandom | tpm2_create -C prim.ctx -u key.pub -r key.priv -c key.ctx -L require-signed.policy -i-
+  # Y mover key.priv/key.pub a /boot/tpm-boot
+  #
+  # Usage: tpm2-grant-next-boot < signing-key.priv
+  # Genera auth.policy y auth.sig
+  tpm2-grant-next-boot = pkgs.writeShellApplication {
+    name = "tpm2-grant-next-boot";
+    runtimeInputs = [ pkgs.openssl pkgs.tpm2-tools ];
+
+    text = ''
+      ctx_dir="$(mktemp -d)"
+      trap 'rm -rf -- "$ctx_dir"' EXIT
+
+      tpm2_createprimary -Q -C owner -g sha256 -G ecc -c "$ctx_dir/prim.ctx"
+
+      tpm2_startauthsession -Q -S "$ctx_dir/session.ctx"
+      tpm_resets=$(tpm2_readclock | grep reset_count | sed 's/.*: //')
+      tpm2_policycountertimer -Q -S "$ctx_dir/session.ctx" resets="$((tpm_resets+1))"
+      tpm2_policypcr -Q -S "$ctx_dir/session.ctx" -L auth.policy -l sha256:${pcrList}
+      tpm2_flushcontext -Q "$ctx_dir/session.ctx"
+
+      openssl dgst -sha256 -sign /dev/stdin -out auth.sig auth.policy
+    '';
+  };
 in
 {
   options.local.boot.tpm = {
     enable = mkEnableOption "Trusted Platform Module 2.0";
+
+    initrd = {
+      enable = mkEnableOption "TPM2 in initrd";
+
+      pcrs = mkOption {
+        type = with types; listOf (ints.between 0 23);
+        default = [ 0 2 7 8 11 ];
+      };
+    };
   };
 
   config = mkIf cfg.enable {
@@ -13,6 +60,26 @@ in
         assertion = config.local.boot.efi.enable;
         message = "TPM2 requires EFI";
       }
+      {
+        assertion = cfg.initrd.enable -> cfg.enable;
+        message = "TPM2 in initrd requires TPM2";
+      }
+    ];
+
+    boot.initrd = mkIf cfg.initrd.enable {
+      extraUtilsCommands = ''
+        copy_bin_and_libs ${pkgs.tpm2-tools}/bin/.tpm2-wrapped
+        mv $out/bin/{.tpm2-wrapped,tpm2}
+        cp {${pkgs.tpm2-tss},$out}/lib/libtss2-tcti-device.so.0
+      '';
+
+      kernelModules = [
+        "tpm_tis"
+      ];
+    };
+
+    environment.systemPackages = optionals cfg.initrd.enable [
+      tpm2-grant-next-boot
     ];
 
     security.tpm2 = {
@@ -21,7 +88,5 @@ in
       pkcs11.enable = true;
       tctiEnvironment.enable = true;
     };
-
-    environment.systemPackages = [ pkgs.tpm2-tools ];
   };
 }