#!/bin/sh
chip=$(ipcinfo -c)
vendor=$(ipcinfo -v)
keydir=/etc

if [ -e /etc/datalink.conf ]; then
	. /etc/datalink.conf
fi

if [ -e /etc/wfb.conf ]; then
	. /etc/wfb.conf
fi

set_mcs() {
	if [ "$vendor" = "ingenic" ]; then
		mcs=$(ls -l /lib/firmware | grep "htc_9271" | awk {'print $11'} | cut -d "." -f3)
	else
		mcs=$(ls -l /lib/firmware/ath9k_htc | grep "htc_9271-1.4.0.fw" | cut -d "." -f6)
	fi

	if [ -z "$mcs" ]; then
		setmcs "$mcs_index"
	fi

	if [ "$mcs_index" -eq 1 ] || [ "$mcs_index" -eq 3 ]; then
		if [ ! "$mcs_index" -eq "$mcs" ]; then
			setmcs "$mcs_index"
			sleep 3
		fi
	fi
}

# "0bda:8813" -> (8814) -> 8814au
# "0846:9052" -> (8811) -> 8821au

detect_wifi_card() {
	echo "Detecting wifi card vendor..."
	devices=$(lsusb | cut -d ' ' -f6 | sort | uniq)
	for card in $devices; do
		case "$card" in
			"0bda:8812" | "0bda:881a" | "0b05:17d2" | "2357:0101" | "2604:0012")
				driver="realtek"
				modprobe 88XXau rtw_tx_pwr_idx_override="$driver_txpower_override"
				;;

			"0cf3:9271" | "040d:3801")
				driver="atheros"
				if ! [ "$unit" = "gs" ]; then
					set_mcs
				fi
				modprobe mac80211
				modprobe ath9k_hw txpower="$txpower"
				modprobe ath9k_htc
				;;
		esac
	done

	if [ -z "$driver" ]; then
		echo "No usb wifi card detected. Check wifi stick connection, usb power or possible bad soldering."
		exit 1
	else
		echo "Detected: $driver"
	fi

	echo "Awaiting interface $wlan in system..."

	local n=0
	while ! ifconfig -a | grep -q "$wlan"; do
		if [ "$n" -ge 5 ]; then
			echo "No interface $wlan. Check wifi stick connection, usb power or possible bad soldering."
			exit 1
		fi
		sleep 0.5
		n=$(expr $n + 1)
	done
}

load_modules() {
	modprobe cfg80211
	detect_wifi_card
}

load_interface() {
	if [ "$driver" = "realtek" ]; then
		ifconfig "$wlan" up
		iwconfig "$wlan" mode monitor
	elif [ "$driver" = "atheros" ]; then
		iwconfig "$wlan" mode monitor
		ifconfig "$wlan" up
	fi

	iw reg set "$region"

	if [ -n "$frequency" ]; then
		iwconfig "$wlan" freq "$frequency"
	else
		iwconfig "$wlan" channel "$channel"
	fi

	# dirty fix crash if txpower set. setting txpower disabled because patched driver always set txpower level 58
	# iw dev $wlan set txpower fixed $(($txpower * 100))
}

start_drone_wfb() {
	wfb_tx -p "$stream" -u "$udp_port" -R "$rcv_buf" -K "$keydir/$unit.key" -B "$bandwidth" \
		-M "$mcs_index" -S "$stbc" -L "$ldpc" -G "$guard_interval" -k "$fec_k" -n "$fec_n" \
		-T "$pool_timeout" -i "$link_id" -f "$frame_type" "$wlan" > /dev/null &
}

start_gs_wfb() {
	wfb_rx -c "$udp_addr" -u "$udp_port" -p "$stream" -K "$keydir/$unit.key" -i "$link_id" "$wlan" > /dev/null &
}

case "$1" in
	start)
		if [ "$daemon" -eq 1 ]; then
			echo "Loading modules and wifi card driver..."
			load_modules

			echo "Preparing interface wlan..."
			load_interface

			if ! cat "$keydir/$unit.key" > /dev/null 2>&1; then
				echo "Generating drone & ground station keys..."
				cd $keydir && wfb_keygen
			else
				echo "$unit key exist..."
			fi

			echo "Starting Wifibroadcast service..."
			start_"$unit"_wfb
			echo "Done."

			if [ "$telemetry" = "true" ]; then
				if [ "$chip" = "gk7205v200" ]; then
					# UART2_RX mux
					devmem 0x120c0010 32 0x1e04
				fi
				telemetry start
			fi
		else
			echo "Wifibroadcast service disabled in wfb.conf..."
		fi
		;;

	stop)
		echo "Stopping Wifibroadcast services..."
		killall -q wfb_tx
		killall -q wfb_rx
		killall -q telemetry_rx
		killall -q telemetry_tx
		killall -q mavlink-routerd
		killall -q mavfwd
		;;

	*)
		echo "Usage: $0 {start|stop}"
		exit 1
		;;
esac