blob: 66294da580587b06d42aba3384a87d119927c6b8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#!/sbin/runscript
# Copyright 1999-2013 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Id$
extra_commands="save"
extra_started_commands="reload"
IPSET_SAVE=${IPSET_SAVE:-/var/lib/ipset/rules-save}
depend() {
before iptables ip6tables
}
checkconfig() {
if [ ! -f "${IPSET_SAVE}" ] ; then
eerror "Not starting ${SVCNAME}. First create some rules then run:"
eerror "/etc/init.d/${SVCNAME} save"
return 1
fi
return 0
}
start() {
checkconfig || return 1
ebegin "Loading ipset session"
ipset restore < "${IPSET_SAVE}"
eend $?
}
stop() {
# check if there are any references to current sets
if ! ipset list | gawk '
($1 == "References:") { refcnt += $2 }
($1 == "Type:" && $2 == "list:set") { set = 1 }
(scan) { if ($0 != "") setcnt++; else { scan = 0; set = 0 } }
(set && $1 == "Members:") {scan = 1}
END { if ((refcnt - setcnt) > 0) exit 1 }
'; then
eerror "ipset is in use, can't stop"
return 1
fi
if [ "${SAVE_ON_STOP}" = "yes" ] ; then
save || return 1
fi
ebegin "Removing kernel IP sets"
ipset flush
ipset destroy
eend $?
}
reload() {
ebegin "Reloading ipsets"
# Loading sets from a save file is only additive (there is no
# automatic flushing or replacing). And, we can not remove sets
# that are currently used in existing iptables rules.
#
# Instead, we create new temp sets for any set that is already
# in use, and then atomically swap them into place.
#
# XXX: This does not clean out previously used ipsets that are
# not in the new saved policy--it can't, because they may still
# be referenced in the current iptables rules.
# Build a list of all currently used sets (if any).
running_ipset_list=$(ipset save | gawk '/^create/{printf "%s ",$2}')
running_ipset_list="${running_ipset_list% }"
# Build a regular expression that matches those set names.
running_ipset_list_regex="${running_ipset_list// /|}"
# Load up sets from the save file, but rename any set that already
# exists to a temporary name that we will swap later.
if ! cat ${IPSET_SAVE} | sed -r "s/^(create|add) (${running_ipset_list_regex}) /\1 \2_atomic_temp /" | ipset restore ; then
eend $? "Failed to load new ipsets"
fi
# Now for every set name that currently exists, atomically swap it
# with the temporary new one we created, and then destroy the old set.
for ipset_name in ${running_ipset_list} ; do
ipset swap ${ipset_name} ${ipset_name}_atomic_temp || eend $? "Failed to swap in new ipset $ipset_name"
ipset destroy ${ipset_name}_atomic_temp || eend $? "Failed to delete obsolete ipset ${ipset_name}_atomic_temp"
done
eend 0
}
save() {
ebegin "Saving ipset session"
touch "${IPSET_SAVE}"
chmod 0600 "${IPSET_SAVE}"
ipset save > "${IPSET_SAVE}"
eend $?
}
|
GitWeb