freebsd-src/usr.sbin/bsdinstall/scripts/wlanconfig
Renato Botelho 55d2990548 bsdinstall: Fix wifi network selection size
Use correct variable while creating dialog used to select among
available wireless networks

Approved by:	asiciliano
Sponsored by:	Rubicon Communications, LLC ("Netgate")
Differential Revision:	https://reviews.freebsd.org/D45271
2024-05-20 13:39:29 -03:00

364 lines
11 KiB
Bash
Executable file

#!/bin/sh
#-
# Copyright (c) 2011 Nathan Whitehorn
# Copyright (c) 2013-2020 Devin Teske
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#
#
############################################################ INCLUDES
BSDCFG_SHARE="/usr/share/bsdconfig"
. $BSDCFG_SHARE/common.subr || exit 1
f_include $BSDCFG_SHARE/dialog.subr
f_dialog_backtitle "$OSNAME Installer"
############################################################ FUNCTIONS
country_set()
{
local error_str iface_up ifconfig_args=
#
# Setup what was selected
# NB: Do not change order of arguments (or regdomain will be ignored)
#
[ "$2" ] && ifconfig_args="$ifconfig_args country $2"
[ "$1" ] && ifconfig_args="$ifconfig_args regdomain $1"
[ "$ifconfig_args" ] || return $SUCCESS # Nothing to do
ifconfig_args="${ifconfig_args# }"
# Regdomain/country cannot be applied while interface is running
iface_up=$( ifconfig -lu | grep -w "$WLAN_IFACE" )
[ "$iface_up" ] && ifconfig "$WLAN_IFACE" down
f_eval_catch -dk error_str wlanconfig ifconfig "ifconfig %s %s" \
"$WLAN_IFACE" "$ifconfig_args"
error_str="${error_str#ifconfig: }"
# Restart wpa_supplicant(8) (should not fail).
[ "$iface_up" ] && f_eval_catch -d wlanconfig wpa_supplicant \
'wpa_supplicant -B -i "%s" -c "%s/wpa_supplicant.conf"' \
"$WLAN_IFACE" "$BSDINSTALL_TMPETC"
if [ "$error_str" ]; then
$DIALOG --title "$msg_error" \
--backtitle "$DIALOG_BACKTITLE" \
--yes-label Change \
--no-label Ignore \
--yesno \
"Error while applying chosen settings ($error_str)" \
0 0 || return $SUCCESS # Skip
return $FAILURE # Restart
else
cat > "$BSDINSTALL_TMPETC/rc.conf.net.wlan" <<-EOF
create_args_$WLAN_IFACE="$ifconfig_args"
EOF
fi
return $SUCCESS
}
dialog_country_select()
{
local input regdomains countries regdomain country prompt
local no_default="<not selected>"
local default_regdomain="${1:-$no_default}"
local default_country="${2:-$no_default}"
#
# Parse available countries/regdomains
#
input=$( ifconfig "$WLAN_IFACE" list countries | sed -e 's/DEBUG//gi' )
regdomains=$( echo "$input" | awk '
sub(/.*domains:/, ""), /[^[:alnum:][[:space:]]/ {
n = split($0, domains)
for (i = 1; i <= n; i++)
printf "'\''%s'\'' '\'\''", domains[i]
}
' | sort )
countries=$( echo "$input" | awk '
sub(/Country codes:/, ""), sub(/Regulatory.*/, "") {
while (match($0, /[[:upper:]][[:upper:][:digit:]] /)) {
country = substr($0, RSTART)
sub(/ [[:upper:]][[:upper:][:digit:]].*/, "",
country)
code = substr(country, 1, 2)
desc = substr(country, 4)
sub(/[[:space:]]*$/, "", desc)
printf "'\''%s'\'' '\''%s'\''\n", code, desc
$0 = substr($0, RSTART + RLENGTH)
}
}
' | sort )
f_dialog_title "Regdomain selection"
prompt="Select your regdomain."
eval f_dialog_menu_size height width rows \
\"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
\"\$prompt\" \"\" $regdomains
regdomain=$( eval $DIALOG \
--title \"\$DIALOG_TITLE\" \
--backtitle \"\$DIALOG_BACKTITLE\" \
--cancel-label \"\$msg_skip\" \
--default-item \"\$default_regdomain\" \
--menu \"\$prompt\" \
$height $width $rows \
$regdomains \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
)
f_dialog_data_sanitize regdomain
f_dialog_title "Country selection"
prompt="Select your country."
eval f_dialog_menu_size height width rows \
\"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
\"\$prompt\" \"\" $countries
country=$( eval $DIALOG \
--title \"\$DIALOG_TITLE\" \
--backtitle \"\$DIALOG_BACKTITLE\" \
--cancel-label \"\$msg_skip\" \
--default-item \"\$default_country\" \
--menu \"\$prompt\" \
$height $width $rows \
$countries \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
)
f_dialog_data_sanitize country
country_set "$regdomain" "$country"
}
############################################################ MAIN
: > "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
chmod 0600 "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<EOF
ctrl_interface=/var/run/wpa_supplicant
eapol_version=2
ap_scan=1
fast_reauth=1
EOF
#
# Try to reach wpa_supplicant. If it isn't running and we can modify the
# existing system, start it. Otherwise, fail.
#
if ! f_eval_catch -d wlanconfig wpa_cli "wpa_cli ping"; then
if [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
f_show_err "Wireless cannot be configured without %s" \
"making changes to the local system!"
exit 1
fi
f_eval_catch wlanconfig wpa_supplicant \
'wpa_supplicant -B -i "%s" -c "%s/wpa_supplicant.conf"' \
"$1" "$BSDINSTALL_TMPETC" || exit 1
# See if we succeeded
f_eval_catch wlanconfig wpa_cli "wpa_cli ping" || exit 1
fi
#
# There is no way to check country/regdomain without (possible)
# interface state modification
#
if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
# Get current country/regdomain for selected interface
WLAN_IFACE=$( wpa_cli ifname | tail -n 1 )
INPUT=$( ifconfig "$WLAN_IFACE" list regdomain | head -n 1 )
DEF_REGDOMAIN=$( echo "$INPUT" | cut -w -f 2 )
DEF_COUNTRY=$( echo "$INPUT" | cut -w -f 4 )
[ "$DEF_REGDOMAIN" = 0 ] && DEF_REGDOMAIN="<not selected>"
[ "$DEF_COUNTRY" = 0 ] && DEF_COUNTRY="<not selected>"
f_dialog_title "Regdomain/country"
if f_yesno "Change regdomain/country ($DEF_REGDOMAIN/$DEF_COUNTRY)?"
then
while ! dialog_country_select "$DEF_REGDOMAIN" "$DEF_COUNTRY"
do :; done
fi
fi
while :; do
SCANSSID=0
f_eval_catch -d wlanconfig wpa_cli "wpa_cli scan"
f_dialog_title "Scanning"
f_dialog_pause "Waiting 5 seconds to scan for wireless networks..." 5 ||
exit 1
f_eval_catch -dk SCAN_RESULTS wlanconfig wpa_cli "wpa_cli scan_results"
NETWORKS=$( echo "$SCAN_RESULTS" | awk -F '\t' '
/..:..:..:..:..:../ && $5 { printf "\"%s\"\t\"%s\"\n", $5, $4 }
' | sort | uniq )
if [ ! "$NETWORKS" ]; then
f_dialog_title "$msg_error"
f_yesno "No wireless networks were found. Rescan?" && continue
else
f_dialog_title "Network Selection"
prompt="Select a wireless network to connect to."
f_dialog_menu_size height width rows "$DIALOG_TITLE" \
"$DIALOG_BACKTITLE" "$prompt" "" $NETWORKS
NETWORK=$( eval $DIALOG \
--title \"\$DIALOG_TITLE\" \
--backtitle \"\$DIALOG_BACKTITLE\" \
--extra-button \
--extra-label \"Rescan\" \
--menu \"\$prompt\" \
$height $width $rows \
$NETWORKS \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
)
fi
retval=$?
f_dialog_data_sanitize NETWORK
case $retval in
$DIALOG_OK) break ;;
$DIALOG_CANCEL)
# Ask if the user wants to select network manually
f_dialog_title "Network Selection"
f_yesno "Do you want to select the network manually?" || exit 1
f_dialog_input NETWORK "Enter SSID" || exit 1
prompt="Select encryption type"
menu_list="
'1 WPA/WPA2 PSK' ''
'2 WPA/WPA2 EAP' ''
'3 WEP' ''
'0 None' ''
" # END-QUOTE
eval f_dialog_menu_size height width rows \"\$DIALOG_TITLE\" \
\"\$DIALOG_BACKTITLE\" \"\$prompt\" \"\" $menu_list
ENCRYPTION=$( eval $DIALOG \
--title \"\$DIALOG_TITLE\" \
--backtitle \"\$DIALOG_BACKTITLE\" \
--menu \"\$prompt\" \
$height $width $rows \
$menu_list \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
) || exit 1
SCANSSID=1
break
;;
$DIALOG_EXTRA) # Rescan
;;
esac
done
[ "$ENCRYPTION" ] || ENCRYPTION=$( echo "$NETWORKS" |
awk -F '\t' "/^\"$NETWORK\"\t/ { print \$2 }" )
if echo "$ENCRYPTION" | grep -q PSK; then
PASS=$( $DIALOG \
--title "WPA Setup" \
--backtitle "$DIALOG_BACKTITLE" \
--insecure \
--mixedform "" \
0 0 0 \
"SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
"Password" 2 0 "" 2 12 15 63 1 \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
) || exec "$0" "$@"
awk 'sub(/^\\/,"")||1' \
>> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
network={
\ ssid="$NETWORK"
\ scan_ssid=$SCANSSID
\ psk="$PASS"
\ priority=5
}
EOF
elif echo "$ENCRYPTION" | grep -q EAP; then
USERPASS=$( $DIALOG \
--title "WPA-Enterprise Setup" \
--backtitle "$DIALOG_BACKTITLE" \
--insecure \
--mixedform "" \
0 0 0 \
"SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
"Username" 2 0 "" 2 12 25 63 0 \
"Password" 3 0 "" 3 12 25 63 1 \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
) || exec "$0" "$@"
awk 'sub(/^\\/,"")||1' \
>> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
network={
\ ssid="$NETWORK"
\ scan_ssid=$SCANSSID
\ key_mgmt=WPA-EAP$(
echo "$USERPASS" | awk '
NR == 1 { printf "\n\tidentity=\"%s\"", $1 }
NR == 2 { printf "\n\tpassword=\"%s\"", $1 }
' )
\ priority=5
}
EOF
elif echo "$ENCRYPTION" | grep -q WEP; then
WEPKEY=$( $DIALOG \
--title "WEP Setup" \
--backtitle "$DIALOG_BACKTITLE" \
--insecure \
--mixedform "" \
0 0 0 \
"SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
"WEP Key 0" 2 0 "" 2 12 15 0 1 \
2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
) || exec "$0" "$@"
awk 'sub(/^\\/,"")||1' \
>> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
network={
\ ssid="$NETWORK"
\ scan_ssid=$SCANSSID
\ key_mgmt=NONE
\ wep_key0="$WEPKEY"
\ wep_tx_keyidx=0
\ priority=5
}
EOF
else # Open
awk 'sub(/^\\/,"")||1' \
>> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
network={
\ ssid="$NETWORK"
\ scan_ssid=$SCANSSID
\ key_mgmt=NONE
\ priority=5
}
EOF
fi
# Connect to any open networks policy
cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<EOF
network={
priority=0
key_mgmt=NONE
}
EOF
# Bring up new network
[ "$BSDINSTALL_CONFIGCURRENT" ] &&
f_eval_catch -d wlanconfig wpa_cli "wpa_cli reconfigure"
exit $SUCCESS
################################################################################
# END
################################################################################