X-Git-Url: https://git.harvie.cz/?p=svn%2FPrometheus-QoS%2F.git;a=blobdiff_plain;f=optional-tools%2Fmake-iptables-restore;h=950a2dd8db3c8c9308fbf8288c1a920106c5c581;hp=a4499a37d84fb19cc45c51f6a3c637f62b44237f;hb=HEAD;hpb=dec30356f770ebe6a6f1648041c4285bf3e5bff3

diff --git a/optional-tools/make-iptables-restore b/optional-tools/make-iptables-restore
index a4499a3..950a2dd 100755
--- a/optional-tools/make-iptables-restore
+++ b/optional-tools/make-iptables-restore
@@ -1,143 +1,197 @@
 #!/bin/bash
+# $Id: make-iptables-restore 172 2012-05-13 16:10:52Z aquarius $
 iptables="/sbin/iptables"
 iptablesrestore="/sbin/iptables-restore"
 ifconfig="/sbin/ifconfig"
+grep="/bin/grep"
+cut="/usr/bin/cut"
+ipcalc="/usr/bin/ipcalc"
 
 #pimp files must be generated by optional-tools/make-pimp utility
-pimp_2way_nat="/mnt/mtdblock0/pimp-2way-nat.tmp"
-pimp_snat="/mnt/mtdblock0/pimp-snat.tmp"
+pimp_2way_nat="/dev/shm/pimp-2way-nat.tmp"
+pimp_snat="/dev/shm/pimp-snat.tmp"
 etchosts="/mnt/mtdblock0/hosts"
+restoretmp="/dev/shm/iptables-restore.tmp"
 restoredata="/mnt/mtdblock0/iptables-restore.in"
 wan1="vlan770"
 wan2="vlan771"
-#just comment-out all lines mentioning wan3 on two-interface gateways
 wan3="vlan772"
-
-echo "*nat" > $restoredata
-echo ":PREROUTING ACCEPT [0:0]" >> $restoredata
-echo ":POSTROUTING ACCEPT [0:0]" >> $restoredata
-echo ":OUTPUT ACCEPT [0:0]" >> $restoredata
+wan4="vlan774"
+czffirstbitmask="19"
+czfsecondbitmask="22"
+czfthirdbitmask="25"
+czffourthbitmask="28"
+pubfirstbitmask="26"
+pubsecondbitmask="29"
+chaintrack="_"
+
+echo "*nat" > $restoretmp
+echo ":PREROUTING ACCEPT [0:0]" >> $restoretmp
+echo ":POSTROUTING ACCEPT [0:0]" >> $restoretmp
+echo ":OUTPUT ACCEPT [0:0]" >> $restoretmp
 
 # ===============================================================
 #  Symetrical SNAT-DNAT using indexed iptables
 # ===============================================================
-
 echo -n "Generating new iptables-restore data - two way SNAT/DNAT "
 
-for czfip in `grep -v ^# $pimp_2way_nat|cut -f 1 -d " "`
+for czfip in `$grep -v ^# $pimp_2way_nat|$cut -f 1 -d " "`
 do
- pubip=`grep "$czfip " $pimp_2way_nat|cut -f 2 -d " "`
- czffirstindex=priv_`ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ |tr [./] _`
- czfsecondindex=priv_`ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ |tr [./] _`
- czfthirdindex=priv_`ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ |tr [./] _`
- pubfirstindex=pub_`ipcalc -n $pubip/27|grep Network|cut -f 4 -d \ |tr [./] _`
- pubsecondindex=pub_`ipcalc -n $pubip/29|grep Network|cut -f 4 -d \ |tr [./] _`
-
- if ! grep $czffirstindex $restoredata > /dev/null
+ pubip=`$grep "$czfip " $pimp_2way_nat|$cut -f 2 -d " "`
+ czffirstindex=priv_`$ipcalc -n $czfip/$czffirstbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czfsecondindex=priv_`$ipcalc -n $czfip/$czfsecondbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czfthirdindex=priv_`$ipcalc -n $czfip/$czfthirdbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czffourthindex=priv_`$ipcalc -n $czfip/$czffourthbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ pubfirstindex=pub_`$ipcalc -n $pubip/$pubfirstbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ pubsecondindex=pub_`$ipcalc -n $pubip/$pubsecondbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+
+ if ! [[ "$chaintrack" == *"$czffirstindex"* ]]
  then
-  echo :$czffirstindex "- [0:0]" >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czffirstindex >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czffirstindex >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czffirstindex >> $restoredata
+  echo :$czffirstindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czffirstbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan1 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan2 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan3 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan4 -j $czffirstindex >> $restoretmp
+  chaintrack=\ ${czffirstindex}\ ${chaintrack}
  fi
 
- if ! grep $czfsecondindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czfsecondindex"* ]]
  then
-  echo :$czfsecondindex "- [0:0]" >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czfsecondindex >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czfsecondindex >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czfsecondindex >> $restoredata
+  echo :$czfsecondindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czfsecondbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czffirstindex -s $s -o $wan1 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan2 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan3 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan4 -j $czfsecondindex >> $restoretmp
+  chaintrack=\ ${czfsecondindex}\ ${chaintrack}
  fi
 
- if ! grep $czfthirdindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czfthirdindex"* ]]
  then
-  echo :$czfthirdindex "- [0:0]" >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czfthirdindex >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czfthirdindex >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czfthirdindex >> $restoredata
+  echo :$czfthirdindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czfthirdbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czfsecondindex -s $s -o $wan1 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan2 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan3 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan4 -j $czfthirdindex >> $restoretmp
+  chaintrack=\ ${czfthirdindex}\ ${chaintrack}
  fi
 
- if ! grep $pubfirstindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czffourthindex"* ]]
  then
-  echo :$pubfirstindex "- [0:0]" >> $restoredata
-  echo -A PREROUTING -i $wan1 -d `ipcalc -n $pubip/27|grep Network|cut -f 4 -d \ ` -j $pubfirstindex >> $restoredata
-  echo -A PREROUTING -i $wan2 -d `ipcalc -n $pubip/27|grep Network|cut -f 4 -d \ ` -j $pubfirstindex >> $restoredata
-  echo -A PREROUTING -i $wan3 -d `ipcalc -n $pubip/27|grep Network|cut -f 4 -d \ ` -j $pubfirstindex >> $restoredata
+  echo :$czffourthindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czffourthbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czfthirdindex -s $s -o $wan1 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan2 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan3 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan4 -j $czffourthindex >> $restoretmp
+  chaintrack=\ ${czffourthindex}\ ${chaintrack}
  fi
 
- if ! grep $pubsecondindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$pubfirstindex"* ]]
  then
-  echo :$pubsecondindex "- [0:0]" >> $restoredata
-  echo -A $pubfirstindex -i $wan1 -d `ipcalc -n $pubip/29|grep Network|cut -f 4 -d \ ` -j $pubsecondindex >> $restoredata
-  echo -A $pubfirstindex -i $wan2 -d `ipcalc -n $pubip/29|grep Network|cut -f 4 -d \ ` -j $pubsecondindex >> $restoredata
-  echo -A $pubfirstindex -i $wan3 -d `ipcalc -n $pubip/29|grep Network|cut -f 4 -d \ ` -j $pubsecondindex >> $restoredata
+  echo :$pubfirstindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $pubip/$pubfirstbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A PREROUTING -i $wan1 -d $s -j $pubfirstindex >> $restoretmp
+  echo -A PREROUTING -i $wan2 -d $s -j $pubfirstindex >> $restoretmp
+  echo -A PREROUTING -i $wan3 -d $s -j $pubfirstindex >> $restoretmp
+  echo -A PREROUTING -i $wan4 -d $s -j $pubfirstindex >> $restoretmp
+  chaintrack=\ ${pubfirstindex}\ ${chaintrack}
  fi
 
- echo -A $pubsecondindex -i $wan1 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoredata
- echo -A $pubsecondindex -i $wan2 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoredata
- echo -A $pubsecondindex -i $wan3 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoredata
- echo -A $pubsecondindex -i $wan1 -d $pubip/32 -j ACCEPT >> $restoredata
- echo -A $pubsecondindex -i $wan2 -d $pubip/32 -j ACCEPT >> $restoredata
- echo -A $pubsecondindex -i $wan3 -d $pubip/32 -j ACCEPT >> $restoredata
+ if ! [[ "$chaintrack" == *"$pubsecondindex"* ]]
+ then
+  echo :$pubsecondindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $pubip/$pubsecondbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $pubfirstindex -i $wan1 -d $s -j $pubsecondindex >> $restoretmp
+  echo -A $pubfirstindex -i $wan2 -d $s -j $pubsecondindex >> $restoretmp
+  echo -A $pubfirstindex -i $wan3 -d $s -j $pubsecondindex >> $restoretmp
+  echo -A $pubfirstindex -i $wan4 -d $s -j $pubsecondindex >> $restoretmp
+  chaintrack=\ ${pubsecondindex}\ ${chaintrack}
+ fi
 
+ echo -A $pubsecondindex -i $wan1 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoretmp
+ echo -A $pubsecondindex -i $wan2 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoretmp
+ echo -A $pubsecondindex -i $wan3 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoretmp
+ echo -A $pubsecondindex -i $wan4 -d $pubip/32 -j DNAT --to-destination $czfip >> $restoretmp
 
- echo -A $czfthirdindex -s $czfip/32 -o $wan1 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan2 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan3 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan1 -j ACCEPT >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan2 -j ACCEPT >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan3 -j ACCEPT >> $restoredata
+ echo -A $czffourthindex -s $czfip/32 -o $wan1 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan2 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan3 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan4 -j SNAT --to-source $pubip >> $restoretmp
 
  echo -n .
+
 done
 echo " done."
 
-echo -n "Generating new iptables-restore data - one way SNAT "
-
 # ===============================================================
 #  SNAT only using indexed iptables (should be rather function, hmm)
 # ===============================================================
+echo -n "Generating new iptables-restore data - one way SNAT "
 
-for czfip in `grep -v ^# $pimp_snat|cut -f 1 -d " "`
+for czfip in `$grep -v ^# $pimp_snat|$cut -f 1 -d " "`
 do
- pubip=`grep "$czfip " $pimp_snat|cut -f 2 -d " "`
- czffirstindex=priv_`ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ |tr [./] _`
- czfsecondindex=priv_`ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ |tr [./] _`
- czfthirdindex=priv_`ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ |tr [./] _`
+ pubip=`$grep "$czfip " $pimp_snat|$cut -f 2 -d " "`
+ czffirstindex=priv_`$ipcalc -n $czfip/$czffirstbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czfsecondindex=priv_`$ipcalc -n $czfip/$czfsecondbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czfthirdindex=priv_`$ipcalc -n $czfip/$czfthirdbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+ czffourthindex=priv_`$ipcalc -n $czfip/$czffourthbitmask|$grep Network|$cut -f 4 -d \ |tr [./] _`
+
+ if ! [[ "$chaintrack" == *"$czffirstindex"* ]]
+ then
+  echo :$czffirstindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czffirstbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan1 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan2 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan3 -j $czffirstindex >> $restoretmp
+  echo -A POSTROUTING -d ! 10.0.0.0/8 -s $s -o $wan4 -j $czffirstindex >> $restoretmp
+  chaintrack=\ ${czffirstindex}\ ${chaintrack}
+ fi
 
- if ! grep $czffirstindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czfsecondindex"* ]]
  then
-  echo :$czffirstindex "- [0:0]" >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czffirstindex >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czffirstindex >> $restoredata
-  echo -A POSTROUTING -d ! 10.0.0.0/8 -s `ipcalc -n $czfip/20|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czffirstindex >> $restoredata
+  echo :$czfsecondindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czfsecondbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czffirstindex -s $s -o $wan1 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan2 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan3 -j $czfsecondindex >> $restoretmp
+  echo -A $czffirstindex -s $s -o $wan4 -j $czfsecondindex >> $restoretmp
+  chaintrack=\ ${czfsecondindex}\ ${chaintrack}
  fi
 
- if ! grep $czfsecondindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czfthirdindex"* ]]
  then
-  echo :$czfsecondindex "- [0:0]" >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czfsecondindex >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czfsecondindex >> $restoredata
-  echo -A $czffirstindex -s `ipcalc -n $czfip/23|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czfsecondindex >> $restoredata
+  echo :$czfthirdindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czfthirdbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czfsecondindex -s $s -o $wan1 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan2 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan3 -j $czfthirdindex >> $restoretmp
+  echo -A $czfsecondindex -s $s -o $wan4 -j $czfthirdindex >> $restoretmp
+  chaintrack=\ ${czfthirdindex}\ ${chaintrack}
  fi
 
- if ! grep $czfthirdindex $restoredata > /dev/null
+ if ! [[ "$chaintrack" == *"$czffourthindex"* ]]
  then
-  echo :$czfthirdindex "- [0:0]" >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan1 -j $czfthirdindex >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan2 -j $czfthirdindex >> $restoredata
-  echo -A $czfsecondindex -s `ipcalc -n $czfip/26|grep Network|cut -f 4 -d \ ` -o $wan3 -j $czfthirdindex >> $restoredata
+  echo :$czffourthindex "- [0:0]" >> $restoretmp
+  s=`$ipcalc -n $czfip/$czffourthbitmask|$grep Network|$cut -f 4 -d \ `
+  echo -A $czfthirdindex -s $s -o $wan1 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan2 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan3 -j $czffourthindex >> $restoretmp
+  echo -A $czfthirdindex -s $s -o $wan4 -j $czffourthindex >> $restoretmp
+  chaintrack=\ ${czffourthindex}\ ${chaintrack}
  fi
 
- echo -A $czfthirdindex -s $czfip/32 -o $wan1 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan2 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan3 -j SNAT --to-source $pubip >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan1 -j ACCEPT >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan2 -j ACCEPT >> $restoredata
- echo -A $czfthirdindex -s $czfip/32 -o $wan3 -j ACCEPT >> $restoredata
+ echo -A $czffourthindex -s $czfip/32 -o $wan1 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan2 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan3 -j SNAT --to-source $pubip >> $restoretmp
+ echo -A $czffourthindex -s $czfip/32 -o $wan4 -j SNAT --to-source $pubip >> $restoretmp
 
  echo -n .
 done
 echo " done."
 
-echo COMMIT >> $restoredata
+echo COMMIT >> $restoretmp
+echo -n "Writing $restoredata"
+mv $restoretmp $restoredata