ucsi wakelock

Signed-off-by: Jishnu Prakash <jishnu.prakash@oss.qualcomm.com>

Author: jprakash-qc

drivers/usb/typec/ucsi/ucsi_glink.c

@@ -7,13 +7,23 @@
 #include <linux/module.h>
 #include <linux/mutex.h>
 #include <linux/of_device.h>
+#include <linux/pm_wakeup.h>
 #include <linux/property.h>
 #include <linux/soc/qcom/pdr.h>
 #include <linux/usb/typec_mux.h>
 #include <linux/gpio/consumer.h>
 #include <linux/soc/qcom/pmic_glink.h>
 #include "ucsi.h"
 
+/*
+ * Wakeup timeout covering two async hops:
+ *   1. pmic_glink_ucsi_notify() work runs ucsi_notify_common()
+ *   2. ucsi_handle_connector_change() work runs and notifies USB
+ * ucsi_handle_connector_change() involves GLINK round-trips with up to 5s
+ * timeouts, so 10s gives sufficient margin for the full chain to complete.
+ */
+#define UCSI_GLINK_WAKEUP_TIMEOUT_MS	200
+
 #define PMIC_GLINK_MAX_PORTS		3
 
 #define UCSI_BUF_V1_SIZE		(UCSI_MESSAGE_OUT + (UCSI_MESSAGE_OUT - UCSI_MESSAGE_IN))
@@ -342,6 +352,7 @@ static void pmic_glink_ucsi_callback(const void *data, size_t len, void *priv)
 		pmic_glink_ucsi_write_ack(ucsi, data, len);
 		break;
 	case UC_UCSI_USBC_NOTIFY_IND:
+		pm_wakeup_event(ucsi->dev, UCSI_GLINK_WAKEUP_TIMEOUT_MS);
 		schedule_work(&ucsi->notify_work);
 		break;
 	}
@@ -401,6 +412,8 @@ static int pmic_glink_ucsi_probe(struct auxiliary_device *adev,
 	ucsi->dev = dev;
 	dev_set_drvdata(dev, ucsi);
 
+	device_init_wakeup(dev, true);
+
 	INIT_WORK(&ucsi->notify_work, pmic_glink_ucsi_notify);
 	INIT_WORK(&ucsi->register_work, pmic_glink_ucsi_register);
 	init_completion(&ucsi->read_ack);