usr.sbin/bsdinstall/scripts/wlanconfig

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2011 Nathan Whitehorn
   4 # Copyright (c) 2013-2016 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         error_str=$( ifconfig "$WLAN_IFACE" $ifconfig_args 2>&1 |
  56                 sed -e 's/ifconfig: //' )
  57         if [ "$iface_up" ]; then
  58                 # Restart wpa_supplicant(8) (should not fail).
  59                 wpa_supplicant -B -i "$WLAN_IFACE" -c \
  60                         "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
  61         fi
  62         if [ "$error_str" ]; then
  63                 $DIALOG \
  64                         --title "$msg_error" \
  65                         --backtitle "$DIALOG_BACKTITLE" \
  66                         --yes-label Change \
  67                         --no-label Ignore \
  68                         --yesno \
  69                         "Error while applying chosen settings ($error_str)" \
  70                         0 0 || return $SUCCESS # Skip
  71                 return $FAILURE # Restart
  72         else
  73                 awk 'sub(/^\t\t/,"")||1' \
  74                         > "$BSDINSTALL_TMPETC/rc.conf.net.wlan" <<-EOF
  75                 create_args_$WLAN_IFACE="$ifconfig_args"
  76                 EOF
  77         fi
  78
  79         return $SUCCESS
  80 }
  81
  82 dialog_country_select()
  83 {
  84         local input regdomains countries regdomain country prompt
  85         local no_default="<not selected>"
  86         local default_regdomain="${1:-$no_default}"
  87         local default_country="${2:-$no_default}"
  88
  89         #
  90         # Parse available countries/regdomains
  91         #
  92         input=$( ifconfig "$WLAN_IFACE" list countries | sed -e 's/DEBUG//gi' )
  93         regdomains=$( echo "$input" | awk '
  94                 sub(/.*domains:/, ""), /[^[:alnum:][[:space:]]/ {
  95                         n = split($0, domains)
  96                         for (i = 1; i <= n; i++)
  97                                 printf "'\''%s'\'' '\'\''", domains[i]
  98                 }
  99         ' | sort )
 100         countries=$( echo "$input" | awk '
 101                 sub(/Country codes:/, ""), sub(/Regulatory.*/, "") {
 102                         while (match($0, /[[:upper:]][[:upper:][:digit:]] /)) {
 103                                 country = substr($0, RSTART)
 104                                 sub(/ [[:upper:]][[:upper:][:digit:]].*/, "", country)
 105                                 code = substr(country, 1, 2)
 106                                 desc = substr(country, 4)
 107                                 sub(/[[:space:]]*$/, "", desc)
 108                                 printf "'\''%s'\'' '\''%s'\''\n", code, desc
 109                                 $0 = substr($0, RSTART + RLENGTH)
 110                         }
 111                 }
 112         ' | sort )
 113
 114         f_dialog_title "Regdomain selection"
 115         prompt="Select your regdomain."
 116         eval f_dialog_menu_size height width rows \
 117                 \"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
 118                 \"\$prompt\" \"\" $regdomains
 119         regdomain=$( eval $DIALOG \
 120                 --title \"\$DIALOG_TITLE\"             \
 121                 --backtitle \"\$DIALOG_BACKTITLE\"     \
 122                 --cancel-label \"\$msg_skip\"          \
 123                 --default-item \"\$default_regdomain\" \
 124                 --menu \"\$prompt\"                    \
 125                 $height $width $rows                   \
 126                 $regdomains                            \
 127                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 128         )
 129         f_dialog_data_sanitize regdomain
 130
 131         f_dialog_title "Country selection"
 132         prompt="Select your country."
 133         eval f_dialog_menu_size height width rows \
 134                 \"\$DIALOG_TITLE\" \"\$DIALOG_BACKTITLE\" \
 135                 \"\$prompt\" \"\" $countries
 136         country=$( eval $DIALOG \
 137                 --title \"\$DIALOG_TITLE\"           \
 138                 --backtitle \"\$DIALOG_BACKTITLE\"   \
 139                 --cancel-label \"\$msg_skip\"        \
 140                 --default-item \"\$default_country\" \
 141                 --menu \"\$prompt\"                  \
 142                 $height $width $rows                 \
 143                 $countries                           \
 144                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 145         )
 146         f_dialog_data_sanitize country
 147
 148         country_set "$regdomain" "$country"
 149 }
 150
 151 ############################################################ MAIN
 152
 153 : > "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
 154 chmod 0600 "$BSDINSTALL_TMPETC/wpa_supplicant.conf"
 155
 156 cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" << EOF
 157 ctrl_interface=/var/run/wpa_supplicant
 158 eapol_version=2
 159 ap_scan=1
 160 fast_reauth=1
 161
 162 EOF
 163
 164 #
 165 # Try to reach wpa_supplicant. If it isn't running and we can modify the
 166 # existing system, start it. Otherwise, fail.
 167 #
 168 if ! wpa_cli ping > /dev/null 2>&1; then
 169         if [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
 170                 $DIALOG --backtitle "$DIALOG_BACKTITLE" --title "$msg_error" --msgbox \
 171                         "Wireless cannot be configured without making changes to the local system!" 0 0
 172                 exit 1
 173         fi
 174         wpa_supplicant -B -i $1 -c "$BSDINSTALL_TMPETC/wpa_supplicant.conf") || exit 1
 175
 176         if ! wpa_cli ping > /dev/null 2>&1 && [ ! "$BSDINSTALL_CONFIGCURRENT" ]; then
 177                 $DIALOG \
 178                         --title "$msg_error" \
 179                         --backtitle "$DIALOG_BACKTITLE" \
 180                         --msgbox "Could not start wpa_supplicant!" \
 181                         0 0
 182                 exit 1
 183         fi
 184 fi
 185
 186 #
 187 # There is no way to check country/regdomain without (possible)
 188 # interface state modification
 189 #
 190 if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
 191         # Get current country/regdomain for selected interface
 192         WLAN_IFACE=$( wpa_cli ifname | tail -1 )
 193         INPUT=$( ifconfig "$WLAN_IFACE" list regdomain | head -1 )
 194         DEF_REGDOMAIN=$( echo "$INPUT" | cut -w -f 2 )
 195         DEF_COUNTRY=$( echo "$INPUT" | cut -w -f 4 )
 196         [ "$DEF_REGDOMAIN" = 0 ] && DEF_REGDOMAIN="<not selected>"
 197         [ "$DEF_COUNTRY" = 0 ] && DEF_COUNTRY="<not selected>"
 198         f_dialog_title "Regdomain/country"
 199         $DIALOG \
 200                 --title "$DIALOG_TITLE" \
 201                 --backtitle "$DIALOG_BACKTITLE" \
 202                 --yesno "Change regdomain/country (now $DEF_REGDOMAIN/$DEF_COUNTRY)?" \
 203                 0 0
 204         if [ $? -eq 0 ]; then
 205                 while :; do
 206                         dialog_country_select "$DEF_REGDOMAIN" "$DEF_COUNTRY"
 207                         if [ $? -eq $SUCCESS ]; then
 208                                 break
 209                         fi
 210                 done
 211         fi
 212 fi
 213
 214 while :; do
 215         SCANSSID=0
 216         output=$( wpa_cli scan 2>&1 )
 217         f_dprintf "%s" "$output"
 218         f_dialog_title "Scanning"
 219         $DIALOG \
 220                 --title "$DIALOG_TITLE" \
 221                 --backtitle "$DIALOG_BACKTITLE" \
 222                 --ok-label "$msg_skip" \
 223                 --pause "Waiting 5 seconds to scan for wireless networks..." \
 224                 9 40 5 || exit 1
 225
 226         SCAN_RESULTS=$( wpa_cli scan_results )
 227         NETWORKS=$( echo "$SCAN_RESULTS" | awk -F '\t' '
 228                 /..:..:..:..:..:../ && $5 { printf "\"%s\"\t%s\n", $5, $4 }
 229         ' | sort | uniq )
 230
 231         if [ ! "$NETWORKS" ]; then
 232                 f_dialog_title "$msg_error"
 233                 $DIALOG \
 234                         --title "$DIALOG_TITLE" \
 235                         --backtitle "$DIALOG_BACKTITLE" \
 236                         --yesno "No wireless networks were found. Rescan?" \
 237                         0 0 && continue
 238                 exit 1
 239         fi
 240
 241         f_dialog_title "Network Selection"
 242         prompt="Select a wireless network to connect to."
 243         menu_list=$( echo $NETWORKS | tr '\n' ' ' )
 244         NETWORK=$( eval $DIALOG \
 245                 --title \"\$DIALOG_TITLE\"         \
 246                 --backtitle \"\$DIALOG_BACKTITLE\" \
 247                 --extra-button                     \
 248                 --extra-label \"Rescan\"           \
 249                 --menu \"\$prompt\"                \
 250                 $height $width $rows               \
 251                 $menu_list                         \
 252                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 253         )
 254         retval=$?
 255         f_dialog_data_sanitize NETWORK
 256         case $retval in
 257         $DIALOG_OK) break ;;
 258         $DIALOG_CANCEL)
 259                 # here we ask if the user wants to select the network manually
 260                 f_dialog_title "Network Selection"
 261                 f_yesno "Do you want to select the network manually?" || exit 1
 262                 f_dialog_input NETWORK "Enter SSID" || exit 1
 263                 prompt="Select encryption type"
 264                 menu_list="
 265                         '1 WPA/WPA2 PSK' ''
 266                         '2 WPA/WPA2 EAP' ''
 267                         '3 WEP' ''
 268                         '0 None' ''
 269                 " # END-QUOTE
 270                 eval f_dialog_menu_size height width rows \"\$DIALOG_TITLE\" \
 271                         \"\$DIALOG_BACKTITLE\" \"\$prompt\" \"\" $menu_list
 272                 ENCRYPTION=$( eval $DIALOG \
 273                         --title \"\$DIALOG_TITLE\"         \
 274                         --backtitle \"\$DIALOG_BACKTITLE\" \
 275                         --menu \"\$prompt\"                \
 276                         $height $width $rows               \
 277                         $menu_list                         \
 278                         2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 279                 ) || exit 1
 280                 SCANSSID=1
 281                 break
 282                 ;;
 283         $DIALOG_EXTRA) # Rescan
 284                 ;;
 285         esac
 286 done
 287
 288 [ "$ENCRYPTION" ] || ENCRYPTION=$( echo "$NETWORKS" |
 289         awk -F '\t' "/^\"$NETWORK\"\t/ { print \$2 }" )
 290
 291 if echo "$ENCRYPTION" | grep -q PSK; then
 292         PASS=$( $DIALOG \
 293                 --title "WPA Setup"              \
 294                 --backtitle "$DIALOG_BACKTITLE"  \
 295                 --insecure                       \
 296                 --mixedform ""                   \
 297                 0 0 0                            \
 298                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 299                 "Password" 2 0 "" 2 12 15 63 1   \
 300                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 301         ) || exec "$0" "$@"
 302         awk 'sub(/^\t/,"")||1' \
 303                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 304         network={
 305                 ssid="$NETWORK"
 306                 scan_ssid=$SCANSSID
 307                 psk="$PASS"
 308                 priority=5
 309         }
 310         EOF
 311 elif echo "$ENCRYPTION" | grep -q EAP; then
 312         USERPASS=$( $DIALOG \
 313                 --title "WPA-Enterprise Setup"   \
 314                 --backtitle "$DIALOG_BACKTITLE"  \
 315                 --insecure                       \
 316                 --mixedform ""                   \
 317                 0 0 0                            \
 318                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 319                 "Username" 2 0 "" 2 12 25 63 0   \
 320                 "Password" 3 0 "" 3 12 25 63 1   \
 321                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 322         ) || exec "$0" "$@"
 323         awk 'sub(/^\t/,"")||1' \
 324                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 325         network={
 326                 ssid="$NETWORK"
 327                 scan_ssid=$SCANSSID
 328                 key_mgmt=WPA-EAP$(
 329                 echo "$USERPASS" | awk '
 330                         NR == 1 { printf "\n\t\tidentity=\"%s\"", $1 }
 331                         NR == 2 { printf "\n\t\tpassword=\"%s\"", $1 }
 332                 ' )
 333                 priority=5
 334         }
 335         EOF
 336 elif echo "$ENCRYPTION" | grep -q WEP; then
 337         WEPKEY=$( $DIALOG \
 338                 --title "WEP Setup"              \
 339                 --backtitle "$DIALOG_BACKTITLE"  \
 340                 --insecure                       \
 341                 --mixedform ""                   \
 342                 0 0 0                            \
 343                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 344                 "WEP Key 0" 2 0 "" 2 12 15 0 1   \
 345                 2>&1 >&$DIALOG_TERMINAL_PASSTHRU_FD
 346         ) || exec "$0" "$@"
 347         awk 'sub(/^\t/,"")||1' \
 348                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 349         network={
 350                 ssid="$NETWORK"
 351                 scan_ssid=$SCANSSID
 352                 key_mgmt=NONE
 353                 wep_key0="$WEPKEY"
 354                 wep_tx_keyidx=0
 355                 priority=5
 356         }
 357         EOF
 358 else # Open
 359         awk 'sub(/^\t/,"")||1' \
 360                 >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" <<-EOF
 361         network={
 362                 ssid="$NETWORK"
 363                 scan_ssid=$SCANSSID
 364                 key_mgmt=NONE
 365                 priority=5
 366         }
 367         EOF
 368 fi
 369
 370 # Connect to any open networks policy
 371 cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" << EOF
 372 network={
 373         priority=0
 374         key_mgmt=NONE
 375 }
 376 EOF
 377
 378 # Bring up new network
 379 if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
 380         output=$( wpa_cli reconfigure 2>&1 )
 381         f_dprintf "%s" "$output"
 382 fi
 383
 384 exit $SUCCESS
 385
 386 ################################################################################
 387 # END
 388 ################################################################################