162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Generic MMIO clocksource support
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci#include <linux/clocksource.h>
662306a36Sopenharmony_ci#include <linux/errno.h>
762306a36Sopenharmony_ci#include <linux/init.h>
862306a36Sopenharmony_ci#include <linux/slab.h>
962306a36Sopenharmony_ci
1062306a36Sopenharmony_cistruct clocksource_mmio {
1162306a36Sopenharmony_ci	void __iomem *reg;
1262306a36Sopenharmony_ci	struct clocksource clksrc;
1362306a36Sopenharmony_ci};
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_cistatic inline struct clocksource_mmio *to_mmio_clksrc(struct clocksource *c)
1662306a36Sopenharmony_ci{
1762306a36Sopenharmony_ci	return container_of(c, struct clocksource_mmio, clksrc);
1862306a36Sopenharmony_ci}
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ciu64 clocksource_mmio_readl_up(struct clocksource *c)
2162306a36Sopenharmony_ci{
2262306a36Sopenharmony_ci	return (u64)readl_relaxed(to_mmio_clksrc(c)->reg);
2362306a36Sopenharmony_ci}
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ciu64 clocksource_mmio_readl_down(struct clocksource *c)
2662306a36Sopenharmony_ci{
2762306a36Sopenharmony_ci	return ~(u64)readl_relaxed(to_mmio_clksrc(c)->reg) & c->mask;
2862306a36Sopenharmony_ci}
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ciu64 clocksource_mmio_readw_up(struct clocksource *c)
3162306a36Sopenharmony_ci{
3262306a36Sopenharmony_ci	return (u64)readw_relaxed(to_mmio_clksrc(c)->reg);
3362306a36Sopenharmony_ci}
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ciu64 clocksource_mmio_readw_down(struct clocksource *c)
3662306a36Sopenharmony_ci{
3762306a36Sopenharmony_ci	return ~(u64)readw_relaxed(to_mmio_clksrc(c)->reg) & c->mask;
3862306a36Sopenharmony_ci}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci/**
4162306a36Sopenharmony_ci * clocksource_mmio_init - Initialize a simple mmio based clocksource
4262306a36Sopenharmony_ci * @base:	Virtual address of the clock readout register
4362306a36Sopenharmony_ci * @name:	Name of the clocksource
4462306a36Sopenharmony_ci * @hz:		Frequency of the clocksource in Hz
4562306a36Sopenharmony_ci * @rating:	Rating of the clocksource
4662306a36Sopenharmony_ci * @bits:	Number of valid bits
4762306a36Sopenharmony_ci * @read:	One of clocksource_mmio_read*() above
4862306a36Sopenharmony_ci */
4962306a36Sopenharmony_ciint __init clocksource_mmio_init(void __iomem *base, const char *name,
5062306a36Sopenharmony_ci	unsigned long hz, int rating, unsigned bits,
5162306a36Sopenharmony_ci	u64 (*read)(struct clocksource *))
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	struct clocksource_mmio *cs;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	if (bits > 64 || bits < 16)
5662306a36Sopenharmony_ci		return -EINVAL;
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	cs = kzalloc(sizeof(struct clocksource_mmio), GFP_KERNEL);
5962306a36Sopenharmony_ci	if (!cs)
6062306a36Sopenharmony_ci		return -ENOMEM;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	cs->reg = base;
6362306a36Sopenharmony_ci	cs->clksrc.name = name;
6462306a36Sopenharmony_ci	cs->clksrc.rating = rating;
6562306a36Sopenharmony_ci	cs->clksrc.read = read;
6662306a36Sopenharmony_ci	cs->clksrc.mask = CLOCKSOURCE_MASK(bits);
6762306a36Sopenharmony_ci	cs->clksrc.flags = CLOCK_SOURCE_IS_CONTINUOUS;
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	return clocksource_register_hz(&cs->clksrc, hz);
7062306a36Sopenharmony_ci}
71