mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-12-27 16:04:56 +00:00
257 lines
8.8 KiB
Python
Executable File
257 lines
8.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Script for creating a variable number of virtual routers.
|
|
"""
|
|
import sys
|
|
import os
|
|
import importlib
|
|
import time
|
|
from pprint import pformat
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
|
|
LFCliBase = lfcli_base.LFCliBase
|
|
LFUtils = importlib.import_module("py-json.LANforge.LFUtils")
|
|
realm = importlib.import_module("py-json.realm")
|
|
Realm = realm.Realm
|
|
lf_logger_config = importlib.import_module("py-scripts.lf_logger_config")
|
|
|
|
# TODO - script under development
|
|
'''
|
|
|
|
Notes:
|
|
start with crafting the flags for the virtual router using AddVrFlags,
|
|
my_vr_flags = LFPost.set_flags(AddVrFlags, 0, [ "USE_IPV6", "USE_IPV6_RADVD" ])
|
|
post_request = new LFPost()
|
|
post_request.post_add_vr(alias=myalias, flags=my_vr_flags, height=100, width=100, resource=1, x=100, y=100)
|
|
|
|
but don't use xorp unless you know you want to ("USE_XORP_MULTICAST")
|
|
|
|
so adding the vAP and eth port to the virt router is that an virt router command or need to look for something else?
|
|
|
|
add_vrcx will add a virtual-router-connection, which is effectively a port in a vr
|
|
|
|
you can build it manually, look at what is saved in DB/DFLT/*
|
|
|
|
FREE_LIST means it is out in the wild and not inside a virtual router, btw
|
|
note grep for the router name
|
|
|
|
cli commands
|
|
add_vr
|
|
add_vr_bgp
|
|
add_bgp_peer
|
|
add_vrcx
|
|
add_vrcx2
|
|
|
|
Adding vap to virtual router
|
|
./ports.db.1.1:122:add_vrcx 1 1 vr_router_0 vap4 NA NA NA NA 737 188 10 10 0 NA 0.0.0.0 43200 0.0.0.0 0.0.0.0 0.0.0.0 NA 1 0.0.0.0 1 0.0.0.0 24 1 100 1 'NA' 'NA' 'NA'
|
|
./ports.db.1.1:123:add_vrcx2 1 1 vr_router_0 vap4 NA NA 00:00:00:00:00:00-0 00:00:00:00:00:00-0 00:00:00:00:00:00-0 00:00:00:00:00:00-0
|
|
|
|
Adding dhcp
|
|
|
|
|
|
'''
|
|
|
|
class CreateVR(Realm):
|
|
def __init__(self,
|
|
lfclient_host="localhost",
|
|
lfclient_port=8080,
|
|
debug=False,
|
|
# resource=1, # USE name=1.2.vr0 convention instead
|
|
vr_name=None,
|
|
ports_list=(),
|
|
services_list=(),
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False,
|
|
_proxy_str=None,
|
|
_capture_signal_list=()):
|
|
super().__init__(lfclient_host=lfclient_host,
|
|
lfclient_port=lfclient_port,
|
|
debug_=debug,
|
|
_exit_on_error=_exit_on_error,
|
|
_exit_on_fail=_exit_on_fail,
|
|
_proxy_str=_proxy_str,
|
|
_capture_signal_list=_capture_signal_list)
|
|
|
|
eid_name = self.name_to_eid(vr_name)
|
|
self.vr_name = eid_name
|
|
self.ports_list = ports_list
|
|
self.services_list = services_list
|
|
self.vr_profile = self.new_vr_profile()
|
|
|
|
# TODO this was hardcoded as an example
|
|
def clean(self):
|
|
if (self.vr_name is None) or (self.vr_profile.vr_eid is None) and (
|
|
self.vr_profile.vr_eid) == "":
|
|
print("No vr_eid to clean")
|
|
return
|
|
# self.rm_port("1.1.rd90a", debug_=self.debug)
|
|
# self.rm_port("1.1.rd90b", debug_=self.debug)
|
|
# self.wait_until_ports_disappear(sta_list=["1.1.rd90a", "1.1.rd90b"],
|
|
# debug_=self.debug)
|
|
|
|
if (self.vr_profile.vr_eid is not None) \
|
|
and (self.vr_profile.vr_eid[1] is not None) \
|
|
and (self.vr_profile.vr_eid[2] is not None):
|
|
self.vr_profile.cleanup(debug=self.debug)
|
|
|
|
if (self.vr_name is not None) \
|
|
and (self.vr_name[1] is not None) \
|
|
and (self.vr_name[2] is not None):
|
|
data = {
|
|
"shelf": 1,
|
|
"resource": self.vr_name[1],
|
|
"router_name": self.vr_name[2]
|
|
}
|
|
self.json_post("/cli-json/rm_vr", data, debug_=self.debug)
|
|
time.sleep(1)
|
|
self.json_post("/cli-json/nc_show_vr", {
|
|
"shelf": 1,
|
|
"resource": self.vr_name[1],
|
|
"router": "all"
|
|
}, debug_=self.debug)
|
|
self.json_post("/cli-json/nc_show_vrcx", {
|
|
"shelf": 1,
|
|
"resource": self.vr_name[1],
|
|
"cx_name": "all"
|
|
}, debug_=self.debug)
|
|
|
|
def build(self):
|
|
self.vr_profile.apply_netsmith(
|
|
self.vr_name[1], delay=5)
|
|
# self.json_post("/cli-json/add_rdd", {
|
|
# "shelf": 1,
|
|
# "resource": self.vr_name[1],
|
|
# "port": "rd90a",
|
|
# "peer_ifname": "rd90b",
|
|
# "report_timer": "3000"
|
|
# })
|
|
# self.json_post("/cli-json/add_rdd", {
|
|
# "shelf": 1,
|
|
# "resource": self.vr_name[1],
|
|
# "port": "rd90b",
|
|
# "peer_ifname": "rd90a",
|
|
# "report_timer": "3000"
|
|
# })
|
|
# self.wait_until_ports_appear(
|
|
# sta_list=[
|
|
# "1.1.rd90a",
|
|
# "1.1.rd90b"],
|
|
# debug_=self.debug)
|
|
self.vr_profile.vrcx_list(
|
|
resource=self.vr_name[1],
|
|
do_sync=True) # do_sync
|
|
self.vr_profile.create(vr_name=self.vr_name)
|
|
|
|
self.vr_profile.sync_netsmith(resource=self.vr_name[1])
|
|
self._pass("created router")
|
|
|
|
def start(self):
|
|
"""
|
|
Move a vrcx into a router and then movie it out
|
|
:return: void
|
|
"""
|
|
# move rd90a into router
|
|
# self.vr_profile.refresh_netsmith(
|
|
# resource=self.vr_name[1])
|
|
# logger.info(pformat(("vr_eid", self.vr_name)))
|
|
# self.vr_profile.wait_until_vrcx_appear(
|
|
# resource=self.vr_name[1], name_list=[
|
|
# "rd90a", "rd90b"])
|
|
# self.vr_profile.add_vrcx(
|
|
# vr_eid=self.vr_name,
|
|
# connection_name_list="rd90a")
|
|
|
|
# self.vr_profile.refresh_netsmith(
|
|
# resource=self.vr_name[1])
|
|
# test to make sure that vrcx is inside vr we expect
|
|
self.vr_profile.vrcx_list(resource=self.vr_name[1], do_sync=True)
|
|
vr_list = self.vr_profile.router_list(
|
|
resource=self.vr_name[1], do_refresh=True)
|
|
router = self.vr_profile.find_cached_router(
|
|
resource=self.vr_name[1], router_name=self.vr_name[2])
|
|
logger.info("cached router 120: {router}".format(router=router))
|
|
router_eid = LFUtils.name_to_eid(router["eid"])
|
|
logger.info(pformat("router eid 122: {router_eid}".format(router_eid=router_eid)))
|
|
full_router = self.json_get(
|
|
"/vr/1/%s/%s/%s" %
|
|
(router_eid[0],
|
|
router_eid[1],
|
|
self.vr_name[2]),
|
|
debug_=self.debug)
|
|
logger.info(pformat("full router: {full_router}".format(full_router=full_router)))
|
|
time.sleep(5)
|
|
if router is None:
|
|
self._fail("Unable to find router after vrcx move " + self.vr_name)
|
|
self.exit_fail()
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
|
|
def main():
|
|
# /home/lanforge-scripts/py-json/LANforge/lfcli_base.py - for bare args parser
|
|
parser = LFCliBase.create_bare_argparse(
|
|
prog=__file__,
|
|
description="""\
|
|
{f}
|
|
--------------------
|
|
Command example:
|
|
{f} --vr_name 1.vr0 --ports 1.br0,1.rdd0a --services 1.br0=dhcp,nat --services 1.vr0=radvd
|
|
{f} --vr_name 2.vr0 --ports 2.br0,2.vap2 --services
|
|
|
|
--debug
|
|
""".format(f=__file__))
|
|
required = parser.add_argument_group('required arguments')
|
|
required.add_argument('--vr_name', '--vr_names', required=True,
|
|
help='EID of virtual router, like 1.2.vr0')
|
|
|
|
optional = parser.add_argument_group('optional arguments')
|
|
|
|
optional.add_argument(
|
|
'--ports',
|
|
default=None,
|
|
required=False,
|
|
help='Comma separated list of ports to add to virtual router')
|
|
optional.add_argument('--services', default=None, required=False,
|
|
help='Add router services to a port, "br0=nat,dhcp"')
|
|
|
|
args = parser.parse_args()
|
|
|
|
logger_config = lf_logger_config.lf_logger_config()
|
|
# set the logger level to requested value
|
|
logger_config.set_level(level=args.log_level)
|
|
logger_config.set_json(json_file=args.lf_logger_config_json)
|
|
|
|
create_vr = CreateVR(lfclient_host=args.mgr,
|
|
lfclient_port=args.mgr_port,
|
|
vr_name=args.vr_name,
|
|
ports_list=args.ports,
|
|
services_list=args.services,
|
|
debug=args.debug,
|
|
_exit_on_error=True,
|
|
_exit_on_fail=True)
|
|
# create_vr.clean()
|
|
create_vr.build()
|
|
create_vr.start()
|
|
create_vr.monitor()
|
|
# create_vr.stop()
|
|
# create_vr.clean()
|
|
print('Created Virtual Router')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
#
|