xref: /freebsd/lib/libkvm/kvm_proc.c (revision 840558b9718a99de410d8dab183a56d90eb6fa9b)
158f0484fSRodney W. Grimes /*-
258f0484fSRodney W. Grimes  * Copyright (c) 1989, 1992, 1993
358f0484fSRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
458f0484fSRodney W. Grimes  *
558f0484fSRodney W. Grimes  * This code is derived from software developed by the Computer Systems
658f0484fSRodney W. Grimes  * Engineering group at Lawrence Berkeley Laboratory under DARPA contract
758f0484fSRodney W. Grimes  * BG 91-66 and contributed to Berkeley.
858f0484fSRodney W. Grimes  *
958f0484fSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
1058f0484fSRodney W. Grimes  * modification, are permitted provided that the following conditions
1158f0484fSRodney W. Grimes  * are met:
1258f0484fSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
1358f0484fSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
1458f0484fSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
1558f0484fSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
1658f0484fSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
1758f0484fSRodney W. Grimes  * 3. All advertising materials mentioning features or use of this software
1858f0484fSRodney W. Grimes  *    must display the following acknowledgement:
1958f0484fSRodney W. Grimes  *	This product includes software developed by the University of
2058f0484fSRodney W. Grimes  *	California, Berkeley and its contributors.
2158f0484fSRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
2258f0484fSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
2358f0484fSRodney W. Grimes  *    without specific prior written permission.
2458f0484fSRodney W. Grimes  *
2558f0484fSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
2658f0484fSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2758f0484fSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2858f0484fSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
2958f0484fSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
3058f0484fSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
3158f0484fSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
3258f0484fSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
3358f0484fSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
3458f0484fSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3558f0484fSRodney W. Grimes  * SUCH DAMAGE.
3658f0484fSRodney W. Grimes  */
3758f0484fSRodney W. Grimes 
38f86e3350SBruce Evans #if 0
3958f0484fSRodney W. Grimes #if defined(LIBC_SCCS) && !defined(lint)
4058f0484fSRodney W. Grimes static char sccsid[] = "@(#)kvm_proc.c	8.3 (Berkeley) 9/23/93";
4158f0484fSRodney W. Grimes #endif /* LIBC_SCCS and not lint */
42f86e3350SBruce Evans #endif
43f86e3350SBruce Evans 
44f86e3350SBruce Evans #include <sys/cdefs.h>
45f86e3350SBruce Evans __FBSDID("$FreeBSD$");
4658f0484fSRodney W. Grimes 
4758f0484fSRodney W. Grimes /*
4858f0484fSRodney W. Grimes  * Proc traversal interface for kvm.  ps and w are (probably) the exclusive
4958f0484fSRodney W. Grimes  * users of this code, so we've factored it out into a separate module.
5058f0484fSRodney W. Grimes  * Thus, we keep this grunge out of the other kvm applications (i.e.,
5158f0484fSRodney W. Grimes  * most other applications are interested only in open/close/read/nlist).
5258f0484fSRodney W. Grimes  */
5358f0484fSRodney W. Grimes 
5458f0484fSRodney W. Grimes #include <sys/param.h>
55b20ea179SAlfred Perlstein #define _WANT_UCRED	/* make ucred.h give us 'struct ucred' */
56aa22cbfeSAlfred Perlstein #include <sys/ucred.h>
5758f0484fSRodney W. Grimes #include <sys/user.h>
5858f0484fSRodney W. Grimes #include <sys/proc.h>
5958f0484fSRodney W. Grimes #include <sys/exec.h>
6058f0484fSRodney W. Grimes #include <sys/stat.h>
6158f0484fSRodney W. Grimes #include <sys/ioctl.h>
6258f0484fSRodney W. Grimes #include <sys/tty.h>
63338c7541SDavid Greenman #include <sys/file.h>
6451295a4dSJordan K. Hubbard #include <stdio.h>
6551295a4dSJordan K. Hubbard #include <stdlib.h>
6658f0484fSRodney W. Grimes #include <unistd.h>
6758f0484fSRodney W. Grimes #include <nlist.h>
6858f0484fSRodney W. Grimes #include <kvm.h>
6958f0484fSRodney W. Grimes 
7058f0484fSRodney W. Grimes #include <vm/vm.h>
7158f0484fSRodney W. Grimes #include <vm/vm_param.h>
7258f0484fSRodney W. Grimes #include <vm/swap_pager.h>
7358f0484fSRodney W. Grimes 
7458f0484fSRodney W. Grimes #include <sys/sysctl.h>
7558f0484fSRodney W. Grimes 
7658f0484fSRodney W. Grimes #include <limits.h>
7777721f53SPeter Wemm #include <memory.h>
7858f0484fSRodney W. Grimes #include <paths.h>
7958f0484fSRodney W. Grimes 
8058f0484fSRodney W. Grimes #include "kvm_private.h"
8158f0484fSRodney W. Grimes 
8258f0484fSRodney W. Grimes #define KREAD(kd, addr, obj) \
8358f0484fSRodney W. Grimes 	(kvm_read(kd, addr, (char *)(obj), sizeof(*obj)) != sizeof(*obj))
8458f0484fSRodney W. Grimes 
8558f0484fSRodney W. Grimes /*
8658f0484fSRodney W. Grimes  * Read proc's from memory file into buffer bp, which has space to hold
8758f0484fSRodney W. Grimes  * at most maxcnt procs.
8858f0484fSRodney W. Grimes  */
8958f0484fSRodney W. Grimes static int
9058f0484fSRodney W. Grimes kvm_proclist(kd, what, arg, p, bp, maxcnt)
9158f0484fSRodney W. Grimes 	kvm_t *kd;
9258f0484fSRodney W. Grimes 	int what, arg;
9358f0484fSRodney W. Grimes 	struct proc *p;
9458f0484fSRodney W. Grimes 	struct kinfo_proc *bp;
9558f0484fSRodney W. Grimes 	int maxcnt;
9658f0484fSRodney W. Grimes {
97be04b6d1SDavid E. O'Brien 	int cnt = 0;
981f7d2501SKirk McKusick 	struct kinfo_proc kinfo_proc, *kp;
9958f0484fSRodney W. Grimes 	struct pgrp pgrp;
10058f0484fSRodney W. Grimes 	struct session sess;
10158f0484fSRodney W. Grimes 	struct tty tty;
1021f7d2501SKirk McKusick 	struct vmspace vmspace;
103840558b9SJohn Baldwin 	struct sigacts sigacts;
1041f7d2501SKirk McKusick 	struct pstats pstats;
1051f7d2501SKirk McKusick 	struct ucred ucred;
10671fad9fdSJulian Elischer 	struct thread mtd;
1074f0db5e0SJulian Elischer 	struct kse mke;
1084f0db5e0SJulian Elischer 	struct ksegrp mkg;
10958f0484fSRodney W. Grimes 	struct proc proc;
110a58930d8STor Egge 	struct proc pproc;
11149b33de8SBrian Feldman 	struct timeval tv;
11258f0484fSRodney W. Grimes 
1131f7d2501SKirk McKusick 	kp = &kinfo_proc;
1141f7d2501SKirk McKusick 	kp->ki_structsize = sizeof(kinfo_proc);
11542680b3aSBen Smithurst 	for (; cnt < maxcnt && p != NULL; p = LIST_NEXT(&proc, p_list)) {
1166f8c6a69SPeter Wemm 		memset(kp, 0, sizeof *kp);
11758f0484fSRodney W. Grimes 		if (KREAD(kd, (u_long)p, &proc)) {
11858f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program, "can't read proc at %x", p);
11958f0484fSRodney W. Grimes 			return (-1);
12058f0484fSRodney W. Grimes 		}
12158551c03SJulian Elischer 		if (proc.p_state != PRS_ZOMBIE) {
1223daf63fcSJulian Elischer 			if (KREAD(kd, (u_long)TAILQ_FIRST(&proc.p_threads),
12371fad9fdSJulian Elischer 			    &mtd)) {
12458551c03SJulian Elischer 				_kvm_err(kd, kd->program,
12558551c03SJulian Elischer 				    "can't read thread at %x",
1263daf63fcSJulian Elischer 				    TAILQ_FIRST(&proc.p_threads));
1273daf63fcSJulian Elischer 				return (-1);
1283daf63fcSJulian Elischer 			}
129ac2e4153SJulian Elischer 			if (proc.p_flag & P_THREADED == 0) {
1304f0db5e0SJulian Elischer 				if (KREAD(kd,
1314f0db5e0SJulian Elischer 				    (u_long)TAILQ_FIRST(&proc.p_ksegrps),
1324f0db5e0SJulian Elischer 				    &mkg)) {
1334f0db5e0SJulian Elischer 					_kvm_err(kd, kd->program,
1344f0db5e0SJulian Elischer 					    "can't read ksegrp at %x",
1354f0db5e0SJulian Elischer 					    TAILQ_FIRST(&proc.p_ksegrps));
1364f0db5e0SJulian Elischer 					return (-1);
1374f0db5e0SJulian Elischer 				}
1384f0db5e0SJulian Elischer 				if (KREAD(kd,
1394f0db5e0SJulian Elischer 				    (u_long)TAILQ_FIRST(&mkg.kg_kseq), &mke)) {
1404f0db5e0SJulian Elischer 					_kvm_err(kd, kd->program,
1414f0db5e0SJulian Elischer 					    "can't read kse at %x",
1424f0db5e0SJulian Elischer 					    TAILQ_FIRST(&mkg.kg_kseq));
1434f0db5e0SJulian Elischer 					return (-1);
1444f0db5e0SJulian Elischer 				}
1454f0db5e0SJulian Elischer 			}
14658551c03SJulian Elischer 		}
147b1fc0ec1SRobert Watson 		if (KREAD(kd, (u_long)proc.p_ucred, &ucred) == 0) {
148b1fc0ec1SRobert Watson 			kp->ki_ruid = ucred.cr_ruid;
149b1fc0ec1SRobert Watson 			kp->ki_svuid = ucred.cr_svuid;
150b1fc0ec1SRobert Watson 			kp->ki_rgid = ucred.cr_rgid;
151b1fc0ec1SRobert Watson 			kp->ki_svgid = ucred.cr_svgid;
1521f7d2501SKirk McKusick 			kp->ki_ngroups = ucred.cr_ngroups;
1531f7d2501SKirk McKusick 			bcopy(ucred.cr_groups, kp->ki_groups,
1541f7d2501SKirk McKusick 			    NGROUPS * sizeof(gid_t));
1551f7d2501SKirk McKusick 			kp->ki_uid = ucred.cr_uid;
1561f7d2501SKirk McKusick 		}
15758f0484fSRodney W. Grimes 
15858f0484fSRodney W. Grimes 		switch(what) {
15958f0484fSRodney W. Grimes 
16058f0484fSRodney W. Grimes 		case KERN_PROC_PID:
16158f0484fSRodney W. Grimes 			if (proc.p_pid != (pid_t)arg)
16258f0484fSRodney W. Grimes 				continue;
16358f0484fSRodney W. Grimes 			break;
16458f0484fSRodney W. Grimes 
16558f0484fSRodney W. Grimes 		case KERN_PROC_UID:
1661f7d2501SKirk McKusick 			if (kp->ki_uid != (uid_t)arg)
16758f0484fSRodney W. Grimes 				continue;
16858f0484fSRodney W. Grimes 			break;
16958f0484fSRodney W. Grimes 
17058f0484fSRodney W. Grimes 		case KERN_PROC_RUID:
1711f7d2501SKirk McKusick 			if (kp->ki_ruid != (uid_t)arg)
17258f0484fSRodney W. Grimes 				continue;
17358f0484fSRodney W. Grimes 			break;
17458f0484fSRodney W. Grimes 		}
17558f0484fSRodney W. Grimes 		/*
17658f0484fSRodney W. Grimes 		 * We're going to add another proc to the set.  If this
17758f0484fSRodney W. Grimes 		 * will overflow the buffer, assume the reason is because
17858f0484fSRodney W. Grimes 		 * nprocs (or the proc list) is corrupt and declare an error.
17958f0484fSRodney W. Grimes 		 */
18058f0484fSRodney W. Grimes 		if (cnt >= maxcnt) {
18158f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program, "nprocs corrupt");
18258f0484fSRodney W. Grimes 			return (-1);
18358f0484fSRodney W. Grimes 		}
18458f0484fSRodney W. Grimes 		/*
1851f7d2501SKirk McKusick 		 * gather kinfo_proc
18658f0484fSRodney W. Grimes 		 */
1871f7d2501SKirk McKusick 		kp->ki_paddr = p;
188b40ce416SJulian Elischer 		kp->ki_addr = proc.p_uarea;
189b40ce416SJulian Elischer 		/* kp->ki_kstack = proc.p_thread.td_kstack; XXXKSE */
1901f7d2501SKirk McKusick 		kp->ki_args = proc.p_args;
1918b7a975eSJohn Baldwin 		kp->ki_tracep = proc.p_tracevp;
1921f7d2501SKirk McKusick 		kp->ki_textvp = proc.p_textvp;
1931f7d2501SKirk McKusick 		kp->ki_fd = proc.p_fd;
1941f7d2501SKirk McKusick 		kp->ki_vmspace = proc.p_vmspace;
195840558b9SJohn Baldwin 		if (proc.p_sigacts != NULL) {
196840558b9SJohn Baldwin 			if (KREAD(kd, (u_long)proc.p_sigacts, &sigacts)) {
1971f7d2501SKirk McKusick 				_kvm_err(kd, kd->program,
198840558b9SJohn Baldwin 				    "can't read sigacts at %x", proc.p_sigacts);
1991f7d2501SKirk McKusick 				return (-1);
2001f7d2501SKirk McKusick 			}
201840558b9SJohn Baldwin 			kp->ki_sigignore = sigacts.ps_sigignore;
202840558b9SJohn Baldwin 			kp->ki_sigcatch = sigacts.ps_sigcatch;
2031f7d2501SKirk McKusick 		}
2046d3e7b9bSJohn Baldwin 		if ((proc.p_sflag & PS_INMEM) && proc.p_stats != NULL) {
2051f7d2501SKirk McKusick 			if (KREAD(kd, (u_long)proc.p_stats, &pstats)) {
2061f7d2501SKirk McKusick 				_kvm_err(kd, kd->program,
2071f7d2501SKirk McKusick 				    "can't read stats at %x", proc.p_stats);
2081f7d2501SKirk McKusick 				return (-1);
2091f7d2501SKirk McKusick 			}
2101f7d2501SKirk McKusick 			kp->ki_start = pstats.p_start;
2111f7d2501SKirk McKusick 			kp->ki_rusage = pstats.p_ru;
2121f7d2501SKirk McKusick 			kp->ki_childtime.tv_sec = pstats.p_cru.ru_utime.tv_sec +
2131f7d2501SKirk McKusick 			    pstats.p_cru.ru_stime.tv_sec;
2141f7d2501SKirk McKusick 			kp->ki_childtime.tv_usec =
2151f7d2501SKirk McKusick 			    pstats.p_cru.ru_utime.tv_usec +
2161f7d2501SKirk McKusick 			    pstats.p_cru.ru_stime.tv_usec;
2171f7d2501SKirk McKusick 		}
218a58930d8STor Egge 		if (proc.p_oppid)
2191f7d2501SKirk McKusick 			kp->ki_ppid = proc.p_oppid;
220a58930d8STor Egge 		else if (proc.p_pptr) {
221a58930d8STor Egge 			if (KREAD(kd, (u_long)proc.p_pptr, &pproc)) {
2221f7d2501SKirk McKusick 				_kvm_err(kd, kd->program,
2231f7d2501SKirk McKusick 				    "can't read pproc at %x", proc.p_pptr);
224a58930d8STor Egge 				return (-1);
225a58930d8STor Egge 			}
2261f7d2501SKirk McKusick 			kp->ki_ppid = pproc.p_pid;
227a58930d8STor Egge 		} else
2281f7d2501SKirk McKusick 			kp->ki_ppid = 0;
2296f8c6a69SPeter Wemm 		if (proc.p_pgrp == NULL)
2306f8c6a69SPeter Wemm 			goto nopgrp;
2316f8c6a69SPeter Wemm 		if (KREAD(kd, (u_long)proc.p_pgrp, &pgrp)) {
2326f8c6a69SPeter Wemm 			_kvm_err(kd, kd->program, "can't read pgrp at %x",
2336f8c6a69SPeter Wemm 				 proc.p_pgrp);
2346f8c6a69SPeter Wemm 			return (-1);
2356f8c6a69SPeter Wemm 		}
2361f7d2501SKirk McKusick 		kp->ki_pgid = pgrp.pg_id;
2371f7d2501SKirk McKusick 		kp->ki_jobc = pgrp.pg_jobc;
23858f0484fSRodney W. Grimes 		if (KREAD(kd, (u_long)pgrp.pg_session, &sess)) {
23958f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program, "can't read session at %x",
24058f0484fSRodney W. Grimes 				pgrp.pg_session);
24158f0484fSRodney W. Grimes 			return (-1);
24258f0484fSRodney W. Grimes 		}
2431f7d2501SKirk McKusick 		kp->ki_sid = sess.s_sid;
2441f7d2501SKirk McKusick 		(void)memcpy(kp->ki_login, sess.s_login,
2451f7d2501SKirk McKusick 						sizeof(kp->ki_login));
2461f7d2501SKirk McKusick 		kp->ki_kiflag = sess.s_ttyvp ? KI_CTTY : 0;
2471f7d2501SKirk McKusick 		if (sess.s_leader == p)
2481f7d2501SKirk McKusick 			kp->ki_kiflag |= KI_SLEADER;
24958f0484fSRodney W. Grimes 		if ((proc.p_flag & P_CONTROLT) && sess.s_ttyp != NULL) {
25058f0484fSRodney W. Grimes 			if (KREAD(kd, (u_long)sess.s_ttyp, &tty)) {
25158f0484fSRodney W. Grimes 				_kvm_err(kd, kd->program,
25258f0484fSRodney W. Grimes 					 "can't read tty at %x", sess.s_ttyp);
25358f0484fSRodney W. Grimes 				return (-1);
25458f0484fSRodney W. Grimes 			}
2551f7d2501SKirk McKusick 			kp->ki_tdev = tty.t_dev;
25658f0484fSRodney W. Grimes 			if (tty.t_pgrp != NULL) {
25758f0484fSRodney W. Grimes 				if (KREAD(kd, (u_long)tty.t_pgrp, &pgrp)) {
25858f0484fSRodney W. Grimes 					_kvm_err(kd, kd->program,
259d3cf4981SKris Kennaway 						 "can't read tpgrp at %x",
26058f0484fSRodney W. Grimes 						tty.t_pgrp);
26158f0484fSRodney W. Grimes 					return (-1);
26258f0484fSRodney W. Grimes 				}
2631f7d2501SKirk McKusick 				kp->ki_tpgid = pgrp.pg_id;
26458f0484fSRodney W. Grimes 			} else
2651f7d2501SKirk McKusick 				kp->ki_tpgid = -1;
2661f7d2501SKirk McKusick 			if (tty.t_session != NULL) {
2671f7d2501SKirk McKusick 				if (KREAD(kd, (u_long)tty.t_session, &sess)) {
2681f7d2501SKirk McKusick 					_kvm_err(kd, kd->program,
2691f7d2501SKirk McKusick 					    "can't read session at %x",
2701f7d2501SKirk McKusick 					    tty.t_session);
2711f7d2501SKirk McKusick 					return (-1);
2721f7d2501SKirk McKusick 				}
2731f7d2501SKirk McKusick 				kp->ki_tsid = sess.s_sid;
2741f7d2501SKirk McKusick 			}
2756f8c6a69SPeter Wemm 		} else {
2766f8c6a69SPeter Wemm nopgrp:
2771f7d2501SKirk McKusick 			kp->ki_tdev = NODEV;
2786f8c6a69SPeter Wemm 		}
27971fad9fdSJulian Elischer 		if ((proc.p_state != PRS_ZOMBIE) && mtd.td_wmesg)
28071fad9fdSJulian Elischer 			(void)kvm_read(kd, (u_long)mtd.td_wmesg,
2811f7d2501SKirk McKusick 			    kp->ki_wmesg, WMESGLEN);
28258f0484fSRodney W. Grimes 
28358f0484fSRodney W. Grimes #ifdef sparc
28458f0484fSRodney W. Grimes 		(void)kvm_read(kd, (u_long)&proc.p_vmspace->vm_rssize,
2851f7d2501SKirk McKusick 		    (char *)&kp->ki_rssize,
2861f7d2501SKirk McKusick 		    sizeof(kp->ki_rssize));
28758f0484fSRodney W. Grimes 		(void)kvm_read(kd, (u_long)&proc.p_vmspace->vm_tsize,
2881f7d2501SKirk McKusick 		    (char *)&kp->ki_tsize,
2891f7d2501SKirk McKusick 		    3 * sizeof(kp->ki_rssize));	/* XXX */
29058f0484fSRodney W. Grimes #else
29158f0484fSRodney W. Grimes 		(void)kvm_read(kd, (u_long)proc.p_vmspace,
2921f7d2501SKirk McKusick 		    (char *)&vmspace, sizeof(vmspace));
2931f7d2501SKirk McKusick 		kp->ki_size = vmspace.vm_map.size;
2941f7d2501SKirk McKusick 		kp->ki_rssize = vmspace.vm_swrss; /* XXX */
2951f7d2501SKirk McKusick 		kp->ki_swrss = vmspace.vm_swrss;
2961f7d2501SKirk McKusick 		kp->ki_tsize = vmspace.vm_tsize;
2971f7d2501SKirk McKusick 		kp->ki_dsize = vmspace.vm_dsize;
2981f7d2501SKirk McKusick 		kp->ki_ssize = vmspace.vm_ssize;
29958f0484fSRodney W. Grimes #endif
30058f0484fSRodney W. Grimes 
30158f0484fSRodney W. Grimes 		switch (what) {
30258f0484fSRodney W. Grimes 
30358f0484fSRodney W. Grimes 		case KERN_PROC_PGRP:
3041f7d2501SKirk McKusick 			if (kp->ki_pgid != (pid_t)arg)
30558f0484fSRodney W. Grimes 				continue;
30658f0484fSRodney W. Grimes 			break;
30758f0484fSRodney W. Grimes 
30858f0484fSRodney W. Grimes 		case KERN_PROC_TTY:
30958f0484fSRodney W. Grimes 			if ((proc.p_flag & P_CONTROLT) == 0 ||
3101f7d2501SKirk McKusick 			     kp->ki_tdev != (dev_t)arg)
31158f0484fSRodney W. Grimes 				continue;
31258f0484fSRodney W. Grimes 			break;
31358f0484fSRodney W. Grimes 		}
3141f7d2501SKirk McKusick 		if (proc.p_comm[0] != 0) {
3151f7d2501SKirk McKusick 			strncpy(kp->ki_comm, proc.p_comm, MAXCOMLEN);
3161f7d2501SKirk McKusick 			kp->ki_comm[MAXCOMLEN] = 0;
3171f7d2501SKirk McKusick 		}
318bff4151cSJulian Elischer 		if ((proc.p_state != PRS_ZOMBIE) &&
31971fad9fdSJulian Elischer 		    (mtd.td_blocked != 0)) {
3200d632649SJohn Baldwin 			kp->ki_kiflag |= KI_LOCKBLOCK;
3210d632649SJohn Baldwin 			if (mtd.td_lockname)
322bff4151cSJulian Elischer 				(void)kvm_read(kd,
3230d632649SJohn Baldwin 				    (u_long)mtd.td_lockname,
3240d632649SJohn Baldwin 				    kp->ki_lockname, LOCKNAMELEN);
3250d632649SJohn Baldwin 			kp->ki_lockname[LOCKNAMELEN] = 0;
3261f7d2501SKirk McKusick 		}
32749b33de8SBrian Feldman 		bintime2timeval(&proc.p_runtime, &tv);
32849b33de8SBrian Feldman 		kp->ki_runtime = (u_int64_t)tv.tv_sec * 1000000 + tv.tv_usec;
3291f7d2501SKirk McKusick 		kp->ki_pid = proc.p_pid;
330e8a58a83SJuli Mallett 		kp->ki_siglist = proc.p_siglist;
33142d3ad71SJeff Roberson 		SIGSETOR(kp->ki_siglist, mtd.td_siglist);
33231a9779eSJeff Roberson 		kp->ki_sigmask = mtd.td_sigmask;
3331f7d2501SKirk McKusick 		kp->ki_xstat = proc.p_xstat;
3341f7d2501SKirk McKusick 		kp->ki_acflag = proc.p_acflag;
3354f0db5e0SJulian Elischer 		kp->ki_lock = proc.p_lock;
336bff4151cSJulian Elischer 		if (proc.p_state != PRS_ZOMBIE) {
3371f7d2501SKirk McKusick 			kp->ki_swtime = proc.p_swtime;
3386143c383SJulian Elischer 			kp->ki_flag = proc.p_flag;
3396d3e7b9bSJohn Baldwin 			kp->ki_sflag = proc.p_sflag;
3401f7d2501SKirk McKusick 			kp->ki_traceflag = proc.p_traceflag;
3416143c383SJulian Elischer 			if (proc.p_state == PRS_NORMAL) {
34271fad9fdSJulian Elischer 				if (TD_ON_RUNQ(&mtd) ||
34371fad9fdSJulian Elischer 				    TD_CAN_RUN(&mtd) ||
34471fad9fdSJulian Elischer 				    TD_IS_RUNNING(&mtd)) {
345e602ba25SJulian Elischer 					kp->ki_stat = SRUN;
3464f0db5e0SJulian Elischer 				} else if (mtd.td_state ==
3474f0db5e0SJulian Elischer 				    TDS_INHIBITED) {
34871fad9fdSJulian Elischer 					if (P_SHOULDSTOP(&proc)) {
349e602ba25SJulian Elischer 						kp->ki_stat = SSTOP;
3504f0db5e0SJulian Elischer 					} else if (
3514f0db5e0SJulian Elischer 					    TD_IS_SLEEPING(&mtd)) {
35271fad9fdSJulian Elischer 						kp->ki_stat = SSLEEP;
3530d632649SJohn Baldwin 					} else if (TD_ON_LOCK(&mtd)) {
3540d632649SJohn Baldwin 						kp->ki_stat = SLOCK;
355e602ba25SJulian Elischer 					} else {
356e602ba25SJulian Elischer 						kp->ki_stat = SWAIT;
357e602ba25SJulian Elischer 					}
35871fad9fdSJulian Elischer 				}
359e602ba25SJulian Elischer 			} else {
360e602ba25SJulian Elischer 				kp->ki_stat = SIDL;
361e602ba25SJulian Elischer 			}
3624f0db5e0SJulian Elischer 			/* Stuff from the thread */
36371fad9fdSJulian Elischer 			kp->ki_pri.pri_level = mtd.td_priority;
36471fad9fdSJulian Elischer 			kp->ki_pri.pri_native = mtd.td_base_pri;
36571fad9fdSJulian Elischer 			kp->ki_lastcpu = mtd.td_lastcpu;
3664f0db5e0SJulian Elischer 			kp->ki_wchan = mtd.td_wchan;
367a0ddbf49SJulian Elischer 			kp->ki_oncpu = mtd.td_oncpu;
3684f0db5e0SJulian Elischer 
369ac2e4153SJulian Elischer 			if (!(proc.p_flag & P_THREADED)) {
3704f0db5e0SJulian Elischer 				/* stuff from the ksegrp */
3714f0db5e0SJulian Elischer 				kp->ki_slptime = mkg.kg_slptime;
3724f0db5e0SJulian Elischer 				kp->ki_pri.pri_class = mkg.kg_pri_class;
3734f0db5e0SJulian Elischer 				kp->ki_pri.pri_user = mkg.kg_user_pri;
3744f0db5e0SJulian Elischer 				kp->ki_nice = mkg.kg_nice;
3754f0db5e0SJulian Elischer 				kp->ki_estcpu = mkg.kg_estcpu;
3764f0db5e0SJulian Elischer 
3774f0db5e0SJulian Elischer 				/* Stuff from the kse */
3784f0db5e0SJulian Elischer 				kp->ki_pctcpu = mke.ke_pctcpu;
3794f0db5e0SJulian Elischer 				kp->ki_rqindex = mke.ke_rqindex;
3804f0db5e0SJulian Elischer 			} else {
3814f0db5e0SJulian Elischer 				kp->ki_tdflags = -1;
3824f0db5e0SJulian Elischer 				/* All the rest are 0 for now */
3834f0db5e0SJulian Elischer 			}
3846143c383SJulian Elischer 		} else {
3856143c383SJulian Elischer 			kp->ki_stat = SZOMB;
3866143c383SJulian Elischer 		}
3871f7d2501SKirk McKusick 		bcopy(&kinfo_proc, bp, sizeof(kinfo_proc));
38858f0484fSRodney W. Grimes 		++bp;
38958f0484fSRodney W. Grimes 		++cnt;
39058f0484fSRodney W. Grimes 	}
39158f0484fSRodney W. Grimes 	return (cnt);
39258f0484fSRodney W. Grimes }
39358f0484fSRodney W. Grimes 
39458f0484fSRodney W. Grimes /*
39558f0484fSRodney W. Grimes  * Build proc info array by reading in proc list from a crash dump.
39658f0484fSRodney W. Grimes  * Return number of procs read.  maxcnt is the max we will read.
39758f0484fSRodney W. Grimes  */
39858f0484fSRodney W. Grimes static int
39958f0484fSRodney W. Grimes kvm_deadprocs(kd, what, arg, a_allproc, a_zombproc, maxcnt)
40058f0484fSRodney W. Grimes 	kvm_t *kd;
40158f0484fSRodney W. Grimes 	int what, arg;
40258f0484fSRodney W. Grimes 	u_long a_allproc;
40358f0484fSRodney W. Grimes 	u_long a_zombproc;
40458f0484fSRodney W. Grimes 	int maxcnt;
40558f0484fSRodney W. Grimes {
406be04b6d1SDavid E. O'Brien 	struct kinfo_proc *bp = kd->procbase;
407be04b6d1SDavid E. O'Brien 	int acnt, zcnt;
40858f0484fSRodney W. Grimes 	struct proc *p;
40958f0484fSRodney W. Grimes 
41058f0484fSRodney W. Grimes 	if (KREAD(kd, a_allproc, &p)) {
41158f0484fSRodney W. Grimes 		_kvm_err(kd, kd->program, "cannot read allproc");
41258f0484fSRodney W. Grimes 		return (-1);
41358f0484fSRodney W. Grimes 	}
41458f0484fSRodney W. Grimes 	acnt = kvm_proclist(kd, what, arg, p, bp, maxcnt);
41558f0484fSRodney W. Grimes 	if (acnt < 0)
41658f0484fSRodney W. Grimes 		return (acnt);
41758f0484fSRodney W. Grimes 
41858f0484fSRodney W. Grimes 	if (KREAD(kd, a_zombproc, &p)) {
41958f0484fSRodney W. Grimes 		_kvm_err(kd, kd->program, "cannot read zombproc");
42058f0484fSRodney W. Grimes 		return (-1);
42158f0484fSRodney W. Grimes 	}
42258f0484fSRodney W. Grimes 	zcnt = kvm_proclist(kd, what, arg, p, bp + acnt, maxcnt - acnt);
42358f0484fSRodney W. Grimes 	if (zcnt < 0)
42458f0484fSRodney W. Grimes 		zcnt = 0;
42558f0484fSRodney W. Grimes 
42658f0484fSRodney W. Grimes 	return (acnt + zcnt);
42758f0484fSRodney W. Grimes }
42858f0484fSRodney W. Grimes 
42958f0484fSRodney W. Grimes struct kinfo_proc *
43058f0484fSRodney W. Grimes kvm_getprocs(kd, op, arg, cnt)
43158f0484fSRodney W. Grimes 	kvm_t *kd;
43258f0484fSRodney W. Grimes 	int op, arg;
43358f0484fSRodney W. Grimes 	int *cnt;
43458f0484fSRodney W. Grimes {
435c2ac238cSDoug Rabson 	int mib[4], st, nprocs;
436c2ac238cSDoug Rabson 	size_t size;
43758f0484fSRodney W. Grimes 
43858f0484fSRodney W. Grimes 	if (kd->procbase != 0) {
43958f0484fSRodney W. Grimes 		free((void *)kd->procbase);
44058f0484fSRodney W. Grimes 		/*
44158f0484fSRodney W. Grimes 		 * Clear this pointer in case this call fails.  Otherwise,
44258f0484fSRodney W. Grimes 		 * kvm_close() will free it again.
44358f0484fSRodney W. Grimes 		 */
44458f0484fSRodney W. Grimes 		kd->procbase = 0;
44558f0484fSRodney W. Grimes 	}
44658f0484fSRodney W. Grimes 	if (ISALIVE(kd)) {
44758f0484fSRodney W. Grimes 		size = 0;
44858f0484fSRodney W. Grimes 		mib[0] = CTL_KERN;
44958f0484fSRodney W. Grimes 		mib[1] = KERN_PROC;
45058f0484fSRodney W. Grimes 		mib[2] = op;
45158f0484fSRodney W. Grimes 		mib[3] = arg;
45244ffb5f5SPoul-Henning Kamp 		st = sysctl(mib, op == KERN_PROC_ALL ? 3 : 4, NULL, &size, NULL, 0);
45358f0484fSRodney W. Grimes 		if (st == -1) {
45458f0484fSRodney W. Grimes 			_kvm_syserr(kd, kd->program, "kvm_getprocs");
45558f0484fSRodney W. Grimes 			return (0);
45658f0484fSRodney W. Grimes 		}
45721687047SDima Dorfman 		/*
45821687047SDima Dorfman 		 * We can't continue with a size of 0 because we pass
45921687047SDima Dorfman 		 * it to realloc() (via _kvm_realloc()), and passing 0
46021687047SDima Dorfman 		 * to realloc() results in undefined behavior.
46121687047SDima Dorfman 		 */
46221687047SDima Dorfman 		if (size == 0) {
46321687047SDima Dorfman 			/*
46421687047SDima Dorfman 			 * XXX: We should probably return an invalid,
46521687047SDima Dorfman 			 * but non-NULL, pointer here so any client
46621687047SDima Dorfman 			 * program trying to dereference it will
46721687047SDima Dorfman 			 * crash.  However, _kvm_freeprocs() calls
46821687047SDima Dorfman 			 * free() on kd->procbase if it isn't NULL,
46921687047SDima Dorfman 			 * and free()'ing a junk pointer isn't good.
47021687047SDima Dorfman 			 * Then again, _kvm_freeprocs() isn't used
47121687047SDima Dorfman 			 * anywhere . . .
47221687047SDima Dorfman 			 */
47321687047SDima Dorfman 			kd->procbase = _kvm_malloc(kd, 1);
47421687047SDima Dorfman 			goto liveout;
47521687047SDima Dorfman 		}
476b2d3d0f0SDag-Erling Smørgrav 		do {
477b2d3d0f0SDag-Erling Smørgrav 			size += size / 10;
478b2d3d0f0SDag-Erling Smørgrav 			kd->procbase = (struct kinfo_proc *)
479b2d3d0f0SDag-Erling Smørgrav 			    _kvm_realloc(kd, kd->procbase, size);
48058f0484fSRodney W. Grimes 			if (kd->procbase == 0)
48158f0484fSRodney W. Grimes 				return (0);
482b2d3d0f0SDag-Erling Smørgrav 			st = sysctl(mib, op == KERN_PROC_ALL ? 3 : 4,
483b2d3d0f0SDag-Erling Smørgrav 			    kd->procbase, &size, NULL, 0);
484b2d3d0f0SDag-Erling Smørgrav 		} while (st == -1 && errno == ENOMEM);
48558f0484fSRodney W. Grimes 		if (st == -1) {
48658f0484fSRodney W. Grimes 			_kvm_syserr(kd, kd->program, "kvm_getprocs");
48758f0484fSRodney W. Grimes 			return (0);
48858f0484fSRodney W. Grimes 		}
48921687047SDima Dorfman 		/*
49021687047SDima Dorfman 		 * We have to check the size again because sysctl()
49121687047SDima Dorfman 		 * may "round up" oldlenp if oldp is NULL; hence it
49221687047SDima Dorfman 		 * might've told us that there was data to get when
49321687047SDima Dorfman 		 * there really isn't any.
49421687047SDima Dorfman 		 */
4950627f53bSDavid Malone 		if (size > 0 &&
4960627f53bSDavid Malone 		    kd->procbase->ki_structsize != sizeof(struct kinfo_proc)) {
49758f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program,
4981f7d2501SKirk McKusick 			    "kinfo_proc size mismatch (expected %d, got %d)",
4991f7d2501SKirk McKusick 			    sizeof(struct kinfo_proc),
5001f7d2501SKirk McKusick 			    kd->procbase->ki_structsize);
50158f0484fSRodney W. Grimes 			return (0);
50258f0484fSRodney W. Grimes 		}
50321687047SDima Dorfman liveout:
504d339edcfSDavid Malone 		nprocs = size == 0 ? 0 : size / kd->procbase->ki_structsize;
50558f0484fSRodney W. Grimes 	} else {
50658f0484fSRodney W. Grimes 		struct nlist nl[4], *p;
50758f0484fSRodney W. Grimes 
50858f0484fSRodney W. Grimes 		nl[0].n_name = "_nprocs";
50958f0484fSRodney W. Grimes 		nl[1].n_name = "_allproc";
51058f0484fSRodney W. Grimes 		nl[2].n_name = "_zombproc";
51158f0484fSRodney W. Grimes 		nl[3].n_name = 0;
51258f0484fSRodney W. Grimes 
51358f0484fSRodney W. Grimes 		if (kvm_nlist(kd, nl) != 0) {
51458f0484fSRodney W. Grimes 			for (p = nl; p->n_type != 0; ++p)
51558f0484fSRodney W. Grimes 				;
51658f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program,
51758f0484fSRodney W. Grimes 				 "%s: no such symbol", p->n_name);
51858f0484fSRodney W. Grimes 			return (0);
51958f0484fSRodney W. Grimes 		}
52058f0484fSRodney W. Grimes 		if (KREAD(kd, nl[0].n_value, &nprocs)) {
52158f0484fSRodney W. Grimes 			_kvm_err(kd, kd->program, "can't read nprocs");
52258f0484fSRodney W. Grimes 			return (0);
52358f0484fSRodney W. Grimes 		}
52458f0484fSRodney W. Grimes 		size = nprocs * sizeof(struct kinfo_proc);
52558f0484fSRodney W. Grimes 		kd->procbase = (struct kinfo_proc *)_kvm_malloc(kd, size);
52658f0484fSRodney W. Grimes 		if (kd->procbase == 0)
52758f0484fSRodney W. Grimes 			return (0);
52858f0484fSRodney W. Grimes 
52958f0484fSRodney W. Grimes 		nprocs = kvm_deadprocs(kd, op, arg, nl[1].n_value,
53058f0484fSRodney W. Grimes 				      nl[2].n_value, nprocs);
53158f0484fSRodney W. Grimes #ifdef notdef
53258f0484fSRodney W. Grimes 		size = nprocs * sizeof(struct kinfo_proc);
53358f0484fSRodney W. Grimes 		(void)realloc(kd->procbase, size);
53458f0484fSRodney W. Grimes #endif
53558f0484fSRodney W. Grimes 	}
53658f0484fSRodney W. Grimes 	*cnt = nprocs;
53758f0484fSRodney W. Grimes 	return (kd->procbase);
53858f0484fSRodney W. Grimes }
53958f0484fSRodney W. Grimes 
54058f0484fSRodney W. Grimes void
54158f0484fSRodney W. Grimes _kvm_freeprocs(kd)
54258f0484fSRodney W. Grimes 	kvm_t *kd;
54358f0484fSRodney W. Grimes {
54458f0484fSRodney W. Grimes 	if (kd->procbase) {
54558f0484fSRodney W. Grimes 		free(kd->procbase);
54658f0484fSRodney W. Grimes 		kd->procbase = 0;
54758f0484fSRodney W. Grimes 	}
54858f0484fSRodney W. Grimes }
54958f0484fSRodney W. Grimes 
55058f0484fSRodney W. Grimes void *
55158f0484fSRodney W. Grimes _kvm_realloc(kd, p, n)
55258f0484fSRodney W. Grimes 	kvm_t *kd;
55358f0484fSRodney W. Grimes 	void *p;
55458f0484fSRodney W. Grimes 	size_t n;
55558f0484fSRodney W. Grimes {
55658f0484fSRodney W. Grimes 	void *np = (void *)realloc(p, n);
55758f0484fSRodney W. Grimes 
558e8420087SWarner Losh 	if (np == 0) {
559e8420087SWarner Losh 		free(p);
56058f0484fSRodney W. Grimes 		_kvm_err(kd, kd->program, "out of memory");
561e8420087SWarner Losh 	}
56258f0484fSRodney W. Grimes 	return (np);
56358f0484fSRodney W. Grimes }
56458f0484fSRodney W. Grimes 
56558f0484fSRodney W. Grimes #ifndef MAX
56658f0484fSRodney W. Grimes #define MAX(a, b) ((a) > (b) ? (a) : (b))
56758f0484fSRodney W. Grimes #endif
56858f0484fSRodney W. Grimes 
56958f0484fSRodney W. Grimes /*
5701f7d2501SKirk McKusick  * Read in an argument vector from the user address space of process kp.
57177721f53SPeter Wemm  * addr if the user-space base address of narg null-terminated contiguous
57258f0484fSRodney W. Grimes  * strings.  This is used to read in both the command arguments and
57358f0484fSRodney W. Grimes  * environment strings.  Read at most maxcnt characters of strings.
57458f0484fSRodney W. Grimes  */
57558f0484fSRodney W. Grimes static char **
5761f7d2501SKirk McKusick kvm_argv(kd, kp, addr, narg, maxcnt)
57758f0484fSRodney W. Grimes 	kvm_t *kd;
5781f7d2501SKirk McKusick 	struct kinfo_proc *kp;
579be04b6d1SDavid E. O'Brien 	u_long addr;
580be04b6d1SDavid E. O'Brien 	int narg;
581be04b6d1SDavid E. O'Brien 	int maxcnt;
58258f0484fSRodney W. Grimes {
583be04b6d1SDavid E. O'Brien 	char *np, *cp, *ep, *ap;
584be04b6d1SDavid E. O'Brien 	u_long oaddr = -1;
585be04b6d1SDavid E. O'Brien 	int len, cc;
586be04b6d1SDavid E. O'Brien 	char **argv;
58758f0484fSRodney W. Grimes 
58858f0484fSRodney W. Grimes 	/*
58958f0484fSRodney W. Grimes 	 * Check that there aren't an unreasonable number of agruments,
59058f0484fSRodney W. Grimes 	 * and that the address is in user space.
59158f0484fSRodney W. Grimes 	 */
5926b492ae4SPeter Wemm 	if (narg > 512 || addr < VM_MIN_ADDRESS || addr >= VM_MAXUSER_ADDRESS)
59358f0484fSRodney W. Grimes 		return (0);
59458f0484fSRodney W. Grimes 
5956b492ae4SPeter Wemm 	/*
5966b492ae4SPeter Wemm 	 * kd->argv : work space for fetching the strings from the target
5976b492ae4SPeter Wemm 	 *            process's space, and is converted for returning to caller
5986b492ae4SPeter Wemm 	 */
59958f0484fSRodney W. Grimes 	if (kd->argv == 0) {
60058f0484fSRodney W. Grimes 		/*
60158f0484fSRodney W. Grimes 		 * Try to avoid reallocs.
60258f0484fSRodney W. Grimes 		 */
60358f0484fSRodney W. Grimes 		kd->argc = MAX(narg + 1, 32);
60458f0484fSRodney W. Grimes 		kd->argv = (char **)_kvm_malloc(kd, kd->argc *
60558f0484fSRodney W. Grimes 						sizeof(*kd->argv));
60658f0484fSRodney W. Grimes 		if (kd->argv == 0)
60758f0484fSRodney W. Grimes 			return (0);
60858f0484fSRodney W. Grimes 	} else if (narg + 1 > kd->argc) {
60958f0484fSRodney W. Grimes 		kd->argc = MAX(2 * kd->argc, narg + 1);
61058f0484fSRodney W. Grimes 		kd->argv = (char **)_kvm_realloc(kd, kd->argv, kd->argc *
61158f0484fSRodney W. Grimes 						sizeof(*kd->argv));
61258f0484fSRodney W. Grimes 		if (kd->argv == 0)
61358f0484fSRodney W. Grimes 			return (0);
61458f0484fSRodney W. Grimes 	}
6156b492ae4SPeter Wemm 	/*
6166b492ae4SPeter Wemm 	 * kd->argspc : returned to user, this is where the kd->argv
6176b492ae4SPeter Wemm 	 *              arrays are left pointing to the collected strings.
6186b492ae4SPeter Wemm 	 */
61958f0484fSRodney W. Grimes 	if (kd->argspc == 0) {
6202572133eSPoul-Henning Kamp 		kd->argspc = (char *)_kvm_malloc(kd, PAGE_SIZE);
62158f0484fSRodney W. Grimes 		if (kd->argspc == 0)
62258f0484fSRodney W. Grimes 			return (0);
6232572133eSPoul-Henning Kamp 		kd->arglen = PAGE_SIZE;
62458f0484fSRodney W. Grimes 	}
6256b492ae4SPeter Wemm 	/*
6266b492ae4SPeter Wemm 	 * kd->argbuf : used to pull in pages from the target process.
6276b492ae4SPeter Wemm 	 *              the strings are copied out of here.
6286b492ae4SPeter Wemm 	 */
62977721f53SPeter Wemm 	if (kd->argbuf == 0) {
6302572133eSPoul-Henning Kamp 		kd->argbuf = (char *)_kvm_malloc(kd, PAGE_SIZE);
63177721f53SPeter Wemm 		if (kd->argbuf == 0)
63277721f53SPeter Wemm 			return (0);
63377721f53SPeter Wemm 	}
6346b492ae4SPeter Wemm 
6356b492ae4SPeter Wemm 	/* Pull in the target process'es argv vector */
63677721f53SPeter Wemm 	cc = sizeof(char *) * narg;
6371f7d2501SKirk McKusick 	if (kvm_uread(kd, kp, addr, (char *)kd->argv, cc) != cc)
63877721f53SPeter Wemm 		return (0);
6396b492ae4SPeter Wemm 	/*
6406b492ae4SPeter Wemm 	 * ap : saved start address of string we're working on in kd->argspc
6416b492ae4SPeter Wemm 	 * np : pointer to next place to write in kd->argspc
6426b492ae4SPeter Wemm 	 * len: length of data in kd->argspc
6436b492ae4SPeter Wemm 	 * argv: pointer to the argv vector that we are hunting around the
6446b492ae4SPeter Wemm 	 *       target process space for, and converting to addresses in
6456b492ae4SPeter Wemm 	 *       our address space (kd->argspc).
6466b492ae4SPeter Wemm 	 */
64777721f53SPeter Wemm 	ap = np = kd->argspc;
64858f0484fSRodney W. Grimes 	argv = kd->argv;
64958f0484fSRodney W. Grimes 	len = 0;
65058f0484fSRodney W. Grimes 	/*
65158f0484fSRodney W. Grimes 	 * Loop over pages, filling in the argument vector.
6526b492ae4SPeter Wemm 	 * Note that the argv strings could be pointing *anywhere* in
6536b492ae4SPeter Wemm 	 * the user address space and are no longer contiguous.
6546b492ae4SPeter Wemm 	 * Note that *argv is modified when we are going to fetch a string
6556b492ae4SPeter Wemm 	 * that crosses a page boundary.  We copy the next part of the string
6566b492ae4SPeter Wemm 	 * into to "np" and eventually convert the pointer.
65758f0484fSRodney W. Grimes 	 */
65877721f53SPeter Wemm 	while (argv < kd->argv + narg && *argv != 0) {
6596b492ae4SPeter Wemm 
6606b492ae4SPeter Wemm 		/* get the address that the current argv string is on */
6612572133eSPoul-Henning Kamp 		addr = (u_long)*argv & ~(PAGE_SIZE - 1);
6626b492ae4SPeter Wemm 
6636b492ae4SPeter Wemm 		/* is it the same page as the last one? */
66477721f53SPeter Wemm 		if (addr != oaddr) {
6651f7d2501SKirk McKusick 			if (kvm_uread(kd, kp, addr, kd->argbuf, PAGE_SIZE) !=
6662572133eSPoul-Henning Kamp 			    PAGE_SIZE)
66777721f53SPeter Wemm 				return (0);
66877721f53SPeter Wemm 			oaddr = addr;
66977721f53SPeter Wemm 		}
6706b492ae4SPeter Wemm 
6716b492ae4SPeter Wemm 		/* offset within the page... kd->argbuf */
6722572133eSPoul-Henning Kamp 		addr = (u_long)*argv & (PAGE_SIZE - 1);
6736b492ae4SPeter Wemm 
6746b492ae4SPeter Wemm 		/* cp = start of string, cc = count of chars in this chunk */
67577721f53SPeter Wemm 		cp = kd->argbuf + addr;
6762572133eSPoul-Henning Kamp 		cc = PAGE_SIZE - addr;
6776b492ae4SPeter Wemm 
6786b492ae4SPeter Wemm 		/* dont get more than asked for by user process */
67958f0484fSRodney W. Grimes 		if (maxcnt > 0 && cc > maxcnt - len)
6806b492ae4SPeter Wemm 			cc = maxcnt - len;
6816b492ae4SPeter Wemm 
6826b492ae4SPeter Wemm 		/* pointer to end of string if we found it in this page */
68377721f53SPeter Wemm 		ep = memchr(cp, '\0', cc);
68477721f53SPeter Wemm 		if (ep != 0)
68577721f53SPeter Wemm 			cc = ep - cp + 1;
6866b492ae4SPeter Wemm 		/*
6876b492ae4SPeter Wemm 		 * at this point, cc is the count of the chars that we are
6886b492ae4SPeter Wemm 		 * going to retrieve this time. we may or may not have found
6896b492ae4SPeter Wemm 		 * the end of it.  (ep points to the null if the end is known)
6906b492ae4SPeter Wemm 		 */
6916b492ae4SPeter Wemm 
6926b492ae4SPeter Wemm 		/* will we exceed the malloc/realloced buffer? */
69358f0484fSRodney W. Grimes 		if (len + cc > kd->arglen) {
694be04b6d1SDavid E. O'Brien 			int off;
695be04b6d1SDavid E. O'Brien 			char **pp;
696be04b6d1SDavid E. O'Brien 			char *op = kd->argspc;
69758f0484fSRodney W. Grimes 
69858f0484fSRodney W. Grimes 			kd->arglen *= 2;
69958f0484fSRodney W. Grimes 			kd->argspc = (char *)_kvm_realloc(kd, kd->argspc,
70058f0484fSRodney W. Grimes 							  kd->arglen);
70158f0484fSRodney W. Grimes 			if (kd->argspc == 0)
70258f0484fSRodney W. Grimes 				return (0);
70358f0484fSRodney W. Grimes 			/*
70458f0484fSRodney W. Grimes 			 * Adjust argv pointers in case realloc moved
70558f0484fSRodney W. Grimes 			 * the string space.
70658f0484fSRodney W. Grimes 			 */
70758f0484fSRodney W. Grimes 			off = kd->argspc - op;
70877721f53SPeter Wemm 			for (pp = kd->argv; pp < argv; pp++)
70958f0484fSRodney W. Grimes 				*pp += off;
71077721f53SPeter Wemm 			ap += off;
71177721f53SPeter Wemm 			np += off;
71258f0484fSRodney W. Grimes 		}
7136b492ae4SPeter Wemm 		/* np = where to put the next part of the string in kd->argspc*/
7146b492ae4SPeter Wemm 		/* np is kinda redundant.. could use "kd->argspc + len" */
71577721f53SPeter Wemm 		memcpy(np, cp, cc);
7166b492ae4SPeter Wemm 		np += cc;	/* inc counters */
71758f0484fSRodney W. Grimes 		len += cc;
7186b492ae4SPeter Wemm 
7196b492ae4SPeter Wemm 		/*
7206b492ae4SPeter Wemm 		 * if end of string found, set the *argv pointer to the
7216b492ae4SPeter Wemm 		 * saved beginning of string, and advance. argv points to
7226b492ae4SPeter Wemm 		 * somewhere in kd->argv..  This is initially relative
7236b492ae4SPeter Wemm 		 * to the target process, but when we close it off, we set
7246b492ae4SPeter Wemm 		 * it to point in our address space.
7256b492ae4SPeter Wemm 		 */
72677721f53SPeter Wemm 		if (ep != 0) {
72777721f53SPeter Wemm 			*argv++ = ap;
72877721f53SPeter Wemm 			ap = np;
7296b492ae4SPeter Wemm 		} else {
7306b492ae4SPeter Wemm 			/* update the address relative to the target process */
73177721f53SPeter Wemm 			*argv += cc;
7326b492ae4SPeter Wemm 		}
7336b492ae4SPeter Wemm 
73458f0484fSRodney W. Grimes 		if (maxcnt > 0 && len >= maxcnt) {
73558f0484fSRodney W. Grimes 			/*
73658f0484fSRodney W. Grimes 			 * We're stopping prematurely.  Terminate the
73777721f53SPeter Wemm 			 * current string.
73858f0484fSRodney W. Grimes 			 */
73977721f53SPeter Wemm 			if (ep == 0) {
74077721f53SPeter Wemm 				*np = '\0';
74177721f53SPeter Wemm 				*argv++ = ap;
74277721f53SPeter Wemm 			}
74377721f53SPeter Wemm 			break;
74477721f53SPeter Wemm 		}
74577721f53SPeter Wemm 	}
74677721f53SPeter Wemm 	/* Make sure argv is terminated. */
74777721f53SPeter Wemm 	*argv = 0;
74858f0484fSRodney W. Grimes 	return (kd->argv);
74958f0484fSRodney W. Grimes }
75058f0484fSRodney W. Grimes 
75158f0484fSRodney W. Grimes static void
75258f0484fSRodney W. Grimes ps_str_a(p, addr, n)
75358f0484fSRodney W. Grimes 	struct ps_strings *p;
75458f0484fSRodney W. Grimes 	u_long *addr;
75558f0484fSRodney W. Grimes 	int *n;
75658f0484fSRodney W. Grimes {
75758f0484fSRodney W. Grimes 	*addr = (u_long)p->ps_argvstr;
75858f0484fSRodney W. Grimes 	*n = p->ps_nargvstr;
75958f0484fSRodney W. Grimes }
76058f0484fSRodney W. Grimes 
76158f0484fSRodney W. Grimes static void
76258f0484fSRodney W. Grimes ps_str_e(p, addr, n)
76358f0484fSRodney W. Grimes 	struct ps_strings *p;
76458f0484fSRodney W. Grimes 	u_long *addr;
76558f0484fSRodney W. Grimes 	int *n;
76658f0484fSRodney W. Grimes {
76758f0484fSRodney W. Grimes 	*addr = (u_long)p->ps_envstr;
76858f0484fSRodney W. Grimes 	*n = p->ps_nenvstr;
76958f0484fSRodney W. Grimes }
77058f0484fSRodney W. Grimes 
77158f0484fSRodney W. Grimes /*
77258f0484fSRodney W. Grimes  * Determine if the proc indicated by p is still active.
77358f0484fSRodney W. Grimes  * This test is not 100% foolproof in theory, but chances of
77458f0484fSRodney W. Grimes  * being wrong are very low.
77558f0484fSRodney W. Grimes  */
77658f0484fSRodney W. Grimes static int
7771f7d2501SKirk McKusick proc_verify(curkp)
7781f7d2501SKirk McKusick 	struct kinfo_proc *curkp;
77958f0484fSRodney W. Grimes {
7801f7d2501SKirk McKusick 	struct kinfo_proc newkp;
78160160c5eSBruce Evans 	int mib[4];
782c2ac238cSDoug Rabson 	size_t len;
78358f0484fSRodney W. Grimes 
78435e6b695SPoul-Henning Kamp 	mib[0] = CTL_KERN;
78535e6b695SPoul-Henning Kamp 	mib[1] = KERN_PROC;
78635e6b695SPoul-Henning Kamp 	mib[2] = KERN_PROC_PID;
7871f7d2501SKirk McKusick 	mib[3] = curkp->ki_pid;
7881f7d2501SKirk McKusick 	len = sizeof(newkp);
7891f7d2501SKirk McKusick 	if (sysctl(mib, 4, &newkp, &len, NULL, 0) == -1)
79058f0484fSRodney W. Grimes 		return (0);
7911f7d2501SKirk McKusick 	return (curkp->ki_pid == newkp.ki_pid &&
7921f7d2501SKirk McKusick 	    (newkp.ki_stat != SZOMB || curkp->ki_stat == SZOMB));
79358f0484fSRodney W. Grimes }
79458f0484fSRodney W. Grimes 
79558f0484fSRodney W. Grimes static char **
79658f0484fSRodney W. Grimes kvm_doargv(kd, kp, nchr, info)
79758f0484fSRodney W. Grimes 	kvm_t *kd;
7981f7d2501SKirk McKusick 	struct kinfo_proc *kp;
79958f0484fSRodney W. Grimes 	int nchr;
80077721f53SPeter Wemm 	void (*info)(struct ps_strings *, u_long *, int *);
80158f0484fSRodney W. Grimes {
8021f7d2501SKirk McKusick 	char **ap;
80358f0484fSRodney W. Grimes 	u_long addr;
80458f0484fSRodney W. Grimes 	int cnt;
80560160c5eSBruce Evans 	static struct ps_strings arginfo;
80660160c5eSBruce Evans 	static u_long ps_strings;
8077350dd84SPeter Wemm 	size_t len;
8087350dd84SPeter Wemm 
80935e6b695SPoul-Henning Kamp 	if (ps_strings == NULL) {
81060160c5eSBruce Evans 		len = sizeof(ps_strings);
81160160c5eSBruce Evans 		if (sysctlbyname("kern.ps_strings", &ps_strings, &len, NULL,
81260160c5eSBruce Evans 		    0) == -1)
8137350dd84SPeter Wemm 			ps_strings = PS_STRINGS;
81435e6b695SPoul-Henning Kamp 	}
81558f0484fSRodney W. Grimes 
81658f0484fSRodney W. Grimes 	/*
81758f0484fSRodney W. Grimes 	 * Pointers are stored at the top of the user stack.
81858f0484fSRodney W. Grimes 	 */
8191f7d2501SKirk McKusick 	if (kp->ki_stat == SZOMB ||
8201f7d2501SKirk McKusick 	    kvm_uread(kd, kp, ps_strings, (char *)&arginfo,
82177721f53SPeter Wemm 		      sizeof(arginfo)) != sizeof(arginfo))
82258f0484fSRodney W. Grimes 		return (0);
82358f0484fSRodney W. Grimes 
82458f0484fSRodney W. Grimes 	(*info)(&arginfo, &addr, &cnt);
82577721f53SPeter Wemm 	if (cnt == 0)
82677721f53SPeter Wemm 		return (0);
8271f7d2501SKirk McKusick 	ap = kvm_argv(kd, kp, addr, cnt, nchr);
82858f0484fSRodney W. Grimes 	/*
82958f0484fSRodney W. Grimes 	 * For live kernels, make sure this process didn't go away.
83058f0484fSRodney W. Grimes 	 */
8311f7d2501SKirk McKusick 	if (ap != 0 && ISALIVE(kd) && !proc_verify(kp))
83258f0484fSRodney W. Grimes 		ap = 0;
83358f0484fSRodney W. Grimes 	return (ap);
83458f0484fSRodney W. Grimes }
83558f0484fSRodney W. Grimes 
83658f0484fSRodney W. Grimes /*
83758f0484fSRodney W. Grimes  * Get the command args.  This code is now machine independent.
83858f0484fSRodney W. Grimes  */
83958f0484fSRodney W. Grimes char **
84058f0484fSRodney W. Grimes kvm_getargv(kd, kp, nchr)
84158f0484fSRodney W. Grimes 	kvm_t *kd;
84258f0484fSRodney W. Grimes 	const struct kinfo_proc *kp;
84358f0484fSRodney W. Grimes 	int nchr;
84458f0484fSRodney W. Grimes {
845b9df5231SPoul-Henning Kamp 	int oid[4];
846b7875890SDavid E. O'Brien 	int i;
847b7875890SDavid E. O'Brien 	size_t bufsz;
84815d2f5f0SPeter Wemm 	static unsigned long buflen;
849b9df5231SPoul-Henning Kamp 	static char *buf, *p;
850b9df5231SPoul-Henning Kamp 	static char **bufp;
851b9df5231SPoul-Henning Kamp 	static int argc;
852b9df5231SPoul-Henning Kamp 
853c4a7cdb3SPeter Wemm 	if (!ISALIVE(kd)) {
854c4a7cdb3SPeter Wemm 		_kvm_err(kd, kd->program,
855c4a7cdb3SPeter Wemm 		    "cannot read user space from dead kernel");
856c4a7cdb3SPeter Wemm 		return (0);
857c4a7cdb3SPeter Wemm 	}
858c4a7cdb3SPeter Wemm 
859b9df5231SPoul-Henning Kamp 	if (!buflen) {
860b7875890SDavid E. O'Brien 		bufsz = sizeof(buflen);
861b9df5231SPoul-Henning Kamp 		i = sysctlbyname("kern.ps_arg_cache_limit",
862b7875890SDavid E. O'Brien 		    &buflen, &bufsz, NULL, 0);
863b9df5231SPoul-Henning Kamp 		if (i == -1) {
864b7875890SDavid E. O'Brien 			buflen = 0;
865b9df5231SPoul-Henning Kamp 		} else {
866b9df5231SPoul-Henning Kamp 			buf = malloc(buflen);
867b9df5231SPoul-Henning Kamp 			if (buf == NULL)
868b9df5231SPoul-Henning Kamp 				buflen = 0;
869b9df5231SPoul-Henning Kamp 			argc = 32;
870b9df5231SPoul-Henning Kamp 			bufp = malloc(sizeof(char *) * argc);
871b9df5231SPoul-Henning Kamp 		}
872b9df5231SPoul-Henning Kamp 	}
873b9df5231SPoul-Henning Kamp 	if (buf != NULL) {
874b9df5231SPoul-Henning Kamp 		oid[0] = CTL_KERN;
875b9df5231SPoul-Henning Kamp 		oid[1] = KERN_PROC;
876b9df5231SPoul-Henning Kamp 		oid[2] = KERN_PROC_ARGS;
8771f7d2501SKirk McKusick 		oid[3] = kp->ki_pid;
878b7875890SDavid E. O'Brien 		bufsz = buflen;
879b7875890SDavid E. O'Brien 		i = sysctl(oid, 4, buf, &bufsz, 0, 0);
880b7875890SDavid E. O'Brien 		if (i == 0 && bufsz > 0) {
881b9df5231SPoul-Henning Kamp 			i = 0;
882b9df5231SPoul-Henning Kamp 			p = buf;
883b9df5231SPoul-Henning Kamp 			do {
884b9df5231SPoul-Henning Kamp 				bufp[i++] = p;
885b9df5231SPoul-Henning Kamp 				p += strlen(p) + 1;
886b9df5231SPoul-Henning Kamp 				if (i >= argc) {
887b9df5231SPoul-Henning Kamp 					argc += argc;
888b9df5231SPoul-Henning Kamp 					bufp = realloc(bufp,
889b9df5231SPoul-Henning Kamp 					    sizeof(char *) * argc);
890b9df5231SPoul-Henning Kamp 				}
891b7875890SDavid E. O'Brien 			} while (p < buf + bufsz);
892b9df5231SPoul-Henning Kamp 			bufp[i++] = 0;
893b9df5231SPoul-Henning Kamp 			return (bufp);
894b9df5231SPoul-Henning Kamp 		}
895b9df5231SPoul-Henning Kamp 	}
8961f7d2501SKirk McKusick 	if (kp->ki_flag & P_SYSTEM)
897b9df5231SPoul-Henning Kamp 		return (NULL);
89858f0484fSRodney W. Grimes 	return (kvm_doargv(kd, kp, nchr, ps_str_a));
89958f0484fSRodney W. Grimes }
90058f0484fSRodney W. Grimes 
90158f0484fSRodney W. Grimes char **
90258f0484fSRodney W. Grimes kvm_getenvv(kd, kp, nchr)
90358f0484fSRodney W. Grimes 	kvm_t *kd;
90458f0484fSRodney W. Grimes 	const struct kinfo_proc *kp;
90558f0484fSRodney W. Grimes 	int nchr;
90658f0484fSRodney W. Grimes {
90758f0484fSRodney W. Grimes 	return (kvm_doargv(kd, kp, nchr, ps_str_e));
90858f0484fSRodney W. Grimes }
90958f0484fSRodney W. Grimes 
91058f0484fSRodney W. Grimes /*
91158f0484fSRodney W. Grimes  * Read from user space.  The user context is given by p.
91258f0484fSRodney W. Grimes  */
91358f0484fSRodney W. Grimes ssize_t
9141f7d2501SKirk McKusick kvm_uread(kd, kp, uva, buf, len)
91558f0484fSRodney W. Grimes 	kvm_t *kd;
9161f7d2501SKirk McKusick 	struct kinfo_proc *kp;
917be04b6d1SDavid E. O'Brien 	u_long uva;
918be04b6d1SDavid E. O'Brien 	char *buf;
919be04b6d1SDavid E. O'Brien 	size_t len;
92058f0484fSRodney W. Grimes {
921be04b6d1SDavid E. O'Brien 	char *cp;
922338c7541SDavid Greenman 	char procfile[MAXPATHLEN];
923338c7541SDavid Greenman 	ssize_t amount;
924338c7541SDavid Greenman 	int fd;
92558f0484fSRodney W. Grimes 
9267350dd84SPeter Wemm 	if (!ISALIVE(kd)) {
927d7fb4b13SBruce Evans 		_kvm_err(kd, kd->program,
928d7fb4b13SBruce Evans 		    "cannot read user space from dead kernel");
9297350dd84SPeter Wemm 		return (0);
9307350dd84SPeter Wemm 	}
9317350dd84SPeter Wemm 
9321f7d2501SKirk McKusick 	sprintf(procfile, "/proc/%d/mem", kp->ki_pid);
933338c7541SDavid Greenman 	fd = open(procfile, O_RDONLY, 0);
934338c7541SDavid Greenman 	if (fd < 0) {
935338c7541SDavid Greenman 		_kvm_err(kd, kd->program, "cannot open %s", procfile);
936338c7541SDavid Greenman 		close(fd);
93758f0484fSRodney W. Grimes 		return (0);
93858f0484fSRodney W. Grimes 	}
939338c7541SDavid Greenman 
940d7fb4b13SBruce Evans 	cp = buf;
941338c7541SDavid Greenman 	while (len > 0) {
942d7fb4b13SBruce Evans 		errno = 0;
943ec4f2251SJoerg Wunsch 		if (lseek(fd, (off_t)uva, 0) == -1 && errno != 0) {
944d7fb4b13SBruce Evans 			_kvm_err(kd, kd->program, "invalid address (%x) in %s",
945d7fb4b13SBruce Evans 			    uva, procfile);
94658f0484fSRodney W. Grimes 			break;
94758f0484fSRodney W. Grimes 		}
948567127faSDavid Greenman 		amount = read(fd, cp, len);
949338c7541SDavid Greenman 		if (amount < 0) {
950d7fb4b13SBruce Evans 			_kvm_syserr(kd, kd->program, "error reading %s",
951d7fb4b13SBruce Evans 			    procfile);
952d7fb4b13SBruce Evans 			break;
953d7fb4b13SBruce Evans 		}
954d7fb4b13SBruce Evans 		if (amount == 0) {
955d7fb4b13SBruce Evans 			_kvm_err(kd, kd->program, "EOF reading %s", procfile);
956338c7541SDavid Greenman 			break;
957338c7541SDavid Greenman 		}
958338c7541SDavid Greenman 		cp += amount;
959338c7541SDavid Greenman 		uva += amount;
960338c7541SDavid Greenman 		len -= amount;
961338c7541SDavid Greenman 	}
962338c7541SDavid Greenman 
963338c7541SDavid Greenman 	close(fd);
964d7fb4b13SBruce Evans 	return ((ssize_t)(cp - buf));
96558f0484fSRodney W. Grimes }
966