2
0
mirror of https://github.com/munin-monitoring/contrib.git synced 2018-11-08 00:59:34 +01:00
contrib-munin/plugins/network/shorewall-accounting
2012-02-13 18:24:46 +01:00

132 lines
4.7 KiB
Python
Executable File

#!/usr/bin/python
# shorewall_accounting v1.2
#
# A munin plugin for tracking traffic as recorded by shorewall accounting rules.
# See "man shorewall-accounting" for all possible accounting rules.
# Basically this plugin examines the output of "shorewall -x show accounting".
# See http://atlee.ca/blog/2006/01/20/munin-shorewall/ for a description of
# the original script by Chris AtLee.
#
# Copyright 2010 Lars Kruse <devel@sumpralle.de>
# Copyright 2006 Chris AtLee <chris@atlee.ca>
#
# Original publication: http://atlee.ca/blog/2006/01/20/munin-shorewall/
# Released under the GPL v3
#
# You can use symlinks of this script according to the following pattern:
# shorewall_accounting_prot: use only rules with a specific protocol
# shorewall_accounting_in: use only rules with a specific input interface
# shorewall_accounting_out: use only rules with a specific output interface
# shorewall_accounting_source: use only rules with a specific source IP
# shorewall_accounting_destination: use only rules with a specific destination IP
# shorewall_accounting_details: use only rules with specific details (e.g. a port)
#
# Here "specific" means: non-default (e.g. a protocol was specified).
# Combinations are allowed:
# shorewall_accounting_prot_in_source: use only rules with a specific protocol, input interface and source IP
#
import sys
import os
import commands
import re
PLUGIN_BASE_NAME = "shorewall_accounting"
ACCOUNTING_LINE_EXP = re.compile(r"^\s*\d+\s+(\d+)\s+(?P<prot>\w+)\s+(?P<opt>[\w-]+)\s+(?P<in>[\w*]+)\s+(?P<out>[\w*]+)\s+(?P<source>[\w./+-]+)\s+(?P<destination>[\w./+-]+)\s*(?P<details>.*)\s*$")
KEY_ORDER = ["prot", "in", "out", "source", "destination", "details"]
FILTER_PATTERNS = {
"prot": r"^all$",
"in": r"^\*$",
"out": r"^\*$",
"source": r"^0\.0\.0\.0/0$",
"destination": r"^0\.0\.0\.0/0$",
"details": r"^$",
}
REPLACE_PATTERNS = {
"prot": ("^all$", "allProt"),
"in": (r"^\*$", "allIn"),
"out": (r"^\*$", "allOut"),
"source": (r"^0\.0\.0\.0/0", "allSrc"),
"destination": (r"^0\.0\.0\.0/0", "allDst"),
"details": (r"^multiport\s+", ""),
}
def get_accounting_rule_fieldname(regdict):
items = []
# filter and clean all requested keys
for key in KEY_ORDER:
raw = regdict[key]
pattern, replacement = REPLACE_PATTERNS[key]
value = re.sub(pattern, replacement, raw).strip()
if value:
items.append(value)
result = "_".join(items)
# clean the fieldname: http://munin-monitoring.org/wiki/notes_on_datasource_names
result = re.sub(r"^[^A-Za-z_]", "_", result)
result = re.sub(r"[^A-Za-z0-9_]", "_", result)
return result
def is_wanted(regdict, filter_list):
for item in filter_list:
# is the item empty?
if not regdict[item]:
return False
# is the default value (unfiltered) set?
if re.search(FILTER_PATTERNS[item], regdict[item]):
return False
return True
def get_bytes_by_chain(filter_list):
status, output = commands.getstatusoutput("shorewall -x show accounting")
if status != 0:
raise OSError("Error running command (%s)[%i]: %s" % (trafficCmd, status, output))
chains = {}
for line in output.splitlines():
m = ACCOUNTING_LINE_EXP.match(line)
if m is not None:
# check if this line was filtered
if not is_wanted(m.groupdict(), filter_list):
continue
target = get_accounting_rule_fieldname(m.groupdict())
bytes = int(m.group(1))
if target in chains:
chains[target] += bytes
else:
chains[target] = bytes
retval = []
names = chains.keys()
names.sort()
for name in names:
retval.append((name, chains[name]))
return retval
call_name = os.path.basename(sys.argv[0])
if call_name.startswith(PLUGIN_BASE_NAME):
suffix = call_name[len(PLUGIN_BASE_NAME):]
suffixes = suffix.split("_")
# use only suffixes that are listed in FILTER_PATTERNS
filter_list = [item for item in suffixes if item in FILTER_PATTERNS]
else:
filter_list = []
if len(sys.argv) > 1:
if sys.argv[1] == "autoconf":
print "yes"
sys.exit(0)
elif sys.argv[1] == "config":
print "graph_title Shorewall accounting"
print "graph_category network"
print "graph_vlabel bits per ${graph_period}"
for chain, bytes in get_bytes_by_chain(filter_list):
print "%s.min 0" % chain
print "%s.type DERIVE" % chain
print "%s.label %s" % (chain, chain)
print "%s.cdef %s,8,*" % (chain, chain)
sys.exit(0)
for chain, bytes in get_bytes_by_chain(filter_list):
print "%s.value %i" % (chain, bytes)