/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _LINUX_JUMP_LABEL_RATELIMIT_H
#define _LINUX_JUMP_LABEL_RATELIMIT_H

#include <linux/jump_label.h>
#include <linux/workqueue.h>

#if defined(CONFIG_JUMP_LABEL)
struct static_key_deferred {
        struct static_key key;
        unsigned long timeout;
        struct delayed_work work;
};

struct static_key_true_deferred {
        struct static_key_true key;
        unsigned long timeout;
        struct delayed_work work;
};

struct static_key_false_deferred {
        struct static_key_false key;
        unsigned long timeout;
        struct delayed_work work;
};

#define static_key_slow_dec_deferred(x)                                 \
        __static_key_slow_dec_deferred(&(x)->key, &(x)->work, (x)->timeout)
#define static_branch_slow_dec_deferred(x)                              \
        __static_key_slow_dec_deferred(&(x)->key.key, &(x)->work, (x)->timeout)

#define static_key_deferred_flush(x)                                    \
        __static_key_deferred_flush((x), &(x)->work)

extern void
__static_key_slow_dec_deferred(struct static_key *key,
                               struct delayed_work *work,
                               unsigned long timeout);
extern void __static_key_deferred_flush(void *key, struct delayed_work *work);
extern void
jump_label_rate_limit(struct static_key_deferred *key, unsigned long rl);

extern void jump_label_update_timeout(struct work_struct *work);

#define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)                       \
        struct static_key_true_deferred name = {                        \
                .key =          { STATIC_KEY_INIT_TRUE },               \
                .timeout =      (rl),                                   \
                .work = __DELAYED_WORK_INITIALIZER((name).work,         \
                                                   jump_label_update_timeout, \
                                                   0),                  \
        }

#define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)                      \
        struct static_key_false_deferred name = {                       \
                .key =          { STATIC_KEY_INIT_FALSE },              \
                .timeout =      (rl),                                   \
                .work = __DELAYED_WORK_INITIALIZER((name).work,         \
                                                   jump_label_update_timeout, \
                                                   0),                  \
        }

#else   /* !CONFIG_JUMP_LABEL */
struct static_key_deferred {
        struct static_key  key;
};
struct static_key_true_deferred {
        struct static_key_true key;
};
struct static_key_false_deferred {
        struct static_key_false key;
};
#define DEFINE_STATIC_KEY_DEFERRED_TRUE(name, rl)       \
        struct static_key_true_deferred name = { STATIC_KEY_TRUE_INIT }
#define DEFINE_STATIC_KEY_DEFERRED_FALSE(name, rl)      \
        struct static_key_false_deferred name = { STATIC_KEY_FALSE_INIT }

#define static_branch_slow_dec_deferred(x)      static_branch_dec(&(x)->key)

static inline void static_key_slow_dec_deferred(struct static_key_deferred *key)
{
        STATIC_KEY_CHECK_USE(key);
        static_key_slow_dec(&key->key);
}
static inline void static_key_deferred_flush(void *key)
{
        STATIC_KEY_CHECK_USE(key);
}
static inline void
jump_label_rate_limit(struct static_key_deferred *key,
                unsigned long rl)
{
        STATIC_KEY_CHECK_USE(key);
}
#endif  /* CONFIG_JUMP_LABEL */

#define static_branch_deferred_inc(x)   static_branch_inc(&(x)->key)

#endif  /* _LINUX_JUMP_LABEL_RATELIMIT_H */