create_l4.py

#!/usr/bin/env python3
"""
NAME: create_l4.py

PURPOSE:
create_l4.py will create a user specified number of stations and layer-4 endpoints in a bytes-rd test scenario.

Use './create_l4.py --help' to see command line usage and options

EXAMPLE:
    ./create_l4.py --mgr <ip-address> --radio wiphy2 --num_stations 4 --upstream_port 1.1.eth1 --ssid <ssid> --passwd <passwd> --security wpa2 --debug

COPYRIGHT:
    Copyright 2023 Candela Technologies Inc
    License: Free to distribute and modify. LANforge systems must be licensed.
"""

import sys
import os
import importlib
import argparse
import logging
import traceback
import requests
from pandas import json_normalize
import json

logger = logging.getLogger(__name__)

if sys.version_info[0] != 3:
    logger.critical("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")
lf_logger_config = importlib.import_module("py-scripts.lf_logger_config")
Realm = realm.Realm
TestGroupProfile = realm.TestGroupProfile


class CreateL4(Realm):
    def __init__(
            self,
            ssid,
            security,
            password,
            url,
            sta_list,
            name_prefix,
            upstream,
            radio,
            host="localhost",
            port=8080,
            mode=0,
            ap=None,
            side_a_min_rate=56,
            side_a_max_rate=0,
            side_b_min_rate=56,
            side_b_max_rate=0,
            number_template="00000",
            use_ht160=False,
            _debug_on=False,
            _exit_on_error=False,
            _exit_on_fail=False):
        super().__init__(host, port)
        self.upstream = upstream
        self.host = host
        self.port = port
        self.ssid = ssid
        self.sta_list = sta_list
        self.security = security
        self.password = password
        self.url = url
        self.radio = radio
        self.mode = mode
        self.ap = ap
        self.number_template = number_template
        self.debug = _debug_on
        self.name_prefix = name_prefix
        self.station_profile = self.new_station_profile()
        self.cx_profile = self.new_l4_cx_profile()
        self.station_profile.lfclient_url = self.lfclient_url
        self.station_profile.ssid = self.ssid
        self.station_profile.ssid_pass = self.password
        self.station_profile.security = self.security
        self.station_profile.number_template_ = self.number_template
        self.station_profile.debug = self.debug
        self.station_profile.use_ht160 = use_ht160
        if self.station_profile.use_ht160:
            self.station_profile.mode = 9
        self.station_profile.mode = mode
        if self.ap is not None:
            self.station_profile.set_command_param("add_sta", "ap", self.ap)
        # 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.url = self.url
        self.cx_profile.side_a_min_bps = side_a_min_rate
        self.cx_profile.side_a_max_bps = side_a_max_rate
        self.cx_profile.side_b_min_bps = side_b_min_rate
        self.cx_profile.side_b_max_bps = side_b_max_rate

    def cleanup(self):
        self.cx_profile.cleanup()
        self.station_profile.cleanup()
        if LFUtils.wait_until_ports_disappear(
                base_url=self.lfclient_url,
                port_list=self.station_profile.station_names,
                debug=self.debug):
            self._pass("Ports were properly deleted.")
        else:
            self._fail("Ports were NOT properly deleted.")

    def build(self):
        # Build stations
        self.station_profile.use_security(
            self.security, self.ssid, self.password)
        self.station_profile.set_number_template(self.number_template)
        logger.info("Creating stations")
        self.station_profile.set_command_flag(
            "add_sta", "create_admin_down", 1)
        self.station_profile.set_command_param(
            "set_port", "report_timer", 1500)
        self.station_profile.set_command_flag("set_port", "rpt_timer", 1)

        timeout = len(self.sta_list) * 2 + 20
        if self.station_profile.create(
                radio=self.radio,
                timeout=timeout,
                sta_names_=self.sta_list,
                debug=self.debug):
            self._pass("PASS: Station build finished")

            if self.cx_profile.create(
                    ports=self.station_profile.station_names,
                    sleep_time=0,
                    debug_=self.debug,
                    suppress_related_commands_=True):
                self._pass("CX creation succeeded.")
            else:
                self._fail("CX creation did not succeed.")
        else:
            self._fail("Station build did not succeed.")


def main():
    help_summary = '''\
     This script generates a variable number (N) of stations as specified by the user. For each station, it also creates
      layer-4 endpoints, which are initially set to a stopped state.
    '''
    parser = LFCliBase.create_basic_argparse(
        prog='create_l4.py',
        formatter_class=argparse.RawTextHelpFormatter,
        epilog='''\
            This script will create a specified number of stations and layer-4 endpoints as a bytes-rd scenario.

            ''',

        description='''\
---------------------------
Layer-4 Test Script - create_l4.py
---------------------------
Summary:
This script will create a user specified number of stations and layer-4 endpoints in a bytes-rd test scenario.
---------------------------
Generic command layout:

./create_l4.py
    --mgr <ip_address>
    --upstream_port eth1
    --radio wiphy0
    --num_stations 10
    --security {open|wep|wpa|wpa2|wpa3}
    --mode   1
        {"auto"   : "0",
        "a"      : "1",
        "b"      : "2",
        "g"      : "3",
        "abg"    : "4",
        "abgn"   : "5",
        "bgn"    : "6",
        "bg"     : "7",
        "abgnAC" : "8",
        "anAC"   : "9",
        "an"     : "10",
        "bgnAC"  : "11",
        "abgnAX" : "12",
        "bgnAX"  : "13",
    --ssid <ssid>
    --password <password>
    --a_min 1000
    --b_min 1000
    --ap "00:0e:8e:78:e1:76"
    --debug

EXAMPLE:
    ./create_l4.py --mgr <ip-address> --radio wiphy2 --num_stations 4 --upstream_port 1.1.eth1 --ssid <ssid> --passwd <passwd> --security wpa2 --debug

Tested on 02/13/2023:
         kernel version: 5.19.17+
         gui version: 5.4.6
         the layer-4 bytes-rd scenario was successfully created and tested on a ct523c sta-to-eth cross connection.

            ''')

    parser.add_argument(
        '--a_min',
        help='--a_min bps rate minimum for side_a',
        default=256000)
    parser.add_argument(
        '--b_min',
        help='--b_min bps rate minimum for side_b',
        default=256000)

    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("--lf_user", type=str, help="--lf_user lanforge user name ",
                        default="lanforge")
    parser.add_argument("--lf_passwd", type=str, help="--lf_passwd lanforge password ",
                        default="lanforge")
    args = parser.parse_args()

    if args.help_summary:
        print(help_summary)
        exit(0)

    # set up logger
    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)

    num_sta = 2
    if (args.num_stations is not None) and (int(args.num_stations) > 0):
        num_sta = int(args.num_stations)

    # get ip upstream port
    rv = LFUtils.name_to_eid(args.upstream_port)
    shelf = rv[0]
    resource = rv[1]
    port_name = rv[2]
    request_command = 'http://{lfmgr}:{lfport}/port/{shelf}/{resource}/{port_name}'.format(
        lfmgr=args.mgr, lfport=args.mgr_port, shelf=shelf, resource=resource, port_name=port_name)
    logger.info("port request command: {request_command}".format(request_command=request_command))
    request = requests.get(request_command, auth=(args.lf_user, args.lf_passwd))
    logger.info("port request status_code {status}".format(status=request.status_code))
    lanforge_json = request.json()
    lanforge_json_formatted = json.dumps(lanforge_json, indent=4)
    try:
        key = 'interface'
        df = json_normalize(lanforge_json[key])
        upstream_port_ip = df['ip'].iloc[0]
    except Exception as x:
        traceback.print_exception(Exception, x, x.__traceback__, chain=True)
        logger.error(
            "json returned : {lanforge_json_formatted}".format(lanforge_json_formatted=lanforge_json_formatted))

    url = 'dl http://{upstream_port_ip} /dev/null'.format(upstream_port_ip=upstream_port_ip)

    station_list = LFUtils.portNameSeries(
        prefix_="sta",
        start_id_=0,
        end_id_=num_sta - 1,
        padding_number_=10000,
        radio=args.radio)
    ip_var_test = CreateL4(host=args.mgr,
                           port=args.mgr_port,
                           number_template="0000",
                           sta_list=station_list,
                           name_prefix="VT",
                           upstream=args.upstream_port,
                           ssid=args.ssid,
                           password=args.passwd,
                           url=url,
                           radio=args.radio,
                           security=args.security,
                           use_ht160=False,
                           side_a_min_rate=args.a_min,
                           side_b_min_rate=args.b_min,
                           mode=args.mode,
                           ap=args.ap,
                           _debug_on=args.debug)

    ip_var_test.cleanup()
    ip_var_test.build()

    # TODO:  Clean up by default, and add --no_cleanup option to NOT do cleanup.

    if ip_var_test.passes():
        ip_var_test.exit_success()
    else:
        ip_var_test.exit_fail()


if __name__ == "__main__":
    main()