1 files changed, 5 insertions, 21 deletions
diff --git a/src/soc/intel/apollolake/memmap.c b/src/soc/intel/apollolake/memmap.c
index c54423c944..f9a484a7c4 100644
--- a/src/soc/intel/apollolake/memmap.c
+++ b/src/soc/intel/apollolake/memmap.c
@@ -28,40 +28,24 @@
 #include <cbmem.h>
 #include <device/pci.h>
 #include <fsp/memmap.h>
+#include <intelblocks/smm.h>
 #include <soc/systemagent.h>
 #include <soc/pci_devs.h>
 
-static uintptr_t smm_region_start(void)
-{
-	return ALIGN_DOWN(pci_read_config32(SA_DEV_ROOT, TSEG), 1*MiB);
-}
-
-static size_t smm_region_size(void)
-{
-	uintptr_t smm_end =
-		ALIGN_DOWN(pci_read_config32(SA_DEV_ROOT, BGSM), 1*MiB);
-	return smm_end - smm_region_start();
-}
-
 void *cbmem_top(void)
 {
-	return (void *)smm_region_start();
-}
-
-void smm_region(void **start, size_t *size)
-{
-	*start = (void *)smm_region_start();
-	*size = smm_region_size();
+	return (void *)sa_get_tseg_base();
 }
 
 int smm_subregion(int sub, void **start, size_t *size)
 {
 	uintptr_t sub_base;
 	size_t sub_size;
+	void *smm_base;
 	const size_t cache_size = CONFIG_SMM_RESERVED_SIZE;
 
-	sub_base = smm_region_start();
-	sub_size = smm_region_size();
+	smm_region_info(&smm_base, &sub_size);
+	sub_base = (uintptr_t)smm_base;
 
 	assert(sub_size > CONFIG_SMM_RESERVED_SIZE);