From a91e864cfe730f29ed1d965fb25dbeaf7c965543 Mon Sep 17 00:00:00 2001
From: Chai Feng <chaifeng@chaifeng.com>
Date: Wed, 3 Oct 2018 09:10:57 +0800
Subject: [PATCH] valid port syntax in `service allow`

---
 ufw-docker | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ufw-docker b/ufw-docker
index 6852280..d907be8 100755
--- a/ufw-docker
+++ b/ufw-docker
@@ -124,6 +124,12 @@ function ufw-docker--service-allow() {
     service_name="$1"
     service_port="$2"
 
+    if [[ -n "$service_port" ]] &&
+           ! grep -E '^[0-9]+(/(tcp|udp))?$' <<< "$service_port" &>/dev/null; then
+        die "Invalid port syntax: $service_port"
+        return 1
+    fi
+
 }
 
 function ufw-docker--install() {