867 lines · 21.1 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0-only/* * jump label support * * Copyright (C) 2009 Jason Baron <jbaron@redhat.com> * Copyright (C) 2011 Peter Zijlstra * */#include <linux/memory.h>#include <linux/uaccess.h>#include <linux/module.h>#include <linux/list.h>#include <linux/slab.h>#include <linux/sort.h>#include <linux/err.h>#include <linux/static_key.h>#include <linux/jump_label_ratelimit.h>#include <linux/bug.h>#include <linux/cpu.h>#include <asm/sections.h> /* mutex to protect coming/going of the jump_label table */static DEFINE_MUTEX(jump_label_mutex); void jump_label_lock(void){	mutex_lock(&jump_label_mutex);} void jump_label_unlock(void){	mutex_unlock(&jump_label_mutex);} static int jump_label_cmp(const void *a, const void *b){	const struct jump_entry *jea = a;	const struct jump_entry *jeb = b; 	/*	 * Entrires are sorted by key.	 */	if (jump_entry_key(jea) < jump_entry_key(jeb))		return -1; 	if (jump_entry_key(jea) > jump_entry_key(jeb))		return 1; 	/*	 * In the batching mode, entries should also be sorted by the code	 * inside the already sorted list of entries, enabling a bsearch in	 * the vector.	 */	if (jump_entry_code(jea) < jump_entry_code(jeb))		return -1; 	if (jump_entry_code(jea) > jump_entry_code(jeb))		return 1; 	return 0;} static void jump_label_swap(void *a, void *b, int size){	long delta = (unsigned long)a - (unsigned long)b;	struct jump_entry *jea = a;	struct jump_entry *jeb = b;	struct jump_entry tmp = *jea; 	jea->code	= jeb->code - delta;	jea->target	= jeb->target - delta;	jea->key	= jeb->key - delta; 	jeb->code	= tmp.code + delta;	jeb->target	= tmp.target + delta;	jeb->key	= tmp.key + delta;} static voidjump_label_sort_entries(struct jump_entry *start, struct jump_entry *stop){	unsigned long size;	void *swapfn = NULL; 	if (IS_ENABLED(CONFIG_HAVE_ARCH_JUMP_LABEL_RELATIVE))		swapfn = jump_label_swap; 	size = (((unsigned long)stop - (unsigned long)start)					/ sizeof(struct jump_entry));	sort(start, size, sizeof(struct jump_entry), jump_label_cmp, swapfn);} static void jump_label_update(struct static_key *key); /* * There are similar definitions for the !CONFIG_JUMP_LABEL case in jump_label.h. * The use of 'atomic_read()' requires atomic.h and its problematic for some * kernel headers such as kernel.h and others. Since static_key_count() is not * used in the branch statements as it is for the !CONFIG_JUMP_LABEL case its ok * to have it be a function here. Similarly, for 'static_key_enable()' and * 'static_key_disable()', which require bug.h. This should allow jump_label.h * to be included from most/all places for CONFIG_JUMP_LABEL. */int static_key_count(struct static_key *key){	/*	 * -1 means the first static_key_slow_inc() is in progress.	 *  static_key_enabled() must return true, so return 1 here.	 */	int n = atomic_read(&key->enabled); 	return n >= 0 ? n : 1;}EXPORT_SYMBOL_GPL(static_key_count); /* * static_key_fast_inc_not_disabled - adds a user for a static key * @key: static key that must be already enabled * * The caller must make sure that the static key can't get disabled while * in this function. It doesn't patch jump labels, only adds a user to * an already enabled static key. * * Returns true if the increment was done. Unlike refcount_t the ref counter * is not saturated, but will fail to increment on overflow. */bool static_key_fast_inc_not_disabled(struct static_key *key){	int v; 	STATIC_KEY_CHECK_USE(key);	/*	 * Negative key->enabled has a special meaning: it sends	 * static_key_slow_inc() down the slow path, and it is non-zero	 * so it counts as "enabled" in jump_label_update().  Note that	 * atomic_inc_unless_negative() checks >= 0, so roll our own.	 */	v = atomic_read(&key->enabled);	do {		if (v <= 0 || (v + 1) < 0)			return false;	} while (!likely(atomic_try_cmpxchg(&key->enabled, &v, v + 1))); 	return true;}EXPORT_SYMBOL_GPL(static_key_fast_inc_not_disabled); bool static_key_slow_inc_cpuslocked(struct static_key *key){	lockdep_assert_cpus_held(); 	/*	 * Careful if we get concurrent static_key_slow_inc() calls;	 * later calls must wait for the first one to _finish_ the	 * jump_label_update() process.  At the same time, however,	 * the jump_label_update() call below wants to see	 * static_key_enabled(&key) for jumps to be updated properly.	 */	if (static_key_fast_inc_not_disabled(key))		return true; 	jump_label_lock();	if (atomic_read(&key->enabled) == 0) {		atomic_set(&key->enabled, -1);		jump_label_update(key);		/*		 * Ensure that if the above cmpxchg loop observes our positive		 * value, it must also observe all the text changes.		 */		atomic_set_release(&key->enabled, 1);	} else {		if (WARN_ON_ONCE(!static_key_fast_inc_not_disabled(key))) {			jump_label_unlock();			return false;		}	}	jump_label_unlock();	return true;} bool static_key_slow_inc(struct static_key *key){	bool ret; 	cpus_read_lock();	ret = static_key_slow_inc_cpuslocked(key);	cpus_read_unlock();	return ret;}EXPORT_SYMBOL_GPL(static_key_slow_inc); void static_key_enable_cpuslocked(struct static_key *key){	STATIC_KEY_CHECK_USE(key);	lockdep_assert_cpus_held(); 	if (atomic_read(&key->enabled) > 0) {		WARN_ON_ONCE(atomic_read(&key->enabled) != 1);		return;	} 	jump_label_lock();	if (atomic_read(&key->enabled) == 0) {		atomic_set(&key->enabled, -1);		jump_label_update(key);		/*		 * See static_key_slow_inc().		 */		atomic_set_release(&key->enabled, 1);	}	jump_label_unlock();}EXPORT_SYMBOL_GPL(static_key_enable_cpuslocked); void static_key_enable(struct static_key *key){	cpus_read_lock();	static_key_enable_cpuslocked(key);	cpus_read_unlock();}EXPORT_SYMBOL_GPL(static_key_enable); void static_key_disable_cpuslocked(struct static_key *key){	STATIC_KEY_CHECK_USE(key);	lockdep_assert_cpus_held(); 	if (atomic_read(&key->enabled) != 1) {		WARN_ON_ONCE(atomic_read(&key->enabled) != 0);		return;	} 	jump_label_lock();	if (atomic_cmpxchg(&key->enabled, 1, 0))		jump_label_update(key);	jump_label_unlock();}EXPORT_SYMBOL_GPL(static_key_disable_cpuslocked); void static_key_disable(struct static_key *key){	cpus_read_lock();	static_key_disable_cpuslocked(key);	cpus_read_unlock();}EXPORT_SYMBOL_GPL(static_key_disable); static bool static_key_slow_try_dec(struct static_key *key){	int val; 	val = atomic_fetch_add_unless(&key->enabled, -1, 1);	if (val == 1)		return false; 	/*	 * The negative count check is valid even when a negative	 * key->enabled is in use by static_key_slow_inc(); a	 * __static_key_slow_dec() before the first static_key_slow_inc()	 * returns is unbalanced, because all other static_key_slow_inc()	 * instances block while the update is in progress.	 */	WARN(val < 0, "jump label: negative count!\n");	return true;} static void __static_key_slow_dec_cpuslocked(struct static_key *key){	lockdep_assert_cpus_held(); 	if (static_key_slow_try_dec(key))		return; 	jump_label_lock();	if (atomic_dec_and_test(&key->enabled))		jump_label_update(key);	jump_label_unlock();} static void __static_key_slow_dec(struct static_key *key){	cpus_read_lock();	__static_key_slow_dec_cpuslocked(key);	cpus_read_unlock();} void jump_label_update_timeout(struct work_struct *work){	struct static_key_deferred *key =		container_of(work, struct static_key_deferred, work.work);	__static_key_slow_dec(&key->key);}EXPORT_SYMBOL_GPL(jump_label_update_timeout); void static_key_slow_dec(struct static_key *key){	STATIC_KEY_CHECK_USE(key);	__static_key_slow_dec(key);}EXPORT_SYMBOL_GPL(static_key_slow_dec); void static_key_slow_dec_cpuslocked(struct static_key *key){	STATIC_KEY_CHECK_USE(key);	__static_key_slow_dec_cpuslocked(key);} void __static_key_slow_dec_deferred(struct static_key *key,				    struct delayed_work *work,				    unsigned long timeout){	STATIC_KEY_CHECK_USE(key); 	if (static_key_slow_try_dec(key))		return; 	schedule_delayed_work(work, timeout);}EXPORT_SYMBOL_GPL(__static_key_slow_dec_deferred); void __static_key_deferred_flush(void *key, struct delayed_work *work){	STATIC_KEY_CHECK_USE(key);	flush_delayed_work(work);}EXPORT_SYMBOL_GPL(__static_key_deferred_flush); void jump_label_rate_limit(struct static_key_deferred *key,		unsigned long rl){	STATIC_KEY_CHECK_USE(key);	key->timeout = rl;	INIT_DELAYED_WORK(&key->work, jump_label_update_timeout);}EXPORT_SYMBOL_GPL(jump_label_rate_limit); static int addr_conflict(struct jump_entry *entry, void *start, void *end){	if (jump_entry_code(entry) <= (unsigned long)end &&	    jump_entry_code(entry) + jump_entry_size(entry) > (unsigned long)start)		return 1; 	return 0;} static int __jump_label_text_reserved(struct jump_entry *iter_start,		struct jump_entry *iter_stop, void *start, void *end, bool init){	struct jump_entry *iter; 	iter = iter_start;	while (iter < iter_stop) {		if (init || !jump_entry_is_init(iter)) {			if (addr_conflict(iter, start, end))				return 1;		}		iter++;	} 	return 0;} #ifndef arch_jump_label_transform_staticstatic void arch_jump_label_transform_static(struct jump_entry *entry,					     enum jump_label_type type){	/* nothing to do on most architectures */}#endif static inline struct jump_entry *static_key_entries(struct static_key *key){	WARN_ON_ONCE(key->type & JUMP_TYPE_LINKED);	return (struct jump_entry *)(key->type & ~JUMP_TYPE_MASK);} static inline bool static_key_type(struct static_key *key){	return key->type & JUMP_TYPE_TRUE;} static inline bool static_key_linked(struct static_key *key){	return key->type & JUMP_TYPE_LINKED;} static inline void static_key_clear_linked(struct static_key *key){	key->type &= ~JUMP_TYPE_LINKED;} static inline void static_key_set_linked(struct static_key *key){	key->type |= JUMP_TYPE_LINKED;} /*** * A 'struct static_key' uses a union such that it either points directly * to a table of 'struct jump_entry' or to a linked list of modules which in * turn point to 'struct jump_entry' tables. * * The two lower bits of the pointer are used to keep track of which pointer * type is in use and to store the initial branch direction, we use an access * function which preserves these bits. */static void static_key_set_entries(struct static_key *key,				   struct jump_entry *entries){	unsigned long type; 	WARN_ON_ONCE((unsigned long)entries & JUMP_TYPE_MASK);	type = key->type & JUMP_TYPE_MASK;	key->entries = entries;	key->type |= type;} static enum jump_label_type jump_label_type(struct jump_entry *entry){	struct static_key *key = jump_entry_key(entry);	bool enabled = static_key_enabled(key);	bool branch = jump_entry_is_branch(entry); 	/* See the comment in linux/jump_label.h */	return enabled ^ branch;} static bool jump_label_can_update(struct jump_entry *entry, bool init){	/*	 * Cannot update code that was in an init text area.	 */	if (!init && jump_entry_is_init(entry))		return false; 	if (!kernel_text_address(jump_entry_code(entry))) {		/*		 * This skips patching built-in __exit, which		 * is part of init_section_contains() but is		 * not part of kernel_text_address().		 *		 * Skipping built-in __exit is fine since it		 * will never be executed.		 */		WARN_ONCE(!jump_entry_is_init(entry),			  "can't patch jump_label at %pS",			  (void *)jump_entry_code(entry));		return false;	} 	return true;} #ifndef HAVE_JUMP_LABEL_BATCHstatic void __jump_label_update(struct static_key *key,				struct jump_entry *entry,				struct jump_entry *stop,				bool init){	for (; (entry < stop) && (jump_entry_key(entry) == key); entry++) {		if (jump_label_can_update(entry, init))			arch_jump_label_transform(entry, jump_label_type(entry));	}}#elsestatic void __jump_label_update(struct static_key *key,				struct jump_entry *entry,				struct jump_entry *stop,				bool init){	for (; (entry < stop) && (jump_entry_key(entry) == key); entry++) { 		if (!jump_label_can_update(entry, init))			continue; 		if (!arch_jump_label_transform_queue(entry, jump_label_type(entry))) {			/*			 * Queue is full: Apply the current queue and try again.			 */			arch_jump_label_transform_apply();			BUG_ON(!arch_jump_label_transform_queue(entry, jump_label_type(entry)));		}	}	arch_jump_label_transform_apply();}#endif void __init jump_label_init(void){	struct jump_entry *iter_start = __start___jump_table;	struct jump_entry *iter_stop = __stop___jump_table;	struct static_key *key = NULL;	struct jump_entry *iter; 	/*	 * Since we are initializing the static_key.enabled field with	 * with the 'raw' int values (to avoid pulling in atomic.h) in	 * jump_label.h, let's make sure that is safe. There are only two	 * cases to check since we initialize to 0 or 1.	 */	BUILD_BUG_ON((int)ATOMIC_INIT(0) != 0);

Lines 1–500 of 867.