162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2012 Altera Corporation
462306a36Sopenharmony_ci * Copyright (c) 2011 Picochip Ltd., Jamie Iles
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci * Modified from mach-picoxcell/time.c
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci#include <linux/delay.h>
962306a36Sopenharmony_ci#include <linux/dw_apb_timer.h>
1062306a36Sopenharmony_ci#include <linux/of.h>
1162306a36Sopenharmony_ci#include <linux/of_address.h>
1262306a36Sopenharmony_ci#include <linux/of_irq.h>
1362306a36Sopenharmony_ci#include <linux/clk.h>
1462306a36Sopenharmony_ci#include <linux/reset.h>
1562306a36Sopenharmony_ci#include <linux/sched_clock.h>
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistatic int __init timer_get_base_and_rate(struct device_node *np,
1862306a36Sopenharmony_ci				    void __iomem **base, u32 *rate)
1962306a36Sopenharmony_ci{
2062306a36Sopenharmony_ci	struct clk *timer_clk;
2162306a36Sopenharmony_ci	struct clk *pclk;
2262306a36Sopenharmony_ci	struct reset_control *rstc;
2362306a36Sopenharmony_ci	int ret;
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci	*base = of_iomap(np, 0);
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci	if (!*base)
2862306a36Sopenharmony_ci		panic("Unable to map regs for %pOFn", np);
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci	/*
3162306a36Sopenharmony_ci	 * Reset the timer if the reset control is available, wiping
3262306a36Sopenharmony_ci	 * out the state the firmware may have left it
3362306a36Sopenharmony_ci	 */
3462306a36Sopenharmony_ci	rstc = of_reset_control_get(np, NULL);
3562306a36Sopenharmony_ci	if (!IS_ERR(rstc)) {
3662306a36Sopenharmony_ci		reset_control_assert(rstc);
3762306a36Sopenharmony_ci		reset_control_deassert(rstc);
3862306a36Sopenharmony_ci	}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	/*
4162306a36Sopenharmony_ci	 * Not all implementations use a peripheral clock, so don't panic
4262306a36Sopenharmony_ci	 * if it's not present
4362306a36Sopenharmony_ci	 */
4462306a36Sopenharmony_ci	pclk = of_clk_get_by_name(np, "pclk");
4562306a36Sopenharmony_ci	if (!IS_ERR(pclk))
4662306a36Sopenharmony_ci		if (clk_prepare_enable(pclk))
4762306a36Sopenharmony_ci			pr_warn("pclk for %pOFn is present, but could not be activated\n",
4862306a36Sopenharmony_ci				np);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	if (!of_property_read_u32(np, "clock-freq", rate) ||
5162306a36Sopenharmony_ci	    !of_property_read_u32(np, "clock-frequency", rate))
5262306a36Sopenharmony_ci		return 0;
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	timer_clk = of_clk_get_by_name(np, "timer");
5562306a36Sopenharmony_ci	if (IS_ERR(timer_clk)) {
5662306a36Sopenharmony_ci		ret = PTR_ERR(timer_clk);
5762306a36Sopenharmony_ci		goto out_pclk_disable;
5862306a36Sopenharmony_ci	}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci	ret = clk_prepare_enable(timer_clk);
6162306a36Sopenharmony_ci	if (ret)
6262306a36Sopenharmony_ci		goto out_timer_clk_put;
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	*rate = clk_get_rate(timer_clk);
6562306a36Sopenharmony_ci	if (!(*rate)) {
6662306a36Sopenharmony_ci		ret = -EINVAL;
6762306a36Sopenharmony_ci		goto out_timer_clk_disable;
6862306a36Sopenharmony_ci	}
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	return 0;
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ciout_timer_clk_disable:
7362306a36Sopenharmony_ci	clk_disable_unprepare(timer_clk);
7462306a36Sopenharmony_ciout_timer_clk_put:
7562306a36Sopenharmony_ci	clk_put(timer_clk);
7662306a36Sopenharmony_ciout_pclk_disable:
7762306a36Sopenharmony_ci	if (!IS_ERR(pclk)) {
7862306a36Sopenharmony_ci		clk_disable_unprepare(pclk);
7962306a36Sopenharmony_ci		clk_put(pclk);
8062306a36Sopenharmony_ci	}
8162306a36Sopenharmony_ci	iounmap(*base);
8262306a36Sopenharmony_ci	return ret;
8362306a36Sopenharmony_ci}
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_cistatic int __init add_clockevent(struct device_node *event_timer)
8662306a36Sopenharmony_ci{
8762306a36Sopenharmony_ci	void __iomem *iobase;
8862306a36Sopenharmony_ci	struct dw_apb_clock_event_device *ced;
8962306a36Sopenharmony_ci	u32 irq, rate;
9062306a36Sopenharmony_ci	int ret = 0;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	irq = irq_of_parse_and_map(event_timer, 0);
9362306a36Sopenharmony_ci	if (irq == 0)
9462306a36Sopenharmony_ci		panic("No IRQ for clock event timer");
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	ret = timer_get_base_and_rate(event_timer, &iobase, &rate);
9762306a36Sopenharmony_ci	if (ret)
9862306a36Sopenharmony_ci		return ret;
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	ced = dw_apb_clockevent_init(-1, event_timer->name, 300, iobase, irq,
10162306a36Sopenharmony_ci				     rate);
10262306a36Sopenharmony_ci	if (!ced)
10362306a36Sopenharmony_ci		return -EINVAL;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	dw_apb_clockevent_register(ced);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return 0;
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic void __iomem *sched_io_base;
11162306a36Sopenharmony_cistatic u32 sched_rate;
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic int __init add_clocksource(struct device_node *source_timer)
11462306a36Sopenharmony_ci{
11562306a36Sopenharmony_ci	void __iomem *iobase;
11662306a36Sopenharmony_ci	struct dw_apb_clocksource *cs;
11762306a36Sopenharmony_ci	u32 rate;
11862306a36Sopenharmony_ci	int ret;
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_ci	ret = timer_get_base_and_rate(source_timer, &iobase, &rate);
12162306a36Sopenharmony_ci	if (ret)
12262306a36Sopenharmony_ci		return ret;
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	cs = dw_apb_clocksource_init(300, source_timer->name, iobase, rate);
12562306a36Sopenharmony_ci	if (!cs)
12662306a36Sopenharmony_ci		return -EINVAL;
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_ci	dw_apb_clocksource_start(cs);
12962306a36Sopenharmony_ci	dw_apb_clocksource_register(cs);
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	/*
13262306a36Sopenharmony_ci	 * Fallback to use the clocksource as sched_clock if no separate
13362306a36Sopenharmony_ci	 * timer is found. sched_io_base then points to the current_value
13462306a36Sopenharmony_ci	 * register of the clocksource timer.
13562306a36Sopenharmony_ci	 */
13662306a36Sopenharmony_ci	sched_io_base = iobase + 0x04;
13762306a36Sopenharmony_ci	sched_rate = rate;
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	return 0;
14062306a36Sopenharmony_ci}
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_cistatic u64 notrace read_sched_clock(void)
14362306a36Sopenharmony_ci{
14462306a36Sopenharmony_ci	return ~readl_relaxed(sched_io_base);
14562306a36Sopenharmony_ci}
14662306a36Sopenharmony_ci
14762306a36Sopenharmony_cistatic const struct of_device_id sptimer_ids[] __initconst = {
14862306a36Sopenharmony_ci	{ .compatible = "picochip,pc3x2-rtc" },
14962306a36Sopenharmony_ci	{ /* Sentinel */ },
15062306a36Sopenharmony_ci};
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_cistatic void __init init_sched_clock(void)
15362306a36Sopenharmony_ci{
15462306a36Sopenharmony_ci	struct device_node *sched_timer;
15562306a36Sopenharmony_ci
15662306a36Sopenharmony_ci	sched_timer = of_find_matching_node(NULL, sptimer_ids);
15762306a36Sopenharmony_ci	if (sched_timer) {
15862306a36Sopenharmony_ci		timer_get_base_and_rate(sched_timer, &sched_io_base,
15962306a36Sopenharmony_ci					&sched_rate);
16062306a36Sopenharmony_ci		of_node_put(sched_timer);
16162306a36Sopenharmony_ci	}
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	sched_clock_register(read_sched_clock, 32, sched_rate);
16462306a36Sopenharmony_ci}
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci#ifdef CONFIG_ARM
16762306a36Sopenharmony_cistatic unsigned long dw_apb_delay_timer_read(void)
16862306a36Sopenharmony_ci{
16962306a36Sopenharmony_ci	return ~readl_relaxed(sched_io_base);
17062306a36Sopenharmony_ci}
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_cistatic struct delay_timer dw_apb_delay_timer = {
17362306a36Sopenharmony_ci	.read_current_timer	= dw_apb_delay_timer_read,
17462306a36Sopenharmony_ci};
17562306a36Sopenharmony_ci#endif
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_cistatic int num_called;
17862306a36Sopenharmony_cistatic int __init dw_apb_timer_init(struct device_node *timer)
17962306a36Sopenharmony_ci{
18062306a36Sopenharmony_ci	int ret = 0;
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	switch (num_called) {
18362306a36Sopenharmony_ci	case 1:
18462306a36Sopenharmony_ci		pr_debug("%s: found clocksource timer\n", __func__);
18562306a36Sopenharmony_ci		ret = add_clocksource(timer);
18662306a36Sopenharmony_ci		if (ret)
18762306a36Sopenharmony_ci			return ret;
18862306a36Sopenharmony_ci		init_sched_clock();
18962306a36Sopenharmony_ci#ifdef CONFIG_ARM
19062306a36Sopenharmony_ci		dw_apb_delay_timer.freq = sched_rate;
19162306a36Sopenharmony_ci		register_current_timer_delay(&dw_apb_delay_timer);
19262306a36Sopenharmony_ci#endif
19362306a36Sopenharmony_ci		break;
19462306a36Sopenharmony_ci	default:
19562306a36Sopenharmony_ci		pr_debug("%s: found clockevent timer\n", __func__);
19662306a36Sopenharmony_ci		ret = add_clockevent(timer);
19762306a36Sopenharmony_ci		if (ret)
19862306a36Sopenharmony_ci			return ret;
19962306a36Sopenharmony_ci		break;
20062306a36Sopenharmony_ci	}
20162306a36Sopenharmony_ci
20262306a36Sopenharmony_ci	num_called++;
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_ci	return 0;
20562306a36Sopenharmony_ci}
20662306a36Sopenharmony_ciTIMER_OF_DECLARE(pc3x2_timer, "picochip,pc3x2-timer", dw_apb_timer_init);
20762306a36Sopenharmony_ciTIMER_OF_DECLARE(apb_timer_osc, "snps,dw-apb-timer-osc", dw_apb_timer_init);
20862306a36Sopenharmony_ciTIMER_OF_DECLARE(apb_timer_sp, "snps,dw-apb-timer-sp", dw_apb_timer_init);
20962306a36Sopenharmony_ciTIMER_OF_DECLARE(apb_timer, "snps,dw-apb-timer", dw_apb_timer_init);
210