From 498301b1117eae3f4803a8c1a173a3235b4ea2d3 Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Wed, 25 Jan 2023 12:00:19 -0500
Subject: [PATCH] Salt for Ubuntu

---
 setup/so-functions | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/setup/so-functions b/setup/so-functions
index cff943376..26c66736f 100755
--- a/setup/so-functions
+++ b/setup/so-functions
@@ -2347,7 +2347,12 @@ ubuntu_check() {
 			whiptail_ubuntu_notsupported
 			exit 1
 		else
-			whiptail_ubuntu_warning
+		    if [[ $UBUNTUINSTALL == "needtoupgrade" ]]; then
+				whiptail_ubuntu_warning
+			else
+				whiptail_ubuntu_notsupported
+				exit 1
+			fi
 		fi
 	fi
 }