freebsd-nq/usr.sbin/bsdinstall/scripts/wlanconfig

#!/bin/sh
#-
# Copyright (c) 2011 Nathan Whitehorn
# Copyright (c) 2013-2016 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
f_dialog_backtitle "FreeBSD 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
	error_str=$( ifconfig "$WLAN_IFACE" $ifconfig_args 2>&1 |
		sed -e 's/ifconfig: //' )
	if [ "$iface_up" ]; then
		# Restart wpa_supplicant(8) (should not fail).
		wpa_supplicant -B -i "$WLAN_IFACE" -c \
			"$BSDINSTALL_TMPETC/wpa_supplicant.conf"
	fi
	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
		awk 'sub(/^\t\t/,"")||1' \
			> "$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 ! wpa_cli ping > /dev/null 2>&1; then
	if [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
		$DIALOG --backtitle "$DIALOG_BACKTITLE" --title "$msg_error" --msgbox \
			"Wireless cannot be configured without making changes to the local system!" 0 0
		exit 1
	fi
	wpa_supplicant -B -i $1 -c "$BSDINSTALL_TMPETC/wpa_supplicant.conf") || exit 1

	if ! wpa_cli ping > /dev/null 2>&1 && [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
		$DIALOG \
			--title "$msg_error" \
			--backtitle "$DIALOG_BACKTITLE" \
			--msgbox "Could not start wpa_supplicant!" \
			0 0
		exit 1
	fi
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 -1 )
	INPUT=$( ifconfig "$WLAN_IFACE" list regdomain | head -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"
	$DIALOG \
		--title "$DIALOG_TITLE" \
		--backtitle "$DIALOG_BACKTITLE" \
		--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
	SCANSSID=0
	output=$( wpa_cli scan 2>&1 )
	f_dprintf "%s" "$output"
	f_dialog_title "Scanning"
	$DIALOG \
		--title "$DIALOG_TITLE" \
		--backtitle "$DIALOG_BACKTITLE" \
		--ok-label "$msg_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' '
		/..:..:..:..:..:../ && $5 { printf "\"%s\"\t%s\n", $5, $4 }
	' | sort | uniq )

	if [ ! "$NETWORKS" ]; then
		f_dialog_title "$msg_error"
		$DIALOG \
			--title "$DIALOG_TITLE" \
			--backtitle "$DIALOG_BACKTITLE" \
			--yesno "No wireless networks were found. Rescan?" \
			0 0 && continue
		exit 1
	fi

	f_dialog_title "Network Selection"
	NETWORK=$( sh -c "$DIALOG \
		--title \"$DIALOG_TITLE\" \
		--backtitle \"$DIALOG_BACKTITLE\" \
		--extra-button \
		--extra-label \"Rescan\" \
		--menu \"Select a wireless network to connect to.\" \
		0 0 0 \
		$( echo $NETWORKS | tr '\n' ' ' )" \
		2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
	)
	retval=$?
	f_dialog_data_sanitize NETWORK
	case $retval in
	$DIALOG_OK) break ;;
	$DIALOG_CANCEL)
		# here we ask if the user wants to select the 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
		ENCRYPTION=$( $DIALOG \
			--title "$DIALOG_TITLE" \
			--backtitle "$DIALOG_BACKTITLE" \
			--menu "Select encryption type" \
			0 0 0 \
			"1 WPA/WPA2 PSK" "" \
			"2 WPA/WPA2 EAP" "" \
			"3 WEP" "" \
			"0 None" "" \
			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(/^\t/,"")||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(/^\t/,"")||1' \
		>> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
	network={
		ssid="$NETWORK"
		scan_ssid=$SCANSSID
		key_mgmt=WPA-EAP$(
		echo "$USERPASS" | awk '
			NR == 1 { printf "\n\t\tidentity=\"%s\"", $1 }
			NR == 2 { printf "\n\t\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(/^\t/,"")||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(/^\t/,"")||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
if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
	output=$( wpa_cli reconfigure 2>&1 )
	f_dprintf "%s" "$output"
fi

exit $SUCCESS

################################################################################
# END
################################################################################