1 files changed, 25 insertions, 11 deletions

diff --git a/drivers/acpi/sleep/main.c b/drivers/acpi/sleep/main.c
index 31e8e58e1cec..10db8991bd15 100644
--- a/drivers/acpi/sleep/main.c
+++ b/drivers/acpi/sleep/main.c
@@ -283,22 +283,34 @@ static struct dmi_system_id __initdata acpisleep_dmi_table[] = {
 #ifdef CONFIG_HIBERNATION
 static int acpi_hibernation_begin(void)
 {
+        int error;
+
         acpi_target_sleep_state = ACPI_STATE_S4;
-        return 0;
+        if (new_pts_ordering)
+                return 0;
+
+        error = acpi_sleep_prepare(ACPI_STATE_S4);
+        if (error)
+                acpi_target_sleep_state = ACPI_STATE_S0;
+        else
+                acpi_sleep_finish_wake_up = true;
+
+        return error;
 }
 
 static int acpi_hibernation_prepare(void)
 {
-        int error;
-
-        error = acpi_sleep_prepare(ACPI_STATE_S4);
-        if (error)
-                return error;
+        if (new_pts_ordering) {
+                int error = acpi_sleep_prepare(ACPI_STATE_S4);
 
-        if (!ACPI_SUCCESS(acpi_hw_disable_all_gpes()))
-                error = -EFAULT;
+                if (error) {
+                        acpi_target_sleep_state = ACPI_STATE_S0;
+                        return error;
+                }
+                acpi_sleep_finish_wake_up = true;
+        }
 
-        return error;
+        return ACPI_SUCCESS(acpi_hw_disable_all_gpes()) ? 0 : -EFAULT;
 }
 
 static int acpi_hibernation_enter(void)
@@ -339,15 +351,17 @@ static void acpi_hibernation_finish(void)
         acpi_set_firmware_waking_vector((acpi_physical_address) 0);
 
         acpi_target_sleep_state = ACPI_STATE_S0;
+        acpi_sleep_finish_wake_up = false;
 }
 
 static void acpi_hibernation_end(void)
 {
         /*
          * This is necessary in case acpi_hibernation_finish() is not called
-         * during a failing transition to the sleep state.
+         * directly during a failing transition to the sleep state.
          */
-        acpi_target_sleep_state = ACPI_STATE_S0;
+        if (acpi_sleep_finish_wake_up)
+                acpi_hibernation_finish();
 }
 
 static int acpi_hibernation_pre_restore(void)