1b2441318SGreg Kroah-Hartman /* SPDX-License-Identifier: GPL-2.0 */
2f3109a51SJan Kara /*
3f3109a51SJan Kara * Floating proportions with flexible aging period
4f3109a51SJan Kara *
5f3109a51SJan Kara * Copyright (C) 2011, SUSE, Jan Kara <jack@suse.cz>
6f3109a51SJan Kara */
7f3109a51SJan Kara
8f3109a51SJan Kara #ifndef _LINUX_FLEX_PROPORTIONS_H
9f3109a51SJan Kara #define _LINUX_FLEX_PROPORTIONS_H
10f3109a51SJan Kara
11f3109a51SJan Kara #include <linux/percpu_counter.h>
12f3109a51SJan Kara #include <linux/spinlock.h>
13f3109a51SJan Kara #include <linux/seqlock.h>
1420ae0079STejun Heo #include <linux/gfp.h>
15f3109a51SJan Kara
16f3109a51SJan Kara /*
17f3109a51SJan Kara * When maximum proportion of some event type is specified, this is the
18f3109a51SJan Kara * precision with which we allow limitting. Note that this creates an upper
19f3109a51SJan Kara * bound on the number of events per period like
20f3109a51SJan Kara * ULLONG_MAX >> FPROP_FRAC_SHIFT.
21f3109a51SJan Kara */
22f3109a51SJan Kara #define FPROP_FRAC_SHIFT 10
23f3109a51SJan Kara #define FPROP_FRAC_BASE (1UL << FPROP_FRAC_SHIFT)
24f3109a51SJan Kara
25f3109a51SJan Kara /*
26f3109a51SJan Kara * ---- Global proportion definitions ----
27f3109a51SJan Kara */
28f3109a51SJan Kara struct fprop_global {
29f3109a51SJan Kara /* Number of events in the current period */
30f3109a51SJan Kara struct percpu_counter events;
31f3109a51SJan Kara /* Current period */
32f3109a51SJan Kara unsigned int period;
33f3109a51SJan Kara /* Synchronization with period transitions */
34f3109a51SJan Kara seqcount_t sequence;
35f3109a51SJan Kara };
36f3109a51SJan Kara
3720ae0079STejun Heo int fprop_global_init(struct fprop_global *p, gfp_t gfp);
38f3109a51SJan Kara void fprop_global_destroy(struct fprop_global *p);
39f3109a51SJan Kara bool fprop_new_period(struct fprop_global *p, int periods);
40f3109a51SJan Kara
41f3109a51SJan Kara /*
42f3109a51SJan Kara * ---- PERCPU ----
43f3109a51SJan Kara */
44f3109a51SJan Kara struct fprop_local_percpu {
45f3109a51SJan Kara /* the local events counter */
46f3109a51SJan Kara struct percpu_counter events;
47f3109a51SJan Kara /* Period in which we last updated events */
48f3109a51SJan Kara unsigned int period;
49f3109a51SJan Kara raw_spinlock_t lock; /* Protect period and numerator */
50f3109a51SJan Kara };
51f3109a51SJan Kara
5220ae0079STejun Heo int fprop_local_init_percpu(struct fprop_local_percpu *pl, gfp_t gfp);
53f3109a51SJan Kara void fprop_local_destroy_percpu(struct fprop_local_percpu *pl);
54*be5f1797SMatthew Wilcox (Oracle) void __fprop_add_percpu(struct fprop_global *p, struct fprop_local_percpu *pl,
55*be5f1797SMatthew Wilcox (Oracle) long nr);
56*be5f1797SMatthew Wilcox (Oracle) void __fprop_add_percpu_max(struct fprop_global *p,
57*be5f1797SMatthew Wilcox (Oracle) struct fprop_local_percpu *pl, int max_frac, long nr);
58f3109a51SJan Kara void fprop_fraction_percpu(struct fprop_global *p,
59f3109a51SJan Kara struct fprop_local_percpu *pl, unsigned long *numerator,
60f3109a51SJan Kara unsigned long *denominator);
61f3109a51SJan Kara
62f3109a51SJan Kara static inline
fprop_inc_percpu(struct fprop_global * p,struct fprop_local_percpu * pl)63f3109a51SJan Kara void fprop_inc_percpu(struct fprop_global *p, struct fprop_local_percpu *pl)
64f3109a51SJan Kara {
65f3109a51SJan Kara unsigned long flags;
66f3109a51SJan Kara
67f3109a51SJan Kara local_irq_save(flags);
68*be5f1797SMatthew Wilcox (Oracle) __fprop_add_percpu(p, pl, 1);
69f3109a51SJan Kara local_irq_restore(flags);
70f3109a51SJan Kara }
71f3109a51SJan Kara
72f3109a51SJan Kara #endif
73