xref: /netbsd-src/sys/fs/tmpfs/tmpfs_subr.c (revision c29d51755812ace2e87aeefdb06cb2b4dac7087a)
1 /*	$NetBSD: tmpfs_subr.c,v 1.100 2015/07/07 09:30:24 justin Exp $	*/
2 
3 /*
4  * Copyright (c) 2005-2013 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Julio M. Merino Vidal, developed as part of Google's Summer of Code
9  * 2005 program, and by Mindaugas Rasiukevicius.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30  * POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 /*
34  * Efficient memory file system: interfaces for inode and directory entry
35  * construction, destruction and manipulation.
36  *
37  * Reference counting
38  *
39  *	The link count of inode (tmpfs_node_t::tn_links) is used as a
40  *	reference counter.  However, it has slightly different semantics.
41  *
42  *	For directories - link count represents directory entries, which
43  *	refer to the directories.  In other words, it represents the count
44  *	of sub-directories.  It also takes into account the virtual '.'
45  *	entry (which has no real entry in the list).  For files - link count
46  *	represents the hard links.  Since only empty directories can be
47  *	removed - link count aligns the reference counting requirements
48  *	enough.  Note: to check whether directory is not empty, the inode
49  *	size (tmpfs_node_t::tn_size) can be used.
50  *
51  *	The inode itself, as an object, gathers its first reference when
52  *	directory entry is attached via tmpfs_dir_attach(9).  For instance,
53  *	after regular tmpfs_create(), a file would have a link count of 1,
54  *	while directory after tmpfs_mkdir() would have 2 (due to '.').
55  *
56  * Reclamation
57  *
58  *	It should be noted that tmpfs inodes rely on a combination of vnode
59  *	reference counting and link counting.  That is, an inode can only be
60  *	destroyed if its associated vnode is inactive.  The destruction is
61  *	done on vnode reclamation i.e. tmpfs_reclaim().  It should be noted
62  *	that tmpfs_node_t::tn_links being 0 is a destruction criterion.
63  *
64  *	If an inode has references within the file system (tn_links > 0) and
65  *	its inactive vnode gets reclaimed/recycled - then the association is
66  *	broken in tmpfs_reclaim().  In such case, an inode will always pass
67  *	tmpfs_lookup() and thus vcache_get() to associate a new vnode.
68  *
69  * Lock order
70  *
71  *	vnode_t::v_vlock ->
72  *		vnode_t::v_interlock
73  */
74 
75 #include <sys/cdefs.h>
76 __KERNEL_RCSID(0, "$NetBSD: tmpfs_subr.c,v 1.100 2015/07/07 09:30:24 justin Exp $");
77 
78 #include <sys/param.h>
79 #include <sys/cprng.h>
80 #include <sys/dirent.h>
81 #include <sys/event.h>
82 #include <sys/kmem.h>
83 #include <sys/mount.h>
84 #include <sys/namei.h>
85 #include <sys/time.h>
86 #include <sys/stat.h>
87 #include <sys/systm.h>
88 #include <sys/vnode.h>
89 #include <sys/kauth.h>
90 #include <sys/atomic.h>
91 
92 #include <uvm/uvm.h>
93 
94 #include <miscfs/specfs/specdev.h>
95 #include <miscfs/genfs/genfs.h>
96 #include <fs/tmpfs/tmpfs.h>
97 #include <fs/tmpfs/tmpfs_fifoops.h>
98 #include <fs/tmpfs/tmpfs_specops.h>
99 #include <fs/tmpfs/tmpfs_vnops.h>
100 
101 static void	tmpfs_dir_putseq(tmpfs_node_t *, tmpfs_dirent_t *);
102 
103 /*
104  * Initialize vnode with tmpfs node.
105  */
106 static void
107 tmpfs_init_vnode(struct vnode *vp, tmpfs_node_t *node)
108 {
109 	kmutex_t *slock;
110 
111 	KASSERT(node->tn_vnode == NULL);
112 
113 	/* Share the interlock with the node. */
114 	if (node->tn_type == VREG) {
115 		slock = node->tn_spec.tn_reg.tn_aobj->vmobjlock;
116 		mutex_obj_hold(slock);
117 		uvm_obj_setlock(&vp->v_uobj, slock);
118 	}
119 
120 	vp->v_tag = VT_TMPFS;
121 	vp->v_type = node->tn_type;
122 
123 	/* Type-specific initialization. */
124 	switch (vp->v_type) {
125 	case VBLK:
126 	case VCHR:
127 		vp->v_op = tmpfs_specop_p;
128 		spec_node_init(vp, node->tn_spec.tn_dev.tn_rdev);
129 		break;
130 	case VFIFO:
131 		vp->v_op = tmpfs_fifoop_p;
132 		break;
133 	case VDIR:
134 		if (node->tn_spec.tn_dir.tn_parent == node)
135 			vp->v_vflag |= VV_ROOT;
136 		/* FALLTHROUGH */
137 	case VLNK:
138 	case VREG:
139 	case VSOCK:
140 		vp->v_op = tmpfs_vnodeop_p;
141 		break;
142 	default:
143 		panic("bad node type %d", vp->v_type);
144 		break;
145 	}
146 
147 	vp->v_data = node;
148 	node->tn_vnode = vp;
149 	uvm_vnp_setsize(vp, node->tn_size);
150 }
151 
152 /*
153  * tmpfs_loadvnode: initialise a vnode for a specified inode.
154  */
155 int
156 tmpfs_loadvnode(struct mount *mp, struct vnode *vp,
157     const void *key, size_t key_len, const void **new_key)
158 {
159 	tmpfs_node_t *node;
160 
161 	KASSERT(key_len == sizeof(node));
162 	memcpy(&node, key, key_len);
163 
164 	if (node->tn_links == 0)
165 		return ENOENT;
166 
167 	tmpfs_init_vnode(vp, node);
168 
169 	*new_key = &vp->v_data;
170 
171 	return 0;
172 }
173 
174 /*
175  * tmpfs_newvnode: allocate a new inode of a specified type and
176  * attach the vonode.
177  */
178 int
179 tmpfs_newvnode(struct mount *mp, struct vnode *dvp, struct vnode *vp,
180     struct vattr *vap, kauth_cred_t cred,
181     size_t *key_len, const void **new_key)
182 {
183 	tmpfs_mount_t *tmp = VFS_TO_TMPFS(mp);
184 	tmpfs_node_t *node, *dnode;
185 
186 	if (dvp != NULL) {
187 		KASSERT(VOP_ISLOCKED(dvp));
188 		dnode = VP_TO_TMPFS_DIR(dvp);
189 		if (dnode->tn_links == 0)
190 			return ENOENT;
191 		if (vap->va_type == VDIR) {
192 			/* Check for maximum links limit. */
193 			if (dnode->tn_links == LINK_MAX)
194 				return EMLINK;
195 			KASSERT(dnode->tn_links < LINK_MAX);
196 		}
197 	} else
198 		dnode = NULL;
199 
200 	node = tmpfs_node_get(tmp);
201 	if (node == NULL)
202 		return ENOSPC;
203 
204 	/* Initially, no references and no associations. */
205 	node->tn_links = 0;
206 	node->tn_vnode = NULL;
207 	node->tn_holdcount = 0;
208 	node->tn_dirent_hint = NULL;
209 
210 	/*
211 	 * XXX Where the pool is backed by a map larger than (4GB *
212 	 * sizeof(*node)), this may produce duplicate inode numbers
213 	 * for applications that do not understand 64-bit ino_t.
214 	 */
215 	node->tn_id = (ino_t)((uintptr_t)node / sizeof(*node));
216 	/*
217 	 * Make sure the generation number is not zero.
218 	 * tmpfs_inactive() uses generation zero to mark dead nodes.
219 	 */
220 	do {
221 		node->tn_gen = TMPFS_NODE_GEN_MASK & cprng_fast32();
222 	} while (node->tn_gen == 0);
223 
224 	/* Generic initialization. */
225 	KASSERT((int)vap->va_type != VNOVAL);
226 	node->tn_type = vap->va_type;
227 	node->tn_size = 0;
228 	node->tn_flags = 0;
229 	node->tn_lockf = NULL;
230 
231 	vfs_timestamp(&node->tn_atime);
232 	node->tn_birthtime = node->tn_atime;
233 	node->tn_ctime = node->tn_atime;
234 	node->tn_mtime = node->tn_atime;
235 
236 	if (dvp == NULL) {
237 		KASSERT(vap->va_uid != VNOVAL && vap->va_gid != VNOVAL);
238 		node->tn_uid = vap->va_uid;
239 		node->tn_gid = vap->va_gid;
240 		vp->v_vflag |= VV_ROOT;
241 	} else {
242 		KASSERT(dnode != NULL);
243 		node->tn_uid = kauth_cred_geteuid(cred);
244 		node->tn_gid = dnode->tn_gid;
245 	}
246 	KASSERT(vap->va_mode != VNOVAL);
247 	node->tn_mode = vap->va_mode;
248 
249 	/* Type-specific initialization. */
250 	switch (node->tn_type) {
251 	case VBLK:
252 	case VCHR:
253 		/* Character/block special device. */
254 		KASSERT(vap->va_rdev != VNOVAL);
255 		node->tn_spec.tn_dev.tn_rdev = vap->va_rdev;
256 		break;
257 	case VDIR:
258 		/* Directory. */
259 		TAILQ_INIT(&node->tn_spec.tn_dir.tn_dir);
260 		node->tn_spec.tn_dir.tn_parent = NULL;
261 		node->tn_spec.tn_dir.tn_seq_arena = NULL;
262 		node->tn_spec.tn_dir.tn_next_seq = TMPFS_DIRSEQ_START;
263 		node->tn_spec.tn_dir.tn_readdir_lastp = NULL;
264 
265 		/* Extra link count for the virtual '.' entry. */
266 		node->tn_links++;
267 		break;
268 	case VFIFO:
269 	case VSOCK:
270 		break;
271 	case VLNK:
272 		node->tn_size = 0;
273 		node->tn_spec.tn_lnk.tn_link = NULL;
274 		break;
275 	case VREG:
276 		/* Regular file.  Create an underlying UVM object. */
277 		node->tn_spec.tn_reg.tn_aobj =
278 		    uao_create(INT32_MAX - PAGE_SIZE, 0);
279 		node->tn_spec.tn_reg.tn_aobj_pages = 0;
280 		break;
281 	default:
282 		panic("bad node type %d", vp->v_type);
283 		break;
284 	}
285 
286 	tmpfs_init_vnode(vp, node);
287 
288 	mutex_enter(&tmp->tm_lock);
289 	LIST_INSERT_HEAD(&tmp->tm_nodes, node, tn_entries);
290 	mutex_exit(&tmp->tm_lock);
291 
292 	*key_len = sizeof(vp->v_data);
293 	*new_key = &vp->v_data;
294 
295 	return 0;
296 }
297 
298 /*
299  * tmpfs_free_node: remove the inode from a list in the mount point and
300  * destroy the inode structures.
301  */
302 void
303 tmpfs_free_node(tmpfs_mount_t *tmp, tmpfs_node_t *node)
304 {
305 	size_t objsz;
306 	uint32_t hold;
307 
308 	mutex_enter(&tmp->tm_lock);
309 	hold = atomic_or_32_nv(&node->tn_holdcount, TMPFS_NODE_RECLAIMED);
310 	/* Defer destruction to last thread holding this node. */
311 	if (hold != TMPFS_NODE_RECLAIMED) {
312 		mutex_exit(&tmp->tm_lock);
313 		return;
314 	}
315 	LIST_REMOVE(node, tn_entries);
316 	mutex_exit(&tmp->tm_lock);
317 
318 	switch (node->tn_type) {
319 	case VLNK:
320 		if (node->tn_size > 0) {
321 			tmpfs_strname_free(tmp, node->tn_spec.tn_lnk.tn_link,
322 			    node->tn_size);
323 		}
324 		break;
325 	case VREG:
326 		/*
327 		 * Calculate the size of inode data, decrease the used-memory
328 		 * counter, and destroy the unerlying UVM object (if any).
329 		 */
330 		objsz = PAGE_SIZE * node->tn_spec.tn_reg.tn_aobj_pages;
331 		if (objsz != 0) {
332 			tmpfs_mem_decr(tmp, objsz);
333 		}
334 		if (node->tn_spec.tn_reg.tn_aobj != NULL) {
335 			uao_detach(node->tn_spec.tn_reg.tn_aobj);
336 		}
337 		break;
338 	case VDIR:
339 		KASSERT(node->tn_size == 0);
340 		KASSERT(node->tn_spec.tn_dir.tn_seq_arena == NULL);
341 		KASSERT(TAILQ_EMPTY(&node->tn_spec.tn_dir.tn_dir));
342 		KASSERT(node->tn_spec.tn_dir.tn_parent == NULL ||
343 		    node == tmp->tm_root);
344 		break;
345 	default:
346 		break;
347 	}
348 	KASSERT(node->tn_vnode == NULL);
349 	KASSERT(node->tn_links == 0);
350 
351 	tmpfs_node_put(tmp, node);
352 }
353 
354 /*
355  * tmpfs_construct_node: allocate a new file of specified type and adds it
356  * into the parent directory.
357  *
358  * => Credentials of the caller are used.
359  */
360 int
361 tmpfs_construct_node(vnode_t *dvp, vnode_t **vpp, struct vattr *vap,
362     struct componentname *cnp, char *target)
363 {
364 	tmpfs_mount_t *tmp = VFS_TO_TMPFS(dvp->v_mount);
365 	tmpfs_node_t *dnode = VP_TO_TMPFS_DIR(dvp), *node;
366 	tmpfs_dirent_t *de, *wde;
367 	char *slink = NULL;
368 	int ssize = 0;
369 	int error;
370 
371 	/* Allocate symlink target. */
372 	if (target != NULL) {
373 		KASSERT(vap->va_type == VLNK);
374 		ssize = strlen(target);
375 		KASSERT(ssize < MAXPATHLEN);
376 		if (ssize > 0) {
377 			slink = tmpfs_strname_alloc(tmp, ssize);
378 			if (slink == NULL)
379 				return ENOSPC;
380 			memcpy(slink, target, ssize);
381 		}
382 	}
383 
384 	/* Allocate a directory entry that points to the new file. */
385 	error = tmpfs_alloc_dirent(tmp, cnp->cn_nameptr, cnp->cn_namelen, &de);
386 	if (error) {
387 		if (slink != NULL)
388 			tmpfs_strname_free(tmp, slink, ssize);
389 		return error;
390 	}
391 
392 	/* Allocate a vnode that represents the new file. */
393 	error = vcache_new(dvp->v_mount, dvp, vap, cnp->cn_cred, vpp);
394 	if (error) {
395 		if (slink != NULL)
396 			tmpfs_strname_free(tmp, slink, ssize);
397 		tmpfs_free_dirent(tmp, de);
398 		return error;
399 	}
400 	error = vn_lock(*vpp, LK_EXCLUSIVE);
401 	if (error) {
402 		vrele(*vpp);
403 		*vpp = NULL;
404 		if (slink != NULL)
405 			tmpfs_strname_free(tmp, slink, ssize);
406 		tmpfs_free_dirent(tmp, de);
407 		return error;
408 	}
409 
410 	node = VP_TO_TMPFS_NODE(*vpp);
411 
412 	if (slink != NULL) {
413 		node->tn_spec.tn_lnk.tn_link = slink;
414 		node->tn_size = ssize;
415 	}
416 
417 	/* Remove whiteout before adding the new entry. */
418 	if (cnp->cn_flags & ISWHITEOUT) {
419 		wde = tmpfs_dir_lookup(dnode, cnp);
420 		KASSERT(wde != NULL && wde->td_node == TMPFS_NODE_WHITEOUT);
421 		tmpfs_dir_detach(dnode, wde);
422 		tmpfs_free_dirent(tmp, wde);
423 	}
424 
425 	/* Associate inode and attach the entry into the directory. */
426 	tmpfs_dir_attach(dnode, de, node);
427 
428 	/* Make node opaque if requested. */
429 	if (cnp->cn_flags & ISWHITEOUT)
430 		node->tn_flags |= UF_OPAQUE;
431 
432 	/* Update the parent's timestamps. */
433 	tmpfs_update(dvp, TMPFS_UPDATE_MTIME | TMPFS_UPDATE_CTIME);
434 
435 	VOP_UNLOCK(*vpp);
436 
437 	return 0;
438 }
439 
440 /*
441  * tmpfs_alloc_dirent: allocates a new directory entry for the inode.
442  * The directory entry contains a path name component.
443  */
444 int
445 tmpfs_alloc_dirent(tmpfs_mount_t *tmp, const char *name, uint16_t len,
446     tmpfs_dirent_t **de)
447 {
448 	tmpfs_dirent_t *nde;
449 
450 	nde = tmpfs_dirent_get(tmp);
451 	if (nde == NULL)
452 		return ENOSPC;
453 
454 	nde->td_name = tmpfs_strname_alloc(tmp, len);
455 	if (nde->td_name == NULL) {
456 		tmpfs_dirent_put(tmp, nde);
457 		return ENOSPC;
458 	}
459 	nde->td_namelen = len;
460 	memcpy(nde->td_name, name, len);
461 	nde->td_seq = TMPFS_DIRSEQ_NONE;
462 
463 	*de = nde;
464 	return 0;
465 }
466 
467 /*
468  * tmpfs_free_dirent: free a directory entry.
469  */
470 void
471 tmpfs_free_dirent(tmpfs_mount_t *tmp, tmpfs_dirent_t *de)
472 {
473 	KASSERT(de->td_node == NULL);
474 	KASSERT(de->td_seq == TMPFS_DIRSEQ_NONE);
475 	tmpfs_strname_free(tmp, de->td_name, de->td_namelen);
476 	tmpfs_dirent_put(tmp, de);
477 }
478 
479 /*
480  * tmpfs_dir_attach: associate directory entry with a specified inode,
481  * and attach the entry into the directory, specified by vnode.
482  *
483  * => Increases link count on the associated node.
484  * => Increases link count on directory node if our node is VDIR.
485  * => It is caller's responsibility to check for the LINK_MAX limit.
486  * => Triggers kqueue events here.
487  */
488 void
489 tmpfs_dir_attach(tmpfs_node_t *dnode, tmpfs_dirent_t *de, tmpfs_node_t *node)
490 {
491 	vnode_t *dvp = dnode->tn_vnode;
492 	int events = NOTE_WRITE;
493 
494 	KASSERT(dvp != NULL);
495 	KASSERT(VOP_ISLOCKED(dvp));
496 
497 	/* Get a new sequence number. */
498 	KASSERT(de->td_seq == TMPFS_DIRSEQ_NONE);
499 	de->td_seq = tmpfs_dir_getseq(dnode, de);
500 
501 	/* Associate directory entry and the inode. */
502 	de->td_node = node;
503 	if (node != TMPFS_NODE_WHITEOUT) {
504 		KASSERT(node->tn_links < LINK_MAX);
505 		node->tn_links++;
506 
507 		/* Save the hint (might overwrite). */
508 		node->tn_dirent_hint = de;
509 	} else if ((dnode->tn_gen & TMPFS_WHITEOUT_BIT) == 0) {
510 		/* Flag that there are whiteout entries. */
511 		atomic_or_32(&dnode->tn_gen, TMPFS_WHITEOUT_BIT);
512 	}
513 
514 	/* Insert the entry to the directory (parent of inode). */
515 	TAILQ_INSERT_TAIL(&dnode->tn_spec.tn_dir.tn_dir, de, td_entries);
516 	dnode->tn_size += sizeof(tmpfs_dirent_t);
517 	uvm_vnp_setsize(dvp, dnode->tn_size);
518 
519 	if (node != TMPFS_NODE_WHITEOUT && node->tn_type == VDIR) {
520 		/* Set parent. */
521 		KASSERT(node->tn_spec.tn_dir.tn_parent == NULL);
522 		node->tn_spec.tn_dir.tn_parent = dnode;
523 
524 		/* Increase the link count of parent. */
525 		KASSERT(dnode->tn_links < LINK_MAX);
526 		dnode->tn_links++;
527 		events |= NOTE_LINK;
528 
529 		TMPFS_VALIDATE_DIR(node);
530 	}
531 	VN_KNOTE(dvp, events);
532 }
533 
534 /*
535  * tmpfs_dir_detach: disassociate directory entry and its inode,
536  * and detach the entry from the directory, specified by vnode.
537  *
538  * => Decreases link count on the associated node.
539  * => Decreases the link count on directory node, if our node is VDIR.
540  * => Triggers kqueue events here.
541  *
542  * => Note: dvp and vp may be NULL only if called by tmpfs_unmount().
543  */
544 void
545 tmpfs_dir_detach(tmpfs_node_t *dnode, tmpfs_dirent_t *de)
546 {
547 	tmpfs_node_t *node = de->td_node;
548 	vnode_t *vp, *dvp = dnode->tn_vnode;
549 	int events = NOTE_WRITE;
550 
551 	KASSERT(dvp == NULL || VOP_ISLOCKED(dvp));
552 
553 	if (__predict_true(node != TMPFS_NODE_WHITEOUT)) {
554 		/* Deassociate the inode and entry. */
555 		node->tn_dirent_hint = NULL;
556 
557 		KASSERT(node->tn_links > 0);
558 		node->tn_links--;
559 
560 		if ((vp = node->tn_vnode) != NULL) {
561 			KASSERT(VOP_ISLOCKED(vp));
562 			VN_KNOTE(vp, node->tn_links ? NOTE_LINK : NOTE_DELETE);
563 		}
564 
565 		/* If directory - decrease the link count of parent. */
566 		if (node->tn_type == VDIR) {
567 			KASSERT(node->tn_spec.tn_dir.tn_parent == dnode);
568 			node->tn_spec.tn_dir.tn_parent = NULL;
569 
570 			KASSERT(dnode->tn_links > 0);
571 			dnode->tn_links--;
572 			events |= NOTE_LINK;
573 		}
574 	}
575 	de->td_node = NULL;
576 
577 	/* Remove the entry from the directory. */
578 	if (dnode->tn_spec.tn_dir.tn_readdir_lastp == de) {
579 		dnode->tn_spec.tn_dir.tn_readdir_lastp = NULL;
580 	}
581 	TAILQ_REMOVE(&dnode->tn_spec.tn_dir.tn_dir, de, td_entries);
582 	dnode->tn_size -= sizeof(tmpfs_dirent_t);
583 	tmpfs_dir_putseq(dnode, de);
584 
585 	if (dvp) {
586 		uvm_vnp_setsize(dvp, dnode->tn_size);
587 		VN_KNOTE(dvp, events);
588 	}
589 }
590 
591 /*
592  * tmpfs_dir_lookup: find a directory entry in the specified inode.
593  *
594  * Note that the . and .. components are not allowed as they do not
595  * physically exist within directories.
596  */
597 tmpfs_dirent_t *
598 tmpfs_dir_lookup(tmpfs_node_t *node, struct componentname *cnp)
599 {
600 	const char *name = cnp->cn_nameptr;
601 	const uint16_t nlen = cnp->cn_namelen;
602 	tmpfs_dirent_t *de;
603 
604 	KASSERT(VOP_ISLOCKED(node->tn_vnode));
605 	KASSERT(nlen != 1 || !(name[0] == '.'));
606 	KASSERT(nlen != 2 || !(name[0] == '.' && name[1] == '.'));
607 	TMPFS_VALIDATE_DIR(node);
608 
609 	TAILQ_FOREACH(de, &node->tn_spec.tn_dir.tn_dir, td_entries) {
610 		if (de->td_namelen != nlen)
611 			continue;
612 		if (memcmp(de->td_name, name, nlen) != 0)
613 			continue;
614 		break;
615 	}
616 	return de;
617 }
618 
619 /*
620  * tmpfs_dir_cached: get a cached directory entry if it is valid.  Used to
621  * avoid unnecessary tmpfs_dir_lookup().
622  *
623  * => The vnode must be locked.
624  */
625 tmpfs_dirent_t *
626 tmpfs_dir_cached(tmpfs_node_t *node)
627 {
628 	tmpfs_dirent_t *de = node->tn_dirent_hint;
629 
630 	KASSERT(VOP_ISLOCKED(node->tn_vnode));
631 
632 	if (de == NULL) {
633 		return NULL;
634 	}
635 	KASSERT(de->td_node == node);
636 
637 	/*
638 	 * Directories always have a valid hint.  For files, check if there
639 	 * are any hard links.  If there are - hint might be invalid.
640 	 */
641 	return (node->tn_type != VDIR && node->tn_links > 1) ? NULL : de;
642 }
643 
644 /*
645  * tmpfs_dir_getseq: get a per-directory sequence number for the entry.
646  *
647  * => Shall not be larger than 2^31 for linux32 compatibility.
648  */
649 uint32_t
650 tmpfs_dir_getseq(tmpfs_node_t *dnode, tmpfs_dirent_t *de)
651 {
652 	uint32_t seq = de->td_seq;
653 	vmem_t *seq_arena;
654 	vmem_addr_t off;
655 	int error __diagused;
656 
657 	TMPFS_VALIDATE_DIR(dnode);
658 
659 	if (__predict_true(seq != TMPFS_DIRSEQ_NONE)) {
660 		/* Already set. */
661 		KASSERT(seq >= TMPFS_DIRSEQ_START);
662 		return seq;
663 	}
664 
665 	/*
666 	 * The "." and ".." and the end-of-directory have reserved numbers.
667 	 * The other sequence numbers are allocated as following:
668 	 *
669 	 * - The first half of the 2^31 is assigned incrementally.
670 	 *
671 	 * - If that range is exceeded, then the second half of 2^31
672 	 * is used, but managed by vmem(9).
673 	 */
674 
675 	seq = dnode->tn_spec.tn_dir.tn_next_seq;
676 	KASSERT(seq >= TMPFS_DIRSEQ_START);
677 
678 	if (__predict_true(seq < TMPFS_DIRSEQ_END)) {
679 		/* First half: just increment and return. */
680 		dnode->tn_spec.tn_dir.tn_next_seq++;
681 		return seq;
682 	}
683 
684 	/*
685 	 * First half exceeded, use the second half.  May need to create
686 	 * vmem(9) arena for the directory first.
687 	 */
688 	if ((seq_arena = dnode->tn_spec.tn_dir.tn_seq_arena) == NULL) {
689 		seq_arena = vmem_create("tmpfscoo", 0,
690 		    TMPFS_DIRSEQ_END - 1, 1, NULL, NULL, NULL, 0,
691 		    VM_SLEEP, IPL_NONE);
692 		dnode->tn_spec.tn_dir.tn_seq_arena = seq_arena;
693 		KASSERT(seq_arena != NULL);
694 	}
695 	error = vmem_alloc(seq_arena, 1, VM_SLEEP | VM_BESTFIT, &off);
696 	KASSERT(error == 0);
697 
698 	KASSERT(off < TMPFS_DIRSEQ_END);
699 	seq = off | TMPFS_DIRSEQ_END;
700 	return seq;
701 }
702 
703 static void
704 tmpfs_dir_putseq(tmpfs_node_t *dnode, tmpfs_dirent_t *de)
705 {
706 	vmem_t *seq_arena = dnode->tn_spec.tn_dir.tn_seq_arena;
707 	uint32_t seq = de->td_seq;
708 
709 	TMPFS_VALIDATE_DIR(dnode);
710 
711 	if (seq == TMPFS_DIRSEQ_NONE || seq < TMPFS_DIRSEQ_END) {
712 		/* First half (or no sequence number set yet). */
713 		KASSERT(de->td_seq >= TMPFS_DIRSEQ_START);
714 	} else {
715 		/* Second half. */
716 		KASSERT(seq_arena != NULL);
717 		KASSERT(seq >= TMPFS_DIRSEQ_END);
718 		seq &= ~TMPFS_DIRSEQ_END;
719 		vmem_free(seq_arena, seq, 1);
720 	}
721 	de->td_seq = TMPFS_DIRSEQ_NONE;
722 
723 	/* Empty?  We can reset. */
724 	if (seq_arena && dnode->tn_size == 0) {
725 		dnode->tn_spec.tn_dir.tn_seq_arena = NULL;
726 		dnode->tn_spec.tn_dir.tn_next_seq = TMPFS_DIRSEQ_START;
727 		vmem_destroy(seq_arena);
728 	}
729 }
730 
731 /*
732  * tmpfs_dir_lookupbyseq: lookup a directory entry by the sequence number.
733  */
734 tmpfs_dirent_t *
735 tmpfs_dir_lookupbyseq(tmpfs_node_t *node, off_t seq)
736 {
737 	tmpfs_dirent_t *de = node->tn_spec.tn_dir.tn_readdir_lastp;
738 
739 	TMPFS_VALIDATE_DIR(node);
740 
741 	/*
742 	 * First, check the cache.  If does not match - perform a lookup.
743 	 */
744 	if (de && de->td_seq == seq) {
745 		KASSERT(de->td_seq >= TMPFS_DIRSEQ_START);
746 		KASSERT(de->td_seq != TMPFS_DIRSEQ_NONE);
747 		return de;
748 	}
749 	TAILQ_FOREACH(de, &node->tn_spec.tn_dir.tn_dir, td_entries) {
750 		KASSERT(de->td_seq >= TMPFS_DIRSEQ_START);
751 		KASSERT(de->td_seq != TMPFS_DIRSEQ_NONE);
752 		if (de->td_seq == seq)
753 			return de;
754 	}
755 	return NULL;
756 }
757 
758 /*
759  * tmpfs_dir_getdotents: helper function for tmpfs_readdir() to get the
760  * dot meta entries, that is, "." or "..".  Copy it to the UIO space.
761  */
762 static int
763 tmpfs_dir_getdotents(tmpfs_node_t *node, struct dirent *dp, struct uio *uio)
764 {
765 	tmpfs_dirent_t *de;
766 	off_t next = 0;
767 	int error;
768 
769 	switch (uio->uio_offset) {
770 	case TMPFS_DIRSEQ_DOT:
771 		dp->d_fileno = node->tn_id;
772 		strlcpy(dp->d_name, ".", sizeof(dp->d_name));
773 		next = TMPFS_DIRSEQ_DOTDOT;
774 		break;
775 	case TMPFS_DIRSEQ_DOTDOT:
776 		dp->d_fileno = node->tn_spec.tn_dir.tn_parent->tn_id;
777 		strlcpy(dp->d_name, "..", sizeof(dp->d_name));
778 		de = TAILQ_FIRST(&node->tn_spec.tn_dir.tn_dir);
779 		next = de ? tmpfs_dir_getseq(node, de) : TMPFS_DIRSEQ_EOF;
780 		break;
781 	default:
782 		KASSERT(false);
783 	}
784 	dp->d_type = DT_DIR;
785 	dp->d_namlen = strlen(dp->d_name);
786 	dp->d_reclen = _DIRENT_SIZE(dp);
787 
788 	if (dp->d_reclen > uio->uio_resid) {
789 		return EJUSTRETURN;
790 	}
791 	if ((error = uiomove(dp, dp->d_reclen, uio)) != 0) {
792 		return error;
793 	}
794 
795 	uio->uio_offset = next;
796 	return error;
797 }
798 
799 /*
800  * tmpfs_dir_getdents: helper function for tmpfs_readdir.
801  *
802  * => Returns as much directory entries as can fit in the uio space.
803  * => The read starts at uio->uio_offset.
804  */
805 int
806 tmpfs_dir_getdents(tmpfs_node_t *node, struct uio *uio, off_t *cntp)
807 {
808 	tmpfs_dirent_t *de;
809 	struct dirent dent;
810 	int error = 0;
811 
812 	KASSERT(VOP_ISLOCKED(node->tn_vnode));
813 	TMPFS_VALIDATE_DIR(node);
814 
815 	/*
816 	 * First check for the "." and ".." cases.
817 	 * Note: tmpfs_dir_getdotents() will "seek" for us.
818 	 */
819 	memset(&dent, 0, sizeof(dent));
820 
821 	if (uio->uio_offset == TMPFS_DIRSEQ_DOT) {
822 		if ((error = tmpfs_dir_getdotents(node, &dent, uio)) != 0) {
823 			goto done;
824 		}
825 		(*cntp)++;
826 	}
827 	if (uio->uio_offset == TMPFS_DIRSEQ_DOTDOT) {
828 		if ((error = tmpfs_dir_getdotents(node, &dent, uio)) != 0) {
829 			goto done;
830 		}
831 		(*cntp)++;
832 	}
833 
834 	/* Done if we reached the end. */
835 	if (uio->uio_offset == TMPFS_DIRSEQ_EOF) {
836 		goto done;
837 	}
838 
839 	/* Locate the directory entry given by the given sequence number. */
840 	de = tmpfs_dir_lookupbyseq(node, uio->uio_offset);
841 	if (de == NULL) {
842 		error = EINVAL;
843 		goto done;
844 	}
845 
846 	/*
847 	 * Read as many entries as possible; i.e., until we reach the end
848 	 * of the directory or we exhaust UIO space.
849 	 */
850 	do {
851 		if (de->td_node == TMPFS_NODE_WHITEOUT) {
852 			dent.d_fileno = 1;
853 			dent.d_type = DT_WHT;
854 		} else {
855 			dent.d_fileno = de->td_node->tn_id;
856 			dent.d_type = vtype2dt(de->td_node->tn_type);
857 		}
858 		dent.d_namlen = de->td_namelen;
859 		KASSERT(de->td_namelen < sizeof(dent.d_name));
860 		memcpy(dent.d_name, de->td_name, de->td_namelen);
861 		dent.d_name[de->td_namelen] = '\0';
862 		dent.d_reclen = _DIRENT_SIZE(&dent);
863 
864 		if (dent.d_reclen > uio->uio_resid) {
865 			/* Exhausted UIO space. */
866 			error = EJUSTRETURN;
867 			break;
868 		}
869 
870 		/* Copy out the directory entry and continue. */
871 		error = uiomove(&dent, dent.d_reclen, uio);
872 		if (error) {
873 			break;
874 		}
875 		(*cntp)++;
876 		de = TAILQ_NEXT(de, td_entries);
877 
878 	} while (uio->uio_resid > 0 && de);
879 
880 	/* Cache the last entry or clear and mark EOF. */
881 	uio->uio_offset = de ? tmpfs_dir_getseq(node, de) : TMPFS_DIRSEQ_EOF;
882 	node->tn_spec.tn_dir.tn_readdir_lastp = de;
883 done:
884 	tmpfs_update(node->tn_vnode, TMPFS_UPDATE_ATIME);
885 
886 	if (error == EJUSTRETURN) {
887 		/* Exhausted UIO space - just return. */
888 		error = 0;
889 	}
890 	KASSERT(error >= 0);
891 	return error;
892 }
893 
894 /*
895  * tmpfs_reg_resize: resize the underlying UVM object associated with the
896  * specified regular file.
897  */
898 int
899 tmpfs_reg_resize(struct vnode *vp, off_t newsize)
900 {
901 	tmpfs_mount_t *tmp = VFS_TO_TMPFS(vp->v_mount);
902 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
903 	struct uvm_object *uobj = node->tn_spec.tn_reg.tn_aobj;
904 	size_t newpages, oldpages;
905 	off_t oldsize;
906 
907 	KASSERT(vp->v_type == VREG);
908 	KASSERT(newsize >= 0);
909 
910 	oldsize = node->tn_size;
911 	oldpages = round_page(oldsize) >> PAGE_SHIFT;
912 	newpages = round_page(newsize) >> PAGE_SHIFT;
913 	KASSERT(oldpages == node->tn_spec.tn_reg.tn_aobj_pages);
914 
915 	if (newpages > oldpages) {
916 		/* Increase the used-memory counter if getting extra pages. */
917 		if (!tmpfs_mem_incr(tmp, (newpages - oldpages) << PAGE_SHIFT)) {
918 			return ENOSPC;
919 		}
920 	} else if (newsize < oldsize) {
921 		size_t zerolen;
922 
923 		zerolen = MIN(round_page(newsize), node->tn_size) - newsize;
924 		ubc_zerorange(uobj, newsize, zerolen, UBC_UNMAP_FLAG(vp));
925 	}
926 
927 	node->tn_spec.tn_reg.tn_aobj_pages = newpages;
928 	node->tn_size = newsize;
929 	uvm_vnp_setsize(vp, newsize);
930 
931 	/*
932 	 * Free "backing store".
933 	 */
934 	if (newpages < oldpages) {
935 		KASSERT(uobj->vmobjlock == vp->v_interlock);
936 
937 		mutex_enter(uobj->vmobjlock);
938 		uao_dropswap_range(uobj, newpages, oldpages);
939 		mutex_exit(uobj->vmobjlock);
940 
941 		/* Decrease the used-memory counter. */
942 		tmpfs_mem_decr(tmp, (oldpages - newpages) << PAGE_SHIFT);
943 	}
944 	if (newsize > oldsize) {
945 		VN_KNOTE(vp, NOTE_EXTEND);
946 	}
947 	return 0;
948 }
949 
950 /*
951  * tmpfs_chflags: change flags of the given vnode.
952  */
953 int
954 tmpfs_chflags(vnode_t *vp, int flags, kauth_cred_t cred, lwp_t *l)
955 {
956 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
957 	kauth_action_t action = KAUTH_VNODE_WRITE_FLAGS;
958 	int error;
959 	bool changing_sysflags = false;
960 
961 	KASSERT(VOP_ISLOCKED(vp));
962 
963 	/* Disallow this operation if the file system is mounted read-only. */
964 	if (vp->v_mount->mnt_flag & MNT_RDONLY)
965 		return EROFS;
966 
967 	/*
968 	 * If the new flags have non-user flags that are different than
969 	 * those on the node, we need special permission to change them.
970 	 */
971 	if ((flags & SF_SETTABLE) != (node->tn_flags & SF_SETTABLE)) {
972 		action |= KAUTH_VNODE_WRITE_SYSFLAGS;
973 		changing_sysflags = true;
974 	}
975 
976 	/*
977 	 * Indicate that this node's flags have system attributes in them if
978 	 * that's the case.
979 	 */
980 	if (node->tn_flags & (SF_IMMUTABLE | SF_APPEND)) {
981 		action |= KAUTH_VNODE_HAS_SYSFLAGS;
982 	}
983 
984 	error = kauth_authorize_vnode(cred, action, vp, NULL,
985 	    genfs_can_chflags(cred, vp->v_type, node->tn_uid,
986 	    changing_sysflags));
987 	if (error)
988 		return error;
989 
990 	/*
991 	 * Set the flags. If we're not setting non-user flags, be careful not
992 	 * to overwrite them.
993 	 *
994 	 * XXX: Can't we always assign here? if the system flags are different,
995 	 *      the code above should catch attempts to change them without
996 	 *      proper permissions, and if we're here it means it's okay to
997 	 *      change them...
998 	 */
999 	if (!changing_sysflags) {
1000 		/* Clear all user-settable flags and re-set them. */
1001 		node->tn_flags &= SF_SETTABLE;
1002 		node->tn_flags |= (flags & UF_SETTABLE);
1003 	} else {
1004 		node->tn_flags = flags;
1005 	}
1006 	tmpfs_update(vp, TMPFS_UPDATE_CTIME);
1007 	VN_KNOTE(vp, NOTE_ATTRIB);
1008 	return 0;
1009 }
1010 
1011 /*
1012  * tmpfs_chmod: change access mode on the given vnode.
1013  */
1014 int
1015 tmpfs_chmod(vnode_t *vp, mode_t mode, kauth_cred_t cred, lwp_t *l)
1016 {
1017 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
1018 	int error;
1019 
1020 	KASSERT(VOP_ISLOCKED(vp));
1021 
1022 	/* Disallow this operation if the file system is mounted read-only. */
1023 	if (vp->v_mount->mnt_flag & MNT_RDONLY)
1024 		return EROFS;
1025 
1026 	/* Immutable or append-only files cannot be modified, either. */
1027 	if (node->tn_flags & (IMMUTABLE | APPEND))
1028 		return EPERM;
1029 
1030 	error = kauth_authorize_vnode(cred, KAUTH_VNODE_WRITE_SECURITY, vp,
1031 	    NULL, genfs_can_chmod(vp->v_type, cred, node->tn_uid, node->tn_gid, mode));
1032 	if (error) {
1033 		return error;
1034 	}
1035 	node->tn_mode = (mode & ALLPERMS);
1036 	tmpfs_update(vp, TMPFS_UPDATE_CTIME);
1037 	VN_KNOTE(vp, NOTE_ATTRIB);
1038 	return 0;
1039 }
1040 
1041 /*
1042  * tmpfs_chown: change ownership of the given vnode.
1043  *
1044  * => At least one of uid or gid must be different than VNOVAL.
1045  * => Attribute is unchanged for VNOVAL case.
1046  */
1047 int
1048 tmpfs_chown(vnode_t *vp, uid_t uid, gid_t gid, kauth_cred_t cred, lwp_t *l)
1049 {
1050 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
1051 	int error;
1052 
1053 	KASSERT(VOP_ISLOCKED(vp));
1054 
1055 	/* Assign default values if they are unknown. */
1056 	KASSERT(uid != VNOVAL || gid != VNOVAL);
1057 	if (uid == VNOVAL) {
1058 		uid = node->tn_uid;
1059 	}
1060 	if (gid == VNOVAL) {
1061 		gid = node->tn_gid;
1062 	}
1063 
1064 	/* Disallow this operation if the file system is mounted read-only. */
1065 	if (vp->v_mount->mnt_flag & MNT_RDONLY)
1066 		return EROFS;
1067 
1068 	/* Immutable or append-only files cannot be modified, either. */
1069 	if (node->tn_flags & (IMMUTABLE | APPEND))
1070 		return EPERM;
1071 
1072 	error = kauth_authorize_vnode(cred, KAUTH_VNODE_CHANGE_OWNERSHIP, vp,
1073 	    NULL, genfs_can_chown(cred, node->tn_uid, node->tn_gid, uid,
1074 	    gid));
1075 	if (error) {
1076 		return error;
1077 	}
1078 	node->tn_uid = uid;
1079 	node->tn_gid = gid;
1080 	tmpfs_update(vp, TMPFS_UPDATE_CTIME);
1081 	VN_KNOTE(vp, NOTE_ATTRIB);
1082 	return 0;
1083 }
1084 
1085 /*
1086  * tmpfs_chsize: change size of the given vnode.
1087  */
1088 int
1089 tmpfs_chsize(vnode_t *vp, u_quad_t size, kauth_cred_t cred, lwp_t *l)
1090 {
1091 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
1092 	const off_t length = size;
1093 	int error;
1094 
1095 	KASSERT(VOP_ISLOCKED(vp));
1096 
1097 	/* Decide whether this is a valid operation based on the file type. */
1098 	switch (vp->v_type) {
1099 	case VDIR:
1100 		return EISDIR;
1101 	case VREG:
1102 		if (vp->v_mount->mnt_flag & MNT_RDONLY) {
1103 			return EROFS;
1104 		}
1105 		break;
1106 	case VBLK:
1107 	case VCHR:
1108 	case VFIFO:
1109 		/*
1110 		 * Allow modifications of special files even if in the file
1111 		 * system is mounted read-only (we are not modifying the
1112 		 * files themselves, but the objects they represent).
1113 		 */
1114 		return 0;
1115 	default:
1116 		return EOPNOTSUPP;
1117 	}
1118 
1119 	/* Immutable or append-only files cannot be modified, either. */
1120 	if (node->tn_flags & (IMMUTABLE | APPEND)) {
1121 		return EPERM;
1122 	}
1123 
1124 	if (length < 0) {
1125 		return EINVAL;
1126 	}
1127 	if (node->tn_size == length) {
1128 		return 0;
1129 	}
1130 
1131 	/* Note: tmpfs_reg_resize() will raise NOTE_EXTEND and NOTE_ATTRIB. */
1132 	if ((error = tmpfs_reg_resize(vp, length)) != 0) {
1133 		return error;
1134 	}
1135 	tmpfs_update(vp, TMPFS_UPDATE_CTIME | TMPFS_UPDATE_MTIME);
1136 	return 0;
1137 }
1138 
1139 /*
1140  * tmpfs_chtimes: change access and modification times for vnode.
1141  */
1142 int
1143 tmpfs_chtimes(vnode_t *vp, const struct timespec *atime,
1144     const struct timespec *mtime, const struct timespec *btime,
1145     int vaflags, kauth_cred_t cred, lwp_t *l)
1146 {
1147 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
1148 	int error;
1149 
1150 	KASSERT(VOP_ISLOCKED(vp));
1151 
1152 	/* Disallow this operation if the file system is mounted read-only. */
1153 	if (vp->v_mount->mnt_flag & MNT_RDONLY)
1154 		return EROFS;
1155 
1156 	/* Immutable or append-only files cannot be modified, either. */
1157 	if (node->tn_flags & (IMMUTABLE | APPEND))
1158 		return EPERM;
1159 
1160 	error = kauth_authorize_vnode(cred, KAUTH_VNODE_WRITE_TIMES, vp, NULL,
1161 	    genfs_can_chtimes(vp, vaflags, node->tn_uid, cred));
1162 	if (error)
1163 		return error;
1164 
1165 	if (atime->tv_sec != VNOVAL) {
1166 		node->tn_atime = *atime;
1167 	}
1168 	if (mtime->tv_sec != VNOVAL) {
1169 		node->tn_mtime = *mtime;
1170 	}
1171 	if (btime->tv_sec != VNOVAL) {
1172 		node->tn_birthtime = *btime;
1173 	}
1174 	VN_KNOTE(vp, NOTE_ATTRIB);
1175 	return 0;
1176 }
1177 
1178 /*
1179  * tmpfs_update: update the timestamps as indicated by the flags.
1180  */
1181 void
1182 tmpfs_update(vnode_t *vp, unsigned tflags)
1183 {
1184 	tmpfs_node_t *node = VP_TO_TMPFS_NODE(vp);
1185 	struct timespec nowtm;
1186 
1187 	if (tflags == 0) {
1188 		return;
1189 	}
1190 	vfs_timestamp(&nowtm);
1191 
1192 	if (tflags & TMPFS_UPDATE_ATIME) {
1193 		node->tn_atime = nowtm;
1194 	}
1195 	if (tflags & TMPFS_UPDATE_MTIME) {
1196 		node->tn_mtime = nowtm;
1197 	}
1198 	if (tflags & TMPFS_UPDATE_CTIME) {
1199 		node->tn_ctime = nowtm;
1200 	}
1201 }
1202