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 32BSDCFG_SHARE="/usr/share/bsdconfig" 33. $BSDCFG_SHARE/common.subr || exit 1 34f_include $BSDCFG_SHARE/dialog.subr 35f_dialog_backtitle "FreeBSD Installer" 36 37############################################################ FUNCTIONS 38 39country_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 82dialog_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" 154chmod 0600 "$BSDINSTALL_TMPETC/wpa_supplicant.conf" 155 156cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" << EOF 157ctrl_interface=/var/run/wpa_supplicant 158eapol_version=2 159ap_scan=1 160fast_reauth=1 161 162EOF 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# 168if ! 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 184fi 185 186# 187# There is no way to check country/regdomain without (possible) 188# interface state modification 189# 190if [ "$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 212fi 213 214while :; 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 286done 287 288[ "$ENCRYPTION" ] || ENCRYPTION=$( echo "$NETWORKS" | 289 awk -F '\t' "/^\"$NETWORK\"\t/ { print \$2 }" ) 290 291if 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 311elif 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 336elif 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 358else # 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 368fi 369 370# Connect to any open networks policy 371cat >> "$BSDINSTALL_TMPETC/wpa_supplicant.conf" << EOF 372network={ 373 priority=0 374 key_mgmt=NONE 375} 376EOF 377 378# Bring up new network 379if [ "$BSDINSTALL_CONFIGCURRENT" ]; then 380 output=$( wpa_cli reconfigure 2>&1 ) 381 f_dprintf "%s" "$output" 382fi 383 384exit $SUCCESS 385 386################################################################################ 387# END 388################################################################################ 389