xref: /dragonfly/sys/vfs/procfs/procfs_subr.c (revision 19380330)
1 /*
2  * Copyright (c) 1993 Jan-Simon Pendry
3  * Copyright (c) 1993
4  *	The Regents of the University of California.  All rights reserved.
5  *
6  * This code is derived from software contributed to Berkeley by
7  * Jan-Simon Pendry.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. All advertising materials mentioning features or use of this software
18  *    must display the following acknowledgement:
19  *	This product includes software developed by the University of
20  *	California, Berkeley and its contributors.
21  * 4. Neither the name of the University nor the names of its contributors
22  *    may be used to endorse or promote products derived from this software
23  *    without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35  * SUCH DAMAGE.
36  *
37  *	@(#)procfs_subr.c	8.6 (Berkeley) 5/14/95
38  *
39  * $FreeBSD: src/sys/miscfs/procfs/procfs_subr.c,v 1.26.2.3 2002/02/18 21:28:04 des Exp $
40  */
41 
42 #include <sys/param.h>
43 #include <sys/systm.h>
44 #include <sys/sysctl.h>
45 #include <sys/proc.h>
46 #include <sys/mount.h>
47 #include <sys/vnode.h>
48 #include <sys/malloc.h>
49 #include <sys/thread2.h>
50 
51 #include <vfs/procfs/procfs.h>
52 
53 #define PFS_HSIZE	256
54 #define PFS_HMASK	(PFS_HSIZE - 1)
55 
56 static struct pfsnode *pfshead[PFS_HSIZE];
57 static int pfsvplock;
58 
59 #define PFSHASH(pid)	&pfshead[(pid) & PFS_HMASK]
60 
61 /*
62  * Allocate a pfsnode/vnode pair.  If no error occurs the returned vnode
63  * will be referenced and exclusively locked.
64  *
65  * The pid, pfs_type, and mount point uniquely identify a pfsnode.
66  * The mount point is needed because someone might mount this filesystem
67  * twice.
68  *
69  * All pfsnodes are maintained on a singly-linked list.  new nodes are
70  * only allocated when they cannot be found on this list.  entries on
71  * the list are removed when the vfs reclaim entry is called.
72  *
73  * A single lock is kept for the entire list.  this is needed because the
74  * getnewvnode() function can block waiting for a vnode to become free,
75  * in which case there may be more than one process trying to get the same
76  * vnode.  this lock is only taken if we are going to call getnewvnode,
77  * since the kernel itself is single-threaded.
78  *
79  * If an entry is found on the list, then call vget() to take a reference
80  * and obtain the lock.  This will properly re-reference the vnode if it
81  * had gotten onto the free list.
82  */
83 int
84 procfs_allocvp(struct mount *mp, struct vnode **vpp, long pid, pfstype pfs_type)
85 {
86 	struct pfsnode *pfs;
87 	struct vnode *vp;
88 	struct pfsnode **pp;
89 	int error;
90 
91 	pp = PFSHASH(pid);
92 loop:
93 	for (pfs = *pp; pfs; pfs = pfs->pfs_next) {
94 		if (pfs->pfs_pid == pid && pfs->pfs_type == pfs_type &&
95 		    PFSTOV(pfs)->v_mount == mp) {
96 			vp = PFSTOV(pfs);
97 			vhold_interlocked(vp);
98 			if (vget(vp, LK_EXCLUSIVE)) {
99 				vdrop(vp);
100 				goto loop;
101 			}
102 
103 			/*
104 			 * Make sure the vnode is still in the cache after
105 			 * getting the interlock to avoid racing a free.
106 			 */
107 			for (pfs = *pp; pfs; pfs = pfs->pfs_next) {
108 				if (PFSTOV(pfs) == vp &&
109 				    pfs->pfs_pid == pid &&
110 				    pfs->pfs_type == pfs_type &&
111 				    PFSTOV(pfs)->v_mount == mp) {
112 					break;
113 				}
114 			}
115 			vdrop(vp);
116 			if (pfs == NULL || PFSTOV(pfs) != vp) {
117 				vput(vp);
118 				goto loop;
119 
120 			}
121 			KKASSERT(vp->v_data == pfs);
122 			*vpp = vp;
123 			return (0);
124 		}
125 	}
126 
127 	/*
128 	 * otherwise lock the vp list while we call getnewvnode
129 	 * since that can block.
130 	 */
131 	if (pfsvplock & PROCFS_LOCKED) {
132 		pfsvplock |= PROCFS_WANT;
133 		(void) tsleep((caddr_t) &pfsvplock, 0, "pfsavp", 0);
134 		goto loop;
135 	}
136 	pfsvplock |= PROCFS_LOCKED;
137 
138 	/*
139 	 * Do the MALLOC before the getnewvnode since doing so afterward
140 	 * might cause a bogus v_data pointer to get dereferenced
141 	 * elsewhere if MALLOC should block.
142 	 *
143 	 * XXX this may not matter anymore since getnewvnode now returns
144 	 * a VX locked vnode.
145 	 */
146 	pfs = kmalloc(sizeof(struct pfsnode), M_TEMP, M_WAITOK);
147 
148 	error = getnewvnode(VT_PROCFS, mp, vpp, 0, 0);
149 	if (error) {
150 		kfree(pfs, M_TEMP);
151 		goto out;
152 	}
153 	vp = *vpp;
154 
155 	vp->v_data = pfs;
156 
157 	pfs->pfs_next = 0;
158 	pfs->pfs_pid = (pid_t) pid;
159 	pfs->pfs_type = pfs_type;
160 	pfs->pfs_vnode = vp;
161 	pfs->pfs_flags = 0;
162 	pfs->pfs_lockowner = 0;
163 	pfs->pfs_fileno = PROCFS_FILENO(pid, pfs_type);
164 
165 	switch (pfs_type) {
166 	case Proot:	/* /proc = dr-xr-xr-x */
167 		pfs->pfs_mode = (VREAD|VEXEC) |
168 				(VREAD|VEXEC) >> 3 |
169 				(VREAD|VEXEC) >> 6;
170 		vp->v_type = VDIR;
171 		vp->v_flag = VROOT;
172 		break;
173 
174 	case Pcurproc:	/* /proc/curproc = lr--r--r-- */
175 		pfs->pfs_mode = (VREAD) |
176 				(VREAD >> 3) |
177 				(VREAD >> 6);
178 		vp->v_type = VLNK;
179 		break;
180 
181 	case Pproc:
182 		pfs->pfs_mode = (VREAD|VEXEC) |
183 				(VREAD|VEXEC) >> 3 |
184 				(VREAD|VEXEC) >> 6;
185 		vp->v_type = VDIR;
186 		break;
187 
188 	case Pfile:
189 		pfs->pfs_mode = (VREAD|VEXEC) |
190 				(VREAD|VEXEC) >> 3 |
191 				(VREAD|VEXEC) >> 6;
192 		vp->v_type = VLNK;
193 		break;
194 
195 	case Pmem:
196 		pfs->pfs_mode = (VREAD|VWRITE);
197 		vp->v_type = VREG;
198 		break;
199 
200 	case Pregs:
201 	case Pfpregs:
202 	case Pdbregs:
203 		pfs->pfs_mode = (VREAD|VWRITE);
204 		vp->v_type = VREG;
205 		break;
206 
207 	case Pctl:
208 	case Pnote:
209 	case Pnotepg:
210 		pfs->pfs_mode = (VWRITE);
211 		vp->v_type = VREG;
212 		break;
213 
214 	case Ptype:
215 	case Pmap:
216 	case Pstatus:
217 	case Pcmdline:
218 	case Prlimit:
219 		pfs->pfs_mode = (VREAD) |
220 				(VREAD >> 3) |
221 				(VREAD >> 6);
222 		vp->v_type = VREG;
223 		break;
224 
225 	default:
226 		panic("procfs_allocvp");
227 	}
228 
229 	/* add to procfs vnode list */
230 	pfs->pfs_next = *pp;
231 	*pp = pfs;
232 
233 out:
234 	pfsvplock &= ~PROCFS_LOCKED;
235 
236 	if (pfsvplock & PROCFS_WANT) {
237 		pfsvplock &= ~PROCFS_WANT;
238 		wakeup((caddr_t) &pfsvplock);
239 	}
240 
241 	return (error);
242 }
243 
244 int
245 procfs_freevp(struct vnode *vp)
246 {
247 	struct pfsnode **pfspp;
248 	struct pfsnode *pfs;
249 
250 	pfs = VTOPFS(vp);
251 	vp->v_data = NULL;
252 
253 	pfspp = PFSHASH(pfs->pfs_pid);
254 	while (*pfspp != pfs && *pfspp)
255 		pfspp = &(*pfspp)->pfs_next;
256 	KKASSERT(*pfspp);
257 	*pfspp = pfs->pfs_next;
258 	pfs->pfs_next = NULL;
259 	pfs->pfs_vnode = NULL;
260 	kfree(pfs, M_TEMP);
261 	return (0);
262 }
263 
264 /*
265  * Try to find the calling pid. Note that pfind()
266  * now references the proc structure to be returned
267  * and needs to be released later with PRELE().
268  */
269 struct proc *
270 pfs_pfind(pid_t pfs_pid)
271 {
272 	struct proc *p = NULL;
273 
274 	if (pfs_pid == 0) {
275 		p = &proc0;
276 		PHOLD(p);
277 	} else {
278 		p = pfind(pfs_pid);
279 	}
280 
281 	/*
282 	 * Make sure the process is not in the middle of exiting (where
283 	 * a lot of its structural members may wind up being NULL).  If it
284 	 * is we give up on it.
285 	 */
286 	if (p) {
287 		lwkt_gettoken(&p->p_token);
288 		if (p->p_flags & P_POSTEXIT) {
289 			lwkt_reltoken(&p->p_token);
290 			PRELE(p);
291 			p = NULL;
292 		}
293 	}
294 	return p;
295 }
296 
297 struct proc *
298 pfs_zpfind(pid_t pfs_pid)
299 {
300 	struct proc *p = NULL;
301 
302 	if (pfs_pid == 0) {
303 		p = &proc0;
304 		PHOLD(p);
305 	} else {
306 		p = zpfind(pfs_pid);
307 	}
308 
309 	/*
310 	 * Make sure the process is not in the middle of exiting (where
311 	 * a lot of its structural members may wind up being NULL).  If it
312 	 * is we give up on it.
313 	 */
314 	if (p) {
315 		lwkt_gettoken(&p->p_token);
316 		if (p->p_flags & P_POSTEXIT) {
317 			lwkt_reltoken(&p->p_token);
318 			PRELE(p);
319 			p = NULL;
320 		}
321 	}
322 	return p;
323 }
324 
325 void
326 pfs_pdone(struct proc *p)
327 {
328 	if (p) {
329 		lwkt_reltoken(&p->p_token);
330 		PRELE(p);
331 	}
332 }
333 
334 int
335 procfs_rw(struct vop_read_args *ap)
336 {
337 	struct vnode *vp = ap->a_vp;
338 	struct uio *uio = ap->a_uio;
339 	struct thread *curtd = uio->uio_td;
340 	struct proc *curp;
341 	struct pfsnode *pfs = VTOPFS(vp);
342 	struct proc *p;
343 	struct lwp *lp;
344 	int rtval;
345 
346 	if (curtd == NULL)
347 		return (EINVAL);
348 	if ((curp = curtd->td_proc) == NULL)	/* XXX */
349 		return (EINVAL);
350 
351 	lwkt_gettoken(&proc_token);
352 	p = pfs_pfind(pfs->pfs_pid);
353 	if (p == NULL) {
354 		rtval = EINVAL;
355 		goto out;
356 	}
357 	if (p->p_pid == 1 && securelevel > 0 && uio->uio_rw == UIO_WRITE) {
358 		rtval = EACCES;
359 		goto out;
360 	}
361 	/* XXX lwp */
362 	lp = FIRST_LWP_IN_PROC(p);
363 	LWPHOLD(lp);
364 
365 	while (pfs->pfs_lockowner) {
366 		tsleep(&pfs->pfs_lockowner, 0, "pfslck", 0);
367 	}
368 	pfs->pfs_lockowner = curproc->p_pid;
369 
370 	switch (pfs->pfs_type) {
371 	case Pnote:
372 	case Pnotepg:
373 		rtval = procfs_donote(curp, lp, pfs, uio);
374 		break;
375 
376 	case Pregs:
377 		rtval = procfs_doregs(curp, lp, pfs, uio);
378 		break;
379 
380 	case Pfpregs:
381 		rtval = procfs_dofpregs(curp, lp, pfs, uio);
382 		break;
383 
384         case Pdbregs:
385                 rtval = procfs_dodbregs(curp, lp, pfs, uio);
386                 break;
387 
388 	case Pctl:
389 		rtval = procfs_doctl(curp, lp, pfs, uio);
390 		break;
391 
392 	case Pstatus:
393 		rtval = procfs_dostatus(curp, lp, pfs, uio);
394 		break;
395 
396 	case Pmap:
397 		rtval = procfs_domap(curp, lp, pfs, uio);
398 		break;
399 
400 	case Pmem:
401 		rtval = procfs_domem(curp, lp, pfs, uio);
402 		break;
403 
404 	case Ptype:
405 		rtval = procfs_dotype(curp, lp, pfs, uio);
406 		break;
407 
408 	case Pcmdline:
409 		rtval = procfs_docmdline(curp, lp, pfs, uio);
410 		break;
411 
412 	case Prlimit:
413 		rtval = procfs_dorlimit(curp, lp, pfs, uio);
414 		break;
415 
416 	default:
417 		rtval = EOPNOTSUPP;
418 		break;
419 	}
420 	LWPRELE(lp);
421 
422 	pfs->pfs_lockowner = 0;
423 	wakeup(&pfs->pfs_lockowner);
424 
425 out:
426 	pfs_pdone(p);
427 	lwkt_reltoken(&proc_token);
428 
429 	return rtval;
430 }
431 
432 /*
433  * Get a string from userland into (buf).  Strip a trailing
434  * nl character (to allow easy access from the shell).
435  * The buffer should be *buflenp + 1 chars long.  vfs_getuserstr
436  * will automatically add a nul char at the end.
437  *
438  * Returns 0 on success or the following errors
439  *
440  * EINVAL:    file offset is non-zero.
441  * EMSGSIZE:  message is longer than kernel buffer
442  * EFAULT:    user i/o buffer is not addressable
443  */
444 int
445 vfs_getuserstr(struct uio *uio, char *buf, int *buflenp)
446 {
447 	int xlen;
448 	int error;
449 
450 	if (uio->uio_offset != 0)
451 		return (EINVAL);
452 
453 	xlen = *buflenp;
454 
455 	/* must be able to read the whole string in one go */
456 	if (xlen < uio->uio_resid)
457 		return (EMSGSIZE);
458 	xlen = uio->uio_resid;
459 
460 	if ((error = uiomove(buf, xlen, uio)) != 0)
461 		return (error);
462 
463 	/* allow multiple writes without seeks */
464 	uio->uio_offset = 0;
465 
466 	/* cleanup string and remove trailing newline */
467 	buf[xlen] = '\0';
468 	xlen = strlen(buf);
469 	if (xlen > 0 && buf[xlen-1] == '\n')
470 		buf[--xlen] = '\0';
471 	*buflenp = xlen;
472 
473 	return (0);
474 }
475 
476 vfs_namemap_t *
477 vfs_findname(vfs_namemap_t *nm, char *buf, int buflen)
478 {
479 
480 	for (; nm->nm_name; nm++)
481 		if (bcmp(buf, nm->nm_name, buflen+1) == 0)
482 			return (nm);
483 
484 	return (0);
485 }
486 
487 void
488 procfs_exit(struct thread *td)
489 {
490 	struct pfsnode *pfs;
491 	struct vnode *vp;
492 	pid_t pid;
493 
494 	KKASSERT(td->td_proc);
495 	pid = td->td_proc->p_pid;
496 
497 	/*
498 	 * NOTE: We can't just vgone() the vnode any more, not while
499 	 * 	 it may potentially still be active.  This will clean
500 	 *	 the vp and clear the mount and cause the new VOP subsystem
501 	 *	 to assert or panic when someone tries to do an operation
502 	 *	 on an open (exited) procfs descriptor.
503 	 *
504 	 * Prevent further operations on this pid by setting pfs_pid to -1.
505 	 * Note that a pfs_pid of 0 is used for nodes which do not track
506 	 * any particular pid.
507 	 *
508 	 * Use vx_get() to properly ref/lock a vp which may not have any
509 	 * refs and which may or may not already be reclaimed.  vx_put()
510 	 * will then properly deactivate it and cause it to be recycled.
511 	 *
512 	 * The hash table can also get ripped out from under us when
513 	 * we block so take the easy way out and restart the scan.
514 	 */
515 again:
516 	pfs = *PFSHASH(pid);
517 	while (pfs) {
518 		if (pfs->pfs_pid == pid) {
519 			vp = PFSTOV(pfs);
520 			vx_get(vp);
521 			pfs->pfs_pid |= PFS_DEAD; /* does not effect hash */
522 			vx_put(vp);
523 			goto again;
524 		}
525 		pfs = pfs->pfs_next;
526 	}
527 }
528 
529