fork.c 43.5 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>
L
Linus Torvalds 已提交
37 38 39 40
#include <linux/swap.h>
#include <linux/syscalls.h>
#include <linux/jiffies.h>
#include <linux/futex.h>
41
#include <linux/compat.h>
42
#include <linux/kthread.h>
43
#include <linux/task_io_accounting_ops.h>
44
#include <linux/rcupdate.h>
L
Linus Torvalds 已提交
45 46 47
#include <linux/ptrace.h>
#include <linux/mount.h>
#include <linux/audit.h>
48
#include <linux/memcontrol.h>
49
#include <linux/ftrace.h>
50
#include <linux/proc_fs.h>
L
Linus Torvalds 已提交
51 52
#include <linux/profile.h>
#include <linux/rmap.h>
H
Hugh Dickins 已提交
53
#include <linux/ksm.h>
L
Linus Torvalds 已提交
54
#include <linux/acct.h>
55
#include <linux/tsacct_kern.h>
M
Matt Helsley 已提交
56
#include <linux/cn_proc.h>
R
Rafael J. Wysocki 已提交
57
#include <linux/freezer.h>
58
#include <linux/delayacct.h>
59
#include <linux/taskstats_kern.h>
60
#include <linux/random.h>
M
Miloslav Trmac 已提交
61
#include <linux/tty.h>
62
#include <linux/blkdev.h>
63
#include <linux/fs_struct.h>
64
#include <linux/magic.h>
65
#include <linux/perf_event.h>
66
#include <linux/posix-timers.h>
67
#include <linux/user-return-notifier.h>
Y
Ying Han 已提交
68
#include <linux/oom.h>
A
Andrea Arcangeli 已提交
69
#include <linux/khugepaged.h>
70
#include <linux/signalfd.h>
L
Linus Torvalds 已提交
71 72 73 74 75 76 77 78

#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>

79 80
#include <trace/events/sched.h>

81 82 83
#define CREATE_TRACE_POINTS
#include <trace/events/task.h>

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

int max_threads;		/* tunable limit on nr_threads */

DEFINE_PER_CPU(unsigned long, process_counts) = 0;

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

#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 已提交
103 104 105 106 107 108

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

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

	return total;
}

#ifndef __HAVE_ARCH_TASK_STRUCT_ALLOCATOR
116 117 118 119
# define alloc_task_struct_node(node)		\
		kmem_cache_alloc_node(task_struct_cachep, GFP_KERNEL, node)
# define free_task_struct(tsk)			\
		kmem_cache_free(task_struct_cachep, (tsk))
120
static struct kmem_cache *task_struct_cachep;
L
Linus Torvalds 已提交
121 122
#endif

123
#ifndef __HAVE_ARCH_THREAD_INFO_ALLOCATOR
124 125
static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
126 127 128 129 130 131
{
#ifdef CONFIG_DEBUG_STACK_USAGE
	gfp_t mask = GFP_KERNEL | __GFP_ZERO;
#else
	gfp_t mask = GFP_KERNEL;
#endif
132 133 134
	struct page *page = alloc_pages_node(node, mask, THREAD_SIZE_ORDER);

	return page ? page_address(page) : NULL;
135 136 137 138 139 140 141 142
}

static inline void free_thread_info(struct thread_info *ti)
{
	free_pages((unsigned long)ti, THREAD_SIZE_ORDER);
}
#endif

L
Linus Torvalds 已提交
143
/* SLAB cache for signal_struct structures (tsk->signal) */
144
static struct kmem_cache *signal_cachep;
L
Linus Torvalds 已提交
145 146

/* SLAB cache for sighand_struct structures (tsk->sighand) */
147
struct kmem_cache *sighand_cachep;
L
Linus Torvalds 已提交
148 149

/* SLAB cache for files_struct structures (tsk->files) */
150
struct kmem_cache *files_cachep;
L
Linus Torvalds 已提交
151 152

/* SLAB cache for fs_struct structures (tsk->fs) */
153
struct kmem_cache *fs_cachep;
L
Linus Torvalds 已提交
154 155

/* SLAB cache for vm_area_struct structures */
156
struct kmem_cache *vm_area_cachep;
L
Linus Torvalds 已提交
157 158

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

161 162 163 164 165 166 167
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 已提交
168 169
void free_task(struct task_struct *tsk)
{
170
	account_kernel_stack(tsk->stack, -1);
R
Roman Zippel 已提交
171
	free_thread_info(tsk->stack);
I
Ingo Molnar 已提交
172
	rt_mutex_debug_task_free(tsk);
173
	ftrace_graph_exit_task(tsk);
L
Linus Torvalds 已提交
174 175 176 177
	free_task_struct(tsk);
}
EXPORT_SYMBOL(free_task);

178 179
static inline void free_signal_struct(struct signal_struct *sig)
{
180
	taskstats_tgid_free(sig);
181
	sched_autogroup_exit(sig);
182 183 184 185 186
	kmem_cache_free(signal_cachep, sig);
}

static inline void put_signal_struct(struct signal_struct *sig)
{
187
	if (atomic_dec_and_test(&sig->sigcnt))
188 189 190
		free_signal_struct(sig);
}

191
void __put_task_struct(struct task_struct *tsk)
L
Linus Torvalds 已提交
192
{
E
Eugene Teo 已提交
193
	WARN_ON(!tsk->exit_state);
L
Linus Torvalds 已提交
194 195 196
	WARN_ON(atomic_read(&tsk->usage));
	WARN_ON(tsk == current);

197
	security_task_free(tsk);
198
	exit_creds(tsk);
199
	delayacct_tsk_free(tsk);
200
	put_signal_struct(tsk->signal);
L
Linus Torvalds 已提交
201 202 203 204

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

S
Suresh Siddha 已提交
207 208 209 210 211 212 213
/*
 * macro override instead of weak attribute alias, to workaround
 * gcc 4.1.0 and 4.1.1 bugs with weak attribute and empty functions.
 */
#ifndef arch_task_cache_init
#define arch_task_cache_init()
#endif
214

L
Linus Torvalds 已提交
215 216 217 218 219 220 221 222 223
void __init fork_init(unsigned long mempages)
{
#ifndef __HAVE_ARCH_TASK_STRUCT_ALLOCATOR
#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 已提交
224
			ARCH_MIN_TASKALIGN, SLAB_PANIC | SLAB_NOTRACK, NULL);
L
Linus Torvalds 已提交
225 226
#endif

227 228 229
	/* do the arch specific task caches init */
	arch_task_cache_init();

L
Linus Torvalds 已提交
230 231 232 233 234 235 236 237 238 239
	/*
	 * 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
	 */
240
	if (max_threads < 20)
L
Linus Torvalds 已提交
241 242 243 244 245 246 247 248
		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];
}

249 250 251 252 253 254 255
int __attribute__((weak)) arch_dup_task_struct(struct task_struct *dst,
					       struct task_struct *src)
{
	*dst = *src;
	return 0;
}

L
Linus Torvalds 已提交
256 257 258 259
static struct task_struct *dup_task_struct(struct task_struct *orig)
{
	struct task_struct *tsk;
	struct thread_info *ti;
260
	unsigned long *stackend;
261
	int node = tsk_fork_get_node(orig);
P
Peter Zijlstra 已提交
262
	int err;
L
Linus Torvalds 已提交
263

264
	tsk = alloc_task_struct_node(node);
L
Linus Torvalds 已提交
265 266 267
	if (!tsk)
		return NULL;

268
	ti = alloc_thread_info_node(tsk, node);
L
Linus Torvalds 已提交
269 270 271 272 273
	if (!ti) {
		free_task_struct(tsk);
		return NULL;
	}

274
	err = arch_dup_task_struct(tsk, orig);
275 276 277
	if (err)
		goto out;

R
Roman Zippel 已提交
278
	tsk->stack = ti;
P
Peter Zijlstra 已提交
279

280
	setup_thread_stack(tsk, orig);
281
	clear_user_return_notifier(tsk);
282
	clear_tsk_need_resched(tsk);
283 284
	stackend = end_of_stack(tsk);
	*stackend = STACK_END_MAGIC;	/* for overflow detection */
L
Linus Torvalds 已提交
285

286 287 288 289
#ifdef CONFIG_CC_STACKPROTECTOR
	tsk->stack_canary = get_random_int();
#endif

290 291 292 293 294
	/*
	 * One for us, one for whoever does the "release_task()" (usually
	 * parent)
	 */
	atomic_set(&tsk->usage, 2);
295
#ifdef CONFIG_BLK_DEV_IO_TRACE
296
	tsk->btrace_seq = 0;
297
#endif
298
	tsk->splice_pipe = NULL;
299 300 301

	account_kernel_stack(ti, 1);

L
Linus Torvalds 已提交
302
	return tsk;
303 304 305 306 307

out:
	free_thread_info(ti);
	free_task_struct(tsk);
	return NULL;
L
Linus Torvalds 已提交
308 309 310
}

#ifdef CONFIG_MMU
A
Alexey Dobriyan 已提交
311
static int dup_mmap(struct mm_struct *mm, struct mm_struct *oldmm)
L
Linus Torvalds 已提交
312
{
313
	struct vm_area_struct *mpnt, *tmp, *prev, **pprev;
L
Linus Torvalds 已提交
314 315 316 317 318 319
	struct rb_node **rb_link, *rb_parent;
	int retval;
	unsigned long charge;
	struct mempolicy *pol;

	down_write(&oldmm->mmap_sem);
320
	flush_cache_dup_mm(oldmm);
321 322 323 324
	/*
	 * Not linked in yet - no deadlock potential:
	 */
	down_write_nested(&mm->mmap_sem, SINGLE_DEPTH_NESTING);
325

L
Linus Torvalds 已提交
326 327 328 329
	mm->locked_vm = 0;
	mm->mmap = NULL;
	mm->mmap_cache = NULL;
	mm->free_area_cache = oldmm->mmap_base;
330
	mm->cached_hole_size = ~0UL;
L
Linus Torvalds 已提交
331
	mm->map_count = 0;
332
	cpumask_clear(mm_cpumask(mm));
L
Linus Torvalds 已提交
333 334 335 336
	mm->mm_rb = RB_ROOT;
	rb_link = &mm->mm_rb.rb_node;
	rb_parent = NULL;
	pprev = &mm->mmap;
H
Hugh Dickins 已提交
337
	retval = ksm_fork(mm, oldmm);
A
Andrea Arcangeli 已提交
338 339 340
	if (retval)
		goto out;
	retval = khugepaged_fork(mm, oldmm);
H
Hugh Dickins 已提交
341 342
	if (retval)
		goto out;
L
Linus Torvalds 已提交
343

344
	prev = NULL;
H
Hugh Dickins 已提交
345
	for (mpnt = oldmm->mmap; mpnt; mpnt = mpnt->vm_next) {
L
Linus Torvalds 已提交
346 347 348
		struct file *file;

		if (mpnt->vm_flags & VM_DONTCOPY) {
349 350
			long pages = vma_pages(mpnt);
			mm->total_vm -= pages;
351
			vm_stat_account(mm, mpnt->vm_flags, mpnt->vm_file,
352
								-pages);
L
Linus Torvalds 已提交
353 354 355 356 357
			continue;
		}
		charge = 0;
		if (mpnt->vm_flags & VM_ACCOUNT) {
			unsigned int len = (mpnt->vm_end - mpnt->vm_start) >> PAGE_SHIFT;
358
			if (security_vm_enough_memory_mm(oldmm, len)) /* sic */
L
Linus Torvalds 已提交
359 360 361
				goto fail_nomem;
			charge = len;
		}
362
		tmp = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
363 364 365
		if (!tmp)
			goto fail_nomem;
		*tmp = *mpnt;
366
		INIT_LIST_HEAD(&tmp->anon_vma_chain);
367
		pol = mpol_dup(vma_policy(mpnt));
L
Linus Torvalds 已提交
368 369 370 371
		retval = PTR_ERR(pol);
		if (IS_ERR(pol))
			goto fail_nomem_policy;
		vma_set_policy(tmp, pol);
A
Andrea Arcangeli 已提交
372
		tmp->vm_mm = mm;
373 374
		if (anon_vma_fork(tmp, mpnt))
			goto fail_nomem_anon_vma_fork;
L
Linus Torvalds 已提交
375
		tmp->vm_flags &= ~VM_LOCKED;
376
		tmp->vm_next = tmp->vm_prev = NULL;
L
Linus Torvalds 已提交
377 378
		file = tmp->vm_file;
		if (file) {
379
			struct inode *inode = file->f_path.dentry->d_inode;
H
Hugh Dickins 已提交
380 381
			struct address_space *mapping = file->f_mapping;

L
Linus Torvalds 已提交
382 383 384
			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
385
			mutex_lock(&mapping->i_mmap_mutex);
H
Hugh Dickins 已提交
386 387 388 389
			if (tmp->vm_flags & VM_SHARED)
				mapping->i_mmap_writable++;
			flush_dcache_mmap_lock(mapping);
			/* insert tmp into the share list, just after mpnt */
L
Linus Torvalds 已提交
390
			vma_prio_tree_add(tmp, mpnt);
H
Hugh Dickins 已提交
391
			flush_dcache_mmap_unlock(mapping);
392
			mutex_unlock(&mapping->i_mmap_mutex);
L
Linus Torvalds 已提交
393 394
		}

395 396 397 398 399 400 401 402
		/*
		 * 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 已提交
403
		/*
404
		 * Link in the new vma and copy the page table entries.
L
Linus Torvalds 已提交
405 406 407
		 */
		*pprev = tmp;
		pprev = &tmp->vm_next;
408 409
		tmp->vm_prev = prev;
		prev = tmp;
L
Linus Torvalds 已提交
410 411 412 413 414 415

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

		mm->map_count++;
416
		retval = copy_page_range(mm, oldmm, mpnt);
L
Linus Torvalds 已提交
417 418 419 420 421 422 423

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

		if (retval)
			goto out;
	}
424 425
	/* a new mm has just been created */
	arch_dup_mmap(oldmm, mm);
L
Linus Torvalds 已提交
426 427
	retval = 0;
out:
428
	up_write(&mm->mmap_sem);
H
Hugh Dickins 已提交
429
	flush_tlb_mm(oldmm);
L
Linus Torvalds 已提交
430 431
	up_write(&oldmm->mmap_sem);
	return retval;
432 433
fail_nomem_anon_vma_fork:
	mpol_put(pol);
L
Linus Torvalds 已提交
434 435 436 437 438 439 440 441
fail_nomem_policy:
	kmem_cache_free(vm_area_cachep, tmp);
fail_nomem:
	retval = -ENOMEM;
	vm_unacct_memory(charge);
	goto out;
}

442
static inline int mm_alloc_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
443 444 445 446 447 448 449
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

450
static inline void mm_free_pgd(struct mm_struct *mm)
L
Linus Torvalds 已提交
451
{
452
	pgd_free(mm, mm->pgd);
L
Linus Torvalds 已提交
453 454 455 456 457 458 459
}
#else
#define dup_mmap(mm, oldmm)	(0)
#define mm_alloc_pgd(mm)	(0)
#define mm_free_pgd(mm)
#endif /* CONFIG_MMU */

D
Daniel Walker 已提交
460
__cacheline_aligned_in_smp DEFINE_SPINLOCK(mmlist_lock);
L
Linus Torvalds 已提交
461

462
#define allocate_mm()	(kmem_cache_alloc(mm_cachep, GFP_KERNEL))
L
Linus Torvalds 已提交
463 464
#define free_mm(mm)	(kmem_cache_free(mm_cachep, (mm)))

465 466 467 468 469 470 471 472 473 474 475 476
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 已提交
477 478
#include <linux/init_task.h>

A
Alexey Dobriyan 已提交
479 480 481 482 483 484 485 486
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
}

487
static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p)
L
Linus Torvalds 已提交
488 489 490 491 492
{
	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 已提交
493 494
	mm->flags = (current->mm) ?
		(current->mm->flags & MMF_INIT_MASK) : default_dump_filter;
495
	mm->core_state = NULL;
L
Linus Torvalds 已提交
496
	mm->nr_ptes = 0;
K
KAMEZAWA Hiroyuki 已提交
497
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
498 499
	spin_lock_init(&mm->page_table_lock);
	mm->free_area_cache = TASK_UNMAPPED_BASE;
500
	mm->cached_hole_size = ~0UL;
A
Alexey Dobriyan 已提交
501
	mm_init_aio(mm);
502
	mm_init_owner(mm, p);
L
Linus Torvalds 已提交
503 504 505

	if (likely(!mm_alloc_pgd(mm))) {
		mm->def_flags = 0;
A
Andrea Arcangeli 已提交
506
		mmu_notifier_mm_init(mm);
L
Linus Torvalds 已提交
507 508
		return mm;
	}
509

L
Linus Torvalds 已提交
510 511 512 513
	free_mm(mm);
	return NULL;
}

514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530
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 已提交
531 532 533
/*
 * Allocate and initialize an mm_struct.
 */
534
struct mm_struct *mm_alloc(void)
L
Linus Torvalds 已提交
535
{
536
	struct mm_struct *mm;
L
Linus Torvalds 已提交
537 538

	mm = allocate_mm();
539 540 541 542
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
543 544
	mm_init_cpumask(mm);
	return mm_init(mm, current);
L
Linus Torvalds 已提交
545 546 547 548 549 550 551
}

/*
 * 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.
 */
552
void __mmdrop(struct mm_struct *mm)
L
Linus Torvalds 已提交
553 554 555 556
{
	BUG_ON(mm == &init_mm);
	mm_free_pgd(mm);
	destroy_context(mm);
A
Andrea Arcangeli 已提交
557
	mmu_notifier_mm_destroy(mm);
558
	check_mm(mm);
L
Linus Torvalds 已提交
559 560
	free_mm(mm);
}
561
EXPORT_SYMBOL_GPL(__mmdrop);
L
Linus Torvalds 已提交
562 563 564 565 566 567

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

L
Linus Torvalds 已提交
570 571
	if (atomic_dec_and_test(&mm->mm_users)) {
		exit_aio(mm);
572
		ksm_exit(mm);
A
Andrea Arcangeli 已提交
573
		khugepaged_exit(mm); /* must run before exit_mmap */
L
Linus Torvalds 已提交
574
		exit_mmap(mm);
M
Matt Helsley 已提交
575
		set_mm_exe_file(mm, NULL);
L
Linus Torvalds 已提交
576 577 578 579 580 581
		if (!list_empty(&mm->mmlist)) {
			spin_lock(&mmlist_lock);
			list_del(&mm->mmlist);
			spin_unlock(&mmlist_lock);
		}
		put_swap_token(mm);
582 583
		if (mm->binfmt)
			module_put(mm->binfmt->module);
L
Linus Torvalds 已提交
584 585 586 587 588
		mmdrop(mm);
	}
}
EXPORT_SYMBOL_GPL(mmput);

589 590 591 592 593 594 595 596 597 598 599 600 601
/*
 * We added or removed a vma mapping the executable. The vmas are only mapped
 * during exec and are not mapped with the mmap system call.
 * Callers must hold down_write() on the mm's mmap_sem for these
 */
void added_exe_file_vma(struct mm_struct *mm)
{
	mm->num_exe_file_vmas++;
}

void removed_exe_file_vma(struct mm_struct *mm)
{
	mm->num_exe_file_vmas--;
602
	if ((mm->num_exe_file_vmas == 0) && mm->exe_file) {
603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
		fput(mm->exe_file);
		mm->exe_file = NULL;
	}

}

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;
	mm->num_exe_file_vmas = 0;
}

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

	/* We need mmap_sem to protect against races with removal of
	 * VM_EXECUTABLE vmas */
	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 已提交
640 641 642
/**
 * get_task_mm - acquire a reference to the task's mm
 *
643
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
644 645 646 647 648 649 650 651 652 653 654 655
 * 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) {
656
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
657 658 659 660 661 662 663 664 665
			mm = NULL;
		else
			atomic_inc(&mm->mm_users);
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
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;
}

686
static void complete_vfork_done(struct task_struct *tsk)
687
{
O
Oleg Nesterov 已提交
688
	struct completion *vfork;
689

O
Oleg Nesterov 已提交
690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715
	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;
716 717
}

L
Linus Torvalds 已提交
718 719 720 721 722 723 724 725 726 727 728 729 730 731 732
/* 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)
{
733 734
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
735
	if (unlikely(tsk->robust_list)) {
736
		exit_robust_list(tsk);
737 738
		tsk->robust_list = NULL;
	}
739
#ifdef CONFIG_COMPAT
740
	if (unlikely(tsk->compat_robust_list)) {
741
		compat_exit_robust_list(tsk);
742 743
		tsk->compat_robust_list = NULL;
	}
744
#endif
745 746
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
747 748
#endif

L
Linus Torvalds 已提交
749 750 751
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

752 753
	if (tsk->vfork_done)
		complete_vfork_done(tsk);
754 755 756 757 758

	/*
	 * 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 已提交
759 760
	 * trouble, say, a killed vfork parent shouldn't touch this mm.
	 * Userland only wants this done for a sys_exit.
761
	 */
762 763 764 765 766 767 768 769 770 771 772
	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 已提交
773 774 775 776
		tsk->clear_child_tid = NULL;
	}
}

777 778 779 780
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
781
struct mm_struct *dup_mm(struct task_struct *tsk)
782 783 784 785 786 787 788 789 790 791 792 793
{
	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));
794
	mm_init_cpumask(mm);
795

796 797 798 799
	/* Initializing for Swap token stuff */
	mm->token_priority = 0;
	mm->last_interval = 0;

800 801 802 803
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	mm->pmd_huge_pte = NULL;
#endif

804
	if (!mm_init(mm, tsk))
805 806 807 808 809
		goto fail_nomem;

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

M
Matt Helsley 已提交
810 811
	dup_mm_exe_file(oldmm, mm);

812 813 814 815 816 817 818
	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

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

819 820 821
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

822 823 824
	return mm;

free_pt:
825 826
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
827 828 829 830 831 832 833 834 835 836 837 838 839 840 841
	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;
}

842
static int copy_mm(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
843
{
844
	struct mm_struct *mm, *oldmm;
L
Linus Torvalds 已提交
845 846 847 848
	int retval;

	tsk->min_flt = tsk->maj_flt = 0;
	tsk->nvcsw = tsk->nivcsw = 0;
849 850 851
#ifdef CONFIG_DETECT_HUNG_TASK
	tsk->last_switch_count = tsk->nvcsw + tsk->nivcsw;
#endif
L
Linus Torvalds 已提交
852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871

	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;
872
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
873 874 875 876
	if (!mm)
		goto fail_nomem;

good_mm:
877 878 879 880
	/* Initializing for Swap token stuff */
	mm->token_priority = 0;
	mm->last_interval = 0;

L
Linus Torvalds 已提交
881 882 883 884 885 886 887 888
	tsk->mm = mm;
	tsk->active_mm = mm;
	return 0;

fail_nomem:
	return retval;
}

A
Alexey Dobriyan 已提交
889
static int copy_fs(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
890
{
A
Al Viro 已提交
891
	struct fs_struct *fs = current->fs;
L
Linus Torvalds 已提交
892
	if (clone_flags & CLONE_FS) {
A
Al Viro 已提交
893
		/* tsk->fs is already what we want */
N
Nick Piggin 已提交
894
		spin_lock(&fs->lock);
A
Al Viro 已提交
895
		if (fs->in_exec) {
N
Nick Piggin 已提交
896
			spin_unlock(&fs->lock);
A
Al Viro 已提交
897 898 899
			return -EAGAIN;
		}
		fs->users++;
N
Nick Piggin 已提交
900
		spin_unlock(&fs->lock);
L
Linus Torvalds 已提交
901 902
		return 0;
	}
A
Al Viro 已提交
903
	tsk->fs = copy_fs_struct(fs);
L
Linus Torvalds 已提交
904 905 906 907 908
	if (!tsk->fs)
		return -ENOMEM;
	return 0;
}

909
static int copy_files(unsigned long clone_flags, struct task_struct *tsk)
910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935
{
	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;
}

936
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
937 938 939
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;
940
	struct io_context *new_ioc;
941 942 943

	if (!ioc)
		return 0;
944 945 946 947 948 949 950 951
	/*
	 * Share io context with parent, if CLONE_IO is set
	 */
	if (clone_flags & CLONE_IO) {
		tsk->io_context = ioc_task_link(ioc);
		if (unlikely(!tsk->io_context))
			return -ENOMEM;
	} else if (ioprio_valid(ioc->ioprio)) {
952 953
		new_ioc = get_task_io_context(tsk, GFP_KERNEL, NUMA_NO_NODE);
		if (unlikely(!new_ioc))
954 955
			return -ENOMEM;

956
		new_ioc->ioprio = ioc->ioprio;
957
		put_io_context(new_ioc);
958 959 960 961 962
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
963
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
964 965 966
{
	struct sighand_struct *sig;

Z
Zhaolei 已提交
967
	if (clone_flags & CLONE_SIGHAND) {
L
Linus Torvalds 已提交
968 969 970 971
		atomic_inc(&current->sighand->count);
		return 0;
	}
	sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
I
Ingo Molnar 已提交
972
	rcu_assign_pointer(tsk->sighand, sig);
L
Linus Torvalds 已提交
973 974 975 976 977 978 979
	if (!sig)
		return -ENOMEM;
	atomic_set(&sig->count, 1);
	memcpy(sig->action, current->sighand->action, sizeof(sig->action));
	return 0;
}

980
void __cleanup_sighand(struct sighand_struct *sighand)
981
{
982 983
	if (atomic_dec_and_test(&sighand->count)) {
		signalfd_cleanup(sighand);
984
		kmem_cache_free(sighand_cachep, sighand);
985
	}
986 987
}

988 989 990 991 992 993

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

996 997 998
	/* Thread group counters. */
	thread_group_cputime_init(sig);

999 1000 1001
	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);
1002 1003 1004
		sig->cputimer.running = 1;
	}

1005 1006 1007 1008 1009 1010
	/* 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 已提交
1011
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
1012 1013 1014
{
	struct signal_struct *sig;

1015
	if (clone_flags & CLONE_THREAD)
1016 1017
		return 0;

1018
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
1019 1020 1021 1022
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

1023
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
1024
	atomic_set(&sig->live, 1);
1025
	atomic_set(&sig->sigcnt, 1);
L
Linus Torvalds 已提交
1026
	init_waitqueue_head(&sig->wait_chldexit);
1027 1028
	if (clone_flags & CLONE_NEWPID)
		sig->flags |= SIGNAL_UNKILLABLE;
1029
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
1030 1031 1032
	init_sigpending(&sig->shared_pending);
	INIT_LIST_HEAD(&sig->posix_timers);

1033
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
1034 1035 1036 1037 1038 1039
	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);

1040 1041
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
1042
	tty_audit_fork(sig);
1043
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
1044

1045
#ifdef CONFIG_CGROUPS
1046
	init_rwsem(&sig->group_rwsem);
1047 1048
#endif

1049
	sig->oom_adj = current->signal->oom_adj;
D
David Rientjes 已提交
1050
	sig->oom_score_adj = current->signal->oom_score_adj;
1051
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
1052

1053 1054 1055
	sig->has_child_subreaper = current->signal->has_child_subreaper ||
				   current->signal->is_child_subreaper;

1056 1057
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
1058 1059 1060
	return 0;
}

A
Alexey Dobriyan 已提交
1061
static void copy_flags(unsigned long clone_flags, struct task_struct *p)
L
Linus Torvalds 已提交
1062 1063 1064
{
	unsigned long new_flags = p->flags;

T
Tejun Heo 已提交
1065
	new_flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER);
L
Linus Torvalds 已提交
1066 1067 1068 1069
	new_flags |= PF_FORKNOEXEC;
	p->flags = new_flags;
}

1070
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1071 1072 1073
{
	current->clear_child_tid = tidptr;

1074
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1075 1076
}

A
Alexey Dobriyan 已提交
1077
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1078
{
1079
	raw_spin_lock_init(&p->pi_lock);
1080
#ifdef CONFIG_RT_MUTEXES
1081
	plist_head_init(&p->pi_waiters);
I
Ingo Molnar 已提交
1082 1083 1084 1085
	p->pi_blocked_on = NULL;
#endif
}

1086 1087 1088 1089 1090 1091 1092
#ifdef CONFIG_MM_OWNER
void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
	mm->owner = p;
}
#endif /* CONFIG_MM_OWNER */

1093 1094 1095 1096 1097
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
1098 1099
	tsk->cputime_expires.prof_exp = 0;
	tsk->cputime_expires.virt_exp = 0;
1100 1101 1102 1103 1104 1105
	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 已提交
1106 1107 1108 1109 1110 1111 1112 1113
/*
 * 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.
 */
1114 1115 1116 1117 1118
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 已提交
1119 1120
					struct pid *pid,
					int trace)
L
Linus Torvalds 已提交
1121 1122
{
	int retval;
1123
	struct task_struct *p;
1124
	int cgroup_callbacks_done = 0;
L
Linus Torvalds 已提交
1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143

	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);

1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
	/*
	 * 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 已提交
1154 1155 1156 1157 1158 1159 1160 1161 1162
	retval = security_task_create(clone_flags);
	if (retval)
		goto fork_out;

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

1163 1164
	ftrace_graph_init_task(p);

1165 1166
	rt_mutex_init_task(p);

I
Ingo Molnar 已提交
1167
#ifdef CONFIG_PROVE_LOCKING
1168 1169 1170
	DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled);
	DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled);
#endif
L
Linus Torvalds 已提交
1171
	retval = -EAGAIN;
1172
	if (atomic_read(&p->real_cred->user->processes) >=
1173
			task_rlimit(p, RLIMIT_NPROC)) {
L
Linus Torvalds 已提交
1174
		if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) &&
1175
		    p->real_cred->user != INIT_USER)
L
Linus Torvalds 已提交
1176 1177
			goto bad_fork_free;
	}
1178
	current->flags &= ~PF_NPROC_EXCEEDED;
L
Linus Torvalds 已提交
1179

1180 1181 1182
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1183 1184 1185 1186 1187 1188

	/*
	 * 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.
	 */
1189
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1190 1191 1192
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

A
Al Viro 已提交
1193
	if (!try_module_get(task_thread_info(p)->exec_domain->module))
L
Linus Torvalds 已提交
1194 1195 1196
		goto bad_fork_cleanup_count;

	p->did_exec = 0;
1197
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
L
Linus Torvalds 已提交
1198 1199 1200
	copy_flags(clone_flags, p);
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1201
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1202 1203 1204 1205 1206
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

1207 1208
	p->utime = p->stime = p->gtime = 0;
	p->utimescaled = p->stimescaled = 0;
1209
#ifndef CONFIG_VIRT_CPU_ACCOUNTING
1210
	p->prev_utime = p->prev_stime = 0;
1211
#endif
1212 1213 1214
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1215

1216 1217
	p->default_timer_slack_ns = current->timer_slack_ns;

1218
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1219 1220
	acct_clear_integrals(p);

1221
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1222 1223

	do_posix_clock_monotonic_gettime(&p->start_time);
1224 1225
	p->real_start_time = p->start_time;
	monotonic_to_bootbased(&p->real_start_time);
L
Linus Torvalds 已提交
1226 1227
	p->io_context = NULL;
	p->audit_context = NULL;
1228
	if (clone_flags & CLONE_THREAD)
1229
		threadgroup_change_begin(current);
1230
	cgroup_fork(p);
L
Linus Torvalds 已提交
1231
#ifdef CONFIG_NUMA
1232
	p->mempolicy = mpol_dup(p->mempolicy);
1233 1234 1235 1236 1237
	if (IS_ERR(p->mempolicy)) {
		retval = PTR_ERR(p->mempolicy);
		p->mempolicy = NULL;
		goto bad_fork_cleanup_cgroup;
	}
1238
	mpol_fix_fork_child_flag(p);
L
Linus Torvalds 已提交
1239
#endif
1240 1241 1242
#ifdef CONFIG_CPUSETS
	p->cpuset_mem_spread_rotor = NUMA_NO_NODE;
	p->cpuset_slab_spread_rotor = NUMA_NO_NODE;
1243
	seqcount_init(&p->mems_allowed_seq);
1244
#endif
1245 1246
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
1247 1248 1249
#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
	p->hardirqs_enabled = 1;
#else
1250
	p->hardirqs_enabled = 0;
1251
#endif
1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263
	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 已提交
1264 1265 1266 1267 1268
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
#endif
L
Linus Torvalds 已提交
1269

1270 1271 1272
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
1273 1274 1275 1276
#ifdef CONFIG_CGROUP_MEM_RES_CTLR
	p->memcg_batch.do_batch = 0;
	p->memcg_batch.memcg = NULL;
#endif
1277

1278
	/* Perform scheduler related setup. Assign this task to a CPU. */
1279
	sched_fork(p);
1280

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

1316 1317
	if (pid != &init_struct_pid) {
		retval = -ENOMEM;
1318
		pid = alloc_pid(p->nsproxy->pid_ns);
1319
		if (!pid)
1320
			goto bad_fork_cleanup_io;
1321 1322 1323 1324 1325 1326 1327
	}

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

L
Linus Torvalds 已提交
1328 1329 1330 1331
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
1332
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr : NULL;
1333 1334 1335
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1336
#ifdef CONFIG_FUTEX
1337 1338 1339 1340
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1341 1342
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1343
#endif
1344 1345 1346 1347 1348 1349
	/*
	 * 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 已提交
1350
	/*
1351 1352
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1353
	 */
1354
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1355
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1356 1357 1358
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1359
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1360 1361

	/* ok, now we should be set up.. */
1362 1363 1364 1365 1366 1367 1368
	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 已提交
1369 1370 1371
	p->pdeath_signal = 0;
	p->exit_state = 0;

1372 1373
	p->nr_dirtied = 0;
	p->nr_dirtied_pause = 128 >> (PAGE_SHIFT - 10);
1374
	p->dirty_paused_when = 0;
1375

L
Linus Torvalds 已提交
1376 1377 1378 1379 1380
	/*
	 * Ok, make it visible to the rest of the system.
	 * We dont wake it up yet.
	 */
	p->group_leader = p;
O
Oleg Nesterov 已提交
1381
	INIT_LIST_HEAD(&p->thread_group);
L
Linus Torvalds 已提交
1382

1383 1384 1385 1386 1387 1388
	/* 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 已提交
1389 1390 1391 1392
	/* Need tasklist lock for parent etc handling! */
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1393
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1394
		p->real_parent = current->real_parent;
1395 1396
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1397
		p->real_parent = current;
1398 1399
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1400

1401
	spin_lock(&current->sighand->siglock);
1402 1403 1404 1405 1406 1407 1408 1409

	/*
	 * 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).
1410
	*/
D
Daniel Walker 已提交
1411
	recalc_sigpending();
1412 1413 1414 1415
	if (signal_pending(current)) {
		spin_unlock(&current->sighand->siglock);
		write_unlock_irq(&tasklist_lock);
		retval = -ERESTARTNOINTR;
1416
		goto bad_fork_free_pid;
1417 1418
	}

L
Linus Torvalds 已提交
1419
	if (clone_flags & CLONE_THREAD) {
1420
		current->signal->nr_threads++;
1421
		atomic_inc(&current->signal->live);
1422
		atomic_inc(&current->signal->sigcnt);
L
Linus Torvalds 已提交
1423
		p->group_leader = current->group_leader;
O
Oleg Nesterov 已提交
1424
		list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
L
Linus Torvalds 已提交
1425 1426
	}

1427
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1428
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1429 1430

		if (thread_group_leader(p)) {
1431
			if (is_child_reaper(pid))
1432
				p->nsproxy->pid_ns->child_reaper = p;
1433

1434
			p->signal->leader_pid = pid;
A
Alan Cox 已提交
1435
			p->signal->tty = tty_kref_get(current->signal->tty);
E
Eric W. Biederman 已提交
1436 1437
			attach_pid(p, PIDTYPE_PGID, task_pgrp(current));
			attach_pid(p, PIDTYPE_SID, task_session(current));
1438
			list_add_tail(&p->sibling, &p->real_parent->children);
1439
			list_add_tail_rcu(&p->tasks, &init_task.tasks);
1440
			__this_cpu_inc(process_counts);
1441
		}
1442
		attach_pid(p, PIDTYPE_PID, pid);
1443
		nr_threads++;
L
Linus Torvalds 已提交
1444 1445 1446
	}

	total_forks++;
1447
	spin_unlock(&current->sighand->siglock);
L
Linus Torvalds 已提交
1448
	write_unlock_irq(&tasklist_lock);
1449
	proc_fork_connector(p);
1450
	cgroup_post_fork(p);
1451
	if (clone_flags & CLONE_THREAD)
1452
		threadgroup_change_end(current);
1453
	perf_event_fork(p);
1454 1455 1456

	trace_task_newtask(p, clone_flags);

L
Linus Torvalds 已提交
1457 1458
	return p;

1459 1460 1461
bad_fork_free_pid:
	if (pid != &init_struct_pid)
		free_pid(pid);
1462
bad_fork_cleanup_io:
1463 1464
	if (p->io_context)
		exit_io_context(p);
S
Serge E. Hallyn 已提交
1465
bad_fork_cleanup_namespaces:
1466 1467
	if (unlikely(clone_flags & CLONE_NEWPID))
		pid_ns_release_proc(p->nsproxy->pid_ns);
1468
	exit_task_namespaces(p);
L
Linus Torvalds 已提交
1469
bad_fork_cleanup_mm:
D
David Rientjes 已提交
1470
	if (p->mm)
L
Linus Torvalds 已提交
1471 1472
		mmput(p->mm);
bad_fork_cleanup_signal:
1473
	if (!(clone_flags & CLONE_THREAD))
1474
		free_signal_struct(p->signal);
L
Linus Torvalds 已提交
1475
bad_fork_cleanup_sighand:
1476
	__cleanup_sighand(p->sighand);
L
Linus Torvalds 已提交
1477 1478 1479 1480 1481 1482 1483 1484 1485
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:
1486
	perf_event_free_task(p);
L
Linus Torvalds 已提交
1487
#ifdef CONFIG_NUMA
1488
	mpol_put(p->mempolicy);
1489
bad_fork_cleanup_cgroup:
L
Linus Torvalds 已提交
1490
#endif
1491
	if (clone_flags & CLONE_THREAD)
1492
		threadgroup_change_end(current);
1493
	cgroup_exit(p, cgroup_callbacks_done);
1494
	delayacct_tsk_free(p);
A
Al Viro 已提交
1495
	module_put(task_thread_info(p)->exec_domain->module);
L
Linus Torvalds 已提交
1496
bad_fork_cleanup_count:
D
David Howells 已提交
1497
	atomic_dec(&p->cred->user->processes);
1498
	exit_creds(p);
L
Linus Torvalds 已提交
1499 1500
bad_fork_free:
	free_task(p);
1501 1502
fork_out:
	return ERR_PTR(retval);
L
Linus Torvalds 已提交
1503 1504
}

A
Adrian Bunk 已提交
1505
noinline struct pt_regs * __cpuinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
L
Linus Torvalds 已提交
1506 1507 1508 1509 1510
{
	memset(regs, 0, sizeof(struct pt_regs));
	return regs;
}

1511 1512 1513 1514 1515 1516 1517 1518 1519 1520
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;
	}
}

1521
struct task_struct * __cpuinit fork_idle(int cpu)
L
Linus Torvalds 已提交
1522
{
1523
	struct task_struct *task;
L
Linus Torvalds 已提交
1524 1525
	struct pt_regs regs;

1526
	task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL,
R
Roland McGrath 已提交
1527
			    &init_struct_pid, 0);
1528 1529
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
1530
		init_idle(task, cpu);
1531
	}
1532

L
Linus Torvalds 已提交
1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550
	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;
1551
	long nr;
L
Linus Torvalds 已提交
1552

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
	/*
	 * Do some preliminary argument and permissions checking before we
	 * actually start allocating stuff
	 */
	if (clone_flags & CLONE_NEWUSER) {
		if (clone_flags & CLONE_THREAD)
			return -EINVAL;
		/* hopefully this check will go away when userns support is
		 * complete
		 */
1563 1564
		if (!capable(CAP_SYS_ADMIN) || !capable(CAP_SETUID) ||
				!capable(CAP_SETGID))
1565 1566 1567
			return -EPERM;
	}

R
Roland McGrath 已提交
1568
	/*
T
Tejun Heo 已提交
1569 1570 1571 1572
	 * 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 已提交
1573
	 */
T
Tejun Heo 已提交
1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
	if (likely(user_mode(regs)) && !(clone_flags & CLONE_UNTRACED)) {
		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 已提交
1585

1586
	p = copy_process(clone_flags, stack_start, regs, stack_size,
R
Roland McGrath 已提交
1587
			 child_tidptr, NULL, trace);
L
Linus Torvalds 已提交
1588 1589 1590 1591 1592 1593 1594
	/*
	 * 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;

1595 1596
		trace_sched_process_fork(current, p);

1597
		nr = task_pid_vnr(p);
1598 1599 1600

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

L
Linus Torvalds 已提交
1602 1603 1604
		if (clone_flags & CLONE_VFORK) {
			p->vfork_done = &vfork;
			init_completion(&vfork);
O
Oleg Nesterov 已提交
1605
			get_task_struct(p);
L
Linus Torvalds 已提交
1606 1607
		}

1608
		wake_up_new_task(p);
L
Linus Torvalds 已提交
1609

T
Tejun Heo 已提交
1610 1611 1612
		/* forking complete and child started to run, tell ptracer */
		if (unlikely(trace))
			ptrace_event(trace, nr);
R
Roland McGrath 已提交
1613

L
Linus Torvalds 已提交
1614
		if (clone_flags & CLONE_VFORK) {
O
Oleg Nesterov 已提交
1615 1616
			if (!wait_for_vfork_done(p, &vfork))
				ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
L
Linus Torvalds 已提交
1617 1618
		}
	} else {
1619
		nr = PTR_ERR(p);
L
Linus Torvalds 已提交
1620
	}
1621
	return nr;
L
Linus Torvalds 已提交
1622 1623
}

1624 1625 1626 1627
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

1628
static void sighand_ctor(void *data)
1629 1630 1631
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
1632
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
1633
	init_waitqueue_head(&sighand->signalfd_wqh);
1634 1635
}

L
Linus Torvalds 已提交
1636 1637 1638 1639
void __init proc_caches_init(void)
{
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
V
Vegard Nossum 已提交
1640 1641
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU|
			SLAB_NOTRACK, sighand_ctor);
L
Linus Torvalds 已提交
1642 1643
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
V
Vegard Nossum 已提交
1644
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1645
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
1646
			sizeof(struct files_struct), 0,
V
Vegard Nossum 已提交
1647
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1648
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
1649
			sizeof(struct fs_struct), 0,
V
Vegard Nossum 已提交
1650
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1651 1652 1653 1654 1655 1656 1657
	/*
	 * 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 已提交
1658
	mm_cachep = kmem_cache_create("mm_struct",
1659
			sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
V
Vegard Nossum 已提交
1660
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1661
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC);
1662
	mmap_init();
1663
	nsproxy_cache_init();
L
Linus Torvalds 已提交
1664
}
1665 1666

/*
1667
 * Check constraints on flags passed to the unshare system call.
1668
 */
1669
static int check_unshare_flags(unsigned long unshare_flags)
1670
{
1671 1672 1673 1674
	if (unshare_flags & ~(CLONE_THREAD|CLONE_FS|CLONE_NEWNS|CLONE_SIGHAND|
				CLONE_VM|CLONE_FILES|CLONE_SYSVSEM|
				CLONE_NEWUTS|CLONE_NEWIPC|CLONE_NEWNET))
		return -EINVAL;
1675
	/*
1676 1677 1678
	 * Not implemented, but pretend it works if there is nothing to
	 * unshare. Note that unsharing CLONE_THREAD or CLONE_SIGHAND
	 * needs to unshare vm.
1679
	 */
1680 1681 1682 1683 1684
	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;
	}
1685 1686 1687 1688 1689

	return 0;
}

/*
1690
 * Unshare the filesystem structure if it is being shared
1691 1692 1693 1694 1695
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
1696 1697 1698 1699 1700 1701 1702 1703 1704 1705
	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;
1706 1707 1708 1709 1710

	return 0;
}

/*
1711
 * Unshare file descriptor table if it is being shared
1712 1713 1714 1715
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
1716
	int error = 0;
1717 1718

	if ((unshare_flags & CLONE_FILES) &&
1719 1720 1721 1722 1723
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735

	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.
 */
1736
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
1737 1738 1739
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
1740
	struct nsproxy *new_nsproxy = NULL;
1741
	int do_sysvsem = 0;
1742
	int err;
1743

1744 1745
	err = check_unshare_flags(unshare_flags);
	if (err)
1746 1747
		goto bad_unshare_out;

1748 1749 1750 1751 1752
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
1753 1754 1755 1756 1757 1758
	/*
	 * 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))
1759
		do_sysvsem = 1;
1760 1761
	err = unshare_fs(unshare_flags, &new_fs);
	if (err)
1762
		goto bad_unshare_out;
1763 1764
	err = unshare_fd(unshare_flags, &new_fd);
	if (err)
1765
		goto bad_unshare_cleanup_fs;
1766 1767
	err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy, new_fs);
	if (err)
1768
		goto bad_unshare_cleanup_fd;
1769

1770
	if (new_fs || new_fd || do_sysvsem || new_nsproxy) {
1771 1772 1773 1774 1775 1776
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
S
Serge E. Hallyn 已提交
1777

1778
		if (new_nsproxy) {
1779 1780
			switch_task_namespaces(current, new_nsproxy);
			new_nsproxy = NULL;
1781
		}
1782

1783 1784
		task_lock(current);

1785 1786
		if (new_fs) {
			fs = current->fs;
N
Nick Piggin 已提交
1787
			spin_lock(&fs->lock);
1788
			current->fs = new_fs;
A
Al Viro 已提交
1789 1790 1791 1792
			if (--fs->users)
				new_fs = NULL;
			else
				new_fs = fs;
N
Nick Piggin 已提交
1793
			spin_unlock(&fs->lock);
1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804
		}

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

		task_unlock(current);
	}

1805
	if (new_nsproxy)
1806
		put_nsproxy(new_nsproxy);
1807

1808 1809 1810 1811 1812 1813
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
1814
		free_fs_struct(new_fs);
1815 1816 1817 1818

bad_unshare_out:
	return err;
}
1819 1820 1821 1822 1823 1824 1825 1826 1827 1828

/*
 *	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;
1829
	struct files_struct *copy = NULL;
1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
	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;
}