Fixing SOURE_PORT variables cycle.

This commit is contained in:
2021-06-18 12:27:30 +00:00
parent 47bcb30048
commit 1ace0d51e7

View File

@@ -1,5 +1,6 @@
#!/bin/sh #!/bin/sh
NSENTER=$NSENTER NSENTER=$NSENTER
NAME=$NAME-$COMMENT
CHAIN=$CHAIN CHAIN=$CHAIN
COMMENT=$COMMENT COMMENT=$COMMENT
PROTOCOL=$TYPE PROTOCOL=$TYPE
@@ -18,16 +19,11 @@ COUNT_SOURCE_PORT=$(set |grep SOURCE_PORT |wc -l)
COUNT_TARGET_IP=$(set |grep TARGET_IP |wc -l) COUNT_TARGET_IP=$(set |grep TARGET_IP |wc -l)
COUNT_TARGET_PORT=$(set |grep TARGET_PORT |wc -l) COUNT_TARGET_PORT=$(set |grep TARGET_PORT |wc -l)
for source_ip_index in $(seq 1 $COUNT_SOURCE_IP) ; do for source_ip_index in $(seq 1 $COUNT_SOURCE_IP) ; do
if set |grep SOURCE_IP_ ; then if set |grep SOURCE_IP_ ; then
SOURCE_IP=$(eval "echo \${"SOURCE_IP_$source_ip_index"}") SOURCE_IP=$(eval "echo \${"SOURCE_IP_$source_ip_index"}")
fi fi
for source_port_index in $(seq 1 $COUNT_SOURCE_PORT) ; do
if set |grep SOURCE_PORT_ ; then
SOURCE_PORT=$(eval "echo \${"SOURCE_PORT_$source_port_index"}")
fi
for target_ip_index in $(seq 1 $COUNT_TARGET_IP) ; do for target_ip_index in $(seq 1 $COUNT_TARGET_IP) ; do
if set |grep TARGET_IP_ ; then if set |grep TARGET_IP_ ; then
TARGET_IP=$(eval "echo \${"TARGET_IP_$target_ip_index"}") TARGET_IP=$(eval "echo \${"TARGET_IP_$target_ip_index"}")
@@ -39,39 +35,45 @@ for source_ip_index in $(seq 1 $COUNT_SOURCE_IP) ; do
fi fi
############################# #############################
# NSENTER Specific settings # # NSENTER Specific settings #
if [[ "$NSENTER" == "true" ]] ; then if [[ "$NSENTER" == "true" ]] ; then
iptables="nsenter -t $(docker inspect --format {{.State.Pid}} $NAME) -n -- /sbin/iptables-legacy -t nat"; iptables="nsenter -t $(docker inspect --format {{.State.Pid}} $NAME) -n -- /sbin/iptables-legacy -t nat";
if $iptables --list POSTROUTING |grep ESTABLISHED |grep RELATED|grep ACCEPT ; then if $iptables --list POSTROUTING |grep ESTABLISHED |grep RELATED|grep ACCEPT ; then
echo "nothing to do";
else $iptables -I POSTROUTING -m state --state established,related -j ACCEPT; else $iptables -I POSTROUTING -m state --state established,related -j ACCEPT;
fi fi
# DELETE UNECESSARY LINES FROM PREVIOUS RULES for source_port_index in $(seq 1 $COUNT_SOURCE_PORT) ; do
LINES=$($iptables --line-number -n --list POSTROUTING | grep $SOURCE_IP |grep $TARGET_IP |grep $PROTOCOL |grep $SOURCE_PORT |grep $TARGET_PORT | awk '{print $1}'| tac) if set |grep SOURCE_PORT_ ; then
SOURCE_PORT=$(eval "echo \${"SOURCE_PORT_$source_port_index"}")
fi
if [ -n "$LINES" ] ; then # DELETE UNECESSARY LINES FROM PREVIOUS RULES
LINES=$($iptables --line-number -n --list POSTROUTING | grep $SOURCE_IP |grep $TARGET_IP |grep $PROTOCOL |grep $SOURCE_PORT |grep $TARGET_PORT | awk '{print $1}'| tac)
if [ -n "$LINES" ] ; then
for i in $LINES; do for i in $LINES; do
$iptables -D PREROUTING $i $iptables -D PREROUTING $i
sleep 0.1 sleep 0.1
done done
fi fi
$iptables -I PREROUTING -d $SOURCE_IP -p $PROTOCOL --dport $SOURCE_PORT -m comment --comment "$COMMENT" -j DNAT --to $TARGET_IP:$TARGET_PORT $iptables -I PREROUTING -d $SOURCE_IP -p $PROTOCOL --dport $SOURCE_PORT -m comment --comment "$COMMENT" -j DNAT --to $TARGET_IP:$TARGET_PORT
done
else else
iptables="/sbin/iptables-legacy" iptables="/sbin/iptables-legacy"
if $iptables --list $CHAIN |grep ESTABLISHED |grep RELATED|grep ACCEPT ; then if $iptables --list $CHAIN |grep ESTABLISHED |grep RELATED|grep ACCEPT ; then
echo "nothing to do";
else $iptables -I $CHAIN -m state --state established,related -j ACCEPT; else $iptables -I $CHAIN -m state --state established,related -j ACCEPT;
fi fi
# #
# DELETE UNECESSARY LINES FROM PREVIOUS RULES # DELETE UNECESSARY LINES FROM PREVIOUS RULES
LINES=$($iptables --line-number -n --list $CHAIN | grep $SOURCE_IP |grep $TARGET_IP |grep $PROTOCOL |grep $TARGET_PORT | awk '{print $1}'| tac) LINES=$($iptables --line-number -n --list $CHAIN | grep $SOURCE_IP |grep $TARGET_IP |grep $PROTOCOL |grep $TARGET_PORT | awk '{print $1}'| tac)
if [ -n "$LINES" ] ; then if [ -n "$LINES" ] ; then
for i in $LINES; do for i in $LINES; do
@@ -80,12 +82,14 @@ LINES=$($iptables --line-number -n --list $CHAIN | grep $SOURCE_IP |grep $TARGET
done done
fi fi
$iptables -I $CHAIN -s $SOURCE_IP -d $TARGET_IP -p $PROTOCOL --dport $TARGET_PORT -m comment --comment "$COMMENT" -j ACCEPT $iptables -I $CHAIN -s $SOURCE_IP -d $TARGET_IP -p $PROTOCOL --dport $TARGET_PORT -m comment --comment "$COMMENT" -j ACCEPT
############################# #############################
fi fi
done
done done # target_port
done done # target_ip
done done # source_ip