Munin contributed plugins

#!/usr/bin/env python3

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.

SPDX-License-Identifier: GPL-2.0-only

=head1 MAGIC MARKERS

 #%# family=snmpauto

 #%# capabilities=snmpconf

    logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(levelname)-7s %(message)s')

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

    host = match.group(1)

    request = match.group(2)

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

    if match is not None:

        host = match.group(1)

        port = match.group(2)

except Exception:

    pass

# jnxJsSPUMonitoringIndex = jnxJsSPUMonitoringObjectsTable + '.1.1'

# jnxJsSPUMonitoringFPCIndex = jnxJsSPUMonitoringObjectsTable + '.1.2'

# jnxJsSPUMonitoringSPUIndex = jnxJsSPUMonitoringObjectsTable + '.1.3'

# jnxJsSPUMonitoringCPUUsage = jnxJsSPUMonitoringObjectsTable + '.1.4'

# jnxJsSPUMonitoringMemoryUsage = jnxJsSPUMonitoringObjectsTable + '.1.5'

# jnxJsSPUMonitoringNodeIndex = jnxJsSPUMonitoringObjectsTable + '.1.10'

class JunOSSnmpClient(object):

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

        self.hostname = host

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

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

        self.gen = cmdgen.CommandGenerator()

    def get_data(self):

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

            self.auth,

            self.transport,

            0, 10,

            jnxJsSPUMonitoringObjectsTable)

        # the line below is only available with pysnmp >= 4.2.4 (?) and broken anyway

        # ignoreNonIncreasingOids=True)

        if errorIndication:

            logging.error("SNMP bulkCmd for devices failed: %s, %s, %s"

                          % (errorIndication, errorStatus, errorIndex))

            return {}

        devices = {}

        values = {}

        for row in varBindTable:

            for name, value in row:

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

                    continue

                oid = str(name)

                nodeid = oid[oid.rindex('.'):]

                idx = oid[len(jnxJsSPUMonitoringObjectsTable) + 3:]

                idx = idx[:idx.index('.')]

                if oid.startswith(jnxJsSPUMonitoringNodeDescr):

                    devices[nodeid] = str(value)

                    continue

                values[oid] = int(value)

        return devices, values

    def print_config(self):

        devices, data = self.get_data()

        data_def = [

            ('flow', self.hostname, 'Flow sessions', '--base 1000', '# of sessions',

             jnxJsSPUMonitoringMaxFlowSession),

            ('cp', self.hostname, 'Central point sessions', '--base 1000', '# of sessions',

             jnxJsSPUMonitoringMaxCPSession),

        ]

        for datarow, hostname, title, args, vlabel, max_prefix in data_def:

            print("""multigraph juniper_{datarow}

graph_info {title}""".format(datarow=datarow, hostname=hostname, title=title, args=args,

                             vlabel=vlabel))

            for suffix, node in devices.items():

                ident = "%s_%s" % (datarow, node)

                print("""{label}.info {title} on {node}

{label}.max {max}""".format(title=title, label=ident, node=node,

                            max=data.get(max_prefix + suffix)))

            for suffix, node in devices.items():

                print("""

""".format(datarow=datarow, hostname=hostname, title=title, args=args, vlabel=vlabel, node=node))

    def execute(self):

        devices, data = self.get_data()

        print("multigraph juniper_flow")

        for suffix, node in devices.items():

            print("flow_%s.value %s"

                  % (node, data.get(jnxJsSPUMonitoringCurrentFlowSession + suffix, 0)))

        for suffix, node in devices.items():

            print("multigraph juniper_flow.%s" % node)

            print("flowV4.value %s" % data.get(jnxJsSPUMonitoringFlowSessIPv4 + suffix, 0))

            print("flowV6.value %s" % data.get(jnxJsSPUMonitoringFlowSessIPv6 + suffix, 0))

            print("flowCurrent.value %s"

                  % data.get(jnxJsSPUMonitoringCurrentFlowSession + suffix, 0))

            print("flowMax.value %s" % data.get(jnxJsSPUMonitoringMaxFlowSession + suffix, 0))

        print("multigraph juniper_cp")

        for suffix, node in devices.items():

            print("cp_%s.value %s"

                  % (node, data.get(jnxJsSPUMonitoringCurrentCPSession + suffix, 0)))

        for suffix, node in devices.items():

            print("multigraph juniper_cp.%s" % node)

            print("cpV4.value %s" % data.get(jnxJsSPUMonitoringCPSessIPv4 + suffix, 0))

            print("cpV6.value %s" % data.get(jnxJsSPUMonitoringCPSessIPv6 + suffix, 0))

            print("cpCurrent.value %s" % data.get(jnxJsSPUMonitoringCurrentCPSession + suffix, 0))

            print("cpMax.value %s" % data.get(jnxJsSPUMonitoringMaxCPSession + suffix, 0))

    print("require %s" % (jnxJsSPUMonitoringObjectsTable, ))

    sys.exit(0)

    if not (host and port and community):

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

              % (host, port, community), file=sys.stderr)

        sys.exit(1)

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

        c.print_config()

    else:

        c.execute()