coresight: etm4x: Allow etm4x to be built as a module
[linux-2.6-microblaze.git] / drivers / hwtracing / coresight / coresight-etm4x-core.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2014, The Linux Foundation. All rights reserved.
4  */
5
6 #include <linux/kernel.h>
7 #include <linux/moduleparam.h>
8 #include <linux/init.h>
9 #include <linux/types.h>
10 #include <linux/device.h>
11 #include <linux/io.h>
12 #include <linux/err.h>
13 #include <linux/fs.h>
14 #include <linux/slab.h>
15 #include <linux/delay.h>
16 #include <linux/smp.h>
17 #include <linux/sysfs.h>
18 #include <linux/stat.h>
19 #include <linux/clk.h>
20 #include <linux/cpu.h>
21 #include <linux/cpu_pm.h>
22 #include <linux/coresight.h>
23 #include <linux/coresight-pmu.h>
24 #include <linux/pm_wakeup.h>
25 #include <linux/amba/bus.h>
26 #include <linux/seq_file.h>
27 #include <linux/uaccess.h>
28 #include <linux/perf_event.h>
29 #include <linux/pm_runtime.h>
30 #include <linux/property.h>
31 #include <asm/sections.h>
32 #include <asm/local.h>
33 #include <asm/virt.h>
34
35 #include "coresight-etm4x.h"
36 #include "coresight-etm-perf.h"
37
38 static int boot_enable;
39 module_param(boot_enable, int, 0444);
40 MODULE_PARM_DESC(boot_enable, "Enable tracing on boot");
41
42 #define PARAM_PM_SAVE_FIRMWARE    0 /* save self-hosted state as per firmware */
43 #define PARAM_PM_SAVE_NEVER       1 /* never save any state */
44 #define PARAM_PM_SAVE_SELF_HOSTED 2 /* save self-hosted state only */
45
46 static int pm_save_enable = PARAM_PM_SAVE_FIRMWARE;
47 module_param(pm_save_enable, int, 0444);
48 MODULE_PARM_DESC(pm_save_enable,
49         "Save/restore state on power down: 1 = never, 2 = self-hosted");
50
51 static struct etmv4_drvdata *etmdrvdata[NR_CPUS];
52 static void etm4_set_default_config(struct etmv4_config *config);
53 static int etm4_set_event_filters(struct etmv4_drvdata *drvdata,
54                                   struct perf_event *event);
55 static u64 etm4_get_access_type(struct etmv4_config *config);
56
57 static enum cpuhp_state hp_online;
58
59 static void etm4_os_unlock(struct etmv4_drvdata *drvdata)
60 {
61         /* Writing 0 to TRCOSLAR unlocks the trace registers */
62         writel_relaxed(0x0, drvdata->base + TRCOSLAR);
63         drvdata->os_unlock = true;
64         isb();
65 }
66
67 static void etm4_os_lock(struct etmv4_drvdata *drvdata)
68 {
69         /* Writing 0x1 to TRCOSLAR locks the trace registers */
70         writel_relaxed(0x1, drvdata->base + TRCOSLAR);
71         drvdata->os_unlock = false;
72         isb();
73 }
74
75 static bool etm4_arch_supported(u8 arch)
76 {
77         /* Mask out the minor version number */
78         switch (arch & 0xf0) {
79         case ETM_ARCH_V4:
80                 break;
81         default:
82                 return false;
83         }
84         return true;
85 }
86
87 static int etm4_cpu_id(struct coresight_device *csdev)
88 {
89         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
90
91         return drvdata->cpu;
92 }
93
94 static int etm4_trace_id(struct coresight_device *csdev)
95 {
96         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
97
98         return drvdata->trcid;
99 }
100
101 struct etm4_enable_arg {
102         struct etmv4_drvdata *drvdata;
103         int rc;
104 };
105
106 static int etm4_enable_hw(struct etmv4_drvdata *drvdata)
107 {
108         int i, rc;
109         struct etmv4_config *config = &drvdata->config;
110         struct device *etm_dev = &drvdata->csdev->dev;
111
112         CS_UNLOCK(drvdata->base);
113
114         etm4_os_unlock(drvdata);
115
116         rc = coresight_claim_device_unlocked(drvdata->base);
117         if (rc)
118                 goto done;
119
120         /* Disable the trace unit before programming trace registers */
121         writel_relaxed(0, drvdata->base + TRCPRGCTLR);
122
123         /* wait for TRCSTATR.IDLE to go up */
124         if (coresight_timeout(drvdata->base, TRCSTATR, TRCSTATR_IDLE_BIT, 1))
125                 dev_err(etm_dev,
126                         "timeout while waiting for Idle Trace Status\n");
127
128         writel_relaxed(config->pe_sel, drvdata->base + TRCPROCSELR);
129         writel_relaxed(config->cfg, drvdata->base + TRCCONFIGR);
130         /* nothing specific implemented */
131         writel_relaxed(0x0, drvdata->base + TRCAUXCTLR);
132         writel_relaxed(config->eventctrl0, drvdata->base + TRCEVENTCTL0R);
133         writel_relaxed(config->eventctrl1, drvdata->base + TRCEVENTCTL1R);
134         writel_relaxed(config->stall_ctrl, drvdata->base + TRCSTALLCTLR);
135         writel_relaxed(config->ts_ctrl, drvdata->base + TRCTSCTLR);
136         writel_relaxed(config->syncfreq, drvdata->base + TRCSYNCPR);
137         writel_relaxed(config->ccctlr, drvdata->base + TRCCCCTLR);
138         writel_relaxed(config->bb_ctrl, drvdata->base + TRCBBCTLR);
139         writel_relaxed(drvdata->trcid, drvdata->base + TRCTRACEIDR);
140         writel_relaxed(config->vinst_ctrl, drvdata->base + TRCVICTLR);
141         writel_relaxed(config->viiectlr, drvdata->base + TRCVIIECTLR);
142         writel_relaxed(config->vissctlr,
143                        drvdata->base + TRCVISSCTLR);
144         writel_relaxed(config->vipcssctlr,
145                        drvdata->base + TRCVIPCSSCTLR);
146         for (i = 0; i < drvdata->nrseqstate - 1; i++)
147                 writel_relaxed(config->seq_ctrl[i],
148                                drvdata->base + TRCSEQEVRn(i));
149         writel_relaxed(config->seq_rst, drvdata->base + TRCSEQRSTEVR);
150         writel_relaxed(config->seq_state, drvdata->base + TRCSEQSTR);
151         writel_relaxed(config->ext_inp, drvdata->base + TRCEXTINSELR);
152         for (i = 0; i < drvdata->nr_cntr; i++) {
153                 writel_relaxed(config->cntrldvr[i],
154                                drvdata->base + TRCCNTRLDVRn(i));
155                 writel_relaxed(config->cntr_ctrl[i],
156                                drvdata->base + TRCCNTCTLRn(i));
157                 writel_relaxed(config->cntr_val[i],
158                                drvdata->base + TRCCNTVRn(i));
159         }
160
161         /*
162          * Resource selector pair 0 is always implemented and reserved.  As
163          * such start at 2.
164          */
165         for (i = 2; i < drvdata->nr_resource * 2; i++)
166                 writel_relaxed(config->res_ctrl[i],
167                                drvdata->base + TRCRSCTLRn(i));
168
169         for (i = 0; i < drvdata->nr_ss_cmp; i++) {
170                 /* always clear status bit on restart if using single-shot */
171                 if (config->ss_ctrl[i] || config->ss_pe_cmp[i])
172                         config->ss_status[i] &= ~BIT(31);
173                 writel_relaxed(config->ss_ctrl[i],
174                                drvdata->base + TRCSSCCRn(i));
175                 writel_relaxed(config->ss_status[i],
176                                drvdata->base + TRCSSCSRn(i));
177                 writel_relaxed(config->ss_pe_cmp[i],
178                                drvdata->base + TRCSSPCICRn(i));
179         }
180         for (i = 0; i < drvdata->nr_addr_cmp; i++) {
181                 writeq_relaxed(config->addr_val[i],
182                                drvdata->base + TRCACVRn(i));
183                 writeq_relaxed(config->addr_acc[i],
184                                drvdata->base + TRCACATRn(i));
185         }
186         for (i = 0; i < drvdata->numcidc; i++)
187                 writeq_relaxed(config->ctxid_pid[i],
188                                drvdata->base + TRCCIDCVRn(i));
189         writel_relaxed(config->ctxid_mask0, drvdata->base + TRCCIDCCTLR0);
190         writel_relaxed(config->ctxid_mask1, drvdata->base + TRCCIDCCTLR1);
191
192         for (i = 0; i < drvdata->numvmidc; i++)
193                 writeq_relaxed(config->vmid_val[i],
194                                drvdata->base + TRCVMIDCVRn(i));
195         writel_relaxed(config->vmid_mask0, drvdata->base + TRCVMIDCCTLR0);
196         writel_relaxed(config->vmid_mask1, drvdata->base + TRCVMIDCCTLR1);
197
198         if (!drvdata->skip_power_up) {
199                 /*
200                  * Request to keep the trace unit powered and also
201                  * emulation of powerdown
202                  */
203                 writel_relaxed(readl_relaxed(drvdata->base + TRCPDCR) |
204                                TRCPDCR_PU, drvdata->base + TRCPDCR);
205         }
206
207         /* Enable the trace unit */
208         writel_relaxed(1, drvdata->base + TRCPRGCTLR);
209
210         /* wait for TRCSTATR.IDLE to go back down to '0' */
211         if (coresight_timeout(drvdata->base, TRCSTATR, TRCSTATR_IDLE_BIT, 0))
212                 dev_err(etm_dev,
213                         "timeout while waiting for Idle Trace Status\n");
214
215         /*
216          * As recommended by section 4.3.7 ("Synchronization when using the
217          * memory-mapped interface") of ARM IHI 0064D
218          */
219         dsb(sy);
220         isb();
221
222 done:
223         CS_LOCK(drvdata->base);
224
225         dev_dbg(etm_dev, "cpu: %d enable smp call done: %d\n",
226                 drvdata->cpu, rc);
227         return rc;
228 }
229
230 static void etm4_enable_hw_smp_call(void *info)
231 {
232         struct etm4_enable_arg *arg = info;
233
234         if (WARN_ON(!arg))
235                 return;
236         arg->rc = etm4_enable_hw(arg->drvdata);
237 }
238
239 /*
240  * The goal of function etm4_config_timestamp_event() is to configure a
241  * counter that will tell the tracer to emit a timestamp packet when it
242  * reaches zero.  This is done in order to get a more fine grained idea
243  * of when instructions are executed so that they can be correlated
244  * with execution on other CPUs.
245  *
246  * To do this the counter itself is configured to self reload and
247  * TRCRSCTLR1 (always true) used to get the counter to decrement.  From
248  * there a resource selector is configured with the counter and the
249  * timestamp control register to use the resource selector to trigger the
250  * event that will insert a timestamp packet in the stream.
251  */
252 static int etm4_config_timestamp_event(struct etmv4_drvdata *drvdata)
253 {
254         int ctridx, ret = -EINVAL;
255         int counter, rselector;
256         u32 val = 0;
257         struct etmv4_config *config = &drvdata->config;
258
259         /* No point in trying if we don't have at least one counter */
260         if (!drvdata->nr_cntr)
261                 goto out;
262
263         /* Find a counter that hasn't been initialised */
264         for (ctridx = 0; ctridx < drvdata->nr_cntr; ctridx++)
265                 if (config->cntr_val[ctridx] == 0)
266                         break;
267
268         /* All the counters have been configured already, bail out */
269         if (ctridx == drvdata->nr_cntr) {
270                 pr_debug("%s: no available counter found\n", __func__);
271                 ret = -ENOSPC;
272                 goto out;
273         }
274
275         /*
276          * Searching for an available resource selector to use, starting at
277          * '2' since every implementation has at least 2 resource selector.
278          * ETMIDR4 gives the number of resource selector _pairs_,
279          * hence multiply by 2.
280          */
281         for (rselector = 2; rselector < drvdata->nr_resource * 2; rselector++)
282                 if (!config->res_ctrl[rselector])
283                         break;
284
285         if (rselector == drvdata->nr_resource * 2) {
286                 pr_debug("%s: no available resource selector found\n",
287                          __func__);
288                 ret = -ENOSPC;
289                 goto out;
290         }
291
292         /* Remember what counter we used */
293         counter = 1 << ctridx;
294
295         /*
296          * Initialise original and reload counter value to the smallest
297          * possible value in order to get as much precision as we can.
298          */
299         config->cntr_val[ctridx] = 1;
300         config->cntrldvr[ctridx] = 1;
301
302         /* Set the trace counter control register */
303         val =  0x1 << 16        |  /* Bit 16, reload counter automatically */
304                0x0 << 7         |  /* Select single resource selector */
305                0x1;                /* Resource selector 1, i.e always true */
306
307         config->cntr_ctrl[ctridx] = val;
308
309         val = 0x2 << 16         | /* Group 0b0010 - Counter and sequencers */
310               counter << 0;       /* Counter to use */
311
312         config->res_ctrl[rselector] = val;
313
314         val = 0x0 << 7          | /* Select single resource selector */
315               rselector;          /* Resource selector */
316
317         config->ts_ctrl = val;
318
319         ret = 0;
320 out:
321         return ret;
322 }
323
324 static int etm4_parse_event_config(struct etmv4_drvdata *drvdata,
325                                    struct perf_event *event)
326 {
327         int ret = 0;
328         struct etmv4_config *config = &drvdata->config;
329         struct perf_event_attr *attr = &event->attr;
330
331         if (!attr) {
332                 ret = -EINVAL;
333                 goto out;
334         }
335
336         /* Clear configuration from previous run */
337         memset(config, 0, sizeof(struct etmv4_config));
338
339         if (attr->exclude_kernel)
340                 config->mode = ETM_MODE_EXCL_KERN;
341
342         if (attr->exclude_user)
343                 config->mode = ETM_MODE_EXCL_USER;
344
345         /* Always start from the default config */
346         etm4_set_default_config(config);
347
348         /* Configure filters specified on the perf cmd line, if any. */
349         ret = etm4_set_event_filters(drvdata, event);
350         if (ret)
351                 goto out;
352
353         /* Go from generic option to ETMv4 specifics */
354         if (attr->config & BIT(ETM_OPT_CYCACC)) {
355                 config->cfg |= BIT(4);
356                 /* TRM: Must program this for cycacc to work */
357                 config->ccctlr = ETM_CYC_THRESHOLD_DEFAULT;
358         }
359         if (attr->config & BIT(ETM_OPT_TS)) {
360                 /*
361                  * Configure timestamps to be emitted at regular intervals in
362                  * order to correlate instructions executed on different CPUs
363                  * (CPU-wide trace scenarios).
364                  */
365                 ret = etm4_config_timestamp_event(drvdata);
366
367                 /*
368                  * No need to go further if timestamp intervals can't
369                  * be configured.
370                  */
371                 if (ret)
372                         goto out;
373
374                 /* bit[11], Global timestamp tracing bit */
375                 config->cfg |= BIT(11);
376         }
377
378         if (attr->config & BIT(ETM_OPT_CTXTID))
379                 /* bit[6], Context ID tracing bit */
380                 config->cfg |= BIT(ETM4_CFG_BIT_CTXTID);
381
382         /* return stack - enable if selected and supported */
383         if ((attr->config & BIT(ETM_OPT_RETSTK)) && drvdata->retstack)
384                 /* bit[12], Return stack enable bit */
385                 config->cfg |= BIT(12);
386
387 out:
388         return ret;
389 }
390
391 static int etm4_enable_perf(struct coresight_device *csdev,
392                             struct perf_event *event)
393 {
394         int ret = 0;
395         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
396
397         if (WARN_ON_ONCE(drvdata->cpu != smp_processor_id())) {
398                 ret = -EINVAL;
399                 goto out;
400         }
401
402         /* Configure the tracer based on the session's specifics */
403         ret = etm4_parse_event_config(drvdata, event);
404         if (ret)
405                 goto out;
406         /* And enable it */
407         ret = etm4_enable_hw(drvdata);
408
409 out:
410         return ret;
411 }
412
413 static int etm4_enable_sysfs(struct coresight_device *csdev)
414 {
415         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
416         struct etm4_enable_arg arg = { };
417         int ret;
418
419         spin_lock(&drvdata->spinlock);
420
421         /*
422          * Executing etm4_enable_hw on the cpu whose ETM is being enabled
423          * ensures that register writes occur when cpu is powered.
424          */
425         arg.drvdata = drvdata;
426         ret = smp_call_function_single(drvdata->cpu,
427                                        etm4_enable_hw_smp_call, &arg, 1);
428         if (!ret)
429                 ret = arg.rc;
430         if (!ret)
431                 drvdata->sticky_enable = true;
432         spin_unlock(&drvdata->spinlock);
433
434         if (!ret)
435                 dev_dbg(&csdev->dev, "ETM tracing enabled\n");
436         return ret;
437 }
438
439 static int etm4_enable(struct coresight_device *csdev,
440                        struct perf_event *event, u32 mode)
441 {
442         int ret;
443         u32 val;
444         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
445
446         val = local_cmpxchg(&drvdata->mode, CS_MODE_DISABLED, mode);
447
448         /* Someone is already using the tracer */
449         if (val)
450                 return -EBUSY;
451
452         switch (mode) {
453         case CS_MODE_SYSFS:
454                 ret = etm4_enable_sysfs(csdev);
455                 break;
456         case CS_MODE_PERF:
457                 ret = etm4_enable_perf(csdev, event);
458                 break;
459         default:
460                 ret = -EINVAL;
461         }
462
463         /* The tracer didn't start */
464         if (ret)
465                 local_set(&drvdata->mode, CS_MODE_DISABLED);
466
467         return ret;
468 }
469
470 static void etm4_disable_hw(void *info)
471 {
472         u32 control;
473         struct etmv4_drvdata *drvdata = info;
474         struct etmv4_config *config = &drvdata->config;
475         struct device *etm_dev = &drvdata->csdev->dev;
476         int i;
477
478         CS_UNLOCK(drvdata->base);
479
480         if (!drvdata->skip_power_up) {
481                 /* power can be removed from the trace unit now */
482                 control = readl_relaxed(drvdata->base + TRCPDCR);
483                 control &= ~TRCPDCR_PU;
484                 writel_relaxed(control, drvdata->base + TRCPDCR);
485         }
486
487         control = readl_relaxed(drvdata->base + TRCPRGCTLR);
488
489         /* EN, bit[0] Trace unit enable bit */
490         control &= ~0x1;
491
492         /*
493          * Make sure everything completes before disabling, as recommended
494          * by section 7.3.77 ("TRCVICTLR, ViewInst Main Control Register,
495          * SSTATUS") of ARM IHI 0064D
496          */
497         dsb(sy);
498         isb();
499         writel_relaxed(control, drvdata->base + TRCPRGCTLR);
500
501         /* wait for TRCSTATR.PMSTABLE to go to '1' */
502         if (coresight_timeout(drvdata->base, TRCSTATR,
503                               TRCSTATR_PMSTABLE_BIT, 1))
504                 dev_err(etm_dev,
505                         "timeout while waiting for PM stable Trace Status\n");
506
507         /* read the status of the single shot comparators */
508         for (i = 0; i < drvdata->nr_ss_cmp; i++) {
509                 config->ss_status[i] =
510                         readl_relaxed(drvdata->base + TRCSSCSRn(i));
511         }
512
513         /* read back the current counter values */
514         for (i = 0; i < drvdata->nr_cntr; i++) {
515                 config->cntr_val[i] =
516                         readl_relaxed(drvdata->base + TRCCNTVRn(i));
517         }
518
519         coresight_disclaim_device_unlocked(drvdata->base);
520
521         CS_LOCK(drvdata->base);
522
523         dev_dbg(&drvdata->csdev->dev,
524                 "cpu: %d disable smp call done\n", drvdata->cpu);
525 }
526
527 static int etm4_disable_perf(struct coresight_device *csdev,
528                              struct perf_event *event)
529 {
530         u32 control;
531         struct etm_filters *filters = event->hw.addr_filters;
532         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
533
534         if (WARN_ON_ONCE(drvdata->cpu != smp_processor_id()))
535                 return -EINVAL;
536
537         etm4_disable_hw(drvdata);
538
539         /*
540          * Check if the start/stop logic was active when the unit was stopped.
541          * That way we can re-enable the start/stop logic when the process is
542          * scheduled again.  Configuration of the start/stop logic happens in
543          * function etm4_set_event_filters().
544          */
545         control = readl_relaxed(drvdata->base + TRCVICTLR);
546         /* TRCVICTLR::SSSTATUS, bit[9] */
547         filters->ssstatus = (control & BIT(9));
548
549         return 0;
550 }
551
552 static void etm4_disable_sysfs(struct coresight_device *csdev)
553 {
554         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
555
556         /*
557          * Taking hotplug lock here protects from clocks getting disabled
558          * with tracing being left on (crash scenario) if user disable occurs
559          * after cpu online mask indicates the cpu is offline but before the
560          * DYING hotplug callback is serviced by the ETM driver.
561          */
562         cpus_read_lock();
563         spin_lock(&drvdata->spinlock);
564
565         /*
566          * Executing etm4_disable_hw on the cpu whose ETM is being disabled
567          * ensures that register writes occur when cpu is powered.
568          */
569         smp_call_function_single(drvdata->cpu, etm4_disable_hw, drvdata, 1);
570
571         spin_unlock(&drvdata->spinlock);
572         cpus_read_unlock();
573
574         dev_dbg(&csdev->dev, "ETM tracing disabled\n");
575 }
576
577 static void etm4_disable(struct coresight_device *csdev,
578                          struct perf_event *event)
579 {
580         u32 mode;
581         struct etmv4_drvdata *drvdata = dev_get_drvdata(csdev->dev.parent);
582
583         /*
584          * For as long as the tracer isn't disabled another entity can't
585          * change its status.  As such we can read the status here without
586          * fearing it will change under us.
587          */
588         mode = local_read(&drvdata->mode);
589
590         switch (mode) {
591         case CS_MODE_DISABLED:
592                 break;
593         case CS_MODE_SYSFS:
594                 etm4_disable_sysfs(csdev);
595                 break;
596         case CS_MODE_PERF:
597                 etm4_disable_perf(csdev, event);
598                 break;
599         }
600
601         if (mode)
602                 local_set(&drvdata->mode, CS_MODE_DISABLED);
603 }
604
605 static const struct coresight_ops_source etm4_source_ops = {
606         .cpu_id         = etm4_cpu_id,
607         .trace_id       = etm4_trace_id,
608         .enable         = etm4_enable,
609         .disable        = etm4_disable,
610 };
611
612 static const struct coresight_ops etm4_cs_ops = {
613         .source_ops     = &etm4_source_ops,
614 };
615
616 static void etm4_init_arch_data(void *info)
617 {
618         u32 etmidr0;
619         u32 etmidr1;
620         u32 etmidr2;
621         u32 etmidr3;
622         u32 etmidr4;
623         u32 etmidr5;
624         struct etmv4_drvdata *drvdata = info;
625         int i;
626
627         /* Make sure all registers are accessible */
628         etm4_os_unlock(drvdata);
629
630         CS_UNLOCK(drvdata->base);
631
632         /* find all capabilities of the tracing unit */
633         etmidr0 = readl_relaxed(drvdata->base + TRCIDR0);
634
635         /* INSTP0, bits[2:1] P0 tracing support field */
636         if (BMVAL(etmidr0, 1, 1) && BMVAL(etmidr0, 2, 2))
637                 drvdata->instrp0 = true;
638         else
639                 drvdata->instrp0 = false;
640
641         /* TRCBB, bit[5] Branch broadcast tracing support bit */
642         if (BMVAL(etmidr0, 5, 5))
643                 drvdata->trcbb = true;
644         else
645                 drvdata->trcbb = false;
646
647         /* TRCCOND, bit[6] Conditional instruction tracing support bit */
648         if (BMVAL(etmidr0, 6, 6))
649                 drvdata->trccond = true;
650         else
651                 drvdata->trccond = false;
652
653         /* TRCCCI, bit[7] Cycle counting instruction bit */
654         if (BMVAL(etmidr0, 7, 7))
655                 drvdata->trccci = true;
656         else
657                 drvdata->trccci = false;
658
659         /* RETSTACK, bit[9] Return stack bit */
660         if (BMVAL(etmidr0, 9, 9))
661                 drvdata->retstack = true;
662         else
663                 drvdata->retstack = false;
664
665         /* NUMEVENT, bits[11:10] Number of events field */
666         drvdata->nr_event = BMVAL(etmidr0, 10, 11);
667         /* QSUPP, bits[16:15] Q element support field */
668         drvdata->q_support = BMVAL(etmidr0, 15, 16);
669         /* TSSIZE, bits[28:24] Global timestamp size field */
670         drvdata->ts_size = BMVAL(etmidr0, 24, 28);
671
672         /* base architecture of trace unit */
673         etmidr1 = readl_relaxed(drvdata->base + TRCIDR1);
674         /*
675          * TRCARCHMIN, bits[7:4] architecture the minor version number
676          * TRCARCHMAJ, bits[11:8] architecture major versin number
677          */
678         drvdata->arch = BMVAL(etmidr1, 4, 11);
679         drvdata->config.arch = drvdata->arch;
680
681         /* maximum size of resources */
682         etmidr2 = readl_relaxed(drvdata->base + TRCIDR2);
683         /* CIDSIZE, bits[9:5] Indicates the Context ID size */
684         drvdata->ctxid_size = BMVAL(etmidr2, 5, 9);
685         /* VMIDSIZE, bits[14:10] Indicates the VMID size */
686         drvdata->vmid_size = BMVAL(etmidr2, 10, 14);
687         /* CCSIZE, bits[28:25] size of the cycle counter in bits minus 12 */
688         drvdata->ccsize = BMVAL(etmidr2, 25, 28);
689
690         etmidr3 = readl_relaxed(drvdata->base + TRCIDR3);
691         /* CCITMIN, bits[11:0] minimum threshold value that can be programmed */
692         drvdata->ccitmin = BMVAL(etmidr3, 0, 11);
693         /* EXLEVEL_S, bits[19:16] Secure state instruction tracing */
694         drvdata->s_ex_level = BMVAL(etmidr3, 16, 19);
695         /* EXLEVEL_NS, bits[23:20] Non-secure state instruction tracing */
696         drvdata->ns_ex_level = BMVAL(etmidr3, 20, 23);
697
698         /*
699          * TRCERR, bit[24] whether a trace unit can trace a
700          * system error exception.
701          */
702         if (BMVAL(etmidr3, 24, 24))
703                 drvdata->trc_error = true;
704         else
705                 drvdata->trc_error = false;
706
707         /* SYNCPR, bit[25] implementation has a fixed synchronization period? */
708         if (BMVAL(etmidr3, 25, 25))
709                 drvdata->syncpr = true;
710         else
711                 drvdata->syncpr = false;
712
713         /* STALLCTL, bit[26] is stall control implemented? */
714         if (BMVAL(etmidr3, 26, 26))
715                 drvdata->stallctl = true;
716         else
717                 drvdata->stallctl = false;
718
719         /* SYSSTALL, bit[27] implementation can support stall control? */
720         if (BMVAL(etmidr3, 27, 27))
721                 drvdata->sysstall = true;
722         else
723                 drvdata->sysstall = false;
724
725         /* NUMPROC, bits[30:28] the number of PEs available for tracing */
726         drvdata->nr_pe = BMVAL(etmidr3, 28, 30);
727
728         /* NOOVERFLOW, bit[31] is trace overflow prevention supported */
729         if (BMVAL(etmidr3, 31, 31))
730                 drvdata->nooverflow = true;
731         else
732                 drvdata->nooverflow = false;
733
734         /* number of resources trace unit supports */
735         etmidr4 = readl_relaxed(drvdata->base + TRCIDR4);
736         /* NUMACPAIRS, bits[0:3] number of addr comparator pairs for tracing */
737         drvdata->nr_addr_cmp = BMVAL(etmidr4, 0, 3);
738         /* NUMPC, bits[15:12] number of PE comparator inputs for tracing */
739         drvdata->nr_pe_cmp = BMVAL(etmidr4, 12, 15);
740         /*
741          * NUMRSPAIR, bits[19:16]
742          * The number of resource pairs conveyed by the HW starts at 0, i.e a
743          * value of 0x0 indicate 1 resource pair, 0x1 indicate two and so on.
744          * As such add 1 to the value of NUMRSPAIR for a better representation.
745          *
746          * For ETM v4.3 and later, 0x0 means 0, and no pairs are available -
747          * the default TRUE and FALSE resource selectors are omitted.
748          * Otherwise for values 0x1 and above the number is N + 1 as per v4.2.
749          */
750         drvdata->nr_resource = BMVAL(etmidr4, 16, 19);
751         if ((drvdata->arch < ETM4X_ARCH_4V3) || (drvdata->nr_resource > 0))
752                 drvdata->nr_resource += 1;
753         /*
754          * NUMSSCC, bits[23:20] the number of single-shot
755          * comparator control for tracing. Read any status regs as these
756          * also contain RO capability data.
757          */
758         drvdata->nr_ss_cmp = BMVAL(etmidr4, 20, 23);
759         for (i = 0; i < drvdata->nr_ss_cmp; i++) {
760                 drvdata->config.ss_status[i] =
761                         readl_relaxed(drvdata->base + TRCSSCSRn(i));
762         }
763         /* NUMCIDC, bits[27:24] number of Context ID comparators for tracing */
764         drvdata->numcidc = BMVAL(etmidr4, 24, 27);
765         /* NUMVMIDC, bits[31:28] number of VMID comparators for tracing */
766         drvdata->numvmidc = BMVAL(etmidr4, 28, 31);
767
768         etmidr5 = readl_relaxed(drvdata->base + TRCIDR5);
769         /* NUMEXTIN, bits[8:0] number of external inputs implemented */
770         drvdata->nr_ext_inp = BMVAL(etmidr5, 0, 8);
771         /* TRACEIDSIZE, bits[21:16] indicates the trace ID width */
772         drvdata->trcid_size = BMVAL(etmidr5, 16, 21);
773         /* ATBTRIG, bit[22] implementation can support ATB triggers? */
774         if (BMVAL(etmidr5, 22, 22))
775                 drvdata->atbtrig = true;
776         else
777                 drvdata->atbtrig = false;
778         /*
779          * LPOVERRIDE, bit[23] implementation supports
780          * low-power state override
781          */
782         if (BMVAL(etmidr5, 23, 23))
783                 drvdata->lpoverride = true;
784         else
785                 drvdata->lpoverride = false;
786         /* NUMSEQSTATE, bits[27:25] number of sequencer states implemented */
787         drvdata->nrseqstate = BMVAL(etmidr5, 25, 27);
788         /* NUMCNTR, bits[30:28] number of counters available for tracing */
789         drvdata->nr_cntr = BMVAL(etmidr5, 28, 30);
790         CS_LOCK(drvdata->base);
791 }
792
793 /* Set ELx trace filter access in the TRCVICTLR register */
794 static void etm4_set_victlr_access(struct etmv4_config *config)
795 {
796         u64 access_type;
797
798         config->vinst_ctrl &= ~(ETM_EXLEVEL_S_VICTLR_MASK | ETM_EXLEVEL_NS_VICTLR_MASK);
799
800         /*
801          * TRCVICTLR::EXLEVEL_NS:EXLEVELS: Set kernel / user filtering
802          * bits in vinst_ctrl, same bit pattern as TRCACATRn values returned by
803          * etm4_get_access_type() but with a relative shift in this register.
804          */
805         access_type = etm4_get_access_type(config) << ETM_EXLEVEL_LSHIFT_TRCVICTLR;
806         config->vinst_ctrl |= (u32)access_type;
807 }
808
809 static void etm4_set_default_config(struct etmv4_config *config)
810 {
811         /* disable all events tracing */
812         config->eventctrl0 = 0x0;
813         config->eventctrl1 = 0x0;
814
815         /* disable stalling */
816         config->stall_ctrl = 0x0;
817
818         /* enable trace synchronization every 4096 bytes, if available */
819         config->syncfreq = 0xC;
820
821         /* disable timestamp event */
822         config->ts_ctrl = 0x0;
823
824         /* TRCVICTLR::EVENT = 0x01, select the always on logic */
825         config->vinst_ctrl = BIT(0);
826
827         /* TRCVICTLR::EXLEVEL_NS:EXLEVELS: Set kernel / user filtering */
828         etm4_set_victlr_access(config);
829 }
830
831 static u64 etm4_get_ns_access_type(struct etmv4_config *config)
832 {
833         u64 access_type = 0;
834
835         /*
836          * EXLEVEL_NS, bits[15:12]
837          * The Exception levels are:
838          *   Bit[12] Exception level 0 - Application
839          *   Bit[13] Exception level 1 - OS
840          *   Bit[14] Exception level 2 - Hypervisor
841          *   Bit[15] Never implemented
842          */
843         if (!is_kernel_in_hyp_mode()) {
844                 /* Stay away from hypervisor mode for non-VHE */
845                 access_type =  ETM_EXLEVEL_NS_HYP;
846                 if (config->mode & ETM_MODE_EXCL_KERN)
847                         access_type |= ETM_EXLEVEL_NS_OS;
848         } else if (config->mode & ETM_MODE_EXCL_KERN) {
849                 access_type = ETM_EXLEVEL_NS_HYP;
850         }
851
852         if (config->mode & ETM_MODE_EXCL_USER)
853                 access_type |= ETM_EXLEVEL_NS_APP;
854
855         return access_type;
856 }
857
858 static u64 etm4_get_access_type(struct etmv4_config *config)
859 {
860         u64 access_type = etm4_get_ns_access_type(config);
861         u64 s_hyp = (config->arch & 0x0f) >= 0x4 ? ETM_EXLEVEL_S_HYP : 0;
862
863         /*
864          * EXLEVEL_S, bits[11:8], don't trace anything happening
865          * in secure state.
866          */
867         access_type |= (ETM_EXLEVEL_S_APP       |
868                         ETM_EXLEVEL_S_OS        |
869                         s_hyp                   |
870                         ETM_EXLEVEL_S_MON);
871
872         return access_type;
873 }
874
875 static void etm4_set_comparator_filter(struct etmv4_config *config,
876                                        u64 start, u64 stop, int comparator)
877 {
878         u64 access_type = etm4_get_access_type(config);
879
880         /* First half of default address comparator */
881         config->addr_val[comparator] = start;
882         config->addr_acc[comparator] = access_type;
883         config->addr_type[comparator] = ETM_ADDR_TYPE_RANGE;
884
885         /* Second half of default address comparator */
886         config->addr_val[comparator + 1] = stop;
887         config->addr_acc[comparator + 1] = access_type;
888         config->addr_type[comparator + 1] = ETM_ADDR_TYPE_RANGE;
889
890         /*
891          * Configure the ViewInst function to include this address range
892          * comparator.
893          *
894          * @comparator is divided by two since it is the index in the
895          * etmv4_config::addr_val array but register TRCVIIECTLR deals with
896          * address range comparator _pairs_.
897          *
898          * Therefore:
899          *      index 0 -> compatator pair 0
900          *      index 2 -> comparator pair 1
901          *      index 4 -> comparator pair 2
902          *      ...
903          *      index 14 -> comparator pair 7
904          */
905         config->viiectlr |= BIT(comparator / 2);
906 }
907
908 static void etm4_set_start_stop_filter(struct etmv4_config *config,
909                                        u64 address, int comparator,
910                                        enum etm_addr_type type)
911 {
912         int shift;
913         u64 access_type = etm4_get_access_type(config);
914
915         /* Configure the comparator */
916         config->addr_val[comparator] = address;
917         config->addr_acc[comparator] = access_type;
918         config->addr_type[comparator] = type;
919
920         /*
921          * Configure ViewInst Start-Stop control register.
922          * Addresses configured to start tracing go from bit 0 to n-1,
923          * while those configured to stop tracing from 16 to 16 + n-1.
924          */
925         shift = (type == ETM_ADDR_TYPE_START ? 0 : 16);
926         config->vissctlr |= BIT(shift + comparator);
927 }
928
929 static void etm4_set_default_filter(struct etmv4_config *config)
930 {
931         /* Trace everything 'default' filter achieved by no filtering */
932         config->viiectlr = 0x0;
933
934         /*
935          * TRCVICTLR::SSSTATUS == 1, the start-stop logic is
936          * in the started state
937          */
938         config->vinst_ctrl |= BIT(9);
939         config->mode |= ETM_MODE_VIEWINST_STARTSTOP;
940
941         /* No start-stop filtering for ViewInst */
942         config->vissctlr = 0x0;
943 }
944
945 static void etm4_set_default(struct etmv4_config *config)
946 {
947         if (WARN_ON_ONCE(!config))
948                 return;
949
950         /*
951          * Make default initialisation trace everything
952          *
953          * This is done by a minimum default config sufficient to enable
954          * full instruction trace - with a default filter for trace all
955          * achieved by having no filtering.
956          */
957         etm4_set_default_config(config);
958         etm4_set_default_filter(config);
959 }
960
961 static int etm4_get_next_comparator(struct etmv4_drvdata *drvdata, u32 type)
962 {
963         int nr_comparator, index = 0;
964         struct etmv4_config *config = &drvdata->config;
965
966         /*
967          * nr_addr_cmp holds the number of comparator _pair_, so time 2
968          * for the total number of comparators.
969          */
970         nr_comparator = drvdata->nr_addr_cmp * 2;
971
972         /* Go through the tally of comparators looking for a free one. */
973         while (index < nr_comparator) {
974                 switch (type) {
975                 case ETM_ADDR_TYPE_RANGE:
976                         if (config->addr_type[index] == ETM_ADDR_TYPE_NONE &&
977                             config->addr_type[index + 1] == ETM_ADDR_TYPE_NONE)
978                                 return index;
979
980                         /* Address range comparators go in pairs */
981                         index += 2;
982                         break;
983                 case ETM_ADDR_TYPE_START:
984                 case ETM_ADDR_TYPE_STOP:
985                         if (config->addr_type[index] == ETM_ADDR_TYPE_NONE)
986                                 return index;
987
988                         /* Start/stop address can have odd indexes */
989                         index += 1;
990                         break;
991                 default:
992                         return -EINVAL;
993                 }
994         }
995
996         /* If we are here all the comparators have been used. */
997         return -ENOSPC;
998 }
999
1000 static int etm4_set_event_filters(struct etmv4_drvdata *drvdata,
1001                                   struct perf_event *event)
1002 {
1003         int i, comparator, ret = 0;
1004         u64 address;
1005         struct etmv4_config *config = &drvdata->config;
1006         struct etm_filters *filters = event->hw.addr_filters;
1007
1008         if (!filters)
1009                 goto default_filter;
1010
1011         /* Sync events with what Perf got */
1012         perf_event_addr_filters_sync(event);
1013
1014         /*
1015          * If there are no filters to deal with simply go ahead with
1016          * the default filter, i.e the entire address range.
1017          */
1018         if (!filters->nr_filters)
1019                 goto default_filter;
1020
1021         for (i = 0; i < filters->nr_filters; i++) {
1022                 struct etm_filter *filter = &filters->etm_filter[i];
1023                 enum etm_addr_type type = filter->type;
1024
1025                 /* See if a comparator is free. */
1026                 comparator = etm4_get_next_comparator(drvdata, type);
1027                 if (comparator < 0) {
1028                         ret = comparator;
1029                         goto out;
1030                 }
1031
1032                 switch (type) {
1033                 case ETM_ADDR_TYPE_RANGE:
1034                         etm4_set_comparator_filter(config,
1035                                                    filter->start_addr,
1036                                                    filter->stop_addr,
1037                                                    comparator);
1038                         /*
1039                          * TRCVICTLR::SSSTATUS == 1, the start-stop logic is
1040                          * in the started state
1041                          */
1042                         config->vinst_ctrl |= BIT(9);
1043
1044                         /* No start-stop filtering for ViewInst */
1045                         config->vissctlr = 0x0;
1046                         break;
1047                 case ETM_ADDR_TYPE_START:
1048                 case ETM_ADDR_TYPE_STOP:
1049                         /* Get the right start or stop address */
1050                         address = (type == ETM_ADDR_TYPE_START ?
1051                                    filter->start_addr :
1052                                    filter->stop_addr);
1053
1054                         /* Configure comparator */
1055                         etm4_set_start_stop_filter(config, address,
1056                                                    comparator, type);
1057
1058                         /*
1059                          * If filters::ssstatus == 1, trace acquisition was
1060                          * started but the process was yanked away before the
1061                          * the stop address was hit.  As such the start/stop
1062                          * logic needs to be re-started so that tracing can
1063                          * resume where it left.
1064                          *
1065                          * The start/stop logic status when a process is
1066                          * scheduled out is checked in function
1067                          * etm4_disable_perf().
1068                          */
1069                         if (filters->ssstatus)
1070                                 config->vinst_ctrl |= BIT(9);
1071
1072                         /* No include/exclude filtering for ViewInst */
1073                         config->viiectlr = 0x0;
1074                         break;
1075                 default:
1076                         ret = -EINVAL;
1077                         goto out;
1078                 }
1079         }
1080
1081         goto out;
1082
1083
1084 default_filter:
1085         etm4_set_default_filter(config);
1086
1087 out:
1088         return ret;
1089 }
1090
1091 void etm4_config_trace_mode(struct etmv4_config *config)
1092 {
1093         u32 mode;
1094
1095         mode = config->mode;
1096         mode &= (ETM_MODE_EXCL_KERN | ETM_MODE_EXCL_USER);
1097
1098         /* excluding kernel AND user space doesn't make sense */
1099         WARN_ON_ONCE(mode == (ETM_MODE_EXCL_KERN | ETM_MODE_EXCL_USER));
1100
1101         /* nothing to do if neither flags are set */
1102         if (!(mode & ETM_MODE_EXCL_KERN) && !(mode & ETM_MODE_EXCL_USER))
1103                 return;
1104
1105         etm4_set_victlr_access(config);
1106 }
1107
1108 static int etm4_online_cpu(unsigned int cpu)
1109 {
1110         if (!etmdrvdata[cpu])
1111                 return 0;
1112
1113         if (etmdrvdata[cpu]->boot_enable && !etmdrvdata[cpu]->sticky_enable)
1114                 coresight_enable(etmdrvdata[cpu]->csdev);
1115         return 0;
1116 }
1117
1118 static int etm4_starting_cpu(unsigned int cpu)
1119 {
1120         if (!etmdrvdata[cpu])
1121                 return 0;
1122
1123         spin_lock(&etmdrvdata[cpu]->spinlock);
1124         if (!etmdrvdata[cpu]->os_unlock)
1125                 etm4_os_unlock(etmdrvdata[cpu]);
1126
1127         if (local_read(&etmdrvdata[cpu]->mode))
1128                 etm4_enable_hw(etmdrvdata[cpu]);
1129         spin_unlock(&etmdrvdata[cpu]->spinlock);
1130         return 0;
1131 }
1132
1133 static int etm4_dying_cpu(unsigned int cpu)
1134 {
1135         if (!etmdrvdata[cpu])
1136                 return 0;
1137
1138         spin_lock(&etmdrvdata[cpu]->spinlock);
1139         if (local_read(&etmdrvdata[cpu]->mode))
1140                 etm4_disable_hw(etmdrvdata[cpu]);
1141         spin_unlock(&etmdrvdata[cpu]->spinlock);
1142         return 0;
1143 }
1144
1145 static void etm4_init_trace_id(struct etmv4_drvdata *drvdata)
1146 {
1147         drvdata->trcid = coresight_get_trace_id(drvdata->cpu);
1148 }
1149
1150 static int etm4_cpu_save(struct etmv4_drvdata *drvdata)
1151 {
1152         int i, ret = 0;
1153         struct etmv4_save_state *state;
1154         struct device *etm_dev = &drvdata->csdev->dev;
1155
1156         /*
1157          * As recommended by 3.4.1 ("The procedure when powering down the PE")
1158          * of ARM IHI 0064D
1159          */
1160         dsb(sy);
1161         isb();
1162
1163         CS_UNLOCK(drvdata->base);
1164
1165         /* Lock the OS lock to disable trace and external debugger access */
1166         etm4_os_lock(drvdata);
1167
1168         /* wait for TRCSTATR.PMSTABLE to go up */
1169         if (coresight_timeout(drvdata->base, TRCSTATR,
1170                               TRCSTATR_PMSTABLE_BIT, 1)) {
1171                 dev_err(etm_dev,
1172                         "timeout while waiting for PM Stable Status\n");
1173                 etm4_os_unlock(drvdata);
1174                 ret = -EBUSY;
1175                 goto out;
1176         }
1177
1178         state = drvdata->save_state;
1179
1180         state->trcprgctlr = readl(drvdata->base + TRCPRGCTLR);
1181         state->trcprocselr = readl(drvdata->base + TRCPROCSELR);
1182         state->trcconfigr = readl(drvdata->base + TRCCONFIGR);
1183         state->trcauxctlr = readl(drvdata->base + TRCAUXCTLR);
1184         state->trceventctl0r = readl(drvdata->base + TRCEVENTCTL0R);
1185         state->trceventctl1r = readl(drvdata->base + TRCEVENTCTL1R);
1186         state->trcstallctlr = readl(drvdata->base + TRCSTALLCTLR);
1187         state->trctsctlr = readl(drvdata->base + TRCTSCTLR);
1188         state->trcsyncpr = readl(drvdata->base + TRCSYNCPR);
1189         state->trcccctlr = readl(drvdata->base + TRCCCCTLR);
1190         state->trcbbctlr = readl(drvdata->base + TRCBBCTLR);
1191         state->trctraceidr = readl(drvdata->base + TRCTRACEIDR);
1192         state->trcqctlr = readl(drvdata->base + TRCQCTLR);
1193
1194         state->trcvictlr = readl(drvdata->base + TRCVICTLR);
1195         state->trcviiectlr = readl(drvdata->base + TRCVIIECTLR);
1196         state->trcvissctlr = readl(drvdata->base + TRCVISSCTLR);
1197         state->trcvipcssctlr = readl(drvdata->base + TRCVIPCSSCTLR);
1198         state->trcvdctlr = readl(drvdata->base + TRCVDCTLR);
1199         state->trcvdsacctlr = readl(drvdata->base + TRCVDSACCTLR);
1200         state->trcvdarcctlr = readl(drvdata->base + TRCVDARCCTLR);
1201
1202         for (i = 0; i < drvdata->nrseqstate - 1; i++)
1203                 state->trcseqevr[i] = readl(drvdata->base + TRCSEQEVRn(i));
1204
1205         state->trcseqrstevr = readl(drvdata->base + TRCSEQRSTEVR);
1206         state->trcseqstr = readl(drvdata->base + TRCSEQSTR);
1207         state->trcextinselr = readl(drvdata->base + TRCEXTINSELR);
1208
1209         for (i = 0; i < drvdata->nr_cntr; i++) {
1210                 state->trccntrldvr[i] = readl(drvdata->base + TRCCNTRLDVRn(i));
1211                 state->trccntctlr[i] = readl(drvdata->base + TRCCNTCTLRn(i));
1212                 state->trccntvr[i] = readl(drvdata->base + TRCCNTVRn(i));
1213         }
1214
1215         for (i = 0; i < drvdata->nr_resource * 2; i++)
1216                 state->trcrsctlr[i] = readl(drvdata->base + TRCRSCTLRn(i));
1217
1218         for (i = 0; i < drvdata->nr_ss_cmp; i++) {
1219                 state->trcssccr[i] = readl(drvdata->base + TRCSSCCRn(i));
1220                 state->trcsscsr[i] = readl(drvdata->base + TRCSSCSRn(i));
1221                 state->trcsspcicr[i] = readl(drvdata->base + TRCSSPCICRn(i));
1222         }
1223
1224         for (i = 0; i < drvdata->nr_addr_cmp * 2; i++) {
1225                 state->trcacvr[i] = readq(drvdata->base + TRCACVRn(i));
1226                 state->trcacatr[i] = readq(drvdata->base + TRCACATRn(i));
1227         }
1228
1229         /*
1230          * Data trace stream is architecturally prohibited for A profile cores
1231          * so we don't save (or later restore) trcdvcvr and trcdvcmr - As per
1232          * section 1.3.4 ("Possible functional configurations of an ETMv4 trace
1233          * unit") of ARM IHI 0064D.
1234          */
1235
1236         for (i = 0; i < drvdata->numcidc; i++)
1237                 state->trccidcvr[i] = readq(drvdata->base + TRCCIDCVRn(i));
1238
1239         for (i = 0; i < drvdata->numvmidc; i++)
1240                 state->trcvmidcvr[i] = readq(drvdata->base + TRCVMIDCVRn(i));
1241
1242         state->trccidcctlr0 = readl(drvdata->base + TRCCIDCCTLR0);
1243         state->trccidcctlr1 = readl(drvdata->base + TRCCIDCCTLR1);
1244
1245         state->trcvmidcctlr0 = readl(drvdata->base + TRCVMIDCCTLR0);
1246         state->trcvmidcctlr0 = readl(drvdata->base + TRCVMIDCCTLR1);
1247
1248         state->trcclaimset = readl(drvdata->base + TRCCLAIMCLR);
1249
1250         state->trcpdcr = readl(drvdata->base + TRCPDCR);
1251
1252         /* wait for TRCSTATR.IDLE to go up */
1253         if (coresight_timeout(drvdata->base, TRCSTATR, TRCSTATR_IDLE_BIT, 1)) {
1254                 dev_err(etm_dev,
1255                         "timeout while waiting for Idle Trace Status\n");
1256                 etm4_os_unlock(drvdata);
1257                 ret = -EBUSY;
1258                 goto out;
1259         }
1260
1261         drvdata->state_needs_restore = true;
1262
1263         /*
1264          * Power can be removed from the trace unit now. We do this to
1265          * potentially save power on systems that respect the TRCPDCR_PU
1266          * despite requesting software to save/restore state.
1267          */
1268         writel_relaxed((state->trcpdcr & ~TRCPDCR_PU),
1269                         drvdata->base + TRCPDCR);
1270
1271 out:
1272         CS_LOCK(drvdata->base);
1273         return ret;
1274 }
1275
1276 static void etm4_cpu_restore(struct etmv4_drvdata *drvdata)
1277 {
1278         int i;
1279         struct etmv4_save_state *state = drvdata->save_state;
1280
1281         CS_UNLOCK(drvdata->base);
1282
1283         writel_relaxed(state->trcclaimset, drvdata->base + TRCCLAIMSET);
1284
1285         writel_relaxed(state->trcprgctlr, drvdata->base + TRCPRGCTLR);
1286         writel_relaxed(state->trcprocselr, drvdata->base + TRCPROCSELR);
1287         writel_relaxed(state->trcconfigr, drvdata->base + TRCCONFIGR);
1288         writel_relaxed(state->trcauxctlr, drvdata->base + TRCAUXCTLR);
1289         writel_relaxed(state->trceventctl0r, drvdata->base + TRCEVENTCTL0R);
1290         writel_relaxed(state->trceventctl1r, drvdata->base + TRCEVENTCTL1R);
1291         writel_relaxed(state->trcstallctlr, drvdata->base + TRCSTALLCTLR);
1292         writel_relaxed(state->trctsctlr, drvdata->base + TRCTSCTLR);
1293         writel_relaxed(state->trcsyncpr, drvdata->base + TRCSYNCPR);
1294         writel_relaxed(state->trcccctlr, drvdata->base + TRCCCCTLR);
1295         writel_relaxed(state->trcbbctlr, drvdata->base + TRCBBCTLR);
1296         writel_relaxed(state->trctraceidr, drvdata->base + TRCTRACEIDR);
1297         writel_relaxed(state->trcqctlr, drvdata->base + TRCQCTLR);
1298
1299         writel_relaxed(state->trcvictlr, drvdata->base + TRCVICTLR);
1300         writel_relaxed(state->trcviiectlr, drvdata->base + TRCVIIECTLR);
1301         writel_relaxed(state->trcvissctlr, drvdata->base + TRCVISSCTLR);
1302         writel_relaxed(state->trcvipcssctlr, drvdata->base + TRCVIPCSSCTLR);
1303         writel_relaxed(state->trcvdctlr, drvdata->base + TRCVDCTLR);
1304         writel_relaxed(state->trcvdsacctlr, drvdata->base + TRCVDSACCTLR);
1305         writel_relaxed(state->trcvdarcctlr, drvdata->base + TRCVDARCCTLR);
1306
1307         for (i = 0; i < drvdata->nrseqstate - 1; i++)
1308                 writel_relaxed(state->trcseqevr[i],
1309                                drvdata->base + TRCSEQEVRn(i));
1310
1311         writel_relaxed(state->trcseqrstevr, drvdata->base + TRCSEQRSTEVR);
1312         writel_relaxed(state->trcseqstr, drvdata->base + TRCSEQSTR);
1313         writel_relaxed(state->trcextinselr, drvdata->base + TRCEXTINSELR);
1314
1315         for (i = 0; i < drvdata->nr_cntr; i++) {
1316                 writel_relaxed(state->trccntrldvr[i],
1317                                drvdata->base + TRCCNTRLDVRn(i));
1318                 writel_relaxed(state->trccntctlr[i],
1319                                drvdata->base + TRCCNTCTLRn(i));
1320                 writel_relaxed(state->trccntvr[i],
1321                                drvdata->base + TRCCNTVRn(i));
1322         }
1323
1324         for (i = 0; i < drvdata->nr_resource * 2; i++)
1325                 writel_relaxed(state->trcrsctlr[i],
1326                                drvdata->base + TRCRSCTLRn(i));
1327
1328         for (i = 0; i < drvdata->nr_ss_cmp; i++) {
1329                 writel_relaxed(state->trcssccr[i],
1330                                drvdata->base + TRCSSCCRn(i));
1331                 writel_relaxed(state->trcsscsr[i],
1332                                drvdata->base + TRCSSCSRn(i));
1333                 writel_relaxed(state->trcsspcicr[i],
1334                                drvdata->base + TRCSSPCICRn(i));
1335         }
1336
1337         for (i = 0; i < drvdata->nr_addr_cmp * 2; i++) {
1338                 writeq_relaxed(state->trcacvr[i],
1339                                drvdata->base + TRCACVRn(i));
1340                 writeq_relaxed(state->trcacatr[i],
1341                                drvdata->base + TRCACATRn(i));
1342         }
1343
1344         for (i = 0; i < drvdata->numcidc; i++)
1345                 writeq_relaxed(state->trccidcvr[i],
1346                                drvdata->base + TRCCIDCVRn(i));
1347
1348         for (i = 0; i < drvdata->numvmidc; i++)
1349                 writeq_relaxed(state->trcvmidcvr[i],
1350                                drvdata->base + TRCVMIDCVRn(i));
1351
1352         writel_relaxed(state->trccidcctlr0, drvdata->base + TRCCIDCCTLR0);
1353         writel_relaxed(state->trccidcctlr1, drvdata->base + TRCCIDCCTLR1);
1354
1355         writel_relaxed(state->trcvmidcctlr0, drvdata->base + TRCVMIDCCTLR0);
1356         writel_relaxed(state->trcvmidcctlr0, drvdata->base + TRCVMIDCCTLR1);
1357
1358         writel_relaxed(state->trcclaimset, drvdata->base + TRCCLAIMSET);
1359
1360         writel_relaxed(state->trcpdcr, drvdata->base + TRCPDCR);
1361
1362         drvdata->state_needs_restore = false;
1363
1364         /*
1365          * As recommended by section 4.3.7 ("Synchronization when using the
1366          * memory-mapped interface") of ARM IHI 0064D
1367          */
1368         dsb(sy);
1369         isb();
1370
1371         /* Unlock the OS lock to re-enable trace and external debug access */
1372         etm4_os_unlock(drvdata);
1373         CS_LOCK(drvdata->base);
1374 }
1375
1376 static int etm4_cpu_pm_notify(struct notifier_block *nb, unsigned long cmd,
1377                               void *v)
1378 {
1379         struct etmv4_drvdata *drvdata;
1380         unsigned int cpu = smp_processor_id();
1381
1382         if (!etmdrvdata[cpu])
1383                 return NOTIFY_OK;
1384
1385         drvdata = etmdrvdata[cpu];
1386
1387         if (!drvdata->save_state)
1388                 return NOTIFY_OK;
1389
1390         if (WARN_ON_ONCE(drvdata->cpu != cpu))
1391                 return NOTIFY_BAD;
1392
1393         switch (cmd) {
1394         case CPU_PM_ENTER:
1395                 /* save the state if self-hosted coresight is in use */
1396                 if (local_read(&drvdata->mode))
1397                         if (etm4_cpu_save(drvdata))
1398                                 return NOTIFY_BAD;
1399                 break;
1400         case CPU_PM_EXIT:
1401         case CPU_PM_ENTER_FAILED:
1402                 if (drvdata->state_needs_restore)
1403                         etm4_cpu_restore(drvdata);
1404                 break;
1405         default:
1406                 return NOTIFY_DONE;
1407         }
1408
1409         return NOTIFY_OK;
1410 }
1411
1412 static struct notifier_block etm4_cpu_pm_nb = {
1413         .notifier_call = etm4_cpu_pm_notify,
1414 };
1415
1416 /* Setup PM. Deals with error conditions and counts */
1417 static int __init etm4_pm_setup(void)
1418 {
1419         int ret;
1420
1421         ret = cpu_pm_register_notifier(&etm4_cpu_pm_nb);
1422         if (ret)
1423                 return ret;
1424
1425         ret = cpuhp_setup_state_nocalls(CPUHP_AP_ARM_CORESIGHT_STARTING,
1426                                         "arm/coresight4:starting",
1427                                         etm4_starting_cpu, etm4_dying_cpu);
1428
1429         if (ret)
1430                 goto unregister_notifier;
1431
1432         ret = cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN,
1433                                         "arm/coresight4:online",
1434                                         etm4_online_cpu, NULL);
1435
1436         /* HP dyn state ID returned in ret on success */
1437         if (ret > 0) {
1438                 hp_online = ret;
1439                 return 0;
1440         }
1441
1442         /* failed dyn state - remove others */
1443         cpuhp_remove_state_nocalls(CPUHP_AP_ARM_CORESIGHT_STARTING);
1444
1445 unregister_notifier:
1446         cpu_pm_unregister_notifier(&etm4_cpu_pm_nb);
1447         return ret;
1448 }
1449
1450 static void etm4_pm_clear(void)
1451 {
1452         cpu_pm_unregister_notifier(&etm4_cpu_pm_nb);
1453         cpuhp_remove_state_nocalls(CPUHP_AP_ARM_CORESIGHT_STARTING);
1454         if (hp_online) {
1455                 cpuhp_remove_state_nocalls(hp_online);
1456                 hp_online = 0;
1457         }
1458 }
1459
1460 static int etm4_probe(struct amba_device *adev, const struct amba_id *id)
1461 {
1462         int ret;
1463         void __iomem *base;
1464         struct device *dev = &adev->dev;
1465         struct coresight_platform_data *pdata = NULL;
1466         struct etmv4_drvdata *drvdata;
1467         struct resource *res = &adev->res;
1468         struct coresight_desc desc = { 0 };
1469
1470         drvdata = devm_kzalloc(dev, sizeof(*drvdata), GFP_KERNEL);
1471         if (!drvdata)
1472                 return -ENOMEM;
1473
1474         dev_set_drvdata(dev, drvdata);
1475
1476         if (pm_save_enable == PARAM_PM_SAVE_FIRMWARE)
1477                 pm_save_enable = coresight_loses_context_with_cpu(dev) ?
1478                                PARAM_PM_SAVE_SELF_HOSTED : PARAM_PM_SAVE_NEVER;
1479
1480         if (pm_save_enable != PARAM_PM_SAVE_NEVER) {
1481                 drvdata->save_state = devm_kmalloc(dev,
1482                                 sizeof(struct etmv4_save_state), GFP_KERNEL);
1483                 if (!drvdata->save_state)
1484                         return -ENOMEM;
1485         }
1486
1487         if (fwnode_property_present(dev_fwnode(dev), "qcom,skip-power-up"))
1488                 drvdata->skip_power_up = true;
1489
1490         /* Validity for the resource is already checked by the AMBA core */
1491         base = devm_ioremap_resource(dev, res);
1492         if (IS_ERR(base))
1493                 return PTR_ERR(base);
1494
1495         drvdata->base = base;
1496
1497         spin_lock_init(&drvdata->spinlock);
1498
1499         drvdata->cpu = coresight_get_cpu(dev);
1500         if (drvdata->cpu < 0)
1501                 return drvdata->cpu;
1502
1503         desc.name = devm_kasprintf(dev, GFP_KERNEL, "etm%d", drvdata->cpu);
1504         if (!desc.name)
1505                 return -ENOMEM;
1506
1507         if (smp_call_function_single(drvdata->cpu,
1508                                 etm4_init_arch_data,  drvdata, 1))
1509                 dev_err(dev, "ETM arch init failed\n");
1510
1511         if (etm4_arch_supported(drvdata->arch) == false)
1512                 return -EINVAL;
1513
1514         etm4_init_trace_id(drvdata);
1515         etm4_set_default(&drvdata->config);
1516
1517         pdata = coresight_get_platform_data(dev);
1518         if (IS_ERR(pdata))
1519                 return PTR_ERR(pdata);
1520
1521         adev->dev.platform_data = pdata;
1522
1523         desc.type = CORESIGHT_DEV_TYPE_SOURCE;
1524         desc.subtype.source_subtype = CORESIGHT_DEV_SUBTYPE_SOURCE_PROC;
1525         desc.ops = &etm4_cs_ops;
1526         desc.pdata = pdata;
1527         desc.dev = dev;
1528         desc.groups = coresight_etmv4_groups;
1529         drvdata->csdev = coresight_register(&desc);
1530         if (IS_ERR(drvdata->csdev))
1531                 return PTR_ERR(drvdata->csdev);
1532
1533         ret = etm_perf_symlink(drvdata->csdev, true);
1534         if (ret) {
1535                 coresight_unregister(drvdata->csdev);
1536                 return ret;
1537         }
1538
1539         etmdrvdata[drvdata->cpu] = drvdata;
1540
1541         pm_runtime_put(&adev->dev);
1542         dev_info(&drvdata->csdev->dev, "CPU%d: ETM v%d.%d initialized\n",
1543                  drvdata->cpu, drvdata->arch >> 4, drvdata->arch & 0xf);
1544
1545         if (boot_enable) {
1546                 coresight_enable(drvdata->csdev);
1547                 drvdata->boot_enable = true;
1548         }
1549
1550         return 0;
1551 }
1552
1553 static struct amba_cs_uci_id uci_id_etm4[] = {
1554         {
1555                 /*  ETMv4 UCI data */
1556                 .devarch        = 0x47704a13,
1557                 .devarch_mask   = 0xfff0ffff,
1558                 .devtype        = 0x00000013,
1559         }
1560 };
1561
1562 static void __exit clear_etmdrvdata(void *info)
1563 {
1564         int cpu = *(int *)info;
1565
1566         etmdrvdata[cpu] = NULL;
1567 }
1568
1569 static int __exit etm4_remove(struct amba_device *adev)
1570 {
1571         struct etmv4_drvdata *drvdata = dev_get_drvdata(&adev->dev);
1572
1573         etm_perf_symlink(drvdata->csdev, false);
1574
1575         /*
1576          * Taking hotplug lock here to avoid racing between etm4_remove and
1577          * CPU hotplug call backs.
1578          */
1579         cpus_read_lock();
1580         /*
1581          * The readers for etmdrvdata[] are CPU hotplug call backs
1582          * and PM notification call backs. Change etmdrvdata[i] on
1583          * CPU i ensures these call backs has consistent view
1584          * inside one call back function.
1585          */
1586         if (smp_call_function_single(drvdata->cpu, clear_etmdrvdata, &drvdata->cpu, 1))
1587                 etmdrvdata[drvdata->cpu] = NULL;
1588
1589         cpus_read_unlock();
1590
1591         coresight_unregister(drvdata->csdev);
1592
1593         return 0;
1594 }
1595
1596 static const struct amba_id etm4_ids[] = {
1597         CS_AMBA_ID(0x000bb95d),                 /* Cortex-A53 */
1598         CS_AMBA_ID(0x000bb95e),                 /* Cortex-A57 */
1599         CS_AMBA_ID(0x000bb95a),                 /* Cortex-A72 */
1600         CS_AMBA_ID(0x000bb959),                 /* Cortex-A73 */
1601         CS_AMBA_UCI_ID(0x000bb9da, uci_id_etm4),/* Cortex-A35 */
1602         CS_AMBA_UCI_ID(0x000bbd0c, uci_id_etm4),/* Neoverse N1 */
1603         CS_AMBA_UCI_ID(0x000f0205, uci_id_etm4),/* Qualcomm Kryo */
1604         CS_AMBA_UCI_ID(0x000f0211, uci_id_etm4),/* Qualcomm Kryo */
1605         CS_AMBA_UCI_ID(0x000bb802, uci_id_etm4),/* Qualcomm Kryo 385 Cortex-A55 */
1606         CS_AMBA_UCI_ID(0x000bb803, uci_id_etm4),/* Qualcomm Kryo 385 Cortex-A75 */
1607         CS_AMBA_UCI_ID(0x000bb805, uci_id_etm4),/* Qualcomm Kryo 4XX Cortex-A55 */
1608         CS_AMBA_UCI_ID(0x000bb804, uci_id_etm4),/* Qualcomm Kryo 4XX Cortex-A76 */
1609         CS_AMBA_UCI_ID(0x000cc0af, uci_id_etm4),/* Marvell ThunderX2 */
1610         CS_AMBA_UCI_ID(0x000b6d01, uci_id_etm4),/* HiSilicon-Hip08 */
1611         CS_AMBA_UCI_ID(0x000b6d02, uci_id_etm4),/* HiSilicon-Hip09 */
1612         {},
1613 };
1614
1615 MODULE_DEVICE_TABLE(amba, etm4_ids);
1616
1617 static struct amba_driver etm4x_driver = {
1618         .drv = {
1619                 .name   = "coresight-etm4x",
1620                 .owner  = THIS_MODULE,
1621                 .suppress_bind_attrs = true,
1622         },
1623         .probe          = etm4_probe,
1624         .remove         = etm4_remove,
1625         .id_table       = etm4_ids,
1626 };
1627
1628 static int __init etm4x_init(void)
1629 {
1630         int ret;
1631
1632         ret = etm4_pm_setup();
1633
1634         /* etm4_pm_setup() does its own cleanup - exit on error */
1635         if (ret)
1636                 return ret;
1637
1638         ret = amba_driver_register(&etm4x_driver);
1639         if (ret) {
1640                 pr_err("Error registering etm4x driver\n");
1641                 etm4_pm_clear();
1642         }
1643
1644         return ret;
1645 }
1646
1647 static void __exit etm4x_exit(void)
1648 {
1649         amba_driver_unregister(&etm4x_driver);
1650         etm4_pm_clear();
1651 }
1652
1653 module_init(etm4x_init);
1654 module_exit(etm4x_exit);
1655
1656 MODULE_AUTHOR("Pratik Patel <pratikp@codeaurora.org>");
1657 MODULE_AUTHOR("Mathieu Poirier <mathieu.poirier@linaro.org>");
1658 MODULE_DESCRIPTION("Arm CoreSight Program Flow Trace v4.x driver");
1659 MODULE_LICENSE("GPL v2");