Files
Takiiiiiiii a377789c1f 8/26 Initial
8/26 Initial
2024-08-26 17:59:43 +08:00

346 lines
10 KiB
Bash

#!/usr/bin/env bash
# This file is part of The RetroPie Project
#
# The RetroPie Project is the legal property of its developers, whose names are
# too numerous to list here. Please refer to the COPYRIGHT.md file distributed with this source.
#
# See the LICENSE.md file at the top-level directory of this distribution and
# at https://raw.githubusercontent.com/RetroPie/RetroPie-Setup/master/LICENSE.md
#
#2019-10-21 - Script adapted for use with MiSTer FPGA project (http://misterfpga.org) by MiSterAddons (https://misteraddons.com)
#rp_module_id="wifi"
#rp_module_desc="Configure Wifi"
#rp_module_section="config"
#rp_module_flags="!x11"
## @fn printMsgs()
## @param type style of display to use - dialog, console or heading
## @param message string or array of messages to display
## @brief Prints messages in a variety of ways.
function printMsgs() {
local type="$1"
shift
if [[ "$__nodialog" == "1" && "$type" == "dialog" ]]; then
type="console"
fi
for msg in "$@"; do
[[ "$type" == "dialog" ]] && dialog --backtitle "$__backtitle" --cr-wrap --no-collapse --msgbox "$msg" 20 60 >/dev/tty2
[[ "$type" == "console" ]] && echo -e "$msg"
[[ "$type" == "heading" ]] && echo -e "\n= = = = = = = = = = = = = = = = = = = = =\n$msg\n= = = = = = = = = = = = = = = = = = = = =\n"
done
return 0
}
function _set_interface_wifi() {
local state="$1"
if [[ "$state" == "up" ]]; then
if ! ifup wlan0; then
ip link set wlan0 up
fi
elif [[ "$state" == "down" ]]; then
if ! ifdown wlan0; then
ip link set wlan0 down
fi
fi
}
function parse_wpa_supplicant() {
#create wpa_supplicant if it doesn't exist
if [[ ! -e "/media/fat/linux/wpa_supplicant.conf" ]]; then
echo "ctrl_interface=/run/wpa_supplicant\n
update_config=1\n
country=US\n
\n
network={\n
ssid=""\n
psk=""\n
}" >> /media/fat/linux/wpa_supplicant.conf
fi
dialog --backtitle "$__backtitle" --infobox "\nScanning Existing Wifi Entries ..." 5 40 >/dev/tty2
grep -i "ssid=" '/media/fat/linux/wpa_supplicant.conf' > 'media/fat/linux/wpa_supplicant_tmp'
sed -i 's/\tssid="//' 'media/fat/linux/wpa_supplicant_tmp'
sed -i 's/"//' 'media/fat/linux/wpa_supplicant_tmp'
}
function remove_wifi() {
#dialog --backtitle "$__backtitle" --infobox "\nScanning Existing Wifi Entries ..." 5 40 >/dev/tty2
#list_wpa_supplicants
#sed -i '/$wpa_config//d' "/media/fat/linux/wpa_supplicant.conf"
sed -i '/network={/,/}/d' "/media/fat/linux/wpa_supplicant.conf"
#_set_interface_wifi down 2>/dev/null
}
function add_wifi() {
cat >> "/media/fat/linux/wpa_supplicant.conf" <<_EOF_
network={
$wpa_config
}
_EOF_
}
function list_wifi() {
local line
local essid
local type
while read line; do
[[ "$line" =~ ^Cell && -n "$essid" ]] && echo -e "$essid\n$type"
[[ "$line" =~ ^ESSID ]] && essid=$(echo "$line" | cut -d\" -f2)
[[ "$line" == "Encryption key:off" ]] && type="open"
[[ "$line" == "Encryption key:on" ]] && type="wep"
[[ "$line" =~ ^IE:.*WPA ]] && type="wpa"
done < <(iwlist wlan0 scan | grep -o "Cell .*\|ESSID:\".*\"\|IE: .*WPA\|Encryption key:.*")
echo -e "$essid\n$type"
}
function list_wpa_supplicants() {
local line
local essids=()
local essid
# while read line; do
# [[ "$line" =~ ssid= $essid ]] && echo -e "$essid\n$type"
# echo $options
# echo $essid
# echo $type
# essids+=("$essid")
# types+=("$type")
# options+=("$i" "$essid")
# ((i++))
# done < <(grep -o "ssid=.*" '/media/fat/linux/wpa_supplicant.conf')
# echo -e "$essid\n$type"
# local cmd=(dialog --backtitle "$__backtitle" --menu "Please choose the network you would like to remove" 22 76 16)
# choice=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty2)
# [[ -z "$choice" ]] && return
# essid=${essids[choice]}
# type=${types[choice]}
# echo $type
# echo $essid
# echo $key
# create_config_wifi "$type" "$essid" "$key"
}
function parse_wpa_supplicants() {
local essids=()
local essid
#local options=()
i=0
_set_interface_wifi up 2>/dev/null
sleep 1
while read essid; read type; do
essids+=("$essid")
types+=("$type")
options+=("$i" "$essid")
((i++))
done < <(list_wpa_supplicants)
local cmd=(dialog --backtitle "$__backtitle" --menu "Please choose the network you would like to remove" 22 76 16)
choice=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty2)
[[ -z "$choice" ]] && return
essid=${essids[choice]}
type=${types[choice]}
echo $type
echo $essid
echo $key
create_config_wifi "$type" "$essid" "$key"
}
function connect_wifi() {
if [[ ! -d "/sys/class/net/wlan0/" ]]; then
printMsgs "dialog" "No wlan0 interface detected"
return 1
fi
dialog --backtitle "$__backtitle" --infobox "\nSearching for WiFi networks......" 5 40 >/dev/tty2
local essids=()
local essid
local types=()
local type
local options=()
i=0
_set_interface_wifi up 2>/dev/null
sleep 1
while read essid; read type; do
essids+=("$essid")
types+=("$type")
options+=("$i" "$essid")
((i++))
done < <(list_wifi)
options+=("H" "Hidden ESSID")
local cmd=(dialog --backtitle "$__backtitle" --menu "Please choose the network you would like to connect to" 22 76 16)
choice=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty2)
[[ -z "$choice" ]] && return
local hidden=0
if [[ "$choice" == "H" ]]; then
cmd=(dialog --backtitle "$__backtitle" --inputbox "Please enter the ESSID" 10 60)
essid=$("${cmd[@]}" 2>&1 >/dev/tty2)
[[ -z "$essid" ]] && return
cmd=(dialog --backtitle "$__backtitle" --nocancel --menu "Please choose the WiFi type" 12 40 6)
options=(
wpa "WPA/WPA2"
wep "WEP"
open "Open"
)
type=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty2)
hidden=1
else
essid=${essids[choice]}
type=${types[choice]}
fi
if [[ "$type" == "wpa" || "$type" == "wep" ]]; then
local key=""
cmd=(dialog --backtitle "$__backtitle" --insecure --passwordbox "Please enter the WiFi key/password for $essid" 10 63)
local key_ok=0
while [[ $key_ok -eq 0 ]]; do
key=$("${cmd[@]}" 2>&1 >/dev/tty2) || return
key_ok=1
if [[ ${#key} -lt 8 || ${#key} -gt 63 ]] && [[ "$type" == "wpa" ]]; then
printMsgs "dialog" "Password must be between 8 and 63 characters"
key_ok=0
fi
if [[ -z "$key" && $type == "wep" ]]; then
printMsgs "dialog" "Password cannot be empty"
key_ok=0
fi
done
fi
remove_wifi
create_config_wifi "$type" "$essid" "$key"
gui_connect_wifi
}
function create_config_wifi() {
local type="$1"
local essid="$2"
local key="$3"
local wpa_config
wpa_config+="\tssid=\"$essid\"\n"
case $type in
wpa)
wpa_config+="\tpsk=\"$key\"\n"
;;
wep)
wpa_config+="\tkey_mgmt=NONE\n"
wpa_config+="\twep_tx_keyidx=0\n"
wpa_config+="\twep_key0=$key\n"
;;
open)
wpa_config+="\tkey_mgmt=NONE\n"
;;
esac
[[ $hidden -eq 1 ]] && wpa_config+="\tscan_ssid=1\n"
wpa_config=$(echo -e "$wpa_config")
cat >> "/media/fat/linux/wpa_supplicant.conf" <<_EOF_
network={
$wpa_config
}
_EOF_
}
function gui_connect_wifi() {
_set_interface_wifi down 2>/dev/null
_set_interface_wifi up 2>/dev/null
# BEGIN workaround for dhcpcd trigger failure on Raspbian stretch
systemctl restart dhcpcd &>/dev/null
# END workaround
dialog --backtitle "$__backtitle" --infobox "\nConnecting ..." 5 40 >/dev/tty2
local id=""
i=0
while [[ -z "$id" && $i -lt 30 ]]; do
sleep 1
id=$(iwgetid -r)
((i++))
dialog --backtitle "$__backtitle" --infobox "\nSaving ..." 5 40 >/dev/tty2
done
if [[ -z "$id" ]]; then
printMsgs "dialog" "Unable to connect to network $essid"
_set_interface_wifi down 2>/dev/null
fi
sleep 5
}
function _check_country_wifi() {
[[ ! -f /media/fat/linux/wpa_supplicant.conf ]] && return
iniConfig "=" "" /media/fat/linux/wpa_supplicant.conf
iniGet "country"
if [[ -z "$ini_value" ]]; then
dialog --backtitle "$__backtitle" --infobox "You don't currently have your WiFi country set in /media/fat/linux/wpa_supplicant.conf\n\nPlease run 'Set Timezone'." 5 40 >/dev/tty2
fi
}
function gui_wifi() {
_check_country_wifi
local default
while true; do
local ip_current="$(getIPAddress)"
local ip_wlan="$(getIPAddress wlan0)"
local cmd=(dialog --backtitle "$__backtitle" --cancel-label "Exit" --item-help --help-button --default-item "$default" --menu "Configure WiFi\nCurrent IP: ${ip_current:-(unknown)}\nWireless IP: ${ip_wlan:-(unknown)}\nWireless ESSID: $(iwgetid -r)" 22 76 16)
local options=(
1 "Connect to WiFi network"
"1 Connect to your WiFi network"
2 "Remove WiFi network"
"2 Remove WiFi network configuration(s)"
3 "Enable WiFi"
"3 Enable WiFi network adapter"
4 "Disable WiFi"
"4 Disable WiFi network adapter"
)
local choice=$("${cmd[@]}" "${options[@]}" 2>&1 >/dev/tty2)
if [[ "${choice[@]:0:4}" == "HELP" ]]; then
choice="${choice[@]:5}"
default="${choice/%\ */}"
choice="${choice#* }"
printMsgs "dialog" "$choice"
continue
fi
default="$choice"
if [[ -n "$choice" ]]; then
case "$choice" in
1)
connect_wifi
;;
2)
dialog --defaultno --yesno "This will remove the WiFi configuration and stop the WiFi.\n\nAre you sure you want to continue ?" 12 35 2>&1 >/dev/tty
[[ $? -ne 0 ]] && continue
remove_wifi
;;
3)
_set_interface_wifi up 2>/dev/null
;;
4) _set_interface_wifi down 2>/dev/null
;;
esac
else
break
fi
done
}
#show_wpa_supplicants
sleep 5
connect_wifi
#gui_wifi
clear