^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0-only
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) * Kernel module for testing static keys.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) * Copyright 2015 Akamai Technologies Inc. All Rights Reserved
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) * Authors:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) * Jason Baron <jbaron@akamai.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #include <linux/module.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) #include <linux/jump_label.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) /* old keys */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) struct static_key base_old_true_key = STATIC_KEY_INIT_TRUE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) EXPORT_SYMBOL_GPL(base_old_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) struct static_key base_inv_old_true_key = STATIC_KEY_INIT_TRUE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) EXPORT_SYMBOL_GPL(base_inv_old_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) struct static_key base_old_false_key = STATIC_KEY_INIT_FALSE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) EXPORT_SYMBOL_GPL(base_old_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) struct static_key base_inv_old_false_key = STATIC_KEY_INIT_FALSE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) EXPORT_SYMBOL_GPL(base_inv_old_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) /* new keys */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) DEFINE_STATIC_KEY_TRUE(base_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) EXPORT_SYMBOL_GPL(base_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) DEFINE_STATIC_KEY_TRUE(base_inv_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) EXPORT_SYMBOL_GPL(base_inv_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) DEFINE_STATIC_KEY_FALSE(base_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) EXPORT_SYMBOL_GPL(base_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) DEFINE_STATIC_KEY_FALSE(base_inv_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) EXPORT_SYMBOL_GPL(base_inv_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) static void invert_key(struct static_key *key)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) if (static_key_enabled(key))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) static_key_disable(key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) static_key_enable(key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) static int __init test_static_key_base_init(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) invert_key(&base_inv_old_true_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) invert_key(&base_inv_old_false_key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) invert_key(&base_inv_true_key.key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) invert_key(&base_inv_false_key.key);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) static void __exit test_static_key_base_exit(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) module_init(test_static_key_base_init);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) module_exit(test_static_key_base_exit);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) MODULE_AUTHOR("Jason Baron <jbaron@akamai.com>");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) MODULE_LICENSE("GPL");