From b555c18e83a6a782908db54ff97bb5c43fd75816 Mon Sep 17 00:00:00 2001
From: nikstur <nikstur@outlook.com>
Date: Thu, 24 Nov 2022 11:45:09 +0100
Subject: [PATCH] lanzatool: add wrapper

---
 flake.nix | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/flake.nix b/flake.nix
index d277974..60f828a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -73,7 +73,19 @@
 
       lanzaboote = buildRustEfiApp ./rust/lanzaboote;
 
-      lanzatool = buildRustLinuxApp ./rust/lanzatool;
+      lanzatoolBin = naersk-nightly.buildPackage {
+        src = ./rust/lanzatool;
+        buildInputs = [ pkgs.binutils ];
+      };
+
+      lanzatool = pkgs.writeShellScriptBin "lanzatool" ''
+        set -euo pipefail
+
+        export LANZABOOTE_STUB=${lanzaboote}/bin/lanzaboote.efi
+        export LANZABOOTE_INITRD_STUB=${initrd-stub}/bin/initrd-stub.efi
+
+        ${lanzatoolBin}/bin/lanzatool "$@"
+      '';
 
       # A script that takes an initrd and turns it into a PE image.
       wrapInitrd = pkgs.writeShellScriptBin "wrap-initrd" ''