Make deletion optional in vif_down
[vsys-scripts.git] / exec / ipfw-be
index 801f041..b3b9439 100755 (executable)
 # - configures the firewall
 # - writes results on the output vsys pipe
 #
-# Configurable variables are at the beginning
+# Configurable variables are at the beginning (only HOOK so far)
 
-# If HOOK is set the program is called before configuring a rule.
+# If HOOK is set, ${HOOK} is called before configuring a rule.
 # A sample hook can be found in the ipfwroot.rpm package,
 # it can be used to collect statistical information on dummynet usage.
-# To configure an hook, set the HOOK variable as follow:
+# To configure a hook, set the HOOK variable as follow:
 # HOOK=/tmp/sample_hook
 
-# You should not touch anything below.
+#--- You should not touch anything below this line. ----
+# For documentation see ARCHITECTURE near the end of the file.
 
-# We assume three type of connections
-#  SERVER we know the local port P, and do the
-#      bind/listen/accept on the local socket.
-#              pipe_in in dst-port P
-#              pipe_out out src-port P
-#
-#  CLIENT we know the remote port P, and do a connect to it
-#      (src and dst are swapped wrt the previous case)
-#              pipe_in in src-port P
-#              pipe_out out dst-port P
-#
-#  SERVICE we run a server on local port P, and also connect
-#      from local clients to remote servers on port P.
-#              pipe_in in { dst-port P or src-port P }
-#              pipe_out out { src-port P or dst-port P }
-# 
-#  On a given port a user can have one CLIENT and/or one SERVER
-#  configuration or one SERVICE configuration.
-#  When a SERVICE configuration is installed any existing CLIENT
-#  and SERVER configuration on the same port are removed.
-#  When a CLIENT or SERVER configuration is installed any existing
-#  SERVICE configuration on the same port is removed.
-#
-#  The following is a case that is implemented as SERVER
-#  D   we run a server on local port P, and also connect
-#      to remote servers but doing a bind(P) before connect().
-#      In terms of rules, this is not distinguishable from
-#      the SERVER case, however it would be different if we
-#      had a way to tell SERVER from CLIENT sockets
-#              pipe_in in dst-port P
-#              pipe_out out src-port P
-#
-# The database of current ipfw and dummynet configuration is in a
-# file which is regenerated on errors. The format is
-#
-#      slice_id type arg rule_base pipe_base timeout
-#
-# (lines starting with '#' are comments and are ignored)
-# For each configuration we allocate one rule number in ipfw,
-# and two sequential pipe numbers.
-
-# globals, do not touch below
+#--- global variables ---
 VERBOSE=0      # set to !0 to enable debug messages
 TEST=0         # set to 1 for test mode
 
+# The database and the lock file
 DBFILE=/tmp/ff
 lockfile=/var/lock/ipfw.lock
 
-# These values are the keys used in the database for blocks,
-# rules and pipes
-# The index rule numbers allocated to a slice can be computed
-# with the following formula:
-#      index_min_rule = $(($RULE_BASE + $(($M*$block_n))))
-# where block_n is the block number associated with the slice
-# and M is the block size.
+# Min and max value (inclusive) for block_index
 BLOCK_MIN=1
 BLOCK_MAX=1000
-M=50           # block size
-RULE_BASE=10001 # the hightest rule is RULE_BASE + (M*BLOCK_MAX)
+M=50           # size of per-slice block of rules
+# Min and max value (inclusive) for pipe_index
 PIPE_MIN=1
 PIPE_MAX=25000
+
 # These are the actual rule numbers used in ipfw
-IPFW_RULE_MIN=10000
-IPFW_RULE_MAX=59999
-IPFW_PIPE_MIN=10000
-IPFW_PIPE_MAX=59999
-# The mapping between keys and ipfw configuration number follow:
-# rule_nr 10001..20000 are mapped to ipfw_rules 50000..59999 (n+39999)
-# pipe_nr 1..25000 are mapped to ipfw_pipes 10000-59999 (n*2+9998)
-# Rule index can be mapped to ipfw rules with the following formula:
-#      ipfw_rule = $(($index_rule + 39999))
-# Pipes index can be mapped to ipfw pipes with:
-#      ipfw_pipein = $(($index_pipe + $index_pipe + 9998))
-#      ipfw_pipeout = $(($ipfw_pipein + $1))
-#
-# the skipto and the generic default rule
+IPFW_RULE_MIN=10000    # initial per-slice rule number
+IPFW_PIPE_MIN=10000    # initial pipe number
+
+# The skipto and the generic default rule
 # these values are used to initialize the firewall
 SLICE_TABLE=1          # table number used for slice ids lookup
 S=1000                 # firewall rule number for the skipto rule
@@ -105,11 +51,8 @@ D=2000                      # default rule for reserved section
 # set slicename and slice_id
 # these are the credential of the user invoking the backend
 SLICENAME=$1
-if [ $SLICENAME = 0 ]; then
-       SLICE_ID=0
-else
-       SLICE_ID=`id -u $SLICENAME`
-fi
+SLICE_ID=`id -u $SLICENAME`
+[ x"$SLICE_ID" = x"" ] && echo "No sliver present." && exit
 
 # programs
 # XXX check consistency for variables {}
@@ -128,6 +71,7 @@ set_verbose() {
     [ x"$1" = x"-v" -o x"$2" = x"-v" ] && VERBOSE=1
 }
 
+# set test mode if -q is found
 set_test() {
     [ x"$1" = x"-q" -o x"$2" = x"-q" ] || return
     TEST=1
@@ -142,13 +86,13 @@ abort() { # $1 message to be displayed in case of error
 }
 
 # remove dangerous characters from user input
-# if present, the leading '-v/-t' will be removed
+# if present, the leading '-v/-q' will be removed
 filter() { # $* variables to be filtered
        [ x${1} = x"-v" -o x${1} = x"-q" ] && shift
        [ x${1} = x"-v" -o x${1} = x"-q" ] && shift
        # allowed chars are: numbers, uppercase and lowercase letters,
        # spaces, and the following symbols: .,_-/
-       echo "$*" | ${SED} ${SEDOPT} 's/[^\t0-9a-zA-Z., _\/\{}-]*//g'
+       echo "$*" | ${SED} ${SEDOPT} 's/[^\t0-9a-zA-Z., _\/\{}@-]*//g'
 }
 
 # remove all entries from the ipfw config, and create an empty db
@@ -159,46 +103,68 @@ clean_db() {
        # is not supported so for the time being we kill them all
        ${IPFW} -q flush
        ${IPFW} -q pipe flush
-       # ${IPFW} delete ${IPFW_RULE_MIN}-${IPFW_RULE_MAX}
-       # ${IPFW} pipe delete ${IPFW_PIPE_MIN}-${IPFW_PIPE_MAX}
+       ${IPFW} -q table $SLICE_TABLE flush
+       #${IPFW} delete ${IPFW_RULE_MIN}-${IPFW_RULE_MAX}
+       #${IPFW} pipe delete ${IPFW_PIPE_MIN}-${IPFW_PIPE_MAX}
        # since all rules are now deleted, we should initialize the firewall 
        ipfw_init
 }
 
+#
 # Add the ipfw rule/pipe and update the database.
-# The pipe-in and pipe_out config are through global variables
-# CONFIG_IN CONFIG_OUT because they may be long.
+# The pipe-in and pipe-out config are through global variables
+# rule_in rule_out because they may be long. XXX why ?
 # Other arguments are on the command line
-add_rule() { # new_rule type arg rule pipe_base timeout
-    local new_rule=$1 type=$2 arg=$3
-    local rule_base=$4 pipe_base=$5 timeout=$6
-    local pipe_in pipe_out rule_in rule_out check_timeout
+#
+# the new_rule variable is set if the rule to be installed is new
+# we need to know this because we do not want to clean
+# rule counters on pipes reconfiguration
+add_rule() { # slice_id new_rule type arg ipfw_rule pipe_index timeout
+    local slice_id=$1 new_rule=$2 type=$3 arg=$4
+    local ipfw_rule=$5 pipe_index=$6 timeout=$7
+    local ipfw_pipe_in ipfw_pipe_out check_timeout
+    local p h # used to split the argument
+
+    local h_in h_out
+    # local rule_in rule_out # XXX test if this works
+    # find actual pipe numbers
+    ipfw_pipe_in=$(($IPFW_PIPE_MIN + $((2 * $(($pipe_index - 1)))) ))
+    ipfw_pipe_out=$(($ipfw_pipe_in + 1))
+    local del          # used to delete incompatible configurations
+
+    # split the argument, and prepare PORTLIST (p) and ADDRLIST (h)
+    p=`echo $arg | cut -s -d "@" -f1-` # empty if no separator
+    if [ "$p" = "" ] ; then
+       p=$arg
+    else
+       p=`echo $arg | cut -d "@" -f1`
+       h=`echo $arg | cut -d "@" -f2`
+    fi
 
-    # If we use a profile file, locate the user directory
-    # move in the slice root dir XXX todo
-    [ "$TEST" != "1" ] && cd /vservers/${SLICENAME}/root
-    #echo ${CONFIG_STRING} | ${SED} -e "s/ profile \(.[^ ]\)/ profile \/vservers\/${SLICENAME}\/\1/g"
+    if [ "$h" = "" ] ; then
+       h_in=""
+       h_out=""
+    else
+       h_in=" src-ip ${h} "
+       h_out=" dst-ip ${h} "
+    fi
 
     # first, call ipfw -n to check syntax, if ok move on and do the action
-    pipe_in=$(($pipe_base + $pipe_base + 9998))
-    pipe_out=$(($pipe_in + 1))
-    local del  # anything to delete ?
-    local rule_nr=$(($rule_base + 39999))  # formula for individual rules
     if [ x"$new_rule" != x"0" ] ; then
        case $type in
-       server)
-           rule_in="dst-port $arg"
-           rule_out="src-port $arg"
+       SERVER|server)
+           rule_in="dst-port $p"
+           rule_out="src-port $p"
            del=service
            ;;
-       client)
-           rule_in="src-port $arg"
-           rule_out="dst-port $arg"
+       CLIENT|client)
+           rule_in="src-port $p"
+           rule_out="dst-port $p"
            del=service
            ;;
-       service)
-           rule_in="{ src-port $arg or dst-port $arg }"
-           rule_out="{ src-port $arg or dst-port $arg }"
+       SERVICE|service)
+           rule_in="{ src-port $p or dst-port $p }"
+           rule_out="{ src-port $p or dst-port $p }"
            del="cli_ser"
            ;;
        *)
@@ -206,74 +172,89 @@ add_rule() { # new_rule type arg rule pipe_base timeout
            ;;
        esac
 
-       rule_in="pipe ${pipe_in} in jail $SLICE_ID ${rule_in} // $type $arg"
-       rule_out="pipe ${pipe_out} out jail $SLICE_ID ${rule_out} // $type $arg"
-       ${IPFW_CHECK} add ${rule_nr} $rule_in > /dev/null || \
-               abort "ipfw syntax error $rule_in"
-       ${IPFW_CHECK} add ${rule_nr} $rule_out > /dev/null || \
-               abort "ipfw syntax error $rule_out"
+       rule_in="pipe ${ipfw_pipe_in} in ${h_in} ${rule_in} // $type $arg $slice_id"
+       rule_out="pipe ${ipfw_pipe_out} out ${h_out} ${rule_out} // $type $arg $slice_id"
+
+       # Move into the user root directory. The profile should be located there
+       ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW_CHECK} add ${ipfw_rule} ${rule_in} ) > /dev/null || \
+               abort "ipfw syntax error ${rule_in}" 
+       ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW_CHECK} add ${ipfw_rule} ${rule_out} ) > /dev/null || \
+               abort "ipfw syntax error ${rule_out}" 
     fi
 
     # check error reporting
-    ${IPFW_CHECK} pipe ${pipe_in} config ${CONFIG_PIPE_IN} > /dev/null || \
-               abort "ipfw syntax error pipe_in"
-    ${IPFW_CHECK} pipe ${pipe_out} config ${CONFIG_PIPE_OUT} > /dev/null || \
+    ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW_CHECK} pipe ${ipfw_pipe_in} config ${CONFIG_PIPE_IN} ) > /dev/null || \
+               abort "ipfw syntax error pipe_in" 
+    ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW_CHECK} pipe ${ipfw_pipe_out} config ${CONFIG_PIPE_OUT} ) > /dev/null || \
                abort "ipfw syntax error pipe_out"
 
     # all good, delete and add rules if necessary
-    [ "$del" = "service" ] && do_delete $SLICE_ID service $arg
-    [ "$del" = "cli_ser" ] && do_delete $SLICE_ID client $arg
-    [ "$del" = "cli_ser" ] && do_delete $SLICE_ID server $arg
-    [ "$new_rule" != "0" ] && ${IPFW} add ${rule_nr} $rule_in > /dev/null
-    [ "$new_rule" != "0" ] && ${IPFW} add ${rule_nr} $rule_out > /dev/null
+    [ "$del" = "service" ] && do_delete 0 $slice_id service $arg
+    [ "$del" = "cli_ser" ] && do_delete 0 $slice_id client $arg
+    [ "$del" = "cli_ser" ] && do_delete 0 $slice_id server $arg
+    [ "$new_rule" != "0" ] && ${IPFW} add ${ipfw_rule} $rule_in > /dev/null
+    [ "$new_rule" != "0" ] && ${IPFW} add ${ipfw_rule} $rule_out > /dev/null
     # config pipes
-    ${IPFW} pipe ${pipe_in} config ${CONFIG_PIPE_IN}
-    ${IPFW} pipe ${pipe_out} config ${CONFIG_PIPE_OUT}
+    ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW} pipe ${ipfw_pipe_in} config ${CONFIG_PIPE_IN} )
+    ( cd /vservers/${SLICENAME}/`pwd`/ ; ${IPFW} pipe ${ipfw_pipe_out} config ${CONFIG_PIPE_OUT} )
 
     # send output to the user
-    ${IPFW} show ${rule_nr}
-    ${IPFW} pipe ${pipe_in} show
-    ${IPFW} pipe ${pipe_out} show
+    ${IPFW} show ${ipfw_rule}
+    ${IPFW} pipe ${ipfw_pipe_in} show
+    ${IPFW} pipe ${ipfw_pipe_out} show
 
+    # do not write on the database on test-only
     [ "$TEST" = "1" ] && return
     # add to the database
-    ( grep -v -- "^${SLICE_ID} ${type} ${arg} " $DBFILE;  \
-       echo "${SLICE_ID} ${type} ${arg} ${rule_base} ${pipe_base} ${timeout}" ) > ${DBFILE}.tmp
+    ( grep -iv -- "^${slice_id} ${type} ${arg} " $DBFILE;  \
+       echo "${slice_id} ${type} ${arg} ${ipfw_rule} ${pipe_index} ${timeout}" ) > ${DBFILE}.tmp
     mv ${DBFILE}.tmp ${DBFILE}
 }
 
+#
 # Delete a given configuration
-do_delete() { # type arg
-    local pipe_in pipe_out pipe_base rule_base rule_nr
-    local type=$1 arg=$2
-
-    [ "${type}" = "BLOCK" ] && abort "A Block can not be deleted"
-    [ "${arg}" = "" ] && abort "Missing args on 'delete', expected on of {SERVICE|SERVER|CLIENT} port_number"
-    set `find_rule $SLICE_ID $type $arg`
-    rule_base=$1; pipe_base=$2
-    [ "$rule_base" = "0" ] && return           # no rules found
-
-    rule_nr=$(($rule_base + 39999))            # XXX only individual rules
-    pipe_in=$(($pipe_base + $pipe_base + 9998))
-    pipe_out=$(($pipe_in + 1))
-
-    $IPFW delete ${rule_nr}
-    $IPFW pipe delete ${pipe_in}
-    $IPFW pipe delete ${pipe_out}
-    echo "removed configuration ${SLICE_ID} ${type} ${arg}"
-    [ "$TEST" = "1" ] && return
-    # remove from the database
-    grep -v -- "^${SLICE_ID} ${type} ${arg} " $DBFILE > ${DBFILE}.tmp
+# if block_deletion !0 free block resources (if necessary)
+# otherwise leave the block allocated in case
+# we are adding the first rule
+do_delete() { # block_deletion slice_id type arg
+    local ipfw_pipe_in ipfw_pipe_out pipe_index ipfw_rule
+    local block_deletion=$1 slice_id=$2 type=$3 arg=$4
+
+    [ "${type}" = "BLOCK" ] && abort "A BLOCK can not be deleted"
+    [ "${arg}" = "" ] && abort "Missing args on 'delete', expected on of {CLIENT|SERVER|SERVICE} arg"
+    set `find_rule $slice_id $type $arg`
+    ipfw_rule=$1; pipe_index=$2
+    [ "$ipfw_rule" = "0" ] && return           # no rules found
+
+    # find actual pipe numbers XXX do as function
+    ipfw_pipe_in=$(($IPFW_PIPE_MIN + $((2 * $(($pipe_index - 1)))) ))
+    ipfw_pipe_out=$(($ipfw_pipe_in + 1))
+
+    echo "removing configuration ${slice_id} ${type} ${arg}"
+    [ "$TEST" = "1" ] && return 0
+    $IPFW delete ${ipfw_rule}
+    $IPFW pipe delete ${ipfw_pipe_in}
+    $IPFW pipe delete ${ipfw_pipe_out}
+    # remove from the database (case insensitive)
+    grep -iv -- "^${slice_id} ${type} ${arg} " $DBFILE > ${DBFILE}.tmp
     mv ${DBFILE}.tmp ${DBFILE}
 
-    # XXX if the use block is empty
+    # if there are no more rules for the user
     # remove the table entry from ipfw and from the db
-    # not yet implemented
+    [ $block_deletion = 0 ] && return 0
+
+    local rule_counter=`grep ^${slice_id} ${DBFILE} | wc -l`
+    [ $rule_counter -gt 1 ] && return 0        # there are still user rules
+    # delete the block and clean the table
+    local block_n=`grep "^${slice_id} BLOCK" ${DBFILE} | cut -d " " -f 3`
+    debug "Deleting BLOCK <${block_n}> entry from ipfw and from the database"
+    table_remove $slice_id $block_n 
 }
 
 # compare the argument with the first two field of
 # the database.
 # On match returns the block number, otherwise returns 0.
+# no echo inside
 find_block() { # $1 slice_id
     local ret
     ret=`grep -- "^$1 BLOCK " $DBFILE`
@@ -285,15 +266,79 @@ find_block() { # $1 slice_id
     echo "$3"
 }
 
+#
+# remove the default user rule and
+# the a BLOCK entry from ipfw and update the db
+# no echo inside
+table_remove() { # $slice_id $block_n
+    [ "$TEST" = "1" ] && return 0
+
+    # compute and delete the last user rule
+    local ipfw_rulemax=$(($IPFW_RULE_MIN + $(($M *${block_n})) -1))
+    ${IPFW} table $SLICE_TABLE delete $slice_id
+    ${IPFW} delete ${ipfw_rulemax}
+    ( grep -iv -- "^${slice_id} BLOCK ${block_n}" $DBFILE; ) > ${DBFILE}.tmp
+    mv ${DBFILE}.tmp ${DBFILE}
+    return 0
+}
+
+#
+# Find a rule and pipe_index for the given key (xid type arg)
+# Allocate a new block if first entry for this xid.
+# Rule and pipe are not written into the database, only the block is.
+#
+# Return ipfw_rule pipe_index new_rule
+# 'new_rule' is 0 if the rule existed, 1 if it is new
+#
+# return ipfw_rule = 0 if there are no resources available
+find_allocate() { # slice_id type arg
+    local slice_id=$1 type=$2 arg=$3
+    local ipfw_rule pipe_index new_block=0
+
+    # search for already allocated rule and pipes
+    set `find_rule $slice_id $type $arg`
+    ipfw_rule=$1; pipe_index=$2
+    [ ! ${ipfw_rule} = 0 ] && echo $ipfw_rule $pipe_index "0" && return 0      # rules found, return
+
+    # no rules found, search for an already existing block, or
+    # allocate a new one
+    local block_n=`find_block ${slice_id}`
+    [ ${block_n} = "0" ] && new_block=1 && block_n=`find_free_block`
+    [ ${block_n} = "0" -o ${block_n} -gt $BLOCK_MAX ] && echo 0 && return 0;
+
+    # We have a valid block, compute the range for user rules
+    local ipfw_rulemin=$(($IPFW_RULE_MIN + $(($M *$(($block_n - 1))))))
+    local ipfw_rulemax=$(($(($ipfw_rulemin + $M)) - 1 ))
+
+    # Find rule and pipes, reserve the last rule for the user's
+    # default rule that catches regular traffic.
+    set `allocate_resources $ipfw_rulemin $(($ipfw_rulemax - 1))`
+    ipfw_rule=$1; pipe_index=$2
+    [ $ipfw_rule = 0 ] && echo 0 && return 0   # no resources
+
+    # If this is a new block, add the slice to the lookup table
+    # and put a default rule at the end of the block.
+    if [ "$TEST" = "0" -a $new_block = 1 ] ; then
+       ${IPFW} table $SLICE_TABLE add ${slice_id} ${ipfw_rulemin} > /dev/null
+       ${IPFW} add ${ipfw_rulemax} allow all from any to any > /dev/null
+       ( echo "${slice_id} BLOCK ${block_n}" ) >> ${DBFILE}
+    fi
+
+    echo $ipfw_rule $pipe_index "1"
+    return 0
+}
+
+#
 # called with the database file as input
 # compare the tuple <slice_id type arg> with
 # the current firewall configuration. The database contains
-#      slice_id type arg rule_base pipe_base timeout
-# On match returns <rule_base pipe_base timeout>
+#      slice_id type arg ipfw_rule pipe_index timeout
+# On match returns <ipfw_rule pipe_index timeout>
 # On non match returns 0 0 0
-find_rule() { # $1 slice_id $2 type $3 arg
+# no echo inside
+find_rule() { # slice_id type arg
     local ret
-    ret=`grep -- "^$1 $2 $3 " $DBFILE`
+    ret=`grep -i -- "^$1 $2 $3 " $DBFILE | grep -v BLOCK`
 
     [ x"$ret" = x ] && echo "0 0 0 " && return # nothing found
     # ignore multiple matches. If the db is corrupt we are
@@ -302,7 +347,7 @@ find_rule() { # $1 slice_id $2 type $3 arg
     echo "$4 $5 $6"
 }
 
-
+#
 # Find a hole in a list of numbers within a range (boundaries included)
 # The input is passed as a sorted list of numbers on stdin.
 # Return a "0" rule if there is no rule free
@@ -317,10 +362,12 @@ find_hole() {  # min max
     echo $cand
 }
 
+# XXX despite the name this does not allocate but only finds holes.
 # returns a free rule and pipe base for client|server|service
 # within a block
 # Returns r=0 if there are no resources available
-allocate_resources() {
+# no echo inside
+allocate_resources() { # ipfw_minrule ipfw_maxrule
     local p r
     # remove comments, extract field, sort
     p=`grep -v '^#' $DBFILE | grep -v BLOCK | awk '{print $5}' | sort -n | \
@@ -331,7 +378,40 @@ allocate_resources() {
     echo $r $p
 }
 
+
+# Returns the index of a free block
+# Returns 0 if there are no resources available
+# no debug inside
+find_free_block() {
+    b=`grep -v '^#' $DBFILE | grep BLOCK | awk '{print $3}' | sort -n | \
+       find_hole $BLOCK_MIN $BLOCK_MAX`
+    echo $b
+}
+
+# parse the ipfw database and remove expired rules
 #
+# Each timeout value stored in the database is compared against
+# the current time.  If the timeout is older than current,
+# the rules and related pipes will be deleted.
+kill_expired() { # slice_id type arg
+    local match timeout
+
+    # if there is no database file exit
+    [ ! -f ${DBFILE} ] && return 0
+
+    # Get the current time
+    now=`date -u +%s`
+
+    cp ${DBFILE} ${DBFILE}.kill
+    cat ${DBFILE}.kill | grep -v BLOCK |
+    while read line; do
+       match=`echo $line|cut -d " " -f 1-3`
+       timeout=`echo $line|cut -d " " -f 6`
+       [ $now -gt $timeout ] && do_delete 1 $match
+    done
+    rm ${DBFILE}.kill
+}
+
 # execute functions from root context
 # can be used from root context as follow:
 # echo "super $command $args" | /vsys/ipfw-be 0
@@ -343,56 +423,31 @@ do_super() { # $arguments...
        dbcleanup)
            clean_db; return 0
            ;;
+       killexpired)
+           kill_expired; return 0
+           ;;
        *)
            abort "Invalid super command"
            ;;
        esac
 }
 
-#
-# show ipfw rules and pipes filtering on slice_id
-# If the first argument is 0 filter rules,
-# otherwise filter pipes
-do_show() { #$1 show rules or pipes
-    local list ipfw_list
-
-    if [ $1 == "0" ]; then
-       list=`grep "^$SLICE_ID " $DBFILE | grep -v BLOCK | cut -d " " -f 4`
-       for i in $list; do ipfw_list="$ipfw_list $(($i + 39999))"; done
-       [ -n "${ipfw_list}" ] && ${IPFW} show $ipfw_list
-    else
-       # ipfw pipe show does not selectively filter pipes
-       # XXX so leave this code commented and show all pipes
-       #list=`grep "^$SLICE_ID " $DBFILE | grep -v BLOCK | cut -d " " -f 5`
-       # the pipe list is build adding two ipfw pipes for each pipe index
-       #for i in $list; do ipfw_list="$ipfw_list $(($i + $i + 9998)) $(($i + $i + 9998 +1)) "; done
-       #[ -n "${ipfw_list}" ] && ${IPFW} pipe show $ipfw_list
-       ${IPFW} pipe show
-    fi
-}
-
-#
 # refresh the rule timeout
-do_refresh() { # type arg timeout
-    local pipe_in pipe_out pipe_base rule_base rule_nr
-    local type=$1 arg=$2 timeout=$3
+do_refresh() { # slice_id type arg timeout
+    local ipfw_pipe_in ipfw_pipe_out pipe_index 
+    local slice_id=$1 type=$2 arg=$3 timeout=$4
 
     debug "do_refresh type: <$type> arg: <$arg> timeout: <$timeout>"
     [ "${type}" = "BLOCK" ] && abort "BLOCK rule not valid"
     [ "${timeout}" = "" ] && abort "Missing args on 'refresh', expected on of {SERVICE|SERVER|CLIENT} port_number"
-    set `find_rule $SLICE_ID $type $arg`
-    rule_base=$1; pipe_base=$2
-    [ "${rule_base}" = "0" ] && debug "no rules found" && return 0             # no rules found
-
-    rule_nr=$(($rule_base + 39999))            # XXX only individual rules
-    pipe_in=$(($pipe_base + $pipe_base + 9998))
-    pipe_out=$(($pipe_in + 1))
-    debug "ipfw rule and pipes value: rule: <$rule_nr> pipe in: <$pipe_in> pipe_out: <$pipe_out>"
+    set `find_rule $slice_id $type $arg`
+    ipfw_rule=$1; pipe_index=$2
+    [ "${ipfw_rule}" = "0" ] && debug "no rules found" && return 0             # no rules found
 
     [ "$TEST" = "1" ] && return
     # update the database with the new timeout value
-    ( grep -v -- "^${SLICE_ID} ${type} ${arg} " $DBFILE;  \
-       echo "${SLICE_ID} ${type} ${arg} ${rule_base} ${pipe_base} ${timeout}" ) > ${DBFILE}.tmp
+    ( grep -iv -- "^${slice_id} ${type} ${arg} " $DBFILE;  \
+       echo "${slice_id} ${type} ${arg} ${ipfw_rule} ${pipe_index} ${timeout}" ) > ${DBFILE}.tmp
     mv ${DBFILE}.tmp ${DBFILE}
     echo "refreshed timeout for rule ${type} ${arg}"
 }
@@ -400,9 +455,10 @@ do_refresh() { # type arg timeout
 # process a request.
 # A request is made by a set of arguments formatted as follow:
 #
-# config {server|client|service} arg [-t timeout] PIPE_IN <pipe_conf> PIPE_OUT <pipe_conf>
+# config {server|client|service} arg [-t timeout] IN <pipe_conf> OUT <pipe_conf>
 # show {rules|pipes} [args]
 # delete type arg
+# refresh type arg [-t timeout]
 #
 # The timeout value is expressed as:
 # week, day, month or anything else accepted by the date command.
@@ -411,7 +467,6 @@ do_refresh() { # type arg timeout
 process() {
     local new_pipe=0
     local timeout TMP i rule_base pipe_base
-    local slicename=${SLICENAME}
     local cmd=$1 ; shift
     local debug_args="$*";
     local type=$1 ; shift
@@ -421,11 +476,12 @@ process() {
     # set the timeout value
     # if present, extract the '-t timeout' substring from the command line
     timeout=`echo ${args} | ${SED} ${SEDOPT} 's/(.+)( -t [a-zA-Z0-9]+ )(.*)/\2/'`
-    # if the '-t timeout' is specified, use the user define timeout value
+    # if the '-t timeout' is specified, use the timeout provided by the user
     if [ "${timeout}" != "${args}" ] ; then    # match
        # remove the '-t ' option
        timeout=`echo ${timeout} | ${SED} ${SEDOPT} 's/-t //'`
-       check_timeout ${timeout}        # abort on error
+       timeout=`check_timeout ${timeout}`
+       [ $timeout = 0 ] && abort "Date format $1 not valid"
        # clean the arguments
        args=`echo ${args} | ${SED} ${SEDOPT} 's/(.+)( -t [a-zA-Z0-9]+ )(.*)/\1 \3/'`
     else
@@ -433,24 +489,31 @@ process() {
        timeout=`date --date="1day" +%s`                # default to 1 day
     fi
 
+    # if the table rule is not present, add it
+    local table_rule=`${IPFW} show $S | grep "skipto tablearg" | grep "lookup jail $SLICE_TABLE"`
+    [ -z "$table_rule" ] && ipfw_init
+
     debug "Timeout $timeout"
     # Handle special requests: show and delete
     case x"$cmd" in 
     x"config") 
-       [ "$type" = "server" ] && do_config $timeout $type $args && return 0
-       [ "$type" = "client" ] && do_config $timeout $type $args && return 0
-       [ "$type" = "service" ] && do_config $timeout $type $args && return 0
-       abort "'config' should be followed by {server|client|service}"
+       case x"$type" in 
+               xserver|xSERVER|xclient|xCLIENT|xservice|xSERVICE)
+                       do_config $SLICE_ID $timeout $type $args && return 0
+               ;;
+       esac
+       abort "'config' should be followed by {CLIENT|SERVER|SERVICE}"
        ;;
     x"delete") 
-       do_delete ${SLICE_ID} $type $args
+       do_delete 1 $SLICE_ID $type $args
        ;;
     x"refresh") 
-       do_refresh $type $args $timeout && return 0
+       do_refresh $SLICE_ID $type $args $timeout && return 0
        ;;
     x"show")
-       [ "$type" = "rules" ] && do_show 0 && return 0
-       [ "$type" = "pipes" ] && do_show 1 && return 0
+       # XXX filter out sliver rules
+       [ "$type" = "rules" ] && ${IPFW} show && return 0
+       [ "$type" = "pipes" ] && ${IPFW} pipe show && return 0
        abort "'show' should be followed by {rules|pipes}"
        ;;
     x"super")
@@ -470,27 +533,30 @@ process() {
 # validate the timeout
 check_timeout() { # timeout
     local tt=`date --date="${1}" +%s`
-    [ "$?" != "0" ] && abort "Date format $1 not valid"
+    [ "$?" != "0" ] && echo 0 && return
+    echo $tt
 }
 
-do_config() { # timeout type arg PIPE_IN pipe_conf PIPE_OUT pipe_conf
+do_config() { # slice_id timeout type arg IN pipe_conf OUT pipe_conf
+    local slice_id=$1; shift
     local timeout=$1; shift
     local type=$1; shift
     local arg=$1; shift        # XXX addr not yet implemented
+    local p h;                 # port and optional hostname
 
-    [ "$1" != "PIPE_IN" ] && abort "Missing addr:port, or PIPE_IN requested"
+    [ "$1" != "IN" ] && abort "Missing addr:port, or IN requested"
     shift
 
     # read pipe in configuration
     i=""
-    while [ "$1" != "" -a "$1" != "PIPE_OUT" ] ; do
+    while [ "$1" != "" -a "$1" != "OUT" ] ; do
        i="$i $1"
        shift
     done
     CONFIG_PIPE_IN="$i"                # XXX local ?
     [ "$CONFIG_PIPE_IN" = "" ] && abort "Missing pipe in configuration"
 
-    [ "$1" != "PIPE_OUT" ] && abort "Missing pipe in configuration, or missing PIPE_OUT"
+    [ "$1" != "OUT" ] && abort "Missing pipe in configuration, or missing OUT"
     shift
 
     # read pipe out configuration
@@ -502,80 +568,46 @@ do_config() { # timeout type arg PIPE_IN pipe_conf PIPE_OUT pipe_conf
     CONFIG_PIPE_OUT="$i"       # XXX local ?
     [ "$CONFIG_PIPE_OUT" = "" ] && abort "Missing pipe out configuration"
 
+
+    # process the argument (port and hostname are separated by a @)
+    # split the argument, and prepare the remote host configuration string
+    p=`echo $arg | cut -s -d "@" -f1-` # empty it there is no separator
+    if [ "$p" = "" ] ; then
+       p=$arg
+    else
+       p=`echo $arg | cut -d "@" -f1`
+       h=`echo $arg | cut -d "@" -f2`
+    fi
+
+    # A port value is mandatory
+    [ "$p" = "" ] && abort "A port value is mandatory."
+
+    # SERVICE do not support remote hostname filtering
+    [ $type = "service" ] && [ "$h" != "" ] && \
+       abort "The service configuration do not support filtering remote hostnames."
+
     debug "Configuration Required:"
     debug "slice_id: $SLICE_ID"
     debug "type: $type"
-    debug "arg: $arg"
+    debug "full arg: $arg"
+    debug "mandatory port(s): $p optional hostname(s): $h"
     debug "timeout: $timeout"
-    debug "PIPE_IN: $CONFIG_PIPE_IN"
-    debug "PIPE_OUT: $CONFIG_PIPE_OUT"
+    debug "IN: $CONFIG_PIPE_IN"
+    debug "OUT: $CONFIG_PIPE_OUT"
     debug "-----------------------"
 
     # check if the link is already configured
-    debug "Search for ${SLICE_ID} ${type} ${arg}"
-
-    set `find_rule ${SLICE_ID} ${type} ${arg}`
-    local rule_base=$1
-    local pipe_base=$2
-    local new_pipe=0
-
-    if [ ! ${rule_base} = "0" ] ; then
-       debug "Rule found, just changing the pipe configuration"
-       add_rule $new_pipe $type $arg $rule_base $pipe_base $timeout
-       hook_call $type $port $rule_base $pipe_base $timeout
-       return 0; # link configured, exit
-    fi
+    debug "Search for slice_id: ${slice_id} type: ${type} port: ${arg}"
 
-    debug "link not found, search for a block already allocated to the user"
-
-    # Search if there is a block already allocated to the slice_id
-    set `find_block ${SLICE_ID}`
-    local block_n=$1
-    if [ ${block_n} = "0" ] ; then
-       debug "Block not found, allocate a new block"
-       # blocks are allocated in sequence, get the first free
-       block_n=`grep BLOCK $DBFILE | tail -1 | cut -d " " -f 3`
-       if [ -z $block_n ]; then
-               block_n=$(($BLOCK_MIN - 1))
-       fi
-       RULE_IN_MIN=$(($RULE_BASE + $(($M*$block_n))))
-       block_n=$(($block_n +1))
-       debug "Allocated new block $block_nr to user $SLICE_ID"
-       [ $block_n -gt $BLOCK_MAX ] && abort "no block resources available"
-
-       # add the rule into the firewall table.
-       # note that the rule number into the table are not database number,
-       # so we need to compute the firewall number before the table insertion
-       local ipfw_rule_nr=$(($RULE_IN_MIN + 39999))  # XXX formula for individual rules
-       debug "Configuring table: <${IPFW_CHECK} table $SLICE_TABLE add ${SLICE_ID} ${ipfw_rule_nr}>"
-       ${IPFW_CHECK} table $SLICE_TABLE add ${SLICE_ID} ${rule_nr} > /dev/null || \
-               abort "ipfw syntax error $rule_out"
-       ${IPFW} table $SLICE_TABLE add ${SLICE_ID} ${ipfw_rule_nr} > /dev/null
-
-       [ "$TEST" = "1" ] && return
-       # add the block declaration to the database
-       ( grep -v -- "^${SLICE_ID} BLOCK " $DBFILE;  \
-               echo "${SLICE_ID} BLOCK ${block_n}" ) > ${DBFILE}.tmp
-       mv ${DBFILE}.tmp ${DBFILE}
-    else
-       debug "Block $block_n found for user $SLICE_ID"
-    fi
-
-    RULE_IN_MAX=$(($RULE_BASE + $(($M * $block_n))))
-    RULE_IN_MIN=$(($RULE_IN_MAX - $M))
-    debug  "Block $block_n, with rules <${RULE_IN_MIN}:${RULE_IN_MAX}>" 
-    debug  "Corresponding to ipfw rules <$(($RULE_IN_MIN + 39999)):$(($RULE_IN_MAX + 39999))>"
-    debug "where the last rule number belongs to the next slice."
-
-    new_pipe=1
-    set `allocate_resources $RULE_IN_MIN $RULE_IN_MAX`
-    rule_base=$1; pipe_base=$2
+    set `find_allocate ${slice_id} ${type} ${arg}`
+    local ipfw_rule=$1 pipe_index=$2 new_rule=$3
 
-    [ $rule_base = 0 ] && abort "no resources available"
-    debug "found free resources rule: $rule_base pipe: $pipe_base"
+    [ ${ipfw_rule} = 0 ] && abort "No resources available"
+    debug "Found or allocated resources ipfw_rule: ${ipfw_rule} and pipe_index: ${pipe_index}"
 
-    add_rule $new_pipe $type $arg $rule_base $pipe_base $timeout
+    add_rule $slice_id $new_rule $type $arg $ipfw_rule $pipe_index $timeout
     hook_call $type $port $rule_base $pipe_base $timeout
+    return 0; # link configured, exit
 }
 
 #
@@ -598,6 +630,8 @@ release_lock() {
 #
 # initialize the firewall with PlanetLab default rules
 ipfw_init() {
+       ${IPFW} -q delete $S
+       ${IPFW} -q delete $D
        ${IPFW} add $S skipto tablearg lookup jail $SLICE_TABLE
        ${IPFW} add $D allow all from any to any
 }
@@ -608,6 +642,7 @@ ipfw_init() {
 # slice_id type port rule_base pipe_base timeout
 hook_call() {
        if [ -n "${HOOK}" -a -x "${HOOK}" ]; then
+               debug "Calling the hook function."
                ${HOOK} ${SLICE_ID} "$*" &
        fi
 }
@@ -615,27 +650,36 @@ hook_call() {
 do_help() {
        cat << EOF
 Usage:
-        ./neconfig [SERVER|CLIENT|SERVICE] port [-t timeout]    \
-                PIPE_IN <pipe in configuration> PIPE_OUT <pipe out configuration>
-        ./netconfig show [rules|pipes]
-        ./netconfig delete [SERVER|CLIENT|SERVICE] port
-        ./netconfig refresh [-t timeout] [SERVER|CLIENT|SERVICE] port
-
-We assume three type of connections
-  SERVER we know the local port P, and do the
-       bind/listen/accept on the local socket.
-               pipe_in in dst-port P
-               pipe_out out src-port P
-
-  CLIENT we know the remote port P, and do a connect to it
-       (src and dst are swapped wrt the previous case)
-               pipe_in in src-port P
-               pipe_out out dst-port P
-
-  SERVICE we run a server on local port P, and also connect
-       from local clients to remote servers on port P.
-               pipe_in in { dst-port P or src-port P }
-               pipe_out out { src-port P or dst-port P }
+        ./neconfig {CLIENT|SERVER|SERVICE} arg [-t timeout]    \
+                IN <pipe in configuration> OUT <pipe out configuration>
+        ./netconfig show {rules|pipes}
+        ./netconfig delete {CLIENT|SERVER|SERVICE} arg
+        ./netconfig refresh [-t timeout] {CLIENT|SERVER|SERVICE} arg
+
+We support three modes of operation:
+
+  CLIENT programs on the node connect to remote ports
+       and/or addresses. Emulation intercepts traffic
+       involving those ports/addresses
+
+  SERVER programs on the node listen on specific ports.
+       Emulation intercepts traffic on those ports,
+       optionally limited to specific client addresses.
+
+  SERVICE the node runs both clients and servers,
+       we can only specify the ports on which emulation
+       is configured.
+
+  'arg' has the form PORTLIST[@ADDRLIST], where ADDRLIST is
+  optional and only supported for CLIENT and SERVER modes.
+  PORTLIST and ADDRLIST can be specified as any valid port
+  or address specifier in ipfw, e.g.
+    - a single value           443 or 10.20.30.40/24
+    - a comma-separated list   1111,2222,3333 1.2.3.4,5.6.7.8
+    - a range                  1111-2222 (only for ports)
+  Addresses can also be specified as symbolic hostnames, and
+  they are resolved when the rule is installed.
+  Note that they always indicate the remote endpoint.
 
   On a given port a user can have one CLIENT and/or one SERVER
   configuration or one SERVICE configuration.
@@ -644,13 +688,15 @@ We assume three type of connections
   When a CLIENT or SERVER configuration is installed any existing
   SERVICE configuration on the same port is removed.
 
-The pipe configuration, both for the upstream and downstream link,
-follow the dummynet syntax. A quick and not exaustive example
+The pipe's configuration, both for the upstream and downstream link,
+follows the dummynet syntax. A quick and not exaustive example
 of the parameters that can be used to configure the delay,
 the bandwidth and the packet loss rate for a link follow:
 
-        PIPE_IN|PIPE_OUT delay 100ms bw 1Mbit/s plr 0.1
+        IN|OUT delay 100ms bw 1Mbit/s plr 0.1
 
+The profile file, if present, should be located into the sliver's
+root directory.
 The full documentation is on the manpage[1].
 
 The timeout value follow the linux 'date' command format[2]
@@ -665,15 +711,45 @@ and can be specified as follow:
 EOF
 }
 
-# ALLOCATION OF RULES AND PIPES
-# The ruleset is composed by different sections, as follow:
-# - a first set of rules is reserved and is configurable by
+#--- DOCUMENTATION AND INTERNAL ARCHITECTURE ---
+#
+# When a user configures an emulated link, we need to allocate
+# two pipes and one ipfw rule number to store the parameters.
+# Reconfigurations of existing links reuse the previous resources.
+# We keep track of all resources (pipes, rules and blocks of rules)
+# in a database stored in a text file, see DATABASE FORMAT below.
+#
+# Pipes are allocated in pairs. In the database each pair is numbered
+# from PIPE_MIN to PIPE_MAX. The actual pipe numbers for each pair are
+#
+#      ipfw_pipein = IPFW_PIPE_MIN + 2*(pipe_index-1)
+#      ipfw_pipeout = ipfw_pipein + 1
+#
+# The rules number is allocated within a block of M consecutive rules
+# for each slice. The block is allocated at the first configuration
+# of an emulated link, and deallocated when the last link is removed.
+# In the database, blocks are numbered from BLOCK_MIN to BLOCK_MAX,
+# and the range of rules for a given block_index is
+#
+#      ipfw_min_rule = RULE_BASE
+#      ipfw_max_rule = RULE_BASE + ((M-1)*block_index) -1
+#
+# All lookups, and the block allocation, are done in find_allocate().
+# The rule_number and pipe_index are written in the database
+# by add_rule() after checking the correctness of the request.
+#
+#
+#--- RULESET STRUCTURE ---
+# The ruleset is made of different sections, as follows:
+# - an initial block of rules, reserved and configurable by
 #   the root context only;
-# - the skipto rule (S), used to optimize the slice rule search;
-# - a second block of reserved rules;
-# - a default (D) rule for the generic configuration;
-# - the slice reserved rules, a block of M rules for each slice;
-# - the firewall default rule.
+# - a skipto rule (S), used to jump directly to the block
+#   associated with a given slice;
+# - a second block of reserved rules, to catch remaining traffic.
+#   This ends with rule number D which is an 'accept all';
+# - after D, we have a block of M rule numbers for each slice.
+#   Each of these blocks ends with an 'accept all' rule;
+# - finally, rule 65535 is the firewall's default rule.
 #
 # To summarize:
 #      1...S-1 first block of reserved rules
@@ -685,50 +761,48 @@ EOF
 #      RULE_BASE+M <block of M entry for second user ...>
 #      ...
 #
-# Out of 64k rules, we allocate a block of M=50 consecutive
-# rules to each slice using emulation. Within this block,
-# each configuration uses one rule number and two pipes.
-#
-# Pipes are allocated starting from PIPE_BASE, a couple
-# of pipes for each configuration.
+#--- DATABASE FORMAT ---
+# The database is stored in a text file, and contains one record per
+# line with the following structure
 #
-# DATABASE FORMAT
-# The database is stored on a file, and contains
-# one line per record with this general structure
 #      XID     TYPE    arg1    arg2    ...
-# whitespace separates the fields. arg1, arg2, ...
-# have different meaning depending on the type.
+#
+# Whitespace separates the fields. arg1, arg2, ... have different
+# meaning depending on the TYPE. XID is the slice ID.
 #
 # In the database we have the following records:
-# - one entry for each slice that has active emulation entries.
-#   For each of these slices we reserve a block of M ipfw rules
-#   starting at some RULE_BASE rule number.
-#   The database entry for this info has the form
-#      XID     BLOCK   block_number
-#   where blocks are numbered sequentially from 1.
-#   The actual rule number is RULE_BASE + M*(block_number)
-#   (we don't care if we waste some rules)
+# - one entry of type BLOCK for each slice with configured links.
+#   This entry represents the block_index of the block of M ipfw
+#   rules allocated to the slice, as follows:
 #
-# - one entry for each predefined config (CLIENT, SERVER, SERVICE).
+#      XID     BLOCK   block_index
+#   (BLOCK_MIN <= block_index <= BLOCK_MAX)
+#
+# - one entry for each link (CLIENT, SERVER, SERVICE).
 #   The database entry for this info has the form
-#      XID     {CLIENT|SERVER|SERVICE} arg     rule_nr pipe_index
-#   rule_nr is the absolute rule number for this configuration
-#   (it must be within the block of M rules allocated to the slice)
-#   pipe_index is the index of the couple of pipes used for the
-#   configuration. pipe_index starts from 1.
 #
+#      XID {CLIENT|SERVER|SERVICE} arg ipfw_rule pipe_index timeout
+# 
+#   'TYPE' reflects the configuration mode;
+#   'arg' is PORTLIST@ADDRLIST and is used as a search key together
+#      with the XID and TYPE;
+#   'ipfw_rule' is the unique ipfw rule number used for this
+#      emulated link. It must be within the block of M rule numbers
+#      allocated to the slice;
+#   'pipe_index' is the index of the pair of pipes used for the
+#      configuration;
 
 #-- main starts here
 debug "--- $0 START for $SLICENAME ---"
 
-# If the db does not exist, create it and we clean rules and pipes
+# If the db does not exist, create it and clean rules and pipes
 [ ! -e ${DBFILE} ] && clean_db
 
 # A request to the vsys backend is composed by a single line of input
 read REQ                       # read one line, ignore the rest
 set_verbose ${REQ}             # use inital -v if present
 set_test ${REQ}                # use inital -t if present
-REQ="`filter ${REQ}`"  # remove -v and -t and invalid chars
+REQ="`filter ${REQ}`"  # remove -v and -q and invalid chars
 debug "--- processing <${REQ}>"
 acquire_lock                   # critical section
 process ${REQ}