mirror of
				https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
				synced 2025-11-04 04:38:02 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			140 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			140 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.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 = self.new_l3_cx_profile()
 | 
						|
        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.cx_profile.start_cx()
 | 
						|
        self._pass("PASS: Cross-connect build finished")
 | 
						|
 | 
						|
 | 
						|
def main(args):
 | 
						|
 | 
						|
    num_sta = 0
 | 
						|
    # if (args.num_stations is not None) and (int(args.num_stations) > 0):
 | 
						|
    #     num_sta = int(args.num_stations)
 | 
						|
 | 
						|
    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(f'Created {num_sta} stations and connections')
 | 
						|
 | 
						|
 | 
						|
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=[],
 | 
						|
        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)
 |