#!/bin/sh

export PATH=/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
UPDATE_TARBALL=update.tar.gz

echo "Update Script started"
pre_vers=`head -n 1 /etc/cabversion`

path=$PWD

# Working directory for passwd, group, and shadow merges
pwdupdir="/tmp/pwdupdir"
mkdir -p ${pwdupdir}
#<PASSWD>
#<GROUP>
#<SHADOW>

# Save password files before extracting the TAR archive
for f in passwd group shadow; do
	cp "/etc/${f}" "${pwdupdir}/${f}.bkp"
done

cd /
## tar will fail immediately if a directory is about to be replaced
## by a file or a symlink. Thus, check here for this problem and
## remove directories before unpacking the tar file.
tar tf ${path}/${UPDATE_TARBALL} | awk '
	/[^\/]$/ {
		if( system("test -d \"" $0 "\" -a ! -h \"" $0 "\"") == 0 ) {
			print("  Replacing directory " $0 " with file or symlink");
			system("rm -rf \"" $0 "\"" );
		}
	}
'
tar xzf ${path}/${UPDATE_TARBALL}
sync

#<REMOVE>
rm -f /etc/factory/scripts.d/monitor/sanemonitor.sh
rm -f /etc/saneclient.d/hooks/call.inc
rm -f /etc/saneclient.d/hooks/disable-bondingproxy.sh
rm -f /etc/saneclient.d/hooks/enable-bondingproxy.sh
rm -f /etc/saneclient.d/hooks/ipsec.inc
rm -f /etc/saneclient.d/hooks/tunnel-addroute.sh
rm -f /etc/saneclient.d/hooks/tunnel-connect.sh
rm -f /etc/saneclient.d/hooks/tunnel-delroute.sh
rm -f /etc/saneclient.d/hooks/tunnel-disconnect.sh
rm -f /etc/ssl/certs/.gitignore
rm -f /lib/modules/6.1.76/kernel/arch/arm/crypto/chacha-neon.ko
rm -f /lib/modules/6.1.76/kernel/arch/arm/crypto/poly1305-arm.ko
rm -f /lib/modules/6.1.76/kernel/crypto/crypto_engine.ko
rm -f /lib/modules/6.1.76/kernel/crypto/curve25519-generic.ko
rm -f /lib/modules/6.1.76/kernel/crypto/ecb.ko
rm -f /lib/modules/6.1.76/kernel/crypto/xts.ko
rm -f /lib/modules/6.1.76/kernel/drivers/char/tpm/tpm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/char/tpm/tpm_i2c_atmel.ko
rm -f /lib/modules/6.1.76/kernel/drivers/char/tpm/tpm_tis_core.ko
rm -f /lib/modules/6.1.76/kernel/drivers/char/tpm/tpm_tis_spi.ko
rm -f /lib/modules/6.1.76/kernel/drivers/crypto/caam/caam.ko
rm -f /lib/modules/6.1.76/kernel/drivers/crypto/caam/caam_jr.ko
rm -f /lib/modules/6.1.76/kernel/drivers/crypto/caam/caamalg_desc.ko
rm -f /lib/modules/6.1.76/kernel/drivers/crypto/caam/caamhash_desc.ko
rm -f /lib/modules/6.1.76/kernel/drivers/crypto/caam/error.ko
rm -f /lib/modules/6.1.76/kernel/drivers/i2c/algos/i2c-algo-pca.ko
rm -f /lib/modules/6.1.76/kernel/drivers/i2c/algos/i2c-algo-pcf.ko
rm -f /lib/modules/6.1.76/kernel/drivers/input/serio/serport.ko
rm -f /lib/modules/6.1.76/kernel/drivers/irqchip/irq-imx-mu-msi.ko
rm -f /lib/modules/6.1.76/kernel/drivers/misc/eeprom/eeprom_93cx6.ko
rm -f /lib/modules/6.1.76/kernel/drivers/modules/gpio/gpio.ko
rm -f /lib/modules/6.1.76/kernel/drivers/modules/pae/pae.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/bareudp.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/ethernet/freescale/enetc/fsl-enetc-ierb.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/ethernet/freescale/enetc/fsl-enetc-mdio.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/ethernet/freescale/enetc/fsl-enetc.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/ipvlan/ipvlan.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/ipvlan/ipvtap.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/mdio/mdio-i2c.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/mdio/mdio-netlink.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/mhi_net.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/pcs/pcs-lynx.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/phy/ax88796b.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/phy/phylink.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/phy/sfp.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/tap.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/asix.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/cdc_ether.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/cdc_mbim.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/cdc_ncm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/huawei_cdc_ncm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/qmi_wwan.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/r8153_ecm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/sierra_net.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/smsc75xx.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/usb/usbnet.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireguard/wireguard.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ath/ath.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ath/ath11k/ath11k.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ath/ath11k/ath11k_pci.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/intel/iwlwifi/iwlwifi.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/intel/iwlwifi/mvm/iwlmvm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/mediatek/mt76/mt76-connac-lib.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/mediatek/mt76/mt76.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/mediatek/mt76/mt7915/mt7915e.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/mediatek/mt76/mt7921/mt7921-common.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/mediatek/mt76/mt7921/mt7921e.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ralink/rt2x00/rt2800lib.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ralink/rt2x00/rt2800usb.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ralink/rt2x00/rt2x00lib.ko
rm -f /lib/modules/6.1.76/kernel/drivers/net/wireless/ralink/rt2x00/rt2x00usb.ko
rm -f /lib/modules/6.1.76/kernel/drivers/ptp/ptp-qoriq.ko
rm -f /lib/modules/6.1.76/kernel/drivers/soc/qcom/qmi_helpers.ko
rm -f /lib/modules/6.1.76/kernel/drivers/tty/serial/sc16is7xx.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/class/cdc-acm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/class/cdc-wdm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/function/u_ether.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/function/usb_f_ecm.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/function/usb_f_ecm_subset.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/function/usb_f_mass_storage.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/function/usb_f_rndis.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/legacy/g_ether.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/legacy/g_mass_storage.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/gadget/libcomposite.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/cp210x.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/ftdi_sio.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/option.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/pl2303.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/sierra.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/usb-serial-simple.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/usb_wwan.ko
rm -f /lib/modules/6.1.76/kernel/drivers/usb/serial/usbserial.ko
rm -f /lib/modules/6.1.76/kernel/fs/binfmt_misc.ko
rm -f /lib/modules/6.1.76/kernel/fs/configfs/configfs.ko
rm -f /lib/modules/6.1.76/kernel/kernel/configs.ko
rm -f /lib/modules/6.1.76/kernel/lib/crc-itu-t.ko
rm -f /lib/modules/6.1.76/kernel/lib/crc7.ko
rm -f /lib/modules/6.1.76/kernel/lib/crypto/libchacha20poly1305.ko
rm -f /lib/modules/6.1.76/kernel/lib/crypto/libcurve25519-generic.ko
rm -f /lib/modules/6.1.76/kernel/lib/crypto/libcurve25519.ko
rm -f /lib/modules/6.1.76/kernel/lib/libcrc32c.ko
rm -f /lib/modules/6.1.76/kernel/net/batman-adv/batman-adv.ko
rm -f /lib/modules/6.1.76/kernel/net/ipv4/esp4_offload.ko
rm -f /lib/modules/6.1.76/kernel/net/ipv4/udp_tunnel.ko
rm -f /lib/modules/6.1.76/kernel/net/ipv6/ip6_udp_tunnel.ko
rm -f /lib/modules/6.1.76/kernel/net/netfilter/xt_TCPMSS.ko
rm -f /lib/modules/6.1.76/kernel/net/xfrm/xfrm_interface.ko
rm -f /lib/modules/6.1.76/modules.alias
rm -f /lib/modules/6.1.76/modules.alias.bin
rm -f /lib/modules/6.1.76/modules.builtin
rm -f /lib/modules/6.1.76/modules.builtin.bin
rm -f /lib/modules/6.1.76/modules.builtin.modinfo
rm -f /lib/modules/6.1.76/modules.dep
rm -f /lib/modules/6.1.76/modules.dep.bin
rm -f /lib/modules/6.1.76/modules.devname
rm -f /lib/modules/6.1.76/modules.order
rm -f /lib/modules/6.1.76/modules.softdep
rm -f /lib/modules/6.1.76/modules.symbols
rm -f /lib/modules/6.1.76/modules.symbols.bin
rm -f /opt/bondix/saneclient
rm -f /sbin/devmem
rm -f /sbin/ifdown
rm -f /sbin/ifup
rm -f /usr/lib/libcurses.so
rm -f /usr/lib/libform.so
rm -f /usr/lib/libform.so.6
rm -f /usr/lib/libform.so.6.1
rm -f /usr/lib/libite.so
rm -f /usr/lib/libite.so.5
rm -f /usr/lib/libite.so.5.0.1
rm -f /usr/lib/libmenu.so
rm -f /usr/lib/libmenu.so.6
rm -f /usr/lib/libmenu.so.6.1
rm -f /usr/lib/libncurses.so
rm -f /usr/lib/libncurses.so.6
rm -f /usr/lib/libncurses.so.6.1
rm -f /usr/lib/libpanel.so
rm -f /usr/lib/libpanel.so.6
rm -f /usr/lib/libpanel.so.6.1
rm -f /usr/lib/libpcap.so.1.9.1
rm -f /usr/sbin/tcpdump
sync

# Unpack tar file again.
# If directories are replaced by a symlink,
# the removes above might delete files.
# Restore them here again.
tar xzf ${path}/${UPDATE_TARBALL}
sync

## Merge password files if they have changed.
## Therefore, determine which entries are new
## or have changed and apply them to the new
## file from the update.
## A changed entry will be dropped if the entry
## is removed by the update.
for file in passwd group shadow; do
	# Variables
	fnew="/etc/${file}"
	fbkp="${pwdupdir}/${file}.bkp"
	fold="${pwdupdir}/${file}"
	fdiff="${pwdupdir}/${file}.diff"
	mode="$(stat -c '%a' ${fnew})"

	# Skip if file has not changed
	diff -q ${fnew} ${fbkp} &>/dev/null && continue

	# Removing all exactly matching lines of old and new from the
	# backup file. The result is changed or new entries.
	grep -v -x -F -f ${fnew} -f ${fold} ${fbkp} > ${fdiff}

	# Create a temporary file for the modifications
	tmp="$(mktemp ${pwdupdir}/${file}-XXXXXX)"
	# Apply the changes to the new file
	awk -F: '
	function pr(type, msg) {
		printf( "  %-7s %7s %s\n", "'"${file}"':", type, msg) > "/dev/stderr";
	}
	function FileToMap(file, map,    a, cmd, line, n) {
		# Get entries from file and store them in an array
		# using the first token until ":" as key.
		while ( (getline line < file) > 0 ) {
			n = split(line, a, ":");
			if ( n < 2 )
				continue;
			map[a[1]] = line;
		}
		close(file);
	}
	function append(a, b) {
		if ( a != "" )
			a = a ", ";

		return a b;
	}
	BEGIN {
		# Store file entries in arrays
		# Note that awk passes arrays as reference
		FileToMap("'"${fdiff}"'", dMap);
		FileToMap("'"${fold}"'", oMap);
		FileToMap("'"${fbkp}"'", bMap);
	}
	/^[^:]+:/ {
		# Check if this entry has been changed
		if ( $1 in dMap ) {
			# Yes, use changed entry
			print dMap[$1];
			delete dMap[$1];
			msg["MERGED"] = append(msg["MERGED"], $1);
		} else {
			# No, keep the entry
			if ( !($1 in oMap) && !($1 in bMap) ) {
				# Was neither in old nor in bkp file,
				# so it must be a new entry
				msg["NEW"] = append(msg["NEW"], $1);
			}
			print $0;
		}
		# Remove entries from old map. Eventually, the remaining
		# entries in the old map are the entries that were removed.
		delete oMap[$1];
		next;
	}
	{
		# Other lines, keep them as they are
		print $0;
		next;
	}
	END {
		# Loop over entries that were in the old but not in the
		# new file anymore.
		for ( i in oMap ) {
			if ( i in dMap ) {
				# Entry has been changed but is removed now.
				delete dMap[i];
				msg["DROPPED"] = append(msg["DROPPED"], i);
			} else {
				# Entry was removed during the update.
				msg["REMOVED"] = append(msg["REMOVED"], i);
			}
		}

		# Loop over remaining changed entries
		for ( i in dMap ) {
			# Entries that were neither in old nor new file.
			print dMap[i];
			msg["EXTRA"] = append(msg["EXTRA"], i);
		}

		# Print results
		for ( i in msg )
			pr(i, msg[i]);

	}
	' ${fnew} 2>&1 >${tmp}
	if [ $? -ne 0 ]; then
		echo -e "${file}:\tMERGE FAILED, CHECK PASSWORDS AND GROUPS!!!"
		continue
	fi

	# Update new file but keep the previous file as backup
	mv ${fbkp} ${fnew}-
	mv ${tmp} ${fnew}
	# ${tmp} had mode 600, so change it correctly
	chmod ${mode} ${fnew} ${fnew}-
done

# Clean-up
rm -rf ${pwdupdir}
sync

cd ${path}

## Update Kernel if present
if [ -r /root/apf6-linux.bin ]; then
	echo "Installing new kernel"
	mount -t ext4 /dev/mmcblk2p1 /mnt

	mv /root/apf6-linux.bin /mnt/

	sync
	umount /mnt
fi

## Update Device Trees if present
if ls /root/imx6*.dtb &> /dev/null; then
	echo "Installing new Device Trees"
	mount -t ext4 /dev/mmcblk2p1 /mnt

	for i in /root/imx6*.dtb; do
		if [ -r "$i" ]; then
			echo "  * $(basename "$i")"
			mv $i /mnt/
		fi
	done
	sync
	umount /mnt
fi

## Update Bootloader (u-boot) if present
if [ -r /root/apf6-u-boot.img -a -r /root/apf6-u-boot.spl ]; then
	echo "Installing new bootloader"

	echo 0 > /sys/block/mmcblk2boot0/force_ro

	# Flash SPL
	dd if=/root/apf6-u-boot.spl of=/dev/mmcblk2boot0 bs=512 seek=2 status=none
	# Flash bootloader
	dd if=/root/apf6-u-boot.img of=/dev/mmcblk2boot0 bs=512 seek=138 status=none
	# Erase environment and backup environment
	dd if=/dev/zero of=/dev/mmcblk2boot0 bs=512 seek=2048 count=2048 status=none

	echo 1 > /sys/block/mmcblk2boot0/force_ro

	rm /root/apf6-u-boot.*

	sync
fi

ldconfig

echo "Update finished."

post_vers=`head -n 1 /etc/cabversion`
echo "Updated from ${pre_vers} to ${post_vers}"

echo "The config file /etc/cablynx.conf was untouched."
echo "Please use the new config file in /etc/conf.d/cablynx.factory"

