#!/bin/sh

exec 1> /opt/wz_mini/log/wz_user.log 2>&1

set -x

echo "welcome to wz_user.sh"
echo "PID $$"

export WZMINI_CFG=/opt/wz_mini/wz_mini.conf

[ -f $WZMINI_CFG ] && source $WZMINI_CFG

hostname_set() {
	echo "set hostname"
	hostname $HOSTNAME
}

first_run_check() {
	if [[ -e /opt/wz_mini/tmp/.run_mmc_firstrun ]]; then
	echo "run_mmc.sh already run once, exit."
	exit 0
	fi
}

wait_sdroot() {
##Stall execution if the micro-sd card isn't mounted yet, iCamera controls this internally.
    while true
    do
	if [[ -d /media/mmc/wz_mini ]] || [[ -d /media/mmcblk0p1/wz_mini ]]; then
	echo "sd card ready"
	break
	fi
        echo "sdcard not ready yet..."
        sleep 5
    done

}

store_mac() {
	echo "store original mac"
	cat /sys/class/net/wlan0/address | tr '[:lower:]' '[:upper:]' > /opt/wz_mini/tmp/wlan0_mac
}

wait_wlan() {
##Check if the driver has been loaded for the onboard wlan0, store the MAC.
    while true
    do
        if  ifconfig wlan0 | grep "inet addr";
        then
	store_mac
        break
	elif [[ "$ENABLE_USB_ETH" == "true" || "$ENABLE_USB_DIRECT" == "true" ]]; then
	store_mac
	break
        fi
        echo " wlan0 not ready yet..."
        sleep 5
    done
}

rename_interface() {
##Fool iCamera by renaming the hardline interface to wlan0
	echo "renaming interfaces"
	ifconfig $1 down
	ifconfig wlan0 down
        /media/mmc/wz_mini/bin/busybox ip link set wlan0 name wlanold
        /media/mmc/wz_mini/bin/busybox ip addr flush dev wlanold
        /media/mmc/wz_mini/bin/busybox ip link set $1 name wlan0
	eth_wlan_up
}

eth_wlan_up() {
##Run DHCP client, and bind mount our fake wpa_cli.sh to fool iCamera
        ifconfig wlan0 up
	pkill udhcpc
        udhcpc -i wlan0 -x hostname:$HOSTNAME -p /var/run/udhcpc.pid -b
	if [[ "$V2" == "true" ]]; then
        mount -o bind /media/mmc/wz_mini/bin/wpa_cli.sh /system/bin/wpa_cli
	else
        mount -o bind /media/mmc/wz_mini/bin/wpa_cli.sh /bin/wpa_cli
	fi
	break
}

wpa_check() {
#Check if wpa_supplicant has been created by iCamera
	if [[ -e /tmp/wpa_supplicant.conf ]]; then
		echo "wpa_supplicant.conf ready"
		wlanold_check $1
	else
		echo "wpa_supplicant.conf not ready, wait some time for creation."
		COUNT=0
		ATTEMPTS=15
		until [[ -e /tmp/wpa_supplicant.conf ]] || [[ $COUNT -eq $ATTEMPTS ]]; do
		echo -e "$(( COUNT++ ))... \c"
		sleep 5
		done
		[[ $COUNT -eq $ATTEMPTS ]] && echo "time exceeded waiting for iCamera, continue potentially broken condition without network." && wlanold_check $1
	fi
}

wlanold_check() {
#Have we renamed interfaces yet?
	if [[ -d /sys/class/net/wlanold ]]; then
		echo "wlanold exist"
		eth_wlan_up
	else
		echo "wlanold doesn't exist"
		rename_interface $1
	fi
}

netloop() {
##While loop for check
        while true
        do
        wpa_check $1
        echo "wlan0 not ready yet..."
        sleep 5
        done
}

swap_enable() {
        if [[ -e /media/mmc/wz_mini/swap ]]; then
                echo "Swap exists, enable"
                swapon /media/mmc/wz_mini/swap
        else
                echo "Swap file missing!"
        fi
}

first_run_check
wait_sdroot
wait_wlan

if cat /params/config/.product_config | grep WYZEC1-JZ; then
V2="true"
KMOD_PATH="/opt/wz_mini/lib/modules/3.10.14_v2"
else
V2="false"
KMOD_PATH="/opt/wz_mini/lib/modules/3.10.14__isvp_swan_1.0__"
fi

if [[ "$ENABLE_SWAP" == "true" ]]; then
        if cat /proc/swaps | grep "mini" ; then
        echo "Swap is already enabled"
        else
        echo "Swap not enabled, enabling"
	swap_enable
        fi
fi

if [[ "$ENABLE_IPV6" == "true" ]]; then
	echo "ipv6 enabled"
else
	echo "ipv6 disabled"
	sysctl -w net.ipv6.conf.all.disable_ipv6=1
fi

if [[ "$ENABLE_NFSv4" == "true" ]]; then
        insmod $KMOD_PATH/kernel/lib/oid_registry.ko
        insmod $KMOD_PATH/kernel/net/dns_resolver/dns_resolver.ko
        insmod $KMOD_PATH/kernel/fs/nfs/nfsv4.ko
        insmod $KMOD_PATH/kernel/net/sunrpc/auth_gss/auth_rpcgss.ko
        echo nfsv4 enabled
else
        echo nfsv4 disabled
fi

if [[ "$ENABLE_IPTABLES" == "true"  ]]; then
	if [[ "$V2" == "true" ]]; then
		echo "v2 has iptables built in"
	else
		insmod /lib/modules/3.10.14__isvp_swan_1.0__/kernel/net/netfilter/x_tables.ko
		insmod /lib/modules/3.10.14__isvp_swan_1.0__/kernel/net/ipv4/netfilter/ip_tables.ko
		insmod /lib/modules/3.10.14__isvp_swan_1.0__/kernel/net/ipv4/netfilter/ipt_REJECT.ko
		insmod /lib/modules/3.10.14__isvp_swan_1.0__/kernel/net/ipv4/netfilter/iptable_filter.ko
		insmod /lib/modules/3.10.14__isvp_swan_1.0__/kernel/net/ipv4/netfilter/iptable_mangle.ko
		echo "iptables ipv4 enabled"
	fi

	if [[ "$ENABLE_IPV6" == "true" ]]; then
		insmod $KMOD_PATH/kernel/net/ipv6/netfilter/ip6_tables.ko
		insmod $KMOD_PATH/kernel/net/ipv6/netfilter/ip6t_REJECT.ko
		insmod $KMOD_PATH/kernel/net/ipv6/netfilter/ip6table_filter.ko
		insmod $KMOD_PATH/kernel/net/ipv6/netfilter/ip6table_mangle.ko
		echo "iptables ipv6 enabled"
	fi
else
	echo "iptables disabled"
fi

if [[ "$ENABLE_USB_ETH" == "true" ]]; then

	insmod $KMOD_PATH/kernel/drivers/net/usb/usbnet.ko
        insmod $KMOD_PATH/kernel/drivers/net/usb/asix.ko

	if [[ "$ENABLE_SWAP" == "true" ]]; then
	echo "swap already enabled"
	else
	swap_enable
	fi

	netloop eth0

	else
	echo "usb ethernet disabled"
fi

if [[ "$ENABLE_USB_DIRECT" == "true" ]]; then

        host_macaddr=$(echo $HOSTNAME|md5sum|sed 's/^\(..\)\(..\)\(..\)\(..\)\(..\).*$/02:\1:\2:\3:\4:\5/')

	if [[ "$V2" == "true" ]]; then
		echo connect > /sys/devices/platform/jz-dwc2/dwc2/udc/dwc2/soft_connect
		sleep 1
		devmem 0x10000040 32 0x0b800096
		sleep 1
		devmem 0x13500000 32 0x001100cc
	else
		#Set dwc2 ID_PIN driver memory
		devmem 0x13500000 32 0x001100cc
		devmem 0x10000040 32 0x0b000096
		#wipe the bits to set the ID_PIN, only for the V3.
		devmem 0x10000040 32 0x0b000FFF
	fi

	if [[ "$V2" == "false" ]]; then
	insmod /opt/wz_mini/lib/modules/3.10.14__isvp_swan_1.0__/kernel/drivers/usb/gadget/u_ether.ko
	insmod /opt/wz_mini/lib/modules/3.10.14__isvp_swan_1.0__/kernel/drivers/usb/gadget/usb_f_ncm.ko
	fi

	insmod $KMOD_PATH/kernel/drivers/usb/gadget/libcomposite.ko
	insmod $KMOD_PATH/kernel/drivers/usb/gadget/g_ncm.ko iManufacturer=wz_mini_ncm host_addr=$host_macaddr dev_addr=$USB_DIRECT_MAC_ADDR

	sleep 1

	if [[ "$ENABLE_SWAP" == "true" ]]; then
	echo "swap already enabled"
	else
	swap_enable
	fi

	#loop begin
	while true
	do
	wpa_check usb0
	echo "wlan0 not ready yet..."
        sleep 1
	done
	else
	echo "usb direct disabled"
fi

if [[ "$ENABLE_USB_RNDIS" == "true" ]]; then

	insmod $KMOD_PATH/kernel/drivers/net/usb/usbnet.ko
	insmod $KMOD_PATH/kernel/drivers/net/usb/cdc_ether.ko
	insmod $KMOD_PATH/kernel/drivers/net/usb/rndis_host.ko

	sleep 1

	if [[ "$ENABLE_SWAP" == "true" ]]; then
	echo "swap already enabled"
	else
	swap_enable
	fi

	#loop begin
	while true
	do
	wpa_check usb0
	echo "wlan0 not ready yet..."
        sleep 1
	done
	else
	echo "usb direct disabled"
fi

if [[ "$ENABLE_WIREGUARD" == "true" ]]; then
	insmod $KMOD_PATH/kernel/net/ipv4/tunnel4.ko
	insmod $KMOD_PATH/kernel/net/ipv4/ip_tunnel.ko
	insmod $KMOD_PATH/kernel/net/wireguard/wireguard.ko

  if [[ "$WIREGUARD_IPV4" != "" ]]; then
    mkdir -p /opt/wz_mini/etc/wireguard

    if [ ! -f /opt/wz_mini/etc/wireguard/privatekey ]; then
      (umask 277 && /media/mmc/wz_mini/bin/wg  genkey | /media/mmc/wz_mini/bin/busybox tee /opt/wz_mini/etc/wireguard/privatekey | /media/mmc/wz_mini/bin/wg  pubkey > /opt/wz_mini/etc/wireguard/publickey)
    fi

    /media/mmc/wz_mini/bin/busybox ip link add dev wg0 type wireguard
    /media/mmc/wz_mini/bin/busybox ip address add dev wg0 $WIREGUARD_IPV4
    /media/mmc/wz_mini/bin/wg set wg0 private-key /opt/wz_mini/etc/wireguard/privatekey
    /media/mmc/wz_mini/bin/busybox ip link set wg0 up
  fi

  if [[ "$WIREGUARD_PEER_PUBLIC_KEY" != "" ]] && [[ "$WIREGUARD_PEER_ALLOWED_IPS" != "" ]] && [[ "$WIREGUARD_PEER_ENDPOINT" != "" ]] && [[ "$WIREGUARD_PEER_KEEP_ALIVE" != "" ]]; then
    /media/mmc/wz_mini/bin/wg set wg0 peer $WIREGUARD_PEER_PUBLIC_KEY allowed-ips $WIREGUARD_PEER_ALLOWED_IPS endpoint $WIREGUARD_PEER_ENDPOINT persistent-keepalive $WIREGUARD_PEER_KEEP_ALIVE
    /media/mmc/wz_mini/bin/busybox ip route add $WIREGUARD_PEER_ALLOWED_IPS dev wg0
  fi
else
	echo "wireguard disabled"
fi

if [[ "$ENABLE_CIFS" == "true" ]]; then
	insmod $KMOD_PATH/kernel/fs/cifs/cifs.ko
else
	echo "cifs disabled"
fi

if [[ "$ENABLE_USB_STORAGE" == "true" ]]; then
	insmod $KMOD_PATH/kernel/drivers/scsi/scsi_mod.ko
	insmod $KMOD_PATH/kernel/drivers/scsi/sd_mod.ko
	insmod $KMOD_PATH/kernel/drivers/usb/storage/usb-storage.ko
else
	echo "usb_storage disabled"
fi

if [[ "$ENABLE_EXT4" == "true" ]]; then
	if [[ "$V2" == "true" ]]; then
	insmod $KMOD_PATH/kernel/lib/crc16.ko
	fi

	insmod $KMOD_PATH/kernel/fs/jbd2/jbd2.ko
	insmod $KMOD_PATH/kernel/fs/mbcache.ko
	insmod $KMOD_PATH/kernel/fs/ext4/ext4.ko
else
	echo "ext4 disabled"
fi

if [[ "$DISABLE_FW_UPGRADE" == "true" ]]; then
	mkdir /tmp/Upgrade
	mount -t tmpfs -o size=1,nr_inodes=1 none /tmp/Upgrade
	echo -e "127.0.0.1 localhost \n127.0.0.1 wyze-upgrade-service.wyzecam.com" > /opt/wz_mini/tmp/.storage/hosts
	mount --bind /opt/wz_mini/tmp/.storage/hosts /etc/hosts
else
        mkdir /tmp/Upgrade
        /opt/wz_mini/bin/busybox inotifyd /opt/wz_mini/usr/bin/watch_up.sh /tmp:n &
fi

if [[ "$REMOTE_SPOTLIGHT" == "true" ]]; then
	/media/mmc/wz_mini/bin/socat pty,link=/dev/ttyUSB0,raw tcp:$REMOTE_SPOTLIGHT_HOST:9000 &
	echo "remote accessory enabled"
else
	echo "remote accessory disabled"
fi

if [[ "$ENABLE_MP4_WRITE" == "true" ]]; then
        if [[ "$V2" == "true" ]]; then
		echo "mp4_write not supported on v2"
	else
		/opt/wz_mini/bin/cmd mp4write on
		echo "mp4_write enabled"
	fi
else
	echo "mp4 write disabled"
fi

if [[ "$RTSP_HI_RES_ENABLED" == "true" ]]; then

        if [[ "$V2" == "true" ]]; then
	HI_VIDEO_DEV="/dev/video6"
	else
	HI_VIDEO_DEV="/dev/video1"
	fi

	if [[ "$ENABLE_SWAP" == "true" ]]; then
	echo "swap already enabled"
	else
	swap_enable
	fi

	if [[ "$RTSP_PASSWORD" = "" ]]; then
	RTSP_PASSWORD=$(cat /opt/wz_mini/tmp/wlan0_mac)
	fi

	/opt/wz_mini/bin/cmd video on

        if [[ "$RTSP_HI_RES_ENABLE_AUDIO" == "true" ]]; then
		/opt/wz_mini/bin/cmd audio on
		AUDIO_CH="-C 1"
		AUDIO_FMT="-a S16_LE"
		DEVICE1="$HI_VIDEO_DEV,hw:Loopback,0"
        else
                DEVICE1="$HI_VIDEO_DEV"
		echo "rtsp audio disabled"
        fi

	if [[ "$RTSP_HI_RES_ENC_PARAMETER" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:0:4:$RTSP_LOW_RES_ENC_PARAMETER" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:44:4:$RTSP_HI_RES_ENC_PARAMETER" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_HI_RES_MAX_BITRATE" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:28:4:$RTSP_LOW_RES_MAX_BITRATE" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:48:4:$RTSP_HI_RES_MAX_BITRATE" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_HI_RES_TARGET_BITRATE" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			echo "not supported on v2"
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:52:4:$RTSP_HI_RES_TARGET_BITRATE" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_HI_RES_FPS" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:8:4:$RTSP_HI_RES_FPS" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 0:80:4:$RTSP_HI_RES_FPS" > /dev/null 2>&1 &
		fi
	fi

        else
        echo "rtsp disabled"

fi


if [[ "$RTSP_LOW_RES_ENABLED" == "true" ]]; then

	if [[ "$V2" == "true" ]]; then
	LOW_VIDEO_DEV="/dev/video7"
	else
	LOW_VIDEO_DEV="/dev/video2"
	fi


	if [[ "$ENABLE_SWAP" == "true" ]]; then
	echo "swap already enabled"
	else
	swap_enable
	fi

	/opt/wz_mini/bin/cmd video on1

	if [[ "$RTSP_PASSWORD" = "" ]]; then
	RTSP_PASSWORD=$(cat /opt/wz_mini/tmp/wlan0_mac)
	fi

        if [[ "$RTSP_LOW_RES_ENABLE_AUDIO" == "true" ]]; then
		/opt/wz_mini/bin/cmd audio on1
		AUDIO_CH="-C 1"
		AUDIO_FMT="-a S16_LE"
		DEVICE2="$LOW_VIDEO_DEV,hw:Loopback,1"
        else
                DEVICE2="$LOW_VIDEO_DEV"
                echo "rtsp audio disabled"
        fi

	if [[ "$RTSP_LOW_RES_ENC_PARAMETER" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:0:4:$RTSP_LOW_RES_ENC_PARAMETER" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:44:4:$RTSP_LOW_RES_ENC_PARAMETER" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_LOW_RES_MAX_BITRATE" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:28:4:$RTSP_LOW_RES_MAX_BITRATE" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:48:4:$RTSP_LOW_RES_MAX_BITRATE" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_LOW_RES_TARGET_BITRATE" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			echo "not supported on v2"
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:52:4:$RTSP_LOW_RES_TARGET_BITRATE" > /dev/null 2>&1 &
		fi
	fi

	if [[ "$RTSP_LOW_RES_FPS" != "" ]]; then
		if [[ "$V2" == "true" ]]; then
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:8:4:$RTSP_LOW_RES_FPS" > /dev/null 2>&1 &
		else
			watch -n30 -t "/system/bin/impdbg --enc_rc_s 1:80:4:$RTSP_LOW_RES_FPS" > /dev/null 2>&1 &
		fi
	fi

        else
        echo "rtsp disabled"

fi

if [[ "$RTSP_LOW_RES_ENABLED" == "true" ]] || [[ "$RTSP_HI_RES_ENABLED" == "true" ]]; then
	LD_LIBRARY_PATH=/media/mmc/wz_mini/lib /media/mmc/wz_mini/bin/v4l2rtspserver $AUDIO_CH $AUDIO_FMT -U $RTSP_LOGIN:$RTSP_PASSWORD -P $RTSP_PORT $DEVICE1 $DEVICE2 &
fi

hostname_set
touch /opt/wz_mini/tmp/.run_mmc_firstrun
pkill -f dumpload #Kill dumpload so it won't waste cpu or ram gathering cores when something crashes
sysctl -w kernel.core_pattern='|/bin/false'
sync;echo 3 > /proc/sys/vm/drop_caches


if [ -f "$CUSTOM_SCRIPT_PATH" ]; then
	echo "starting custom script"
	$CUSTOM_SCRIPT_PATH &
else
	echo "custom script not found"
fi