Christian Brauner (Amutable)exec_state: relocate dumpable information6b1c66c5 months agoHistory
1738 lines · 47.6 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0-only/* Kernel thread helper functions. *   Copyright (C) 2004 IBM Corporation, Rusty Russell. *   Copyright (C) 2009 Red Hat, Inc. * * Creation is done via kthreadd, so that we get a clean environment * even if we're invoked from userspace (think modprobe, hotplug cpu, * etc.). */#include <uapi/linux/sched/types.h>#include <linux/mm.h>#include <linux/mmu_context.h>#include <linux/sched.h>#include <linux/sched/mm.h>#include <linux/sched/task.h>#include <linux/kthread.h>#include <linux/completion.h>#include <linux/err.h>#include <linux/cgroup.h>#include <linux/cpuset.h>#include <linux/unistd.h>#include <linux/file.h>#include <linux/export.h>#include <linux/mutex.h>#include <linux/slab.h>#include <linux/freezer.h>#include <linux/ptrace.h>#include <linux/uaccess.h>#include <linux/numa.h>#include <linux/sched/isolation.h>#include <trace/events/sched.h>  static DEFINE_SPINLOCK(kthread_create_lock);static LIST_HEAD(kthread_create_list);struct task_struct *kthreadd_task; static LIST_HEAD(kthread_affinity_list);static DEFINE_MUTEX(kthread_affinity_lock); struct kthread_create_info{	/* Information passed to kthread() from kthreadd. */	char *full_name;	int (*threadfn)(void *data);	void *data;	int node; 	/* Result passed back to kthread_create() from kthreadd. */	struct task_struct *result;	struct completion *done; 	struct list_head list;}; struct kthread {	unsigned long flags;	unsigned int cpu;	unsigned int node;	int started;	int result;	int (*threadfn)(void *);	void *data;	struct completion parked;	struct completion exited;#ifdef CONFIG_BLK_CGROUP	struct cgroup_subsys_state *blkcg_css;#endif	/* To store the full name if task comm is truncated. */	char *full_name;	struct task_struct *task;	struct list_head affinity_node;	struct cpumask *preferred_affinity;}; enum KTHREAD_BITS {	KTHREAD_IS_PER_CPU = 0,	KTHREAD_SHOULD_STOP,	KTHREAD_SHOULD_PARK,}; static inline struct kthread *to_kthread(struct task_struct *k){	WARN_ON(!(k->flags & PF_KTHREAD));	return k->worker_private;} void get_kthread_comm(char *buf, size_t buf_size, struct task_struct *tsk){	struct kthread *kthread = to_kthread(tsk); 	if (!kthread || !kthread->full_name) {		strscpy(buf, tsk->comm, buf_size);		return;	} 	strscpy_pad(buf, kthread->full_name, buf_size);} bool set_kthread_struct(struct task_struct *p){	struct kthread *kthread; 	if (WARN_ON_ONCE(to_kthread(p)))		return false; 	kthread = kzalloc_obj(*kthread);	if (!kthread)		return false; 	init_completion(&kthread->exited);	init_completion(&kthread->parked);	INIT_LIST_HEAD(&kthread->affinity_node);	p->vfork_done = &kthread->exited; 	kthread->task = p;	kthread->node = tsk_fork_get_node(current);	p->worker_private = kthread;	return true;} void free_kthread_struct(struct task_struct *k){	struct kthread *kthread; 	/*	 * Can be NULL if kmalloc() in set_kthread_struct() failed.	 */	kthread = to_kthread(k);	if (!kthread)		return; #ifdef CONFIG_BLK_CGROUP	WARN_ON_ONCE(kthread->blkcg_css);#endif	k->worker_private = NULL;	kfree(kthread->full_name);	kfree(kthread);} /** * kthread_should_stop - should this kthread return now? * * When someone calls kthread_stop() on your kthread, it will be woken * and this will return true.  You should then return, and your return * value will be passed through to kthread_stop(). */bool kthread_should_stop(void){	return test_bit(KTHREAD_SHOULD_STOP, &to_kthread(current)->flags);}EXPORT_SYMBOL(kthread_should_stop); static bool __kthread_should_park(struct task_struct *k){	return test_bit(KTHREAD_SHOULD_PARK, &to_kthread(k)->flags);} /** * kthread_should_park - should this kthread park now? * * When someone calls kthread_park() on your kthread, it will be woken * and this will return true.  You should then do the necessary * cleanup and call kthread_parkme() * * Similar to kthread_should_stop(), but this keeps the thread alive * and in a park position. kthread_unpark() "restarts" the thread and * calls the thread function again. */bool kthread_should_park(void){	return __kthread_should_park(current);}EXPORT_SYMBOL_GPL(kthread_should_park); bool kthread_should_stop_or_park(void){	struct kthread *kthread = tsk_is_kthread(current); 	if (!kthread)		return false; 	return kthread->flags & (BIT(KTHREAD_SHOULD_STOP) | BIT(KTHREAD_SHOULD_PARK));} /** * kthread_freezable_should_stop - should this freezable kthread return now? * @was_frozen: optional out parameter, indicates whether %current was frozen * * kthread_should_stop() for freezable kthreads, which will enter * refrigerator if necessary.  This function is safe from kthread_stop() / * freezer deadlock and freezable kthreads should use this function instead * of calling try_to_freeze() directly. */bool kthread_freezable_should_stop(bool *was_frozen){	bool frozen = false; 	might_sleep(); 	if (unlikely(freezing(current)))		frozen = __refrigerator(true); 	if (was_frozen)		*was_frozen = frozen; 	return kthread_should_stop();}EXPORT_SYMBOL_GPL(kthread_freezable_should_stop); /** * kthread_func - return the function specified on kthread creation * @task: kthread task in question * * Returns NULL if the task is not a kthread. */void *kthread_func(struct task_struct *task){	struct kthread *kthread = tsk_is_kthread(task);	if (kthread)		return kthread->threadfn;	return NULL;}EXPORT_SYMBOL_GPL(kthread_func); /** * kthread_data - return data value specified on kthread creation * @task: kthread task in question * * Return the data value specified when kthread @task was created. * The caller is responsible for ensuring the validity of @task when * calling this function. */void *kthread_data(struct task_struct *task){	return to_kthread(task)->data;}EXPORT_SYMBOL_GPL(kthread_data); /** * kthread_probe_data - speculative version of kthread_data() * @task: possible kthread task in question * * @task could be a kthread task.  Return the data value specified when it * was created if accessible.  If @task isn't a kthread task or its data is * inaccessible for any reason, %NULL is returned.  This function requires * that @task itself is safe to dereference. */void *kthread_probe_data(struct task_struct *task){	struct kthread *kthread = tsk_is_kthread(task);	void *data = NULL; 	if (kthread)		copy_from_kernel_nofault(&data, &kthread->data, sizeof(data));	return data;} static void __kthread_parkme(struct kthread *self){	for (;;) {		/*		 * TASK_PARKED is a special state; we must serialize against		 * possible pending wakeups to avoid store-store collisions on		 * task->state.		 *		 * Such a collision might possibly result in the task state		 * changin from TASK_PARKED and us failing the		 * wait_task_inactive() in kthread_park().		 */		set_special_state(TASK_PARKED);		if (!test_bit(KTHREAD_SHOULD_PARK, &self->flags))			break; 		/*		 * Thread is going to call schedule(), do not preempt it,		 * or the caller of kthread_park() may spend more time in		 * wait_task_inactive().		 */		preempt_disable();		complete(&self->parked);		schedule_preempt_disabled();		preempt_enable();	}	__set_current_state(TASK_RUNNING);} void kthread_parkme(void){	__kthread_parkme(to_kthread(current));}EXPORT_SYMBOL_GPL(kthread_parkme); void kthread_do_exit(struct kthread *kthread, long result){	kthread->result = result;	if (!list_empty(&kthread->affinity_node)) {		mutex_lock(&kthread_affinity_lock);		list_del(&kthread->affinity_node);		mutex_unlock(&kthread_affinity_lock); 		if (kthread->preferred_affinity) {			kfree(kthread->preferred_affinity);			kthread->preferred_affinity = NULL;		}	}} /** * kthread_complete_and_exit - Exit the current kthread. * @comp: Completion to complete * @code: The integer value to return to kthread_stop(). * * If present, complete @comp and then return code to kthread_stop(). * * A kernel thread whose module may be removed after the completion of * @comp can use this function to exit safely. * * Does not return. */void __noreturn kthread_complete_and_exit(struct completion *comp, long code){	if (comp)		complete(comp); 	kthread_exit(code);}EXPORT_SYMBOL(kthread_complete_and_exit); static void kthread_fetch_affinity(struct kthread *kthread, struct cpumask *cpumask){	const struct cpumask *pref; 	guard(rcu)(); 	if (kthread->preferred_affinity) {		pref = kthread->preferred_affinity;	} else {		if (kthread->node == NUMA_NO_NODE)			pref = housekeeping_cpumask(HK_TYPE_DOMAIN);		else			pref = cpumask_of_node(kthread->node);	} 	cpumask_and(cpumask, pref, housekeeping_cpumask(HK_TYPE_DOMAIN));	if (cpumask_empty(cpumask))		cpumask_copy(cpumask, housekeeping_cpumask(HK_TYPE_DOMAIN));} static void kthread_affine_node(void){	struct kthread *kthread = to_kthread(current);	cpumask_var_t affinity; 	if (WARN_ON_ONCE(kthread_is_per_cpu(current)))		return; 	if (!zalloc_cpumask_var(&affinity, GFP_KERNEL)) {		WARN_ON_ONCE(1);		return;	} 	mutex_lock(&kthread_affinity_lock);	WARN_ON_ONCE(!list_empty(&kthread->affinity_node));	list_add_tail(&kthread->affinity_node, &kthread_affinity_list);	/*	 * The node cpumask is racy when read from kthread() but:	 * - a racing CPU going down will either fail on the subsequent	 *   call to set_cpus_allowed_ptr() or be migrated to housekeepers	 *   afterwards by the scheduler.	 * - a racing CPU going up will be handled by kthreads_online_cpu()	 */	kthread_fetch_affinity(kthread, affinity);	set_cpus_allowed_ptr(current, affinity);	mutex_unlock(&kthread_affinity_lock); 	free_cpumask_var(affinity);} static int kthread(void *_create){	static const struct sched_param param = { .sched_priority = 0 };	/* Copy data: it's on kthread's stack */	struct kthread_create_info *create = _create;	int (*threadfn)(void *data) = create->threadfn;	void *data = create->data;	struct completion *done;	struct kthread *self;	int ret; 	self = to_kthread(current); 	/* Release the structure when caller killed by a fatal signal. */	done = xchg(&create->done, NULL);	if (!done) {		kfree(create->full_name);		kfree(create);		kthread_exit(-EINTR);	} 	self->full_name = create->full_name;	self->threadfn = threadfn;	self->data = data; 	/*	 * The new thread inherited kthreadd's priority and CPU mask. Reset	 * back to default in case they have been changed.	 */	sched_setscheduler_nocheck(current, SCHED_NORMAL, &param); 	/* OK, tell user we're spawned, wait for stop or wakeup */	__set_current_state(TASK_UNINTERRUPTIBLE);	create->result = current;	/*	 * Thread is going to call schedule(), do not preempt it,	 * or the creator may spend more time in wait_task_inactive().	 */	preempt_disable();	complete(done);	schedule_preempt_disabled();	preempt_enable(); 	self->started = 1; 	/*	 * Apply default node affinity if no call to kthread_bind[_mask]() nor	 * kthread_affine_preferred() was issued before the first wake-up.	 */	if (!(current->flags & PF_NO_SETAFFINITY) && !self->preferred_affinity)		kthread_affine_node(); 	ret = -EINTR;	if (!test_bit(KTHREAD_SHOULD_STOP, &self->flags)) {		cgroup_kthread_ready();		__kthread_parkme(self);		ret = threadfn(data);	}	kthread_exit(ret);} /* called from kernel_clone() to get node information for about to be created task */int tsk_fork_get_node(struct task_struct *tsk){#ifdef CONFIG_NUMA	if (tsk == kthreadd_task)		return tsk->pref_node_fork;#endif	return NUMA_NO_NODE;} static void create_kthread(struct kthread_create_info *create){	int pid; #ifdef CONFIG_NUMA	current->pref_node_fork = create->node;#endif	/* We want our own signal handler (we take no signals by default). */	pid = kernel_thread(kthread, create, create->full_name,			    CLONE_FS | CLONE_FILES | SIGCHLD);	if (pid < 0) {		/* Release the structure when caller killed by a fatal signal. */		struct completion *done = xchg(&create->done, NULL); 		kfree(create->full_name);		if (!done) {			kfree(create);			return;		}		create->result = ERR_PTR(pid);		complete(done);	}} static __printf(4, 0)struct task_struct *__kthread_create_on_node(int (*threadfn)(void *data),						    void *data, int node,						    const char namefmt[],						    va_list args){	DECLARE_COMPLETION_ONSTACK(done);	struct task_struct *task;	struct kthread_create_info *create = kmalloc_obj(*create); 	if (!create)		return ERR_PTR(-ENOMEM);	create->threadfn = threadfn;	create->data = data;	create->node = node;	create->done = &done;	create->full_name = kvasprintf(GFP_KERNEL, namefmt, args);	if (!create->full_name) {		task = ERR_PTR(-ENOMEM);		goto free_create;	} 	spin_lock(&kthread_create_lock);	list_add_tail(&create->list, &kthread_create_list);	spin_unlock(&kthread_create_lock); 

Lines 1–500 of 1738.