From f2b2c526db5a9c18abfd605c52c9437f701e1858 Mon Sep 17 00:00:00 2001
From: Pratik Kumar Tripathy <22846027+pratiktri@users.noreply.github.com>
Date: Tue, 7 Jul 2020 10:04:12 +0530
Subject: [PATCH] Added support for Ubuntu 20.04

---
 init-linux-harden.sh | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/init-linux-harden.sh b/init-linux-harden.sh
index caefb97..69189c1 100644
--- a/init-linux-harden.sh
+++ b/init-linux-harden.sh
@@ -78,8 +78,8 @@ case "$OS" in
         elif [[ "$VER" -eq 10 ]]; then
             DEB_VER_STR="buster" 
         else
-            printf "This script only supports Debian 8 and Debian 9\\n"
-            printf "\\tUbuntu 14.04, Ubuntu 16.04, Ubuntu 18.04, Ubuntu 18.10\\n"
+            printf "This script only supports Debian 8 and Debian 9, and Debian 10\\n"
+            printf "\\tUbuntu 14.04, Ubuntu 16.04, Ubuntu 18.04, Ubuntu 18.10, and Ubuntu 20.04\\n"
             printf "Your OS is NOT supported.\\n"
             exit 1
         fi
@@ -93,9 +93,11 @@ case "$OS" in
             UBT_VER_STR="bionic"
         elif [[ "$VER" = "18.10" ]]; then
             UBT_VER_STR="cosmic"
+        elif [[ "$VER" = "20.04" ]]; then
+            UBT_VER_STR="focal"
         else
-            printf "This script only supports Debian 8 and Debian 9\\n"
-            printf "\\tUbuntu 14.04, Ubuntu 16.04, Ubuntu 18.04, Ubuntu 18.10\\n"
+            printf "This script only supports Debian 8, Debian 9, and Debian 10\\n"
+            printf "\\tUbuntu 14.04, Ubuntu 16.04, Ubuntu 18.04, Ubuntu 18.10, and Ubuntu 20.04\\n"
             printf "Your OS is NOT supported.\\n"
             exit 1
         fi