1 files changed, 95 insertions, 0 deletions

diff --git a/arch/arm/xen/enlighten.c b/arch/arm/xen/enlighten.c
index fc7ea529f462..75cd7345c654 100644
--- a/arch/arm/xen/enlighten.c
+++ b/arch/arm/xen/enlighten.c
@@ -12,6 +12,7 @@
 #include <xen/page.h>
 #include <xen/interface/sched.h>
 #include <xen/xen-ops.h>
+#include <asm/paravirt.h>
 #include <asm/xen/hypervisor.h>
 #include <asm/xen/hypercall.h>
 #include <asm/system_misc.h>
@@ -25,6 +26,10 @@
 #include <linux/cpufreq.h>
 #include <linux/cpu.h>
 #include <linux/console.h>
+#include <linux/pvclock_gtod.h>
+#include <linux/time64.h>
+#include <linux/timekeeping.h>
+#include <linux/timekeeper_internal.h>
 
 #include <linux/mm.h>
 
@@ -79,6 +84,83 @@ int xen_unmap_domain_gfn_range(struct vm_area_struct *vma,
 }
 EXPORT_SYMBOL_GPL(xen_unmap_domain_gfn_range);
 
+static unsigned long long xen_stolen_accounting(int cpu)
+{
+        struct vcpu_runstate_info state;
+
+        BUG_ON(cpu != smp_processor_id());
+
+        xen_get_runstate_snapshot(&state);
+
+        WARN_ON(state.state != RUNSTATE_running);
+
+        return state.time[RUNSTATE_runnable] + state.time[RUNSTATE_offline];
+}
+
+static void xen_read_wallclock(struct timespec64 *ts)
+{
+        u32 version;
+        struct timespec64 now, ts_monotonic;
+        struct shared_info *s = HYPERVISOR_shared_info;
+        struct pvclock_wall_clock *wall_clock = &(s->wc);
+
+        /* get wallclock at system boot */
+        do {
+                version = wall_clock->version;
+                rmb();          /* fetch version before time */
+                now.tv_sec  = ((uint64_t)wall_clock->sec_hi << 32) | wall_clock->sec;
+                now.tv_nsec = wall_clock->nsec;
+                rmb();          /* fetch time before checking version */
+        } while ((wall_clock->version & 1) || (version != wall_clock->version));
+
+        /* time since system boot */
+        ktime_get_ts64(&ts_monotonic);
+        *ts = timespec64_add(now, ts_monotonic);
+}
+
+static int xen_pvclock_gtod_notify(struct notifier_block *nb,
+                                   unsigned long was_set, void *priv)
+{
+        /* Protected by the calling core code serialization */
+        static struct timespec64 next_sync;
+
+        struct xen_platform_op op;
+        struct timespec64 now, system_time;
+        struct timekeeper *tk = priv;
+
+        now.tv_sec = tk->xtime_sec;
+        now.tv_nsec = (long)(tk->tkr_mono.xtime_nsec >> tk->tkr_mono.shift);
+        system_time = timespec64_add(now, tk->wall_to_monotonic);
+
+        /*
+         * We only take the expensive HV call when the clock was set
+         * or when the 11 minutes RTC synchronization time elapsed.
+         */
+        if (!was_set && timespec64_compare(&now, &next_sync) < 0)
+                return NOTIFY_OK;
+
+        op.cmd = XENPF_settime64;
+        op.u.settime64.mbz = 0;
+        op.u.settime64.secs = now.tv_sec;
+        op.u.settime64.nsecs = now.tv_nsec;
+        op.u.settime64.system_time = timespec64_to_ns(&system_time);
+        (void)HYPERVISOR_platform_op(&op);
+
+        /*
+         * Move the next drift compensation time 11 minutes
+         * ahead. That's emulating the sync_cmos_clock() update for
+         * the hardware RTC.
+         */
+        next_sync = now;
+        next_sync.tv_sec += 11 * 60;
+
+        return NOTIFY_OK;
+}
+
+static struct notifier_block xen_pvclock_gtod_notifier = {
+        .notifier_call = xen_pvclock_gtod_notify,
+};
+
 static void xen_percpu_init(void)
 {
         struct vcpu_register_vcpu_info info;
@@ -104,6 +186,8 @@ static void xen_percpu_init(void)
         BUG_ON(err);
         per_cpu(xen_vcpu, cpu) = vcpup;
 
+        xen_setup_runstate_info(cpu);
+
 after_register_vcpu_info:
         enable_percpu_irq(xen_events_irq, 0);
         put_cpu();
@@ -271,6 +355,11 @@ static int __init xen_guest_init(void)
 
         register_cpu_notifier(&xen_cpu_notifier);
 
+        pv_time_ops.steal_clock = xen_stolen_accounting;
+        static_key_slow_inc(&paravirt_steal_enabled);
+        if (xen_initial_domain())
+                pvclock_gtod_register_notifier(&xen_pvclock_gtod_notifier);
+
         return 0;
 }
 early_initcall(xen_guest_init);
@@ -282,6 +371,11 @@ static int __init xen_pm_init(void)
 
         pm_power_off = xen_power_off;
         arm_pm_restart = xen_restart;
+        if (!xen_initial_domain()) {
+                struct timespec64 ts;
+                xen_read_wallclock(&ts);
+                do_settimeofday64(&ts);
+        }
 
         return 0;
 }
@@ -307,5 +401,6 @@ EXPORT_SYMBOL_GPL(HYPERVISOR_memory_op);
 EXPORT_SYMBOL_GPL(HYPERVISOR_physdev_op);
 EXPORT_SYMBOL_GPL(HYPERVISOR_vcpu_op);
 EXPORT_SYMBOL_GPL(HYPERVISOR_tmem_op);
+EXPORT_SYMBOL_GPL(HYPERVISOR_platform_op);
 EXPORT_SYMBOL_GPL(HYPERVISOR_multicall);
 EXPORT_SYMBOL_GPL(privcmd_call);