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