mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 10:48:02 +00:00
117 lines
4.5 KiB
Python
Executable File
117 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
|
|
Create Layer-3 Cross Connection Using LANforge JSON AP : https://www.candelatech.com/cookbook.php?vol=fire&book=scripted+layer-3+test
|
|
Written by Candela Technologies Inc.
|
|
Updated by: Erin Grimes
|
|
Example Command:
|
|
./create_l3.py --endp_a 'eth1' --endp_b 'eth2' --min_rate_a '56000' --min_rate_b '40000'
|
|
"""
|
|
import sys
|
|
import os
|
|
import importlib
|
|
import argparse
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
LANforge = importlib.import_module("py-json.LANforge")
|
|
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
|
|
TestGroupProfile = realm.TestGroupProfile
|
|
|
|
|
|
class CreateL3(Realm):
|
|
def __init__(self,
|
|
name_prefix,
|
|
endp_b,
|
|
endp_a,
|
|
host="localhost", port=8080, mode=0,
|
|
min_rate_a=56, max_rate_a=0,
|
|
min_rate_b=56, max_rate_b=0,
|
|
_debug_on=False,
|
|
_exit_on_error=False,
|
|
_exit_on_fail=False):
|
|
super().__init__(host, port)
|
|
self.host = host
|
|
self.port = port
|
|
self.endp_b = endp_b
|
|
self.endp_a = endp_a
|
|
self.mode = mode
|
|
self.name_prefix = name_prefix
|
|
self.station_profile = self.new_station_profile()
|
|
self.cx_profile = self.new_l3_cx_profile()
|
|
self.station_profile.lfclient_url = self.lfclient_url
|
|
self.station_list= LFUtils.portNameSeries(prefix_="sta", start_id_=0, end_id_=2, padding_number_=10000, radio='wiphy0') #Make radio a user defined variable from terminal.
|
|
self.cx_profile.host = self.host
|
|
self.cx_profile.port = self.port
|
|
self.cx_profile.name_prefix = self.name_prefix
|
|
self.cx_profile.side_a_min_bps = min_rate_a
|
|
self.cx_profile.side_a_max_bps = max_rate_a
|
|
self.cx_profile.side_b_min_bps = min_rate_b
|
|
self.cx_profile.side_b_max_bps = max_rate_b
|
|
|
|
def pre_cleanup(self):
|
|
self.cx_profile.cleanup_prefix()
|
|
|
|
def build(self):
|
|
self.cx_profile.create(endp_type="lf_udp",
|
|
side_a=self.endp_a,
|
|
side_b=self.endp_b,
|
|
sleep_time=0)
|
|
self._pass("PASS: Cross-connect build finished")
|
|
|
|
|
|
def main(args):
|
|
|
|
num_sta = 2
|
|
if (args.num_stations is not None) and (int(args.num_stations) > 0):
|
|
num_sta = int(args.num_stations)
|
|
|
|
# station_list = LFUtils.portNameSeries(prefix_="sta", start_id_=int(args.number_template), end_id_=num_sta+int(args.number_template) - 1, padding_number_=10000,
|
|
# radio=args.radio)
|
|
ip_var_test = CreateL3(host=args.mgr,
|
|
port=args.mgr_port,
|
|
name_prefix="VT",
|
|
endp_a=args.endp_a,
|
|
endp_b=args.endp_b,
|
|
min_rate_a=args.min_rate_a,
|
|
min_rate_b=args.min_rate_b,
|
|
mode=args.mode,
|
|
_debug_on=args.debug)
|
|
|
|
ip_var_test.pre_cleanup()
|
|
ip_var_test.build()
|
|
if not ip_var_test.passes():
|
|
print(ip_var_test.get_fail_message())
|
|
ip_var_test.exit_fail()
|
|
print('Created %s stations and connections' % num_sta)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = LFCliBase.create_basic_argparse(
|
|
prog='create_l3.py',
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
epilog='''\
|
|
Generate traffic between ports
|
|
''',
|
|
description='''\
|
|
''')
|
|
parser.add_argument('--min_rate_a', help='--min_rate_a bps rate minimum for side_a', default=56000)
|
|
parser.add_argument('--min_rate_b', help='--min_rate_b bps rate minimum for side_b', default=56000)
|
|
parser.add_argument('--endp_a', help='--endp_a station list', default=["eth1"], action="append", required=True)
|
|
parser.add_argument('--endp_b', help='--upstream port', default="eth2", required=True)
|
|
parser.add_argument('--mode', help='Used to force mode of stations', default=0)
|
|
parser.add_argument('--ap', help='Used to force a connection to a particular AP')
|
|
parser.add_argument('--number_template', help='Start the station numbering with a particular number. Default is 0000', default=0000)
|
|
args = parser.parse_args()
|
|
|
|
main(args)
|