From 7cce2b53941606076a1bb39379c330840bee2ea2 Mon Sep 17 00:00:00 2001
From: gyurix <gyorgy.berenyi@format.hu>
Date: Wed, 8 Mar 2023 20:53:13 +0000
Subject: [PATCH] Complete firewall in case of network defined in rules

---
 firewall/firewall-add | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/firewall/firewall-add b/firewall/firewall-add
index a08633c..f960aa1 100755
--- a/firewall/firewall-add
+++ b/firewall/firewall-add
@@ -400,6 +400,11 @@ else
 	IPTABLES_OPTIONS=""
 	GREP_OPTIONS=""
 	if [ "$SOURCE_IP" != "" ]; then
+		
+		if [ "$(echo $SOURCE_IP | cut -d . -f4)" == "0" ]; then
+			SOURCE_IP="$(echo $SOURCE_IP | cut -d . -f1-3).0/24";
+		fi
+		
 		IPTABLES_OPTIONS=$IPTABLES_OPTIONS" -s $SOURCE_IP";
 		GREP_OPTIONS=$GREP_OPTIONS"|grep -e $SOURCE_IP";
 
@@ -410,6 +415,11 @@ else
 	fi
 	
 	if [ "$TARGET_IP" != "" ]; then
+		
+		if [ "$(echo $TARGET_IP | cut -d . -f4)" == "0" ]; then
+			TARGET_IP="$(echo $TARGET_IP | cut -d . -f1-3).0/24";
+		fi
+		
 		IPTABLES_OPTIONS=$IPTABLES_OPTIONS" -d $TARGET_IP";
 		GREP_OPTIONS=$GREP_OPTIONS"|grep -e $TARGET_IP";