tmpfs_subr.c revision 1.97.2.2 1 /* $NetBSD: tmpfs_subr.c,v 1.97.2.2 2015/09/22 12:06:06 skrll 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.97.2.2 2015/09/22 12:06:06 skrll 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