From ccc6ee331662a793d5e26d34720693e9ec9d9ae3 Mon Sep 17 00:00:00 2001
From: Markus Rudy <mr@edgeless.systems>
Date: Thu, 23 May 2024 13:18:49 +0200
Subject: [PATCH] genpolicy-msft: add support for volumeDevices

---
 packages/by-name/genpolicy-msft/package.nix | 46 ++++++++++++++-------
 1 file changed, 31 insertions(+), 15 deletions(-)

diff --git a/packages/by-name/genpolicy-msft/package.nix b/packages/by-name/genpolicy-msft/package.nix
index 5c5f784775..169355597b 100644
--- a/packages/by-name/genpolicy-msft/package.nix
+++ b/packages/by-name/genpolicy-msft/package.nix
@@ -3,8 +3,9 @@
 
 { lib
 , fetchFromGitHub
-, fetchurl
+, fetchpatch
 , applyPatches
+, pkgs
 , rustPlatform
 , openssl
 , pkg-config
@@ -25,6 +26,17 @@ rustPlatform.buildRustPackage rec {
     hash = "sha256-W36RJFf0MVRIBV4ahpv6pqdAwgRYrlqmu4Y/8qiILS8=";
   };
 
+  patches = [
+    # TODO(burgerdev): drop this patch after upstream applied it
+    (fetchpatch {
+      name = "genpolicy_device_support.patch";
+      url = "https://github.com/kata-containers/kata-containers/commit/f61b43777834f097fcca26864ee634125d9266ef.patch";
+      sha256 = "sha256-wBOyrFY4ZdWBjF5bIrHm7CFy6lVclcvwhF85wXpFZoc=";
+    })
+  ];
+
+  patchFlags = [ "-p4" ];
+
   sourceRoot = "${src.name}/src/tools/genpolicy";
 
   cargoHash = "sha256-YxIwsjs4K0TNVlwwA+PrOrCf16h7ZW+zU/jXeFfIMZo=";
@@ -49,14 +61,16 @@ rustPlatform.buildRustPackage rec {
   '';
 
   passthru = rec {
-    settings = fetchurl {
+    settings = pkgs.stdenv.mkDerivation {
       name = "${pname}-${version}-settings";
-      # TODO(burgerdev): see whether future releases contain this file as an asset again (not true for 3.2.0.azl1).
-      url = "https://raw.githubusercontent.com/microsoft/kata-containers/${version}/src/tools/genpolicy/genpolicy-settings.json";
-      hash = "sha256-jrhzDqesm16yCV3aex48c2OcEimCUrxwhoaJUtAMPvo=";
-      downloadToTemp = true;
-      recursiveHash = true;
-      postFetch = "install -D $downloadedFile $out/genpolicy-settings.json";
+      inherit src sourceRoot patches patchFlags;
+
+      phases = [ "unpackPhase" "patchPhase" "installPhase" ];
+      installPhase = ''
+        runHook preInstall
+        install -D genpolicy-settings.json $out/genpolicy-settings.json
+        runHook postInstall
+      '';
     };
 
     # Settings that allow exec into CVM pods - not safe for production use!
@@ -65,14 +79,16 @@ rustPlatform.buildRustPackage rec {
       patches = [ ./genpolicy_msft_settings_dev.patch ];
     };
 
-    rules = fetchurl {
+    rules = pkgs.stdenv.mkDerivation {
       name = "${pname}-${version}-rules";
-      # TODO(burgerdev): see whether future releases contain this file as an asset again (not true for 3.2.0.azl1).
-      url = "https://raw.githubusercontent.com/microsoft/kata-containers/${version}/src/tools/genpolicy/rules.rego";
-      hash = "sha256-fhE5hDND5QeZtEw3u+qgSVsFO+00cc41k/r/Y+km6TU=";
-      downloadToTemp = true;
-      recursiveHash = true;
-      postFetch = "install -D $downloadedFile $out/genpolicy-rules.rego";
+      inherit src sourceRoot patches patchFlags;
+
+      phases = [ "unpackPhase" "patchPhase" "installPhase" ];
+      installPhase = ''
+        runHook preInstall
+        install -D rules.rego $out/genpolicy-rules.rego
+        runHook postInstall
+      '';
     };
 
     rules-coordinator = applyPatches {