mirror of
https://github.com/opnsense/src.git
synced 2026-03-14 22:53:16 -04:00
Allow to setup country and/or regulatory domain before scanning (since channel list is affected by regulatory restrictions this may impact scan results). PR: 182600 (originally) Differential Revision: https://reviews.freebsd.org/D6325
326 lines
9.6 KiB
Bash
Executable file
326 lines
9.6 KiB
Bash
Executable file
#!/bin/sh
|
|
#-
|
|
# Copyright (c) 2011 Nathan Whitehorn
|
|
# Copyright (c) 2013-2015 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.
|
|
#
|
|
# $FreeBSD$
|
|
#
|
|
############################################################ INCLUDES
|
|
|
|
BSDCFG_SHARE="/usr/share/bsdconfig"
|
|
. $BSDCFG_SHARE/common.subr || exit 1
|
|
f_include $BSDCFG_SHARE/dialog.subr
|
|
|
|
############################################################ MAIN
|
|
|
|
echo -n > $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
chmod 0600 $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
|
|
echo "ctrl_interface=/var/run/wpa_supplicant" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo "eapol_version=2" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo "ap_scan=1" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo "fast_reauth=1" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
|
|
# Try to reach wpa_supplicant. If it isn't running and we can modify the
|
|
# existing system, start it. Otherwise, fail.
|
|
(wpa_cli ping >/dev/null 2>/dev/null || ([ ! -z $BSDINSTALL_CONFIGCURRENT ] && \
|
|
wpa_supplicant -B -i $1 -c $BSDINSTALL_TMPETC/wpa_supplicant.conf)) || \
|
|
(dialog --backtitle "FreeBSD Installer" --title "Error" --msgbox \
|
|
"Could not start wpa_supplicant!" 0 0; exit 1) || exit 1
|
|
|
|
# See if we succeeded
|
|
wpa_cli ping >/dev/null 2>/dev/null
|
|
if [ $? -ne 0 -a -z $BSDINSTALL_CONFIGCURRENT ]; then
|
|
dialog --backtitle "FreeBSD Installer" --title "Error" --msgbox \
|
|
"Wireless cannot be configured without making changes to the local system!" \ 0 0
|
|
exit 1
|
|
fi
|
|
|
|
country_set()
|
|
{
|
|
local error_str=
|
|
local iface_up=
|
|
local ifconfig_args=
|
|
|
|
# Setup what was selected
|
|
# NB: do not change order of arguments (or regdomain will be ignored)
|
|
if [ ! -z "$2" ]; then
|
|
ifconfig_args="${ifconfig_args}country $2"
|
|
fi
|
|
if [ ! -z "$1" ]; then
|
|
if [ ! -z "$2" ]; then
|
|
ifconfig_args="${ifconfig_args} "
|
|
fi
|
|
ifconfig_args="${ifconfig_args}regdomain $1"
|
|
fi
|
|
if [ -z "$ifconfig_args" ]; then
|
|
# Nothing to do (everything was skipped)
|
|
return $SUCCESS
|
|
fi
|
|
|
|
# Regdomain/country cannot be applied while interface is running
|
|
iface_up=`ifconfig -lu | grep -w $WLAN_IFACE`
|
|
if [ ! -z "$iface_up" ]; then
|
|
ifconfig $WLAN_IFACE down
|
|
fi
|
|
error_str=`ifconfig $WLAN_IFACE $ifconfig_args 2>&1 | \
|
|
sed 's/ifconfig: //'`
|
|
if [ ! -z "$iface_up" ]; then
|
|
# Restart wpa_supplicant(8) (should not fail).
|
|
wpa_supplicant -B -i $WLAN_IFACE -c \
|
|
$BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
fi
|
|
if [ ! -z "$error_str" ]; then
|
|
dialog --backtitle "FreeBSD Installer" --title "Error" \
|
|
--yes-label Change --no-label Ignore --yesno \
|
|
"Error while applying chosen settings ($error_str)" 0 0
|
|
if [ $? -eq $DIALOG_OK ]; then
|
|
return $FAILURE # Restart
|
|
else
|
|
return $SUCCESS # Skip
|
|
fi
|
|
else
|
|
: > $BSDINSTALL_TMPETC/rc.conf.net.wlan
|
|
echo create_args_$WLAN_IFACE=\"$ifconfig_args\" >> \
|
|
$BSDINSTALL_TMPETC/rc.conf.net.wlan
|
|
fi
|
|
|
|
return $SUCCESS
|
|
}
|
|
|
|
dialog_country_select()
|
|
{
|
|
local input=
|
|
local def_item_regdomain=
|
|
local def_item_country=
|
|
local regdomains=
|
|
local countries=
|
|
local regdomain=
|
|
local country=
|
|
|
|
# Parse available countries/regdomains
|
|
input=`ifconfig $WLAN_IFACE list countries | sed 's/DEBUG//gi'`
|
|
regdomains=`echo $input | sed 's/.*domains://' | tr ' ' '\n' | \
|
|
sort | tr '\n' ' '`
|
|
countries=`echo $input | sed 's/Country codes://' | \
|
|
sed 's/Regulatory.*//' | awk '{
|
|
for (i = 1; i <= NF; i++) {
|
|
printf "%s", $i
|
|
if (match($i, "[[:lower:]]"))
|
|
if (match($(i+1), "[[:lower:]]"))
|
|
printf "\\\\\\ "
|
|
else
|
|
printf "\n"
|
|
else
|
|
printf " "
|
|
}
|
|
}' | sort -k 2 | tr '\n' ' '`
|
|
|
|
# Change default cursor position (if required).
|
|
if [ "$1" != "<not selected>" ]; then
|
|
def_item_regdomain="--default-item $1"
|
|
fi
|
|
if [ "$2" != "<not selected>" ]; then
|
|
def_item_country="--default-item $2"
|
|
fi
|
|
|
|
f_dialog_menu_size height width rows \"Regdomain selection\" \
|
|
\"FreeBSD Installer\" \"Select your regdomain.\" \
|
|
\"\" $regdomains
|
|
regdomain=`sh -c "dialog \
|
|
--backtitle \"FreeBSD Installer\" \
|
|
--title \"Regdomain selection\" \
|
|
--cancel-label \"Skip\" \
|
|
$def_item_regdomain \
|
|
--no-items \
|
|
--stdout \
|
|
--menu \"Select your regdomain.\" \
|
|
$height $width $rows $regdomains"`
|
|
|
|
f_dialog_menu_size height width rows \"Country selection\" \
|
|
\"FreeBSD Installer\" \"Select your country.\" \
|
|
\"\" $countries
|
|
country=`sh -c "dialog \
|
|
--backtitle \"FreeBSD Installer\" \
|
|
--title \"Country selection\" \
|
|
--cancel-label \"Skip\" \
|
|
$def_item_country \
|
|
--stdout \
|
|
--menu \"Select your country.\" \
|
|
$height $width $rows $countries"`
|
|
|
|
country_set "$regdomain" "$country"
|
|
|
|
return $?
|
|
}
|
|
|
|
# There is no way to check country/regdomain without (possible)
|
|
# interface state modification
|
|
if [ ! -z $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`
|
|
if [ "$DEF_REGDOMAIN" = "0" ]; then
|
|
DEF_REGDOMAIN="<not selected>"
|
|
fi
|
|
DEF_COUNTRY=`echo $INPUT | cut -w -f 4`
|
|
if [ "$DEF_COUNTRY" = "0" ]; then
|
|
DEF_COUNTRY="<not selected>"
|
|
fi
|
|
dialog --backtitle "FreeBSD Installer" --title "Regdomain/country" \
|
|
--yesno "Change regdomain/country (now \
|
|
$DEF_REGDOMAIN/$DEF_COUNTRY)?" 0 0
|
|
if [ $? -eq 0 ]; then
|
|
while :
|
|
do
|
|
dialog_country_select "$DEF_REGDOMAIN" "$DEF_COUNTRY"
|
|
if [ $? -eq $SUCCESS ]; then
|
|
break
|
|
fi
|
|
done
|
|
fi
|
|
fi
|
|
|
|
while :
|
|
do
|
|
output=$( wpa_cli scan 2>&1 )
|
|
f_dprintf "%s" "$output"
|
|
dialog --backtitle "FreeBSD Installer" --title "Scanning" \
|
|
--ok-label "Skip" \
|
|
--pause "Waiting 5 seconds to scan for wireless networks..." \
|
|
9 40 5 || exit 1
|
|
|
|
SCAN_RESULTS=`wpa_cli scan_results`
|
|
NETWORKS=`echo "$SCAN_RESULTS" | awk -F '\t' \
|
|
'/..:..:..:..:..:../ {if (length($5) > 0) \
|
|
printf("\"%s\"\t%s\n", $5, $4);}' | sort | uniq`
|
|
|
|
if [ -z "$NETWORKS" ]; then
|
|
dialog --backtitle "FreeBSD Installer" --title "Error" \
|
|
--yesno "No wireless networks were found. Rescan?" 0 0 && \
|
|
continue
|
|
exit 1
|
|
fi
|
|
|
|
exec 3>&1
|
|
NETWORK=`sh -c "dialog --extra-button --extra-label \"Rescan\" \
|
|
--backtitle \"FreeBSD Installer\" --title \"Network Selection\" \
|
|
--menu \"Select a wireless network to connect to.\" 0 0 0 \
|
|
$(echo $NETWORKS | tr '\n' ' ')" 2>&1 1>&3`
|
|
case $? in
|
|
0) # OK
|
|
break
|
|
;;
|
|
1) # Cancel
|
|
exit 1
|
|
;;
|
|
3) # Rescan
|
|
;;
|
|
esac
|
|
exec 3>&-
|
|
done
|
|
|
|
ENCRYPTION=`echo "$NETWORKS" | awk -F '\t' \
|
|
"/^\"$NETWORK\"\t/ {printf(\"%s\n\", \\\$2 );}"`
|
|
|
|
if echo $ENCRYPTION | grep -q 'PSK'; then
|
|
exec 3>&1
|
|
PASS=`dialog --insecure --backtitle "FreeBSD Installer" \
|
|
--title "WPA Setup" --mixedform "" 0 0 0 \
|
|
"SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
|
|
"Password" 2 0 "" 2 12 15 63 1 \
|
|
2>&1 1>&3` \
|
|
|| exec $0 $@
|
|
exec 3>&-
|
|
echo "network={
|
|
ssid=\"$NETWORK\"
|
|
psk=\"$PASS\"
|
|
priority=5
|
|
}" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
elif echo $ENCRYPTION | grep -q EAP; then
|
|
exec 3>&1
|
|
USERPASS=`dialog --insecure --backtitle "FreeBSD Installer" \
|
|
--title "WPA-Enterprise Setup" --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 1>&3` \
|
|
|| exec $0 $@
|
|
exec 3>&-
|
|
echo "network={
|
|
ssid=\"$NETWORK\"
|
|
key_mgmt=WPA-EAP" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo "$USERPASS" | awk '
|
|
{
|
|
if (NR == 1) {
|
|
printf " identity=\"%s\"\n", $1;
|
|
} else if (NR == 2) {
|
|
printf " password=\"%s\"\n", $1;
|
|
}
|
|
}' >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
echo " priority=5
|
|
}" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
elif echo $ENCRYPTION | grep -q WEP; then
|
|
exec 3>&1
|
|
WEPKEY=`dialog --insecure --backtitle "FreeBSD Installer" \
|
|
--title "WEP Setup" --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 1>&3` \
|
|
|| exec $0 $@
|
|
echo "network={
|
|
ssid=\"$NETWORK\"
|
|
key_mgmt=NONE
|
|
wep_key0=\"$WEPKEY\"
|
|
wep_tx_keyidx=0
|
|
priority=5
|
|
}" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
else # Open
|
|
echo "network={
|
|
ssid=\"$NETWORK\"
|
|
key_mgmt=NONE
|
|
priority=5
|
|
}" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
fi
|
|
|
|
# Connect to any open networks policy
|
|
echo "network={
|
|
priority=0
|
|
key_mgmt=NONE
|
|
}" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
|
|
|
|
# Bring up new network
|
|
if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
|
|
output=$( wpa_cli reconfigure 2>&1 )
|
|
f_dprintf "%s" "$output"
|
|
fi
|
|
|
|
exit $SUCCESS
|
|
|
|
################################################################################
|
|
# END
|
|
################################################################################
|