xref: /csrg-svn/sys/kern/kern_sysctl.c (revision 50909)
1 /*
2  * Copyright (c) 1982, 1986, 1989 Regents of the University of California.
3  * All rights reserved.
4  *
5  * %sccs.include.redist.c%
6  *
7  *	@(#)kern_sysctl.c	7.18 (Berkeley) 08/28/91
8  */
9 
10 #include "param.h"
11 #include "proc.h"
12 #include "kinfo.h"
13 #include "ioctl.h"
14 #include "tty.h"
15 #include "buf.h"
16 #include "file.h"
17 
18 #include "vm/vm.h"
19 
20 #include "kinfo_proc.h"
21 
22 #define snderr(e) { error = (e); goto release;}
23 extern int kinfo_doproc(), kinfo_rtable(), kinfo_vnode(), kinfo_file();
24 extern int kinfo_meter();
25 struct kinfo_lock kinfo_lock;
26 
27 /* ARGSUSED */
28 getkerninfo(p, uap, retval)
29 	struct proc *p;
30 	register struct args {
31 		int	op;
32 		char	*where;
33 		int	*size;
34 		int	arg;
35 	} *uap;
36 	int *retval;
37 {
38 
39 	int bufsize;		/* max size of users buffer */
40 	int needed, locked, (*server)(), error = 0;
41 
42 	if (error = copyin((caddr_t)uap->size, (caddr_t)&bufsize,
43 	    sizeof (bufsize)))
44 		goto done;
45 
46 	switch (ki_type(uap->op)) {
47 
48 	case KINFO_PROC:
49 		server = kinfo_doproc;
50 		break;
51 
52 	case KINFO_RT:
53 		server = kinfo_rtable;
54 		break;
55 
56 	case KINFO_VNODE:
57 		server = kinfo_vnode;
58 		break;
59 
60 	case KINFO_FILE:
61 		server = kinfo_file;
62 		break;
63 
64 	case KINFO_METER:
65 		server = kinfo_meter;
66 		break;
67 
68 	default:
69 		error = EINVAL;
70 		goto done;
71 	}
72 	if (uap->where == NULL || uap->size == NULL) {
73 		error = (*server)(uap->op, NULL, NULL, uap->arg, &needed);
74 		goto done;
75 	}
76 	while (kinfo_lock.kl_lock) {
77 		kinfo_lock.kl_want++;
78 		sleep(&kinfo_lock, PRIBIO+1);
79 		kinfo_lock.kl_want--;
80 		kinfo_lock.kl_locked++;
81 	}
82 	kinfo_lock.kl_lock++;
83 
84 	if (!useracc(uap->where, bufsize, B_WRITE))
85 		snderr(EFAULT);
86 	if (server != kinfo_vnode)	/* XXX */
87 		vslock(uap->where, bufsize);
88 	locked = bufsize;
89 	error = (*server)(uap->op, uap->where, &bufsize, uap->arg, &needed);
90 	if (server != kinfo_vnode)	/* XXX */
91 		vsunlock(uap->where, locked, B_WRITE);
92 	if (error == 0)
93 		error = copyout((caddr_t)&bufsize,
94 				(caddr_t)uap->size, sizeof (bufsize));
95 release:
96 	kinfo_lock.kl_lock--;
97 	if (kinfo_lock.kl_want)
98 		wakeup(&kinfo_lock);
99 done:
100 	if (!error)
101 		*retval = needed;
102 	return (error);
103 }
104 
105 /*
106  * try over estimating by 5 procs
107  */
108 #define KINFO_PROCSLOP	(5 * sizeof (struct kinfo_proc))
109 
110 kinfo_doproc(op, where, acopysize, arg, aneeded)
111 	char *where;
112 	int *acopysize, *aneeded;
113 {
114 	register struct proc *p;
115 	register struct kinfo_proc *dp = (struct kinfo_proc *)where;
116 	register needed = 0;
117 	int buflen;
118 	int doingzomb;
119 	struct eproc eproc;
120 	int error = 0;
121 
122 	if (where != NULL)
123 		buflen = *acopysize;
124 
125 	p = allproc;
126 	doingzomb = 0;
127 again:
128 	for (; p != NULL; p = p->p_nxt) {
129 		/*
130 		 * TODO - make more efficient (see notes below).
131 		 * do by session.
132 		 */
133 		switch (ki_op(op)) {
134 
135 		case KINFO_PROC_PID:
136 			/* could do this with just a lookup */
137 			if (p->p_pid != (pid_t)arg)
138 				continue;
139 			break;
140 
141 		case KINFO_PROC_PGRP:
142 			/* could do this by traversing pgrp */
143 			if (p->p_pgrp->pg_id != (pid_t)arg)
144 				continue;
145 			break;
146 
147 		case KINFO_PROC_TTY:
148 			if ((p->p_flag&SCTTY) == 0 ||
149 			    p->p_session->s_ttyp == NULL ||
150 			    p->p_session->s_ttyp->t_dev != (dev_t)arg)
151 				continue;
152 			break;
153 
154 		case KINFO_PROC_UID:
155 			if (p->p_ucred->cr_uid != (uid_t)arg)
156 				continue;
157 			break;
158 
159 		case KINFO_PROC_RUID:
160 			if (p->p_cred->p_ruid != (uid_t)arg)
161 				continue;
162 			break;
163 		}
164 		if (where != NULL && buflen >= sizeof (struct kinfo_proc)) {
165 			fill_eproc(p, &eproc);
166 			if (error = copyout((caddr_t)p, &dp->kp_proc,
167 			    sizeof (struct proc)))
168 				return (error);
169 			if (error = copyout((caddr_t)&eproc, &dp->kp_eproc,
170 			    sizeof (eproc)))
171 				return (error);
172 			dp++;
173 			buflen -= sizeof (struct kinfo_proc);
174 		}
175 		needed += sizeof (struct kinfo_proc);
176 	}
177 	if (doingzomb == 0) {
178 		p = zombproc;
179 		doingzomb++;
180 		goto again;
181 	}
182 	if (where != NULL)
183 		*acopysize = (caddr_t)dp - where;
184 	else
185 		needed += KINFO_PROCSLOP;
186 	*aneeded = needed;
187 
188 	return (0);
189 }
190 
191 /*
192  * Fill in an eproc structure for the specified process.
193  */
194 void
195 fill_eproc(p, ep)
196 	register struct proc *p;
197 	register struct eproc *ep;
198 {
199 	register struct tty *tp;
200 
201 	ep->e_paddr = p;
202 	ep->e_sess = p->p_pgrp->pg_session;
203 	ep->e_pcred = *p->p_cred;
204 	ep->e_ucred = *p->p_ucred;
205 	ep->e_vm = *p->p_vmspace;
206 	if (p->p_pptr)
207 		ep->e_ppid = p->p_pptr->p_pid;
208 	else
209 		ep->e_ppid = 0;
210 	ep->e_pgid = p->p_pgrp->pg_id;
211 	ep->e_jobc = p->p_pgrp->pg_jobc;
212 	if ((p->p_flag&SCTTY) &&
213 	     (tp = ep->e_sess->s_ttyp)) {
214 		ep->e_tdev = tp->t_dev;
215 		ep->e_tpgid = tp->t_pgrp ? tp->t_pgrp->pg_id : NO_PID;
216 		ep->e_tsess = tp->t_session;
217 	} else
218 		ep->e_tdev = NODEV;
219 	ep->e_flag = ep->e_sess->s_ttyvp ? EPROC_CTTY : 0;
220 	if (SESS_LEADER(p))
221 		ep->e_flag |= EPROC_SLEADER;
222 	if (p->p_wmesg)
223 		strncpy(ep->e_wmesg, p->p_wmesg, WMESGLEN);
224 	ep->e_xsize = ep->e_xrssize = 0;
225 	ep->e_xccount = ep->e_xswrss = 0;
226 }
227 
228 /*
229  * Get file structures.
230  */
231 kinfo_file(op, where, acopysize, arg, aneeded)
232 	register char *where;
233 	int *acopysize, *aneeded;
234 {
235 	int buflen, needed, error;
236 	struct file *fp;
237 	char *start = where;
238 
239 	if (where == NULL) {
240 		/*
241 		 * overestimate by 10 files
242 		 */
243 		*aneeded = sizeof (filehead) +
244 			(nfiles + 10) * sizeof (struct file);
245 		return (0);
246 	}
247 	buflen = *acopysize;
248 	needed = 0;
249 
250 	/*
251 	 * first copyout filehead
252 	 */
253 	if (buflen > sizeof (filehead)) {
254 		if (error = copyout((caddr_t)&filehead, where,
255 		    sizeof (filehead)))
256 			return (error);
257 		buflen -= sizeof (filehead);
258 		where += sizeof (filehead);
259 	}
260 	needed += sizeof (filehead);
261 
262 	/*
263 	 * followed by an array of file structures
264 	 */
265 	for (fp = filehead; fp != NULL; fp = fp->f_filef) {
266 		if (buflen > sizeof (struct file)) {
267 			if (error = copyout((caddr_t)fp, where,
268 			    sizeof (struct file)))
269 				return (error);
270 			buflen -= sizeof (struct file);
271 			where += sizeof (struct file);
272 		}
273 		needed += sizeof (struct file);
274 	}
275 	*acopysize = where - start;
276 	*aneeded = needed;
277 
278 	return (0);
279 }
280