Something went wrong. Try again.
Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241// SPDX-License-Identifier: GPL-2.0-only/* * Kernel module for testing static keys. * * Copyright 2015 Akamai Technologies Inc. All Rights Reserved * * Authors: * Jason Baron <jbaron@akamai.com> */
#include <linux/module.h>#include <linux/jump_label.h>
/* old keys */struct static_key old_true_key = STATIC_KEY_INIT_TRUE;struct static_key old_false_key = STATIC_KEY_INIT_FALSE;
/* new api */DEFINE_STATIC_KEY_TRUE(true_key);DEFINE_STATIC_KEY_FALSE(false_key);
/* external */extern struct static_key base_old_true_key;extern struct static_key base_inv_old_true_key;extern struct static_key base_old_false_key;extern struct static_key base_inv_old_false_key;
/* new api */extern struct static_key_true base_true_key;extern struct static_key_true base_inv_true_key;extern struct static_key_false base_false_key;extern struct static_key_false base_inv_false_key;
struct test_key { bool init_state; struct static_key *key; bool (*test_key)(void);};
#define test_key_func(key, branch) \static bool key ## _ ## branch(void) \{ \ return branch(&key); \}
static void invert_key(struct static_key *key){ if (static_key_enabled(key)) static_key_disable(key); else static_key_enable(key);}
static void invert_keys(struct test_key *keys, int size){ struct static_key *previous = NULL; int i;
for (i = 0; i < size; i++) { if (previous != keys[i].key) { invert_key(keys[i].key); previous = keys[i].key; } }}
static int verify_keys(struct test_key *keys, int size, bool invert){ int i; bool ret, init;
for (i = 0; i < size; i++) { ret = static_key_enabled(keys[i].key); init = keys[i].init_state; if (ret != (invert ? !init : init)) return -EINVAL; ret = keys[i].test_key(); if (static_key_enabled(keys[i].key)) { if (!ret) return -EINVAL; } else { if (ret) return -EINVAL; } } return 0;}
test_key_func(old_true_key, static_key_true)test_key_func(old_false_key, static_key_false)test_key_func(true_key, static_branch_likely)test_key_func(true_key, static_branch_unlikely)test_key_func(false_key, static_branch_likely)test_key_func(false_key, static_branch_unlikely)test_key_func(base_old_true_key, static_key_true)test_key_func(base_inv_old_true_key, static_key_true)test_key_func(base_old_false_key, static_key_false)test_key_func(base_inv_old_false_key, static_key_false)test_key_func(base_true_key, static_branch_likely)test_key_func(base_true_key, static_branch_unlikely)test_key_func(base_inv_true_key, static_branch_likely)test_key_func(base_inv_true_key, static_branch_unlikely)test_key_func(base_false_key, static_branch_likely)test_key_func(base_false_key, static_branch_unlikely)test_key_func(base_inv_false_key, static_branch_likely)test_key_func(base_inv_false_key, static_branch_unlikely)
static int __init test_static_key_init(void){ int ret; int size;
struct test_key static_key_tests[] = { /* internal keys - old keys */ { .init_state = true, .key = &old_true_key, .test_key = &old_true_key_static_key_true, }, { .init_state = false, .key = &old_false_key, .test_key = &old_false_key_static_key_false, }, /* internal keys - new keys */ { .init_state = true, .key = &true_key.key, .test_key = &true_key_static_branch_likely, }, { .init_state = true, .key = &true_key.key, .test_key = &true_key_static_branch_unlikely, }, { .init_state = false, .key = &false_key.key, .test_key = &false_key_static_branch_likely, }, { .init_state = false, .key = &false_key.key, .test_key = &false_key_static_branch_unlikely, }, /* external keys - old keys */ { .init_state = true, .key = &base_old_true_key, .test_key = &base_old_true_key_static_key_true, }, { .init_state = false, .key = &base_inv_old_true_key, .test_key = &base_inv_old_true_key_static_key_true, }, { .init_state = false, .key = &base_old_false_key, .test_key = &base_old_false_key_static_key_false, }, { .init_state = true, .key = &base_inv_old_false_key, .test_key = &base_inv_old_false_key_static_key_false, }, /* external keys - new keys */ { .init_state = true, .key = &base_true_key.key, .test_key = &base_true_key_static_branch_likely, }, { .init_state = true, .key = &base_true_key.key, .test_key = &base_true_key_static_branch_unlikely, }, { .init_state = false, .key = &base_inv_true_key.key, .test_key = &base_inv_true_key_static_branch_likely, }, { .init_state = false, .key = &base_inv_true_key.key, .test_key = &base_inv_true_key_static_branch_unlikely, }, { .init_state = false, .key = &base_false_key.key, .test_key = &base_false_key_static_branch_likely, }, { .init_state = false, .key = &base_false_key.key, .test_key = &base_false_key_static_branch_unlikely, }, { .init_state = true, .key = &base_inv_false_key.key, .test_key = &base_inv_false_key_static_branch_likely, }, { .init_state = true, .key = &base_inv_false_key.key, .test_key = &base_inv_false_key_static_branch_unlikely, }, };
size = ARRAY_SIZE(static_key_tests);
ret = verify_keys(static_key_tests, size, false); if (ret) goto out;
invert_keys(static_key_tests, size); ret = verify_keys(static_key_tests, size, true); if (ret) goto out;
invert_keys(static_key_tests, size); ret = verify_keys(static_key_tests, size, false); if (ret) goto out; return 0;out: return ret;}
static void __exit test_static_key_exit(void){}
module_init(test_static_key_init);module_exit(test_static_key_exit);
MODULE_AUTHOR("Jason Baron <jbaron@akamai.com>");MODULE_DESCRIPTION("Kernel module for testing static keys");MODULE_LICENSE("GPL");