summaryrefslogtreecommitdiff
path: root/examples/v3arch/oneliner/agent/ntforg/inform-async-multiple-snmp-engines.py
blob: fcceaf0d52b495b1096c9d7140bba2918b0c42e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#
# Notification Originator
#
# Send SNMP notifications in behalf of multiple independend SNMP engines 
# using the following options:
#
# * with a single transport dispatcher and two independent SNMP engines
# * SNMPv2c and SNMPv3
# * with community name 'public' or USM username usr-md5-des
# * over IPv4/UDP
# * send IMFORM notification
# * to multiple Managers
# * with TRAP ID 'coldStart' specified as a MIB symbol
# * include managed object information specified as var-bind objects pair
#
# Within this script we have a single asynchronous TransportDispatcher
# and a single UDP-based transport serving two independent SNMP engines.
# We use a single instance of AsyncNotificationOriginator with each of 
# SNMP Engines to communicate INFORM notification to remote systems.
#
# When we receive a [response] message from remote system we use
# a custom message router to choose what of the two SNMP engines
# data packet should be handed over. The selection criteria we
# employ here is based on peer's UDP port number. Other selection
# criterias are also possible.
# 
from pysnmp.entity.rfc3413.oneliner import ntforg
from pysnmp.entity import engine
from pysnmp.carrier.asyncore.dispatch import AsyncoreDispatcher

# List of targets in the following format:
# ( ( authData, transportTarget ), ... )
targets = (
    # 1-st target (SNMPv2c over IPv4/UDP)
    ( ntforg.CommunityData('public'),
      ntforg.UdpTransportTarget(('localhost', 1162)),
      ntforg.ContextData() ),
    # 2-nd target (SNMPv3 over IPv4/UDP)
    ( ntforg.UsmUserData('usr-md5-des', 'authkey1', 'privkey1'),
      ntforg.UdpTransportTarget(('localhost', 162)),
      ntforg.ContextData() ),
)

def cbFun(snmpEngine, sendRequestHandle, errorIndication, 
          errorStatus, errorIndex, varBinds, cbCtx):
    snmpEngine = cbCtx
    if errorIndication:
        print('Notification %s for %s not sent: %s' % (sendRequestHandle, snmpEngine.snmpEngineID.prettyPrint(), errorIndication))
    elif errorStatus:
        print('Notification Receiver returned error for request %s, SNMP Engine %s: %s @%s' % (sendRequestHandle, snmpEngine.snmpEngineID.prettyPrint(), errorStatus, errorIndex))
    else:
        print('Notification %s for SNMP Engine %s delivered:' % (sendRequestHandle, snmpEngine.snmpEngineID.prettyPrint()))
        for name, val in varBinds:
            print('%s = %s' % (name.prettyPrint(), val.prettyPrint()))

# Instantiate the single transport dispatcher object
transportDispatcher = AsyncoreDispatcher()

# Setup a custom data routing function to select snmpEngine by transportDomain
transportDispatcher.registerRoutingCbFun(
    lambda td,ta,d: ta[1] % 3 and 'A' or 'B'
)

snmpEngineA = engine.SnmpEngine()
snmpEngineA.registerTransportDispatcher(transportDispatcher, 'A')

snmpEngineB = engine.SnmpEngine()
snmpEngineB.registerTransportDispatcher(transportDispatcher, 'B')

ntfOrg = ntforg.AsyncNotificationOriginator()

for authData, transportTarget, contextData in targets:
    snmpEngine = transportTarget.getTransportInfo()[1][1] % 3 and \
            snmpEngineA or snmpEngineB
    sendPduHandle = ntfOrg.sendNotification(
        snmpEngine,
        authData,
        transportTarget,
        contextData,
        'inform',       # NotifyType
        ntforg.NotificationType(
            ntforg.ObjectIdentity('SNMPv2-MIB', 'coldStart')
        ).addVarBinds( ( '1.3.6.1.2.1.1.1.0', 'my name' ) ),
        cbInfo=(cbFun, snmpEngine)
    )

transportDispatcher.runDispatcher()