From b251f9e87074975aacff372908d525c517a901c6 Mon Sep 17 00:00:00 2001
From: TommyTran732 <57488583+tommytran732@users.noreply.github.com>
Date: Wed, 7 Apr 2021 02:39:31 -0400
Subject: [PATCH] Update Ubuntu-Desktop-20.04-LTS.sh

---
 Ubuntu-Desktop-20.04-LTS.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Ubuntu-Desktop-20.04-LTS.sh b/Ubuntu-Desktop-20.04-LTS.sh
index 30b59f0..c29b93c 100644
--- a/Ubuntu-Desktop-20.04-LTS.sh
+++ b/Ubuntu-Desktop-20.04-LTS.sh
@@ -184,6 +184,15 @@ sudo sed -i 's^DRIVER=="nvidia", RUN+="/usr/lib/gdm3/gdm-disable-wayland"^#DRIVE
 #Signing ashmem kernel module
 sudo kmodsign sha512 /var/lib/shim-signed/mok/MOK.priv /var/lib/shim-signed/mok/MOK.der /lib/modules/`uname -r`/kernel/drivers/staging/android/ashmem_linux.ko
 
+#Kind of an ugly hack, but since I use Nvidia and it needs to sign the Nvidia driver everytime theres a kernel update anyways, so I am doing this for now
+sudo bash -c 'cat > /etc/dkms/sign_helper.sh' <<-'EOF'
+#!/bin/sh
+/lib/modules/"$1"/build/scripts/sign-file sha512 /root/mok.priv /root/mok.der "$2"
+echo "kmodsign sha512 /var/lib/shim-signed/mok/MOK.priv /var/lib/shim-signed/mok/MOK.der /lib/modules/`uname -r`/kernel/drivers/staging/android/ashmem_linux.ko"
+EOF
+
+sudo chmod 755 /etc/dkms/sign_helper.sh
+
 #Randomize MAC address
 sudo bash -c 'cat > /etc/NetworkManager/conf.d/00-macrandomize.conf' <<-'EOF'
 [device]