initial checkin

vpp-snmp-agent.py

@@ -0,0 +1,317 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from vppstats import VPPStats
+import time
+import pyagentx
+import logging
+import threading
+
+vppstat_lastread = 0
+vppstat_ifstat = {
+    'ifNames': [],
+    'ifHCInOctets': [],
+    'ifHCInUcastPkts': [],
+    'ifHCInMulticastPkts': [],
+    'ifHCInBroadcastPkts': [],
+    'ifHCOutOctets': [],
+    'ifHCOutUcastPkts': [],
+    'ifHCOutMulticastPkts': [],
+    'ifHCOutBroadcastPkts': [],
+    'ifHighSpeed': []
+}
+
+vppstat_lock = threading.Lock()
+
+
+class NullHandler(logging.Handler):
+    def emit(self, record):
+        pass
+
+
+logger = logging.getLogger('pyagentx.vpp')
+logger.addHandler(NullHandler())
+
+
+def vppstat_update():
+    global vppstat_lastread, vppstat_lock, vppstat_ifstat, logger
+
+    vppstat_lock.acquire()
+    try:
+        if time.time() - vppstat_lastread < 9.0:
+            logger.debug("Skipping, cache still fresh")
+            vppstat_lock.release()
+            return
+
+        logger.info("Fetching interface data from VPP")
+        vppstat = VPPStats(socketname='/run/vpp/stats.sock', timeout=2)
+        vppstat.connect()
+        vppstat_ifstat['ifNames'] = vppstat['/if/names']
+        vppstat_ifstat['ifHCInOctets'].clear()
+        vppstat_ifstat['ifHCInUcastPkts'].clear()
+        vppstat_ifstat['ifHCInMulticastPkts'].clear()
+        vppstat_ifstat['ifHCInBroadcastPkts'].clear()
+        vppstat_ifstat['ifHCOutOctets'].clear()
+        vppstat_ifstat['ifHCOutUcastPkts'].clear()
+        vppstat_ifstat['ifHCOutMulticastPkts'].clear()
+        vppstat_ifstat['ifHCOutBroadcastPkts'].clear()
+        vppstat_ifstat['ifHighSpeed'].clear()
+
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            vppstat_ifstat['ifHCInOctets'].append(
+                vppstat['/if/rx'][:, i].sum_octets())
+            vppstat_ifstat['ifHCInUcastPkts'].append(
+                vppstat['/if/rx'][:, i].sum_packets())
+            vppstat_ifstat['ifHCInMulticastPkts'].append(
+                vppstat['/if/rx-multicast'][:, i].sum_packets())
+            vppstat_ifstat['ifHCInBroadcastPkts'].append(
+                vppstat['/if/rx-broadcast'][:, i].sum_packets())
+
+            vppstat_ifstat['ifHCOutOctets'].append(
+                vppstat['/if/tx'][:, i].sum_octets())
+            vppstat_ifstat['ifHCOutUcastPkts'].append(
+                vppstat['/if/tx'][:, i].sum_packets())
+            vppstat_ifstat['ifHCOutMulticastPkts'].append(
+                vppstat['/if/tx-multicast'][:, i].sum_packets())
+            vppstat_ifstat['ifHCOutBroadcastPkts'].append(
+                vppstat['/if/tx-broadcast'][:, i].sum_packets())
+
+            # TODO(pim) retrieve from vpp_papi
+            # IF-MIB::ifHighSpeed.2 = Gauge32: 1000
+            vppstat_ifstat['ifHighSpeed'].append(1000)
+
+
+# TODO(pim) retrieve from linux namespace if present
+# IF-MIB::ifAlias.2 = STRING: Infra: nikhef-core-1.nl.switch.coloclue.net e1/34
+
+# Initializing these to defaults:
+# IF-MIB::ifPromiscuousMode.2 = INTEGER: false(2)
+# IF-MIB::ifConnectorPresent.2 = INTEGER: true(1)
+# IF-MIB::ifCounterDiscontinuityTime.2 = Timeticks: (0) 0:00:00.00
+
+        logger.info("Fetched data for %u interfaces" %
+                    len(vppstat_ifstat['ifNames']))
+        vppstat_lastread = time.time()
+        vppstat.disconnect()
+    except Exception as e:
+        logger.error("Error occured, releasing lock: ", e)
+    vppstat_lock.release()
+    return
+
+
+class ifName(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_OCTETSTRING(str(i + 1), vppstat_ifstat['ifNames'][i])
+
+
+class ifAlias(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_OCTETSTRING(str(i + 1), vppstat_ifstat['ifNames'][i])
+
+
+class ifInMulticastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER32(str(i + 1),
+                               vppstat_ifstat['ifHCInMulticastPkts'][i])
+
+
+class ifInBroadcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER32(str(i + 1),
+                               vppstat_ifstat['ifHCInBroadcastPkts'][i])
+
+
+class ifOutMulticastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER32(str(i + 1),
+                               vppstat_ifstat['ifHCOutMulticastPkts'][i])
+
+
+class ifOutBroadcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER32(str(i + 1),
+                               vppstat_ifstat['ifHCOutBroadcastPkts'][i])
+
+
+class ifHCInOctets(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1), vppstat_ifstat['ifHCInOctets'][i])
+
+
+class ifHCInUcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCInUcastPkts'][i])
+
+
+class ifHCInMulticastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCInMulticastPkts'][i])
+
+
+class ifHCInBroadcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCInBroadcastPkts'][i])
+
+
+class ifHCOutOctets(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1), vppstat_ifstat['ifHCOutOctets'][i])
+
+
+class ifHCOutUcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCOutUcastPkts'][i])
+
+
+class ifHCOutMulticastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCOutMulticastPkts'][i])
+
+
+class ifHCOutBroadcastPkts(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_COUNTER64(str(i + 1),
+                               vppstat_ifstat['ifHCOutBroadcastPkts'][i])
+
+
+class ifHighSpeed(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            self.set_GAUGE32(str(i + 1), vppstat_ifstat['ifHighSpeed'][i])
+
+
+class ifPromiscuousMode(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            # Hardcode to false(2)
+            self.set_INTEGER(str(i + 1), 2)
+
+
+class ifConnectorPresent(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            # Hardcode to true(1)
+            self.set_INTEGER(str(i + 1), 1)
+
+
+class ifCounterDiscontinuityTime(pyagentx.Updater):
+    def update(self):
+        global vppstat_ifstat
+
+        vppstat_update()
+        for i in range(len(vppstat_ifstat['ifNames'])):
+            # Hardcode to Timeticks: (0) 0:00:00.00
+            self.set_TIMETICKS(str(i + 1), 0)
+
+
+class MyAgent(pyagentx.Agent):
+    def setup(self):
+        self.register('1.3.6.1.2.1.31.1.1.1.1', ifName)
+        self.register('1.3.6.1.2.1.31.1.1.1.2', ifInMulticastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.3', ifInBroadcastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.4', ifOutMulticastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.5', ifOutBroadcastPkts)
+
+        self.register('1.3.6.1.2.1.31.1.1.1.6', ifHCInOctets)
+        self.register('1.3.6.1.2.1.31.1.1.1.7', ifHCInUcastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.8', ifHCInMulticastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.9', ifHCInBroadcastPkts)
+
+        self.register('1.3.6.1.2.1.31.1.1.1.10', ifHCOutOctets)
+        self.register('1.3.6.1.2.1.31.1.1.1.11', ifHCOutUcastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.12', ifHCOutMulticastPkts)
+        self.register('1.3.6.1.2.1.31.1.1.1.13', ifHCOutBroadcastPkts)
+
+        self.register('1.3.6.1.2.1.31.1.1.1.14', ifHighSpeed)
+        self.register('1.3.6.1.2.1.31.1.1.1.15', ifPromiscuousMode)
+        self.register('1.3.6.1.2.1.31.1.1.1.16', ifConnectorPresent)
+        self.register('1.3.6.1.2.1.31.1.1.1.18', ifCounterDiscontinuityTime)
+
+        self.register('1.3.6.1.2.1.31.1.1.1.17', ifAlias)
+
+
+def main():
+    pyagentx.setup_logging()
+
+    try:
+        a = MyAgent()
+        a.start()
+    except Exception as e:
+        print("Unhandled exception:", e)
+        a.stop()
+    except KeyboardInterrupt:
+        a.stop()
+
+
+if __name__ == "__main__":
+    main()