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