diff options
Diffstat (limited to 'net-firewall/ipset/files/ipset.initd-r7')
-rw-r--r-- | net-firewall/ipset/files/ipset.initd-r7 | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/net-firewall/ipset/files/ipset.initd-r7 b/net-firewall/ipset/files/ipset.initd-r7 new file mode 100644 index 000000000000..a96e69617e8a --- /dev/null +++ b/net-firewall/ipset/files/ipset.initd-r7 @@ -0,0 +1,127 @@ +#!/sbin/openrc-run +# Copyright 1999-2013 Gentoo Foundation +# Distributed under the terms of the GNU General Public License v2 + +extra_commands="save" +extra_started_commands="reload" + +IPSET_SAVE=${IPSET_SAVE:-/var/lib/ipset/rules-save} +ipset_name_regex='^[-_.a-zA-Z0-9]+$' + +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); + # also make sure all sets' names are fully supported. + IFS_O=$IFS + NEWLINE=' +' + local IFS=$NEWLINE + for running_ipset in $(ipset -n list) ; do + if ! echo "${running_ipset}" | grep -q -E "${ipset_name_regex}" ; then + eend 1 "Running set name ('$running_ipset') does not match regex '${ipset_name_regex}'" + return 1 + fi + running_ipset_list="$running_ipset_list $running_ipset" + done + IFS=$IFS_O + running_ipset_list="${running_ipset_list# }" + + # Check the configured suffix, and make sure there are no collisions + if test -z "${TEMP_SUFFIX}" ; then + eend 1 "TEMP_SUFFIX cannot be empty" + return 1 + fi + + # Make sure TEMP_SUFFIX uses only known-safe characters. + if ! echo "${TEMP_SUFFIX}" | grep -q -E "${ipset_name_regex}" ; then + eend 1 "TEMP_SUFFIX pattern ('$TEMP_SUFFIX') does not match regex '${ipset_name_regex}'" + return 1 + fi + # We don't block ., but it must be escaped. + temp_suffix_regex=$(echo "${TEMP_SUFFIX}" | sed 's/\./\\./g') + + if echo "$running_ipset_list" | grep -q -E "${temp_suffix_regex}( |$)" ; then + eend 1 "Existing set(s) match TEMP_SUFFIX pattern ('${TEMP_SUFFIX}'), cannot continue" + return 1 + fi + + # Build a regular expression that matches those set names. + running_ipset_list_regex="$(echo "$running_ipset_list" | tr -s ' ' '|' )" + + # 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${temp_suffix_regex} /" | 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}${TEMP_SUFFIX}" || eend $? "Failed to swap in new ipset $ipset_name" + ipset destroy "${ipset_name}${TEMP_SUFFIX}" || eend $? "Failed to delete obsolete ipset ${ipset_name}${TEMP_SUFFIX}" + done + eend 0 +} + +save() { + ebegin "Saving ipset session" + checkpath --file --mode 0600 "${IPSET_SAVE}" + ipset -output save list > "${IPSET_SAVE}" + eend $? +} |