4 files changed, 33 insertions, 0 deletions
diff --git a/src/southbridge/intel/i82801dx/smi.c b/src/southbridge/intel/i82801dx/smi.c
index 93e28452e0..945fcf2db8 100644
--- a/src/southbridge/intel/i82801dx/smi.c
+++ b/src/southbridge/intel/i82801dx/smi.c
@@ -18,6 +18,7 @@
 #include <device/device.h>
 #include <device/pci.h>
 #include <console/console.h>
+#include <arch/acpi.h>
 #include <arch/io.h>
 #include <cpu/cpu.h>
 #include <cpu/x86/cache.h>
@@ -229,6 +230,7 @@ static void smi_set_eos(void)
 }
 
 extern uint8_t smm_relocation_start, smm_relocation_end;
+static void *default_smm_area = NULL;
 
 static void smm_relocate(void)
 {
@@ -246,6 +248,8 @@ static void smm_relocate(void)
 		return;
 	}
 
+	default_smm_area = backup_default_smm_area();
+
 	/* copy the SMM relocation code */
 	memcpy((void *)0x38000, &smm_relocation_start,
 			&smm_relocation_end - &smm_relocation_start);
@@ -339,6 +343,11 @@ void smm_init(void)
 	smi_set_eos();
 }
 
+void smm_init_completion(void)
+{
+	restore_default_smm_area(default_smm_area);
+}
+
 void smm_lock(void)
 {
 	/* LOCK the SMM memory window and enable normal SMM.
diff --git a/src/southbridge/intel/i82801gx/smi.c b/src/southbridge/intel/i82801gx/smi.c
index 17a81c1757..7355d117a4 100644
--- a/src/southbridge/intel/i82801gx/smi.c
+++ b/src/southbridge/intel/i82801gx/smi.c
@@ -287,6 +287,7 @@ static void smi_set_eos(void)
 }
 
 extern uint8_t smm_relocation_start, smm_relocation_end;
+static void *default_smm_area = NULL;
 
 static void smm_relocate(void)
 {
@@ -304,6 +305,8 @@ static void smm_relocate(void)
 		return;
 	}
 
+	default_smm_area = backup_default_smm_area();
+
 	/* copy the SMM relocation code */
 	memcpy((void *)0x38000, &smm_relocation_start,
 			&smm_relocation_end - &smm_relocation_start);
@@ -412,6 +415,11 @@ void smm_init(void)
 	smi_set_eos();
 }
 
+void smm_init_completion(void)
+{
+	restore_default_smm_area(default_smm_area);
+}
+
 void smm_lock(void)
 {
 	/* LOCK the SMM memory window and enable normal SMM.
diff --git a/src/southbridge/intel/i82801ix/smi.c b/src/southbridge/intel/i82801ix/smi.c
index a6c28cbbc2..bf3aa6b0d5 100644
--- a/src/southbridge/intel/i82801ix/smi.c
+++ b/src/southbridge/intel/i82801ix/smi.c
@@ -234,6 +234,7 @@ static void smi_set_eos(void)
 }
 
 extern uint8_t smm_relocation_start, smm_relocation_end;
+static void *default_smm_area = NULL;
 
 static void smm_relocate(void)
 {
@@ -251,6 +252,8 @@ static void smm_relocate(void)
 		return;
 	}
 
+	default_smm_area = backup_default_smm_area();
+
 	/* copy the SMM relocation code */
 	memcpy((void *)0x38000, &smm_relocation_start,
 			&smm_relocation_end - &smm_relocation_start);
@@ -354,6 +357,11 @@ void smm_init(void)
 	smi_set_eos();
 }
 
+void smm_init_completion(void)
+{
+	restore_default_smm_area(default_smm_area);
+}
+
 void smm_lock(void)
 {
 	/* LOCK the SMM memory window and enable normal SMM.
diff --git a/src/southbridge/intel/i82801jx/smi.c b/src/southbridge/intel/i82801jx/smi.c
index 06fa5856c6..8d87291d68 100644
--- a/src/southbridge/intel/i82801jx/smi.c
+++ b/src/southbridge/intel/i82801jx/smi.c
@@ -234,6 +234,7 @@ static void smi_set_eos(void)
 }
 
 extern uint8_t smm_relocation_start, smm_relocation_end;
+static void *default_smm_area = NULL;
 
 static void smm_relocate(void)
 {
@@ -251,6 +252,8 @@ static void smm_relocate(void)
 		return;
 	}
 
+	default_smm_area = backup_default_smm_area();
+
 	/* copy the SMM relocation code */
 	memcpy((void *)0x38000, &smm_relocation_start,
 			&smm_relocation_end - &smm_relocation_start);
@@ -354,6 +357,11 @@ void smm_init(void)
 	smi_set_eos();
 }
 
+void smm_init_completion(void)
+{
+	restore_default_smm_area(default_smm_area);
+}
+
 void smm_lock(void)
 {
 	/* LOCK the SMM memory window and enable normal SMM.