usr.sbin/bsdinstall/scripts/wlanconfig

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2011 Nathan Whitehorn
   4 # Copyright (c) 2013 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
  35 ############################################################ MAIN
  36
  37 echo -n > $BSDINSTALL_TMPETC/wpa_supplicant.conf
  38 chmod 0600 $BSDINSTALL_TMPETC/wpa_supplicant.conf
  39
  40 echo "ctrl_interface=/var/run/wpa_supplicant" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
  41 echo "eapol_version=2" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
  42 echo "ap_scan=1" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
  43 echo "fast_reauth=1" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
  44 echo >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
  45
  46 # Try to reach wpa_supplicant. If it isn't running and we can modify the
  47 # existing system, start it. Otherwise, fail.
  48 (wpa_cli ping >/dev/null 2>/dev/null || ([ ! -z $BSDINSTALL_CONFIGCURRENT ] && \
  49         wpa_supplicant -B -i $1 -c $BSDINSTALL_TMPETC/wpa_supplicant.conf)) || \
  50         (dialog --backtitle "FreeBSD Installer" --title "Error" --msgbox \
  51         "Could not start wpa_supplicant!" 0 0; exit 1) || exit 1
  52
  53 # See if we succeeded
  54 wpa_cli ping >/dev/null 2>/dev/null
  55 if [ $? -ne 0 -a -z $BSDINSTALL_CONFIGCURRENT ]; then
  56         dialog --backtitle "FreeBSD Installer" --title "Error" --msgbox \
  57         "Wireless cannot be configured without making changes to the local system!" \ 0 0
  58         exit 1
  59 fi
  60
  61 output=$( wpa_cli scan 2>&1 )
  62 f_dprintf "%s" "$output"
  63 dialog --backtitle "FreeBSD Installer" --title "Scanning" --ok-label "Skip" \
  64         --pause "Waiting 5 seconds to scan for wireless networks..." \
  65         9 40 5 || exit 1
  66
  67 SCAN_RESULTS=`wpa_cli scan_results`
  68 NETWORKS=`echo "$SCAN_RESULTS" | awk -F '\t' \
  69    '/..:..:..:..:..:../ {if (length($5) > 0) printf("\"%s\"\t%s\n", $5, $4);}' |
  70    sort | uniq`
  71
  72 if [ -z "$NETWORKS" ]; then
  73         dialog --backtitle "FreeBSD Installer" --title "Error" \
  74             --yesno "No wireless networks were found. Rescan?" 0 0 && \
  75             exec $0 $@
  76         exit 1
  77 fi
  78
  79 exec 3>&1
  80 NETWORK=`sh -c "dialog --extra-button --extra-label \"Rescan\" \
  81     --backtitle \"FreeBSD Installer\" --title \"Network Selection\" --menu \
  82     \"Select a wireless network to connect to.\" 0 0 0 \
  83     $(echo $NETWORKS | tr '\n' ' ')" 2>&1 1>&3`
  84 case $? in
  85 0)      # OK
  86         ;;
  87 1)      # Cancel
  88         exit 1
  89         ;;
  90 3)      # Rescan
  91         exec $0 $@
  92         ;;
  93 esac
  94 exec 3>&-
  95
  96 ENCRYPTION=`echo "$NETWORKS" | awk -F '\t' \
  97     "/^\"$NETWORK\"\t/ {printf(\"%s\n\", \\\$2 );}"`
  98
  99 if echo $ENCRYPTION | grep -q 'PSK'; then
 100         exec 3>&1
 101         PASS=`dialog --insecure --backtitle "FreeBSD Installer" \
 102             --title "WPA Setup" --mixedform "" 0 0 0 \
 103                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 104                 "Password" 2 0 "" 2 12 15 63 1 \
 105                 2>&1 1>&3` \
 106         || exec $0 $@
 107         exec 3>&-
 108 echo "network={
 109         ssid=\"$NETWORK\"
 110         psk=\"$PASS\"
 111         priority=5
 112 }" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 113 elif echo $ENCRYPTION | grep -q EAP; then
 114         exec 3>&1
 115         USERPASS=`dialog --insecure --backtitle "FreeBSD Installer" \
 116             --title "WPA-Enterprise Setup" --mixedform "" 0 0 0 \
 117                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 118                 "Username" 2 0 "" 2 12 25 63 0 \
 119                 "Password" 3 0 "" 3 12 25 63 1 \
 120                 2>&1 1>&3` \
 121         || exec $0 $@
 122         exec 3>&-
 123 echo "network={
 124         ssid=\"$NETWORK\"
 125         key_mgmt=WPA-EAP" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 126 echo "$USERPASS" | awk '
 127 {
 128         if (NR == 1) {
 129                 printf "        identity=\"%s\"\n", $1;
 130         } else if (NR == 2) {
 131                 printf "        password=\"%s\"\n", $1;
 132         }
 133 }' >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 134 echo "  priority=5
 135 }" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 136 elif echo $ENCRYPTION | grep -q WEP; then
 137         exec 3>&1
 138         WEPKEY=`dialog --insecure --backtitle "FreeBSD Installer" \
 139             --title "WEP Setup" --mixedform "" 0 0 0 \
 140                 "SSID" 1 0 "$NETWORK" 1 12 0 0 2 \
 141                 "WEP Key 0" 2 0 "" 2 12 15 0 1 \
 142                 2>&1 1>&3` \
 143         || exec $0 $@
 144 echo "network={
 145         ssid=\"$NETWORK\"
 146         key_mgmt=NONE
 147         wep_key0=\"$WEPKEY\"
 148         wep_tx_keyidx=0
 149         priority=5
 150 }" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 151 else    # Open
 152 echo "network={
 153         ssid=\"$NETWORK\"
 154         key_mgmt=NONE
 155         priority=5
 156 }" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 157 fi
 158
 159 # Connect to any open networks policy
 160 echo "network={
 161         priority=0
 162         key_mgmt=NONE
 163 }" >> $BSDINSTALL_TMPETC/wpa_supplicant.conf
 164
 165 # Bring up new network
 166 if [ "$BSDINSTALL_CONFIGCURRENT" ]; then
 167         output=$( wpa_cli reconfigure 2>&1 )
 168         f_dprintf "%s" "$output"
 169 fi
 170
 171 exit 0
 172
 173 ################################################################################
 174 # END
 175 ################################################################################