firewall: allow local redirection of ports

[openwrt/openwrt.git] / package / firewall / files / lib / core_redirect.sh

diff --git a/package/firewall/files/lib/core_redirect.sh b/package/firewall/files/lib/core_redirect.sh
index 87f584e37bf44912639491c5ef8d16e8a98743ad..64c619e434fe0b8203875ccbb8af2d3c11c4ba46 100644 (file)
--- a/package/firewall/files/lib/core_redirect.sh
+++ b/package/firewall/files/lib/core_redirect.sh
@@ -17,6 +17,7 @@ fw_config_get_redirect() {
                string dest_port "" \
                string proto "tcpudp" \
                string family "" \
+               string target "DNAT" \
        } || return
        [ -n "$redirect_name" ] || redirect_name=$redirect__name
 }
@@ -26,36 +27,84 @@ fw_load_redirect() {
 
        fw_callback pre redirect
 
-       [ -n "$redirect_src" -a -n "$redirect_dest_ip" ] || {
-               fw_die "redirect ${redirect_name}: needs src and dest_ip"
-       }
+       local fwdchain natchain natopt nataddr natports srcdaddr srcdports
+       if [ "$redirect_target" == "DNAT" ]; then
+               [ -n "$redirect_src" -a -n "$redirect_dest_ip$redirect_dest_port" ] || {
+                       fw_log error "DNAT redirect ${redirect_name}: needs src and dest_ip or dest_port, skipping"
+                       return 0
+               }
+
+               fwdchain="zone_${redirect_src}${redirect_dest_ip:+_forward}"
+
+               natopt="--to-destination"
+               natchain="zone_${redirect_src}_prerouting"
+               nataddr="$redirect_dest_ip"
+               fw_get_port_range natports "$redirect_dest_port" "-"
+
+               fw_get_negation srcdaddr '-d' "${redirect_src_dip:+$redirect_src_dip/$redirect_src_dip_prefixlen}"
+               fw_get_port_range srcdports "$redirect_src_dport" ":"
+
+               list_contains FW_CONNTRACK_ZONES $redirect_src || \
+                       append FW_CONNTRACK_ZONES $redirect_src
+
+       elif [ "$redirect_target" == "SNAT" ]; then
+               [ -n "$redirect_dest" -a -n "$redirect_src_dip" ] || {
+                       fw_log error "SNAT redirect ${redirect_name}: needs dest and src_dip, skipping"
+                       return 0
+               }
+
+               fwdchain="${redirect_src:+zone_${redirect_src}_forward}"
 
-       local mode=$(fw_get_family_mode ${redirect_family:-x} $redirect_src I)
+               natopt="--to-source"
+               natchain="zone_${redirect_dest}_nat"
+               nataddr="$redirect_src_dip"
+               fw_get_port_range natports "$redirect_src_dport" "-"
 
-       local nat_dest_port=$redirect_dest_port
-       redirect_dest_port=$(fw_get_port_range $redirect_dest_port)
-       redirect_src_port=$(fw_get_port_range $redirect_src_port)
-       redirect_src_dport=$(fw_get_port_range $redirect_src_dport)
-       local fwd_dest_port=${redirect_dest_port:-$redirect_src_dport}
+               fw_get_negation srcdaddr '-d' "${redirect_dest_ip:+$redirect_dest_ip/$redirect_dest_ip_prefixlen}"
+               fw_get_port_range srcdports "$redirect_dest_port" ":"
+
+               list_contains FW_CONNTRACK_ZONES $redirect_dest || \
+                       append FW_CONNTRACK_ZONES $redirect_dest
+
+       else
+               fw_log error "redirect ${redirect_name}: target must be either DNAT or SNAT, skipping"
+               return 0
+       fi
+
+       local mode
+       fw_get_family_mode mode ${redirect_family:-x} ${redirect_src:-$redirect_dest} I
+
+       local srcaddr
+       fw_get_negation srcaddr '-s' "${redirect_src_ip:+$redirect_src_ip/$redirect_src_ip_prefixlen}"
+
+       local srcports
+       fw_get_port_range srcports "$redirect_src_port" ":"
+
+       local destaddr
+       fw_get_negation destaddr '-d' "${redirect_dest_ip:+$redirect_dest_ip/$redirect_dest_ip_prefixlen}"
+
+       local destports
+       fw_get_port_range destports "${redirect_dest_port:-$redirect_src_dport}" ":"
 
        [ "$redirect_proto" == "tcpudp" ] && redirect_proto="tcp udp"
        for redirect_proto in $redirect_proto; do
-               fw add $mode n zone_${redirect_src}_prerouting DNAT $ { $redirect_src_ip $redirect_dest_ip } { \
+               local pos
+               eval 'pos=$((++FW__REDIR_COUNT_'${mode#G}'_'$natchain'))'
+
+               fw add $mode n $natchain $redirect_target $pos { $redirect_src_ip $redirect_dest_ip } { \
+                       $srcaddr $srcdaddr \
                        ${redirect_proto:+-p $redirect_proto} \
-                       ${redirect_src_ip:+-s $redirect_src_ip/$redirect_src_ip_prefixlen} \
-                       ${redirect_src_dip:+-d $redirect_src_dip/$redirect_src_dip_prefixlen} \
-                       ${redirect_src_port:+--sport $redirect_src_port} \
-                       ${redirect_src_dport:+--dport $redirect_src_dport} \
+                       ${srcports:+--sport $srcports} \
+                       ${srcdports:+--dport $srcdports} \
                        ${redirect_src_mac:+-m mac --mac-source $redirect_src_mac} \
-                       --to-destination ${redirect_dest_ip}${redirect_dest_port:+:$nat_dest_port} \
+                       $natopt $nataddr${natports:+:$natports} \
                }
 
-               fw add $mode f zone_${redirect_src}_forward ACCEPT ^ { $redirect_src_ip $redirect_dest_ip } { \
-                       -d $redirect_dest_ip \
+               fw add $mode f ${fwdchain:-forward} ACCEPT ^ { $redirect_src_ip $redirect_dest_ip } { \
+                       $srcaddr ${destaddr:--m conntrack --ctstate DNAT} \
                        ${redirect_proto:+-p $redirect_proto} \
-                       ${redirect_src_ip:+-s $redirect_src_ip/$redirect_src_ip_prefixlen} \
-                       ${redirect_src_port:+--sport $redirect_src_port} \
-                       ${fwd_dest_port:+--dport $fwd_dest_port} \
+                       ${srcports:+--sport $srcports} \
+                       ${destports:+--dport $destports} \
                        ${redirect_src_mac:+-m mac --mac-source $redirect_src_mac} \
                }
        done