162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2017, Linaro Ltd.  All rights reserved.
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Author: Daniel Lezcano <daniel.lezcano@linaro.org>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci#include <linux/clk.h>
862306a36Sopenharmony_ci#include <linux/interrupt.h>
962306a36Sopenharmony_ci#include <linux/of.h>
1062306a36Sopenharmony_ci#include <linux/of_address.h>
1162306a36Sopenharmony_ci#include <linux/of_irq.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#include "timer-of.h"
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci/**
1762306a36Sopenharmony_ci * timer_of_irq_exit - Release the interrupt
1862306a36Sopenharmony_ci * @of_irq: an of_timer_irq structure pointer
1962306a36Sopenharmony_ci *
2062306a36Sopenharmony_ci * Free the irq resource
2162306a36Sopenharmony_ci */
2262306a36Sopenharmony_cistatic __init void timer_of_irq_exit(struct of_timer_irq *of_irq)
2362306a36Sopenharmony_ci{
2462306a36Sopenharmony_ci	struct timer_of *to = container_of(of_irq, struct timer_of, of_irq);
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci	struct clock_event_device *clkevt = &to->clkevt;
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci	if (of_irq->percpu)
2962306a36Sopenharmony_ci		free_percpu_irq(of_irq->irq, clkevt);
3062306a36Sopenharmony_ci	else
3162306a36Sopenharmony_ci		free_irq(of_irq->irq, clkevt);
3262306a36Sopenharmony_ci}
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci/**
3562306a36Sopenharmony_ci * timer_of_irq_init - Request the interrupt
3662306a36Sopenharmony_ci * @np: a device tree node pointer
3762306a36Sopenharmony_ci * @of_irq: an of_timer_irq structure pointer
3862306a36Sopenharmony_ci *
3962306a36Sopenharmony_ci * Get the interrupt number from the DT from its definition and
4062306a36Sopenharmony_ci * request it. The interrupt is gotten by falling back the following way:
4162306a36Sopenharmony_ci *
4262306a36Sopenharmony_ci * - Get interrupt number by name
4362306a36Sopenharmony_ci * - Get interrupt number by index
4462306a36Sopenharmony_ci *
4562306a36Sopenharmony_ci * When the interrupt is per CPU, 'request_percpu_irq()' is called,
4662306a36Sopenharmony_ci * otherwise 'request_irq()' is used.
4762306a36Sopenharmony_ci *
4862306a36Sopenharmony_ci * Returns 0 on success, < 0 otherwise
4962306a36Sopenharmony_ci */
5062306a36Sopenharmony_cistatic __init int timer_of_irq_init(struct device_node *np,
5162306a36Sopenharmony_ci				    struct of_timer_irq *of_irq)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	int ret;
5462306a36Sopenharmony_ci	struct timer_of *to = container_of(of_irq, struct timer_of, of_irq);
5562306a36Sopenharmony_ci	struct clock_event_device *clkevt = &to->clkevt;
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	if (of_irq->name) {
5862306a36Sopenharmony_ci		of_irq->irq = ret = of_irq_get_byname(np, of_irq->name);
5962306a36Sopenharmony_ci		if (ret < 0) {
6062306a36Sopenharmony_ci			pr_err("Failed to get interrupt %s for %pOF\n",
6162306a36Sopenharmony_ci			       of_irq->name, np);
6262306a36Sopenharmony_ci			return ret;
6362306a36Sopenharmony_ci		}
6462306a36Sopenharmony_ci	} else	{
6562306a36Sopenharmony_ci		of_irq->irq = irq_of_parse_and_map(np, of_irq->index);
6662306a36Sopenharmony_ci	}
6762306a36Sopenharmony_ci	if (!of_irq->irq) {
6862306a36Sopenharmony_ci		pr_err("Failed to map interrupt for %pOF\n", np);
6962306a36Sopenharmony_ci		return -EINVAL;
7062306a36Sopenharmony_ci	}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci	ret = of_irq->percpu ?
7362306a36Sopenharmony_ci		request_percpu_irq(of_irq->irq, of_irq->handler,
7462306a36Sopenharmony_ci				   np->full_name, clkevt) :
7562306a36Sopenharmony_ci		request_irq(of_irq->irq, of_irq->handler,
7662306a36Sopenharmony_ci			    of_irq->flags ? of_irq->flags : IRQF_TIMER,
7762306a36Sopenharmony_ci			    np->full_name, clkevt);
7862306a36Sopenharmony_ci	if (ret) {
7962306a36Sopenharmony_ci		pr_err("Failed to request irq %d for %pOF\n", of_irq->irq, np);
8062306a36Sopenharmony_ci		return ret;
8162306a36Sopenharmony_ci	}
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	clkevt->irq = of_irq->irq;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	return 0;
8662306a36Sopenharmony_ci}
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci/**
8962306a36Sopenharmony_ci * timer_of_clk_exit - Release the clock resources
9062306a36Sopenharmony_ci * @of_clk: a of_timer_clk structure pointer
9162306a36Sopenharmony_ci *
9262306a36Sopenharmony_ci * Disables and releases the refcount on the clk
9362306a36Sopenharmony_ci */
9462306a36Sopenharmony_cistatic __init void timer_of_clk_exit(struct of_timer_clk *of_clk)
9562306a36Sopenharmony_ci{
9662306a36Sopenharmony_ci	of_clk->rate = 0;
9762306a36Sopenharmony_ci	clk_disable_unprepare(of_clk->clk);
9862306a36Sopenharmony_ci	clk_put(of_clk->clk);
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci/**
10262306a36Sopenharmony_ci * timer_of_clk_init - Initialize the clock resources
10362306a36Sopenharmony_ci * @np: a device tree node pointer
10462306a36Sopenharmony_ci * @of_clk: a of_timer_clk structure pointer
10562306a36Sopenharmony_ci *
10662306a36Sopenharmony_ci * Get the clock by name or by index, enable it and get the rate
10762306a36Sopenharmony_ci *
10862306a36Sopenharmony_ci * Returns 0 on success, < 0 otherwise
10962306a36Sopenharmony_ci */
11062306a36Sopenharmony_cistatic __init int timer_of_clk_init(struct device_node *np,
11162306a36Sopenharmony_ci				    struct of_timer_clk *of_clk)
11262306a36Sopenharmony_ci{
11362306a36Sopenharmony_ci	int ret;
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	of_clk->clk = of_clk->name ? of_clk_get_by_name(np, of_clk->name) :
11662306a36Sopenharmony_ci		of_clk_get(np, of_clk->index);
11762306a36Sopenharmony_ci	if (IS_ERR(of_clk->clk)) {
11862306a36Sopenharmony_ci		ret = PTR_ERR(of_clk->clk);
11962306a36Sopenharmony_ci		if (ret != -EPROBE_DEFER)
12062306a36Sopenharmony_ci			pr_err("Failed to get clock for %pOF\n", np);
12162306a36Sopenharmony_ci		goto out;
12262306a36Sopenharmony_ci	}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	ret = clk_prepare_enable(of_clk->clk);
12562306a36Sopenharmony_ci	if (ret) {
12662306a36Sopenharmony_ci		pr_err("Failed for enable clock for %pOF\n", np);
12762306a36Sopenharmony_ci		goto out_clk_put;
12862306a36Sopenharmony_ci	}
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	of_clk->rate = clk_get_rate(of_clk->clk);
13162306a36Sopenharmony_ci	if (!of_clk->rate) {
13262306a36Sopenharmony_ci		ret = -EINVAL;
13362306a36Sopenharmony_ci		pr_err("Failed to get clock rate for %pOF\n", np);
13462306a36Sopenharmony_ci		goto out_clk_disable;
13562306a36Sopenharmony_ci	}
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ci	of_clk->period = DIV_ROUND_UP(of_clk->rate, HZ);
13862306a36Sopenharmony_ciout:
13962306a36Sopenharmony_ci	return ret;
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ciout_clk_disable:
14262306a36Sopenharmony_ci	clk_disable_unprepare(of_clk->clk);
14362306a36Sopenharmony_ciout_clk_put:
14462306a36Sopenharmony_ci	clk_put(of_clk->clk);
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	goto out;
14762306a36Sopenharmony_ci}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_cistatic __init void timer_of_base_exit(struct of_timer_base *of_base)
15062306a36Sopenharmony_ci{
15162306a36Sopenharmony_ci	iounmap(of_base->base);
15262306a36Sopenharmony_ci}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_cistatic __init int timer_of_base_init(struct device_node *np,
15562306a36Sopenharmony_ci				     struct of_timer_base *of_base)
15662306a36Sopenharmony_ci{
15762306a36Sopenharmony_ci	of_base->base = of_base->name ?
15862306a36Sopenharmony_ci		of_io_request_and_map(np, of_base->index, of_base->name) :
15962306a36Sopenharmony_ci		of_iomap(np, of_base->index);
16062306a36Sopenharmony_ci	if (IS_ERR_OR_NULL(of_base->base)) {
16162306a36Sopenharmony_ci		pr_err("Failed to iomap (%s:%s)\n", np->name, of_base->name);
16262306a36Sopenharmony_ci		return of_base->base ? PTR_ERR(of_base->base) : -ENOMEM;
16362306a36Sopenharmony_ci	}
16462306a36Sopenharmony_ci
16562306a36Sopenharmony_ci	return 0;
16662306a36Sopenharmony_ci}
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ciint __init timer_of_init(struct device_node *np, struct timer_of *to)
16962306a36Sopenharmony_ci{
17062306a36Sopenharmony_ci	int ret = -EINVAL;
17162306a36Sopenharmony_ci	int flags = 0;
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_ci	if (to->flags & TIMER_OF_BASE) {
17462306a36Sopenharmony_ci		ret = timer_of_base_init(np, &to->of_base);
17562306a36Sopenharmony_ci		if (ret)
17662306a36Sopenharmony_ci			goto out_fail;
17762306a36Sopenharmony_ci		flags |= TIMER_OF_BASE;
17862306a36Sopenharmony_ci	}
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci	if (to->flags & TIMER_OF_CLOCK) {
18162306a36Sopenharmony_ci		ret = timer_of_clk_init(np, &to->of_clk);
18262306a36Sopenharmony_ci		if (ret)
18362306a36Sopenharmony_ci			goto out_fail;
18462306a36Sopenharmony_ci		flags |= TIMER_OF_CLOCK;
18562306a36Sopenharmony_ci	}
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci	if (to->flags & TIMER_OF_IRQ) {
18862306a36Sopenharmony_ci		ret = timer_of_irq_init(np, &to->of_irq);
18962306a36Sopenharmony_ci		if (ret)
19062306a36Sopenharmony_ci			goto out_fail;
19162306a36Sopenharmony_ci		flags |= TIMER_OF_IRQ;
19262306a36Sopenharmony_ci	}
19362306a36Sopenharmony_ci
19462306a36Sopenharmony_ci	if (!to->clkevt.name)
19562306a36Sopenharmony_ci		to->clkevt.name = np->full_name;
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_ci	to->np = np;
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_ci	return ret;
20062306a36Sopenharmony_ci
20162306a36Sopenharmony_ciout_fail:
20262306a36Sopenharmony_ci	if (flags & TIMER_OF_IRQ)
20362306a36Sopenharmony_ci		timer_of_irq_exit(&to->of_irq);
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci	if (flags & TIMER_OF_CLOCK)
20662306a36Sopenharmony_ci		timer_of_clk_exit(&to->of_clk);
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci	if (flags & TIMER_OF_BASE)
20962306a36Sopenharmony_ci		timer_of_base_exit(&to->of_base);
21062306a36Sopenharmony_ci	return ret;
21162306a36Sopenharmony_ci}
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci/**
21462306a36Sopenharmony_ci * timer_of_cleanup - release timer_of resources
21562306a36Sopenharmony_ci * @to: timer_of structure
21662306a36Sopenharmony_ci *
21762306a36Sopenharmony_ci * Release the resources that has been used in timer_of_init().
21862306a36Sopenharmony_ci * This function should be called in init error cases
21962306a36Sopenharmony_ci */
22062306a36Sopenharmony_civoid __init timer_of_cleanup(struct timer_of *to)
22162306a36Sopenharmony_ci{
22262306a36Sopenharmony_ci	if (to->flags & TIMER_OF_IRQ)
22362306a36Sopenharmony_ci		timer_of_irq_exit(&to->of_irq);
22462306a36Sopenharmony_ci
22562306a36Sopenharmony_ci	if (to->flags & TIMER_OF_CLOCK)
22662306a36Sopenharmony_ci		timer_of_clk_exit(&to->of_clk);
22762306a36Sopenharmony_ci
22862306a36Sopenharmony_ci	if (to->flags & TIMER_OF_BASE)
22962306a36Sopenharmony_ci		timer_of_base_exit(&to->of_base);
23062306a36Sopenharmony_ci}
231