gentoo-overlay/net-firewall/ipset/files/ipset.initd-r7

#!/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 $?
}