cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

jump_label_ratelimit.h (2841B)

      1/* SPDX-License-Identifier: GPL-2.0 */
      2#ifndef _LINUX_JUMP_LABEL_RATELIMIT_H
      3#define _LINUX_JUMP_LABEL_RATELIMIT_H
      4
      5#include <linux/jump_label.h>
      6#include <linux/workqueue.h>
      7
      8#if defined(CONFIG_JUMP_LABEL)
      9struct static_key_deferred {
     10	struct static_key key;
     11	unsigned long timeout;
     12	struct delayed_work work;
     13};
     14
     15struct static_key_true_deferred {
     16	struct static_key_true key;
     17	unsigned long timeout;
     18	struct delayed_work work;
     19};
     20
     21struct static_key_false_deferred {
     22	struct static_key_false key;
     23	unsigned long timeout;
     24	struct delayed_work work;
     25};
     26
     27#define static_key_slow_dec_deferred(x)					\
     28	__static_key_slow_dec_deferred(&(x)->key, &(x)->work, (x)->timeout)
     29#define static_branch_slow_dec_deferred(x)				\
     30	__static_key_slow_dec_deferred(&(x)->key.key, &(x)->work, (x)->timeout)
     31
     32#define static_key_deferred_flush(x)					\
     33	__static_key_deferred_flush((x), &(x)->work)
     34
     35extern void
     36__static_key_slow_dec_deferred(struct static_key *key,
     37			       struct delayed_work *work,
     38			       unsigned long timeout);
     39extern void __static_key_deferred_flush(void *key, struct delayed_work *work);
     40extern void
     41jump_label_rate_limit(struct static_key_deferred *key, unsigned long rl);
     42
     43extern void jump_label_update_timeout(struct work_struct *work);
     44
     45#define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)			\
     46	struct static_key_true_deferred name = {			\
     47		.key =		{ STATIC_KEY_INIT_TRUE },		\
     48		.timeout =	(rl),					\
     49		.work =	__DELAYED_WORK_INITIALIZER((name).work,		\
     50						   jump_label_update_timeout, \
     51						   0),			\
     52	}
     53
     54#define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)			\
     55	struct static_key_false_deferred name = {			\
     56		.key =		{ STATIC_KEY_INIT_FALSE },		\
     57		.timeout =	(rl),					\
     58		.work =	__DELAYED_WORK_INITIALIZER((name).work,		\
     59						   jump_label_update_timeout, \
     60						   0),			\
     61	}
     62
     63#else	/* !CONFIG_JUMP_LABEL */
     64struct static_key_deferred {
     65	struct static_key  key;
     66};
     67struct static_key_true_deferred {
     68	struct static_key_true key;
     69};
     70struct static_key_false_deferred {
     71	struct static_key_false key;
     72};
     73#define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)	\
     74	struct static_key_true_deferred name = { STATIC_KEY_TRUE_INIT }
     75#define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)	\
     76	struct static_key_false_deferred name = { STATIC_KEY_FALSE_INIT }
     77
     78#define static_branch_slow_dec_deferred(x)	static_branch_dec(&(x)->key)
     79
     80static inline void static_key_slow_dec_deferred(struct static_key_deferred *key)
     81{
     82	STATIC_KEY_CHECK_USE(key);
     83	static_key_slow_dec(&key->key);
     84}
     85static inline void static_key_deferred_flush(void *key)
     86{
     87	STATIC_KEY_CHECK_USE(key);
     88}
     89static inline void
     90jump_label_rate_limit(struct static_key_deferred *key,
     91		unsigned long rl)
     92{
     93	STATIC_KEY_CHECK_USE(key);
     94}
     95#endif	/* CONFIG_JUMP_LABEL */
     96
     97#define static_branch_deferred_inc(x)	static_branch_inc(&(x)->key)
     98
     99#endif	/* _LINUX_JUMP_LABEL_RATELIMIT_H */