1 /*
2  *  linux/drivers/clocksource/arm_arch_timer.c
3  *
4  *  Copyright (C) 2011 ARM Ltd.
5  *  All Rights Reserved
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  */
11 #include <linux/init.h>
12 #include <linux/kernel.h>
13 #include <linux/device.h>
14 #include <linux/smp.h>
15 #include <linux/cpu.h>
16 #include <linux/cpu_pm.h>
17 #include <linux/clockchips.h>
18 #include <linux/interrupt.h>
19 #include <linux/of_irq.h>
20 #include <linux/of_address.h>
21 #include <linux/io.h>
22 #include <linux/slab.h>
23 #include <linux/sched_clock.h>
24 
25 #include <asm/arch_timer.h>
26 #include <asm/virt.h>
27 
28 #include <clocksource/arm_arch_timer.h>
29 
30 #define CNTTIDR		0x08
31 #define CNTTIDR_VIRT(n)	(BIT(1) << ((n) * 4))
32 
33 #define CNTVCT_LO	0x08
34 #define CNTVCT_HI	0x0c
35 #define CNTFRQ		0x10
36 #define CNTP_TVAL	0x28
37 #define CNTP_CTL	0x2c
38 #define CNTV_TVAL	0x38
39 #define CNTV_CTL	0x3c
40 
41 #define ARCH_CP15_TIMER	BIT(0)
42 #define ARCH_MEM_TIMER	BIT(1)
43 static unsigned arch_timers_present __initdata;
44 
45 static void __iomem *arch_counter_base;
46 
47 struct arch_timer {
48 	void __iomem *base;
49 	struct clock_event_device evt;
50 };
51 
52 #define to_arch_timer(e) container_of(e, struct arch_timer, evt)
53 
54 static u32 arch_timer_rate;
55 
56 enum ppi_nr {
57 	PHYS_SECURE_PPI,
58 	PHYS_NONSECURE_PPI,
59 	VIRT_PPI,
60 	HYP_PPI,
61 	MAX_TIMER_PPI
62 };
63 
64 static int arch_timer_ppi[MAX_TIMER_PPI];
65 
66 static struct clock_event_device __percpu *arch_timer_evt;
67 
68 static bool arch_timer_use_virtual = true;
69 static bool arch_timer_c3stop;
70 static bool arch_timer_mem_use_virtual;
71 
72 /*
73  * Architected system timer support.
74  */
75 
76 static __always_inline
77 void arch_timer_reg_write(int access, enum arch_timer_reg reg, u32 val,
78 			  struct clock_event_device *clk)
79 {
80 	if (access == ARCH_TIMER_MEM_PHYS_ACCESS) {
81 		struct arch_timer *timer = to_arch_timer(clk);
82 		switch (reg) {
83 		case ARCH_TIMER_REG_CTRL:
84 			writel_relaxed(val, timer->base + CNTP_CTL);
85 			break;
86 		case ARCH_TIMER_REG_TVAL:
87 			writel_relaxed(val, timer->base + CNTP_TVAL);
88 			break;
89 		}
90 	} else if (access == ARCH_TIMER_MEM_VIRT_ACCESS) {
91 		struct arch_timer *timer = to_arch_timer(clk);
92 		switch (reg) {
93 		case ARCH_TIMER_REG_CTRL:
94 			writel_relaxed(val, timer->base + CNTV_CTL);
95 			break;
96 		case ARCH_TIMER_REG_TVAL:
97 			writel_relaxed(val, timer->base + CNTV_TVAL);
98 			break;
99 		}
100 	} else {
101 		arch_timer_reg_write_cp15(access, reg, val);
102 	}
103 }
104 
105 static __always_inline
106 u32 arch_timer_reg_read(int access, enum arch_timer_reg reg,
107 			struct clock_event_device *clk)
108 {
109 	u32 val;
110 
111 	if (access == ARCH_TIMER_MEM_PHYS_ACCESS) {
112 		struct arch_timer *timer = to_arch_timer(clk);
113 		switch (reg) {
114 		case ARCH_TIMER_REG_CTRL:
115 			val = readl_relaxed(timer->base + CNTP_CTL);
116 			break;
117 		case ARCH_TIMER_REG_TVAL:
118 			val = readl_relaxed(timer->base + CNTP_TVAL);
119 			break;
120 		}
121 	} else if (access == ARCH_TIMER_MEM_VIRT_ACCESS) {
122 		struct arch_timer *timer = to_arch_timer(clk);
123 		switch (reg) {
124 		case ARCH_TIMER_REG_CTRL:
125 			val = readl_relaxed(timer->base + CNTV_CTL);
126 			break;
127 		case ARCH_TIMER_REG_TVAL:
128 			val = readl_relaxed(timer->base + CNTV_TVAL);
129 			break;
130 		}
131 	} else {
132 		val = arch_timer_reg_read_cp15(access, reg);
133 	}
134 
135 	return val;
136 }
137 
138 static __always_inline irqreturn_t timer_handler(const int access,
139 					struct clock_event_device *evt)
140 {
141 	unsigned long ctrl;
142 
143 	ctrl = arch_timer_reg_read(access, ARCH_TIMER_REG_CTRL, evt);
144 	if (ctrl & ARCH_TIMER_CTRL_IT_STAT) {
145 		ctrl |= ARCH_TIMER_CTRL_IT_MASK;
146 		arch_timer_reg_write(access, ARCH_TIMER_REG_CTRL, ctrl, evt);
147 		evt->event_handler(evt);
148 		return IRQ_HANDLED;
149 	}
150 
151 	return IRQ_NONE;
152 }
153 
154 static irqreturn_t arch_timer_handler_virt(int irq, void *dev_id)
155 {
156 	struct clock_event_device *evt = dev_id;
157 
158 	return timer_handler(ARCH_TIMER_VIRT_ACCESS, evt);
159 }
160 
161 static irqreturn_t arch_timer_handler_phys(int irq, void *dev_id)
162 {
163 	struct clock_event_device *evt = dev_id;
164 
165 	return timer_handler(ARCH_TIMER_PHYS_ACCESS, evt);
166 }
167 
168 static irqreturn_t arch_timer_handler_phys_mem(int irq, void *dev_id)
169 {
170 	struct clock_event_device *evt = dev_id;
171 
172 	return timer_handler(ARCH_TIMER_MEM_PHYS_ACCESS, evt);
173 }
174 
175 static irqreturn_t arch_timer_handler_virt_mem(int irq, void *dev_id)
176 {
177 	struct clock_event_device *evt = dev_id;
178 
179 	return timer_handler(ARCH_TIMER_MEM_VIRT_ACCESS, evt);
180 }
181 
182 static __always_inline void timer_set_mode(const int access, int mode,
183 				  struct clock_event_device *clk)
184 {
185 	unsigned long ctrl;
186 	switch (mode) {
187 	case CLOCK_EVT_MODE_UNUSED:
188 	case CLOCK_EVT_MODE_SHUTDOWN:
189 		ctrl = arch_timer_reg_read(access, ARCH_TIMER_REG_CTRL, clk);
190 		ctrl &= ~ARCH_TIMER_CTRL_ENABLE;
191 		arch_timer_reg_write(access, ARCH_TIMER_REG_CTRL, ctrl, clk);
192 		break;
193 	default:
194 		break;
195 	}
196 }
197 
198 static void arch_timer_set_mode_virt(enum clock_event_mode mode,
199 				     struct clock_event_device *clk)
200 {
201 	timer_set_mode(ARCH_TIMER_VIRT_ACCESS, mode, clk);
202 }
203 
204 static void arch_timer_set_mode_phys(enum clock_event_mode mode,
205 				     struct clock_event_device *clk)
206 {
207 	timer_set_mode(ARCH_TIMER_PHYS_ACCESS, mode, clk);
208 }
209 
210 static void arch_timer_set_mode_virt_mem(enum clock_event_mode mode,
211 					 struct clock_event_device *clk)
212 {
213 	timer_set_mode(ARCH_TIMER_MEM_VIRT_ACCESS, mode, clk);
214 }
215 
216 static void arch_timer_set_mode_phys_mem(enum clock_event_mode mode,
217 					 struct clock_event_device *clk)
218 {
219 	timer_set_mode(ARCH_TIMER_MEM_PHYS_ACCESS, mode, clk);
220 }
221 
222 static __always_inline void set_next_event(const int access, unsigned long evt,
223 					   struct clock_event_device *clk)
224 {
225 	unsigned long ctrl;
226 	ctrl = arch_timer_reg_read(access, ARCH_TIMER_REG_CTRL, clk);
227 	ctrl |= ARCH_TIMER_CTRL_ENABLE;
228 	ctrl &= ~ARCH_TIMER_CTRL_IT_MASK;
229 	arch_timer_reg_write(access, ARCH_TIMER_REG_TVAL, evt, clk);
230 	arch_timer_reg_write(access, ARCH_TIMER_REG_CTRL, ctrl, clk);
231 }
232 
233 static int arch_timer_set_next_event_virt(unsigned long evt,
234 					  struct clock_event_device *clk)
235 {
236 	set_next_event(ARCH_TIMER_VIRT_ACCESS, evt, clk);
237 	return 0;
238 }
239 
240 static int arch_timer_set_next_event_phys(unsigned long evt,
241 					  struct clock_event_device *clk)
242 {
243 	set_next_event(ARCH_TIMER_PHYS_ACCESS, evt, clk);
244 	return 0;
245 }
246 
247 static int arch_timer_set_next_event_virt_mem(unsigned long evt,
248 					      struct clock_event_device *clk)
249 {
250 	set_next_event(ARCH_TIMER_MEM_VIRT_ACCESS, evt, clk);
251 	return 0;
252 }
253 
254 static int arch_timer_set_next_event_phys_mem(unsigned long evt,
255 					      struct clock_event_device *clk)
256 {
257 	set_next_event(ARCH_TIMER_MEM_PHYS_ACCESS, evt, clk);
258 	return 0;
259 }
260 
261 static void __arch_timer_setup(unsigned type,
262 			       struct clock_event_device *clk)
263 {
264 	clk->features = CLOCK_EVT_FEAT_ONESHOT;
265 
266 	if (type == ARCH_CP15_TIMER) {
267 		if (arch_timer_c3stop)
268 			clk->features |= CLOCK_EVT_FEAT_C3STOP;
269 		clk->name = "arch_sys_timer";
270 		clk->rating = 450;
271 		clk->cpumask = cpumask_of(smp_processor_id());
272 		if (arch_timer_use_virtual) {
273 			clk->irq = arch_timer_ppi[VIRT_PPI];
274 			clk->set_mode = arch_timer_set_mode_virt;
275 			clk->set_next_event = arch_timer_set_next_event_virt;
276 		} else {
277 			clk->irq = arch_timer_ppi[PHYS_SECURE_PPI];
278 			clk->set_mode = arch_timer_set_mode_phys;
279 			clk->set_next_event = arch_timer_set_next_event_phys;
280 		}
281 	} else {
282 		clk->features |= CLOCK_EVT_FEAT_DYNIRQ;
283 		clk->name = "arch_mem_timer";
284 		clk->rating = 400;
285 		clk->cpumask = cpu_all_mask;
286 		if (arch_timer_mem_use_virtual) {
287 			clk->set_mode = arch_timer_set_mode_virt_mem;
288 			clk->set_next_event =
289 				arch_timer_set_next_event_virt_mem;
290 		} else {
291 			clk->set_mode = arch_timer_set_mode_phys_mem;
292 			clk->set_next_event =
293 				arch_timer_set_next_event_phys_mem;
294 		}
295 	}
296 
297 	clk->set_mode(CLOCK_EVT_MODE_SHUTDOWN, clk);
298 
299 	clockevents_config_and_register(clk, arch_timer_rate, 0xf, 0x7fffffff);
300 }
301 
302 static void arch_timer_evtstrm_enable(int divider)
303 {
304 	u32 cntkctl = arch_timer_get_cntkctl();
305 
306 	cntkctl &= ~ARCH_TIMER_EVT_TRIGGER_MASK;
307 	/* Set the divider and enable virtual event stream */
308 	cntkctl |= (divider << ARCH_TIMER_EVT_TRIGGER_SHIFT)
309 			| ARCH_TIMER_VIRT_EVT_EN;
310 	arch_timer_set_cntkctl(cntkctl);
311 	elf_hwcap |= HWCAP_EVTSTRM;
312 #ifdef CONFIG_COMPAT
313 	compat_elf_hwcap |= COMPAT_HWCAP_EVTSTRM;
314 #endif
315 }
316 
317 static void arch_timer_configure_evtstream(void)
318 {
319 	int evt_stream_div, pos;
320 
321 	/* Find the closest power of two to the divisor */
322 	evt_stream_div = arch_timer_rate / ARCH_TIMER_EVT_STREAM_FREQ;
323 	pos = fls(evt_stream_div);
324 	if (pos > 1 && !(evt_stream_div & (1 << (pos - 2))))
325 		pos--;
326 	/* enable event stream */
327 	arch_timer_evtstrm_enable(min(pos, 15));
328 }
329 
330 static void arch_counter_set_user_access(void)
331 {
332 	u32 cntkctl = arch_timer_get_cntkctl();
333 
334 	/* Disable user access to the timers and the physical counter */
335 	/* Also disable virtual event stream */
336 	cntkctl &= ~(ARCH_TIMER_USR_PT_ACCESS_EN
337 			| ARCH_TIMER_USR_VT_ACCESS_EN
338 			| ARCH_TIMER_VIRT_EVT_EN
339 			| ARCH_TIMER_USR_PCT_ACCESS_EN);
340 
341 	/* Enable user access to the virtual counter */
342 	cntkctl |= ARCH_TIMER_USR_VCT_ACCESS_EN;
343 
344 	arch_timer_set_cntkctl(cntkctl);
345 }
346 
347 static int arch_timer_setup(struct clock_event_device *clk)
348 {
349 	__arch_timer_setup(ARCH_CP15_TIMER, clk);
350 
351 	if (arch_timer_use_virtual)
352 		enable_percpu_irq(arch_timer_ppi[VIRT_PPI], 0);
353 	else {
354 		enable_percpu_irq(arch_timer_ppi[PHYS_SECURE_PPI], 0);
355 		if (arch_timer_ppi[PHYS_NONSECURE_PPI])
356 			enable_percpu_irq(arch_timer_ppi[PHYS_NONSECURE_PPI], 0);
357 	}
358 
359 	arch_counter_set_user_access();
360 	if (IS_ENABLED(CONFIG_ARM_ARCH_TIMER_EVTSTREAM))
361 		arch_timer_configure_evtstream();
362 
363 	return 0;
364 }
365 
366 static void
367 arch_timer_detect_rate(void __iomem *cntbase, struct device_node *np)
368 {
369 	/* Who has more than one independent system counter? */
370 	if (arch_timer_rate)
371 		return;
372 
373 	/* Try to determine the frequency from the device tree or CNTFRQ */
374 	if (of_property_read_u32(np, "clock-frequency", &arch_timer_rate)) {
375 		if (cntbase)
376 			arch_timer_rate = readl_relaxed(cntbase + CNTFRQ);
377 		else
378 			arch_timer_rate = arch_timer_get_cntfrq();
379 	}
380 
381 	/* Check the timer frequency. */
382 	if (arch_timer_rate == 0)
383 		pr_warn("Architected timer frequency not available\n");
384 }
385 
386 static void arch_timer_banner(unsigned type)
387 {
388 	pr_info("Architected %s%s%s timer(s) running at %lu.%02luMHz (%s%s%s).\n",
389 		     type & ARCH_CP15_TIMER ? "cp15" : "",
390 		     type == (ARCH_CP15_TIMER | ARCH_MEM_TIMER) ?  " and " : "",
391 		     type & ARCH_MEM_TIMER ? "mmio" : "",
392 		     (unsigned long)arch_timer_rate / 1000000,
393 		     (unsigned long)(arch_timer_rate / 10000) % 100,
394 		     type & ARCH_CP15_TIMER ?
395 			arch_timer_use_virtual ? "virt" : "phys" :
396 			"",
397 		     type == (ARCH_CP15_TIMER | ARCH_MEM_TIMER) ?  "/" : "",
398 		     type & ARCH_MEM_TIMER ?
399 			arch_timer_mem_use_virtual ? "virt" : "phys" :
400 			"");
401 }
402 
403 u32 arch_timer_get_rate(void)
404 {
405 	return arch_timer_rate;
406 }
407 
408 static u64 arch_counter_get_cntvct_mem(void)
409 {
410 	u32 vct_lo, vct_hi, tmp_hi;
411 
412 	do {
413 		vct_hi = readl_relaxed(arch_counter_base + CNTVCT_HI);
414 		vct_lo = readl_relaxed(arch_counter_base + CNTVCT_LO);
415 		tmp_hi = readl_relaxed(arch_counter_base + CNTVCT_HI);
416 	} while (vct_hi != tmp_hi);
417 
418 	return ((u64) vct_hi << 32) | vct_lo;
419 }
420 
421 /*
422  * Default to cp15 based access because arm64 uses this function for
423  * sched_clock() before DT is probed and the cp15 method is guaranteed
424  * to exist on arm64. arm doesn't use this before DT is probed so even
425  * if we don't have the cp15 accessors we won't have a problem.
426  */
427 u64 (*arch_timer_read_counter)(void) = arch_counter_get_cntvct;
428 
429 static cycle_t arch_counter_read(struct clocksource *cs)
430 {
431 	return arch_timer_read_counter();
432 }
433 
434 static cycle_t arch_counter_read_cc(const struct cyclecounter *cc)
435 {
436 	return arch_timer_read_counter();
437 }
438 
439 static struct clocksource clocksource_counter = {
440 	.name	= "arch_sys_counter",
441 	.rating	= 400,
442 	.read	= arch_counter_read,
443 	.mask	= CLOCKSOURCE_MASK(56),
444 	.flags	= CLOCK_SOURCE_IS_CONTINUOUS | CLOCK_SOURCE_SUSPEND_NONSTOP,
445 };
446 
447 static struct cyclecounter cyclecounter = {
448 	.read	= arch_counter_read_cc,
449 	.mask	= CLOCKSOURCE_MASK(56),
450 };
451 
452 static struct timecounter timecounter;
453 
454 struct timecounter *arch_timer_get_timecounter(void)
455 {
456 	return &timecounter;
457 }
458 
459 static void __init arch_counter_register(unsigned type)
460 {
461 	u64 start_count;
462 
463 	/* Register the CP15 based counter if we have one */
464 	if (type & ARCH_CP15_TIMER) {
465 		arch_timer_read_counter = arch_counter_get_cntvct;
466 	} else {
467 		arch_timer_read_counter = arch_counter_get_cntvct_mem;
468 
469 		/* If the clocksource name is "arch_sys_counter" the
470 		 * VDSO will attempt to read the CP15-based counter.
471 		 * Ensure this does not happen when CP15-based
472 		 * counter is not available.
473 		 */
474 		clocksource_counter.name = "arch_mem_counter";
475 	}
476 
477 	start_count = arch_timer_read_counter();
478 	clocksource_register_hz(&clocksource_counter, arch_timer_rate);
479 	cyclecounter.mult = clocksource_counter.mult;
480 	cyclecounter.shift = clocksource_counter.shift;
481 	timecounter_init(&timecounter, &cyclecounter, start_count);
482 
483 	/* 56 bits minimum, so we assume worst case rollover */
484 	sched_clock_register(arch_timer_read_counter, 56, arch_timer_rate);
485 }
486 
487 static void arch_timer_stop(struct clock_event_device *clk)
488 {
489 	pr_debug("arch_timer_teardown disable IRQ%d cpu #%d\n",
490 		 clk->irq, smp_processor_id());
491 
492 	if (arch_timer_use_virtual)
493 		disable_percpu_irq(arch_timer_ppi[VIRT_PPI]);
494 	else {
495 		disable_percpu_irq(arch_timer_ppi[PHYS_SECURE_PPI]);
496 		if (arch_timer_ppi[PHYS_NONSECURE_PPI])
497 			disable_percpu_irq(arch_timer_ppi[PHYS_NONSECURE_PPI]);
498 	}
499 
500 	clk->set_mode(CLOCK_EVT_MODE_UNUSED, clk);
501 }
502 
503 static int arch_timer_cpu_notify(struct notifier_block *self,
504 					   unsigned long action, void *hcpu)
505 {
506 	/*
507 	 * Grab cpu pointer in each case to avoid spurious
508 	 * preemptible warnings
509 	 */
510 	switch (action & ~CPU_TASKS_FROZEN) {
511 	case CPU_STARTING:
512 		arch_timer_setup(this_cpu_ptr(arch_timer_evt));
513 		break;
514 	case CPU_DYING:
515 		arch_timer_stop(this_cpu_ptr(arch_timer_evt));
516 		break;
517 	}
518 
519 	return NOTIFY_OK;
520 }
521 
522 static struct notifier_block arch_timer_cpu_nb = {
523 	.notifier_call = arch_timer_cpu_notify,
524 };
525 
526 #ifdef CONFIG_CPU_PM
527 static unsigned int saved_cntkctl;
528 static int arch_timer_cpu_pm_notify(struct notifier_block *self,
529 				    unsigned long action, void *hcpu)
530 {
531 	if (action == CPU_PM_ENTER)
532 		saved_cntkctl = arch_timer_get_cntkctl();
533 	else if (action == CPU_PM_ENTER_FAILED || action == CPU_PM_EXIT)
534 		arch_timer_set_cntkctl(saved_cntkctl);
535 	return NOTIFY_OK;
536 }
537 
538 static struct notifier_block arch_timer_cpu_pm_notifier = {
539 	.notifier_call = arch_timer_cpu_pm_notify,
540 };
541 
542 static int __init arch_timer_cpu_pm_init(void)
543 {
544 	return cpu_pm_register_notifier(&arch_timer_cpu_pm_notifier);
545 }
546 #else
547 static int __init arch_timer_cpu_pm_init(void)
548 {
549 	return 0;
550 }
551 #endif
552 
553 static int __init arch_timer_register(void)
554 {
555 	int err;
556 	int ppi;
557 
558 	arch_timer_evt = alloc_percpu(struct clock_event_device);
559 	if (!arch_timer_evt) {
560 		err = -ENOMEM;
561 		goto out;
562 	}
563 
564 	if (arch_timer_use_virtual) {
565 		ppi = arch_timer_ppi[VIRT_PPI];
566 		err = request_percpu_irq(ppi, arch_timer_handler_virt,
567 					 "arch_timer", arch_timer_evt);
568 	} else {
569 		ppi = arch_timer_ppi[PHYS_SECURE_PPI];
570 		err = request_percpu_irq(ppi, arch_timer_handler_phys,
571 					 "arch_timer", arch_timer_evt);
572 		if (!err && arch_timer_ppi[PHYS_NONSECURE_PPI]) {
573 			ppi = arch_timer_ppi[PHYS_NONSECURE_PPI];
574 			err = request_percpu_irq(ppi, arch_timer_handler_phys,
575 						 "arch_timer", arch_timer_evt);
576 			if (err)
577 				free_percpu_irq(arch_timer_ppi[PHYS_SECURE_PPI],
578 						arch_timer_evt);
579 		}
580 	}
581 
582 	if (err) {
583 		pr_err("arch_timer: can't register interrupt %d (%d)\n",
584 		       ppi, err);
585 		goto out_free;
586 	}
587 
588 	err = register_cpu_notifier(&arch_timer_cpu_nb);
589 	if (err)
590 		goto out_free_irq;
591 
592 	err = arch_timer_cpu_pm_init();
593 	if (err)
594 		goto out_unreg_notify;
595 
596 	/* Immediately configure the timer on the boot CPU */
597 	arch_timer_setup(this_cpu_ptr(arch_timer_evt));
598 
599 	return 0;
600 
601 out_unreg_notify:
602 	unregister_cpu_notifier(&arch_timer_cpu_nb);
603 out_free_irq:
604 	if (arch_timer_use_virtual)
605 		free_percpu_irq(arch_timer_ppi[VIRT_PPI], arch_timer_evt);
606 	else {
607 		free_percpu_irq(arch_timer_ppi[PHYS_SECURE_PPI],
608 				arch_timer_evt);
609 		if (arch_timer_ppi[PHYS_NONSECURE_PPI])
610 			free_percpu_irq(arch_timer_ppi[PHYS_NONSECURE_PPI],
611 					arch_timer_evt);
612 	}
613 
614 out_free:
615 	free_percpu(arch_timer_evt);
616 out:
617 	return err;
618 }
619 
620 static int __init arch_timer_mem_register(void __iomem *base, unsigned int irq)
621 {
622 	int ret;
623 	irq_handler_t func;
624 	struct arch_timer *t;
625 
626 	t = kzalloc(sizeof(*t), GFP_KERNEL);
627 	if (!t)
628 		return -ENOMEM;
629 
630 	t->base = base;
631 	t->evt.irq = irq;
632 	__arch_timer_setup(ARCH_MEM_TIMER, &t->evt);
633 
634 	if (arch_timer_mem_use_virtual)
635 		func = arch_timer_handler_virt_mem;
636 	else
637 		func = arch_timer_handler_phys_mem;
638 
639 	ret = request_irq(irq, func, IRQF_TIMER, "arch_mem_timer", &t->evt);
640 	if (ret) {
641 		pr_err("arch_timer: Failed to request mem timer irq\n");
642 		kfree(t);
643 	}
644 
645 	return ret;
646 }
647 
648 static const struct of_device_id arch_timer_of_match[] __initconst = {
649 	{ .compatible   = "arm,armv7-timer",    },
650 	{ .compatible   = "arm,armv8-timer",    },
651 	{},
652 };
653 
654 static const struct of_device_id arch_timer_mem_of_match[] __initconst = {
655 	{ .compatible   = "arm,armv7-timer-mem", },
656 	{},
657 };
658 
659 static bool __init
660 arch_timer_probed(int type, const struct of_device_id *matches)
661 {
662 	struct device_node *dn;
663 	bool probed = true;
664 
665 	dn = of_find_matching_node(NULL, matches);
666 	if (dn && of_device_is_available(dn) && !(arch_timers_present & type))
667 		probed = false;
668 	of_node_put(dn);
669 
670 	return probed;
671 }
672 
673 static void __init arch_timer_common_init(void)
674 {
675 	unsigned mask = ARCH_CP15_TIMER | ARCH_MEM_TIMER;
676 
677 	/* Wait until both nodes are probed if we have two timers */
678 	if ((arch_timers_present & mask) != mask) {
679 		if (!arch_timer_probed(ARCH_MEM_TIMER, arch_timer_mem_of_match))
680 			return;
681 		if (!arch_timer_probed(ARCH_CP15_TIMER, arch_timer_of_match))
682 			return;
683 	}
684 
685 	arch_timer_banner(arch_timers_present);
686 	arch_counter_register(arch_timers_present);
687 	arch_timer_arch_init();
688 }
689 
690 static void __init arch_timer_init(struct device_node *np)
691 {
692 	int i;
693 
694 	if (arch_timers_present & ARCH_CP15_TIMER) {
695 		pr_warn("arch_timer: multiple nodes in dt, skipping\n");
696 		return;
697 	}
698 
699 	arch_timers_present |= ARCH_CP15_TIMER;
700 	for (i = PHYS_SECURE_PPI; i < MAX_TIMER_PPI; i++)
701 		arch_timer_ppi[i] = irq_of_parse_and_map(np, i);
702 	arch_timer_detect_rate(NULL, np);
703 
704 	/*
705 	 * If HYP mode is available, we know that the physical timer
706 	 * has been configured to be accessible from PL1. Use it, so
707 	 * that a guest can use the virtual timer instead.
708 	 *
709 	 * If no interrupt provided for virtual timer, we'll have to
710 	 * stick to the physical timer. It'd better be accessible...
711 	 */
712 	if (is_hyp_mode_available() || !arch_timer_ppi[VIRT_PPI]) {
713 		arch_timer_use_virtual = false;
714 
715 		if (!arch_timer_ppi[PHYS_SECURE_PPI] ||
716 		    !arch_timer_ppi[PHYS_NONSECURE_PPI]) {
717 			pr_warn("arch_timer: No interrupt available, giving up\n");
718 			return;
719 		}
720 	}
721 
722 	arch_timer_c3stop = !of_property_read_bool(np, "always-on");
723 
724 	arch_timer_register();
725 	arch_timer_common_init();
726 }
727 CLOCKSOURCE_OF_DECLARE(armv7_arch_timer, "arm,armv7-timer", arch_timer_init);
728 CLOCKSOURCE_OF_DECLARE(armv8_arch_timer, "arm,armv8-timer", arch_timer_init);
729 
730 static void __init arch_timer_mem_init(struct device_node *np)
731 {
732 	struct device_node *frame, *best_frame = NULL;
733 	void __iomem *cntctlbase, *base;
734 	unsigned int irq;
735 	u32 cnttidr;
736 
737 	arch_timers_present |= ARCH_MEM_TIMER;
738 	cntctlbase = of_iomap(np, 0);
739 	if (!cntctlbase) {
740 		pr_err("arch_timer: Can't find CNTCTLBase\n");
741 		return;
742 	}
743 
744 	cnttidr = readl_relaxed(cntctlbase + CNTTIDR);
745 	iounmap(cntctlbase);
746 
747 	/*
748 	 * Try to find a virtual capable frame. Otherwise fall back to a
749 	 * physical capable frame.
750 	 */
751 	for_each_available_child_of_node(np, frame) {
752 		int n;
753 
754 		if (of_property_read_u32(frame, "frame-number", &n)) {
755 			pr_err("arch_timer: Missing frame-number\n");
756 			of_node_put(best_frame);
757 			of_node_put(frame);
758 			return;
759 		}
760 
761 		if (cnttidr & CNTTIDR_VIRT(n)) {
762 			of_node_put(best_frame);
763 			best_frame = frame;
764 			arch_timer_mem_use_virtual = true;
765 			break;
766 		}
767 		of_node_put(best_frame);
768 		best_frame = of_node_get(frame);
769 	}
770 
771 	base = arch_counter_base = of_iomap(best_frame, 0);
772 	if (!base) {
773 		pr_err("arch_timer: Can't map frame's registers\n");
774 		of_node_put(best_frame);
775 		return;
776 	}
777 
778 	if (arch_timer_mem_use_virtual)
779 		irq = irq_of_parse_and_map(best_frame, 1);
780 	else
781 		irq = irq_of_parse_and_map(best_frame, 0);
782 	of_node_put(best_frame);
783 	if (!irq) {
784 		pr_err("arch_timer: Frame missing %s irq",
785 		       arch_timer_mem_use_virtual ? "virt" : "phys");
786 		return;
787 	}
788 
789 	arch_timer_detect_rate(base, np);
790 	arch_timer_mem_register(base, irq);
791 	arch_timer_common_init();
792 }
793 CLOCKSOURCE_OF_DECLARE(armv7_arch_timer_mem, "arm,armv7-timer-mem",
794 		       arch_timer_mem_init);
795