179 lines
6.7 KiB
Python
179 lines
6.7 KiB
Python
|
import random
|
||
|
|
import time
|
||
|
|
import string
|
||
|
|
import threading
|
||
|
|
|
||
|
|
from TCAction import TCActionBase
|
||
|
|
from NativeLog import NativeLog
|
||
|
|
from TCAction import PerformanceTCBase
|
||
|
|
from Utility import Encoding
|
||
|
|
|
||
|
|
|
||
|
|
class STAJAPThread(threading.Thread):
|
||
|
|
def __init__(self, test_action, port_name, ssid, password, delay1, delay2, change_mac):
|
||
|
|
threading.Thread.__init__(self)
|
||
|
|
self.setDaemon(True)
|
||
|
|
self.test_action = test_action
|
||
|
|
self.port_name = port_name
|
||
|
|
self.ssid = ssid
|
||
|
|
self.password = password
|
||
|
|
self.delay1 = delay1
|
||
|
|
self.delay2 = delay2
|
||
|
|
self.change_mac = change_mac
|
||
|
|
self.exit_flag = threading.Event()
|
||
|
|
pass
|
||
|
|
|
||
|
|
def exit(self):
|
||
|
|
self.exit_flag.set()
|
||
|
|
pass
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
total_test_count = 0
|
||
|
|
fail_count = 0
|
||
|
|
while self.exit_flag.isSet() is False:
|
||
|
|
# change mac
|
||
|
|
if self.change_mac is True:
|
||
|
|
mac = Encoding.generate_random_mac()
|
||
|
|
self.test_action.serial_write_line(self.port_name, "mac -S -o 1 -m %s" % mac)
|
||
|
|
self.test_action.check_response(self.port_name, "+MAC:STA,OK")
|
||
|
|
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# JAP
|
||
|
|
total_test_count += 1
|
||
|
|
# flush current port data
|
||
|
|
self.test_action.flush_data(self.port_name)
|
||
|
|
self.test_action.serial_write_line(self.port_name, "sta -C -s %s -p %s" % (self.ssid, self.password))
|
||
|
|
if self.test_action.check_response(self.port_name, "+JAP:CONNECTED", 45) is False:
|
||
|
|
fail_count += 1
|
||
|
|
NativeLog.add_trace_critical("[%s] Failed to JAP, Failed/Total : %d/%d"
|
||
|
|
% (self.port_name, fail_count, total_test_count))
|
||
|
|
continue
|
||
|
|
time.sleep(random.randint(self.delay1[0], self.delay1[1]))
|
||
|
|
|
||
|
|
# QAP
|
||
|
|
self.test_action.serial_write_line(self.port_name, "sta -D")
|
||
|
|
if self.test_action.check_response(self.port_name, "+QAP:OK", 5) is False:
|
||
|
|
NativeLog.add_trace_critical("[%s] Failed to QAP" % self.port_name)
|
||
|
|
time.sleep(random.randint(self.delay2[0], self.delay2[1]))
|
||
|
|
|
||
|
|
# make sure quit AP
|
||
|
|
self.test_action.serial_write_line(self.port_name, "sta -D")
|
||
|
|
pass
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class SoftAPNSTA(PerformanceTCBase.PerformanceTCBase):
|
||
|
|
|
||
|
|
def __init__(self, name, test_env, cmd_set, timeout=30, log_path=TCActionBase.LOG_PATH):
|
||
|
|
PerformanceTCBase.PerformanceTCBase.__init__(self, name, test_env, cmd_set=cmd_set,
|
||
|
|
timeout=timeout, log_path=log_path)
|
||
|
|
self.sta_num = 0
|
||
|
|
self.max_sta = 4
|
||
|
|
self.test_time = 60
|
||
|
|
self.delay1 = [5, 30]
|
||
|
|
self.delay2 = [5, 10]
|
||
|
|
self.change_mac = True
|
||
|
|
self.channel = 11
|
||
|
|
# load param from excel
|
||
|
|
for i in range(1, len(cmd_set)):
|
||
|
|
if cmd_set[i][0] != "dummy" and cmd_set[i][0] != "":
|
||
|
|
cmd_string = "self." + cmd_set[i][0]
|
||
|
|
exec cmd_string
|
||
|
|
pass
|
||
|
|
|
||
|
|
def process(self):
|
||
|
|
# configurable parameters
|
||
|
|
try:
|
||
|
|
sta_num = self.sta_num
|
||
|
|
max_sta = self.max_sta
|
||
|
|
test_time = self.test_time
|
||
|
|
# delay between JAP succeed and QAP
|
||
|
|
delay1 = self.delay1
|
||
|
|
# delay between QAP and JAP
|
||
|
|
delay2 = self.delay2
|
||
|
|
# if change mac each time before JAP
|
||
|
|
change_mac = self.change_mac
|
||
|
|
# channel
|
||
|
|
channel = self.channel
|
||
|
|
except StandardError, e:
|
||
|
|
raise StandardError("miss mandatory parameters")
|
||
|
|
|
||
|
|
# step 0, set op mode and enable dhcp
|
||
|
|
self.serial_write_line("SSC1", "op -S -o 2")
|
||
|
|
if self.check_response("SSC1", "+MODE:OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to set ap mode")
|
||
|
|
return
|
||
|
|
self.serial_write_line("SSC1", "dhcp -E -o 2")
|
||
|
|
if self.check_response("SSC1", "+DHCP:AP,OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to enable ap dhcp")
|
||
|
|
return
|
||
|
|
self.serial_write_line("SSC1", "dhcp -L -s 192.168.4.2 -e 192.168.4.100 -t 1")
|
||
|
|
if self.check_response("SSC1", "+DHCP:LEASE,OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to enable ap dhcp")
|
||
|
|
return
|
||
|
|
self.serial_write_line("SSC1", "dhcp -S -o 2")
|
||
|
|
if self.check_response("SSC1", "+DHCP:AP,OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to enable ap dhcp")
|
||
|
|
return
|
||
|
|
|
||
|
|
for i in range(sta_num):
|
||
|
|
self.serial_write_line("SSC%d" % (i+2), "op -S -o 1")
|
||
|
|
if self.check_response("SSC%d" % (i+2), "+MODE:OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to set sta mode")
|
||
|
|
return
|
||
|
|
self.serial_write_line("SSC%d" % (i+2), "dhcp -S -o 1")
|
||
|
|
if self.check_response("SSC%d" % (i+2), "+DHCP:STA,OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to enable sta dhcp")
|
||
|
|
|
||
|
|
# step 1, set ap config and load
|
||
|
|
ap_ssid = "".join([random.choice(string.uppercase) for m in range(15)])
|
||
|
|
ap_password = "".join([random.choice(string.lowercase) for m in range(15)])
|
||
|
|
|
||
|
|
self.serial_write_line("SSC1", "ap -S -s %s -p %s -t 3 -m %s -n %s"
|
||
|
|
% (ap_ssid, ap_password, max_sta, channel))
|
||
|
|
if self.check_response("SSC1", "+SAP:OK", 2) is False:
|
||
|
|
NativeLog.add_trace_critical("Failed to set AP")
|
||
|
|
return
|
||
|
|
|
||
|
|
# step 2, start thread to let STA JAP
|
||
|
|
sta_thread_list = []
|
||
|
|
for i in range(sta_num):
|
||
|
|
sta_thread_list.append(STAJAPThread(self, "SSC%d" % (i+2),
|
||
|
|
ap_ssid, ap_password, delay1, delay2, change_mac))
|
||
|
|
for sta_thread in sta_thread_list:
|
||
|
|
sta_thread.start()
|
||
|
|
|
||
|
|
# step 3, sleep for test time
|
||
|
|
for i in range(test_time):
|
||
|
|
self.flush_data("SSC1")
|
||
|
|
time.sleep(60)
|
||
|
|
|
||
|
|
# step 4, close all thread, will disconnect when exit thread
|
||
|
|
for sta_thread in sta_thread_list:
|
||
|
|
sta_thread.exit()
|
||
|
|
for sta_thread in sta_thread_list:
|
||
|
|
sta_thread.join()
|
||
|
|
# wait and make sure disconnect done
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# step 5, join AP and check
|
||
|
|
sta_num_temp = max_sta if sta_num > max_sta else sta_num
|
||
|
|
|
||
|
|
for i in range(sta_num_temp):
|
||
|
|
self.serial_write_line("SSC%d" % (i+2), "sta -C -s %s -p %s" % (ap_ssid, ap_password))
|
||
|
|
if self.check_response("SSC%d" % (i+2), "+JAP:CONNECTED", 45) is False:
|
||
|
|
self.set_result("Fail")
|
||
|
|
break
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
self.set_result("Succeed")
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
pass
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|
||
|
|
|