From 37cd69fc1a2c6e8f3bc7f3f94bae800c38a4f6b6 Mon Sep 17 00:00:00 2001
From: Hin-Tak Leung <htl10@users.sourceforge.net>
Date: Mon, 24 Aug 2020 12:03:46 +0100
Subject: [PATCH] Allow kernel images to be elsewhere - Ubuntu kernel is not at
 Raspbian location

---
 install.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/install.sh b/install.sh
index 3268e69..2665d29 100755
--- a/install.sh
+++ b/install.sh
@@ -66,6 +66,7 @@ function get_kernel_version() {
   _VER_RUN=""
   [ -z "$_VER_RUN" ] && {
     ZIMAGE=/boot/kernel.img
+    [ -f /boot/firmware/vmlinuz ] && ZIMAGE=/boot/firmware/vmlinuz
     IMG_OFFSET=$(LC_ALL=C grep -abo $'\x1f\x8b\x08\x00' $ZIMAGE | head -n 1 | cut -d ':' -f 1)
     _VER_RUN=$(dd if=$ZIMAGE obs=64K ibs=4 skip=$(( IMG_OFFSET / 4)) | zcat | grep -a -m1 "Linux version" | strings | awk '{ print $3; }')
   }