327 lines
11 KiB
Python
Executable file
327 lines
11 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# SPDX-License-Identifier: ISC
|
|
|
|
#
|
|
# Copyright (c) 2022 Donatas Abraitis <donatas@opensourcerouting.org>
|
|
#
|
|
|
|
"""
|
|
Test some of the BGP4V2-MIB entries.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
from time import sleep
|
|
import pytest
|
|
import functools
|
|
|
|
CWD = os.path.dirname(os.path.realpath(__file__))
|
|
sys.path.append(os.path.join(CWD, "../"))
|
|
|
|
# pylint: disable=C0413
|
|
# Import topogen and topotest helpers
|
|
from lib.topogen import Topogen, TopoRouter, get_topogen
|
|
from lib.snmptest import SnmpTester
|
|
from lib import topotest
|
|
from lib.topolog import logger
|
|
|
|
pytestmark = [pytest.mark.bgpd, pytest.mark.snmp]
|
|
|
|
|
|
def build_topo(tgen):
|
|
tgen.add_router("r1")
|
|
tgen.add_router("r2")
|
|
tgen.add_router("r3")
|
|
tgen.add_router("rr")
|
|
|
|
switch = tgen.add_switch("s1")
|
|
switch.add_link(tgen.gears["r1"])
|
|
switch.add_link(tgen.gears["r2"])
|
|
switch.add_link(tgen.gears["r3"])
|
|
switch.add_link(tgen.gears["rr"])
|
|
|
|
|
|
def setup_module(mod):
|
|
snmpd = os.system("which snmpd")
|
|
if snmpd:
|
|
error_msg = "SNMP not installed - skipping"
|
|
pytest.skip(error_msg)
|
|
|
|
tgen = Topogen(build_topo, mod.__name__)
|
|
tgen.start_topology()
|
|
|
|
for rname, router in tgen.routers().items():
|
|
router.load_config(
|
|
TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
|
|
)
|
|
router.load_config(
|
|
TopoRouter.RD_BGP,
|
|
os.path.join(CWD, "{}/bgpd.conf".format(rname)),
|
|
"-M snmp",
|
|
)
|
|
|
|
r2 = tgen.gears["r2"]
|
|
r2.load_config(
|
|
TopoRouter.RD_SNMP,
|
|
os.path.join(CWD, "{}/snmpd.conf".format(r2.name)),
|
|
"-Le -Ivacm_conf,usmConf,iquery -V -DAgentX",
|
|
)
|
|
r2.load_config(
|
|
TopoRouter.RD_TRAP,
|
|
os.path.join(CWD, "{}/snmptrapd.conf".format(r2.name)),
|
|
" -On -OQ ",
|
|
)
|
|
|
|
tgen.start_router()
|
|
|
|
|
|
def teardown_module(mod):
|
|
tgen = get_topogen()
|
|
tgen.stop_topology()
|
|
|
|
|
|
def test_bgp_snmp_bgp4v2():
|
|
tgen = get_topogen()
|
|
|
|
r1 = tgen.gears["r1"]
|
|
r2 = tgen.gears["r2"]
|
|
rr = tgen.gears["rr"]
|
|
|
|
def _bgp_converge_summary():
|
|
output = json.loads(r2.vtysh_cmd("show bgp summary json"))
|
|
expected = {
|
|
"ipv4Unicast": {
|
|
"peers": {
|
|
"192.168.12.4": {
|
|
"state": "Established",
|
|
"pfxRcd": 6,
|
|
}
|
|
}
|
|
},
|
|
"ipv6Unicast": {
|
|
"peers": {
|
|
"2001:db8::12:4": {
|
|
"state": "Established",
|
|
"pfxRcd": 4,
|
|
}
|
|
}
|
|
},
|
|
}
|
|
# tgen.mininet_cli()
|
|
return topotest.json_cmp(output, expected)
|
|
|
|
test_func = functools.partial(_bgp_converge_summary)
|
|
_, result = topotest.run_and_expect(test_func, None, count=30, wait=1)
|
|
assert result is None, "Can't see connections established"
|
|
|
|
def _bgp_converge_prefixes():
|
|
output = json.loads(r2.vtysh_cmd("show bgp all json"))
|
|
expected = {
|
|
"ipv4Unicast": {
|
|
"routes": {
|
|
"10.0.0.0/31": [
|
|
{
|
|
"metric": 1,
|
|
"origin": "IGP",
|
|
}
|
|
],
|
|
"10.0.0.2/32": [
|
|
{
|
|
"metric": 2,
|
|
"origin": "incomplete",
|
|
}
|
|
],
|
|
}
|
|
},
|
|
"ipv6Unicast": {
|
|
"routes": {
|
|
"2001:db8::1/128": [
|
|
{
|
|
"metric": 1,
|
|
"origin": "IGP",
|
|
}
|
|
],
|
|
"2001:db8:1::/56": [
|
|
{
|
|
"metric": 2,
|
|
"origin": "incomplete",
|
|
}
|
|
],
|
|
}
|
|
},
|
|
}
|
|
# tgen.mininet_cli()
|
|
return topotest.json_cmp(output, expected)
|
|
|
|
test_func = functools.partial(_bgp_converge_prefixes)
|
|
_, result = topotest.run_and_expect(test_func, None, count=30, wait=1)
|
|
assert result is None, "Can't see prefixes from R1"
|
|
|
|
snmp = SnmpTester(r2, "localhost", "public", "2c", "-Ln -On")
|
|
|
|
def _snmpwalk_remote_addr():
|
|
expected = {
|
|
"1.3.6.1.3.5.1.1.2.1.5.1.1.192.168.12.4": "C0 A8 0C 04",
|
|
"1.3.6.1.3.5.1.1.2.1.5.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4": "20 01 0D B8 00 00 00 00 00 00 00 00 00 12 00 04",
|
|
}
|
|
|
|
# bgp4V2PeerRemoteAddr
|
|
# tgen.mininet_cli()
|
|
output, _ = snmp.walk(".1.3.6.1.3.5.1.1.2.1.5")
|
|
return output == expected
|
|
|
|
_, result = topotest.run_and_expect(_snmpwalk_remote_addr, True, count=10, wait=1)
|
|
assertmsg = "Can't fetch SNMP for bgp4V2PeerRemoteAddr"
|
|
assert result, assertmsg
|
|
|
|
def _snmpwalk_peer_state():
|
|
expected = {
|
|
"1.3.6.1.3.5.1.1.2.1.13.1.1.192.168.12.4": "6",
|
|
"1.3.6.1.3.5.1.1.2.1.13.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4": "6",
|
|
}
|
|
|
|
# bgp4V2PeerState
|
|
output, _ = snmp.walk(".1.3.6.1.3.5.1.1.2.1.13")
|
|
return output == expected
|
|
|
|
_, result = topotest.run_and_expect(_snmpwalk_peer_state, True, count=10, wait=1)
|
|
assertmsg = "Can't fetch SNMP for bgp4V2PeerState"
|
|
assert result, assertmsg
|
|
|
|
def _snmpwalk_peer_last_error_code_received():
|
|
expected = {
|
|
"1.3.6.1.3.5.1.1.3.1.1.1.1.192.168.12.4": "0",
|
|
"1.3.6.1.3.5.1.1.3.1.1.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4": "0",
|
|
}
|
|
|
|
# bgp4V2PeerLastErrorCodeReceived
|
|
output, _ = snmp.walk(".1.3.6.1.3.5.1.1.3.1.1")
|
|
return output == expected
|
|
|
|
_, result = topotest.run_and_expect(
|
|
_snmpwalk_peer_last_error_code_received, True, count=10, wait=1
|
|
)
|
|
assertmsg = "Can't fetch SNMP for bgp4V2PeerLastErrorCodeReceived"
|
|
assert result, assertmsg
|
|
|
|
def _snmpwalk_origin():
|
|
expected = {
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.10.10.10.32.1.192.168.12.4.1": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.10.10.10.32.1.192.168.12.4.2": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.0.0.0.31.1.192.168.12.4.1": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.0.0.0.31.1.192.168.12.4.2": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.0.0.2.32.1.192.168.12.4.1": "3",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.1.1.1.10.0.0.2.32.1.192.168.12.4.2": "3",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.2.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.0.0.1.128.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.1": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.2.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.0.0.1.128.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.2": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.2.1.2.32.1.13.184.0.1.0.0.0.0.0.0.0.0.0.0.56.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.1": "3",
|
|
"1.3.6.1.3.5.1.1.9.1.9.1.2.1.2.32.1.13.184.0.1.0.0.0.0.0.0.0.0.0.0.56.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.2": "3",
|
|
}
|
|
|
|
# bgp4V2NlriOrigin
|
|
output, _ = snmp.walk(".1.3.6.1.3.5.1.1.9.1.9")
|
|
return output == expected
|
|
|
|
_, result = topotest.run_and_expect(_snmpwalk_origin, True, count=10, wait=1)
|
|
assertmsg = "Can't fetch SNMP for bgp4V2NlriOrigin"
|
|
assert result, assertmsg
|
|
|
|
def _snmpwalk_med():
|
|
expected = {
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.10.10.10.32.1.192.168.12.4.1": "0",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.10.10.10.32.1.192.168.12.4.2": "0",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.0.0.0.31.1.192.168.12.4.1": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.0.0.0.31.1.192.168.12.4.2": "0",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.0.0.2.32.1.192.168.12.4.1": "2",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.1.1.1.10.0.0.2.32.1.192.168.12.4.2": "0",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.2.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.0.0.1.128.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.1": "1",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.2.1.2.32.1.13.184.0.0.0.0.0.0.0.0.0.0.0.1.128.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.2": "0",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.2.1.2.32.1.13.184.0.1.0.0.0.0.0.0.0.0.0.0.56.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.1": "2",
|
|
"1.3.6.1.3.5.1.1.9.1.17.1.2.1.2.32.1.13.184.0.1.0.0.0.0.0.0.0.0.0.0.56.2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4.2": "0",
|
|
}
|
|
|
|
# bgp4V2NlriMed
|
|
output, _ = snmp.walk(".1.3.6.1.3.5.1.1.9.1.17")
|
|
# tgen.mininet_cli()
|
|
return output == expected
|
|
|
|
_, result = topotest.run_and_expect(_snmpwalk_med, True, count=10, wait=1)
|
|
assertmsg = "Can't fetch SNMP for bgp4V2NlriMed"
|
|
assert result, assertmsg
|
|
|
|
#
|
|
# traps
|
|
#
|
|
|
|
#
|
|
# bgp4 traps
|
|
#
|
|
def _snmptrap_ipv4():
|
|
def __get_notif_bgp4_in_trap_file(router):
|
|
snmptrapfile = "{}/{}/snmptrapd.log".format(router.logdir, router.name)
|
|
outputfile = open(snmptrapfile).read()
|
|
output = snmp.get_notif_bgp4(outputfile)
|
|
|
|
return output
|
|
|
|
output = __get_notif_bgp4_in_trap_file(r2)
|
|
logger.info("output bgp4")
|
|
logger.info(output)
|
|
return snmp.is_notif_bgp4_valid(output, "192.168.12.4")
|
|
|
|
# skip tests is SNMP not installed
|
|
if not os.path.isfile("/usr/sbin/snmptrapd"):
|
|
error_msg = "SNMP not installed - skipping"
|
|
pytest.skip(error_msg)
|
|
|
|
rr.vtysh_cmd("clear bgp *")
|
|
_, result = topotest.run_and_expect(_snmptrap_ipv4, True, count=30, wait=1)
|
|
assertmsg = "Can't fetch SNMP trap for ipv4"
|
|
assert result, assertmsg
|
|
|
|
#
|
|
# bgp4v2 traps
|
|
#
|
|
def _snmptrap_ipv6():
|
|
def __get_notif_bgp4v2_in_trap_file(router):
|
|
snmptrapfile = "{}/{}/snmptrapd.log".format(router.logdir, router.name)
|
|
outputfile = open(snmptrapfile).read()
|
|
output = snmp.get_notif_bgp4v2(outputfile)
|
|
|
|
return output
|
|
|
|
# tgen.mininet_cli()
|
|
output = __get_notif_bgp4v2_in_trap_file(r2)
|
|
logger.info("output bgp4v2")
|
|
logger.info(output)
|
|
p_ipv4_addr = "1.192.168.12.4"
|
|
p_ipv6_addr = "2.32.1.13.184.0.0.0.0.0.0.0.0.0.18.0.4"
|
|
return (
|
|
snmp.is_notif_bgp4v2_valid(output, p_ipv4_addr, "Estab")
|
|
and snmp.is_notif_bgp4v2_valid(output, p_ipv6_addr, "Estab")
|
|
and snmp.is_notif_bgp4v2_valid(output, p_ipv4_addr, "Backward")
|
|
and snmp.is_notif_bgp4v2_valid(output, p_ipv6_addr, "Backward")
|
|
)
|
|
|
|
sleep(10)
|
|
r2.vtysh_cmd("conf\nbgp snmp traps bgp4-mibv2")
|
|
r2.vtysh_cmd("conf\nno bgp snmp traps rfc4273")
|
|
rr.vtysh_cmd("clear bgp *")
|
|
_, result = topotest.run_and_expect(_snmptrap_ipv6, True, count=60, wait=1)
|
|
assertmsg = "Can't fetch SNMP trap for ipv6"
|
|
assert result, assertmsg
|
|
|
|
|
|
def test_memory_leak():
|
|
"Run the memory leak test and report results."
|
|
tgen = get_topogen()
|
|
if not tgen.is_memleak_enabled():
|
|
pytest.skip("Memory leak test/report is disabled")
|
|
|
|
tgen.report_memory_leaks()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = ["-s"] + sys.argv[1:]
|
|
sys.exit(pytest.main(args))
|