161 lines
6.9 KiB
Python
161 lines
6.9 KiB
Python
|
import time
|
||
|
|
import random
|
||
|
|
import threading
|
||
|
|
|
||
|
|
from TCAction import TCActionBase
|
||
|
|
from NativeLog import NativeLog
|
||
|
|
|
||
|
|
|
||
|
|
class StableCase1(TCActionBase.CommonTCActionBase):
|
||
|
|
|
||
|
|
def __init__(self, name, test_env, cmd_set, timeout=30, log_path=TCActionBase.LOG_PATH):
|
||
|
|
TCActionBase.CommonTCActionBase.__init__(self, name, test_env, cmd_set=cmd_set,
|
||
|
|
timeout=timeout, log_path=log_path)
|
||
|
|
# load param from excel
|
||
|
|
for i in range(1, len(cmd_set)):
|
||
|
|
if cmd_set[i][0] != "dummy":
|
||
|
|
cmd_string = "self." + cmd_set[i][0]
|
||
|
|
exec cmd_string
|
||
|
|
self.exit_event = threading.Event()
|
||
|
|
self.result_cntx = TCActionBase.ResultCheckContext(self, test_env, self.tc_name)
|
||
|
|
pass
|
||
|
|
|
||
|
|
def check_wifi_status(self, data):
|
||
|
|
if data.find("+JAP:DISCONNECTED") != -1:
|
||
|
|
self.exit_event.set()
|
||
|
|
NativeLog.add_trace_critical("[Wifi] Disconnected")
|
||
|
|
pass
|
||
|
|
|
||
|
|
def execute(self):
|
||
|
|
TCActionBase.TCActionBase.execute(self)
|
||
|
|
self.result_cntx.start()
|
||
|
|
|
||
|
|
try:
|
||
|
|
# configurable params
|
||
|
|
# target role
|
||
|
|
target_role = self.target_role
|
||
|
|
# enable tcp send/recv
|
||
|
|
tcp_enable = self.tcp_enable
|
||
|
|
# enable udp send/recv
|
||
|
|
udp_enable = self.udp_enable
|
||
|
|
# enable ping
|
||
|
|
ping_enable = self.ping_enable
|
||
|
|
# delay range
|
||
|
|
delay_range = self.delay_range
|
||
|
|
# test time in hours
|
||
|
|
test_time = self.test_time * 3600
|
||
|
|
# configurable params
|
||
|
|
except StandardError, e:
|
||
|
|
NativeLog.add_trace_critical("Error configuration for TCPTransparent script, error is %s" % e)
|
||
|
|
raise StandardError("Error configuration")
|
||
|
|
|
||
|
|
if target_role == "AP":
|
||
|
|
pc_ip = "<pc_ip_wifi>"
|
||
|
|
target_ip = "<target_ap_ip>"
|
||
|
|
elif target_role == "STA":
|
||
|
|
pc_ip = "<pc_ip>"
|
||
|
|
target_ip = "<target_ip>"
|
||
|
|
else:
|
||
|
|
raise StandardError("target role only support AP or STA")
|
||
|
|
|
||
|
|
# step 1, create UDP socket and TCP server
|
||
|
|
checker_stings = ["R SSC1 A <udp_socket>:BIND:(\d+),OK"]
|
||
|
|
test_action_string = ["SSC SSC1 soc -B -t UDP -p <test_udp_port1>"]
|
||
|
|
fail_string = "Fail, Fail to create UDP socket"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
return
|
||
|
|
|
||
|
|
checker_stings = ["R SSC1 A <tcp_server>:BIND:(\d+),OK"]
|
||
|
|
test_action_string = ["SSC SSC1 soc -B -t TCP -p <random_port>"]
|
||
|
|
fail_string = "Fail, Fail to create tcp server"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
return
|
||
|
|
|
||
|
|
checker_stings = ["R SSC1 RE LISTEN:(\d+),OK"]
|
||
|
|
test_action_string = ["SSC SSC1 soc -L -s <tcp_server>"]
|
||
|
|
fail_string = "Fail, Fail to listen"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
return
|
||
|
|
|
||
|
|
# step 2, PC connect to 8266 tcp server, PC create UDP socket
|
||
|
|
checker_stings = ["R SOC_COM C OK"]
|
||
|
|
test_action_string = ["SOC SOC1 BIND <test_udp_port1> %s" % pc_ip]
|
||
|
|
fail_string = "Fail, Fail to create udp socket on PC"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
return
|
||
|
|
|
||
|
|
checker_stings = ["P SSC1 A <tcp_socket>:ACCEPT:(\d+),\d+", "P SOC_COM C OK"]
|
||
|
|
test_action_string = ["SOC SOC2 CONNECT <random_port> %s 0 %s" % (target_ip, pc_ip)]
|
||
|
|
fail_string = "Fail, Fail to create tcp socket on PC"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
return
|
||
|
|
|
||
|
|
start_time = time.time()
|
||
|
|
total_test_count = ping_fail_count = tcp_fail_count = udp_fail_count = 0
|
||
|
|
|
||
|
|
# step 3, start do tcp/udp/ping
|
||
|
|
while time.time() - start_time < test_time and self.exit_event.isSet() is False:
|
||
|
|
total_test_count += 1
|
||
|
|
if ping_enable is True:
|
||
|
|
# do ping
|
||
|
|
checker_stings = ["P PC_COM RE \+PING:\d+ms"]
|
||
|
|
test_action_string = ["PING %s -n 1 -w 1000" % target_ip]
|
||
|
|
fail_string = "Fail, Fail to ping target"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
ping_fail_count += 1
|
||
|
|
NativeLog.add_prompt_trace("[ping fail] fail/total = %s/%s"
|
||
|
|
% (ping_fail_count, total_test_count))
|
||
|
|
pass
|
||
|
|
|
||
|
|
data_len = random.randint(1, 1460)
|
||
|
|
|
||
|
|
if tcp_enable is True:
|
||
|
|
# do tcp send/recv
|
||
|
|
checker_stings = ["P SSC1 SL <tcp_socket>+%s" % data_len, "P SOC2 RL %s" % data_len]
|
||
|
|
test_action_string = ["SSC SSC1 soc -S -s <tcp_socket> -l %s" % data_len,
|
||
|
|
"SOC SOC2 SEND %s" % data_len]
|
||
|
|
fail_string = "Fail, Fail to send/recv tcp"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string) is False:
|
||
|
|
tcp_fail_count += 1
|
||
|
|
NativeLog.add_prompt_trace("[tcp fail] fail/total = %s/%s"
|
||
|
|
% (tcp_fail_count, total_test_count))
|
||
|
|
# tcp fail, break
|
||
|
|
self.exit_event.set()
|
||
|
|
pass
|
||
|
|
|
||
|
|
if udp_enable is True:
|
||
|
|
# do udp send/recv
|
||
|
|
checker_stings = ["P SSC1 SL <udp_socket>+%s" % data_len, "P SOC1 RL %s" % data_len]
|
||
|
|
test_action_string = ["SSC SSC1 soc -S -s <udp_socket> "
|
||
|
|
"-i %s -p <test_udp_port1> -l %s" % (pc_ip, data_len),
|
||
|
|
"SOC SOC1 SENDTO %s <test_udp_port1> %s" % (data_len, target_ip)]
|
||
|
|
fail_string = "Fail, Fail to sendto/recvfrom udp"
|
||
|
|
if self.load_and_exe_one_step(checker_stings, test_action_string, fail_string, check_time=20) is False:
|
||
|
|
udp_fail_count += 1
|
||
|
|
NativeLog.add_prompt_trace("[udp fail] fail/total = %s/%s"
|
||
|
|
% (udp_fail_count, total_test_count))
|
||
|
|
pass
|
||
|
|
|
||
|
|
# sleep
|
||
|
|
time.sleep(random.randint(delay_range[0], delay_range[1]))
|
||
|
|
pass
|
||
|
|
|
||
|
|
# finally, execute done
|
||
|
|
if self.exit_event.isSet() is False:
|
||
|
|
self.result_cntx.set_result("Succeed")
|
||
|
|
|
||
|
|
def result_check(self, port_name, data):
|
||
|
|
self.result_cntx.append_data(port_name, data)
|
||
|
|
if port_name != "SOC1" and port_name != "SOC2":
|
||
|
|
# socket received data do not need to be logged
|
||
|
|
TCActionBase.CommonTCActionBase.result_check(self, port_name, data)
|
||
|
|
if port_name == "SSC1":
|
||
|
|
self.check_wifi_status(data)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
pass
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|