# Copyright (c) 2007-2008 Roy Marples <roy@marples.name>
# All rights reserved. Released under the 2-clause BSD license.

_config_vars="$_config_vars ssid mode associate_timeout sleep_scan"
_config_vars="$_config_vars preferred_aps blacklist_aps"

iwconfig_depend()
{
	program /sbin/iwconfig
	after plug
	before interface
	provide wireless
}

iwconfig_get_wep_status()
{
	local mode= status="disabled"

	# No easy way of doing this grep in bash regex :/
	if LC_ALL=C iwconfig "${IFACE}" | \
	grep -qE "^ +Encryption key:[*0-9,A-F]"; then
		status="enabled"
		mode=$(LC_ALL=C iwconfig "${IFACE}" | \
		sed -n -e 's/^.*Security mode:\(.*[^ ]\).*/\1/p')
		[ -n "${mode}" ] && mode=" - ${mode}"
	fi

	echo "(WEP ${status}${mode})"
}

_get_ssid()
{
	local i=5 ssid=

	while [ ${i} -gt 0 ]; do
		ssid=$(iwgetid --raw "${IFACE}")
		if [ -n "${ssid}" ]; then
			echo "${ssid}"
			return 0
		fi
		sleep 1
		i=$((${i} - 1))
	done

	return 1
}

_get_ap_mac_address()
{
	local mac="$(iwgetid --raw --ap "${IFACE}")" 
	case "${mac}" in
		"00:00:00:00:00:00") return 1;;
		"44:44:44:44:44:44") return 1;;
		"FF:00:00:00:00:00") return 1;;
		"FF:FF:FF:FF:FF:FF") return 1;;
		*) echo "${mac}";;
	esac
}

iwconfig_get_mode()
{
	LC_ALL=C iwgetid --mode "${IFACE}" | \
	sed -n -e 's/^.*Mode:\(.*\)/\1/p' | \
	tr '[:upper:]' '[:lower:]'
}

iwconfig_set_mode()
{
	local mode="$1"
	[ "${mode}" = "$(iwconfig_get_mode)" ] && return 0

	# Devicescape stack requires the interface to be down
	_down
	iwconfig "${IFACE}" mode "${mode}" || return 1
	_up
}

iwconfig_get_type()
{
	LC_ALL=C iwconfig "${IFACE}" | \
	sed -n -e 's/^'"$1"' *\([^ ]* [^ ]*\).*/\1/p'
}

iwconfig_report()
{
	local mac= m="connected to"
	local ssid="$(_get_ssid)"
	local wep_status="$(iwconfig_get_wep_status)"
	local channel="$(iwgetid --raw --channel "${iface}")"
	[ -n "${channel}" ] && channel="on channel ${channel} "
	local mode="$(iwconfig_get_mode)"
	if [ "${mode}" = "master" ]; then
		m="configured as"
	else
		mac="$(_get_ap_mac_address)"
		[ -n "${mac}" ] && mac=" at ${mac}"
	fi

	eindent
	einfo "${IFACE} ${m} SSID \"${SSID}\"${mac}"
	einfo "in ${mode} mode ${channel}${wep_status}"
	eoutdent
}

iwconfig_get_wep_key()
{
	local mac="$1" key=
	[ -n "${mac}" ] && mac="$(echo "${mac}" | sed -e 's/://g')"
	eval key=\$mac_key_${mac}
	[ -z "${key}" ] && eval key=\$key_${SSIDVAR}
	if [ -z "${key}" ]; then
		echo "off"
	else
		set -- ${key}
		local x= e=false
		for x; do
			if [ "${x}" = "enc" ]; then
				e=true
				break
			fi
		done
		${e} || key="${key} enc open"
		echo "${key}"
	fi
}

iwconfig_user_config()
{
	local conf= var=${SSIDVAR} config=
	[ -z "${var}" ] && var=${IFVAR}

	config="$(_get_array "iwconfig_${var}")"
	local IFS="$__IFS"
	for conf in ${config}; do
		unset IFS
		if ! eval iwconfig "${IFACE}" "${conf}"; then
			ewarn "${IFACE} does not support the following configuration commands"
			ewarn "  ${conf}"
		fi
	done
	unset IFS

	config="$(_get_array "iwpriv_${var}")"
	local IFS="$__IFS"
	for conf in ${config}; do
		unset IFS
		if ! eval iwpriv "${IFACE}" "${conf}"; then
			ewarn "${IFACE} does not support the following private ioctls"
			ewarn "  ${conf}"
		fi
	done
}

iwconfig_setup_specific()
{
	local mode="$1" channel=
	if [ -z "${SSID}" ]; then
		eerror "${IFACE} requires an SSID to be set to operate in ${mode} mode"
		eerror "adjust the ssid_${IFVAR} setting in /etc/conf.d/net"
		return 1
	fi
	SSIDVAR=$(shell_var "${SSID}")
	local key=$(iwconfig_get_wep_key)

	iwconfig_set_mode "${mode}"
	
	# Now set the key
	if ! eval iwconfig "${IFACE}" key "${key}"; then
		if [ "${key}" != "off" ]; then
			ewarn "${IFACE} does not support setting keys"
			ewarn "or the parameter \"mac_key_${SSIDVAR}\" or \"key_${SSIDVAR}\" is incorrect"
		fi
	fi

	# Then set the SSID
	if ! iwconfig "${IFACE}" essid "${SSID}"; then
		eerror "${IFACE} does not support setting SSID to \"${SSID}\""
		return 1
	fi

	eval channel=\$channel_${SSIDVAR}
	[ -z "${channel}" ] && eval channel=\$channel_${IFVAR}
	# We default the channel to 3
	if ! iwconfig "${IFACE}" channel "${channel:-3}"; then
		ewarn "${IFACE} does not support setting the channel to \"${channel:-3}\""
		return 1
	fi
	
	# Finally apply the user Config
	iwconfig_user_config
	
	iwconfig_report
	return 0
}

iwconfig_wait_for_association()
{
	local timeout= i=0
	eval timeout=\$associate_timeout_${IFVAR}
	timeout=${timeout:-10}

	[ ${timeout} -eq 0 ] \
		&& vewarn "WARNING: infinite timeout set for association on ${IFACE}"

	while true; do
		# Use sysfs if we can
		if [ -e /sys/class/net/"${IFACE}"/carrier ]; then
			if [ "$(cat /sys/class/net/"${IFACE}"/carrier)" = "1" ]; then
				# Double check we have an ssid and a non-zero
				# mac address.  This is mainly for buggy
				# prism54 drivers that always set their
				# carrier on or buggy madwifi drivers that
				# sometimes have carrier on and ssid set
				# without being associated.  :/
				[ -n "$(iwgetid --raw "${IFACE}")" ] && [ "$(iwgetid --ap --raw "${IFACE}")" != "00:00:00:00:00:00" ] && return 0
			fi
		else
			local atest=
			eval atest=\$associate_test_${IFVAR}
			atest=${atest:-mac}
			if [ "${atest}" = "mac" -o "${atest}" = "all" ]; then
				[ -n "$(_get_ap_mac_address)" ] && return 0
			fi
			if [ "${atest}" = "quality" -o "${atest}" = "all" ]; then
				[ "$(sed -n -e 's/^.*'"${IFACE}"': *[0-9]* *\([0-9]*\).*/\1/p' \
					/proc/net/wireless)" != "0" ] && return 0
			fi
		fi
		
		sleep 1
		[ ${timeout} -eq 0 ] && continue
		i=$((${i} +  1))
		[ ${i} -ge ${timeout} ] && return 1
	done
	return 1
}

iwconfig_associate()
{
	local mode="${1:-managed}" mac="$2" wep_required="$3"
	local freq="$4" chan="$5"
	local w="(WEP Disabled)" key=

	iwconfig_set_mode "${mode}"

	if [ "${SSID}" = "any" ]; then
		iwconfig "${IFACE}" ap any 2>/dev/null
		unset SSIDVAR
	else
		SSIDVAR=$(shell_var "${SSID}")
		key="$(iwconfig_get_wep_key "${mac}")"
		if [ "${wep_required}" = "on" -a "${key}" = "off" ]; then
			ewarn "WEP key is not set for \"${SSID}\""
			return 1
		fi
		if [ "${wep_required}" = "off" -a "${key}" != "off" ]; then
			key="off"
			ewarn "\"${SSID}\" is not WEP enabled"
		fi

		if ! eval iwconfig "${IFACE}" key "${key}"; then
			if [ "${key}" != "off" ]; then
				ewarn "${IFACE} does not support setting keys"
				ewarn "or the parameter \"mac_key_${SSIDVAR}\" or \"key_${SSIDVAR}\" is incorrect"
				return 1
			fi
		fi
		[ "${key}" != "off" ] && w="$(iwconfig_get_wep_status "${iface}")"
	fi

	if ! iwconfig "${IFACE}" essid "${SSID}"; then
		if [ "${SSID}" != "any" ]; then
			ewarn "${IFACE} does not support setting SSID to \"${SSID}\""
		fi
	fi

	# Only use channel or frequency
	if [ -n "${chan}" ]; then
		iwconfig "${IFACE}" channel "${chan}"
	elif [ -n "${freq}" ]; then
		iwconfig "${IFACE}" freq "${freq}"
	fi
	[ -n "${mac}" ] && iwconfig "${IFACE}" ap "${mac}"
	
	# Finally apply the user Config
	iwconfig_user_config

	ebegin "Connecting to \"${SSID}\" in ${mode} mode ${w}"

	if [ "${SSID}" != "any" ] && type preassociate >/dev/null 2>&1; then
		veinfo "Running preassociate function"
		veindent
		( preassociate )
		local e=$?
		veoutdent
		if [ ${e} -eq 0 ]; then
			veend 1 "preassociate \"${SSID}\" on ${IFACE} failed"
			return 1
		fi
	fi

	if ! iwconfig_wait_for_association; then
		eend 1
		return 1
	fi
	eend 0

	if [ "${SSID}"  = "any" ]; then
		SSID="$(_get_ssid)"
		iwconfig_associate
		return $?
	fi

	iwconfig_report

	if type postassociate >/dev/null 2>&1; then
		veinfo "Running postassociate function"
		veindent
		( postassociate )
		veoutdent
	fi

	return 0
}

iwconfig_scan()
{
	local x= i=0 scan=
	einfo "Scanning for access points"
	eindent

	# Sleep if required
	eval x=\$sleep_scan_${IFVAR}
	[ -n "${x}" ] && sleep "${x}"

	while [ ${i} -lt 3 ]; do
	    local scan="${scan}${scan:+ }$(LC_ALL=C iwlist "${IFACE}" scan 2>/dev/null | sed -e "s/'/'\\\\''/g" -e "s/$/'/g" -e "s/^/'/g")"
		# If this is the first pass and txpower as off and we have no
		# results then we need to wait for at least 2 seconds whilst
		# the interface does an initial scan.
		if [ "${i}" = "0" -a "${txpowerwasoff}" = "0" ]; then
			case "${scan}" in
				"'${IFACE} "*"No scan results"*)
					sleep 2
					txpowerwasoff=1
					continue
					;;
			esac
		fi
	    i=$((${i} + 1))
	done

	if [ -z "${scan}" ]; then
		ewarn "${iface} does not support scanning"
		eoutdent
		eval x=\$adhoc_ssid_${IFVAR}
		[ -n "${x}" ] && return 0
		if [ -n "${preferred_aps}" ]; then
			[ "${associate_order}" = "forcepreferred" ] || \
			[ "${associate_order}" = "forcepreferredonly" ] && return 0
		fi
		eerror "You either need to set a preferred_aps list in /etc/conf.d/wireless"
		eerror "   preferred_aps=\"SSID1 SSID2\""
		eerror "   and set associate_order_${IFVAR}=\"forcepreferred\""
		eerror "   or set associate_order_${IFVAR}=\"forcepreferredonly\""
		eerror "or hardcode the  SSID to \"any\" and let the driver find an Access Point"
		eerror "   ssid_${IFVAR}=\"any\""
		eerror "or configure defaulting to Ad-Hoc when Managed fails"
		eerror "   adhoc_ssid_${IFVAR}=\"WLAN\""
		eerror "or hardcode the SSID against the interface (not recommended)"
		eerror "   ssid_${IFVAR}=\"SSID\""
		return 1
	fi

	APS=-1
	eval set -- ${scan}
	for line; do
		case "${line}" in
			*Address:*)
				APS=$((${APS} + 1)) 
				eval MAC_${APS}="\""$(echo "${line#*: }" | tr '[:lower:]' '[:upper:]')"\""
				eval QUALITY_${APS}=0
				;;	
			*ESSID:*)
				x=${line#*\"}
				x=${x%*\"}
				eval SSID_${APS}=\$x
				;;
			*Mode:*)
				x="$(echo "${line#*:}" | tr '[:upper:]' '[:lower:]')"
				if [ "${x}" = "master" ]; then
					eval MODE_${APS}=managed
				else
					eval MODE_${APS}=\$x
				fi
				;;
			*'Encryption key:'*)
				x=${line#*:}
				eval ENC_${APS}=\$x
				;;
			#*Frequency:*)
			#		freq[i]="${line#*:}"
			#		x="${freq[i]#* }"
			#		freq[i]="${freq[i]%% *}${x:0:1}"
			#		;;
			*Channel:*)
				x=${line#*:}
				x=${x%% *}
				eval CHAN_${APS}=\$x
				;;
			*Quality*)
				x=${line#*:}
				x=${x%/*}
				x="$(echo "${x}" | sed -e 's/[^[:digit:]]//g')"
				x=${x:-0}
				eval QUALITY_${APS}=\$x
				;;
		esac
	done

	if [ -z "${MAC_0}" ]; then
		ewarn "no access points found"
		eoutdent
		return 1
	fi

	# Sort based on quality
	local i=0 k=1 a= b= x= t=
	while [ ${i} -lt ${APS} ]; do
	    k=$((${i} + 1))
	    while [ ${k} -le ${APS} ]; do
		eval a=\$QUALITY_${i}
		[ -z "${a}" ] && break
		eval b=\$QUALITY_${k}
		if [ -n "${b}" -a "${a}" -lt "${b}" ]; then
		    for x in MAC SSID MODE CHAN QUALITY ENC; do
			eval t=\$${x}_${i}
			eval ${x}_${i}=\$${x}_${k}
			eval ${x}_${k}=\$t
		    done
		fi
		k=$((${k} + 1))
	    done
	    i=$((${i} + 1))
	done

	# Strip any duplicates
	local i=0 k=1 a= b=
	while [ ${i} -lt ${APS} ]; do
		k=$((${i} + 1))
		while [ ${k} -le ${APS} ]; do
			eval a=\$MAC_${i}
			eval b=\$MAC_${k}
			if [ "${a}" = "${b}" ]; then
				eval a=\$QUALITY_${i}
				eval b=\$QUALITY_${k}
				local u=${k}
				# We need to split this into two tests, otherwise bash errors 
				[ -n "${a}" -a -n "${b}" ] && [ "${a}" -lt "${b}" ] && u=${i}
				unset MAC_${u} SSID_${u} MODE_${u} CHAN_${u} QUALITY_${u} ENC_${u}
			fi
			k=$((${k} + 1))
		done
		i=$((${i} + 1))
	done

	local i=0 e= m= s=

	while [ ${i} -le ${APS} ]; do
		eval x=\$MAC_${i}
		if [ -z "${x}" ]; then
		    i=$((${i} + 1))
		    continue
		fi

		eval m=\$MODE_${i}
		eval s=\$SSID_${i}
		eval q=\$QUALITY_${i}
		eval e=\$ENC_${i}
		if [ -n "${e}" -a "${e}" != "off" ]; then
		    e=", encrypted"
		else
		    e=""
		fi
		if [ -z "${s}" ]; then
			einfo "Found ${x}, ${m}${e}"
		else
			einfo "Found \"${s}\" at ${x}, ${m}${e}"
		fi

		x="$(echo "${x}" | sed -e 's/://g')"
		eval x=\$mac_ssid_${x}
		if [ -n "${x}" ]; then
			eval SSID_${i}=\$x
			s=${x}
			eindent
			einfo "mapping to \"${x}\""
			eoutdent
		fi

		eval set -- $(_flatten_array "blacklist_aps_${IFVAR}")
		[ $# = 0 ] && eval set -- $(_flatten_array "blacklist_aps")
		for x; do
			if [ "${x}" = "${s}" ]; then
				ewarn "${s} has been blacklisted - not connecting"
				unset SSID_${i} MAC_${i} ${MODE}_${i} CHAN_${i} QUALITY_${i} ENC_${i}
			fi
		done
		i=$((${i} + 1))
	done
	eoutdent
}

iwconfig_force_preferred()
{
	eval set -- $(_flatten_array "preferred_aps_${IFVAR}")
	[ $# = 0 ] && eval set -- $(_flatten_array "preferred_aps")
	[ $# = 0 ] && return 1

	ewarn "Trying to force preferred in case they are hidden"
	for ssid; do
		local found_AP=false i=0 e=
		while [ ${i} -le ${APS} ]; do
			eval e=\$SSID_${i}
			if [ "${e}" = "${ssid}" ]; then
				found_AP=true
				break
			fi
			i=$((${i} + 1))
		done
		if ! ${found_AP}; then
			SSID=${ssid}
			iwconfig_associate && return 0
		fi
	done

	ewarn "Failed to associate with any preferred access points on ${IFACE}"
	return 1
}

iwconfig_connect_preferred()
{
	local ssid= i= mode= mac= enc= freq= chan=
	eval set -- $(_flatten_array "preferred_aps_${IFVAR}")
	[ $# = 0 ] && eval set -- $(_flatten_array "preferred_aps")

	for ssid; do
		unset IFS
		i=0
		while [ ${i} -le ${APS} ]; do
			eval e=\$SSID_${i}
			if [ "${e}" = "${ssid}" ]; then
				SSID=${e}
				eval mode=\$MODE_${i}
				eval mac=\$MAC_${i}
				eval enc=\$ENC_${i}
				eval freq=\$FREQ_${i}
				eval chan=\$CHAN_${i}
				iwconfig_associate "${mode}" "${mac}" "${enc}" "${freq}" \
					"${chan}" && return 0
			fi
			i=$((${i} + 1))
		done
	done

	return 1
}

iwconfig_connect_not_preferred()
{
	local ssid= i=0 mode= mac= enc= freq= chan= pref=false

	while [ ${i} -le ${APS} ]; do
		eval e=\$SSID_${i}
		if [ -n "${e}" ]; then
			eval set -- $(_flatten_array "preferred_aps_${IFVAR}")
			[ $# = 0 ] && eval set -- $(_flatten_array "preferred_aps")
			for ssid; do
				if [ "${e}" = "${ssid}" ]; then
					pref=true
					break
				fi
			done
	
			if ! ${pref}; then
				SSID=${e}
				eval mode=\$MODE_${i}
				eval mac=\$MAC_${i}
				eval enc=\$ENC_${i}
				eval freq=\$FREQ_${i}
				eval chan=\$CHAN_${i}
				iwconfig_associate "${mode}" "${mac}" "${enc}" "${freq}" \
					"${chan}" && return 0
			fi
		fi
		i=$((${i} + 1))
	done

	return 1
}

iwconfig_defaults()
{
	# Turn on the radio
	iwconfig "${IFACE}" txpower on 2>/dev/null

	# Release the AP forced
	# Must do ap and then ssid otherwise scanning borks
	iwconfig "${IFACE}" ap off 2>/dev/null
	iwconfig "${IFACE}" essid off 2>/dev/null
}

iwconfig_configure()
{
	local x= APS=-1
	eval SSID=\$ssid_${IFVAR}

	# Support old variable
	[ -z "${SSID}" ] && eval SSID=\$essid_${IFVAR}

	# Setup ad-hoc mode?
	eval x=\$mode_${IFVAR}
	x=${x:-managed}
	if [ "${x}" = "ad-hoc" -o "${x}" = "master" ]; then
		iwconfig_setup_specific "${x}"
		return $?
	fi

	if [ "${x}" != "managed" -a "${x}" != "auto" ]; then
		eerror "Only managed, ad-hoc, master and auto modes are supported"
		return 1
	fi

	# Has an SSID been forced?
	if [ -n "${SSID}" ]; then
		iwconfig_set_mode "${x}"
		iwconfig_associate && return 0
		[ "${SSID}" = "any" ] && iwconfig_force_preferred && return 0

		eval SSID=\$adhoc_ssid_${IFVAR}
		if [ -n "${SSID}" ]; then
			iwconfig_setup_specific ad-hoc
			return $?
		fi
		return 1
	fi

	eval x=\$preferred_aps_${IFVAR}
	[ -n "${x}" ] && preferred_aps=${x}

	eval x=\$blacklist_aps_${IFVAR}
	[ -n "${x}" ] && blacklist_aps=${x}

	eval x=\$associate_order_${IFVAR}
	[ -n "${x}" ] && associate_order=${x}
	associate_order=${associate_order:-any}

	if [ "${associate_order}" = "forcepreferredonly" ]; then
		iwconfig_force_preferred && return 0
	else
		iwconfig_scan || return 1
		iwconfig_connect_preferred && return 0
		[ "${associate_order}" = "forcepreferred" ] || \
		[ "${associate_order}" = "forceany" ] && \
		iwconfig_force_preferred && return 0
		[ "${associate_order}" = "any" ] || \
		[ "${associate_order}" = "forceany" ] && \
		iwconfig_connect_not_preferred && return 0
	fi

	e="associate with"
	[ -z "${MAC_0}" ] && e="find"
	[ "${preferred_aps}" = "force" ] || \
	[ "${preferred_aps}" = "forceonly" ] && \
	e="force"
	e="Couldn't ${e} any access points on ${IFACE}"

	eval SSID=\$adhoc_ssid_${IFVAR}
	if [ -n "${SSID}" ]; then
		ewarn "${e}"
		iwconfig_setup_specific ad-hoc
		return $?
	fi

	eerror "${e}"
	return 1
}

iwconfig_pre_start()
{
	# We don't configure wireless if we're being called from
	# the background
	yesno ${IN_BACKGROUND} && return 0

	service_set_value "SSID" ""
	_exists || return 0

	if ! _is_wireless; then
		veinfo "Wireless extensions not found for ${IFACE}"
		return 0
	fi

	# Warn about old file - we want to punt it really
	if [ -e /etc/conf.d/wireless ]; then
		ewarn "/etc/conf.d/wireless is deprecated"
		ewarn "Please put all settings into /etc/conf.d/net"
		. /etc/conf.d/wireless
	fi

	# Store the fact that tx-power was off so we default to a longer
	# wait if our scan returns nothing
	LC_ALL=C iwconfig "${IFACE}" | sed -e '1d' | grep -q "Tx-Power=off"
	local txpowerwasoff=$?
	
	iwconfig_defaults
	iwconfig_user_config
	
	# Set the base metric to be 2000
	metric=2000

	# Check for rf_kill - only ipw supports this at present, but other
	# cards may in the future.
	if [ -e /sys/class/net/"${IFACE}"/device/rf_kill ]; then
		if [ $(cat /sys/class/net/"${IFACE}"/device/rf_kill) != "0" ]; then
			eerror "Wireless radio has been killed for interface ${IFACE}"
			return 1
		fi
	fi

	einfo "Configuring wireless network for ${IFACE}"

	# Are we a proper IEEE device?
	# Most devices reutrn IEEE 802.11b/g - but intel cards return IEEE
	# in lower case and RA cards return RAPCI or similar
	# which really sucks :(
	# For the time being, we will test prism54 not loading firmware
	# which reports NOT READY!
	x="$(iwconfig_get_type)"
	if [ "${x}" = "NOT READY!" ]; then
		eerror "Looks like there was a problem loading the firmware for ${IFACE}"
		return 1
	fi

	if iwconfig_configure; then
		service_set_value "SSID" "${SSID}"
		return 0
	fi

	eerror "Failed to configure wireless for ${IFACE}"
	iwconfig_defaults
	iwconfig "${IFACE}" txpower off 2>/dev/null
	unset SSID SSIDVAR
	_down
	return 1
}

iwconfig_post_stop()
{
	yesno ${IN_BACKGROUND} && return 0
	_exists || return 0
	iwconfig_defaults
	iwconfig "${IFACE}" txpower off 2>/dev/null
}