From 51cd55bca0e7c9fb2fc91a1904e7925853e283b4 Mon Sep 17 00:00:00 2001
From: TommyTran732 <57488583+tommytran732@users.noreply.github.com>
Date: Thu, 2 Jan 2020 17:35:34 -0500
Subject: [PATCH] Update cloudflare.sh

---
 cloudflare.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cloudflare.sh b/cloudflare.sh
index 8debf04..20a2431 100644
--- a/cloudflare.sh
+++ b/cloudflare.sh
@@ -4,6 +4,10 @@ output(){
     echo -e '\e[36m'$1'\e[0m';
 }
 
+if [ -r /etc/os-release ]; then
+    lsb_dist="$(. /etc/os-release && echo "$ID")"
+fi
+
 if [ "$lsb_dist" =  "ubuntu" ] || [ "$lsb_dist" =  "debian" ]; then
      apt -y install ufw
      # Opening Port 22 just in case so that we do not lose the internet connection when the rules are applied.