#include "coresight-etm4x.h"
 #include "coresight-etm-perf.h"
 #include "coresight-etm4x-cfg.h"
+#include "coresight-self-hosted-trace.h"
 #include "coresight-syscfg.h"
 
 static int boot_enable;
        if (is_kernel_in_hyp_mode())
                trfcr |= TRFCR_EL2_CX;
 
-       write_sysreg_s(trfcr, SYS_TRFCR_EL1);
+       write_trfcr(trfcr);
 }
 
 static void etm4_init_arch_data(void *info)
        drvdata->trcid = coresight_get_trace_id(drvdata->cpu);
 }
 
-static int etm4_cpu_save(struct etmv4_drvdata *drvdata)
+static int __etm4_cpu_save(struct etmv4_drvdata *drvdata)
 {
        int i, ret = 0;
        struct etmv4_save_state *state;
        return ret;
 }
 
-static void etm4_cpu_restore(struct etmv4_drvdata *drvdata)
+static int etm4_cpu_save(struct etmv4_drvdata *drvdata)
+{
+       int ret = 0;
+
+       /* Save the TRFCR irrespective of whether the ETM is ON */
+       if (drvdata->trfc)
+               drvdata->save_trfcr = read_trfcr();
+       /*
+        * Save and restore the ETM Trace registers only if
+        * the ETM is active.
+        */
+       if (local_read(&drvdata->mode) && drvdata->save_state)
+               ret = __etm4_cpu_save(drvdata);
+       return ret;
+}
+
+static void __etm4_cpu_restore(struct etmv4_drvdata *drvdata)
 {
        int i;
        struct etmv4_save_state *state = drvdata->save_state;
        etm4_cs_lock(drvdata, csa);
 }
 
+static void etm4_cpu_restore(struct etmv4_drvdata *drvdata)
+{
+       if (drvdata->trfc)
+               write_trfcr(drvdata->save_trfcr);
+       if (drvdata->state_needs_restore)
+               __etm4_cpu_restore(drvdata);
+}
+
 static int etm4_cpu_pm_notify(struct notifier_block *nb, unsigned long cmd,
                              void *v)
 {
 
        drvdata = etmdrvdata[cpu];
 
-       if (!drvdata->save_state)
-               return NOTIFY_OK;
-
        if (WARN_ON_ONCE(drvdata->cpu != cpu))
                return NOTIFY_BAD;
 
        switch (cmd) {
        case CPU_PM_ENTER:
-               /* save the state if self-hosted coresight is in use */
-               if (local_read(&drvdata->mode))
-                       if (etm4_cpu_save(drvdata))
-                               return NOTIFY_BAD;
+               if (etm4_cpu_save(drvdata))
+                       return NOTIFY_BAD;
                break;
        case CPU_PM_EXIT:
        case CPU_PM_ENTER_FAILED:
-               if (drvdata->state_needs_restore)
-                       etm4_cpu_restore(drvdata);
+               etm4_cpu_restore(drvdata);
                break;
        default:
                return NOTIFY_DONE;
 
  * @lpoverride:        If the implementation can support low-power state over.
  * @trfc:      If the implementation supports Arm v8.4 trace filter controls.
  * @config:    structure holding configuration parameters.
+ * @save_trfcr:        Saved TRFCR_EL1 register during a CPU PM event.
  * @save_state:        State to be preserved across power loss
  * @state_needs_restore: True when there is context to restore after PM exit
  * @skip_power_up: Indicates if an implementation can skip powering up
        bool                            lpoverride;
        bool                            trfc;
        struct etmv4_config             config;
+       u64                             save_trfcr;
        struct etmv4_save_state         *save_state;
        bool                            state_needs_restore;
        bool                            skip_power_up;
 
--- /dev/null
+/* SPDX-License-Identifier: GPL-2.0-only */
+/*
+ * Arm v8 Self-Hosted trace support.
+ *
+ * Copyright (C) 2021 ARM Ltd.
+ */
+
+#ifndef __CORESIGHT_SELF_HOSTED_TRACE_H
+#define __CORESIGHT_SELF_HOSTED_TRACE_H
+
+#include <asm/sysreg.h>
+
+static inline u64 read_trfcr(void)
+{
+       return read_sysreg_s(SYS_TRFCR_EL1);
+}
+
+static inline void write_trfcr(u64 val)
+{
+       write_sysreg_s(val, SYS_TRFCR_EL1);
+       isb();
+}
+
+#endif /*  __CORESIGHT_SELF_HOSTED_TRACE_H */