usr.sbin/bsdinstall/scripts/wlanconfig

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2011 Nathan Whitehorn
   4 # Copyright (c) 2013-2020 Devin Teske
   5 # All rights reserved.
   6 #
   7 # Redistribution and use in source and binary forms, with or without
   8 # modification, are permitted provided that the following conditions
   9 # are met:
  10 # 1. Redistributions of source code must retain the above copyright
  11 #    notice, this list of conditions and the following disclaimer.
  12 # 2. Redistributions in binary form must reproduce the above copyright
  13 #    notice, this list of conditions and the following disclaimer in the
  14 #    documentation and/or other materials provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26 # SUCH DAMAGE.
  27 #
  28 # $FreeBSD$
  29 #
  30 ############################################################ INCLUDES
  31
  32 BSDCFG_SHARE="/usr/share/bsdconfig"
  33 . $BSDCFG_SHARE/common.subr || exit 1
  34 f_include $BSDCFG_SHARE/dialog.subr
  35 f_dialog_backtitle "FreeBSD Installer"
  36
  37 ############################################################ FUNCTIONS
  38
  39 country_set()
  40 {
  41         local error_str iface_up ifconfig_args=
  42
  43         #
  44         # Setup what was selected
  45         # NB: Do not change order of arguments (or regdomain will be ignored)
  46         #
  47         [ "$2" ] && ifconfig_args="$ifconfig_args country $2"
  48         [ "$1" ] && ifconfig_args="$ifconfig_args regdomain $1"
  49         [ "$ifconfig_args" ] || return $SUCCESS # Nothing to do
  50         ifconfig_args="${ifconfig_args# }"
  51
  52         # Regdomain/country cannot be applied while interface is running
  53         iface_up=$( ifconfig -lu | grep -w "$WLAN_IFACE" )
  54         [ "$iface_up" ] && ifconfig "$WLAN_IFACE" down
  55         f_eval_catch -dk error_str wlanconfig ifconfig "ifconfig %s %s" \
  56                 "$WLAN_IFACE" "$ifconfig_args"
  57         error_str="${error_str#ifconfig: }"
  58         # Restart wpa_supplicant(8) (should not fail).
  59         [ "$iface_up" ] && f_eval_catch -d wlanconfig wpa_supplicant \
  60                 'wpa_supplicant -B -i "%s" -c "%s/wpa_supplicant.conf"' \
  61                 "$WLAN_IFACE" "$BSDINSTALL_TMPETC"
  62         if [ "$error_str" ]; then
  63                 $DIALOG --title "$msg_error" \
  64                         --backtitle "$DIALOG_BACKTITLE" \
  65                         --yes-label Change \
  66                         --no-label Ignore \
  67                         --yesno \
  68                         "Error while applying chosen settings ($error_str)" \
  69                         0 0 || return $SUCCESS # Skip
  70                 return $FAILURE # Restart
  71         else
  72                 cat > "$BSDINSTALL_TMPETC/rc.conf.net.wlan" <<-EOF
  73                 create_args_$WLAN_IFACE="$ifconfig_args"
  74                 EOF
  75         fi
  76
  77         return $SUCCESS
  78 }
  79
  80 dialog_country_select()
  81 {
  82         local input regdomains countries regdomain country prompt
  83         local no_default="<not selected>"
  84         local default_regdomain="${1:-$no_default}"
  85         local default_country="${2:-$no_default}"
  86
  87         #
  88         # Parse available countries/regdomains
  89         #
  90         input=$( ifconfig "$WLAN_IFACE" list countries | sed -e 's/DEBUG//gi' )
  91         regdomains=$( echo "$input" | awk '
  92                 sub(/.*domains:/, ""), /[^[:alnum:][[:space:]]/ {
  93                         n = split($0, domains)
  94                         for (i = 1; i <= n; i++)
  95                                 printf "'\''%s'\'' '\'\''", domains[i]
  96                 }
  97         ' | sort )
  98         countries=$( echo "$input" | awk '
  99                 sub(/Country codes:/, ""), sub(/Regulatory.*/, "") {
 100                         while (match($0, /[[:upper:]][[:upper:][:digit:]] /)) {
 101                                 country = substr($0, RSTART)
 102                                 sub(/ [[:upper:]][[:upper:][:digit:]].*/, "",
 103                                         country)
 104                                 code = substr(country, 1, 2)
 105                                 desc = substr(country, 4)
 106                                 sub(/[[:space:]]*$/, "", desc)
 107                                 printf "'\''%s'\'' '\''%s'\''\n", code, desc
 108                                 $0 = substr($0, RSTART + RLENGTH)
 109                         }
 110                 }
 111         ' | sort )
 112
 113         f_dialog_title "Regdomain selection"
 114         prompt="Select your regdomain."
 115         eval f_dialog_menu_size height width rows \
 116                 \"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
 117                 \"\$prompt\" \"\" $regdomains
 118         regdomain=$( eval $DIALOG \
 119                 --title \"\$DIALOG_TITLE\"             \
 120                 --backtitle \"\$DIALOG_BACKTITLE\"     \
 121                 --cancel-label \"\$msg_skip\"          \
 122                 --default-item \"\$default_regdomain\" \
 123                 --menu \"\$prompt\"                    \
 124                 $height $width $rows                   \
 125                 $regdomains                            \
 126                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 127         )
 128         f_dialog_data_sanitize regdomain
 129
 130         f_dialog_title "Country selection"
 131         prompt="Select your country."
 132         eval f_dialog_menu_size height width rows \
 133                 \"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
 134                 \"\$prompt\" \"\" $countries
 135         country=$( eval $DIALOG \
 136                 --title \"\$DIALOG_TITLE\"           \
 137                 --backtitle \"\$DIALOG_BACKTITLE\"   \
 138                 --cancel-label \"\$msg_skip\"        \
 139                 --default-item \"\$default_country\" \
 140                 --menu \"\$prompt\"                  \
 141                 $height $width $rows                 \
 142                 $countries                           \
 143                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 144         )
 145         f_dialog_data_sanitize country
 146
 147         country_set "$regdomain" "$country"
 148 }
 149
 150 ############################################################ MAIN
 151
 152 : > "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
 153 chmod 0600 "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
 154
 155 cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<EOF
 156 ctrl_interface=/var/run/wpa_supplicant
 157 eapol_version=2
 158 ap_scan=1
 159 fast_reauth=1
 160
 161 EOF
 162
 163 #
 164 # Try to reach wpa_supplicant. If it isn't running and we can modify the
 165 # existing system, start it. Otherwise, fail.
 166 #
 167 if ! f_eval_catch -d wlanconfig wpa_cli "wpa_cli ping"; then
 168         if [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
 169                 f_show_err "Wireless cannot be configured without %s" \
 170                            "making changes to the local system!"
 171                 exit 1
 172         fi
 173         f_eval_catch wlanconfig wpa_supplicant \
 174                 'wpa_supplicant -B -i "%s" -c "%s/wpa_supplicant.conf"' \
 175                 "$1" "$BSDINSTALL_TMPETC" || exit 1
 176
 177         # See if we succeeded
 178         f_eval_catch wlanconfig wpa_cli "wpa_cli ping" || exit 1
 179 fi
 180
 181 #
 182 # There is no way to check country/regdomain without (possible)
 183 # interface state modification
 184 #
 185 if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
 186         # Get current country/regdomain for selected interface
 187         WLAN_IFACE=$( wpa_cli ifname | tail -n 1 )
 188         INPUT=$( ifconfig "$WLAN_IFACE" list regdomain | head -n 1 )
 189         DEF_REGDOMAIN=$( echo "$INPUT" | cut -w -f 2 )
 190         DEF_COUNTRY=$( echo "$INPUT" | cut -w -f 4 )
 191         [ "$DEF_REGDOMAIN" = 0 ] && DEF_REGDOMAIN="<not selected>"
 192         [ "$DEF_COUNTRY" = 0 ] && DEF_COUNTRY="<not selected>"
 193         f_dialog_title "Regdomain/country"
 194         if f_yesno "Change regdomain/country ($DEF_REGDOMAIN/$DEF_COUNTRY)?"
 195         then
 196                 while ! dialog_country_select "$DEF_REGDOMAIN" "$DEF_COUNTRY"
 197                 do :; done
 198         fi
 199 fi
 200
 201 while :; do
 202         SCANSSID=0
 203         f_eval_catch -d wlanconfig wpa_cli "wpa_cli scan"
 204         f_dialog_title "Scanning"
 205         f_dialog_pause "Waiting 5 seconds to scan for wireless networks..." 5 ||
 206                 exit 1
 207
 208         f_eval_catch -dk SCAN_RESULTS wlanconfig wpa_cli "wpa_cli scan_results"
 209         NETWORKS=$( echo "$SCAN_RESULTS" | awk -F '\t' '
 210                 /..:..:..:..:..:../ && $5 { printf "\"%s\"\t\"%s\"\n", $5, $4 }
 211         ' | sort | uniq )
 212
 213         if [ ! "$NETWORKS" ]; then
 214                 f_dialog_title "$msg_error"
 215                 f_yesno "No wireless networks were found. Rescan?" && continue
 216                 exit 1
 217         fi
 218
 219         f_dialog_title "Network Selection"
 220         prompt="Select a wireless network to connect to."
 221         f_dialog_menu_size height width rows "$DIALOG_TITLE" \
 222                 "$DIALOG_BACKTITLE" "$prompt" "" $menu_list
 223         NETWORK=$( eval $DIALOG \
 224                 --title \"\$DIALOG_TITLE\"         \
 225                 --backtitle \"\$DIALOG_BACKTITLE\" \
 226                 --extra-button                     \
 227                 --extra-label \"Rescan\"           \
 228                 --menu \"\$prompt\"                \
 229                 $height $width $rows               \
 230                 $NETWORKS                          \
 231                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 232         )
 233         retval=$?
 234         f_dialog_data_sanitize NETWORK
 235         case $retval in
 236         $DIALOG_OK) break ;;
 237         $DIALOG_CANCEL)
 238                 # Ask if the user wants to select network manually
 239                 f_dialog_title "Network Selection"
 240                 f_yesno "Do you want to select the network manually?" || exit 1
 241                 f_dialog_input NETWORK "Enter SSID" || exit 1
 242                 prompt="Select encryption type"
 243                 menu_list="
 244                         '1 WPA/WPA2 PSK' ''
 245                         '2 WPA/WPA2 EAP' ''
 246                         '3 WEP' ''
 247                         '0 None' ''
 248                 " # END-QUOTE
 249                 eval f_dialog_menu_size height width rows \"\$DIALOG_TITLE\" \
 250                         \"\$DIALOG_BACKTITLE\" \"\$prompt\" \"\" $menu_list
 251                 ENCRYPTION=$( eval $DIALOG \
 252                         --title \"\$DIALOG_TITLE\"         \
 253                         --backtitle \"\$DIALOG_BACKTITLE\" \
 254                         --menu \"\$prompt\"                \
 255                         $height $width $rows               \
 256                         $menu_list                         \
 257                         2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 258                 ) || exit 1
 259                 SCANSSID=1
 260                 break
 261                 ;;
 262         $DIALOG_EXTRA) # Rescan
 263                 ;;
 264         esac
 265 done
 266
 267 [ "$ENCRYPTION" ] || ENCRYPTION=$( echo "$NETWORKS" |
 268         awk -F '\t' "/^\"$NETWORK\"\t/ { print \$2 }" )
 269
 270 if echo "$ENCRYPTION" | grep -q PSK; then
 271         PASS=$( $DIALOG \
 272                 --title "WPA Setup"              \
 273                 --backtitle "$DIALOG_BACKTITLE"  \
 274                 --insecure                       \
 275                 --mixedform ""                   \
 276                 0 0 0                            \
 277                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 278                 "Password" 2 0 "" 2 12 15 63 1   \
 279                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 280         ) || exec "$0" "$@"
 281         awk 'sub(/^\\/,"")||1' \
 282                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 283         network={
 284         \       ssid="$NETWORK"
 285         \       scan_ssid=$SCANSSID
 286         \       psk="$PASS"
 287         \       priority=5
 288         }
 289         EOF
 290 elif echo "$ENCRYPTION" | grep -q EAP; then
 291         USERPASS=$( $DIALOG \
 292                 --title "WPA-Enterprise Setup"   \
 293                 --backtitle "$DIALOG_BACKTITLE"  \
 294                 --insecure                       \
 295                 --mixedform ""                   \
 296                 0 0 0                            \
 297                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 298                 "Username" 2 0 "" 2 12 25 63 0   \
 299                 "Password" 3 0 "" 3 12 25 63 1   \
 300                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 301         ) || exec "$0" "$@"
 302         awk 'sub(/^\\/,"")||1' \
 303                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 304         network={
 305         \       ssid="$NETWORK"
 306         \       scan_ssid=$SCANSSID
 307         \       key_mgmt=WPA-EAP$(
 308                 echo "$USERPASS" | awk '
 309                         NR == 1 { printf "\n\tidentity=\"%s\"", $1 }
 310                         NR == 2 { printf "\n\tpassword=\"%s\"", $1 }
 311                 ' )
 312         \       priority=5
 313         }
 314         EOF
 315 elif echo "$ENCRYPTION" | grep -q WEP; then
 316         WEPKEY=$( $DIALOG \
 317                 --title "WEP Setup"              \
 318                 --backtitle "$DIALOG_BACKTITLE"  \
 319                 --insecure                       \
 320                 --mixedform ""                   \
 321                 0 0 0                            \
 322                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 323                 "WEP Key 0" 2 0 "" 2 12 15 0 1   \
 324                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 325         ) || exec "$0" "$@"
 326         awk 'sub(/^\\/,"")||1' \
 327                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 328         network={
 329         \       ssid="$NETWORK"
 330         \       scan_ssid=$SCANSSID
 331         \       key_mgmt=NONE
 332         \       wep_key0="$WEPKEY"
 333         \       wep_tx_keyidx=0
 334         \       priority=5
 335         }
 336         EOF
 337 else # Open
 338         awk 'sub(/^\\/,"")||1' \
 339                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 340         network={
 341         \       ssid="$NETWORK"
 342         \       scan_ssid=$SCANSSID
 343         \       key_mgmt=NONE
 344         \       priority=5
 345         }
 346         EOF
 347 fi
 348
 349 # Connect to any open networks policy
 350 cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<EOF
 351 network={
 352         priority=0
 353         key_mgmt=NONE
 354 }
 355 EOF
 356
 357 # Bring up new network
 358 [ "$BSDINSTALL_CONFIGCURRENT" ] &&
 359         f_eval_catch -d wlanconfig wpa_cli "wpa_cli reconfigure"
 360
 361 exit $SUCCESS
 362
 363 ################################################################################
 364 # END
 365 ################################################################################