xref: /xnu-10002.1.13/bsd/kern/kern_fork.c (revision 1031c584a5e37aff177559b9f69dbd3c8c3fd30a)
1 /*
2  * Copyright (c) 2000-2020 Apple Inc. All rights reserved.
3  *
4  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5  *
6  * This file contains Original Code and/or Modifications of Original Code
7  * as defined in and that are subject to the Apple Public Source License
8  * Version 2.0 (the 'License'). You may not use this file except in
9  * compliance with the License. The rights granted to you under the License
10  * may not be used to create, or enable the creation or redistribution of,
11  * unlawful or unlicensed copies of an Apple operating system, or to
12  * circumvent, violate, or enable the circumvention or violation of, any
13  * terms of an Apple operating system software license agreement.
14  *
15  * Please obtain a copy of the License at
16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17  *
18  * The Original Code and all software distributed under the License are
19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23  * Please see the License for the specific language governing rights and
24  * limitations under the License.
25  *
26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27  */
28 /* Copyright (c) 1995, 1997 Apple Computer, Inc. All Rights Reserved */
29 /*
30  * Copyright (c) 1982, 1986, 1989, 1991, 1993
31  *	The Regents of the University of California.  All rights reserved.
32  * (c) UNIX System Laboratories, Inc.
33  * All or some portions of this file are derived from material licensed
34  * to the University of California by American Telephone and Telegraph
35  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
36  * the permission of UNIX System Laboratories, Inc.
37  *
38  * Redistribution and use in source and binary forms, with or without
39  * modification, are permitted provided that the following conditions
40  * are met:
41  * 1. Redistributions of source code must retain the above copyright
42  *    notice, this list of conditions and the following disclaimer.
43  * 2. Redistributions in binary form must reproduce the above copyright
44  *    notice, this list of conditions and the following disclaimer in the
45  *    documentation and/or other materials provided with the distribution.
46  * 3. All advertising materials mentioning features or use of this software
47  *    must display the following acknowledgement:
48  *	This product includes software developed by the University of
49  *	California, Berkeley and its contributors.
50  * 4. Neither the name of the University nor the names of its contributors
51  *    may be used to endorse or promote products derived from this software
52  *    without specific prior written permission.
53  *
54  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
55  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
56  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
57  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
58  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
59  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
60  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
61  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
62  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
63  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
64  * SUCH DAMAGE.
65  *
66  *	@(#)kern_fork.c	8.8 (Berkeley) 2/14/95
67  */
68 /*
69  * NOTICE: This file was modified by McAfee Research in 2004 to introduce
70  * support for mandatory and extensible security protections.  This notice
71  * is included in support of clause 2.2 (b) of the Apple Public License,
72  * Version 2.0.
73  */
74 /*
75  * NOTICE: This file was modified by SPARTA, Inc. in 2005 to introduce
76  * support for mandatory and extensible security protections.  This notice
77  * is included in support of clause 2.2 (b) of the Apple Public License,
78  * Version 2.0.
79  */
80 
81 #include <kern/assert.h>
82 #include <kern/bits.h>
83 #include <sys/param.h>
84 #include <sys/systm.h>
85 #include <sys/filedesc.h>
86 #include <sys/kernel.h>
87 #include <sys/malloc.h>
88 #include <sys/proc_internal.h>
89 #include <sys/kauth.h>
90 #include <sys/user.h>
91 #include <sys/reason.h>
92 #include <sys/resourcevar.h>
93 #include <sys/vnode_internal.h>
94 #include <sys/file_internal.h>
95 #include <sys/acct.h>
96 #include <sys/codesign.h>
97 #include <sys/sysent.h>
98 #include <sys/sysproto.h>
99 #include <sys/ulock.h>
100 #if CONFIG_PERSONAS
101 #include <sys/persona.h>
102 #endif
103 #include <sys/doc_tombstone.h>
104 #if CONFIG_DTRACE
105 /* Do not include dtrace.h, it redefines kmem_[alloc/free] */
106 extern void (*dtrace_proc_waitfor_exec_ptr)(proc_t);
107 extern void dtrace_proc_fork(proc_t, proc_t, int);
108 
109 /*
110  * Since dtrace_proc_waitfor_exec_ptr can be added/removed in dtrace_subr.c,
111  * we will store its value before actually calling it.
112  */
113 static void (*dtrace_proc_waitfor_hook)(proc_t) = NULL;
114 
115 #include <sys/dtrace_ptss.h>
116 #endif
117 
118 #include <security/audit/audit.h>
119 
120 #include <mach/mach_types.h>
121 #include <kern/coalition.h>
122 #include <kern/kern_types.h>
123 #include <kern/kalloc.h>
124 #include <kern/mach_param.h>
125 #include <kern/task.h>
126 #include <kern/thread.h>
127 #include <kern/thread_call.h>
128 #include <kern/zalloc.h>
129 
130 #include <os/log.h>
131 
132 #if CONFIG_MACF
133 #include <security/mac_framework.h>
134 #include <security/mac_mach_internal.h>
135 #endif
136 
137 #include <vm/vm_map.h>
138 #include <vm/vm_protos.h>
139 #include <vm/vm_shared_region.h>
140 
141 #include <sys/shm_internal.h>   /* for shmfork() */
142 #include <mach/task.h>          /* for thread_create() */
143 #include <mach/thread_act.h>    /* for thread_resume() */
144 
145 #include <sys/sdt.h>
146 
147 #if CONFIG_MEMORYSTATUS
148 #include <sys/kern_memorystatus.h>
149 #endif
150 
151 /* XXX routines which should have Mach prototypes, but don't */
152 void thread_set_parent(thread_t parent, int pid);
153 extern void act_thread_catt(void *ctx);
154 void thread_set_child(thread_t child, int pid);
155 boolean_t thread_is_active(thread_t thread);
156 void *act_thread_csave(void);
157 extern boolean_t task_is_exec_copy(task_t);
158 int nextpidversion = 0;
159 
160 void ipc_task_enable(task_t task);
161 
162 proc_t forkproc(proc_t, cloneproc_flags_t);
163 void forkproc_free(proc_t);
164 thread_t fork_create_child(task_t parent_task,
165     coalition_t *parent_coalitions,
166     proc_t child,
167     int is_64bit_addr,
168     int is_64bit_data,
169     cloneproc_flags_t clone_flags);
170 
171 __private_extern__ const size_t uthread_size = sizeof(struct uthread);
172 static LCK_GRP_DECLARE(rethrottle_lock_grp, "rethrottle");
173 
174 os_refgrp_decl(, p_refgrp, "proc", NULL);
175 
176 extern const size_t task_alignment;
177 const size_t proc_alignment = _Alignof(struct proc);
178 
179 extern size_t task_struct_size;
180 size_t proc_struct_size = sizeof(struct proc);
181 size_t proc_and_task_size;
182 
183 ZONE_DECLARE_ID(ZONE_ID_PROC_TASK, struct proc);
184 SECURITY_READ_ONLY_LATE(zone_t) proc_task_zone;
185 
186 KALLOC_TYPE_DEFINE(proc_stats_zone, struct pstats, KT_DEFAULT);
187 
188 /*
189  * fork1
190  *
191  * Description:	common code used by all new process creation other than the
192  *		bootstrap of the initial process on the system
193  *
194  * Parameters: parent_proc		parent process of the process being
195  *		child_threadp		pointer to location to receive the
196  *					Mach thread_t of the child process
197  *					created
198  *		kind			kind of creation being requested
199  *		coalitions		if spawn, the set of coalitions the
200  *					child process should join, or NULL to
201  *					inherit the parent's. On non-spawns,
202  *					this param is ignored and the child
203  *					always inherits the parent's
204  *					coalitions.
205  *
206  * Notes:	Permissable values for 'kind':
207  *
208  *		PROC_CREATE_FORK	Create a complete process which will
209  *					return actively running in both the
210  *					parent and the child; the child copies
211  *					the parent address space.
212  *		PROC_CREATE_SPAWN	Create a complete process which will
213  *					return actively running in the parent
214  *					only after returning actively running
215  *					in the child; the child address space
216  *					is newly created by an image activator,
217  *					after which the child is run.
218  *
219  *		At first it may seem strange that we return the child thread
220  *		address rather than process structure, since the process is
221  *		the only part guaranteed to be "new"; however, since we do
222  *		not actualy adjust other references between Mach and BSD, this
223  *		is the only method which guarantees us the ability to get
224  *		back to the other information.
225  */
226 int
fork1(proc_t parent_proc,thread_t * child_threadp,int kind,coalition_t * coalitions)227 fork1(proc_t parent_proc, thread_t *child_threadp, int kind, coalition_t *coalitions)
228 {
229 	proc_t child_proc = NULL;       /* set in switch, but compiler... */
230 	thread_t child_thread = NULL;
231 	uid_t uid;
232 	size_t count;
233 	int err = 0;
234 	int spawn = 0;
235 	rlim_t rlimit_nproc_cur;
236 
237 	/*
238 	 * Although process entries are dynamically created, we still keep
239 	 * a global limit on the maximum number we will create.  Don't allow
240 	 * a nonprivileged user to use the last process; don't let root
241 	 * exceed the limit. The variable nprocs is the current number of
242 	 * processes, maxproc is the limit.
243 	 */
244 	uid = kauth_getruid();
245 	proc_list_lock();
246 	if ((nprocs >= maxproc - 1 && uid != 0) || nprocs >= maxproc) {
247 #if (DEVELOPMENT || DEBUG) && !defined(XNU_TARGET_OS_OSX)
248 		/*
249 		 * On the development kernel, panic so that the fact that we hit
250 		 * the process limit is obvious, as this may very well wedge the
251 		 * system.
252 		 */
253 		panic("The process table is full; parent pid=%d", proc_getpid(parent_proc));
254 #endif
255 		proc_list_unlock();
256 		tablefull("proc");
257 		return EAGAIN;
258 	}
259 	proc_list_unlock();
260 
261 	/*
262 	 * Increment the count of procs running with this uid. Don't allow
263 	 * a nonprivileged user to exceed their current limit, which is
264 	 * always less than what an rlim_t can hold.
265 	 * (locking protection is provided by list lock held in chgproccnt)
266 	 */
267 	count = chgproccnt(uid, 1);
268 	rlimit_nproc_cur = proc_limitgetcur(parent_proc, RLIMIT_NPROC);
269 	if (uid != 0 &&
270 	    (rlim_t)count > rlimit_nproc_cur) {
271 #if (DEVELOPMENT || DEBUG) && !defined(XNU_TARGET_OS_OSX)
272 		/*
273 		 * On the development kernel, panic so that the fact that we hit
274 		 * the per user process limit is obvious.  This may be less dire
275 		 * than hitting the global process limit, but we cannot rely on
276 		 * that.
277 		 */
278 		panic("The per-user process limit has been hit; parent pid=%d, uid=%d", proc_getpid(parent_proc), uid);
279 #endif
280 		err = EAGAIN;
281 		goto bad;
282 	}
283 
284 #if CONFIG_MACF
285 	/*
286 	 * Determine if MAC policies applied to the process will allow
287 	 * it to fork.  This is an advisory-only check.
288 	 */
289 	err = mac_proc_check_fork(parent_proc);
290 	if (err != 0) {
291 		goto bad;
292 	}
293 #endif
294 
295 	switch (kind) {
296 	case PROC_CREATE_SPAWN:
297 		/*
298 		 * A spawned process differs from a forked process in that
299 		 * the spawned process does not carry around the parents
300 		 * baggage with regard to address space copying, dtrace,
301 		 * and so on.
302 		 */
303 		spawn = 1;
304 
305 		OS_FALLTHROUGH;
306 
307 	case PROC_CREATE_FORK:
308 		/*
309 		 * When we clone the parent process, we are going to inherit
310 		 * its task attributes and memory, since when we fork, we
311 		 * will, in effect, create a duplicate of it, with only minor
312 		 * differences.  Contrarily, spawned processes do not inherit.
313 		 */
314 		if ((child_thread = cloneproc(proc_task(parent_proc),
315 		    spawn ? coalitions : NULL,
316 		    parent_proc,
317 		    spawn ? CLONEPROC_FLAGS_NONE : CLONEPROC_FLAGS_INHERIT_MEMORY)) == NULL) {
318 			/* Failed to create thread */
319 			err = EAGAIN;
320 			goto bad;
321 		}
322 
323 		/* child_proc = child_thread->task->proc; */
324 		child_proc = (proc_t)(get_bsdtask_info(get_threadtask(child_thread)));
325 
326 		if (!spawn) {
327 			/* Copy current thread state into the child thread (only for fork) */
328 			thread_dup(child_thread);
329 		}
330 
331 // XXX BEGIN: wants to move to be common code (and safe)
332 #if CONFIG_MACF
333 		/*
334 		 * allow policies to associate the credential/label that
335 		 * we referenced from the parent ... with the child
336 		 * JMM - this really isn't safe, as we can drop that
337 		 *       association without informing the policy in other
338 		 *       situations (keep long enough to get policies changed)
339 		 */
340 		mac_cred_label_associate_fork(proc_ucred_unsafe(child_proc),
341 		    child_proc);
342 #endif
343 
344 		/*
345 		 * Propogate change of PID - may get new cred if auditing.
346 		 */
347 		set_security_token(child_proc, proc_ucred_unsafe(child_proc));
348 
349 		AUDIT_ARG(pid, proc_getpid(child_proc));
350 
351 // XXX END: wants to move to be common code (and safe)
352 
353 		/*
354 		 * Blow thread state information; this is what gives the child
355 		 * process its "return" value from a fork() call.
356 		 *
357 		 * Note: this should probably move to fork() proper, since it
358 		 * is not relevent to spawn, and the value won't matter
359 		 * until we resume the child there.  If you are in here
360 		 * refactoring code, consider doing this at the same time.
361 		 */
362 		thread_set_child(child_thread, proc_getpid(child_proc));
363 
364 		child_proc->p_acflag = AFORK;   /* forked but not exec'ed */
365 
366 #if CONFIG_DTRACE
367 		dtrace_proc_fork(parent_proc, child_proc, spawn);
368 #endif  /* CONFIG_DTRACE */
369 		if (!spawn) {
370 			/*
371 			 * Of note, we need to initialize the bank context behind
372 			 * the protection of the proc_trans lock to prevent a race with exit.
373 			 */
374 			task_bank_init(get_threadtask(child_thread));
375 		}
376 
377 		break;
378 
379 	default:
380 		panic("fork1 called with unknown kind %d", kind);
381 		break;
382 	}
383 
384 
385 	/* return the thread pointer to the caller */
386 	*child_threadp = child_thread;
387 
388 bad:
389 	/*
390 	 * In the error case, we return a 0 value for the returned pid (but
391 	 * it is ignored in the trampoline due to the error return); this
392 	 * is probably not necessary.
393 	 */
394 	if (err) {
395 		(void)chgproccnt(uid, -1);
396 	}
397 
398 	return err;
399 }
400 
401 
402 
403 
404 /*
405  * fork_create_child
406  *
407  * Description:	Common operations associated with the creation of a child
408  *		process. Return with new task and first thread's control port movable
409  *      and not pinned.
410  *
411  * Parameters:	parent_task		parent task
412  *		parent_coalitions	parent's set of coalitions
413  *		child_proc			child process
414  *		inherit_memory		TRUE, if the parents address space is
415  *							to be inherited by the child
416  *		is_64bit_addr		TRUE, if the child being created will
417  *							be associated with a 64 bit address space
418  *		is_64bit_data		TRUE if the child being created will use a
419  *                                                       64-bit register state
420  *		in_exec				TRUE, if called from execve or posix spawn set exec
421  *							FALSE, if called from fork or vfexec
422  *
423  * Note:	This code is called in the fork() case, from the execve() call
424  *		graph, from the posix_spawn() call graph (which implicitly
425  *		includes a vfork() equivalent call, and in the system
426  *		bootstrap case.
427  *
428  *		It creates a new task and thread (and as a side effect of the
429  *		thread creation, a uthread) in the parent coalition set, which is
430  *		then associated with the process 'child'.  If the parent
431  *		process address space is to be inherited, then a flag
432  *		indicates that the newly created task should inherit this from
433  *		the child task.
434  *
435  *		As a special concession to bootstrapping the initial process
436  *		in the system, it's possible for 'parent_task' to be TASK_NULL;
437  *		in this case, 'inherit_memory' MUST be FALSE.
438  */
439 thread_t
fork_create_child(task_t parent_task,coalition_t * parent_coalitions,proc_t child_proc,int is_64bit_addr,int is_64bit_data,cloneproc_flags_t clone_flags)440 fork_create_child(task_t parent_task,
441     coalition_t *parent_coalitions,
442     proc_t child_proc,
443     int is_64bit_addr,
444     int is_64bit_data,
445     cloneproc_flags_t clone_flags)
446 {
447 	thread_t        child_thread = NULL;
448 	task_t          child_task;
449 	kern_return_t   result;
450 	proc_ro_t       proc_ro;
451 	bool inherit_memory = !!(clone_flags & CLONEPROC_FLAGS_INHERIT_MEMORY);
452 	bool in_exec = !!(clone_flags & CLONEPROC_FLAGS_FOR_EXEC);
453 	/*
454 	 * Exec complete hook should be called for spawn and exec, but not for fork.
455 	 */
456 	uint8_t returnwaitflags = (!inherit_memory ? TRW_LEXEC_COMPLETE : 0) |
457 	    (TRW_LRETURNWAIT | TRW_LRETURNWAITER);
458 
459 	proc_ro = proc_get_ro(child_proc);
460 	if (proc_ro_task(proc_ro) != NULL) {
461 		panic("Proc_ro_task for newly created proc %p is not NULL", child_proc);
462 	}
463 
464 	child_task = proc_get_task_raw(child_proc);
465 
466 	/*
467 	 * Create a new task for the child process, IPC access to the new task will
468 	 * be set up after task has been fully initialized.
469 	 */
470 	result = task_create_internal(parent_task,
471 	    proc_ro,
472 	    parent_coalitions,
473 	    inherit_memory,
474 	    is_64bit_addr,
475 	    is_64bit_data,
476 	    TF_NONE,
477 	    TF_NONE,
478 	    in_exec ? TPF_EXEC_COPY : TPF_NONE,           /* Mark the task exec copy if in execve */
479 	    returnwaitflags,                              /* All created threads will wait in task_wait_to_return */
480 	    child_task);
481 	if (result != KERN_SUCCESS) {
482 		printf("%s: task_create_internal failed.  Code: %d\n",
483 		    __func__, result);
484 		goto bad;
485 	}
486 
487 	/* Set the child proc process to child task */
488 	proc_set_task(child_proc, child_task);
489 
490 	/* Set child task process to child proc */
491 	set_bsdtask_info(child_task, child_proc);
492 
493 	/* Propagate CPU limit timer from parent */
494 	if (timerisset(&child_proc->p_rlim_cpu)) {
495 		task_vtimer_set(child_task, TASK_VTIMER_RLIM);
496 	}
497 
498 	/*
499 	 * Set child process BSD visible scheduler priority if nice value
500 	 * inherited from parent
501 	 */
502 	if (child_proc->p_nice != 0) {
503 		resetpriority(child_proc);
504 	}
505 
506 	/*
507 	 * Create main thread for the child process. Its control port is not immovable/pinned
508 	 * until main_thread_set_immovable_pinned().
509 	 *
510 	 * The new thread is waiting on the event triggered by 'task_clear_return_wait'
511 	 */
512 	result = main_thread_create_waiting(child_task,
513 	    (thread_continue_t)task_wait_to_return,
514 	    task_get_return_wait_event(child_task),
515 	    &child_thread);
516 
517 	if (result != KERN_SUCCESS) {
518 		printf("%s: thread_create failed. Code: %d\n",
519 		    __func__, result);
520 		task_deallocate(child_task);
521 		child_task = NULL;
522 	}
523 
524 	/*
525 	 * Tag thread as being the first thread in its task.
526 	 */
527 	thread_set_tag(child_thread, THREAD_TAG_MAINTHREAD);
528 
529 bad:
530 	thread_yield_internal(1);
531 
532 	return child_thread;
533 }
534 
535 
536 /*
537  * fork
538  *
539  * Description:	fork system call.
540  *
541  * Parameters:	parent			Parent process to fork
542  *		uap (void)		[unused]
543  *		retval			Return value
544  *
545  * Returns:	0			Success
546  *		EAGAIN			Resource unavailable, try again
547  *
548  * Notes:	Attempts to create a new child process which inherits state
549  *		from the parent process.  If successful, the call returns
550  *		having created an initially suspended child process with an
551  *		extra Mach task and thread reference, for which the thread
552  *		is initially suspended.  Until we resume the child process,
553  *		it is not yet running.
554  *
555  *		The return information to the child is contained in the
556  *		thread state structure of the new child, and does not
557  *		become visible to the child through a normal return process,
558  *		since it never made the call into the kernel itself in the
559  *		first place.
560  *
561  *		After resuming the thread, this function returns directly to
562  *		the parent process which invoked the fork() system call.
563  *
564  * Important:	The child thread_resume occurs before the parent returns;
565  *		depending on scheduling latency, this means that it is not
566  *		deterministic as to whether the parent or child is scheduled
567  *		to run first.  It is entirely possible that the child could
568  *		run to completion prior to the parent running.
569  */
570 int
fork(proc_t parent_proc,__unused struct fork_args * uap,int32_t * retval)571 fork(proc_t parent_proc, __unused struct fork_args *uap, int32_t *retval)
572 {
573 	thread_t child_thread;
574 	int err;
575 
576 	retval[1] = 0;          /* flag parent return for user space */
577 
578 	if ((err = fork1(parent_proc, &child_thread, PROC_CREATE_FORK, NULL)) == 0) {
579 		task_t child_task;
580 		proc_t child_proc;
581 
582 		/* Return to the parent */
583 		child_proc = (proc_t)get_bsdthreadtask_info(child_thread);
584 		retval[0] = proc_getpid(child_proc);
585 
586 		child_task = (task_t)get_threadtask(child_thread);
587 		assert(child_task != TASK_NULL);
588 
589 		/* task_control_port_options has been inherited from parent, apply it */
590 		task_set_immovable_pinned(child_task);
591 		main_thread_set_immovable_pinned(child_thread);
592 
593 		/*
594 		 * Since the task ports for this new task are now set to be immovable,
595 		 * we can enable them.
596 		 */
597 		ipc_task_enable(get_threadtask(child_thread));
598 
599 		/*
600 		 * Drop the signal lock on the child which was taken on our
601 		 * behalf by forkproc()/cloneproc() to prevent signals being
602 		 * received by the child in a partially constructed state.
603 		 */
604 		proc_signalend(child_proc, 0);
605 		proc_transend(child_proc, 0);
606 
607 		/* flag the fork has occurred */
608 		proc_knote(parent_proc, NOTE_FORK | proc_getpid(child_proc));
609 		DTRACE_PROC1(create, proc_t, child_proc);
610 
611 #if CONFIG_DTRACE
612 		if ((dtrace_proc_waitfor_hook = dtrace_proc_waitfor_exec_ptr) != NULL) {
613 			(*dtrace_proc_waitfor_hook)(child_proc);
614 		}
615 #endif
616 
617 		/*
618 		 * If current process died during the fork, the child would contain
619 		 * non consistent vmmap, kill the child and reap it internally.
620 		 */
621 		if (parent_proc->p_lflag & P_LEXIT || !thread_is_active(current_thread())) {
622 			task_terminate_internal(child_task);
623 			proc_list_lock();
624 			child_proc->p_listflag |= P_LIST_DEADPARENT;
625 			proc_list_unlock();
626 		}
627 
628 		/* "Return" to the child */
629 		task_clear_return_wait(get_threadtask(child_thread), TCRW_CLEAR_ALL_WAIT);
630 
631 		/* drop the extra references we got during the creation */
632 		task_deallocate(child_task);
633 		thread_deallocate(child_thread);
634 	}
635 
636 	return err;
637 }
638 
639 
640 /*
641  * cloneproc
642  *
643  * Description: Create a new process from a specified process.
644  *
645  * Parameters:	parent_task		The parent task to be cloned, or
646  *					TASK_NULL is task characteristics
647  *					are not to be inherited
648  *					be cloned, or TASK_NULL if the new
649  *					task is not to inherit the VM
650  *					characteristics of the parent
651  *		parent_proc		The parent process to be cloned
652  *		clone_flags		Clone flags to specify if the cloned
653  *					process should inherit memory,
654  *					marked as memory stat internal,
655  *					or if the cloneproc is called for exec.
656  *
657  * Returns:	!NULL			pointer to new child thread
658  *		NULL			Failure (unspecified)
659  *
660  * Note:	On return newly created child process has signal lock held
661  *		to block delivery of signal to it if called with lock set.
662  *		fork() code needs to explicity remove this lock before
663  *		signals can be delivered
664  *
665  *		In the case of bootstrap, this function can be called from
666  *		bsd_utaskbootstrap() in order to bootstrap the first process;
667  *		the net effect is to provide a uthread structure for the
668  *		kernel process associated with the kernel task.
669  *
670  * XXX:		Tristating using the value parent_task as the major key
671  *		and inherit_memory as the minor key is something we should
672  *		refactor later; we owe the current semantics, ultimately,
673  *		to the semantics of task_create_internal.  For now, we will
674  *		live with this being somewhat awkward.
675  */
676 thread_t
cloneproc(task_t parent_task,coalition_t * parent_coalitions,proc_t parent_proc,cloneproc_flags_t clone_flags)677 cloneproc(task_t parent_task, coalition_t *parent_coalitions, proc_t parent_proc, cloneproc_flags_t clone_flags)
678 {
679 #if !CONFIG_MEMORYSTATUS
680 #pragma unused(memstat_internal)
681 #endif
682 	task_t child_task;
683 	proc_t child_proc;
684 	thread_t child_thread = NULL;
685 	bool memstat_internal = !!(clone_flags & CLONEPROC_FLAGS_MEMSTAT_INTERNAL);
686 	bool in_exec = !!(clone_flags & CLONEPROC_FLAGS_FOR_EXEC);
687 
688 	if ((child_proc = forkproc(parent_proc, clone_flags)) == NULL) {
689 		/* Failed to allocate new process */
690 		goto bad;
691 	}
692 
693 	/*
694 	 * In the case where the parent_task is TASK_NULL (during the init path)
695 	 * we make the assumption that the register size will be the same as the
696 	 * address space size since there's no way to determine the possible
697 	 * register size until an image is exec'd.
698 	 *
699 	 * The only architecture that has different address space and register sizes
700 	 * (arm64_32) isn't being used within kernel-space, so the above assumption
701 	 * always holds true for the init path.
702 	 */
703 	const int parent_64bit_addr = parent_proc->p_flag & P_LP64;
704 	const int parent_64bit_data = (parent_task == TASK_NULL) ? parent_64bit_addr : task_get_64bit_data(parent_task);
705 
706 	child_thread = fork_create_child(parent_task,
707 	    parent_coalitions,
708 	    child_proc,
709 	    parent_64bit_addr,
710 	    parent_64bit_data,
711 	    clone_flags);
712 
713 	if (child_thread == NULL) {
714 		/*
715 		 * Failed to create thread; now we must deconstruct the new
716 		 * process previously obtained from forkproc().
717 		 */
718 		forkproc_free(child_proc);
719 		goto bad;
720 	}
721 
722 	child_task = get_threadtask(child_thread);
723 	if (parent_64bit_addr) {
724 		OSBitOrAtomic(P_LP64, (UInt32 *)&child_proc->p_flag);
725 		get_bsdthread_info(child_thread)->uu_flag |= UT_LP64;
726 	} else {
727 		OSBitAndAtomic(~((uint32_t)P_LP64), (UInt32 *)&child_proc->p_flag);
728 		get_bsdthread_info(child_thread)->uu_flag &= ~UT_LP64;
729 	}
730 
731 #if CONFIG_MEMORYSTATUS
732 	if (memstat_internal ||
733 	    (in_exec && (parent_proc->p_memstat_state & P_MEMSTAT_INTERNAL))) {
734 		proc_list_lock();
735 		child_proc->p_memstat_state |= P_MEMSTAT_INTERNAL;
736 		proc_list_unlock();
737 	}
738 	if (in_exec && parent_proc->p_memstat_relaunch_flags != P_MEMSTAT_RELAUNCH_UNKNOWN) {
739 		memorystatus_relaunch_flags_update(child_proc, parent_proc->p_memstat_relaunch_flags);
740 	}
741 #endif
742 
743 	/* make child visible */
744 	pinsertchild(parent_proc, child_proc, in_exec);
745 
746 	/*
747 	 * Make child runnable, set start time.
748 	 */
749 	child_proc->p_stat = SRUN;
750 bad:
751 	return child_thread;
752 }
753 
754 void
proc_set_sigact(proc_t p,int sig,user_addr_t sigact)755 proc_set_sigact(proc_t p, int sig, user_addr_t sigact)
756 {
757 	assert((sig > 0) && (sig < NSIG));
758 
759 	p->p_sigacts.ps_sigact[sig] = sigact;
760 }
761 
762 void
proc_set_trampact(proc_t p,int sig,user_addr_t trampact)763 proc_set_trampact(proc_t p, int sig, user_addr_t trampact)
764 {
765 	assert((sig > 0) && (sig < NSIG));
766 
767 	p->p_sigacts.ps_trampact[sig] = trampact;
768 }
769 
770 void
proc_set_sigact_trampact(proc_t p,int sig,user_addr_t sigact,user_addr_t trampact)771 proc_set_sigact_trampact(proc_t p, int sig, user_addr_t sigact, user_addr_t trampact)
772 {
773 	assert((sig > 0) && (sig < NSIG));
774 
775 	p->p_sigacts.ps_sigact[sig] = sigact;
776 	p->p_sigacts.ps_trampact[sig] = trampact;
777 }
778 
779 void
proc_reset_sigact(proc_t p,sigset_t sigs)780 proc_reset_sigact(proc_t p, sigset_t sigs)
781 {
782 	user_addr_t *sigacts = p->p_sigacts.ps_sigact;
783 	int nc;
784 
785 	while (sigs) {
786 		nc = ffs((unsigned int)sigs);
787 		if (sigacts[nc] != SIG_DFL) {
788 			sigacts[nc] = SIG_DFL;
789 		}
790 		sigs &= ~sigmask(nc);
791 	}
792 }
793 
794 /*
795  * Destroy a process structure that resulted from a call to forkproc(), but
796  * which must be returned to the system because of a subsequent failure
797  * preventing it from becoming active.
798  *
799  * Parameters:	p			The incomplete process from forkproc()
800  *
801  * Returns:	(void)
802  *
803  * Note:	This function should only be used in an error handler following
804  *		a call to forkproc().
805  *
806  *		Operations occur in reverse order of those in forkproc().
807  */
808 void
forkproc_free(proc_t p)809 forkproc_free(proc_t p)
810 {
811 	struct pgrp *pg;
812 
813 #if CONFIG_PERSONAS
814 	persona_proc_drop(p);
815 #endif /* CONFIG_PERSONAS */
816 
817 #if PSYNCH
818 	pth_proc_hashdelete(p);
819 #endif /* PSYNCH */
820 
821 	/* We held signal and a transition locks; drop them */
822 	proc_signalend(p, 0);
823 	proc_transend(p, 0);
824 
825 	/*
826 	 * If we have our own copy of the resource limits structure, we
827 	 * need to free it.  If it's a shared copy, we need to drop our
828 	 * reference on it.
829 	 */
830 	proc_limitdrop(p);
831 
832 #if SYSV_SHM
833 	/* Need to drop references to the shared memory segment(s), if any */
834 	if (p->vm_shm) {
835 		/*
836 		 * Use shmexec(): we have no address space, so no mappings
837 		 *
838 		 * XXX Yes, the routine is badly named.
839 		 */
840 		shmexec(p);
841 	}
842 #endif
843 
844 	/* Need to undo the effects of the fdt_fork(), if any */
845 	fdt_invalidate(p);
846 	fdt_destroy(p);
847 
848 	/*
849 	 * Drop the reference on a text vnode pointer, if any
850 	 * XXX This code is broken in forkproc(); see <rdar://4256419>;
851 	 * XXX if anyone ever uses this field, we will be extremely unhappy.
852 	 */
853 	if (p->p_textvp) {
854 		vnode_rele(p->p_textvp);
855 		p->p_textvp = NULL;
856 	}
857 
858 	/* Update the audit session proc count */
859 	AUDIT_SESSION_PROCEXIT(p);
860 
861 	lck_mtx_destroy(&p->p_mlock, &proc_mlock_grp);
862 	lck_mtx_destroy(&p->p_ucred_mlock, &proc_ucred_mlock_grp);
863 #if CONFIG_AUDIT
864 	lck_mtx_destroy(&p->p_audit_mlock, &proc_ucred_mlock_grp);
865 #endif /* CONFIG_AUDIT */
866 #if CONFIG_DTRACE
867 	lck_mtx_destroy(&p->p_dtrace_sprlock, &proc_lck_grp);
868 #endif
869 	lck_spin_destroy(&p->p_slock, &proc_slock_grp);
870 
871 	proc_list_lock();
872 	/* Decrement the count of processes in the system */
873 	nprocs--;
874 
875 	/* quit the group */
876 	pg = pgrp_leave_locked(p);
877 
878 	/* Take it out of process hash */
879 	assert((os_ref_get_raw_mask(&p->p_refcount) >> P_REF_BITS) == 1);
880 	assert((os_ref_get_raw_mask(&p->p_refcount) & P_REF_NEW) == P_REF_NEW);
881 	os_atomic_xor(&p->p_refcount, P_REF_NEW | P_REF_DEAD, relaxed);
882 
883 	/* Remove from hash if not a shadow proc */
884 	if (!proc_is_shadow(p)) {
885 		phash_remove_locked(p);
886 	}
887 
888 	proc_list_unlock();
889 
890 	pgrp_rele(pg);
891 
892 	thread_call_free(p->p_rcall);
893 
894 	/* Free allocated memory */
895 	zfree(proc_stats_zone, p->p_stats);
896 	p->p_stats = NULL;
897 	if (p->p_subsystem_root_path) {
898 		zfree(ZV_NAMEI, p->p_subsystem_root_path);
899 		p->p_subsystem_root_path = NULL;
900 	}
901 
902 	proc_checkdeadrefs(p);
903 	proc_wait_release(p);
904 }
905 
906 
907 /*
908  * forkproc
909  *
910  * Description:	Create a new process structure, given a parent process
911  *		structure.
912  *
913  * Parameters:	parent_proc		The parent process
914  *
915  * Returns:	!NULL			The new process structure
916  *		NULL			Error (insufficient free memory)
917  *
918  * Note:	When successful, the newly created process structure is
919  *		partially initialized; if a caller needs to deconstruct the
920  *		returned structure, they must call forkproc_free() to do so.
921  */
922 proc_t
forkproc(proc_t parent_proc,cloneproc_flags_t clone_flags)923 forkproc(proc_t parent_proc, cloneproc_flags_t clone_flags)
924 {
925 	static uint64_t nextuniqueid = 0;
926 	static pid_t lastpid = 0;
927 
928 	proc_t child_proc;      /* Our new process */
929 	int error = 0;
930 	struct pgrp *pg;
931 	uthread_t parent_uthread = current_uthread();
932 	rlim_t rlimit_cpu_cur;
933 	pid_t pid;
934 	struct proc_ro_data proc_ro_data = {};
935 	bool in_exec = !!(clone_flags & CLONEPROC_FLAGS_FOR_EXEC);
936 
937 	child_proc = zalloc_flags(proc_task_zone, Z_WAITOK | Z_ZERO);
938 
939 	child_proc->p_stats = zalloc_flags(proc_stats_zone, Z_WAITOK | Z_ZERO);
940 	child_proc->p_sigacts = parent_proc->p_sigacts;
941 	os_ref_init_mask(&child_proc->p_refcount, P_REF_BITS, &p_refgrp, P_REF_NEW);
942 	os_ref_init_raw(&child_proc->p_waitref, &p_refgrp);
943 	proc_ref_hold_proc_task_struct(child_proc);
944 
945 	/* allocate a callout for use by interval timers */
946 	child_proc->p_rcall = thread_call_allocate((thread_call_func_t)realitexpire, child_proc);
947 
948 
949 	/*
950 	 * Find an unused PID.
951 	 */
952 
953 	fdt_init(child_proc);
954 
955 	proc_list_lock();
956 
957 	if (!in_exec) {
958 		pid = lastpid;
959 		do {
960 			/*
961 			 * If the process ID prototype has wrapped around,
962 			 * restart somewhat above 0, as the low-numbered procs
963 			 * tend to include daemons that don't exit.
964 			 */
965 			if (++pid >= PID_MAX) {
966 				pid = 100;
967 			}
968 			if (pid == lastpid) {
969 				panic("Unable to allocate a new pid");
970 			}
971 
972 			/* if the pid stays in hash both for zombie and runniing state */
973 		} while (phash_find_locked(pid) != PROC_NULL ||
974 		    pghash_exists_locked(pid) ||
975 		    session_find_locked(pid) != SESSION_NULL);
976 
977 		lastpid = pid;
978 		nprocs++;
979 
980 		child_proc->p_pid = pid;
981 		proc_ro_data.p_idversion = OSIncrementAtomic(&nextpidversion);
982 		/* kernel process is handcrafted and not from fork, so start from 1 */
983 		proc_ro_data.p_uniqueid = ++nextuniqueid;
984 
985 		/* Insert in the hash, and inherit our group (and session) */
986 		phash_insert_locked(child_proc);
987 
988 		/* Check if the proc is from App Cryptex */
989 		if (parent_proc->p_ladvflag & P_RSR) {
990 			os_atomic_or(&child_proc->p_ladvflag, P_RSR, relaxed);
991 		}
992 	} else {
993 		/* For exec copy of the proc, copy the pid, pidversion and uniqueid of original proc */
994 		pid = parent_proc->p_pid;
995 		child_proc->p_pid = pid;
996 		proc_ro_data.p_idversion = parent_proc->p_proc_ro->p_idversion;
997 		proc_ro_data.p_uniqueid = parent_proc->p_proc_ro->p_uniqueid;
998 
999 		nprocs++;
1000 		os_atomic_or(&child_proc->p_refcount, P_REF_SHADOW, relaxed);
1001 	}
1002 	pg = pgrp_enter_locked(parent_proc, child_proc);
1003 	proc_list_unlock();
1004 
1005 	if (proc_ro_data.p_uniqueid == startup_serial_num_procs) {
1006 		/*
1007 		 * Turn off startup serial logging now that we have reached
1008 		 * the defined number of startup processes.
1009 		 */
1010 		startup_serial_logging_active = false;
1011 	}
1012 
1013 	/*
1014 	 * We've identified the PID we are going to use;
1015 	 * initialize the new process structure.
1016 	 */
1017 	child_proc->p_stat = SIDL;
1018 
1019 	/*
1020 	 * The zero'ing of the proc was at the allocation time due to need
1021 	 * for insertion to hash.  Copy the section that is to be copied
1022 	 * directly from the parent.
1023 	 */
1024 	child_proc->p_forkcopy = parent_proc->p_forkcopy;
1025 
1026 	proc_ro_data.syscall_filter_mask = proc_syscall_filter_mask(parent_proc);
1027 	proc_ro_data.p_platform_data = proc_get_ro(parent_proc)->p_platform_data;
1028 
1029 	/*
1030 	 * Some flags are inherited from the parent.
1031 	 * Duplicate sub-structures as needed.
1032 	 * Increase reference counts on shared objects.
1033 	 * The p_stats substruct is set in vm_fork.
1034 	 */
1035 #if CONFIG_DELAY_IDLE_SLEEP
1036 	child_proc->p_flag = (parent_proc->p_flag & (P_LP64 | P_TRANSLATED | P_DISABLE_ASLR | P_DELAYIDLESLEEP | P_SUGID | P_AFFINITY));
1037 #else /* CONFIG_DELAY_IDLE_SLEEP */
1038 	child_proc->p_flag = (parent_proc->p_flag & (P_LP64 | P_TRANSLATED | P_DISABLE_ASLR | P_SUGID | P_AFFINITY));
1039 #endif /* CONFIG_DELAY_IDLE_SLEEP */
1040 
1041 	child_proc->p_vfs_iopolicy = (parent_proc->p_vfs_iopolicy & (P_VFS_IOPOLICY_INHERITED_MASK));
1042 
1043 	child_proc->p_responsible_pid = parent_proc->p_responsible_pid;
1044 
1045 	/*
1046 	 * Note that if the current thread has an assumed identity, this
1047 	 * credential will be granted to the new process.
1048 	 */
1049 	kauth_cred_set(&proc_ro_data.p_ucred.__smr_ptr, kauth_cred_get());
1050 
1051 	lck_mtx_init(&child_proc->p_mlock, &proc_mlock_grp, &proc_lck_attr);
1052 	lck_mtx_init(&child_proc->p_ucred_mlock, &proc_ucred_mlock_grp, &proc_lck_attr);
1053 #if CONFIG_AUDIT
1054 	lck_mtx_init(&child_proc->p_audit_mlock, &proc_ucred_mlock_grp, &proc_lck_attr);
1055 #endif /* CONFIG_AUDIT */
1056 #if CONFIG_DTRACE
1057 	lck_mtx_init(&child_proc->p_dtrace_sprlock, &proc_lck_grp, &proc_lck_attr);
1058 #endif
1059 	lck_spin_init(&child_proc->p_slock, &proc_slock_grp, &proc_lck_attr);
1060 
1061 	klist_init(&child_proc->p_klist);
1062 
1063 	if (child_proc->p_textvp != NULLVP) {
1064 		/* bump references to the text vnode */
1065 		/* Need to hold iocount across the ref call */
1066 		if ((error = vnode_getwithref(child_proc->p_textvp)) == 0) {
1067 			error = vnode_ref(child_proc->p_textvp);
1068 			vnode_put(child_proc->p_textvp);
1069 		}
1070 
1071 		if (error != 0) {
1072 			child_proc->p_textvp = NULLVP;
1073 		}
1074 	}
1075 
1076 	/* Inherit the parent flags for code sign */
1077 	proc_ro_data.p_csflags = ((uint32_t)proc_getcsflags(parent_proc) & ~CS_KILLED);
1078 
1079 	child_proc->p_proc_ro = proc_ro_alloc(child_proc, &proc_ro_data, NULL, NULL);
1080 
1081 	/* update cred on proc */
1082 	proc_update_creds_onproc(child_proc, proc_ucred_unsafe(child_proc));
1083 
1084 	/* update audit session proc count */
1085 	AUDIT_SESSION_PROCNEW(child_proc);
1086 
1087 	/*
1088 	 * Copy the parents per process open file table to the child; if
1089 	 * there is a per-thread current working directory, set the childs
1090 	 * per-process current working directory to that instead of the
1091 	 * parents.
1092 	 */
1093 	if (fdt_fork(&child_proc->p_fd, parent_proc, parent_uthread->uu_cdir, in_exec) != 0) {
1094 		forkproc_free(child_proc);
1095 		child_proc = NULL;
1096 		goto bad;
1097 	}
1098 
1099 #if SYSV_SHM
1100 	if (parent_proc->vm_shm && !in_exec) {
1101 		/* XXX may fail to attach shm to child */
1102 		(void)shmfork(parent_proc, child_proc);
1103 	}
1104 #endif
1105 
1106 	/*
1107 	 * Child inherits the parent's plimit
1108 	 */
1109 	proc_limitfork(parent_proc, child_proc);
1110 
1111 	rlimit_cpu_cur = proc_limitgetcur(child_proc, RLIMIT_CPU);
1112 	if (rlimit_cpu_cur != RLIM_INFINITY) {
1113 		child_proc->p_rlim_cpu.tv_sec = (rlimit_cpu_cur > __INT_MAX__) ? __INT_MAX__ : rlimit_cpu_cur;
1114 	}
1115 
1116 	if (in_exec) {
1117 		/* Keep the original start time for exec'ed proc */
1118 		child_proc->p_stats->ps_start = parent_proc->p_stats->ps_start;
1119 		child_proc->p_start.tv_sec = parent_proc->p_start.tv_sec;
1120 		child_proc->p_start.tv_usec = parent_proc->p_start.tv_usec;
1121 	} else {
1122 		/* Intialize new process stats, including start time */
1123 		/* <rdar://6640543> non-zeroed portion contains garbage AFAICT */
1124 		microtime_with_abstime(&child_proc->p_start, &child_proc->p_stats->ps_start);
1125 	}
1126 
1127 	if (pg->pg_session->s_ttyvp != NULL && parent_proc->p_flag & P_CONTROLT) {
1128 		os_atomic_or(&child_proc->p_flag, P_CONTROLT, relaxed);
1129 	}
1130 
1131 	/*
1132 	 * block all signals to reach the process.
1133 	 * no transition race should be occuring with the child yet,
1134 	 * but indicate that the process is in (the creation) transition.
1135 	 */
1136 	proc_signalstart(child_proc, 0);
1137 	proc_transstart(child_proc, 0, 0);
1138 
1139 	child_proc->p_pcaction = 0;
1140 
1141 	TAILQ_INIT(&child_proc->p_uthlist);
1142 	TAILQ_INIT(&child_proc->p_aio_activeq);
1143 	TAILQ_INIT(&child_proc->p_aio_doneq);
1144 
1145 	/*
1146 	 * Copy work queue information
1147 	 *
1148 	 * Note: This should probably only happen in the case where we are
1149 	 *	creating a child that is a copy of the parent; since this
1150 	 *	routine is called in the non-duplication case of vfork()
1151 	 *	or posix_spawn(), then this information should likely not
1152 	 *	be duplicated.
1153 	 *
1154 	 * <rdar://6640553> Work queue pointers that no longer point to code
1155 	 */
1156 	child_proc->p_wqthread = parent_proc->p_wqthread;
1157 	child_proc->p_threadstart = parent_proc->p_threadstart;
1158 	child_proc->p_pthsize = parent_proc->p_pthsize;
1159 	if ((parent_proc->p_lflag & P_LREGISTER) != 0) {
1160 		child_proc->p_lflag |= P_LREGISTER;
1161 	}
1162 	child_proc->p_dispatchqueue_offset = parent_proc->p_dispatchqueue_offset;
1163 	child_proc->p_dispatchqueue_serialno_offset = parent_proc->p_dispatchqueue_serialno_offset;
1164 	child_proc->p_dispatchqueue_label_offset = parent_proc->p_dispatchqueue_label_offset;
1165 	child_proc->p_return_to_kernel_offset = parent_proc->p_return_to_kernel_offset;
1166 	child_proc->p_mach_thread_self_offset = parent_proc->p_mach_thread_self_offset;
1167 	child_proc->p_pth_tsd_offset = parent_proc->p_pth_tsd_offset;
1168 	child_proc->p_pthread_wq_quantum_offset = parent_proc->p_pthread_wq_quantum_offset;
1169 #if PSYNCH
1170 	pth_proc_hashinit(child_proc);
1171 #endif /* PSYNCH */
1172 
1173 #if CONFIG_PERSONAS
1174 	child_proc->p_persona = NULL;
1175 	if (parent_proc->p_persona) {
1176 		struct persona *persona = proc_persona_get(parent_proc);
1177 
1178 		if (persona) {
1179 			error = persona_proc_adopt(child_proc, persona, NULL);
1180 			if (error != 0) {
1181 				printf("forkproc: persona_proc_inherit failed (persona %d being destroyed?)\n",
1182 				    persona_get_id(persona));
1183 				forkproc_free(child_proc);
1184 				child_proc = NULL;
1185 				goto bad;
1186 			}
1187 		}
1188 	}
1189 #endif
1190 
1191 #if CONFIG_MEMORYSTATUS
1192 	/* Memorystatus init */
1193 	child_proc->p_memstat_state = 0;
1194 	child_proc->p_memstat_effectivepriority = JETSAM_PRIORITY_DEFAULT;
1195 	child_proc->p_memstat_requestedpriority = JETSAM_PRIORITY_DEFAULT;
1196 	child_proc->p_memstat_assertionpriority = 0;
1197 	child_proc->p_memstat_userdata          = 0;
1198 	child_proc->p_memstat_idle_start        = 0;
1199 	child_proc->p_memstat_idle_delta        = 0;
1200 	child_proc->p_memstat_memlimit          = 0;
1201 	child_proc->p_memstat_memlimit_active   = 0;
1202 	child_proc->p_memstat_memlimit_inactive = 0;
1203 	child_proc->p_memstat_relaunch_flags    = P_MEMSTAT_RELAUNCH_UNKNOWN;
1204 #if CONFIG_FREEZE
1205 	child_proc->p_memstat_freeze_sharedanon_pages = 0;
1206 #endif
1207 	child_proc->p_memstat_dirty = 0;
1208 	child_proc->p_memstat_idledeadline = 0;
1209 #endif /* CONFIG_MEMORYSTATUS */
1210 
1211 	if (parent_proc->p_subsystem_root_path) {
1212 		size_t parent_length = strlen(parent_proc->p_subsystem_root_path) + 1;
1213 		assert(parent_length <= MAXPATHLEN);
1214 		child_proc->p_subsystem_root_path = zalloc_flags(ZV_NAMEI,
1215 		    Z_WAITOK | Z_ZERO);
1216 		memcpy(child_proc->p_subsystem_root_path, parent_proc->p_subsystem_root_path, parent_length);
1217 	}
1218 
1219 bad:
1220 	return child_proc;
1221 }
1222 
1223 void
proc_lock(proc_t p)1224 proc_lock(proc_t p)
1225 {
1226 	LCK_MTX_ASSERT(&proc_list_mlock, LCK_MTX_ASSERT_NOTOWNED);
1227 	lck_mtx_lock(&p->p_mlock);
1228 }
1229 
1230 void
proc_unlock(proc_t p)1231 proc_unlock(proc_t p)
1232 {
1233 	lck_mtx_unlock(&p->p_mlock);
1234 }
1235 
1236 void
proc_spinlock(proc_t p)1237 proc_spinlock(proc_t p)
1238 {
1239 	lck_spin_lock_grp(&p->p_slock, &proc_slock_grp);
1240 }
1241 
1242 void
proc_spinunlock(proc_t p)1243 proc_spinunlock(proc_t p)
1244 {
1245 	lck_spin_unlock(&p->p_slock);
1246 }
1247 
1248 void
proc_list_lock(void)1249 proc_list_lock(void)
1250 {
1251 	lck_mtx_lock(&proc_list_mlock);
1252 }
1253 
1254 void
proc_list_unlock(void)1255 proc_list_unlock(void)
1256 {
1257 	lck_mtx_unlock(&proc_list_mlock);
1258 }
1259 
1260 void
proc_ucred_lock(proc_t p)1261 proc_ucred_lock(proc_t p)
1262 {
1263 	lck_mtx_lock(&p->p_ucred_mlock);
1264 }
1265 
1266 void
proc_ucred_unlock(proc_t p)1267 proc_ucred_unlock(proc_t p)
1268 {
1269 	lck_mtx_unlock(&p->p_ucred_mlock);
1270 }
1271 
1272 void
proc_update_creds_onproc(proc_t p,kauth_cred_t cred)1273 proc_update_creds_onproc(proc_t p, kauth_cred_t cred)
1274 {
1275 	p->p_uid = kauth_cred_getuid(cred);
1276 	p->p_gid = kauth_cred_getgid(cred);
1277 	p->p_ruid = kauth_cred_getruid(cred);
1278 	p->p_rgid = kauth_cred_getrgid(cred);
1279 	p->p_svuid = kauth_cred_getsvuid(cred);
1280 	p->p_svgid = kauth_cred_getsvgid(cred);
1281 }
1282 
1283 
1284 bool
uthread_is64bit(struct uthread * uth)1285 uthread_is64bit(struct uthread *uth)
1286 {
1287 	return uth->uu_flag & UT_LP64;
1288 }
1289 
1290 void
uthread_init(task_t task,uthread_t uth,thread_ro_t tro_tpl,int workq_thread)1291 uthread_init(task_t task, uthread_t uth, thread_ro_t tro_tpl, int workq_thread)
1292 {
1293 	uthread_t uth_parent = current_uthread();
1294 
1295 	lck_spin_init(&uth->uu_rethrottle_lock, &rethrottle_lock_grp,
1296 	    LCK_ATTR_NULL);
1297 
1298 	/*
1299 	 * Lazily set the thread on the kernel VFS context
1300 	 * to the first thread made which will be vm_pageout_scan_thread.
1301 	 */
1302 	if (__improbable(vfs_context0.vc_thread == NULL)) {
1303 		extern thread_t vm_pageout_scan_thread;
1304 
1305 		assert(task == kernel_task);
1306 		assert(get_machthread(uth) == vm_pageout_scan_thread);
1307 		vfs_context0.vc_thread = get_machthread(uth);
1308 	}
1309 
1310 	if (task_get_64bit_addr(task)) {
1311 		uth->uu_flag |= UT_LP64;
1312 	}
1313 
1314 	/*
1315 	 * Thread inherits credential from the creating thread, if both
1316 	 * are in the same task.
1317 	 *
1318 	 * If the creating thread has no credential or is from another
1319 	 * task we can leave the new thread credential NULL.  If it needs
1320 	 * one later, it will be lazily assigned from the task's process.
1321 	 */
1322 	if (task == kernel_task) {
1323 		kauth_cred_set(&tro_tpl->tro_cred, vfs_context0.vc_ucred);
1324 		tro_tpl->tro_proc = kernproc;
1325 		tro_tpl->tro_proc_ro = kernproc->p_proc_ro;
1326 	} else if (!task_is_a_corpse(task)) {
1327 		thread_ro_t curtro = current_thread_ro();
1328 		proc_t p = get_bsdtask_info(task);
1329 
1330 		if (task == curtro->tro_task &&
1331 		    ((curtro->tro_flags & TRO_SETUID) == 0 || !workq_thread)) {
1332 			kauth_cred_set(&tro_tpl->tro_cred, curtro->tro_cred);
1333 			tro_tpl->tro_flags = (curtro->tro_flags & TRO_SETUID);
1334 			tro_tpl->tro_proc_ro = curtro->tro_proc_ro;
1335 		} else {
1336 			kauth_cred_t cred = kauth_cred_proc_ref(p);
1337 			kauth_cred_set_and_unref(&tro_tpl->tro_cred, &cred);
1338 			tro_tpl->tro_proc_ro = task_get_ro(task);
1339 		}
1340 		tro_tpl->tro_proc = p;
1341 
1342 		proc_lock(p);
1343 		if (workq_thread) {
1344 			/* workq_thread threads will not inherit masks */
1345 			uth->uu_sigmask = ~workq_threadmask;
1346 		} else if (uth_parent->uu_flag & UT_SAS_OLDMASK) {
1347 			uth->uu_sigmask = uth_parent->uu_oldmask;
1348 		} else {
1349 			uth->uu_sigmask = uth_parent->uu_sigmask;
1350 		}
1351 
1352 		TAILQ_INSERT_TAIL(&p->p_uthlist, uth, uu_list);
1353 		proc_unlock(p);
1354 
1355 #if CONFIG_DTRACE
1356 		if (p->p_dtrace_ptss_pages != NULL) {
1357 			uth->t_dtrace_scratch = dtrace_ptss_claim_entry(p);
1358 		}
1359 #endif
1360 	} else {
1361 		tro_tpl->tro_proc_ro = task_get_ro(task);
1362 	}
1363 
1364 	uth->uu_pending_sigreturn = 0;
1365 	uthread_init_proc_refcount(uth);
1366 }
1367 
1368 mach_port_name_t
uthread_joiner_port(struct uthread * uth)1369 uthread_joiner_port(struct uthread *uth)
1370 {
1371 	return uth->uu_save.uus_bsdthread_terminate.kport;
1372 }
1373 
1374 user_addr_t
uthread_joiner_address(uthread_t uth)1375 uthread_joiner_address(uthread_t uth)
1376 {
1377 	return uth->uu_save.uus_bsdthread_terminate.ulock_addr;
1378 }
1379 
1380 void
uthread_joiner_wake(task_t task,uthread_t uth)1381 uthread_joiner_wake(task_t task, uthread_t uth)
1382 {
1383 	struct _bsdthread_terminate bts = uth->uu_save.uus_bsdthread_terminate;
1384 
1385 	assert(bts.ulock_addr);
1386 	bzero(&uth->uu_save.uus_bsdthread_terminate, sizeof(bts));
1387 
1388 	int flags = UL_UNFAIR_LOCK | ULF_WAKE_ALL | ULF_WAKE_ALLOW_NON_OWNER;
1389 	(void)ulock_wake(task, flags, bts.ulock_addr, 0);
1390 	mach_port_deallocate(get_task_ipcspace(task), bts.kport);
1391 }
1392 
1393 /*
1394  * This routine frees the thread name field of the uthread_t structure. Split out of
1395  * uthread_cleanup() so thread name does not get deallocated while generating a corpse fork.
1396  */
1397 void
uthread_cleanup_name(uthread_t uth)1398 uthread_cleanup_name(uthread_t uth)
1399 {
1400 	/*
1401 	 * <rdar://17834538>
1402 	 * Set pth_name to NULL before calling free().
1403 	 * Previously there was a race condition in the
1404 	 * case this code was executing during a stackshot
1405 	 * where the stackshot could try and copy pth_name
1406 	 * after it had been freed and before if was marked
1407 	 * as null.
1408 	 */
1409 	if (uth->pth_name != NULL) {
1410 		void *pth_name = uth->pth_name;
1411 		uth->pth_name = NULL;
1412 		kfree_data(pth_name, MAXTHREADNAMESIZE);
1413 	}
1414 	return;
1415 }
1416 
1417 /*
1418  * This routine frees all the BSD context in uthread except the credential.
1419  * It does not free the uthread structure as well
1420  */
1421 void
uthread_cleanup(uthread_t uth,thread_ro_t tro)1422 uthread_cleanup(uthread_t uth, thread_ro_t tro)
1423 {
1424 	task_t task = tro->tro_task;
1425 	proc_t p    = tro->tro_proc;
1426 
1427 	uthread_assert_zero_proc_refcount(uth);
1428 
1429 	if (uth->uu_lowpri_window || uth->uu_throttle_info) {
1430 		/*
1431 		 * task is marked as a low priority I/O type
1432 		 * and we've somehow managed to not dismiss the throttle
1433 		 * through the normal exit paths back to user space...
1434 		 * no need to throttle this thread since its going away
1435 		 * but we do need to update our bookeeping w/r to throttled threads
1436 		 *
1437 		 * Calling this routine will clean up any throttle info reference
1438 		 * still inuse by the thread.
1439 		 */
1440 		throttle_lowpri_io(0);
1441 	}
1442 
1443 #if CONFIG_AUDIT
1444 	/*
1445 	 * Per-thread audit state should never last beyond system
1446 	 * call return.  Since we don't audit the thread creation/
1447 	 * removal, the thread state pointer should never be
1448 	 * non-NULL when we get here.
1449 	 */
1450 	assert(uth->uu_ar == NULL);
1451 #endif
1452 
1453 	if (uth->uu_select.nbytes) {
1454 		select_cleanup_uthread(&uth->uu_select);
1455 	}
1456 
1457 	if (uth->uu_cdir) {
1458 		vnode_rele(uth->uu_cdir);
1459 		uth->uu_cdir = NULLVP;
1460 	}
1461 
1462 	if (uth->uu_selset) {
1463 		select_set_free(uth->uu_selset);
1464 		uth->uu_selset = NULL;
1465 	}
1466 
1467 	os_reason_free(uth->uu_exit_reason);
1468 
1469 	if ((task != kernel_task) && p) {
1470 		/*
1471 		 * Remove the thread from the process list and
1472 		 * transfer [appropriate] pending signals to the process.
1473 		 * Do not remove the uthread from proc uthlist for exec
1474 		 * copy task, since they does not have a ref on proc and
1475 		 * would not have been added to the list.
1476 		 */
1477 		if (uth->uu_kqr_bound) {
1478 			kqueue_threadreq_unbind(p, uth->uu_kqr_bound);
1479 		}
1480 
1481 		if (get_bsdtask_info(task) == p) {
1482 			proc_lock(p);
1483 			TAILQ_REMOVE(&p->p_uthlist, uth, uu_list);
1484 			p->p_siglist |= (uth->uu_siglist & execmask & (~p->p_sigignore | sigcantmask));
1485 			proc_unlock(p);
1486 		}
1487 
1488 #if CONFIG_DTRACE
1489 		struct dtrace_ptss_page_entry *tmpptr = uth->t_dtrace_scratch;
1490 		uth->t_dtrace_scratch = NULL;
1491 		if (tmpptr != NULL) {
1492 			dtrace_ptss_release_entry(p, tmpptr);
1493 		}
1494 #endif
1495 	} else {
1496 		assert(!uth->uu_kqr_bound);
1497 	}
1498 }
1499 
1500 /* This routine releases the credential stored in uthread */
1501 void
uthread_cred_ref(struct ucred * ucred)1502 uthread_cred_ref(struct ucred *ucred)
1503 {
1504 	kauth_cred_ref(ucred);
1505 }
1506 
1507 void
uthread_cred_free(struct ucred * ucred)1508 uthread_cred_free(struct ucred *ucred)
1509 {
1510 	kauth_cred_set(&ucred, NOCRED);
1511 }
1512 
1513 /* This routine frees the uthread structure held in thread structure */
1514 void
uthread_destroy(uthread_t uth)1515 uthread_destroy(uthread_t uth)
1516 {
1517 	uthread_destroy_proc_refcount(uth);
1518 
1519 	if (uth->t_tombstone) {
1520 		kfree_type(struct doc_tombstone, uth->t_tombstone);
1521 		uth->t_tombstone = NULL;
1522 	}
1523 
1524 #if CONFIG_DEBUG_SYSCALL_REJECTION
1525 	size_t const bitstr_len = BITMAP_SIZE(mach_trap_count + nsysent);
1526 
1527 	if (uth->syscall_rejection_mask) {
1528 		kfree_data(uth->syscall_rejection_mask, bitstr_len);
1529 		uth->syscall_rejection_mask = NULL;
1530 	}
1531 
1532 	if (uth->syscall_rejection_once_mask) {
1533 		kfree_data(uth->syscall_rejection_once_mask, bitstr_len);
1534 		uth->syscall_rejection_once_mask = NULL;
1535 	}
1536 #endif /* CONFIG_DEBUG_SYSCALL_REJECTION */
1537 
1538 	lck_spin_destroy(&uth->uu_rethrottle_lock, &rethrottle_lock_grp);
1539 
1540 	uthread_cleanup_name(uth);
1541 }
1542 
1543 user_addr_t
thread_get_sigreturn_token(thread_t thread)1544 thread_get_sigreturn_token(thread_t thread)
1545 {
1546 	uthread_t ut = (struct uthread *) get_bsdthread_info(thread);
1547 	return ut->uu_sigreturn_token;
1548 }
1549 
1550 uint32_t
thread_get_sigreturn_diversifier(thread_t thread)1551 thread_get_sigreturn_diversifier(thread_t thread)
1552 {
1553 	uthread_t ut = (struct uthread *) get_bsdthread_info(thread);
1554 	return ut->uu_sigreturn_diversifier;
1555 }
1556