fork.c 44.2 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/*
 *  linux/kernel/fork.c
 *
 *  Copyright (C) 1991, 1992  Linus Torvalds
 */

/*
 *  'fork.c' contains the help-routines for the 'fork' system call
 * (see also entry.S and others).
 * Fork is rather simple, once you get the hang of it, but the memory
 * management can be a bitch. See 'mm/memory.c': 'copy_page_range()'
 */

#include <linux/slab.h>
#include <linux/init.h>
#include <linux/unistd.h>
#include <linux/module.h>
#include <linux/vmalloc.h>
#include <linux/completion.h>
#include <linux/personality.h>
#include <linux/mempolicy.h>
#include <linux/sem.h>
#include <linux/file.h>
A
Al Viro 已提交
24
#include <linux/fdtable.h>
25
#include <linux/iocontext.h>
L
Linus Torvalds 已提交
26 27 28
#include <linux/key.h>
#include <linux/binfmts.h>
#include <linux/mman.h>
A
Andrea Arcangeli 已提交
29
#include <linux/mmu_notifier.h>
L
Linus Torvalds 已提交
30
#include <linux/fs.h>
S
Serge E. Hallyn 已提交
31
#include <linux/nsproxy.h>
32
#include <linux/capability.h>
L
Linus Torvalds 已提交
33
#include <linux/cpu.h>
34
#include <linux/cgroup.h>
L
Linus Torvalds 已提交
35
#include <linux/security.h>
36
#include <linux/hugetlb.h>
37
#include <linux/seccomp.h>
L
Linus Torvalds 已提交
38 39 40 41
#include <linux/swap.h>
#include <linux/syscalls.h>
#include <linux/jiffies.h>
#include <linux/futex.h>
42
#include <linux/compat.h>
43
#include <linux/kthread.h>
44
#include <linux/task_io_accounting_ops.h>
45
#include <linux/rcupdate.h>
L
Linus Torvalds 已提交
46 47 48
#include <linux/ptrace.h>
#include <linux/mount.h>
#include <linux/audit.h>
49
#include <linux/memcontrol.h>
50
#include <linux/ftrace.h>
51
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
52 53
#include <linux/profile.h>
#include <linux/rmap.h>
H
Hugh Dickins 已提交
54
#include <linux/ksm.h>
L
Linus Torvalds 已提交
55
#include <linux/acct.h>
56
#include <linux/tsacct_kern.h>
M
Matt Helsley 已提交
57
#include <linux/cn_proc.h>
R
Rafael J. Wysocki 已提交
58
#include <linux/freezer.h>
59
#include <linux/delayacct.h>
60
#include <linux/taskstats_kern.h>
61
#include <linux/random.h>
M
Miloslav Trmac 已提交
62
#include <linux/tty.h>
63
#include <linux/blkdev.h>
64
#include <linux/fs_struct.h>
65
#include <linux/magic.h>
66
#include <linux/perf_event.h>
67
#include <linux/posix-timers.h>
68
#include <linux/user-return-notifier.h>
Y
Ying Han 已提交
69
#include <linux/oom.h>
A
Andrea Arcangeli 已提交
70
#include <linux/khugepaged.h>
71
#include <linux/signalfd.h>
72
#include <linux/uprobes.h>
L
Linus Torvalds 已提交
73 74 75 76 77 78 79 80

#include <asm/pgtable.h>
#include <asm/pgalloc.h>
#include <asm/uaccess.h>
#include <asm/mmu_context.h>
#include <asm/cacheflush.h>
#include <asm/tlbflush.h>

81 82
#include <trace/events/sched.h>

83 84 85
#define CREATE_TRACE_POINTS
#include <trace/events/task.h>

L
Linus Torvalds 已提交
86 87 88 89
/*
 * Protected counters by write_lock_irq(&tasklist_lock)
 */
unsigned long total_forks;	/* Handle normal Linux uptimes. */
90
int nr_threads;			/* The idle threads do not count.. */
L
Linus Torvalds 已提交
91 92 93 94 95

int max_threads;		/* tunable limit on nr_threads */

DEFINE_PER_CPU(unsigned long, process_counts) = 0;

96
__cacheline_aligned DEFINE_RWLOCK(tasklist_lock);  /* outer */
97 98 99 100 101 102 103 104

#ifdef CONFIG_PROVE_RCU
int lockdep_tasklist_lock_is_held(void)
{
	return lockdep_is_held(&tasklist_lock);
}
EXPORT_SYMBOL_GPL(lockdep_tasklist_lock_is_held);
#endif /* #ifdef CONFIG_PROVE_RCU */
L
Linus Torvalds 已提交
105 106 107 108 109 110

int nr_processes(void)
{
	int cpu;
	int total = 0;

111
	for_each_possible_cpu(cpu)
L
Linus Torvalds 已提交
112 113 114 115 116
		total += per_cpu(process_counts, cpu);

	return total;
}

117 118 119 120
void __weak arch_release_task_struct(struct task_struct *tsk)
{
}

121
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
122
static struct kmem_cache *task_struct_cachep;
123 124 125 126 127 128 129 130 131 132

static inline struct task_struct *alloc_task_struct_node(int node)
{
	return kmem_cache_alloc_node(task_struct_cachep, GFP_KERNEL, node);
}

static inline void free_task_struct(struct task_struct *tsk)
{
	kmem_cache_free(task_struct_cachep, tsk);
}
L
Linus Torvalds 已提交
133 134
#endif

135 136 137 138
void __weak arch_release_thread_info(struct thread_info *ti)
{
}

139
#ifndef CONFIG_ARCH_THREAD_INFO_ALLOCATOR
140

141 142 143 144 145
/*
 * Allocate pages if THREAD_SIZE is >= PAGE_SIZE, otherwise use a
 * kmemcache based allocator.
 */
# if THREAD_SIZE >= PAGE_SIZE
146 147
static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
148
{
149 150
	struct page *page = alloc_pages_node(node, THREADINFO_GFP,
					     THREAD_SIZE_ORDER);
151 152

	return page ? page_address(page) : NULL;
153 154 155 156 157 158
}

static inline void free_thread_info(struct thread_info *ti)
{
	free_pages((unsigned long)ti, THREAD_SIZE_ORDER);
}
159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
# else
static struct kmem_cache *thread_info_cache;

static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
{
	return kmem_cache_alloc_node(thread_info_cache, THREADINFO_GFP, node);
}

static void free_thread_info(struct thread_info *ti)
{
	kmem_cache_free(thread_info_cache, ti);
}

void thread_info_cache_init(void)
{
	thread_info_cache = kmem_cache_create("thread_info", THREAD_SIZE,
					      THREAD_SIZE, 0, NULL);
	BUG_ON(thread_info_cache == NULL);
}
# endif
180 181
#endif

L
Linus Torvalds 已提交
182
/* SLAB cache for signal_struct structures (tsk->signal) */
183
static struct kmem_cache *signal_cachep;
L
Linus Torvalds 已提交
184 185

/* SLAB cache for sighand_struct structures (tsk->sighand) */
186
struct kmem_cache *sighand_cachep;
L
Linus Torvalds 已提交
187 188

/* SLAB cache for files_struct structures (tsk->files) */
189
struct kmem_cache *files_cachep;
L
Linus Torvalds 已提交
190 191

/* SLAB cache for fs_struct structures (tsk->fs) */
192
struct kmem_cache *fs_cachep;
L
Linus Torvalds 已提交
193 194

/* SLAB cache for vm_area_struct structures */
195
struct kmem_cache *vm_area_cachep;
L
Linus Torvalds 已提交
196 197

/* SLAB cache for mm_struct structures (tsk->mm) */
198
static struct kmem_cache *mm_cachep;
L
Linus Torvalds 已提交
199

200 201 202 203 204 205 206
static void account_kernel_stack(struct thread_info *ti, int account)
{
	struct zone *zone = page_zone(virt_to_page(ti));

	mod_zone_page_state(zone, NR_KERNEL_STACK, account);
}

L
Linus Torvalds 已提交
207 208
void free_task(struct task_struct *tsk)
{
209
	account_kernel_stack(tsk->stack, -1);
210
	arch_release_thread_info(tsk->stack);
R
Roman Zippel 已提交
211
	free_thread_info(tsk->stack);
I
Ingo Molnar 已提交
212
	rt_mutex_debug_task_free(tsk);
213
	ftrace_graph_exit_task(tsk);
214
	put_seccomp_filter(tsk);
215
	arch_release_task_struct(tsk);
L
Linus Torvalds 已提交
216 217 218 219
	free_task_struct(tsk);
}
EXPORT_SYMBOL(free_task);

220 221
static inline void free_signal_struct(struct signal_struct *sig)
{
222
	taskstats_tgid_free(sig);
223
	sched_autogroup_exit(sig);
224 225 226 227 228
	kmem_cache_free(signal_cachep, sig);
}

static inline void put_signal_struct(struct signal_struct *sig)
{
229
	if (atomic_dec_and_test(&sig->sigcnt))
230 231 232
		free_signal_struct(sig);
}

233
void __put_task_struct(struct task_struct *tsk)
L
Linus Torvalds 已提交
234
{
E
Eugene Teo 已提交
235
	WARN_ON(!tsk->exit_state);
L
Linus Torvalds 已提交
236 237 238
	WARN_ON(atomic_read(&tsk->usage));
	WARN_ON(tsk == current);

239
	security_task_free(tsk);
240
	exit_creds(tsk);
241
	delayacct_tsk_free(tsk);
242
	put_signal_struct(tsk->signal);
L
Linus Torvalds 已提交
243 244 245 246

	if (!profile_handoff_task(tsk))
		free_task(tsk);
}
247
EXPORT_SYMBOL_GPL(__put_task_struct);
L
Linus Torvalds 已提交
248

T
Thomas Gleixner 已提交
249
void __init __weak arch_task_cache_init(void) { }
250

L
Linus Torvalds 已提交
251 252
void __init fork_init(unsigned long mempages)
{
253
#ifndef CONFIG_ARCH_TASK_STRUCT_ALLOCATOR
L
Linus Torvalds 已提交
254 255 256 257 258 259
#ifndef ARCH_MIN_TASKALIGN
#define ARCH_MIN_TASKALIGN	L1_CACHE_BYTES
#endif
	/* create a slab on which task_structs can be allocated */
	task_struct_cachep =
		kmem_cache_create("task_struct", sizeof(struct task_struct),
V
Vegard Nossum 已提交
260
			ARCH_MIN_TASKALIGN, SLAB_PANIC | SLAB_NOTRACK, NULL);
L
Linus Torvalds 已提交
261 262
#endif

263 264 265
	/* do the arch specific task caches init */
	arch_task_cache_init();

L
Linus Torvalds 已提交
266 267 268 269 270 271 272 273 274 275
	/*
	 * The default maximum number of threads is set to a safe
	 * value: the thread structures can take up at most half
	 * of memory.
	 */
	max_threads = mempages / (8 * THREAD_SIZE / PAGE_SIZE);

	/*
	 * we need to allow at least 20 threads to boot a system
	 */
276
	if (max_threads < 20)
L
Linus Torvalds 已提交
277 278 279 280 281 282 283 284
		max_threads = 20;

	init_task.signal->rlim[RLIMIT_NPROC].rlim_cur = max_threads/2;
	init_task.signal->rlim[RLIMIT_NPROC].rlim_max = max_threads/2;
	init_task.signal->rlim[RLIMIT_SIGPENDING] =
		init_task.signal->rlim[RLIMIT_NPROC];
}

285 286 287 288 289 290 291
int __attribute__((weak)) arch_dup_task_struct(struct task_struct *dst,
					       struct task_struct *src)
{
	*dst = *src;
	return 0;
}

L
Linus Torvalds 已提交
292 293 294 295
static struct task_struct *dup_task_struct(struct task_struct *orig)
{
	struct task_struct *tsk;
	struct thread_info *ti;
296
	unsigned long *stackend;
297
	int node = tsk_fork_get_node(orig);
P
Peter Zijlstra 已提交
298
	int err;
L
Linus Torvalds 已提交
299

300
	tsk = alloc_task_struct_node(node);
L
Linus Torvalds 已提交
301 302 303
	if (!tsk)
		return NULL;

304
	ti = alloc_thread_info_node(tsk, node);
305 306
	if (!ti)
		goto free_tsk;
L
Linus Torvalds 已提交
307

308
	err = arch_dup_task_struct(tsk, orig);
309
	if (err)
310
		goto free_ti;
311

312 313 314
	tsk->stack = ti;

	setup_thread_stack(tsk, orig);
315
	clear_user_return_notifier(tsk);
316
	clear_tsk_need_resched(tsk);
317 318
	stackend = end_of_stack(tsk);
	*stackend = STACK_END_MAGIC;	/* for overflow detection */
L
Linus Torvalds 已提交
319

320 321 322 323
#ifdef CONFIG_CC_STACKPROTECTOR
	tsk->stack_canary = get_random_int();
#endif

324 325 326 327 328
	/*
	 * One for us, one for whoever does the "release_task()" (usually
	 * parent)
	 */
	atomic_set(&tsk->usage, 2);
329
#ifdef CONFIG_BLK_DEV_IO_TRACE
330
	tsk->btrace_seq = 0;
331
#endif
332
	tsk->splice_pipe = NULL;
333
	tsk->task_frag.page = NULL;
334 335 336

	account_kernel_stack(ti, 1);

L
Linus Torvalds 已提交
337
	return tsk;
338

339
free_ti:
340
	free_thread_info(ti);
341
free_tsk:
342 343
	free_task_struct(tsk);
	return NULL;
L
Linus Torvalds 已提交
344 345 346
}

#ifdef CONFIG_MMU
A
Alexey Dobriyan 已提交
347
static int dup_mmap(struct mm_struct *mm, struct mm_struct *oldmm)
L
Linus Torvalds 已提交
348
{
349
	struct vm_area_struct *mpnt, *tmp, *prev, **pprev;
L
Linus Torvalds 已提交
350 351 352 353 354 355
	struct rb_node **rb_link, *rb_parent;
	int retval;
	unsigned long charge;
	struct mempolicy *pol;

	down_write(&oldmm->mmap_sem);
356
	flush_cache_dup_mm(oldmm);
357
	uprobe_dup_mmap(oldmm, mm);
358 359 360 361
	/*
	 * Not linked in yet - no deadlock potential:
	 */
	down_write_nested(&mm->mmap_sem, SINGLE_DEPTH_NESTING);
362

L
Linus Torvalds 已提交
363 364 365 366
	mm->locked_vm = 0;
	mm->mmap = NULL;
	mm->mmap_cache = NULL;
	mm->free_area_cache = oldmm->mmap_base;
367
	mm->cached_hole_size = ~0UL;
L
Linus Torvalds 已提交
368
	mm->map_count = 0;
369
	cpumask_clear(mm_cpumask(mm));
L
Linus Torvalds 已提交
370 371 372 373
	mm->mm_rb = RB_ROOT;
	rb_link = &mm->mm_rb.rb_node;
	rb_parent = NULL;
	pprev = &mm->mmap;
H
Hugh Dickins 已提交
374
	retval = ksm_fork(mm, oldmm);
A
Andrea Arcangeli 已提交
375 376 377
	if (retval)
		goto out;
	retval = khugepaged_fork(mm, oldmm);
H
Hugh Dickins 已提交
378 379
	if (retval)
		goto out;
L
Linus Torvalds 已提交
380

381
	prev = NULL;
H
Hugh Dickins 已提交
382
	for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
L
Linus Torvalds 已提交
383 384 385
		struct file *file;

		if (mpnt->vm_flags & VM_DONTCOPY) {
386
			vm_stat_account(mm, mpnt->vm_flags, mpnt->vm_file,
387
							-vma_pages(mpnt));
L
Linus Torvalds 已提交
388 389 390 391
			continue;
		}
		charge = 0;
		if (mpnt->vm_flags & VM_ACCOUNT) {
392 393
			unsigned long len = vma_pages(mpnt);

394
			if (security_vm_enough_memory_mm(oldmm, len)) /* sic */
L
Linus Torvalds 已提交
395 396 397
				goto fail_nomem;
			charge = len;
		}
398
		tmp = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
399 400 401
		if (!tmp)
			goto fail_nomem;
		*tmp = *mpnt;
402
		INIT_LIST_HEAD(&tmp->anon_vma_chain);
403
		pol = mpol_dup(vma_policy(mpnt));
L
Linus Torvalds 已提交
404 405 406 407
		retval = PTR_ERR(pol);
		if (IS_ERR(pol))
			goto fail_nomem_policy;
		vma_set_policy(tmp, pol);
A
Andrea Arcangeli 已提交
408
		tmp->vm_mm = mm;
409 410
		if (anon_vma_fork(tmp, mpnt))
			goto fail_nomem_anon_vma_fork;
L
Linus Torvalds 已提交
411
		tmp->vm_flags &= ~VM_LOCKED;
412
		tmp->vm_next = tmp->vm_prev = NULL;
L
Linus Torvalds 已提交
413 414
		file = tmp->vm_file;
		if (file) {
415
			struct inode *inode = file->f_path.dentry->d_inode;
H
Hugh Dickins 已提交
416 417
			struct address_space *mapping = file->f_mapping;

L
Linus Torvalds 已提交
418 419 420
			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
421
			mutex_lock(&mapping->i_mmap_mutex);
H
Hugh Dickins 已提交
422 423 424 425
			if (tmp->vm_flags & VM_SHARED)
				mapping->i_mmap_writable++;
			flush_dcache_mmap_lock(mapping);
			/* insert tmp into the share list, just after mpnt */
M
Michel Lespinasse 已提交
426 427 428 429 430 431
			if (unlikely(tmp->vm_flags & VM_NONLINEAR))
				vma_nonlinear_insert(tmp,
						&mapping->i_mmap_nonlinear);
			else
				vma_interval_tree_insert_after(tmp, mpnt,
							&mapping->i_mmap);
H
Hugh Dickins 已提交
432
			flush_dcache_mmap_unlock(mapping);
433
			mutex_unlock(&mapping->i_mmap_mutex);
L
Linus Torvalds 已提交
434 435
		}

436 437 438 439 440 441 442 443
		/*
		 * Clear hugetlb-related page reserves for children. This only
		 * affects MAP_PRIVATE mappings. Faults generated by the child
		 * are not guaranteed to succeed, even if read-only
		 */
		if (is_vm_hugetlb_page(tmp))
			reset_vma_resv_huge_pages(tmp);

L
Linus Torvalds 已提交
444
		/*
445
		 * Link in the new vma and copy the page table entries.
L
Linus Torvalds 已提交
446 447 448
		 */
		*pprev = tmp;
		pprev = &tmp->vm_next;
449 450
		tmp->vm_prev = prev;
		prev = tmp;
L
Linus Torvalds 已提交
451 452 453 454 455 456

		__vma_link_rb(mm, tmp, rb_link, rb_parent);
		rb_link = &tmp->vm_rb.rb_right;
		rb_parent = &tmp->vm_rb;

		mm->map_count++;
457
		retval = copy_page_range(mm, oldmm, mpnt);
L
Linus Torvalds 已提交
458 459 460 461 462 463 464

		if (tmp->vm_ops && tmp->vm_ops->open)
			tmp->vm_ops->open(tmp);

		if (retval)
			goto out;
	}
465 466
	/* a new mm has just been created */
	arch_dup_mmap(oldmm, mm);
L
Linus Torvalds 已提交
467 468
	retval = 0;
out:
469
	up_write(&mm->mmap_sem);
H
Hugh Dickins 已提交
470
	flush_tlb_mm(oldmm);
L
Linus Torvalds 已提交
471 472
	up_write(&oldmm->mmap_sem);
	return retval;
473 474
fail_nomem_anon_vma_fork:
	mpol_put(pol);
L
Linus Torvalds 已提交
475 476 477 478 479 480 481 482
fail_nomem_policy:
	kmem_cache_free(vm_area_cachep, tmp);
fail_nomem:
	retval = -ENOMEM;
	vm_unacct_memory(charge);
	goto out;
}

483
static inline int mm_alloc_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
484 485 486 487 488 489 490
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

491
static inline void mm_free_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
492
{
493
	pgd_free(mm, mm->pgd);
L
Linus Torvalds 已提交
494 495 496 497 498 499 500
}
#else
#define dup_mmap(mm, oldmm)	(0)
#define mm_alloc_pgd(mm)	(0)
#define mm_free_pgd(mm)
#endif /* CONFIG_MMU */

D
Daniel Walker 已提交
501
__cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
L
Linus Torvalds 已提交
502

503
#define allocate_mm()	(kmem_cache_alloc(mm_cachep, GFP_KERNEL))
L
Linus Torvalds 已提交
504 505
#define free_mm(mm)	(kmem_cache_free(mm_cachep, (mm)))

506 507 508 509 510 511 512 513 514 515 516 517
static unsigned long default_dump_filter = MMF_DUMP_FILTER_DEFAULT;

static int __init coredump_filter_setup(char *s)
{
	default_dump_filter =
		(simple_strtoul(s, NULL, 0) << MMF_DUMP_FILTER_SHIFT) &
		MMF_DUMP_FILTER_MASK;
	return 1;
}

__setup("coredump_filter=", coredump_filter_setup);

L
Linus Torvalds 已提交
518 519
#include <linux/init_task.h>

A
Alexey Dobriyan 已提交
520 521 522 523 524 525 526 527
static void mm_init_aio(struct mm_struct *mm)
{
#ifdef CONFIG_AIO
	spin_lock_init(&mm->ioctx_lock);
	INIT_HLIST_HEAD(&mm->ioctx_list);
#endif
}

528
static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p)
L
Linus Torvalds 已提交
529 530 531 532 533
{
	atomic_set(&mm->mm_users, 1);
	atomic_set(&mm->mm_count, 1);
	init_rwsem(&mm->mmap_sem);
	INIT_LIST_HEAD(&mm->mmlist);
H
Hugh Dickins 已提交
534 535
	mm->flags = (current->mm) ?
		(current->mm->flags & MMF_INIT_MASK) : default_dump_filter;
536
	mm->core_state = NULL;
L
Linus Torvalds 已提交
537
	mm->nr_ptes = 0;
K
KAMEZAWA Hiroyuki 已提交
538
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
539 540
	spin_lock_init(&mm->page_table_lock);
	mm->free_area_cache = TASK_UNMAPPED_BASE;
541
	mm->cached_hole_size = ~0UL;
A
Alexey Dobriyan 已提交
542
	mm_init_aio(mm);
543
	mm_init_owner(mm, p);
L
Linus Torvalds 已提交
544 545 546

	if (likely(!mm_alloc_pgd(mm))) {
		mm->def_flags = 0;
A
Andrea Arcangeli 已提交
547
		mmu_notifier_mm_init(mm);
L
Linus Torvalds 已提交
548 549
		return mm;
	}
550

L
Linus Torvalds 已提交
551 552 553 554
	free_mm(mm);
	return NULL;
}

555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
static void check_mm(struct mm_struct *mm)
{
	int i;

	for (i = 0; i < NR_MM_COUNTERS; i++) {
		long x = atomic_long_read(&mm->rss_stat.count[i]);

		if (unlikely(x))
			printk(KERN_ALERT "BUG: Bad rss-counter state "
					  "mm:%p idx:%d val:%ld\n", mm, i, x);
	}

#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	VM_BUG_ON(mm->pmd_huge_pte);
#endif
}

L
Linus Torvalds 已提交
572 573 574
/*
 * Allocate and initialize an mm_struct.
 */
575
struct mm_struct *mm_alloc(void)
L
Linus Torvalds 已提交
576
{
577
	struct mm_struct *mm;
L
Linus Torvalds 已提交
578 579

	mm = allocate_mm();
580 581 582 583
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
584 585
	mm_init_cpumask(mm);
	return mm_init(mm, current);
L
Linus Torvalds 已提交
586 587 588 589 590 591 592
}

/*
 * Called when the last reference to the mm
 * is dropped: either by a lazy thread or by
 * mmput. Free the page directory and the mm.
 */
593
void __mmdrop(struct mm_struct *mm)
L
Linus Torvalds 已提交
594 595 596 597
{
	BUG_ON(mm == &init_mm);
	mm_free_pgd(mm);
	destroy_context(mm);
A
Andrea Arcangeli 已提交
598
	mmu_notifier_mm_destroy(mm);
599
	check_mm(mm);
L
Linus Torvalds 已提交
600 601
	free_mm(mm);
}
602
EXPORT_SYMBOL_GPL(__mmdrop);
L
Linus Torvalds 已提交
603 604 605 606 607 608

/*
 * Decrement the use count and release all resources for an mm.
 */
void mmput(struct mm_struct *mm)
{
A
Andrew Morton 已提交
609 610
	might_sleep();

L
Linus Torvalds 已提交
611
	if (atomic_dec_and_test(&mm->mm_users)) {
612
		uprobe_clear_state(mm);
L
Linus Torvalds 已提交
613
		exit_aio(mm);
614
		ksm_exit(mm);
A
Andrea Arcangeli 已提交
615
		khugepaged_exit(mm); /* must run before exit_mmap */
L
Linus Torvalds 已提交
616
		exit_mmap(mm);
M
Matt Helsley 已提交
617
		set_mm_exe_file(mm, NULL);
L
Linus Torvalds 已提交
618 619 620 621 622
		if (!list_empty(&mm->mmlist)) {
			spin_lock(&mmlist_lock);
			list_del(&mm->mmlist);
			spin_unlock(&mmlist_lock);
		}
623 624
		if (mm->binfmt)
			module_put(mm->binfmt->module);
L
Linus Torvalds 已提交
625 626 627 628 629
		mmdrop(mm);
	}
}
EXPORT_SYMBOL_GPL(mmput);

630 631 632 633 634 635 636 637 638 639 640 641 642
void set_mm_exe_file(struct mm_struct *mm, struct file *new_exe_file)
{
	if (new_exe_file)
		get_file(new_exe_file);
	if (mm->exe_file)
		fput(mm->exe_file);
	mm->exe_file = new_exe_file;
}

struct file *get_mm_exe_file(struct mm_struct *mm)
{
	struct file *exe_file;

643
	/* We need mmap_sem to protect against races with removal of exe_file */
644 645 646 647 648 649 650 651 652 653 654 655 656 657 658
	down_read(&mm->mmap_sem);
	exe_file = mm->exe_file;
	if (exe_file)
		get_file(exe_file);
	up_read(&mm->mmap_sem);
	return exe_file;
}

static void dup_mm_exe_file(struct mm_struct *oldmm, struct mm_struct *newmm)
{
	/* It's safe to write the exe_file pointer without exe_file_lock because
	 * this is called during fork when the task is not yet in /proc */
	newmm->exe_file = get_mm_exe_file(oldmm);
}

L
Linus Torvalds 已提交
659 660 661
/**
 * get_task_mm - acquire a reference to the task's mm
 *
662
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
663 664 665 666 667 668 669 670 671 672 673 674
 * this kernel workthread has transiently adopted a user mm with use_mm,
 * to do its AIO) is not set and if so returns a reference to it, after
 * bumping up the use count.  User must release the mm via mmput()
 * after use.  Typically used by /proc and ptrace.
 */
struct mm_struct *get_task_mm(struct task_struct *task)
{
	struct mm_struct *mm;

	task_lock(task);
	mm = task->mm;
	if (mm) {
675
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
676 677 678 679 680 681 682 683 684
			mm = NULL;
		else
			atomic_inc(&mm->mm_users);
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704
struct mm_struct *mm_access(struct task_struct *task, unsigned int mode)
{
	struct mm_struct *mm;
	int err;

	err =  mutex_lock_killable(&task->signal->cred_guard_mutex);
	if (err)
		return ERR_PTR(err);

	mm = get_task_mm(task);
	if (mm && mm != current->mm &&
			!ptrace_may_access(task, mode)) {
		mmput(mm);
		mm = ERR_PTR(-EACCES);
	}
	mutex_unlock(&task->signal->cred_guard_mutex);

	return mm;
}

705
static void complete_vfork_done(struct task_struct *tsk)
706
{
O
Oleg Nesterov 已提交
707
	struct completion *vfork;
708

O
Oleg Nesterov 已提交
709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734
	task_lock(tsk);
	vfork = tsk->vfork_done;
	if (likely(vfork)) {
		tsk->vfork_done = NULL;
		complete(vfork);
	}
	task_unlock(tsk);
}

static int wait_for_vfork_done(struct task_struct *child,
				struct completion *vfork)
{
	int killed;

	freezer_do_not_count();
	killed = wait_for_completion_killable(vfork);
	freezer_count();

	if (killed) {
		task_lock(child);
		child->vfork_done = NULL;
		task_unlock(child);
	}

	put_task_struct(child);
	return killed;
735 736
}

L
Linus Torvalds 已提交
737 738 739 740 741 742 743 744 745 746 747 748 749 750 751
/* Please note the differences between mmput and mm_release.
 * mmput is called whenever we stop holding onto a mm_struct,
 * error success whatever.
 *
 * mm_release is called after a mm_struct has been removed
 * from the current process.
 *
 * This difference is important for error handling, when we
 * only half set up a mm_struct for a new process and need to restore
 * the old one.  Because we mmput the new mm_struct before
 * restoring the old one. . .
 * Eric Biederman 10 January 1998
 */
void mm_release(struct task_struct *tsk, struct mm_struct *mm)
{
752 753
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
754
	if (unlikely(tsk->robust_list)) {
755
		exit_robust_list(tsk);
756 757
		tsk->robust_list = NULL;
	}
758
#ifdef CONFIG_COMPAT
759
	if (unlikely(tsk->compat_robust_list)) {
760
		compat_exit_robust_list(tsk);
761 762
		tsk->compat_robust_list = NULL;
	}
763
#endif
764 765
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
766 767
#endif

768 769
	uprobe_free_utask(tsk);

L
Linus Torvalds 已提交
770 771 772
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

773 774 775 776
	/*
	 * If we're exiting normally, clear a user-space tid field if
	 * requested.  We leave this alone when dying by signal, to leave
	 * the value intact in a core dump, and to save the unnecessary
O
Oleg Nesterov 已提交
777 778
	 * trouble, say, a killed vfork parent shouldn't touch this mm.
	 * Userland only wants this done for a sys_exit.
779
	 */
780 781 782 783 784 785 786 787 788 789 790
	if (tsk->clear_child_tid) {
		if (!(tsk->flags & PF_SIGNALED) &&
		    atomic_read(&mm->mm_users) > 1) {
			/*
			 * We don't check the error code - if userspace has
			 * not set up a proper pointer then tough luck.
			 */
			put_user(0, tsk->clear_child_tid);
			sys_futex(tsk->clear_child_tid, FUTEX_WAKE,
					1, NULL, NULL, 0);
		}
L
Linus Torvalds 已提交
791 792
		tsk->clear_child_tid = NULL;
	}
793 794 795 796 797 798 799

	/*
	 * All done, finally we can wake up parent and return this mm to him.
	 * Also kthread_stop() uses this completion for synchronization.
	 */
	if (tsk->vfork_done)
		complete_vfork_done(tsk);
L
Linus Torvalds 已提交
800 801
}

802 803 804 805
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
806
struct mm_struct *dup_mm(struct task_struct *tsk)
807 808 809 810 811 812 813 814 815 816 817 818
{
	struct mm_struct *mm, *oldmm = current->mm;
	int err;

	if (!oldmm)
		return NULL;

	mm = allocate_mm();
	if (!mm)
		goto fail_nomem;

	memcpy(mm, oldmm, sizeof(*mm));
819
	mm_init_cpumask(mm);
820

821 822 823
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	mm->pmd_huge_pte = NULL;
#endif
824
	if (!mm_init(mm, tsk))
825 826 827 828 829
		goto fail_nomem;

	if (init_new_context(tsk, mm))
		goto fail_nocontext;

M
Matt Helsley 已提交
830 831
	dup_mm_exe_file(oldmm, mm);

832 833 834 835 836 837 838
	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

	mm->hiwater_rss = get_mm_rss(mm);
	mm->hiwater_vm = mm->total_vm;

839 840 841
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

842 843 844
	return mm;

free_pt:
845 846
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
847 848 849 850 851 852 853 854 855 856 857 858 859 860 861
	mmput(mm);

fail_nomem:
	return NULL;

fail_nocontext:
	/*
	 * If init_new_context() failed, we cannot use mmput() to free the mm
	 * because it calls destroy_context()
	 */
	mm_free_pgd(mm);
	free_mm(mm);
	return NULL;
}

862
static int copy_mm(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
863
{
864
	struct mm_struct *mm, *oldmm;
L
Linus Torvalds 已提交
865 866 867 868
	int retval;

	tsk->min_flt = tsk->maj_flt = 0;
	tsk->nvcsw = tsk->nivcsw = 0;
869 870 871
#ifdef CONFIG_DETECT_HUNG_TASK
	tsk->last_switch_count = tsk->nvcsw + tsk->nivcsw;
#endif
L
Linus Torvalds 已提交
872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891

	tsk->mm = NULL;
	tsk->active_mm = NULL;

	/*
	 * Are we cloning a kernel thread?
	 *
	 * We need to steal a active VM for that..
	 */
	oldmm = current->mm;
	if (!oldmm)
		return 0;

	if (clone_flags & CLONE_VM) {
		atomic_inc(&oldmm->mm_users);
		mm = oldmm;
		goto good_mm;
	}

	retval = -ENOMEM;
892
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
893 894 895 896 897 898 899 900 901 902 903 904
	if (!mm)
		goto fail_nomem;

good_mm:
	tsk->mm = mm;
	tsk->active_mm = mm;
	return 0;

fail_nomem:
	return retval;
}

A
Alexey Dobriyan 已提交
905
static int copy_fs(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
906
{
A
Al Viro 已提交
907
	struct fs_struct *fs = current->fs;
L
Linus Torvalds 已提交
908
	if (clone_flags & CLONE_FS) {
A
Al Viro 已提交
909
		/* tsk->fs is already what we want */
N
Nick Piggin 已提交
910
		spin_lock(&fs->lock);
A
Al Viro 已提交
911
		if (fs->in_exec) {
N
Nick Piggin 已提交
912
			spin_unlock(&fs->lock);
A
Al Viro 已提交
913 914 915
			return -EAGAIN;
		}
		fs->users++;
N
Nick Piggin 已提交
916
		spin_unlock(&fs->lock);
L
Linus Torvalds 已提交
917 918
		return 0;
	}
A
Al Viro 已提交
919
	tsk->fs = copy_fs_struct(fs);
L
Linus Torvalds 已提交
920 921 922 923 924
	if (!tsk->fs)
		return -ENOMEM;
	return 0;
}

925
static int copy_files(unsigned long clone_flags, struct task_struct *tsk)
926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951
{
	struct files_struct *oldf, *newf;
	int error = 0;

	/*
	 * A background process may not have any files ...
	 */
	oldf = current->files;
	if (!oldf)
		goto out;

	if (clone_flags & CLONE_FILES) {
		atomic_inc(&oldf->count);
		goto out;
	}

	newf = dup_fd(oldf, &error);
	if (!newf)
		goto out;

	tsk->files = newf;
	error = 0;
out:
	return error;
}

952
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
953 954 955
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;
956
	struct io_context *new_ioc;
957 958 959

	if (!ioc)
		return 0;
960 961 962 963
	/*
	 * Share io context with parent, if CLONE_IO is set
	 */
	if (clone_flags & CLONE_IO) {
T
Tejun Heo 已提交
964 965
		ioc_task_link(ioc);
		tsk->io_context = ioc;
966
	} else if (ioprio_valid(ioc->ioprio)) {
967 968
		new_ioc = get_task_io_context(tsk, GFP_KERNEL, NUMA_NO_NODE);
		if (unlikely(!new_ioc))
969 970
			return -ENOMEM;

971
		new_ioc->ioprio = ioc->ioprio;
972
		put_io_context(new_ioc);
973 974 975 976 977
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
978
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
979 980 981
{
	struct sighand_struct *sig;

Z
Zhaolei 已提交
982
	if (clone_flags & CLONE_SIGHAND) {
L
Linus Torvalds 已提交
983 984 985 986
		atomic_inc(&current->sighand->count);
		return 0;
	}
	sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
I
Ingo Molnar 已提交
987
	rcu_assign_pointer(tsk->sighand, sig);
L
Linus Torvalds 已提交
988 989 990 991 992 993 994
	if (!sig)
		return -ENOMEM;
	atomic_set(&sig->count, 1);
	memcpy(sig->action, current->sighand->action, sizeof(sig->action));
	return 0;
}

995
void __cleanup_sighand(struct sighand_struct *sighand)
996
{
997 998
	if (atomic_dec_and_test(&sighand->count)) {
		signalfd_cleanup(sighand);
999
		kmem_cache_free(sighand_cachep, sighand);
1000
	}
1001 1002
}

1003 1004 1005 1006 1007 1008

/*
 * Initialize POSIX timer handling for a thread group.
 */
static void posix_cpu_timers_init_group(struct signal_struct *sig)
{
1009 1010
	unsigned long cpu_limit;

1011 1012 1013
	/* Thread group counters. */
	thread_group_cputime_init(sig);

1014 1015 1016
	cpu_limit = ACCESS_ONCE(sig->rlim[RLIMIT_CPU].rlim_cur);
	if (cpu_limit != RLIM_INFINITY) {
		sig->cputime_expires.prof_exp = secs_to_cputime(cpu_limit);
1017 1018 1019
		sig->cputimer.running = 1;
	}

1020 1021 1022 1023 1024 1025
	/* The timer lists. */
	INIT_LIST_HEAD(&sig->cpu_timers[0]);
	INIT_LIST_HEAD(&sig->cpu_timers[1]);
	INIT_LIST_HEAD(&sig->cpu_timers[2]);
}

A
Alexey Dobriyan 已提交
1026
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1027 1028 1029
{
	struct signal_struct *sig;

1030
	if (clone_flags & CLONE_THREAD)
1031 1032
		return 0;

1033
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
1034 1035 1036 1037
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

1038
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
1039
	atomic_set(&sig->live, 1);
1040
	atomic_set(&sig->sigcnt, 1);
L
Linus Torvalds 已提交
1041
	init_waitqueue_head(&sig->wait_chldexit);
1042
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
1043 1044 1045
	init_sigpending(&sig->shared_pending);
	INIT_LIST_HEAD(&sig->posix_timers);

1046
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
1047 1048 1049 1050 1051 1052
	sig->real_timer.function = it_real_fn;

	task_lock(current->group_leader);
	memcpy(sig->rlim, current->signal->rlim, sizeof sig->rlim);
	task_unlock(current->group_leader);

1053 1054
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
1055
	tty_audit_fork(sig);
1056
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
1057

1058
#ifdef CONFIG_CGROUPS
1059
	init_rwsem(&sig->group_rwsem);
1060 1061
#endif

D
David Rientjes 已提交
1062
	sig->oom_score_adj = current->signal->oom_score_adj;
1063
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
1064

1065 1066 1067
	sig->has_child_subreaper = current->signal->has_child_subreaper ||
				   current->signal->is_child_subreaper;

1068 1069
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
1070 1071 1072
	return 0;
}

A
Alexey Dobriyan 已提交
1073
static void copy_flags(unsigned long clone_flags, struct task_struct *p)
L
Linus Torvalds 已提交
1074 1075 1076
{
	unsigned long new_flags = p->flags;

T
Tejun Heo 已提交
1077
	new_flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER);
L
Linus Torvalds 已提交
1078 1079 1080 1081
	new_flags |= PF_FORKNOEXEC;
	p->flags = new_flags;
}

1082
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1083 1084 1085
{
	current->clear_child_tid = tidptr;

1086
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1087 1088
}

A
Alexey Dobriyan 已提交
1089
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1090
{
1091
	raw_spin_lock_init(&p->pi_lock);
1092
#ifdef CONFIG_RT_MUTEXES
1093
	plist_head_init(&p->pi_waiters);
I
Ingo Molnar 已提交
1094 1095 1096 1097
	p->pi_blocked_on = NULL;
#endif
}

1098 1099 1100 1101 1102 1103 1104
#ifdef CONFIG_MM_OWNER
void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
	mm->owner = p;
}
#endif /* CONFIG_MM_OWNER */

1105 1106 1107 1108 1109
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
1110 1111
	tsk->cputime_expires.prof_exp = 0;
	tsk->cputime_expires.virt_exp = 0;
1112 1113 1114 1115 1116 1117
	tsk->cputime_expires.sched_exp = 0;
	INIT_LIST_HEAD(&tsk->cpu_timers[0]);
	INIT_LIST_HEAD(&tsk->cpu_timers[1]);
	INIT_LIST_HEAD(&tsk->cpu_timers[2]);
}

L
Linus Torvalds 已提交
1118 1119 1120 1121 1122 1123 1124 1125
/*
 * This creates a new process as a copy of the old one,
 * but does not actually start it yet.
 *
 * It copies the registers, and all the appropriate
 * parts of the process environment (as per the clone
 * flags). The actual kick-off is left to the caller.
 */
1126 1127 1128 1129 1130
static struct task_struct *copy_process(unsigned long clone_flags,
					unsigned long stack_start,
					struct pt_regs *regs,
					unsigned long stack_size,
					int __user *child_tidptr,
R
Roland McGrath 已提交
1131 1132
					struct pid *pid,
					int trace)
L
Linus Torvalds 已提交
1133 1134
{
	int retval;
1135
	struct task_struct *p;
1136
	int cgroup_callbacks_done = 0;
L
Linus Torvalds 已提交
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155

	if ((clone_flags & (CLONE_NEWNS|CLONE_FS)) == (CLONE_NEWNS|CLONE_FS))
		return ERR_PTR(-EINVAL);

	/*
	 * Thread groups must share signals as well, and detached threads
	 * can only be started up within the thread group.
	 */
	if ((clone_flags & CLONE_THREAD) && !(clone_flags & CLONE_SIGHAND))
		return ERR_PTR(-EINVAL);

	/*
	 * Shared signal handlers imply shared VM. By way of the above,
	 * thread groups also imply shared VM. Blocking this case allows
	 * for various simplifications in other code.
	 */
	if ((clone_flags & CLONE_SIGHAND) && !(clone_flags & CLONE_VM))
		return ERR_PTR(-EINVAL);

1156 1157 1158 1159 1160 1161 1162 1163 1164 1165
	/*
	 * Siblings of global init remain as zombies on exit since they are
	 * not reaped by their parent (swapper). To solve this and to avoid
	 * multi-rooted process trees, prevent global and container-inits
	 * from creating siblings.
	 */
	if ((clone_flags & CLONE_PARENT) &&
				current->signal->flags & SIGNAL_UNKILLABLE)
		return ERR_PTR(-EINVAL);

L
Linus Torvalds 已提交
1166 1167 1168 1169 1170 1171 1172 1173 1174
	retval = security_task_create(clone_flags);
	if (retval)
		goto fork_out;

	retval = -ENOMEM;
	p = dup_task_struct(current);
	if (!p)
		goto fork_out;

1175
	ftrace_graph_init_task(p);
1176
	get_seccomp_filter(p);
1177

1178 1179
	rt_mutex_init_task(p);

I
Ingo Molnar 已提交
1180
#ifdef CONFIG_PROVE_LOCKING
1181 1182 1183
	DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled);
	DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled);
#endif
L
Linus Torvalds 已提交
1184
	retval = -EAGAIN;
1185
	if (atomic_read(&p->real_cred->user->processes) >=
1186
			task_rlimit(p, RLIMIT_NPROC)) {
L
Linus Torvalds 已提交
1187
		if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) &&
1188
		    p->real_cred->user != INIT_USER)
L
Linus Torvalds 已提交
1189 1190
			goto bad_fork_free;
	}
1191
	current->flags &= ~PF_NPROC_EXCEEDED;
L
Linus Torvalds 已提交
1192

1193 1194 1195
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1196 1197 1198 1199 1200 1201

	/*
	 * If multiple threads are within copy_process(), then this check
	 * triggers too late. This doesn't hurt, the check is only there
	 * to stop root fork bombs.
	 */
1202
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1203 1204 1205
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

A
Al Viro 已提交
1206
	if (!try_module_get(task_thread_info(p)->exec_domain->module))
L
Linus Torvalds 已提交
1207 1208 1209
		goto bad_fork_cleanup_count;

	p->did_exec = 0;
1210
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
L
Linus Torvalds 已提交
1211 1212 1213
	copy_flags(clone_flags, p);
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1214
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1215 1216 1217 1218 1219
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

1220 1221
	p->utime = p->stime = p->gtime = 0;
	p->utimescaled = p->stimescaled = 0;
1222
#ifndef CONFIG_VIRT_CPU_ACCOUNTING
1223
	p->prev_utime = p->prev_stime = 0;
1224
#endif
1225 1226 1227
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1228

1229 1230
	p->default_timer_slack_ns = current->timer_slack_ns;

1231
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1232 1233
	acct_clear_integrals(p);

1234
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1235 1236

	do_posix_clock_monotonic_gettime(&p->start_time);
1237 1238
	p->real_start_time = p->start_time;
	monotonic_to_bootbased(&p->real_start_time);
L
Linus Torvalds 已提交
1239 1240
	p->io_context = NULL;
	p->audit_context = NULL;
1241
	if (clone_flags & CLONE_THREAD)
1242
		threadgroup_change_begin(current);
1243
	cgroup_fork(p);
L
Linus Torvalds 已提交
1244
#ifdef CONFIG_NUMA
1245
	p->mempolicy = mpol_dup(p->mempolicy);
1246 1247 1248 1249 1250
	if (IS_ERR(p->mempolicy)) {
		retval = PTR_ERR(p->mempolicy);
		p->mempolicy = NULL;
		goto bad_fork_cleanup_cgroup;
	}
1251
	mpol_fix_fork_child_flag(p);
L
Linus Torvalds 已提交
1252
#endif
1253 1254 1255
#ifdef CONFIG_CPUSETS
	p->cpuset_mem_spread_rotor = NUMA_NO_NODE;
	p->cpuset_slab_spread_rotor = NUMA_NO_NODE;
1256
	seqcount_init(&p->mems_allowed_seq);
1257
#endif
1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
	p->hardirqs_enabled = 0;
	p->hardirq_enable_ip = 0;
	p->hardirq_enable_event = 0;
	p->hardirq_disable_ip = _THIS_IP_;
	p->hardirq_disable_event = 0;
	p->softirqs_enabled = 1;
	p->softirq_enable_ip = _THIS_IP_;
	p->softirq_enable_event = 0;
	p->softirq_disable_ip = 0;
	p->softirq_disable_event = 0;
	p->hardirq_context = 0;
	p->softirq_context = 0;
#endif
I
Ingo Molnar 已提交
1273 1274 1275 1276 1277
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
#endif
L
Linus Torvalds 已提交
1278

1279 1280 1281
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
A
Andrew Morton 已提交
1282
#ifdef CONFIG_MEMCG
1283 1284 1285
	p->memcg_batch.do_batch = 0;
	p->memcg_batch.memcg = NULL;
#endif
1286

1287
	/* Perform scheduler related setup. Assign this task to a CPU. */
1288
	sched_fork(p);
1289

1290
	retval = perf_event_init_task(p);
1291 1292
	if (retval)
		goto bad_fork_cleanup_policy;
1293 1294
	retval = audit_alloc(p);
	if (retval)
1295
		goto bad_fork_cleanup_policy;
L
Linus Torvalds 已提交
1296
	/* copy all the process information */
1297 1298
	retval = copy_semundo(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1299
		goto bad_fork_cleanup_audit;
1300 1301
	retval = copy_files(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1302
		goto bad_fork_cleanup_semundo;
1303 1304
	retval = copy_fs(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1305
		goto bad_fork_cleanup_files;
1306 1307
	retval = copy_sighand(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1308
		goto bad_fork_cleanup_fs;
1309 1310
	retval = copy_signal(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1311
		goto bad_fork_cleanup_sighand;
1312 1313
	retval = copy_mm(clone_flags, p);
	if (retval)
L
Linus Torvalds 已提交
1314
		goto bad_fork_cleanup_signal;
1315 1316
	retval = copy_namespaces(clone_flags, p);
	if (retval)
D
David Howells 已提交
1317
		goto bad_fork_cleanup_mm;
1318 1319
	retval = copy_io(clone_flags, p);
	if (retval)
1320
		goto bad_fork_cleanup_namespaces;
A
Alexey Dobriyan 已提交
1321
	retval = copy_thread(clone_flags, stack_start, stack_size, p, regs);
L
Linus Torvalds 已提交
1322
	if (retval)
1323
		goto bad_fork_cleanup_io;
L
Linus Torvalds 已提交
1324

1325 1326
	if (pid != &init_struct_pid) {
		retval = -ENOMEM;
1327
		pid = alloc_pid(p->nsproxy->pid_ns);
1328
		if (!pid)
1329
			goto bad_fork_cleanup_io;
1330 1331 1332 1333 1334 1335 1336
	}

	p->pid = pid_nr(pid);
	p->tgid = p->pid;
	if (clone_flags & CLONE_THREAD)
		p->tgid = current->tgid;

L
Linus Torvalds 已提交
1337 1338 1339 1340
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
1341
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr : NULL;
1342 1343 1344
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1345
#ifdef CONFIG_FUTEX
1346 1347 1348 1349
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1350 1351
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1352
#endif
1353
	uprobe_copy_process(p);
1354 1355 1356 1357 1358 1359
	/*
	 * sigaltstack should be cleared when sharing the same VM
	 */
	if ((clone_flags & (CLONE_VM|CLONE_VFORK)) == CLONE_VM)
		p->sas_ss_sp = p->sas_ss_size = 0;

L
Linus Torvalds 已提交
1360
	/*
1361 1362
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1363
	 */
1364
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1365
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1366 1367 1368
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1369
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1370 1371

	/* ok, now we should be set up.. */
1372 1373 1374 1375 1376 1377 1378
	if (clone_flags & CLONE_THREAD)
		p->exit_signal = -1;
	else if (clone_flags & CLONE_PARENT)
		p->exit_signal = current->group_leader->exit_signal;
	else
		p->exit_signal = (clone_flags & CSIGNAL);

L
Linus Torvalds 已提交
1379 1380 1381
	p->pdeath_signal = 0;
	p->exit_state = 0;

1382 1383
	p->nr_dirtied = 0;
	p->nr_dirtied_pause = 128 >> (PAGE_SHIFT - 10);
1384
	p->dirty_paused_when = 0;
1385

L
Linus Torvalds 已提交
1386 1387 1388 1389 1390
	/*
	 * Ok, make it visible to the rest of the system.
	 * We dont wake it up yet.
	 */
	p->group_leader = p;
O
Oleg Nesterov 已提交
1391
	INIT_LIST_HEAD(&p->thread_group);
A
Al Viro 已提交
1392
	p->task_works = NULL;
L
Linus Torvalds 已提交
1393

1394 1395 1396 1397 1398 1399
	/* Now that the task is set up, run cgroup callbacks if
	 * necessary. We need to run them before the task is visible
	 * on the tasklist. */
	cgroup_fork_callbacks(p);
	cgroup_callbacks_done = 1;

L
Linus Torvalds 已提交
1400 1401 1402 1403
	/* Need tasklist lock for parent etc handling! */
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1404
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1405
		p->real_parent = current->real_parent;
1406 1407
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1408
		p->real_parent = current;
1409 1410
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1411

1412
	spin_lock(&current->sighand->siglock);
1413 1414 1415 1416 1417 1418 1419 1420

	/*
	 * Process group and session signals need to be delivered to just the
	 * parent before the fork or both the parent and the child after the
	 * fork. Restart if a signal comes in before we add the new process to
	 * it's process group.
	 * A fatal signal pending means that current will exit, so the new
	 * thread can't slip out of an OOM kill (or normal SIGKILL).
1421
	*/
D
Daniel Walker 已提交
1422
	recalc_sigpending();
1423 1424 1425 1426
	if (signal_pending(current)) {
		spin_unlock(&current->sighand->siglock);
		write_unlock_irq(&tasklist_lock);
		retval = -ERESTARTNOINTR;
1427
		goto bad_fork_free_pid;
1428 1429
	}

L
Linus Torvalds 已提交
1430
	if (clone_flags & CLONE_THREAD) {
1431
		current->signal->nr_threads++;
1432
		atomic_inc(&current->signal->live);
1433
		atomic_inc(&current->signal->sigcnt);
L
Linus Torvalds 已提交
1434
		p->group_leader = current->group_leader;
O
Oleg Nesterov 已提交
1435
		list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
L
Linus Torvalds 已提交
1436 1437
	}

1438
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1439
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1440 1441

		if (thread_group_leader(p)) {
1442
			if (is_child_reaper(pid)) {
1443
				ns_of_pid(pid)->child_reaper = p;
1444 1445
				p->signal->flags |= SIGNAL_UNKILLABLE;
			}
1446

1447
			p->signal->leader_pid = pid;
A
Alan Cox 已提交
1448
			p->signal->tty = tty_kref_get(current->signal->tty);
E
Eric W. Biederman 已提交
1449 1450
			attach_pid(p, PIDTYPE_PGID, task_pgrp(current));
			attach_pid(p, PIDTYPE_SID, task_session(current));
1451
			list_add_tail(&p->sibling, &p->real_parent->children);
1452
			list_add_tail_rcu(&p->tasks, &init_task.tasks);
1453
			__this_cpu_inc(process_counts);
1454
		}
1455
		attach_pid(p, PIDTYPE_PID, pid);
1456
		nr_threads++;
L
Linus Torvalds 已提交
1457 1458 1459
	}

	total_forks++;
1460
	spin_unlock(&current->sighand->siglock);
L
Linus Torvalds 已提交
1461
	write_unlock_irq(&tasklist_lock);
1462
	proc_fork_connector(p);
1463
	cgroup_post_fork(p);
1464
	if (clone_flags & CLONE_THREAD)
1465
		threadgroup_change_end(current);
1466
	perf_event_fork(p);
1467 1468 1469

	trace_task_newtask(p, clone_flags);

L
Linus Torvalds 已提交
1470 1471
	return p;

1472 1473 1474
bad_fork_free_pid:
	if (pid != &init_struct_pid)
		free_pid(pid);
1475
bad_fork_cleanup_io:
1476 1477
	if (p->io_context)
		exit_io_context(p);
S
Serge E. Hallyn 已提交
1478
bad_fork_cleanup_namespaces:
1479
	exit_task_namespaces(p);
L
Linus Torvalds 已提交
1480
bad_fork_cleanup_mm:
D
David Rientjes 已提交
1481
	if (p->mm)
L
Linus Torvalds 已提交
1482 1483
		mmput(p->mm);
bad_fork_cleanup_signal:
1484
	if (!(clone_flags & CLONE_THREAD))
1485
		free_signal_struct(p->signal);
L
Linus Torvalds 已提交
1486
bad_fork_cleanup_sighand:
1487
	__cleanup_sighand(p->sighand);
L
Linus Torvalds 已提交
1488 1489 1490 1491 1492 1493 1494 1495 1496
bad_fork_cleanup_fs:
	exit_fs(p); /* blocking */
bad_fork_cleanup_files:
	exit_files(p); /* blocking */
bad_fork_cleanup_semundo:
	exit_sem(p);
bad_fork_cleanup_audit:
	audit_free(p);
bad_fork_cleanup_policy:
1497
	perf_event_free_task(p);
L
Linus Torvalds 已提交
1498
#ifdef CONFIG_NUMA
1499
	mpol_put(p->mempolicy);
1500
bad_fork_cleanup_cgroup:
L
Linus Torvalds 已提交
1501
#endif
1502
	if (clone_flags & CLONE_THREAD)
1503
		threadgroup_change_end(current);
1504
	cgroup_exit(p, cgroup_callbacks_done);
1505
	delayacct_tsk_free(p);
A
Al Viro 已提交
1506
	module_put(task_thread_info(p)->exec_domain->module);
L
Linus Torvalds 已提交
1507
bad_fork_cleanup_count:
D
David Howells 已提交
1508
	atomic_dec(&p->cred->user->processes);
1509
	exit_creds(p);
L
Linus Torvalds 已提交
1510 1511
bad_fork_free:
	free_task(p);
1512 1513
fork_out:
	return ERR_PTR(retval);
L
Linus Torvalds 已提交
1514 1515
}

A
Adrian Bunk 已提交
1516
noinline struct pt_regs * __cpuinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
L
Linus Torvalds 已提交
1517 1518 1519 1520 1521
{
	memset(regs, 0, sizeof(struct pt_regs));
	return regs;
}

1522 1523 1524 1525 1526 1527 1528 1529 1530 1531
static inline void init_idle_pids(struct pid_link *links)
{
	enum pid_type type;

	for (type = PIDTYPE_PID; type < PIDTYPE_MAX; ++type) {
		INIT_HLIST_NODE(&links[type].node); /* not really needed */
		links[type].pid = &init_struct_pid;
	}
}

1532
struct task_struct * __cpuinit fork_idle(int cpu)
L
Linus Torvalds 已提交
1533
{
1534
	struct task_struct *task;
L
Linus Torvalds 已提交
1535 1536
	struct pt_regs regs;

1537
	task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL,
R
Roland McGrath 已提交
1538
			    &init_struct_pid, 0);
1539 1540
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
1541
		init_idle(task, cpu);
1542
	}
1543

L
Linus Torvalds 已提交
1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561
	return task;
}

/*
 *  Ok, this is the main fork-routine.
 *
 * It copies the process, and if successful kick-starts
 * it and waits for it to finish using the VM if required.
 */
long do_fork(unsigned long clone_flags,
	      unsigned long stack_start,
	      struct pt_regs *regs,
	      unsigned long stack_size,
	      int __user *parent_tidptr,
	      int __user *child_tidptr)
{
	struct task_struct *p;
	int trace = 0;
1562
	long nr;
L
Linus Torvalds 已提交
1563

1564 1565 1566 1567
	/*
	 * Do some preliminary argument and permissions checking before we
	 * actually start allocating stuff
	 */
1568 1569
	if (clone_flags & (CLONE_NEWUSER | CLONE_NEWPID)) {
		if (clone_flags & (CLONE_THREAD|CLONE_PARENT))
1570
			return -EINVAL;
1571
	}
1572

R
Roland McGrath 已提交
1573
	/*
T
Tejun Heo 已提交
1574 1575 1576 1577
	 * Determine whether and which event to report to ptracer.  When
	 * called from kernel_thread or CLONE_UNTRACED is explicitly
	 * requested, no event is reported; otherwise, report if the event
	 * for the type of forking is enabled.
R
Roland McGrath 已提交
1578
	 */
1579
	if (!(clone_flags & CLONE_UNTRACED) && likely(user_mode(regs))) {
T
Tejun Heo 已提交
1580 1581 1582 1583 1584 1585 1586 1587 1588 1589
		if (clone_flags & CLONE_VFORK)
			trace = PTRACE_EVENT_VFORK;
		else if ((clone_flags & CSIGNAL) != SIGCHLD)
			trace = PTRACE_EVENT_CLONE;
		else
			trace = PTRACE_EVENT_FORK;

		if (likely(!ptrace_event_enabled(current, trace)))
			trace = 0;
	}
L
Linus Torvalds 已提交
1590

1591
	p = copy_process(clone_flags, stack_start, regs, stack_size,
R
Roland McGrath 已提交
1592
			 child_tidptr, NULL, trace);
L
Linus Torvalds 已提交
1593 1594 1595 1596 1597 1598 1599
	/*
	 * Do this prior waking up the new thread - the thread pointer
	 * might get invalid after that point, if the thread exits quickly.
	 */
	if (!IS_ERR(p)) {
		struct completion vfork;

1600 1601
		trace_sched_process_fork(current, p);

1602
		nr = task_pid_vnr(p);
1603 1604 1605

		if (clone_flags & CLONE_PARENT_SETTID)
			put_user(nr, parent_tidptr);
1606

L
Linus Torvalds 已提交
1607 1608 1609
		if (clone_flags & CLONE_VFORK) {
			p->vfork_done = &vfork;
			init_completion(&vfork);
O
Oleg Nesterov 已提交
1610
			get_task_struct(p);
L
Linus Torvalds 已提交
1611 1612
		}

1613
		wake_up_new_task(p);
L
Linus Torvalds 已提交
1614

T
Tejun Heo 已提交
1615 1616 1617
		/* forking complete and child started to run, tell ptracer */
		if (unlikely(trace))
			ptrace_event(trace, nr);
R
Roland McGrath 已提交
1618

L
Linus Torvalds 已提交
1619
		if (clone_flags & CLONE_VFORK) {
O
Oleg Nesterov 已提交
1620 1621
			if (!wait_for_vfork_done(p, &vfork))
				ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
L
Linus Torvalds 已提交
1622 1623
		}
	} else {
1624
		nr = PTR_ERR(p);
L
Linus Torvalds 已提交
1625
	}
1626
	return nr;
L
Linus Torvalds 已提交
1627 1628
}

1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639
#ifdef CONFIG_GENERIC_KERNEL_THREAD
/*
 * Create a kernel thread.
 */
pid_t kernel_thread(int (*fn)(void *), void *arg, unsigned long flags)
{
	return do_fork(flags|CLONE_VM|CLONE_UNTRACED, (unsigned long)fn, NULL,
		(unsigned long)arg, NULL, NULL);
}
#endif

1640 1641 1642 1643
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

1644
static void sighand_ctor(void *data)
1645 1646 1647
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
1648
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
1649
	init_waitqueue_head(&sighand->signalfd_wqh);
1650 1651
}

L
Linus Torvalds 已提交
1652 1653 1654 1655
void __init proc_caches_init(void)
{
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
V
Vegard Nossum 已提交
1656 1657
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU|
			SLAB_NOTRACK, sighand_ctor);
L
Linus Torvalds 已提交
1658 1659
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
V
Vegard Nossum 已提交
1660
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1661
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
1662
			sizeof(struct files_struct), 0,
V
Vegard Nossum 已提交
1663
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1664
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
1665
			sizeof(struct fs_struct), 0,
V
Vegard Nossum 已提交
1666
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1667 1668 1669 1670 1671 1672 1673
	/*
	 * FIXME! The "sizeof(struct mm_struct)" currently includes the
	 * whole struct cpumask for the OFFSTACK case. We could change
	 * this to *only* allocate as much of it as required by the
	 * maximum number of CPU's we can ever have.  The cpumask_allocation
	 * is at the end of the structure, exactly for that reason.
	 */
L
Linus Torvalds 已提交
1674
	mm_cachep = kmem_cache_create("mm_struct",
1675
			sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
V
Vegard Nossum 已提交
1676
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1677
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC);
1678
	mmap_init();
1679
	nsproxy_cache_init();
L
Linus Torvalds 已提交
1680
}
1681 1682

/*
1683
 * Check constraints on flags passed to the unshare system call.
1684
 */
1685
static int check_unshare_flags(unsigned long unshare_flags)
1686
{
1687 1688
	if (unshare_flags & ~(CLONE_THREAD|CLONE_FS|CLONE_NEWNS|CLONE_SIGHAND|
				CLONE_VM|CLONE_FILES|CLONE_SYSVSEM|
1689 1690
				CLONE_NEWUTS|CLONE_NEWIPC|CLONE_NEWNET|
				CLONE_NEWPID))
1691
		return -EINVAL;
1692
	/*
1693 1694 1695
	 * Not implemented, but pretend it works if there is nothing to
	 * unshare. Note that unsharing CLONE_THREAD or CLONE_SIGHAND
	 * needs to unshare vm.
1696
	 */
1697 1698 1699 1700 1701
	if (unshare_flags & (CLONE_THREAD | CLONE_SIGHAND | CLONE_VM)) {
		/* FIXME: get_task_mm() increments ->mm_users */
		if (atomic_read(&current->mm->mm_users) > 1)
			return -EINVAL;
	}
1702 1703 1704 1705 1706

	return 0;
}

/*
1707
 * Unshare the filesystem structure if it is being shared
1708 1709 1710 1711 1712
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
1713 1714 1715 1716 1717 1718 1719 1720 1721 1722
	if (!(unshare_flags & CLONE_FS) || !fs)
		return 0;

	/* don't need lock here; in the worst case we'll do useless copy */
	if (fs->users == 1)
		return 0;

	*new_fsp = copy_fs_struct(fs);
	if (!*new_fsp)
		return -ENOMEM;
1723 1724 1725 1726 1727

	return 0;
}

/*
1728
 * Unshare file descriptor table if it is being shared
1729 1730 1731 1732
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
1733
	int error = 0;
1734 1735

	if ((unshare_flags & CLONE_FILES) &&
1736 1737 1738 1739 1740
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752

	return 0;
}

/*
 * unshare allows a process to 'unshare' part of the process
 * context which was originally shared using clone.  copy_*
 * functions used by do_fork() cannot be used here directly
 * because they modify an inactive task_struct that is being
 * constructed. Here we are modifying the current, active,
 * task_struct.
 */
1753
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
1754 1755 1756
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
1757
	struct nsproxy *new_nsproxy = NULL;
1758
	int do_sysvsem = 0;
1759
	int err;
1760

1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775
	/*
	 * If unsharing a pid namespace must also unshare the thread.
	 */
	if (unshare_flags & CLONE_NEWPID)
		unshare_flags |= CLONE_THREAD;
	/*
	 * If unsharing a thread from a thread group, must also unshare vm.
	 */
	if (unshare_flags & CLONE_THREAD)
		unshare_flags |= CLONE_VM;
	/*
	 * If unsharing vm, must also unshare signal handlers.
	 */
	if (unshare_flags & CLONE_VM)
		unshare_flags |= CLONE_SIGHAND;
1776 1777 1778 1779 1780
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
1781 1782 1783 1784

	err = check_unshare_flags(unshare_flags);
	if (err)
		goto bad_unshare_out;
1785 1786 1787 1788 1789 1790
	/*
	 * CLONE_NEWIPC must also detach from the undolist: after switching
	 * to a new ipc namespace, the semaphore arrays from the old
	 * namespace are unreachable.
	 */
	if (unshare_flags & (CLONE_NEWIPC|CLONE_SYSVSEM))
1791
		do_sysvsem = 1;
1792 1793
	err = unshare_fs(unshare_flags, &new_fs);
	if (err)
1794
		goto bad_unshare_out;
1795 1796
	err = unshare_fd(unshare_flags, &new_fd);
	if (err)
1797
		goto bad_unshare_cleanup_fs;
1798 1799
	err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy, new_fs);
	if (err)
1800
		goto bad_unshare_cleanup_fd;
1801

1802
	if (new_fs || new_fd || do_sysvsem || new_nsproxy) {
1803 1804 1805 1806 1807 1808
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
S
Serge E. Hallyn 已提交
1809

1810
		if (new_nsproxy) {
1811 1812
			switch_task_namespaces(current, new_nsproxy);
			new_nsproxy = NULL;
1813
		}
1814

1815 1816
		task_lock(current);

1817 1818
		if (new_fs) {
			fs = current->fs;
N
Nick Piggin 已提交
1819
			spin_lock(&fs->lock);
1820
			current->fs = new_fs;
A
Al Viro 已提交
1821 1822 1823 1824
			if (--fs->users)
				new_fs = NULL;
			else
				new_fs = fs;
N
Nick Piggin 已提交
1825
			spin_unlock(&fs->lock);
1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836
		}

		if (new_fd) {
			fd = current->files;
			current->files = new_fd;
			new_fd = fd;
		}

		task_unlock(current);
	}

1837
	if (new_nsproxy)
1838
		put_nsproxy(new_nsproxy);
1839

1840 1841 1842 1843 1844 1845
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
1846
		free_fs_struct(new_fs);
1847 1848 1849 1850

bad_unshare_out:
	return err;
}
1851 1852 1853 1854 1855 1856 1857 1858 1859 1860

/*
 *	Helper to unshare the files of the current task.
 *	We don't want to expose copy_files internals to
 *	the exec layer of the kernel.
 */

int unshare_files(struct files_struct **displaced)
{
	struct task_struct *task = current;
1861
	struct files_struct *copy = NULL;
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874
	int error;

	error = unshare_fd(CLONE_FILES, &copy);
	if (error || !copy) {
		*displaced = NULL;
		return error;
	}
	*displaced = task->files;
	task_lock(task);
	task->files = copy;
	task_unlock(task);
	return 0;
}