From fe051fecb5e4de6d8f5961715d9022d719e8c95d Mon Sep 17 00:00:00 2001
From: Logan Lipke <lolipke@gmail.com>
Date: Fri, 3 Jul 2020 15:23:39 -0700
Subject: [PATCH] Added build method, allowed start method to bring stations up

---
 py-scripts/test_ipv4_l4.py | 52 +++++++++++++++++++++++++++++++++-----
 1 file changed, 45 insertions(+), 7 deletions(-)
 mode change 100644 => 100755 py-scripts/test_ipv4_l4.py

diff --git a/py-scripts/test_ipv4_l4.py b/py-scripts/test_ipv4_l4.py
old mode 100644
new mode 100755
index 431c8422..ec4d54dd
--- a/py-scripts/test_ipv4_l4.py
+++ b/py-scripts/test_ipv4_l4.py
@@ -12,35 +12,73 @@ if 'py-json' not in sys.path:
 import argparse
 from LANforge.lfcli_base import LFCliBase
 from LANforge.LFUtils import *
+from LANforge import LFUtils
 import realm
 import time
 import datetime
 
-class IPV4L4:
-    def __init__(self, host, port, ssid, security, password, num_stations, side_a_min_rate=56, side_b_min_rate=56, side_a_max_rate=0,
-                 side_b_max_rate=0, prefix="00000", test_duration="5m",
+
+class IPV4L4(LFCliBase):
+    def __init__(self, host, port, ssid, security, password, url, requests_per_ten, station_list, prefix="00000",
+                 resource=1,
                  _debug_on=False,
                  _exit_on_error=False,
                  _exit_on_fail=False):
         super().__init__(host, port, _debug=_debug_on, _halt_on_error=_exit_on_error, _exit_on_fail=_exit_on_fail)
+        self.host = host
+        self.port = port
+        self.ssid = ssid
+        self.security = security
+        self.password = password
+        self.url = url
+        self.requests_per_ten = requests_per_ten
+        self.prefix = prefix
+        self.sta_list = station_list
+        self.resource = resource
+
+        self.local_realm = realm.Realm(lfclient_host=self.host, lfclient_port=self.port)
+        self.profile = realm.StationProfile(self.lfclient_url, ssid=self.ssid, ssid_pass=self.password,
+                                            security=self.security, number_template_=self.prefix, mode=0, up=False,
+                                            dhcp=True,
+                                            debug_=False)
+        self.cx_profile = realm.L4CXProfile(lfclient_host=self.host, lfclient_port=self.port,
+                                            local_realm=self.local_realm, debug_=False)
+        self.cx_profile.url = self.url
+        self.cx_profile.requests_per_ten = self.requests_per_ten
 
     def build(self):
-        pass
+        # Build stations
+        self.profile.use_wpa2(True, self.ssid, self.password)
+        self.profile.set_number_template(self.prefix)
+        print("Creating stations")
+        self.profile.set_command_flag("add_sta", "create_admin_down", 1)
+        self.profile.set_command_param("set_port", "report_timer", 1500)
+        self.profile.set_command_flag("set_port", "rpt_timer", 1)
+        self.profile.create(resource=1, radio="wiphy0", sta_names_=self.sta_list, debug=False)
+        self._pass("PASS: Station build finished")
+        temp_sta_list = []
+        for station in range(len(self.sta_list)):
+            temp_sta_list.append(str(self.resource) + "." + self.sta_list[station])
+
+        self.cx_profile.create(ports=temp_sta_list, sleep_time=.5, debug_=False, suppress_related_commands_=None)
 
     def start(self):
-        pass
+        self.profile.admin_up(1)
 
     def stop(self):
         pass
 
-    def cleanup(self):
+    def cleanup(self, resource):
         pass
 
 
 def main():
     lfjson_host = "localhost"
     lfjson_port = 8080
-
+    station_list = LFUtils.portNameSeries(prefix_="sta", start_id_=0, end_id_=4, padding_number_=10000)
+    ip_test = IPV4L4(lfjson_host, lfjson_port, ssid="jedway-wpa2-x2048-4-4", password="jedway-wpa2-x2048-4-4",
+                     security="open", station_list=station_list, url="http://localhost", requests_per_ten=600)
+    ip_test.build()
 
 if __name__ == "__main__":
     main()
\ No newline at end of file