From da065572e4e6dd46438880e3e5e81ba27b97d0af Mon Sep 17 00:00:00 2001
From: NullBite <me@nullbite.com>
Date: Sat, 2 Nov 2024 22:25:15 +0100
Subject: [PATCH] Update NVIDIA driver because security

---
 hosts/slab/nvidia-optimus.nix | 9 ++++++++-
 system/hardware/nvidia.nix    | 9 ++++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/hosts/slab/nvidia-optimus.nix b/hosts/slab/nvidia-optimus.nix
index a42f712..ad69e9e 100644
--- a/hosts/slab/nvidia-optimus.nix
+++ b/hosts/slab/nvidia-optimus.nix
@@ -27,7 +27,14 @@
         nvidiaSettings = false;
 
         # Optionally, you may need to select the appropriate driver version for your specific GPU.
-        package = config.boot.kernelPackages.nvidiaPackages.stable;
+        package = let
+
+          # security vulnerability fix
+          stable = config.boot.kernelPackages.nvidiaPackages.stable;
+          beta = config.boot.kernelPackages.nvidiaPackages.beta;
+
+          version = if (lib.versionOlder stable.version "565.57.01") then beta else stable;
+        in version;
 
         prime = {
           offload = {
diff --git a/system/hardware/nvidia.nix b/system/hardware/nvidia.nix
index bb3a333..5ebf16e 100644
--- a/system/hardware/nvidia.nix
+++ b/system/hardware/nvidia.nix
@@ -68,7 +68,14 @@ in
       nvidiaSettings = lib.mkDefault true;
 
       # Optionally, you may need to select the appropriate driver version for your specific GPU.
-      package = lib.mkDefault config.boot.kernelPackages.nvidiaPackages.stable;
+      package = let
+
+        # security vulnerability fix
+        stable = config.boot.kernelPackages.nvidiaPackages.stable;
+        beta = config.boot.kernelPackages.nvidiaPackages.beta;
+
+        version = if (lib.versionOlder stable.version "565.57.01") then beta else stable;
+      in lib.mkDefault version;
     };
   };
 }