139 lines
2.8 KiB
Bash
139 lines
2.8 KiB
Bash
|
#!/bin/sh
|
||
|
|
path_in_config_tree="/stress/wifi"
|
||
|
|
connect_max=`mjson_fetch $path_in_config_tree/connect_count`
|
||
|
|
connect_count=0
|
||
|
|
fail_count=0
|
||
|
|
write_wpa_conf(){
|
||
|
|
if [ "x$2" = "xNONE" ];then
|
||
|
|
key="key_mgmt=NONE"
|
||
|
|
else
|
||
|
|
key="psk=\"$2\""
|
||
|
|
fi
|
||
|
|
|
||
|
|
(cat <<EOF
|
||
|
|
ctrl_interface=/etc/wifi/sockets
|
||
|
|
disable_scan_offload=1
|
||
|
|
update_config=1
|
||
|
|
|
||
|
|
network={
|
||
|
|
ssid="$1"
|
||
|
|
scan_ssid=1
|
||
|
|
$key
|
||
|
|
priority=1
|
||
|
|
}
|
||
|
|
EOF
|
||
|
|
) >/etc/wifi/wpa_supplicant.conf
|
||
|
|
}
|
||
|
|
|
||
|
|
check_wlan(){
|
||
|
|
ifconfig -a | grep wlan0 || {
|
||
|
|
echo "===========Not Found wlan0========="
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
connect(){
|
||
|
|
ps | grep wpa_supplicant | awk '{print $5}' | grep wpa_supplicant && {
|
||
|
|
killall wpa_supplicant
|
||
|
|
sleep 1
|
||
|
|
}
|
||
|
|
|
||
|
|
ps | grep udhcpc | awk '{print $5}' | grep udhcpc && {
|
||
|
|
killall udhcpc
|
||
|
|
sleep 1
|
||
|
|
}
|
||
|
|
|
||
|
|
wpa_supplicant -iwlan0 -Dnl80211 -c/etc/wifi/wpa_supplicant.conf > /tmp/wifi.log &
|
||
|
|
|
||
|
|
# sleep 1
|
||
|
|
|
||
|
|
while true
|
||
|
|
do
|
||
|
|
sleep 0.5
|
||
|
|
let count++
|
||
|
|
grep "CTRL-EVENT-CONNECTED" /tmp/wifi.log >/dev/null && {
|
||
|
|
ip=$(/sbin/udhcpc -i wlan0 | awk '($2~/^setting/&&$3~/^default/) {print $5}');
|
||
|
|
if [ -z ${ip} ];then
|
||
|
|
exit 1
|
||
|
|
else
|
||
|
|
ttips "Test $connect_count times:Success " \
|
||
|
|
-n "Routers IP:$ip"
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
grep "pre-shared key may be incorrect" /tmp/wifi.log && {
|
||
|
|
ttips "Test $connect_count times:FAILED" \
|
||
|
|
-n "PASSWARD INCORRECT"
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
if [ $count -eq 10000 ]; then
|
||
|
|
ttips "Test $connect_count times:FAILED" \
|
||
|
|
-n "================= Connection timed out ============="
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
disconnect(){
|
||
|
|
killall wpa_supplicant
|
||
|
|
killall udhcpc
|
||
|
|
ifconfig wlan0 down
|
||
|
|
}
|
||
|
|
|
||
|
|
enter_info(){
|
||
|
|
local number=0
|
||
|
|
until [ $number -ge 8 ];do
|
||
|
|
ssid=$(task "Please enter the WIFI SSID")
|
||
|
|
password=$(task "Please enter WIFI password")
|
||
|
|
number=$(echo $password | wc -c)
|
||
|
|
|
||
|
|
[ "$password" = "NONE" ] && break
|
||
|
|
[ $number -lt 8 ] && {
|
||
|
|
ttips "The password must be greater than 8 digits,Please enter again."
|
||
|
|
}
|
||
|
|
done
|
||
|
|
write_wpa_conf $ssid $password
|
||
|
|
}
|
||
|
|
|
||
|
|
check_connect_max(){
|
||
|
|
ttrue "Connect $connect_max times, Please confirm."
|
||
|
|
[ $? = 0 ] || {
|
||
|
|
connect_max=$(task "Please enter the number of connections")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
check_wificonf(){
|
||
|
|
if [ -f /etc/wifi/wpa_supplicant.conf ];then
|
||
|
|
ssid=$(grep "\bssid=" /etc/wifi/wpa_supplicant.conf | sed 's/^[ \t]*//g')
|
||
|
|
if [ -z "$ssid" ];then
|
||
|
|
enter_info
|
||
|
|
else
|
||
|
|
password=$(grep "\bpsk=" /etc/wifi/wpa_supplicant.conf | sed 's/^[ \t]*//g')
|
||
|
|
if [ -z $password ];then
|
||
|
|
ttips "$ssid | No password"
|
||
|
|
else
|
||
|
|
ttips "$ssid | $password"
|
||
|
|
fi
|
||
|
|
|
||
|
|
ttrue "Whether to use the above default wifi configuration"
|
||
|
|
[ $? = 0 ] || enter_info
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
check_wlan
|
||
|
|
check_wificonf
|
||
|
|
check_connect_max
|
||
|
|
for i in $(seq 1 $connect_max)
|
||
|
|
do
|
||
|
|
let connect_count++
|
||
|
|
connect
|
||
|
|
done
|
||
|
|
echo "==============================================="
|
||
|
|
echo "TEST $connect_count times"
|
||
|
|
echo "Success:$((connect_count-fail_count))"
|
||
|
|
echo "Failed:$fail_count"
|
||
|
|
echo "==============================================="
|
||
|
|
exit 0
|