162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci
362306a36Sopenharmony_ci#include <linux/interrupt.h>
462306a36Sopenharmony_ci#include <linux/ioport.h>
562306a36Sopenharmony_ci#include <linux/clocksource.h>
662306a36Sopenharmony_ci#include <linux/clockchips.h>
762306a36Sopenharmony_ci#include <linux/module.h>
862306a36Sopenharmony_ci#include <linux/slab.h>
962306a36Sopenharmony_ci#include <linux/goldfish.h>
1062306a36Sopenharmony_ci#include <clocksource/timer-goldfish.h>
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistruct goldfish_timer {
1362306a36Sopenharmony_ci	struct clocksource cs;
1462306a36Sopenharmony_ci	struct clock_event_device ced;
1562306a36Sopenharmony_ci	struct resource res;
1662306a36Sopenharmony_ci	void __iomem *base;
1762306a36Sopenharmony_ci};
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_cistatic struct goldfish_timer *ced_to_gf(struct clock_event_device *ced)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	return container_of(ced, struct goldfish_timer, ced);
2262306a36Sopenharmony_ci}
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic struct goldfish_timer *cs_to_gf(struct clocksource *cs)
2562306a36Sopenharmony_ci{
2662306a36Sopenharmony_ci	return container_of(cs, struct goldfish_timer, cs);
2762306a36Sopenharmony_ci}
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_cistatic u64 goldfish_timer_read(struct clocksource *cs)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	struct goldfish_timer *timerdrv = cs_to_gf(cs);
3262306a36Sopenharmony_ci	void __iomem *base = timerdrv->base;
3362306a36Sopenharmony_ci	u32 time_low, time_high;
3462306a36Sopenharmony_ci	u64 ticks;
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	/*
3762306a36Sopenharmony_ci	 * time_low: get low bits of current time and update time_high
3862306a36Sopenharmony_ci	 * time_high: get high bits of time at last time_low read
3962306a36Sopenharmony_ci	 */
4062306a36Sopenharmony_ci	time_low = gf_ioread32(base + TIMER_TIME_LOW);
4162306a36Sopenharmony_ci	time_high = gf_ioread32(base + TIMER_TIME_HIGH);
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	ticks = ((u64)time_high << 32) | time_low;
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	return ticks;
4662306a36Sopenharmony_ci}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_cistatic int goldfish_timer_set_oneshot(struct clock_event_device *evt)
4962306a36Sopenharmony_ci{
5062306a36Sopenharmony_ci	struct goldfish_timer *timerdrv = ced_to_gf(evt);
5162306a36Sopenharmony_ci	void __iomem *base = timerdrv->base;
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_ci	gf_iowrite32(0, base + TIMER_ALARM_HIGH);
5462306a36Sopenharmony_ci	gf_iowrite32(0, base + TIMER_ALARM_LOW);
5562306a36Sopenharmony_ci	gf_iowrite32(1, base + TIMER_IRQ_ENABLED);
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	return 0;
5862306a36Sopenharmony_ci}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_cistatic int goldfish_timer_shutdown(struct clock_event_device *evt)
6162306a36Sopenharmony_ci{
6262306a36Sopenharmony_ci	struct goldfish_timer *timerdrv = ced_to_gf(evt);
6362306a36Sopenharmony_ci	void __iomem *base = timerdrv->base;
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci	gf_iowrite32(0, base + TIMER_IRQ_ENABLED);
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	return 0;
6862306a36Sopenharmony_ci}
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_cistatic int goldfish_timer_next_event(unsigned long delta,
7162306a36Sopenharmony_ci				     struct clock_event_device *evt)
7262306a36Sopenharmony_ci{
7362306a36Sopenharmony_ci	struct goldfish_timer *timerdrv = ced_to_gf(evt);
7462306a36Sopenharmony_ci	void __iomem *base = timerdrv->base;
7562306a36Sopenharmony_ci	u64 now;
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci	now = goldfish_timer_read(&timerdrv->cs);
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	now += delta;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	gf_iowrite32(upper_32_bits(now), base + TIMER_ALARM_HIGH);
8262306a36Sopenharmony_ci	gf_iowrite32(lower_32_bits(now), base + TIMER_ALARM_LOW);
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	return 0;
8562306a36Sopenharmony_ci}
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_cistatic irqreturn_t goldfish_timer_irq(int irq, void *dev_id)
8862306a36Sopenharmony_ci{
8962306a36Sopenharmony_ci	struct goldfish_timer *timerdrv = dev_id;
9062306a36Sopenharmony_ci	struct clock_event_device *evt = &timerdrv->ced;
9162306a36Sopenharmony_ci	void __iomem *base = timerdrv->base;
9262306a36Sopenharmony_ci
9362306a36Sopenharmony_ci	gf_iowrite32(1, base + TIMER_CLEAR_INTERRUPT);
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	evt->event_handler(evt);
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	return IRQ_HANDLED;
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ciint __init goldfish_timer_init(int irq, void __iomem *base)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	struct goldfish_timer *timerdrv;
10362306a36Sopenharmony_ci	int ret;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	timerdrv = kzalloc(sizeof(*timerdrv), GFP_KERNEL);
10662306a36Sopenharmony_ci	if (!timerdrv)
10762306a36Sopenharmony_ci		return -ENOMEM;
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	timerdrv->base = base;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	timerdrv->ced = (struct clock_event_device){
11262306a36Sopenharmony_ci		.name			= "goldfish_timer",
11362306a36Sopenharmony_ci		.features		= CLOCK_EVT_FEAT_ONESHOT,
11462306a36Sopenharmony_ci		.set_state_shutdown	= goldfish_timer_shutdown,
11562306a36Sopenharmony_ci		.set_state_oneshot      = goldfish_timer_set_oneshot,
11662306a36Sopenharmony_ci		.set_next_event		= goldfish_timer_next_event,
11762306a36Sopenharmony_ci	};
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	timerdrv->res = (struct resource){
12062306a36Sopenharmony_ci		.name  = "goldfish_timer",
12162306a36Sopenharmony_ci		.start = (unsigned long)base,
12262306a36Sopenharmony_ci		.end   = (unsigned long)base + 0xfff,
12362306a36Sopenharmony_ci	};
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	ret = request_resource(&iomem_resource, &timerdrv->res);
12662306a36Sopenharmony_ci	if (ret) {
12762306a36Sopenharmony_ci		pr_err("Cannot allocate '%s' resource\n", timerdrv->res.name);
12862306a36Sopenharmony_ci		return ret;
12962306a36Sopenharmony_ci	}
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	timerdrv->cs = (struct clocksource){
13262306a36Sopenharmony_ci		.name		= "goldfish_timer",
13362306a36Sopenharmony_ci		.rating		= 400,
13462306a36Sopenharmony_ci		.read		= goldfish_timer_read,
13562306a36Sopenharmony_ci		.mask		= CLOCKSOURCE_MASK(64),
13662306a36Sopenharmony_ci		.flags		= 0,
13762306a36Sopenharmony_ci		.max_idle_ns	= LONG_MAX,
13862306a36Sopenharmony_ci	};
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	clocksource_register_hz(&timerdrv->cs, NSEC_PER_SEC);
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci	ret = request_irq(irq, goldfish_timer_irq, IRQF_TIMER,
14362306a36Sopenharmony_ci			  "goldfish_timer", timerdrv);
14462306a36Sopenharmony_ci	if (ret) {
14562306a36Sopenharmony_ci		pr_err("Couldn't register goldfish-timer interrupt\n");
14662306a36Sopenharmony_ci		return ret;
14762306a36Sopenharmony_ci	}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	clockevents_config_and_register(&timerdrv->ced, NSEC_PER_SEC,
15062306a36Sopenharmony_ci					1, 0xffffffff);
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ci	return 0;
15362306a36Sopenharmony_ci}
154