source: trunk/package/firewall/files/lib/fw.sh @ 21378

Last change on this file since 21378 was 21378, checked in by jow, 7 years ago

[package] firewall: properly unset position for delete command, fixes rule removal in ifdown

File size: 3.2 KB
Line 
1# Copyright (C) 2009-2010 OpenWrt.org
2# Copyright (C) 2009 Malte S. Stretz
3
4export FW_4_ERROR=0
5export FW_6_ERROR=0
6export FW_i_ERROR=0
7export FW_e_ERROR=0
8export FW_a_ERROR=0
9
10#TODO: remove this
11[ "${-#*x}" == "$-" ] && {
12        fw() {
13                fw__exec "$@"
14        }
15} || {
16        fw() {
17                local os=$-
18                set +x
19                fw__exec "$@"
20                local rc=$?
21                set -$os
22                return $rc
23        }
24}
25
26fw__exec() { # <action> <family> <table> <chain> <target> <position> { <rules> }
27        local cmd fam tab chn tgt pos
28        local i
29        for i in cmd fam tab chn tgt pos; do
30                if [ "$1" -a "$1" != '{' ]; then
31                        eval "$i='$1'"
32                        shift
33                else
34                        eval "$i=-"
35                fi
36        done
37
38        fw__rc() {
39                export FW_${fam}_ERROR=$1
40                return $1
41        }
42
43        fw__dualip() {
44                fw $cmd 4 $tab $chn $tgt $pos "$@"
45                fw $cmd 6 $tab $chn $tgt $pos "$@"
46                fw__rc $((FW_4_ERROR | FW_6_ERROR))
47        }
48
49        fw__autoip() {
50                local ip4 ip6
51                shift
52                while [ "$1" != '}' ]; do
53                        case "$1" in
54                                *.*.*.*) ip4=1 ;;
55                                *:*) ip6=1 ;;
56                        esac
57                        shift
58                done
59                shift
60                if [ "${ip4:-4}" == "${ip6:-6}" ]; then
61                        echo "fw: can't mix ip4 and ip6" >&2
62                        return 1
63                fi
64                local ver=${ip4:+4}${ip6:+6}
65                fam=i
66                fw $cmd ${ver:-i} $tab $chn $tgt $pos "$@"
67                fw__rc $?
68        }
69
70        fw__has() {
71                local tab=${1:-$tab}
72                if [ $tab == '-' ]; then
73                        type $app > /dev/null 2> /dev/null
74                        fw__rc $(($? & 1))
75                        return
76                fi
77                local mod
78                eval "mod=\$FW_${fam}_${tab}"
79                if [ "$mod" ]; then
80                        fw__rc $mod
81                        return
82                fi
83                case "$fam" in
84                        4) mod=iptable_${tab} ;;
85                        6) mod=ip6table_${tab} ;;
86                        *) mod=. ;;
87                esac
88                grep "^${mod} " /proc/modules > /dev/null
89                mod=$?
90                export FW_${fam}_${tab}=$mod
91                fw__rc $mod
92        }
93
94        fw__err() {
95                local err
96                eval "err=\$FW_${fam}_ERROR"
97                fw__rc $err
98        }
99
100        local app=
101        local pol=
102        case "$fam" in
103                4) app=iptables ;;
104                6) app=ip6tables ;;
105                i) fw__dualip "$@"; return ;;
106                I) fw__autoip "$@"; return ;;
107                e) app=ebtables ;;
108                a) app=arptables ;;
109                -) fw $cmd i $tab $chn $tgt $pos "$@"; return ;;
110                *) return 254 ;;
111        esac
112        case "$tab" in
113                f) tab=filter ;;
114                m) tab=mangle ;;
115                n) tab=nat ;;
116                r) tab=raw ;;
117                -) tab=filter ;;
118        esac
119        case "$cmd:$chn:$tgt:$pos" in
120                add:*:-:*) cmd=new-chain ;;
121                add:*:*:-) cmd=append ;;
122                add:*:*:$) cmd=append ;;
123                add:*:*:*) cmd=insert ;;
124                del:-:*:*) cmd=delete-chain; fw flush $fam $tab ;;
125                del:*:-:*) cmd=delete-chain; fw flush $fam $tab $chn ;;
126                del:*:*:*) cmd=delete ;;
127                flush:*) ;;
128                policy:*) pol=$tgt; tgt=- ;;
129                has:*) fw__has; return ;;
130                err:*) fw__err; return ;;
131                list:*) cmd="numeric --verbose --$cmd" ;;
132                *) return 254 ;;
133        esac
134        case "$chn" in
135                -) chn= ;;
136        esac
137        case "$tgt" in
138                -) tgt= ;;
139        esac
140        case "$pos" in
141                ^) pos=1 ;;
142                $) pos= ;;
143                -) pos= ;;
144        esac
145
146        if ! fw__has - family || ! fw__has $tab ; then
147                export FW_${fam}_ERROR=0
148                return 0
149        fi
150
151        if [ $# -gt 0 ]; then
152                shift
153                if [ $cmd == delete ]; then
154                        pos=
155                fi
156        fi
157        while [ $# -gt 1 ]; do
158                echo -n  "$1"
159                echo -ne "\0"
160                shift
161        done | xargs -0 ${FW_TRACE:+-t} \
162                $app --table ${tab} --${cmd} ${chn} ${pol} ${pos} ${tgt:+--jump "$tgt"}
163        fw__rc $?
164}
165
166fw_get_port_range() {
167        local ports=$1
168        local delim=${2:-:}
169        if [ "$3" ]; then
170                fw_get_port_range "${ports}-${3}" $delim
171                return
172        fi
173
174        local first=${ports%-*}
175        local last=${ports#*-}
176        if [ "$first" != "$last" ]; then
177                echo "$first$delim$last"
178        else
179                echo "$first"
180        fi
181}
182
Note: See TracBrowser for help on using the repository browser.