Munin contributed plugins

#!/usr/bin/python

# Copyright (C) 2014 Johann Schmitz <johann@j-schmitz.net>

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU Library General Public License as published by

# the Free Software Foundation; version 2 only

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Library General Public License for more details.

#

# You should have received a copy of the GNU Library General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

#

"""

=head1 NAME

snmp__juniper - Health monitoring plugin for Juniper firewalls.

=head1 CONFIGURATION

Make sure your Juniper device is accessible via SNMP (e.g. via snmpwalk) and the munin-node

has been configured correctly

=head1 MAGIC MARKERS

	#%# family=contrib

	#%# capabilities=

=head1 VERSION

0.0.1

=head1 BUGS

Open a ticket at https://github.com/ercpe/contrib if you find one.

=head1 AUTHOR

Johann Schmitz <johann@j-schmitz.net>

=head1 LICENSE

GPLv2

=cut

"""

import re

import sys

import os

import logging

from pysnmp.entity.rfc3413.oneliner import cmdgen

host = None

port = 161

community = os.getenv('community', None)

try:

	match = re.search("^(?:|.*\/)snmp_([^_]+)_juniper$", sys.argv[0])

	host = match.group(1)

	request = match.group(2)

	match = re.search("^([^:]+):(\d+)$", host)

	if match is not None:

		host = match.group(1)

		port = match.group(2)

except:

	pass

if not (host and port and community):

	print "# Bad configuration. Cannot run with Host=%s, port=%s and community=%s" % (host, port, community)

	sys.exit(1)

jnxOperatingTable = '1.3.6.1.4.1.2636.3.1.13.1.5'

jnxOperatingTemp = '1.3.6.1.4.1.2636.3.1.13.1.7'

jnxOperatingCPU = '1.3.6.1.4.1.2636.3.1.13.1.8'

jnxOperatingBuffer = '1.3.6.1.4.1.2636.3.1.13.1.11'

class JunOSSnmpClient(object):

	def __init__(self, host, port, community):

		self.transport = cmdgen.UdpTransportTarget((host, int(port)))

		self.auth = cmdgen.CommunityData('test-agent', community)

		self.gen = cmdgen.CommandGenerator()

	def get_devices(self):

		errorIndication, errorStatus, errorIndex, varBindTable = self.gen.bulkCmd(

			self.auth,

			self.transport,

			0, 20,

			jnxOperatingTable)

#			ignoreNonIncreasingOids=True)

		if errorIndication:

			logging.error("SNMP bulkCmd for devices failed: %s, %s, %s" % (errorIndication, errorStatus, errorIndex))

			return {}

		devices = {}

		for row in varBindTable:

			for name, value in row:

				if not str(name).startswith(jnxOperatingTable):

					continue

				if str(value).endswith(' Routing Engine'):

					devices[str(name)[len(jnxOperatingTable):]] = re.sub("[^\w]", '_', str(value).replace(' Routing Engine', ''))

		return devices

	def get_one(self, prefix_oid, suffix):

		oid = prefix_oid + suffix

		errorIndication, errorStatus, errorIndex, varBindTable = self.gen.getCmd(

			self.auth,

			self.transport,

			oid)

		if errorIndication:

			logging.error("SNMP getCmd for %s failed: %s, %s, %s" % (oid, errorIndication, errorStatus, errorIndex))

			return None

		return int(varBindTable[0][1])

	def get_data(self):

		devs = self.get_devices()

		return {

			'temp': dict([(name, self.get_one(jnxOperatingTemp, suffix)) for suffix, name in devs.iteritems()]),

			'cpu': dict([(name, self.get_one(jnxOperatingCPU, suffix)) for suffix, name in devs.iteritems()]),

			'buffer': dict([(name, self.get_one(jnxOperatingBuffer, suffix)) for suffix, name in devs.iteritems()]),

		}

	def print_config(self):

		devices = self.get_devices()

		tpl = """multigraph juniper_temperature

graph_title $host system temperature

graph_vlabel System temperature in C per ${graph_period}

graph_category system

graph_info System temperature for ${host}

%s"""

		s = ""

		for suffix, node in devices.iteritems():

			ident = "temp_%s" % node

			s += """{label}.info System temperature on {node}

{label}.label {label}

{label}.type GAUGE

{label}.min 0

""".format(label=ident, node=node)

		print tpl % s

		tpl = """multigraph juniper_cpu

graph_title $host CPU usage

graph_vlabel CPU usage in %% per ${graph_period}

graph_category system

graph_info CPU usage for ${host}

%s"""

		s = ""

		for suffix, node in devices.iteritems():

			ident = "cpu_%s" % node

			s += """{label}.info CPU usage on {node}

{label}.label {label}

{label}.type GAUGE

{label}.min 0

{label}.max 100

""".format(label=ident, node=node)

		print tpl % s

		tpl = """multigraph juniper_buffer		

graph_title $host Buffer usage

graph_vlabel Buffer usage in %% per ${graph_period}

graph_category system

graph_info Buffer usage for ${host}

%s"""

		s = ""

		for suffix, node in devices.iteritems():

			ident = "buffer_%s" % node

			s += """{label}.info Buffer usage on {node}

{label}.label {label}

{label}.type GAUGE

{label}.min 0

{label}.max 100

""".format(label=ident, node=node)

		print tpl % s

	def execute(self):

		data = self.get_data()

#		import pprint

#		pprint.pprint(data)

		for pre, values in data.iteritems():

			for node, value in values.iteritems():

				print "%s_%s.value %s" % (pre, node, value)

def print_data():

	pass

c = JunOSSnmpClient(host, port, community)

if "config" in sys.argv[1:]:

	c.print_config()

#elif "snmpconf" in sys.argv[1:]:

#	#print "require 1.3.6.1.4.1.18928.1.2.2.1.8.1.1"

#	sys.exit(0)

else:

	c.execute()