From 1b5288996fb73871ccaf0d4663caeefb7aaa7fbc Mon Sep 17 00:00:00 2001
From: randomshell <randshell@protonmail.com>
Date: Thu, 2 Jul 2020 10:43:15 +0000
Subject: [PATCH] Add validation on $SERVER_PORT

---
 wireguard-install.sh | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/wireguard-install.sh b/wireguard-install.sh
index d10e00f..413febf 100644
--- a/wireguard-install.sh
+++ b/wireguard-install.sh
@@ -76,8 +76,10 @@ function installWireGuard() {
 	read -rp "Server's WireGuard IPv6: " -e -i "$SERVER_WG_IPV6" SERVER_WG_IPV6
 
 	# Generate random number within private ports range
-	SERVER_PORT=$(shuf -i49152-65535 -n1)
-	read -rp "Server's WireGuard port: " -e -i "$SERVER_PORT" SERVER_PORT
+	RANDOM_PORT=$(shuf -i49152-65535 -n1)
+	until [[ $SERVER_PORT =~ ^[0-9]+$ ]] && [ "$SERVER_PORT" -ge 1 ] && [ "$SERVER_PORT" -le 65535 ]; do
+		read -rp "Server's WireGuard port [1-65535]: " -e -i "$RANDOM_PORT" SERVER_PORT
+	done
 
 	# Install WireGuard tools and module
 	if [[ $OS == 'ubuntu' ]]; then