fork.c 42.3 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>
L
Linus Torvalds 已提交
50 51
#include <linux/profile.h>
#include <linux/rmap.h>
H
Hugh Dickins 已提交
52
#include <linux/ksm.h>
L
Linus Torvalds 已提交
53
#include <linux/acct.h>
54
#include <linux/tsacct_kern.h>
M
Matt Helsley 已提交
55
#include <linux/cn_proc.h>
R
Rafael J. Wysocki 已提交
56
#include <linux/freezer.h>
57
#include <linux/delayacct.h>
58
#include <linux/taskstats_kern.h>
59
#include <linux/random.h>
M
Miloslav Trmac 已提交
60
#include <linux/tty.h>
61
#include <linux/blkdev.h>
62
#include <linux/fs_struct.h>
63
#include <linux/magic.h>
64
#include <linux/perf_event.h>
65
#include <linux/posix-timers.h>
66
#include <linux/user-return-notifier.h>
Y
Ying Han 已提交
67
#include <linux/oom.h>
A
Andrea Arcangeli 已提交
68
#include <linux/khugepaged.h>
L
Linus Torvalds 已提交
69 70 71 72 73 74 75 76

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

77 78
#include <trace/events/sched.h>

L
Linus Torvalds 已提交
79 80 81 82 83 84 85 86 87 88
/*
 * Protected counters by write_lock_irq(&tasklist_lock)
 */
unsigned long total_forks;	/* Handle normal Linux uptimes. */
int nr_threads; 		/* The idle threads do not count.. */

int max_threads;		/* tunable limit on nr_threads */

DEFINE_PER_CPU(unsigned long, process_counts) = 0;

89
__cacheline_aligned DEFINE_RWLOCK(tasklist_lock);  /* outer */
90 91 92 93 94 95 96 97

#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 已提交
98 99 100 101 102 103

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

104
	for_each_possible_cpu(cpu)
L
Linus Torvalds 已提交
105 106 107 108 109 110
		total += per_cpu(process_counts, cpu);

	return total;
}

#ifndef __HAVE_ARCH_TASK_STRUCT_ALLOCATOR
111 112 113 114
# 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))
115
static struct kmem_cache *task_struct_cachep;
L
Linus Torvalds 已提交
116 117
#endif

118
#ifndef __HAVE_ARCH_THREAD_INFO_ALLOCATOR
119 120
static struct thread_info *alloc_thread_info_node(struct task_struct *tsk,
						  int node)
121 122 123 124 125 126
{
#ifdef CONFIG_DEBUG_STACK_USAGE
	gfp_t mask = GFP_KERNEL | __GFP_ZERO;
#else
	gfp_t mask = GFP_KERNEL;
#endif
127 128 129
	struct page *page = alloc_pages_node(node, mask, THREAD_SIZE_ORDER);

	return page ? page_address(page) : NULL;
130 131 132 133 134 135 136 137
}

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

L
Linus Torvalds 已提交
138
/* SLAB cache for signal_struct structures (tsk->signal) */
139
static struct kmem_cache *signal_cachep;
L
Linus Torvalds 已提交
140 141

/* SLAB cache for sighand_struct structures (tsk->sighand) */
142
struct kmem_cache *sighand_cachep;
L
Linus Torvalds 已提交
143 144

/* SLAB cache for files_struct structures (tsk->files) */
145
struct kmem_cache *files_cachep;
L
Linus Torvalds 已提交
146 147

/* SLAB cache for fs_struct structures (tsk->fs) */
148
struct kmem_cache *fs_cachep;
L
Linus Torvalds 已提交
149 150

/* SLAB cache for vm_area_struct structures */
151
struct kmem_cache *vm_area_cachep;
L
Linus Torvalds 已提交
152 153

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

156 157 158 159 160 161 162
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 已提交
163 164
void free_task(struct task_struct *tsk)
{
P
Peter Zijlstra 已提交
165
	prop_local_destroy_single(&tsk->dirties);
166
	account_kernel_stack(tsk->stack, -1);
R
Roman Zippel 已提交
167
	free_thread_info(tsk->stack);
I
Ingo Molnar 已提交
168
	rt_mutex_debug_task_free(tsk);
169
	ftrace_graph_exit_task(tsk);
L
Linus Torvalds 已提交
170 171 172 173
	free_task_struct(tsk);
}
EXPORT_SYMBOL(free_task);

174 175
static inline void free_signal_struct(struct signal_struct *sig)
{
176
	taskstats_tgid_free(sig);
177
	sched_autogroup_exit(sig);
178 179 180 181 182
	kmem_cache_free(signal_cachep, sig);
}

static inline void put_signal_struct(struct signal_struct *sig)
{
183
	if (atomic_dec_and_test(&sig->sigcnt))
184 185 186
		free_signal_struct(sig);
}

187
void __put_task_struct(struct task_struct *tsk)
L
Linus Torvalds 已提交
188
{
E
Eugene Teo 已提交
189
	WARN_ON(!tsk->exit_state);
L
Linus Torvalds 已提交
190 191 192
	WARN_ON(atomic_read(&tsk->usage));
	WARN_ON(tsk == current);

193
	exit_creds(tsk);
194
	delayacct_tsk_free(tsk);
195
	put_signal_struct(tsk->signal);
L
Linus Torvalds 已提交
196 197 198 199

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

S
Suresh Siddha 已提交
202 203 204 205 206 207 208
/*
 * 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
209

L
Linus Torvalds 已提交
210 211 212 213 214 215 216 217 218
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 已提交
219
			ARCH_MIN_TASKALIGN, SLAB_PANIC | SLAB_NOTRACK, NULL);
L
Linus Torvalds 已提交
220 221
#endif

222 223 224
	/* do the arch specific task caches init */
	arch_task_cache_init();

L
Linus Torvalds 已提交
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243
	/*
	 * 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
	 */
	if(max_threads < 20)
		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];
}

244 245 246 247 248 249 250
int __attribute__((weak)) arch_dup_task_struct(struct task_struct *dst,
					       struct task_struct *src)
{
	*dst = *src;
	return 0;
}

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

	prepare_to_copy(orig);

261
	tsk = alloc_task_struct_node(node);
L
Linus Torvalds 已提交
262 263 264
	if (!tsk)
		return NULL;

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

271 272 273 274
 	err = arch_dup_task_struct(tsk, orig);
	if (err)
		goto out;

R
Roman Zippel 已提交
275
	tsk->stack = ti;
P
Peter Zijlstra 已提交
276 277

	err = prop_local_init_single(&tsk->dirties);
278 279
	if (err)
		goto out;
P
Peter Zijlstra 已提交
280

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

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

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

	account_kernel_stack(ti, 1);

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

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

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

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

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

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

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

L
Linus Torvalds 已提交
381 382 383
			get_file(file);
			if (tmp->vm_flags & VM_DENYWRITE)
				atomic_dec(&inode->i_writecount);
384
			mutex_lock(&mapping->i_mmap_mutex);
H
Hugh Dickins 已提交
385 386 387 388
			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 已提交
389
			vma_prio_tree_add(tmp, mpnt);
H
Hugh Dickins 已提交
390
			flush_dcache_mmap_unlock(mapping);
391
			mutex_unlock(&mapping->i_mmap_mutex);
L
Linus Torvalds 已提交
392 393
		}

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

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

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

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

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

static inline int mm_alloc_pgd(struct mm_struct * mm)
{
	mm->pgd = pgd_alloc(mm);
	if (unlikely(!mm->pgd))
		return -ENOMEM;
	return 0;
}

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

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

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

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

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

486
static struct mm_struct * mm_init(struct mm_struct * mm, struct task_struct *p)
L
Linus Torvalds 已提交
487 488 489 490 491
{
	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 已提交
492 493
	mm->flags = (current->mm) ?
		(current->mm->flags & MMF_INIT_MASK) : default_dump_filter;
494
	mm->core_state = NULL;
L
Linus Torvalds 已提交
495
	mm->nr_ptes = 0;
K
KAMEZAWA Hiroyuki 已提交
496
	memset(&mm->rss_stat, 0, sizeof(mm->rss_stat));
L
Linus Torvalds 已提交
497 498
	spin_lock_init(&mm->page_table_lock);
	mm->free_area_cache = TASK_UNMAPPED_BASE;
499
	mm->cached_hole_size = ~0UL;
A
Alexey Dobriyan 已提交
500
	mm_init_aio(mm);
501
	mm_init_owner(mm, p);
Y
Ying Han 已提交
502
	atomic_set(&mm->oom_disable_count, 0);
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 514 515 516 517 518 519 520 521
	free_mm(mm);
	return NULL;
}

/*
 * Allocate and initialize an mm_struct.
 */
struct mm_struct * mm_alloc(void)
{
	struct mm_struct * mm;

	mm = allocate_mm();
522 523 524 525
	if (!mm)
		return NULL;

	memset(mm, 0, sizeof(*mm));
526 527
	mm_init_cpumask(mm);
	return mm_init(mm, current);
L
Linus Torvalds 已提交
528 529 530 531 532 533 534
}

/*
 * 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.
 */
535
void __mmdrop(struct mm_struct *mm)
L
Linus Torvalds 已提交
536 537 538 539
{
	BUG_ON(mm == &init_mm);
	mm_free_pgd(mm);
	destroy_context(mm);
A
Andrea Arcangeli 已提交
540
	mmu_notifier_mm_destroy(mm);
541 542 543
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	VM_BUG_ON(mm->pmd_huge_pte);
#endif
L
Linus Torvalds 已提交
544 545
	free_mm(mm);
}
546
EXPORT_SYMBOL_GPL(__mmdrop);
L
Linus Torvalds 已提交
547 548 549 550 551 552

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

L
Linus Torvalds 已提交
555 556
	if (atomic_dec_and_test(&mm->mm_users)) {
		exit_aio(mm);
557
		ksm_exit(mm);
A
Andrea Arcangeli 已提交
558
		khugepaged_exit(mm); /* must run before exit_mmap */
L
Linus Torvalds 已提交
559
		exit_mmap(mm);
M
Matt Helsley 已提交
560
		set_mm_exe_file(mm, NULL);
L
Linus Torvalds 已提交
561 562 563 564 565 566
		if (!list_empty(&mm->mmlist)) {
			spin_lock(&mmlist_lock);
			list_del(&mm->mmlist);
			spin_unlock(&mmlist_lock);
		}
		put_swap_token(mm);
567 568
		if (mm->binfmt)
			module_put(mm->binfmt->module);
L
Linus Torvalds 已提交
569 570 571 572 573
		mmdrop(mm);
	}
}
EXPORT_SYMBOL_GPL(mmput);

574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624
/*
 * 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--;
	if ((mm->num_exe_file_vmas == 0) && mm->exe_file){
		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 已提交
625 626 627
/**
 * get_task_mm - acquire a reference to the task's mm
 *
628
 * Returns %NULL if the task has no mm.  Checks PF_KTHREAD (meaning
L
Linus Torvalds 已提交
629 630 631 632 633 634 635 636 637 638 639 640
 * 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) {
641
		if (task->flags & PF_KTHREAD)
L
Linus Torvalds 已提交
642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667
			mm = NULL;
		else
			atomic_inc(&mm->mm_users);
	}
	task_unlock(task);
	return mm;
}
EXPORT_SYMBOL_GPL(get_task_mm);

/* 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)
{
	struct completion *vfork_done = tsk->vfork_done;

668 669
	/* Get rid of any futexes when releasing the mm */
#ifdef CONFIG_FUTEX
670
	if (unlikely(tsk->robust_list)) {
671
		exit_robust_list(tsk);
672 673
		tsk->robust_list = NULL;
	}
674
#ifdef CONFIG_COMPAT
675
	if (unlikely(tsk->compat_robust_list)) {
676
		compat_exit_robust_list(tsk);
677 678
		tsk->compat_robust_list = NULL;
	}
679
#endif
680 681
	if (unlikely(!list_empty(&tsk->pi_state_list)))
		exit_pi_state_list(tsk);
682 683
#endif

L
Linus Torvalds 已提交
684 685 686 687 688 689 690 691
	/* Get rid of any cached register state */
	deactivate_mm(tsk, mm);

	/* notify parent sleeping on vfork() */
	if (vfork_done) {
		tsk->vfork_done = NULL;
		complete(vfork_done);
	}
692 693 694 695 696 697 698

	/*
	 * 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
	 * trouble otherwise.  Userland only wants this done for a sys_exit.
	 */
699 700 701 702 703 704 705 706 707 708 709
	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 已提交
710 711 712 713
		tsk->clear_child_tid = NULL;
	}
}

714 715 716 717
/*
 * Allocate a new mm structure and copy contents from the
 * mm structure of the passed in task structure.
 */
718
struct mm_struct *dup_mm(struct task_struct *tsk)
719 720 721 722 723 724 725 726 727 728 729 730
{
	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));
731
	mm_init_cpumask(mm);
732

733 734 735 736
	/* Initializing for Swap token stuff */
	mm->token_priority = 0;
	mm->last_interval = 0;

737 738 739 740
#ifdef CONFIG_TRANSPARENT_HUGEPAGE
	mm->pmd_huge_pte = NULL;
#endif

741
	if (!mm_init(mm, tsk))
742 743 744 745 746
		goto fail_nomem;

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

M
Matt Helsley 已提交
747 748
	dup_mm_exe_file(oldmm, mm);

749 750 751 752 753 754 755
	err = dup_mmap(mm, oldmm);
	if (err)
		goto free_pt;

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

756 757 758
	if (mm->binfmt && !try_module_get(mm->binfmt->module))
		goto free_pt;

759 760 761
	return mm;

free_pt:
762 763
	/* don't put binfmt in mmput, we haven't got module yet */
	mm->binfmt = NULL;
764 765 766 767 768 769 770 771 772 773 774 775 776 777 778
	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;
}

L
Linus Torvalds 已提交
779 780 781 782 783 784 785
static int copy_mm(unsigned long clone_flags, struct task_struct * tsk)
{
	struct mm_struct * mm, *oldmm;
	int retval;

	tsk->min_flt = tsk->maj_flt = 0;
	tsk->nvcsw = tsk->nivcsw = 0;
786 787 788
#ifdef CONFIG_DETECT_HUNG_TASK
	tsk->last_switch_count = tsk->nvcsw + tsk->nivcsw;
#endif
L
Linus Torvalds 已提交
789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808

	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;
809
	mm = dup_mm(tsk);
L
Linus Torvalds 已提交
810 811 812 813
	if (!mm)
		goto fail_nomem;

good_mm:
814 815 816
	/* Initializing for Swap token stuff */
	mm->token_priority = 0;
	mm->last_interval = 0;
Y
Ying Han 已提交
817 818
	if (tsk->signal->oom_score_adj == OOM_SCORE_ADJ_MIN)
		atomic_inc(&mm->oom_disable_count);
819

L
Linus Torvalds 已提交
820 821 822 823 824 825 826 827
	tsk->mm = mm;
	tsk->active_mm = mm;
	return 0;

fail_nomem:
	return retval;
}

A
Alexey Dobriyan 已提交
828
static int copy_fs(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
829
{
A
Al Viro 已提交
830
	struct fs_struct *fs = current->fs;
L
Linus Torvalds 已提交
831
	if (clone_flags & CLONE_FS) {
A
Al Viro 已提交
832
		/* tsk->fs is already what we want */
N
Nick Piggin 已提交
833
		spin_lock(&fs->lock);
A
Al Viro 已提交
834
		if (fs->in_exec) {
N
Nick Piggin 已提交
835
			spin_unlock(&fs->lock);
A
Al Viro 已提交
836 837 838
			return -EAGAIN;
		}
		fs->users++;
N
Nick Piggin 已提交
839
		spin_unlock(&fs->lock);
L
Linus Torvalds 已提交
840 841
		return 0;
	}
A
Al Viro 已提交
842
	tsk->fs = copy_fs_struct(fs);
L
Linus Torvalds 已提交
843 844 845 846 847
	if (!tsk->fs)
		return -ENOMEM;
	return 0;
}

848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874
static int copy_files(unsigned long clone_flags, struct task_struct * tsk)
{
	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;
}

875
static int copy_io(unsigned long clone_flags, struct task_struct *tsk)
876 877 878 879 880 881
{
#ifdef CONFIG_BLOCK
	struct io_context *ioc = current->io_context;

	if (!ioc)
		return 0;
882 883 884 885 886 887 888 889
	/*
	 * 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)) {
890 891 892 893 894 895 896 897 898 899
		tsk->io_context = alloc_io_context(GFP_KERNEL, -1);
		if (unlikely(!tsk->io_context))
			return -ENOMEM;

		tsk->io_context->ioprio = ioc->ioprio;
	}
#endif
	return 0;
}

A
Alexey Dobriyan 已提交
900
static int copy_sighand(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
901 902 903
{
	struct sighand_struct *sig;

Z
Zhaolei 已提交
904
	if (clone_flags & CLONE_SIGHAND) {
L
Linus Torvalds 已提交
905 906 907 908
		atomic_inc(&current->sighand->count);
		return 0;
	}
	sig = kmem_cache_alloc(sighand_cachep, GFP_KERNEL);
I
Ingo Molnar 已提交
909
	rcu_assign_pointer(tsk->sighand, sig);
L
Linus Torvalds 已提交
910 911 912 913 914 915 916
	if (!sig)
		return -ENOMEM;
	atomic_set(&sig->count, 1);
	memcpy(sig->action, current->sighand->action, sizeof(sig->action));
	return 0;
}

917
void __cleanup_sighand(struct sighand_struct *sighand)
918 919 920 921 922
{
	if (atomic_dec_and_test(&sighand->count))
		kmem_cache_free(sighand_cachep, sighand);
}

923 924 925 926 927 928

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

931 932 933
	/* Thread group counters. */
	thread_group_cputime_init(sig);

934 935 936
	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);
937 938 939
		sig->cputimer.running = 1;
	}

940 941 942 943 944 945
	/* 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 已提交
946
static int copy_signal(unsigned long clone_flags, struct task_struct *tsk)
L
Linus Torvalds 已提交
947 948 949
{
	struct signal_struct *sig;

950
	if (clone_flags & CLONE_THREAD)
951 952
		return 0;

953
	sig = kmem_cache_zalloc(signal_cachep, GFP_KERNEL);
L
Linus Torvalds 已提交
954 955 956 957
	tsk->signal = sig;
	if (!sig)
		return -ENOMEM;

958
	sig->nr_threads = 1;
L
Linus Torvalds 已提交
959
	atomic_set(&sig->live, 1);
960
	atomic_set(&sig->sigcnt, 1);
L
Linus Torvalds 已提交
961
	init_waitqueue_head(&sig->wait_chldexit);
962 963
	if (clone_flags & CLONE_NEWPID)
		sig->flags |= SIGNAL_UNKILLABLE;
964
	sig->curr_target = tsk;
L
Linus Torvalds 已提交
965 966 967
	init_sigpending(&sig->shared_pending);
	INIT_LIST_HEAD(&sig->posix_timers);

968
	hrtimer_init(&sig->real_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
L
Linus Torvalds 已提交
969 970 971 972 973 974
	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);

975 976
	posix_cpu_timers_init_group(sig);

M
Miloslav Trmac 已提交
977
	tty_audit_fork(sig);
978
	sched_autogroup_fork(sig);
M
Miloslav Trmac 已提交
979

980 981 982 983
#ifdef CONFIG_CGROUPS
	init_rwsem(&sig->threadgroup_fork_lock);
#endif

984
	sig->oom_adj = current->signal->oom_adj;
D
David Rientjes 已提交
985
	sig->oom_score_adj = current->signal->oom_score_adj;
986
	sig->oom_score_adj_min = current->signal->oom_score_adj_min;
987

988 989
	mutex_init(&sig->cred_guard_mutex);

L
Linus Torvalds 已提交
990 991 992
	return 0;
}

A
Alexey Dobriyan 已提交
993
static void copy_flags(unsigned long clone_flags, struct task_struct *p)
L
Linus Torvalds 已提交
994 995 996
{
	unsigned long new_flags = p->flags;

T
Tejun Heo 已提交
997
	new_flags &= ~(PF_SUPERPRIV | PF_WQ_WORKER);
L
Linus Torvalds 已提交
998
	new_flags |= PF_FORKNOEXEC;
R
Roland McGrath 已提交
999
	new_flags |= PF_STARTING;
L
Linus Torvalds 已提交
1000
	p->flags = new_flags;
1001
	clear_freeze_flag(p);
L
Linus Torvalds 已提交
1002 1003
}

1004
SYSCALL_DEFINE1(set_tid_address, int __user *, tidptr)
L
Linus Torvalds 已提交
1005 1006 1007
{
	current->clear_child_tid = tidptr;

1008
	return task_pid_vnr(current);
L
Linus Torvalds 已提交
1009 1010
}

A
Alexey Dobriyan 已提交
1011
static void rt_mutex_init_task(struct task_struct *p)
I
Ingo Molnar 已提交
1012
{
1013
	raw_spin_lock_init(&p->pi_lock);
1014
#ifdef CONFIG_RT_MUTEXES
1015
	plist_head_init_raw(&p->pi_waiters, &p->pi_lock);
I
Ingo Molnar 已提交
1016 1017 1018 1019
	p->pi_blocked_on = NULL;
#endif
}

1020 1021 1022 1023 1024 1025 1026
#ifdef CONFIG_MM_OWNER
void mm_init_owner(struct mm_struct *mm, struct task_struct *p)
{
	mm->owner = p;
}
#endif /* CONFIG_MM_OWNER */

1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
/*
 * Initialize POSIX timer handling for a single task.
 */
static void posix_cpu_timers_init(struct task_struct *tsk)
{
	tsk->cputime_expires.prof_exp = cputime_zero;
	tsk->cputime_expires.virt_exp = cputime_zero;
	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 已提交
1040 1041 1042 1043 1044 1045 1046 1047
/*
 * 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.
 */
1048 1049 1050 1051 1052
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 已提交
1053 1054
					struct pid *pid,
					int trace)
L
Linus Torvalds 已提交
1055 1056
{
	int retval;
1057
	struct task_struct *p;
1058
	int cgroup_callbacks_done = 0;
L
Linus Torvalds 已提交
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077

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

1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
	/*
	 * 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 已提交
1088 1089 1090 1091 1092 1093 1094 1095 1096
	retval = security_task_create(clone_flags);
	if (retval)
		goto fork_out;

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

1097 1098
	ftrace_graph_init_task(p);

1099 1100
	rt_mutex_init_task(p);

I
Ingo Molnar 已提交
1101
#ifdef CONFIG_PROVE_LOCKING
1102 1103 1104
	DEBUG_LOCKS_WARN_ON(!p->hardirqs_enabled);
	DEBUG_LOCKS_WARN_ON(!p->softirqs_enabled);
#endif
L
Linus Torvalds 已提交
1105
	retval = -EAGAIN;
1106
	if (atomic_read(&p->real_cred->user->processes) >=
1107
			task_rlimit(p, RLIMIT_NPROC)) {
L
Linus Torvalds 已提交
1108
		if (!capable(CAP_SYS_ADMIN) && !capable(CAP_SYS_RESOURCE) &&
1109
		    p->real_cred->user != INIT_USER)
L
Linus Torvalds 已提交
1110 1111 1112
			goto bad_fork_free;
	}

1113 1114 1115
	retval = copy_creds(p, clone_flags);
	if (retval < 0)
		goto bad_fork_free;
L
Linus Torvalds 已提交
1116 1117 1118 1119 1120 1121

	/*
	 * 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.
	 */
1122
	retval = -EAGAIN;
L
Linus Torvalds 已提交
1123 1124 1125
	if (nr_threads >= max_threads)
		goto bad_fork_cleanup_count;

A
Al Viro 已提交
1126
	if (!try_module_get(task_thread_info(p)->exec_domain->module))
L
Linus Torvalds 已提交
1127 1128 1129
		goto bad_fork_cleanup_count;

	p->did_exec = 0;
1130
	delayacct_tsk_init(p);	/* Must remain after dup_task_struct() */
L
Linus Torvalds 已提交
1131 1132 1133
	copy_flags(clone_flags, p);
	INIT_LIST_HEAD(&p->children);
	INIT_LIST_HEAD(&p->sibling);
1134
	rcu_copy_process(p);
L
Linus Torvalds 已提交
1135 1136 1137 1138 1139 1140 1141
	p->vfork_done = NULL;
	spin_lock_init(&p->alloc_lock);

	init_sigpending(&p->pending);

	p->utime = cputime_zero;
	p->stime = cputime_zero;
1142
	p->gtime = cputime_zero;
1143 1144
	p->utimescaled = cputime_zero;
	p->stimescaled = cputime_zero;
1145
#ifndef CONFIG_VIRT_CPU_ACCOUNTING
P
Peter Zijlstra 已提交
1146
	p->prev_utime = cputime_zero;
1147
	p->prev_stime = cputime_zero;
1148
#endif
1149 1150 1151
#if defined(SPLIT_RSS_COUNTING)
	memset(&p->rss_stat, 0, sizeof(p->rss_stat));
#endif
1152

1153 1154
	p->default_timer_slack_ns = current->timer_slack_ns;

1155
	task_io_accounting_init(&p->ioac);
L
Linus Torvalds 已提交
1156 1157
	acct_clear_integrals(p);

1158
	posix_cpu_timers_init(p);
L
Linus Torvalds 已提交
1159 1160

	do_posix_clock_monotonic_gettime(&p->start_time);
1161 1162
	p->real_start_time = p->start_time;
	monotonic_to_bootbased(&p->real_start_time);
L
Linus Torvalds 已提交
1163 1164
	p->io_context = NULL;
	p->audit_context = NULL;
1165 1166
	if (clone_flags & CLONE_THREAD)
		threadgroup_fork_read_lock(current);
1167
	cgroup_fork(p);
L
Linus Torvalds 已提交
1168
#ifdef CONFIG_NUMA
1169
	p->mempolicy = mpol_dup(p->mempolicy);
L
Linus Torvalds 已提交
1170 1171 1172
 	if (IS_ERR(p->mempolicy)) {
 		retval = PTR_ERR(p->mempolicy);
 		p->mempolicy = NULL;
1173
 		goto bad_fork_cleanup_cgroup;
L
Linus Torvalds 已提交
1174
 	}
1175
	mpol_fix_fork_child_flag(p);
L
Linus Torvalds 已提交
1176
#endif
1177 1178
#ifdef CONFIG_TRACE_IRQFLAGS
	p->irq_events = 0;
1179 1180 1181
#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
	p->hardirqs_enabled = 1;
#else
1182
	p->hardirqs_enabled = 0;
1183
#endif
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195
	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 已提交
1196 1197 1198 1199 1200
#ifdef CONFIG_LOCKDEP
	p->lockdep_depth = 0; /* no locks held yet */
	p->curr_chain_key = 0;
	p->lockdep_recursion = 0;
#endif
L
Linus Torvalds 已提交
1201

1202 1203 1204
#ifdef CONFIG_DEBUG_MUTEXES
	p->blocked_on = NULL; /* not blocked yet */
#endif
1205 1206 1207 1208
#ifdef CONFIG_CGROUP_MEM_RES_CTLR
	p->memcg_batch.do_batch = 0;
	p->memcg_batch.memcg = NULL;
#endif
1209

1210
	/* Perform scheduler related setup. Assign this task to a CPU. */
1211
	sched_fork(p);
1212

1213
	retval = perf_event_init_task(p);
1214 1215
	if (retval)
		goto bad_fork_cleanup_policy;
1216

L
Linus Torvalds 已提交
1217
	if ((retval = audit_alloc(p)))
1218
		goto bad_fork_cleanup_policy;
L
Linus Torvalds 已提交
1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
	/* copy all the process information */
	if ((retval = copy_semundo(clone_flags, p)))
		goto bad_fork_cleanup_audit;
	if ((retval = copy_files(clone_flags, p)))
		goto bad_fork_cleanup_semundo;
	if ((retval = copy_fs(clone_flags, p)))
		goto bad_fork_cleanup_files;
	if ((retval = copy_sighand(clone_flags, p)))
		goto bad_fork_cleanup_fs;
	if ((retval = copy_signal(clone_flags, p)))
		goto bad_fork_cleanup_sighand;
	if ((retval = copy_mm(clone_flags, p)))
		goto bad_fork_cleanup_signal;
S
Serge E. Hallyn 已提交
1232
	if ((retval = copy_namespaces(clone_flags, p)))
D
David Howells 已提交
1233
		goto bad_fork_cleanup_mm;
1234
	if ((retval = copy_io(clone_flags, p)))
1235
		goto bad_fork_cleanup_namespaces;
A
Alexey Dobriyan 已提交
1236
	retval = copy_thread(clone_flags, stack_start, stack_size, p, regs);
L
Linus Torvalds 已提交
1237
	if (retval)
1238
		goto bad_fork_cleanup_io;
L
Linus Torvalds 已提交
1239

1240 1241
	if (pid != &init_struct_pid) {
		retval = -ENOMEM;
1242
		pid = alloc_pid(p->nsproxy->pid_ns);
1243
		if (!pid)
1244
			goto bad_fork_cleanup_io;
1245 1246 1247 1248 1249 1250 1251
	}

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

L
Linus Torvalds 已提交
1252 1253 1254 1255 1256
	p->set_child_tid = (clone_flags & CLONE_CHILD_SETTID) ? child_tidptr : NULL;
	/*
	 * Clear TID on mm_release()?
	 */
	p->clear_child_tid = (clone_flags & CLONE_CHILD_CLEARTID) ? child_tidptr: NULL;
1257 1258 1259
#ifdef CONFIG_BLOCK
	p->plug = NULL;
#endif
1260
#ifdef CONFIG_FUTEX
1261 1262 1263 1264
	p->robust_list = NULL;
#ifdef CONFIG_COMPAT
	p->compat_robust_list = NULL;
#endif
1265 1266
	INIT_LIST_HEAD(&p->pi_state_list);
	p->pi_state_cache = NULL;
1267
#endif
1268 1269 1270 1271 1272 1273
	/*
	 * 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 已提交
1274
	/*
1275 1276
	 * Syscall tracing and stepping should be turned off in the
	 * child regardless of CLONE_PTRACE.
L
Linus Torvalds 已提交
1277
	 */
1278
	user_disable_single_step(p);
L
Linus Torvalds 已提交
1279
	clear_tsk_thread_flag(p, TIF_SYSCALL_TRACE);
1280 1281 1282
#ifdef TIF_SYSCALL_EMU
	clear_tsk_thread_flag(p, TIF_SYSCALL_EMU);
#endif
A
Arjan van de Ven 已提交
1283
	clear_all_latency_tracing(p);
L
Linus Torvalds 已提交
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294

	/* ok, now we should be set up.. */
	p->exit_signal = (clone_flags & CLONE_THREAD) ? -1 : (clone_flags & CSIGNAL);
	p->pdeath_signal = 0;
	p->exit_state = 0;

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

1297 1298 1299 1300 1301 1302
	/* 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 已提交
1303 1304 1305 1306
	/* Need tasklist lock for parent etc handling! */
	write_lock_irq(&tasklist_lock);

	/* CLONE_PARENT re-uses the old parent */
1307
	if (clone_flags & (CLONE_PARENT|CLONE_THREAD)) {
L
Linus Torvalds 已提交
1308
		p->real_parent = current->real_parent;
1309 1310
		p->parent_exec_id = current->parent_exec_id;
	} else {
L
Linus Torvalds 已提交
1311
		p->real_parent = current;
1312 1313
		p->parent_exec_id = current->self_exec_id;
	}
L
Linus Torvalds 已提交
1314

1315
	spin_lock(&current->sighand->siglock);
1316 1317 1318 1319 1320 1321 1322 1323 1324

	/*
	 * 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).
 	 */
D
Daniel Walker 已提交
1325
	recalc_sigpending();
1326 1327 1328 1329
	if (signal_pending(current)) {
		spin_unlock(&current->sighand->siglock);
		write_unlock_irq(&tasklist_lock);
		retval = -ERESTARTNOINTR;
1330
		goto bad_fork_free_pid;
1331 1332
	}

L
Linus Torvalds 已提交
1333
	if (clone_flags & CLONE_THREAD) {
1334
		current->signal->nr_threads++;
1335
		atomic_inc(&current->signal->live);
1336
		atomic_inc(&current->signal->sigcnt);
L
Linus Torvalds 已提交
1337
		p->group_leader = current->group_leader;
O
Oleg Nesterov 已提交
1338
		list_add_tail_rcu(&p->thread_group, &p->group_leader->thread_group);
L
Linus Torvalds 已提交
1339 1340
	}

1341
	if (likely(p->pid)) {
T
Tejun Heo 已提交
1342
		ptrace_init_task(p, (clone_flags & CLONE_PTRACE) || trace);
1343 1344

		if (thread_group_leader(p)) {
1345
			if (is_child_reaper(pid))
1346
				p->nsproxy->pid_ns->child_reaper = p;
1347

1348
			p->signal->leader_pid = pid;
A
Alan Cox 已提交
1349
			p->signal->tty = tty_kref_get(current->signal->tty);
E
Eric W. Biederman 已提交
1350 1351
			attach_pid(p, PIDTYPE_PGID, task_pgrp(current));
			attach_pid(p, PIDTYPE_SID, task_session(current));
1352
			list_add_tail(&p->sibling, &p->real_parent->children);
1353
			list_add_tail_rcu(&p->tasks, &init_task.tasks);
1354
			__this_cpu_inc(process_counts);
1355
		}
1356
		attach_pid(p, PIDTYPE_PID, pid);
1357
		nr_threads++;
L
Linus Torvalds 已提交
1358 1359 1360
	}

	total_forks++;
1361
	spin_unlock(&current->sighand->siglock);
L
Linus Torvalds 已提交
1362
	write_unlock_irq(&tasklist_lock);
1363
	proc_fork_connector(p);
1364
	cgroup_post_fork(p);
1365 1366
	if (clone_flags & CLONE_THREAD)
		threadgroup_fork_read_unlock(current);
1367
	perf_event_fork(p);
L
Linus Torvalds 已提交
1368 1369
	return p;

1370 1371 1372
bad_fork_free_pid:
	if (pid != &init_struct_pid)
		free_pid(pid);
1373
bad_fork_cleanup_io:
1374 1375
	if (p->io_context)
		exit_io_context(p);
S
Serge E. Hallyn 已提交
1376
bad_fork_cleanup_namespaces:
1377
	exit_task_namespaces(p);
L
Linus Torvalds 已提交
1378
bad_fork_cleanup_mm:
Y
Ying Han 已提交
1379 1380 1381 1382 1383
	if (p->mm) {
		task_lock(p);
		if (p->signal->oom_score_adj == OOM_SCORE_ADJ_MIN)
			atomic_dec(&p->mm->oom_disable_count);
		task_unlock(p);
L
Linus Torvalds 已提交
1384
		mmput(p->mm);
Y
Ying Han 已提交
1385
	}
L
Linus Torvalds 已提交
1386
bad_fork_cleanup_signal:
1387
	if (!(clone_flags & CLONE_THREAD))
1388
		free_signal_struct(p->signal);
L
Linus Torvalds 已提交
1389
bad_fork_cleanup_sighand:
1390
	__cleanup_sighand(p->sighand);
L
Linus Torvalds 已提交
1391 1392 1393 1394 1395 1396 1397 1398 1399
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:
1400
	perf_event_free_task(p);
L
Linus Torvalds 已提交
1401
#ifdef CONFIG_NUMA
1402
	mpol_put(p->mempolicy);
1403
bad_fork_cleanup_cgroup:
L
Linus Torvalds 已提交
1404
#endif
1405 1406
	if (clone_flags & CLONE_THREAD)
		threadgroup_fork_read_unlock(current);
1407
	cgroup_exit(p, cgroup_callbacks_done);
1408
	delayacct_tsk_free(p);
A
Al Viro 已提交
1409
	module_put(task_thread_info(p)->exec_domain->module);
L
Linus Torvalds 已提交
1410
bad_fork_cleanup_count:
D
David Howells 已提交
1411
	atomic_dec(&p->cred->user->processes);
1412
	exit_creds(p);
L
Linus Torvalds 已提交
1413 1414
bad_fork_free:
	free_task(p);
1415 1416
fork_out:
	return ERR_PTR(retval);
L
Linus Torvalds 已提交
1417 1418
}

A
Adrian Bunk 已提交
1419
noinline struct pt_regs * __cpuinit __attribute__((weak)) idle_regs(struct pt_regs *regs)
L
Linus Torvalds 已提交
1420 1421 1422 1423 1424
{
	memset(regs, 0, sizeof(struct pt_regs));
	return regs;
}

1425 1426 1427 1428 1429 1430 1431 1432 1433 1434
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;
	}
}

1435
struct task_struct * __cpuinit fork_idle(int cpu)
L
Linus Torvalds 已提交
1436
{
1437
	struct task_struct *task;
L
Linus Torvalds 已提交
1438 1439
	struct pt_regs regs;

1440
	task = copy_process(CLONE_VM, 0, idle_regs(&regs), 0, NULL,
R
Roland McGrath 已提交
1441
			    &init_struct_pid, 0);
1442 1443
	if (!IS_ERR(task)) {
		init_idle_pids(task->pids);
1444
		init_idle(task, cpu);
1445
	}
1446

L
Linus Torvalds 已提交
1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464
	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;
1465
	long nr;
L
Linus Torvalds 已提交
1466

1467 1468 1469 1470 1471 1472 1473 1474 1475 1476
	/*
	 * 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
		 */
1477 1478
		if (!capable(CAP_SYS_ADMIN) || !capable(CAP_SETUID) ||
				!capable(CAP_SETGID))
1479 1480 1481
			return -EPERM;
	}

R
Roland McGrath 已提交
1482
	/*
T
Tejun Heo 已提交
1483 1484 1485 1486
	 * 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 已提交
1487
	 */
T
Tejun Heo 已提交
1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498
	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 已提交
1499

1500
	p = copy_process(clone_flags, stack_start, regs, stack_size,
R
Roland McGrath 已提交
1501
			 child_tidptr, NULL, trace);
L
Linus Torvalds 已提交
1502 1503 1504 1505 1506 1507 1508
	/*
	 * 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;

1509 1510
		trace_sched_process_fork(current, p);

1511
		nr = task_pid_vnr(p);
1512 1513 1514

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

L
Linus Torvalds 已提交
1516 1517 1518 1519 1520
		if (clone_flags & CLONE_VFORK) {
			p->vfork_done = &vfork;
			init_completion(&vfork);
		}

1521
		audit_finish_fork(p);
T
Tejun Heo 已提交
1522

R
Roland McGrath 已提交
1523 1524 1525
		/*
		 * We set PF_STARTING at creation in case tracing wants to
		 * use this to distinguish a fully live task from one that
T
Tejun Heo 已提交
1526 1527
		 * hasn't finished SIGSTOP raising yet.  Now we clear it
		 * and set the child going.
R
Roland McGrath 已提交
1528 1529 1530
		 */
		p->flags &= ~PF_STARTING;

1531
		wake_up_new_task(p);
L
Linus Torvalds 已提交
1532

T
Tejun Heo 已提交
1533 1534 1535
		/* forking complete and child started to run, tell ptracer */
		if (unlikely(trace))
			ptrace_event(trace, nr);
R
Roland McGrath 已提交
1536

L
Linus Torvalds 已提交
1537
		if (clone_flags & CLONE_VFORK) {
R
Rafael J. Wysocki 已提交
1538
			freezer_do_not_count();
L
Linus Torvalds 已提交
1539
			wait_for_completion(&vfork);
R
Rafael J. Wysocki 已提交
1540
			freezer_count();
T
Tejun Heo 已提交
1541
			ptrace_event(PTRACE_EVENT_VFORK_DONE, nr);
L
Linus Torvalds 已提交
1542 1543
		}
	} else {
1544
		nr = PTR_ERR(p);
L
Linus Torvalds 已提交
1545
	}
1546
	return nr;
L
Linus Torvalds 已提交
1547 1548
}

1549 1550 1551 1552
#ifndef ARCH_MIN_MMSTRUCT_ALIGN
#define ARCH_MIN_MMSTRUCT_ALIGN 0
#endif

1553
static void sighand_ctor(void *data)
1554 1555 1556
{
	struct sighand_struct *sighand = data;

C
Christoph Lameter 已提交
1557
	spin_lock_init(&sighand->siglock);
D
Davide Libenzi 已提交
1558
	init_waitqueue_head(&sighand->signalfd_wqh);
1559 1560
}

L
Linus Torvalds 已提交
1561 1562 1563 1564
void __init proc_caches_init(void)
{
	sighand_cachep = kmem_cache_create("sighand_cache",
			sizeof(struct sighand_struct), 0,
V
Vegard Nossum 已提交
1565 1566
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_DESTROY_BY_RCU|
			SLAB_NOTRACK, sighand_ctor);
L
Linus Torvalds 已提交
1567 1568
	signal_cachep = kmem_cache_create("signal_cache",
			sizeof(struct signal_struct), 0,
V
Vegard Nossum 已提交
1569
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1570
	files_cachep = kmem_cache_create("files_cache",
L
Linus Torvalds 已提交
1571
			sizeof(struct files_struct), 0,
V
Vegard Nossum 已提交
1572
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1573
	fs_cachep = kmem_cache_create("fs_cache",
L
Linus Torvalds 已提交
1574
			sizeof(struct fs_struct), 0,
V
Vegard Nossum 已提交
1575
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1576 1577 1578 1579 1580 1581 1582
	/*
	 * 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 已提交
1583
	mm_cachep = kmem_cache_create("mm_struct",
1584
			sizeof(struct mm_struct), ARCH_MIN_MMSTRUCT_ALIGN,
V
Vegard Nossum 已提交
1585
			SLAB_HWCACHE_ALIGN|SLAB_PANIC|SLAB_NOTRACK, NULL);
1586
	vm_area_cachep = KMEM_CACHE(vm_area_struct, SLAB_PANIC);
1587
	mmap_init();
L
Linus Torvalds 已提交
1588
}
1589 1590

/*
1591
 * Check constraints on flags passed to the unshare system call.
1592
 */
1593
static int check_unshare_flags(unsigned long unshare_flags)
1594
{
1595 1596 1597 1598
	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;
1599
	/*
1600 1601 1602
	 * Not implemented, but pretend it works if there is nothing to
	 * unshare. Note that unsharing CLONE_THREAD or CLONE_SIGHAND
	 * needs to unshare vm.
1603
	 */
1604 1605 1606 1607 1608
	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;
	}
1609 1610 1611 1612 1613

	return 0;
}

/*
1614
 * Unshare the filesystem structure if it is being shared
1615 1616 1617 1618 1619
 */
static int unshare_fs(unsigned long unshare_flags, struct fs_struct **new_fsp)
{
	struct fs_struct *fs = current->fs;

A
Al Viro 已提交
1620 1621 1622 1623 1624 1625 1626 1627 1628 1629
	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;
1630 1631 1632 1633 1634

	return 0;
}

/*
1635
 * Unshare file descriptor table if it is being shared
1636 1637 1638 1639
 */
static int unshare_fd(unsigned long unshare_flags, struct files_struct **new_fdp)
{
	struct files_struct *fd = current->files;
1640
	int error = 0;
1641 1642

	if ((unshare_flags & CLONE_FILES) &&
1643 1644 1645 1646 1647
	    (fd && atomic_read(&fd->count) > 1)) {
		*new_fdp = dup_fd(fd, &error);
		if (!*new_fdp)
			return error;
	}
1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659

	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.
 */
1660
SYSCALL_DEFINE1(unshare, unsigned long, unshare_flags)
1661 1662 1663
{
	struct fs_struct *fs, *new_fs = NULL;
	struct files_struct *fd, *new_fd = NULL;
1664
	struct nsproxy *new_nsproxy = NULL;
1665
	int do_sysvsem = 0;
1666
	int err;
1667

1668 1669
	err = check_unshare_flags(unshare_flags);
	if (err)
1670 1671
		goto bad_unshare_out;

1672 1673 1674 1675 1676
	/*
	 * If unsharing namespace, must also unshare filesystem information.
	 */
	if (unshare_flags & CLONE_NEWNS)
		unshare_flags |= CLONE_FS;
1677 1678 1679 1680 1681 1682
	/*
	 * 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))
1683
		do_sysvsem = 1;
1684
	if ((err = unshare_fs(unshare_flags, &new_fs)))
1685
		goto bad_unshare_out;
1686
	if ((err = unshare_fd(unshare_flags, &new_fd)))
1687
		goto bad_unshare_cleanup_fs;
1688 1689
	if ((err = unshare_nsproxy_namespaces(unshare_flags, &new_nsproxy,
			new_fs)))
1690
		goto bad_unshare_cleanup_fd;
1691

1692
	if (new_fs || new_fd || do_sysvsem || new_nsproxy) {
1693 1694 1695 1696 1697 1698
		if (do_sysvsem) {
			/*
			 * CLONE_SYSVSEM is equivalent to sys_exit().
			 */
			exit_sem(current);
		}
S
Serge E. Hallyn 已提交
1699

1700
		if (new_nsproxy) {
1701 1702
			switch_task_namespaces(current, new_nsproxy);
			new_nsproxy = NULL;
1703
		}
1704

1705 1706
		task_lock(current);

1707 1708
		if (new_fs) {
			fs = current->fs;
N
Nick Piggin 已提交
1709
			spin_lock(&fs->lock);
1710
			current->fs = new_fs;
A
Al Viro 已提交
1711 1712 1713 1714
			if (--fs->users)
				new_fs = NULL;
			else
				new_fs = fs;
N
Nick Piggin 已提交
1715
			spin_unlock(&fs->lock);
1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726
		}

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

		task_unlock(current);
	}

1727
	if (new_nsproxy)
1728
		put_nsproxy(new_nsproxy);
1729

1730 1731 1732 1733 1734 1735
bad_unshare_cleanup_fd:
	if (new_fd)
		put_files_struct(new_fd);

bad_unshare_cleanup_fs:
	if (new_fs)
A
Al Viro 已提交
1736
		free_fs_struct(new_fs);
1737 1738 1739 1740

bad_unshare_out:
	return err;
}
1741 1742 1743 1744 1745 1746 1747 1748 1749 1750

/*
 *	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;
1751
	struct files_struct *copy = NULL;
1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764
	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;
}