2 files changed, 47 insertions, 37 deletions

diff --git a/src/soc/intel/tigerlake/elog.c b/src/soc/intel/tigerlake/elog.c
index 88324d63cf..531f5c11ee 100644
--- a/src/soc/intel/tigerlake/elog.c
+++ b/src/soc/intel/tigerlake/elog.c
@@ -57,54 +57,45 @@ static void pch_log_rp_wake_source(void)
 	}
 }
 
-static void pch_log_add_elog_event(const struct pme_map *ipme_map)
-{
-	/*
-	 * If wake source is XHCI, check for detailed wake source events on
-	 * USB2/3 ports.
-	 */
-	if ((ipme_map->devfn == PCH_DEVFN_XHCI) &&
-			pch_xhci_update_wake_event(soc_get_xhci_usb_info()))
-		return;
-
-	elog_add_event_wake(ipme_map->wake_source, 0);
-}
-
 static void pch_log_pme_internal_wake_source(void)
 {
-	size_t i;
-	bool dev_found = false;
-
 	const struct pme_map ipme_map[] = {
 		{ PCH_DEVFN_HDA, ELOG_WAKE_SOURCE_PME_HDA },
 		{ PCH_DEVFN_GBE, ELOG_WAKE_SOURCE_PME_GBE },
 		{ PCH_DEVFN_SATA, ELOG_WAKE_SOURCE_PME_SATA },
 		{ PCH_DEVFN_CSE, ELOG_WAKE_SOURCE_PME_CSE },
-		{ PCH_DEVFN_XHCI, ELOG_WAKE_SOURCE_PME_XHCI },
 		{ PCH_DEVFN_USBOTG, ELOG_WAKE_SOURCE_PME_XDCI },
 		{ PCH_DEVFN_CNVI_WIFI, ELOG_WAKE_SOURCE_PME_WIFI },
 	};
+	const struct xhci_wake_info xhci_wake_info[] = {
+		{ PCH_DEVFN_XHCI,	ELOG_WAKE_SOURCE_PME_XHCI },
+		{ SA_DEVFN_TCSS_XHCI,	ELOG_WAKE_SOURCE_PME_TCSS_XHCI },
+	};
+	bool dev_found = false;
+	size_t i;
 
 	for (i = 0; i < ARRAY_SIZE(ipme_map); i++) {
-		const struct device *dev = pcidev_path_on_root(ipme_map[i].devfn);
+		const struct device *dev =
+			pcidev_path_on_root(ipme_map[i].devfn);
 		if (!dev)
 			continue;
 
 		if (pci_dev_is_wake_source(dev)) {
-			pch_log_add_elog_event(&ipme_map[i]);
+			elog_add_event_wake(ipme_map[i].wake_source, 0);
 			dev_found = true;
 		}
 	}
 
 	/*
-	 * If device is still not found, but the wake source is internal PME,
-	 * try probing XHCI ports to see if any of the USB2/3 ports indicate
-	 * that it was the wake source. This path would be taken in case of GSMI
-	 * logging with S0ix where the pci_pm_resume_noirq runs and clears the
-	 * PME_STS_BIT in controller register.
+	 * Check the XHCI controllers' USB2 & USB3 ports for wake events. There
+	 * are cases (GSMI logging for S0ix clears PME_STS_BIT) where the XHCI
+	 * controller's PME_STS_BIT may have already been cleared, so the host
+	 * controller wake wouldn't get logged here; therefore, the host
+	 * controller wake event is logged before its corresponding port wake
+	 * event is logged.
 	 */
-	if (!dev_found)
-		dev_found = pch_xhci_update_wake_event(soc_get_xhci_usb_info());
+	dev_found |= xhci_update_wake_event(xhci_wake_info,
+					    ARRAY_SIZE(xhci_wake_info));
 
 	if (!dev_found)
 		elog_add_event_wake(ELOG_WAKE_SOURCE_PME_INTERNAL, 0);
diff --git a/src/soc/intel/tigerlake/xhci.c b/src/soc/intel/tigerlake/xhci.c
index 18bb129983..6f095fa633 100644
--- a/src/soc/intel/tigerlake/xhci.c
+++ b/src/soc/intel/tigerlake/xhci.c
@@ -1,20 +1,39 @@
 /* SPDX-License-Identifier: GPL-2.0-only */
 
+#include <device/pci_type.h>
 #include <intelblocks/xhci.h>
+#include <soc/pci_devs.h>
 
-#define XHCI_USB2_PORT_STATUS_REG	0x480
-#define XHCI_USB3_PORT_STATUS_REG	0x520
-#define XHCI_USB2_PORT_NUM		10
-#define XHCI_USB3_PORT_NUM		4
+#define PCH_XHCI_USB2_PORT_STATUS_REG	0x480
+#define PCH_XHCI_USB3_PORT_STATUS_REG	0x520
+#define PCH_XHCI_USB2_PORT_NUM		10
+#define PCH_XHCI_USB3_PORT_NUM		4
 
-static const struct xhci_usb_info usb_info = {
-	.usb2_port_status_reg = XHCI_USB2_PORT_STATUS_REG,
-	.num_usb2_ports = XHCI_USB2_PORT_NUM,
-	.usb3_port_status_reg = XHCI_USB3_PORT_STATUS_REG,
-	.num_usb3_ports = XHCI_USB3_PORT_NUM,
+#define TCSS_XHCI_USB2_PORT_STATUS_REG	0x480
+#define TCSS_XHCI_USB3_PORT_STATUS_REG	0x490
+#define TCSS_XHCI_USB2_PORT_NUM		1
+#define TCSS_XHCI_USB3_PORT_NUM		4
+
+static const struct xhci_usb_info pch_usb_info = {
+	.usb2_port_status_reg = PCH_XHCI_USB2_PORT_STATUS_REG,
+	.num_usb2_ports = PCH_XHCI_USB2_PORT_NUM,
+	.usb3_port_status_reg = PCH_XHCI_USB3_PORT_STATUS_REG,
+	.num_usb3_ports = PCH_XHCI_USB3_PORT_NUM,
+};
+
+static const struct xhci_usb_info tcss_usb_info = {
+	.usb2_port_status_reg = TCSS_XHCI_USB2_PORT_STATUS_REG,
+	.num_usb2_ports = TCSS_XHCI_USB2_PORT_NUM,
+	.usb3_port_status_reg = TCSS_XHCI_USB3_PORT_STATUS_REG,
+	.num_usb3_ports = TCSS_XHCI_USB3_PORT_NUM,
 };
 
-const struct xhci_usb_info *soc_get_xhci_usb_info(void)
+const struct xhci_usb_info *soc_get_xhci_usb_info(pci_devfn_t xhci_dev)
 {
-	return &usb_info;
+	if (xhci_dev == PCH_DEVFN_XHCI)
+		return &pch_usb_info;
+	else if (xhci_dev == SA_DEVFN_TCSS_XHCI)
+		return &tcss_usb_info;
+
+	return NULL;
 }