1 files changed, 42 insertions, 4 deletions

diff --git a/drivers/xen/manage.c b/drivers/xen/manage.c
index 07e857b0de13..1799bd890315 100644
--- a/drivers/xen/manage.c
+++ b/drivers/xen/manage.c
@@ -9,6 +9,7 @@
 #include <linux/stop_machine.h>
 #include <linux/freezer.h>
 
+#include <xen/xen.h>
 #include <xen/xenbus.h>
 #include <xen/grant_table.h>
 #include <xen/events.h>
@@ -17,6 +18,7 @@
 
 #include <asm/xen/hypercall.h>
 #include <asm/xen/page.h>
+#include <asm/xen/hypervisor.h>
 
 enum shutdown_state {
         SHUTDOWN_INVALID = -1,
@@ -33,10 +35,30 @@ enum shutdown_state {
 static enum shutdown_state shutting_down = SHUTDOWN_INVALID;
 
 #ifdef CONFIG_PM_SLEEP
-static int xen_suspend(void *data)
+static int xen_hvm_suspend(void *data)
 {
+        struct sched_shutdown r = { .reason = SHUTDOWN_suspend };
         int *cancelled = data;
+
+        BUG_ON(!irqs_disabled());
+
+        *cancelled = HYPERVISOR_sched_op(SCHEDOP_shutdown, &r);
+
+        xen_hvm_post_suspend(*cancelled);
+        gnttab_resume();
+
+        if (!*cancelled) {
+                xen_irq_resume();
+                xen_timer_resume();
+        }
+
+        return 0;
+}
+
+static int xen_suspend(void *data)
+{
         int err;
+        int *cancelled = data;
 
         BUG_ON(!irqs_disabled());
 
@@ -106,7 +128,10 @@ static void do_suspend(void)
                 goto out_resume;
         }
 
-        err = stop_machine(xen_suspend, &cancelled, cpumask_of(0));
+        if (xen_hvm_domain())
+                err = stop_machine(xen_hvm_suspend, &cancelled, cpumask_of(0));
+        else
+                err = stop_machine(xen_suspend, &cancelled, cpumask_of(0));
 
         dpm_resume_noirq(PMSG_RESUME);
 
@@ -255,7 +280,19 @@ static int shutdown_event(struct notifier_block *notifier,
         return NOTIFY_DONE;
 }
 
-static int __init setup_shutdown_event(void)
+static int __init __setup_shutdown_event(void)
+{
+        /* Delay initialization in the PV on HVM case */
+        if (xen_hvm_domain())
+                return 0;
+
+        if (!xen_pv_domain())
+                return -ENODEV;
+
+        return xen_setup_shutdown_event();
+}
+
+int xen_setup_shutdown_event(void)
 {
         static struct notifier_block xenstore_notifier = {
                 .notifier_call = shutdown_event
@@ -264,5 +301,6 @@ static int __init setup_shutdown_event(void)
 
         return 0;
 }
+EXPORT_SYMBOL_GPL(xen_setup_shutdown_event);
 
-subsys_initcall(setup_shutdown_event);
+subsys_initcall(__setup_shutdown_event);