From 6f3f36e6a56a24ceb5743b3be65c9d925da7918d Mon Sep 17 00:00:00 2001
From: Katharine Chui <kwchuiaa@connect.ust.hk>
Date: Sat, 3 Feb 2024 02:16:46 +0800
Subject: [PATCH] restart bthal on wake, allow user defined module reprobe

---
 device.mk     |   2 +
 init.sh       |   2 +
 post_sleep.sh | 122 ++++++++++++++++++++++++++++++++++++++++++++++++++
 pre_sleep.sh  |  17 +++++++
 4 files changed, 143 insertions(+)
 create mode 100644 post_sleep.sh
 create mode 100644 pre_sleep.sh

diff --git a/device.mk b/device.mk
index 5f09ebed..d33547d7 100644
--- a/device.mk
+++ b/device.mk
@@ -73,6 +73,8 @@ endif
 PRODUCT_COPY_FILES := \
     $(if $(wildcard $(PRODUCT_DIR)init.rc),$(PRODUCT_DIR)init.rc:root/init.rc) \
     $(if $(wildcard $(PRODUCT_DIR)init.sh),$(PRODUCT_DIR),$(LOCAL_PATH)/)init.sh:system/etc/init.sh \
+    $(if $(wildcard $(PRODUCT_DIR)pre_sleep.sh),$(PRODUCT_DIR),$(LOCAL_PATH)/)pre_sleep.sh:system/etc/pre_sleep.sh \
+    $(if $(wildcard $(PRODUCT_DIR)post_sleep.sh),$(PRODUCT_DIR),$(LOCAL_PATH)/)post_sleep.sh:system/etc/post_sleep.sh \
     $(if $(wildcard $(PRODUCT_DIR)modules.blocklist),$(PRODUCT_DIR),$(LOCAL_PATH)/)modules.blocklist:system/etc/modules.blocklist \
     $(if $(wildcard $(PRODUCT_DIR)modules.options),$(PRODUCT_DIR),$(LOCAL_PATH)/)modules.options:system/etc/modules.options \
     $(if $(wildcard $(PRODUCT_DIR)fstab.$(TARGET_PRODUCT)),$(PRODUCT_DIR)fstab.$(TARGET_PRODUCT),$(LOCAL_PATH)/fstab.x86):root/fstab.$(TARGET_PRODUCT) \
diff --git a/init.sh b/init.sh
index f64b8d75..92884f5a 100644
--- a/init.sh
+++ b/init.sh
@@ -214,8 +214,10 @@ function init_hal_bluetooth()
 
 	if [ "$BTLINUX_HAL" = "1" ]; then
 		start btlinux-1.1
+		set_property "ro.bliss.bthal" "btlinux"
 	else
 		start vendor.bluetooth-1-1
+		set_property "ro.bliss.bthal" "celadon"
 	fi
 }
 
diff --git a/post_sleep.sh b/post_sleep.sh
new file mode 100644
index 00000000..683fe210
--- /dev/null
+++ b/post_sleep.sh
@@ -0,0 +1,122 @@
+# called from modified suspend hal
+
+# prepare env
+DMIPATH=/sys/class/dmi/id
+export BOARD=$(cat $DMIPATH/board_name)
+export PRODUCT=$(cat $DMIPATH/product_name)
+export VENDOR=$(cat $DMIPATH/sys_vendor)
+export UEVENT=$(cat $DMIPATH/uevent)
+
+function reprobe_module_if_loaded()
+{
+	module=$1
+	for modname in $(lsmod | awk '{print $1}')
+	do
+		if [ "$module" == "$modname" ]
+		then
+			modprobe -rv $module
+			modprobe -v $module
+			return
+		fi
+	done
+	echo module $module not loaded, not reprobing
+}
+
+# prep the prebaked list here, if there are known devices that definitely needs this work-around
+MODULE_RELOAD_LIST=""
+# set RESTART_WIFI to true if a wifi module is going to be reloaded, note that users will have to manually turn on wifi again after a wificond restart
+RESTART_WIFI=false
+# set RESET_BT to true if a bluetooth module is going to be reloaded, note that user might have to manually turn on bluetooth again after bthal restart
+RESTART_BT=false
+
+### EXAMPLE:
+### if [ "$BOARD" == "xxx" ]
+### then
+###   MODULE_RELOAD_LIST="$MODULE_RELOAD_LIST iwlwifi btusb"
+###   RESTART_WIFI=true
+###   RESTART_BT=true
+### fi
+
+# usually bluetooth breaks when waking from s3 resets the device, then bthal continues to communicate to it with existing sockets as if the reset had never happened
+# while disabling bthal before suspend then starting it after is enough on the steamdeck, reprobing the module should make it work better in a more general sense
+RESTART_BT=true
+MODULE_RELOAD_LIST="$MODULE_RELOAD_LIST btusb"
+
+# users can use this to flag wificond restart
+USER_RESTART_WIFI_FLAG=/data/etc/wake_reload_wifi
+if [ -e $USER_RESTART_WIFI_FLAG ]
+then
+	RESTART_WIFI=true
+fi
+
+# users can use this to flag bluetooth restart
+USER_RESTART_BT_FLAG=/data/etc/wake_reload_bt
+if [ -e $USER_RESTART_BT_FLAG ]
+then
+	RESTART_BT=true
+fi
+
+# stop services if requested
+if $RESTART_BT
+then
+	setprop ctl.stop btlinux-1.1
+	setprop ctl.stop vendor.bluetooth-1-1
+fi
+
+if $RESTART_WIFI
+then
+	setprop ctl.stop wificond
+fi
+
+# setprop ctl.stop should be instant, but just in case
+if $RESTART_WIFI || $RESTART_BT
+then
+	sleep 0.2
+fi
+
+# perform module reprobe
+MODULE_RELOAD_LOG=/data/wake_module_reload_log
+rm -f $MODULE_RELOAD_LOG
+if [ -n "$MODULE_RELOAD_LIST" ]
+then
+	for m in $MODULE_RELOAD_LIST
+	do
+		reprobe_module_if_loaded $m 2>&1 | cat >> $MODULE_RELOAD_LOG
+	done
+fi
+
+# let users define a list of modules to reload on wake
+USER_MODULE_RELOAD_LIST=/data/etc/wake_module_reload_list
+if [ -e $USER_MODULE_RELOAD_LIST ]
+then
+	for m in $(cat $USER_MODULE_RELOAD_LIST)
+	do
+		reprobe_module_if_loaded $m 2>&1 | cat >> $MODULE_RELOAD_LOG
+	done
+fi
+
+# start services again
+if $RESTART_WIFI
+then
+	setprop ctl.start wificond
+fi
+
+if $RESTART_BT
+then
+	bthal=$(getprop ro.bliss.bthal)
+	case $bthal in
+		"btlinux")
+			setprop ctl.start btlinux-1.1
+			;;
+		"celadon")
+			setprop ctl.start vendor.bluetooth-1-1
+			;;
+	esac
+fi
+
+# allow user defined actions
+USER_SCRIPT=/data/etc/post_sleep.sh
+if [ -e $USER_SCRIPT ]
+then
+	/system/bin/sh $USER_SCRIPT
+fi
diff --git a/pre_sleep.sh b/pre_sleep.sh
new file mode 100644
index 00000000..98be46d8
--- /dev/null
+++ b/pre_sleep.sh
@@ -0,0 +1,17 @@
+# called from modified suspend hal
+
+# prepare env
+DMIPATH=/sys/class/dmi/id
+export BOARD=$(cat $DMIPATH/board_name)
+export PRODUCT=$(cat $DMIPATH/product_name)
+export VENDOR=$(cat $DMIPATH/sys_vendor)
+export UEVENT=$(cat $DMIPATH/uevent)
+
+# put prebaked actions here
+
+# allow user defined actions
+USER_SCRIPT=/data/etc/pre_sleep.sh
+if [ -e $USER_SCRIPT ]
+then
+	/system/bin/sh $USER_SCRIPT
+fi