#
# Author : Stefan Schantl <stefan.schantl@ipfire.org>
#
-# Version : 01.00
+# Version : 01.03
#
# Notes :
#
eval $(/usr/local/bin/readhash /var/ipfire/suricata/settings)
eval $(/usr/local/bin/readhash /var/ipfire/ethernet/settings)
-# Name of the firewall chain.
-FW_CHAIN="IPS"
+# Name of the firewall chains.
+IPS_INPUT_CHAIN="IPS_INPUT"
+IPS_FORWARD_CHAIN="IPS_FORWARD"
+IPS_OUTPUT_CHAIN="IPS_OUTPUT"
# Optional options for the Netfilter queue.
NFQ_OPTS="--queue-bypass "
# Array containing the 4 possible network zones.
-network_zones=( red green blue orange )
+network_zones=( red green blue orange ovpn )
+
+# Array to store the network zones weather the IPS is enabled for.
+enabled_ips_zones=()
# Mark and Mask options.
MARK="0x70000000"
[ "$line" ] && [ -z "${line%processor*}" ] && ((CPUCOUNT++))
done </proc/cpuinfo
- echo $CPUCOUNT
+ # Limit to a maximum of 16 cores, because suricata does not support more than
+ # 16 netfilter queues at the moment.
+ if [ $CPUCOUNT -gt "16" ]; then
+ echo "16"
+ else
+ echo $CPUCOUNT
+ fi
}
-# Function to flush the firewall chain.
+# Function to flush the firewall chains.
function flush_fw_chain {
- # Call iptables and flush the chain
- iptables -F "$FW_CHAIN"
+ # Call iptables and flush the chains
+ iptables -F "$IPS_INPUT_CHAIN"
+ iptables -F "$IPS_FORWARD_CHAIN"
+ iptables -F "$IPS_OUTPUT_CHAIN"
}
# Function to create the firewall rules to pass the traffic to suricata.
function generate_fw_rules {
cpu_count=$(get_cpu_count)
- # Flush the firewall chain.
- flush_fw_chain
-
# Loop through the array of network zones.
for zone in "${network_zones[@]}"; do
# Convert zone into upper case.
if [ "$zone" == "red" ] && [ "$RED_TYPE" == "PPPOE" ]; then
# Set device name to ppp0.
network_device="ppp0"
+ elif [ "$zone" == "ovpn" ]; then
+ # Get all virtual net devices because the RW server and each
+ # N2N connection creates it's own tun device.
+ for virt_dev in /sys/devices/virtual/net/*; do
+ # Cut-off the directory.
+ dev="${virt_dev##*/}"
+
+ # Only process tun devices.
+ if [[ $dev =~ "tun" ]]; then
+ # Add the network device to the array of enabled zones.
+ enabled_ips_zones+=( "$dev" )
+ fi
+ done
+
+ # Process next zone.
+ continue
else
# Generate variable name which contains the device name.
zone_name="$zone_upper"
network_device=${!zone_name}
fi
- # Assign NFQ_OPTS
- NFQ_OPTIONS=$NFQ_OPTS
-
- # Check if there are multiple cpu cores available.
- if [ "$cpu_count" -gt "1" ]; then
- # Balance beetween all queues.
- NFQ_OPTIONS+="--queue-balance 0:$(($cpu_count-1))"
- NFQ_OPTIONS+=" --queue-cpu-fanout"
- else
- # Send all packets to queue 0.
- NFQ_OPTIONS+="--queue-num 0"
- fi
-
- # Create firewall rules to queue the traffic and pass to
- # the IDS.
- iptables -I "$FW_CHAIN" -i "$network_device" -m mark ! --mark "$MARK"/"$MASK" -j NFQUEUE $NFQ_OPTIONS
- iptables -I "$FW_CHAIN" -o "$network_device" -m mark ! --mark "$MARK"/"$MASK" -j NFQUEUE $NFQ_OPTIONS
+ # Add the network device to the array of enabled zones.
+ enabled_ips_zones+=( "$network_device" )
fi
done
- # Clear repeat bit, so that it does not confuse IPsec or QoS
- iptables -A "${FW_CHAIN}" -j MARK --set-xmark "0x0/${MASK}"
+ # Assign NFQ_OPTS
+ NFQ_OPTIONS=$NFQ_OPTS
+
+ # Check if there are multiple cpu cores available.
+ if [ "$cpu_count" -gt "1" ]; then
+ # Balance beetween all queues.
+ NFQ_OPTIONS+="--queue-balance 0:$(($cpu_count-1))"
+ NFQ_OPTIONS+=" --queue-cpu-fanout"
+ else
+ # Send all packets to queue 0.
+ NFQ_OPTIONS+="--queue-num 0"
+ fi
+
+ # Flush the firewall chains.
+ flush_fw_chain
+
+ # Check if the array of enabled_ips_zones contains any elements.
+ if [[ ${enabled_ips_zones[@]} ]]; then
+ # Loop through the array and create firewall rules.
+ for enabled_ips_zone in "${enabled_ips_zones[@]}"; do
+ # Create rules queue input and output related traffic and pass it to the IPS.
+ iptables -I "$IPS_INPUT_CHAIN" -i "$enabled_ips_zone" -m mark ! --mark "$MARK"/"$MASK" -j NFQUEUE $NFQ_OPTIONS
+ iptables -I "$IPS_OUTPUT_CHAIN" -o "$enabled_ips_zone" -m mark ! --mark "$MARK"/"$MASK" -j NFQUEUE $NFQ_OPTIONS
+
+ # Create rules which are required to handle forwarded traffic.
+ for enabled_ips_zone_forward in "${enabled_ips_zones[@]}"; do
+ iptables -I "$IPS_FORWARD_CHAIN" -i "$enabled_ips_zone" -o "$enabled_ips_zone_forward" -m mark ! --mark "$MARK"/"$MASK" -j NFQUEUE $NFQ_OPTIONS
+ done
+ done
+
+ # Clear repeat bit, so that it does not confuse IPsec or QoS
+ iptables -A "${IPS_INPUT_CHAIN}" -j MARK --set-xmark "0x0/${MASK}"
+ iptables -A "${IPS_FORWARD_CHAIN}" -j MARK --set-xmark "0x0/${MASK}"
+ iptables -A "${IPS_OUTPUT_CHAIN}" -j MARK --set-xmark "0x0/${MASK}"
+ fi
}
case "$1" in
cpu_count=$(get_cpu_count)
# Numer of NFQUES.
- NFQUEUES=
+ NFQUEUES="-q 0"
- for i in $(seq 0 $((cpu_count-1)) ); do
- NFQUEUES+="-q $i "
- done
+ if [ $cpu_count -gt "1" ]; then
+ NFQUEUES+=":$(($cpu_count-1))"
+ fi
# Check if the IDS should be started.
if [ "$ENABLE_IDS" == "on" ]; then
# Flush firewall chain.
flush_fw_chain
- # Remove suricata control socket.
+ # Sometimes suricata not correct shutdown. So killall.
+ killall -KILL /usr/bin/suricata 2>/dev/null
+
+ # Remove suricata control socket.
rm /var/run/suricata/* >/dev/null 2>/dev/null
+ # Trash remain pid file if still exists.
+ rm -f $PID_FILE >/dev/null 2>/dev/null
+
# Don't report returncode of rm if suricata was not started
exit 0
;;
projects / people / pmueller / ipfire-2.x.git / blobdiff