kernel - Implement KVABIO API in TMPFS
[dragonfly.git] / sys / vfs / tmpfs / tmpfs_subr.c
CommitLineData
7a2de9a4
MD
1/* $NetBSD: tmpfs_subr.c,v 1.35 2007/07/09 21:10:50 ad Exp $ */
2
3/*-
4 * Copyright (c) 2005 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.
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 supporting functions.
35 */
7a2de9a4
MD
36
37#include <sys/kernel.h>
38#include <sys/param.h>
39#include <sys/namei.h>
40#include <sys/priv.h>
41#include <sys/proc.h>
7a2de9a4
MD
42#include <sys/stat.h>
43#include <sys/systm.h>
44#include <sys/vnode.h>
45#include <sys/vmmeter.h>
46
7a2de9a4
MD
47#include <vm/vm.h>
48#include <vm/vm_object.h>
49#include <vm/vm_page.h>
50#include <vm/vm_pager.h>
51#include <vm/vm_extern.h>
52
53#include <vfs/tmpfs/tmpfs.h>
7a2de9a4
MD
54#include <vfs/tmpfs/tmpfs_vnops.h>
55
f7db522f 56static ino_t tmpfs_fetch_ino(struct tmpfs_mount *);
f5f22af6 57
29ca4fd6
JH
58static int tmpfs_dirtree_compare(struct tmpfs_dirent *a,
59 struct tmpfs_dirent *b);
29ca4fd6
JH
60RB_GENERATE(tmpfs_dirtree, tmpfs_dirent, rb_node, tmpfs_dirtree_compare);
61
f5f22af6
MD
62static int tmpfs_dirtree_compare_cookie(struct tmpfs_dirent *a,
63 struct tmpfs_dirent *b);
64RB_GENERATE(tmpfs_dirtree_cookie, tmpfs_dirent,
65 rb_cookienode, tmpfs_dirtree_compare_cookie);
66
7a2de9a4
MD
67
68/* --------------------------------------------------------------------- */
69
70/*
71 * Allocates a new node of type 'type' inside the 'tmp' mount point, with
72 * its owner set to 'uid', its group to 'gid' and its mode set to 'mode',
73 * using the credentials of the process 'p'.
74 *
75 * If the node type is set to 'VDIR', then the parent parameter must point
76 * to the parent directory of the node being created. It may only be NULL
77 * while allocating the root node.
78 *
79 * If the node type is set to 'VBLK' or 'VCHR', then the rdev parameter
80 * specifies the device the node represents.
81 *
82 * If the node type is set to 'VLNK', then the parameter target specifies
83 * the file name of the target file for the symbolic link that is being
84 * created.
85 *
86 * Note that new nodes are retrieved from the available list if it has
87 * items or, if it is empty, from the node pool as long as there is enough
88 * space to create them.
89 *
90 * Returns zero on success or an appropriate error code on failure.
91 */
92int
93tmpfs_alloc_node(struct tmpfs_mount *tmp, enum vtype type,
6e0c5aab
MD
94 uid_t uid, gid_t gid, mode_t mode,
95 char *target, int rmajor, int rminor,
96 struct tmpfs_node **node)
7a2de9a4
MD
97{
98 struct tmpfs_node *nnode;
99 struct timespec ts;
100 udev_t rdev;
101
7a2de9a4
MD
102 KKASSERT(IFF(type == VLNK, target != NULL));
103 KKASSERT(IFF(type == VBLK || type == VCHR, rmajor != VNOVAL));
104
52ce5c29 105 if (tmp->tm_nodes_inuse >= tmp->tm_nodes_max)
7a2de9a4
MD
106 return (ENOSPC);
107
881dac8b
VS
108 nnode = objcache_get(tmp->tm_node_pool, M_WAITOK | M_NULLOK);
109 if (nnode == NULL)
110 return (ENOSPC);
7a2de9a4
MD
111
112 /* Generic initialization. */
113 nnode->tn_type = type;
114 vfs_timestamp(&ts);
115 nnode->tn_ctime = nnode->tn_mtime = nnode->tn_atime
116 = ts.tv_sec;
117 nnode->tn_ctimensec = nnode->tn_mtimensec = nnode->tn_atimensec
118 = ts.tv_nsec;
119 nnode->tn_uid = uid;
120 nnode->tn_gid = gid;
121 nnode->tn_mode = mode;
f7db522f 122 nnode->tn_id = tmpfs_fetch_ino(tmp);
12a5de0e 123 nnode->tn_advlock.init_done = 0;
6e0c5aab 124 KKASSERT(nnode->tn_links == 0);
7a2de9a4
MD
125
126 /* Type-specific initialization. */
127 switch (nnode->tn_type) {
128 case VBLK:
129 case VCHR:
130 rdev = makeudev(rmajor, rminor);
131 if (rdev == NOUDEV) {
42f6f6b1 132 objcache_put(tmp->tm_node_pool, nnode);
7a2de9a4
MD
133 return(EINVAL);
134 }
135 nnode->tn_rdev = rdev;
136 break;
137
138 case VDIR:
29ca4fd6 139 RB_INIT(&nnode->tn_dir.tn_dirtree);
f5f22af6 140 RB_INIT(&nnode->tn_dir.tn_cookietree);
7a2de9a4 141 nnode->tn_size = 0;
7a2de9a4
MD
142 break;
143
144 case VFIFO:
145 /* FALLTHROUGH */
146 case VSOCK:
147 break;
148
149 case VLNK:
9fc94b5f 150 nnode->tn_size = strlen(target);
d00cd01c 151 nnode->tn_link = kmalloc(nnode->tn_size + 1, tmp->tm_name_zone,
42f6f6b1
VS
152 M_WAITOK | M_NULLOK);
153 if (nnode->tn_link == NULL) {
154 objcache_put(tmp->tm_node_pool, nnode);
155 return (ENOSPC);
156 }
7a2de9a4
MD
157 bcopy(target, nnode->tn_link, nnode->tn_size);
158 nnode->tn_link[nnode->tn_size] = '\0';
159 break;
160
161 case VREG:
162 nnode->tn_reg.tn_aobj =
5a648714 163 swap_pager_alloc(NULL, 0, VM_PROT_DEFAULT, 0);
7a2de9a4
MD
164 nnode->tn_reg.tn_aobj_pages = 0;
165 nnode->tn_size = 0;
46b71cbe 166 vm_object_set_flag(nnode->tn_reg.tn_aobj, OBJ_NOPAGEIN);
7a2de9a4
MD
167 break;
168
169 default:
170 panic("tmpfs_alloc_node: type %p %d", nnode, (int)nnode->tn_type);
171 }
172
173 TMPFS_NODE_LOCK(nnode);
7a2de9a4 174 TMPFS_LOCK(tmp);
0786baf1 175 LIST_INSERT_HEAD(&tmp->tm_nodes_used, nnode, tn_entries);
7a2de9a4
MD
176 tmp->tm_nodes_inuse++;
177 TMPFS_UNLOCK(tmp);
178 TMPFS_NODE_UNLOCK(nnode);
179
180 *node = nnode;
181 return 0;
182}
183
184/* --------------------------------------------------------------------- */
185
186/*
187 * Destroys the node pointed to by node from the file system 'tmp'.
188 * If the node does not belong to the given mount point, the results are
189 * unpredicted.
190 *
191 * If the node references a directory; no entries are allowed because
192 * their removal could need a recursive algorithm, something forbidden in
193 * kernel space. Furthermore, there is not need to provide such
194 * functionality (recursive removal) because the only primitives offered
195 * to the user are the removal of empty directories and the deletion of
196 * individual files.
197 *
198 * Note that nodes are not really deleted; in fact, when a node has been
199 * allocated, it cannot be deleted during the whole life of the file
200 * system. Instead, they are moved to the available list and remain there
201 * until reused.
c5552356
TK
202 *
203 * A caller must have TMPFS_NODE_LOCK(node) and this function unlocks it.
7a2de9a4
MD
204 */
205void
206tmpfs_free_node(struct tmpfs_mount *tmp, struct tmpfs_node *node)
207{
29ffeb28 208 vm_pindex_t pages = 0;
7a2de9a4 209
7a2de9a4
MD
210#ifdef INVARIANTS
211 TMPFS_ASSERT_ELOCKED(node);
212 KKASSERT(node->tn_vnode == NULL);
213 KKASSERT((node->tn_vpstate & TMPFS_VNODE_ALLOCATING) == 0);
214#endif
215
7a2de9a4 216 TMPFS_LOCK(tmp);
0786baf1 217 LIST_REMOVE(node, tn_entries);
7a2de9a4
MD
218 tmp->tm_nodes_inuse--;
219 TMPFS_UNLOCK(tmp);
c5552356 220 TMPFS_NODE_UNLOCK(node); /* Caller has this lock */
7a2de9a4
MD
221
222 switch (node->tn_type) {
223 case VNON:
224 /* Do not do anything. VNON is provided to let the
225 * allocation routine clean itself easily by avoiding
226 * duplicating code in it. */
227 /* FALLTHROUGH */
228 case VBLK:
229 /* FALLTHROUGH */
230 case VCHR:
231 /* FALLTHROUGH */
232 break;
233 case VDIR:
0786baf1
MD
234 /*
235 * The parent link can be NULL if this is the root
6e0c5aab
MD
236 * node or if it is a directory node that was rmdir'd.
237 *
238 * XXX what if node is a directory which still contains
239 * directory entries (e.g. due to a forced umount) ?
0786baf1 240 */
7a2de9a4 241 node->tn_size = 0;
6e0c5aab 242 KKASSERT(node->tn_dir.tn_parent == NULL);
0786baf1
MD
243
244 /*
245 * If the root node is being destroyed don't leave a
246 * dangling pointer in tmpfs_mount.
247 */
248 if (node == tmp->tm_root)
249 tmp->tm_root = NULL;
7a2de9a4
MD
250 break;
251 case VFIFO:
252 /* FALLTHROUGH */
253 case VSOCK:
254 break;
255
256 case VLNK:
d00cd01c 257 kfree(node->tn_link, tmp->tm_name_zone);
9fc94b5f 258 node->tn_link = NULL;
2706b587 259 node->tn_size = 0;
7a2de9a4
MD
260 break;
261
262 case VREG:
263 if (node->tn_reg.tn_aobj != NULL)
f96f2f39 264 vm_object_deallocate(node->tn_reg.tn_aobj);
7a2de9a4
MD
265 node->tn_reg.tn_aobj = NULL;
266 pages = node->tn_reg.tn_aobj_pages;
267 break;
268
269 default:
270 panic("tmpfs_free_node: type %p %d", node, (int)node->tn_type);
271 }
272
0786baf1
MD
273 /*
274 * Clean up fields for the next allocation. The objcache only ctors
275 * new allocations.
276 */
277 tmpfs_node_ctor(node, NULL, 0);
7a2de9a4 278 objcache_put(tmp->tm_node_pool, node);
0786baf1 279 /* node is now invalid */
7a2de9a4 280
eae96dbb
MD
281 if (pages)
282 atomic_add_long(&tmp->tm_pages_used, -(long)pages);
7a2de9a4
MD
283}
284
285/* --------------------------------------------------------------------- */
286
287/*
288 * Allocates a new directory entry for the node node with a name of name.
289 * The new directory entry is returned in *de.
290 *
291 * The link count of node is increased by one to reflect the new object
292 * referencing it.
293 *
294 * Returns zero on success or an appropriate error code on failure.
295 */
296int
297tmpfs_alloc_dirent(struct tmpfs_mount *tmp, struct tmpfs_node *node,
aa1adbf0 298 const char *name, uint16_t len, struct tmpfs_dirent **de)
7a2de9a4
MD
299{
300 struct tmpfs_dirent *nde;
301
42f6f6b1 302 nde = objcache_get(tmp->tm_dirent_pool, M_WAITOK);
d00cd01c 303 nde->td_name = kmalloc(len + 1, tmp->tm_name_zone, M_WAITOK | M_NULLOK);
42f6f6b1
VS
304 if (nde->td_name == NULL) {
305 objcache_put(tmp->tm_dirent_pool, nde);
306 *de = NULL;
307 return (ENOSPC);
308 }
7a2de9a4
MD
309 nde->td_namelen = len;
310 bcopy(name, nde->td_name, len);
311 nde->td_name[len] = '\0';
312
313 nde->td_node = node;
314
315 TMPFS_NODE_LOCK(node);
ff837cd5 316 ++node->tn_links;
7a2de9a4
MD
317 TMPFS_NODE_UNLOCK(node);
318
319 *de = nde;
320
321 return 0;
322}
323
324/* --------------------------------------------------------------------- */
325
326/*
327 * Frees a directory entry. It is the caller's responsibility to destroy
328 * the node referenced by it if needed.
329 *
330 * The link count of node is decreased by one to reflect the removal of an
331 * object that referenced it. This only happens if 'node_exists' is true;
332 * otherwise the function will not access the node referred to by the
333 * directory entry, as it may already have been released from the outside.
334 */
335void
0786baf1 336tmpfs_free_dirent(struct tmpfs_mount *tmp, struct tmpfs_dirent *de)
7a2de9a4 337{
0786baf1 338 struct tmpfs_node *node;
7a2de9a4 339
0786baf1 340 node = de->td_node;
7a2de9a4 341
0786baf1
MD
342 TMPFS_NODE_LOCK(node);
343 TMPFS_ASSERT_ELOCKED(node);
344 KKASSERT(node->tn_links > 0);
ff837cd5 345 node->tn_links--;
0786baf1 346 TMPFS_NODE_UNLOCK(node);
7a2de9a4 347
d00cd01c 348 kfree(de->td_name, tmp->tm_name_zone);
0786baf1 349 de->td_namelen = 0;
9fc94b5f 350 de->td_name = NULL;
0786baf1 351 de->td_node = NULL;
7a2de9a4
MD
352 objcache_put(tmp->tm_dirent_pool, de);
353}
354
355/* --------------------------------------------------------------------- */
356
357/*
358 * Allocates a new vnode for the node node or returns a new reference to
359 * an existing one if the node had already a vnode referencing it. The
360 * resulting locked vnode is returned in *vpp.
361 *
362 * Returns zero on success or an appropriate error code on failure.
363 */
364int
365tmpfs_alloc_vp(struct mount *mp, struct tmpfs_node *node, int lkflag,
2706b587 366 struct vnode **vpp)
7a2de9a4
MD
367{
368 int error = 0;
369 struct vnode *vp;
370
371loop:
2706b587
MD
372 /*
373 * Interlocked extraction from node. This can race many things.
374 * We have to get a soft reference on the vnode while we hold
375 * the node locked, then acquire it properly and check for races.
376 */
7a2de9a4
MD
377 TMPFS_NODE_LOCK(node);
378 if ((vp = node->tn_vnode) != NULL) {
379 KKASSERT((node->tn_vpstate & TMPFS_VNODE_DOOMED) == 0);
ee173d09 380 vhold(vp);
7a2de9a4 381 TMPFS_NODE_UNLOCK(node);
7a2de9a4 382
2706b587
MD
383 if (vget(vp, lkflag | LK_EXCLUSIVE) != 0) {
384 vdrop(vp);
385 goto loop;
386 }
387 if (node->tn_vnode != vp) {
7a2de9a4 388 vput(vp);
2706b587 389 vdrop(vp);
7a2de9a4
MD
390 goto loop;
391 }
2706b587 392 vdrop(vp);
7a2de9a4
MD
393 goto out;
394 }
2706b587 395 /* vp is NULL */
7a2de9a4 396
0786baf1
MD
397 /*
398 * This should never happen.
399 */
400 if (node->tn_vpstate & TMPFS_VNODE_DOOMED) {
7a2de9a4
MD
401 TMPFS_NODE_UNLOCK(node);
402 error = ENOENT;
7a2de9a4
MD
403 goto out;
404 }
405
406 /*
2706b587
MD
407 * Interlock against other calls to tmpfs_alloc_vp() trying to
408 * allocate and assign a vp to node.
7a2de9a4
MD
409 */
410 if (node->tn_vpstate & TMPFS_VNODE_ALLOCATING) {
411 node->tn_vpstate |= TMPFS_VNODE_WANT;
2706b587
MD
412 error = tsleep(&node->tn_vpstate, PINTERLOCKED | PCATCH,
413 "tmpfs_alloc_vp", 0);
7a2de9a4
MD
414 TMPFS_NODE_UNLOCK(node);
415 if (error)
416 return error;
7a2de9a4 417 goto loop;
2706b587
MD
418 }
419 node->tn_vpstate |= TMPFS_VNODE_ALLOCATING;
7a2de9a4
MD
420 TMPFS_NODE_UNLOCK(node);
421
2706b587
MD
422 /*
423 * Allocate a new vnode (may block). The ALLOCATING flag should
424 * prevent a race against someone else assigning node->tn_vnode.
425 */
7a2de9a4
MD
426 error = getnewvnode(VT_TMPFS, mp, &vp, VLKTIMEOUT, LK_CANRECURSE);
427 if (error != 0)
428 goto unlock;
7a2de9a4 429
2706b587
MD
430 KKASSERT(node->tn_vnode == NULL);
431 KKASSERT(vp != NULL);
7a2de9a4
MD
432 vp->v_data = node;
433 vp->v_type = node->tn_type;
434
435 /* Type-specific initialization. */
436 switch (node->tn_type) {
437 case VBLK:
438 /* FALLTHROUGH */
439 case VCHR:
440 /* FALLTHROUGH */
441 case VSOCK:
442 break;
443 case VREG:
48db4e20
MD
444 /*
445 * VMIO is mandatory. Tmpfs also supports KVABIO
446 * for its tmpfs_strategy().
447 */
448 vsetflags(vp, VKVABIO);
24bdeb69 449 vinitvmio(vp, node->tn_size, TMPFS_BLKSIZE, -1);
7a2de9a4
MD
450 break;
451 case VLNK:
7a2de9a4
MD
452 break;
453 case VFIFO:
454 vp->v_ops = &mp->mnt_vn_fifo_ops;
455 break;
456 case VDIR:
7a2de9a4
MD
457 break;
458
459 default:
460 panic("tmpfs_alloc_vp: type %p %d", node, (int)node->tn_type);
461 }
462
7a2de9a4
MD
463
464unlock:
465 TMPFS_NODE_LOCK(node);
466
467 KKASSERT(node->tn_vpstate & TMPFS_VNODE_ALLOCATING);
468 node->tn_vpstate &= ~TMPFS_VNODE_ALLOCATING;
469 node->tn_vnode = vp;
470
471 if (node->tn_vpstate & TMPFS_VNODE_WANT) {
472 node->tn_vpstate &= ~TMPFS_VNODE_WANT;
473 TMPFS_NODE_UNLOCK(node);
2706b587
MD
474 wakeup(&node->tn_vpstate);
475 } else {
7a2de9a4 476 TMPFS_NODE_UNLOCK(node);
2706b587 477 }
7a2de9a4
MD
478
479out:
480 *vpp = vp;
481
482 KKASSERT(IFF(error == 0, *vpp != NULL && vn_islocked(*vpp)));
483#ifdef INVARIANTS
484 TMPFS_NODE_LOCK(node);
485 KKASSERT(*vpp == node->tn_vnode);
486 TMPFS_NODE_UNLOCK(node);
487#endif
488
489 return error;
490}
491
492/* --------------------------------------------------------------------- */
493
494/*
495 * Destroys the association between the vnode vp and the node it
496 * references.
497 */
498void
499tmpfs_free_vp(struct vnode *vp)
500{
501 struct tmpfs_node *node;
502
503 node = VP_TO_TMPFS_NODE(vp);
504
505 TMPFS_NODE_LOCK(node);
506 KKASSERT(lockcount(TMPFS_NODE_MTX(node)) > 0);
507 node->tn_vnode = NULL;
7a2de9a4 508 vp->v_data = NULL;
d86d27a8 509 TMPFS_NODE_UNLOCK(node);
7a2de9a4
MD
510}
511
512/* --------------------------------------------------------------------- */
513
514/*
515 * Allocates a new file of type 'type' and adds it to the parent directory
516 * 'dvp'; this addition is done using the component name given in 'cnp'.
517 * The ownership of the new file is automatically assigned based on the
518 * credentials of the caller (through 'cnp'), the group is set based on
519 * the parent directory and the mode is determined from the 'vap' argument.
520 * If successful, *vpp holds a vnode to the newly created file and zero
521 * is returned. Otherwise *vpp is NULL and the function returns an
522 * appropriate error code.
523 */
524int
525tmpfs_alloc_file(struct vnode *dvp, struct vnode **vpp, struct vattr *vap,
2706b587 526 struct namecache *ncp, struct ucred *cred, char *target)
7a2de9a4
MD
527{
528 int error;
529 struct tmpfs_dirent *de;
530 struct tmpfs_mount *tmp;
531 struct tmpfs_node *dnode;
532 struct tmpfs_node *node;
7a2de9a4
MD
533
534 tmp = VFS_TO_TMPFS(dvp->v_mount);
535 dnode = VP_TO_TMPFS_DIR(dvp);
536 *vpp = NULL;
537
6e0c5aab
MD
538 /*
539 * If the directory was removed but a process was CD'd into it,
540 * we do not allow any more file/dir creation within it. Otherwise
541 * we will lose track of it.
542 */
543 KKASSERT(dnode->tn_type == VDIR);
544 if (dnode != tmp->tm_root && dnode->tn_dir.tn_parent == NULL)
545 return ENOENT;
7a2de9a4 546
6e0c5aab
MD
547 /*
548 * Make sure the link count does not overflow.
549 */
550 if (vap->va_type == VDIR && dnode->tn_links >= LINK_MAX)
551 return EMLINK;
7a2de9a4
MD
552
553 /* Allocate a node that represents the new file. */
554 error = tmpfs_alloc_node(tmp, vap->va_type, cred->cr_uid,
6e0c5aab
MD
555 dnode->tn_gid, vap->va_mode, target,
556 vap->va_rmajor, vap->va_rminor, &node);
7a2de9a4
MD
557 if (error != 0)
558 return error;
0786baf1 559 TMPFS_NODE_LOCK(node);
7a2de9a4
MD
560
561 /* Allocate a directory entry that points to the new file. */
0786baf1 562 error = tmpfs_alloc_dirent(tmp, node, ncp->nc_name, ncp->nc_nlen, &de);
7a2de9a4
MD
563 if (error != 0) {
564 tmpfs_free_node(tmp, node);
0786baf1 565 /* eats node lock */
7a2de9a4
MD
566 return error;
567 }
568
569 /* Allocate a vnode for the new file. */
570 error = tmpfs_alloc_vp(dvp->v_mount, node, LK_EXCLUSIVE, vpp);
571 if (error != 0) {
0786baf1 572 tmpfs_free_dirent(tmp, de);
7a2de9a4 573 tmpfs_free_node(tmp, node);
0786baf1 574 /* eats node lock */
7a2de9a4
MD
575 return error;
576 }
577
6e0c5aab
MD
578 /*
579 * Now that all required items are allocated, we can proceed to
7a2de9a4 580 * insert the new node into the directory, an operation that
6e0c5aab
MD
581 * cannot fail.
582 */
22d3b394 583 tmpfs_dir_attach(dnode, de);
0786baf1 584 TMPFS_NODE_UNLOCK(node);
7a2de9a4
MD
585
586 return error;
587}
588
589/* --------------------------------------------------------------------- */
590
591/*
9855a822 592 * Attaches the directory entry de to the directory represented by dnode.
7a2de9a4
MD
593 * Note that this does not change the link count of the node pointed by
594 * the directory entry, as this is done by tmpfs_alloc_dirent.
595 */
596void
22d3b394 597tmpfs_dir_attach(struct tmpfs_node *dnode, struct tmpfs_dirent *de)
7a2de9a4 598{
6e0c5aab
MD
599 struct tmpfs_node *node = de->td_node;
600
22d3b394 601 TMPFS_NODE_LOCK(dnode);
6e0c5aab
MD
602 if (node && node->tn_type == VDIR) {
603 TMPFS_NODE_LOCK(node);
ff837cd5 604 ++node->tn_links;
6e0c5aab
MD
605 node->tn_status |= TMPFS_NODE_CHANGED;
606 node->tn_dir.tn_parent = dnode;
ff837cd5 607 ++dnode->tn_links;
6e0c5aab
MD
608 TMPFS_NODE_UNLOCK(node);
609 }
29ca4fd6 610 RB_INSERT(tmpfs_dirtree, &dnode->tn_dir.tn_dirtree, de);
f5f22af6 611 RB_INSERT(tmpfs_dirtree_cookie, &dnode->tn_dir.tn_cookietree, de);
7a2de9a4 612 dnode->tn_size += sizeof(struct tmpfs_dirent);
22d3b394
MD
613 dnode->tn_status |= TMPFS_NODE_ACCESSED | TMPFS_NODE_CHANGED |
614 TMPFS_NODE_MODIFIED;
7a2de9a4
MD
615 TMPFS_NODE_UNLOCK(dnode);
616}
617
618/* --------------------------------------------------------------------- */
619
620/*
9855a822 621 * Detaches the directory entry de from the directory represented by dnode.
7a2de9a4
MD
622 * Note that this does not change the link count of the node pointed by
623 * the directory entry, as this is done by tmpfs_free_dirent.
624 */
625void
22d3b394 626tmpfs_dir_detach(struct tmpfs_node *dnode, struct tmpfs_dirent *de)
7a2de9a4 627{
6e0c5aab
MD
628 struct tmpfs_node *node = de->td_node;
629
22d3b394 630 TMPFS_NODE_LOCK(dnode);
29ca4fd6 631 RB_REMOVE(tmpfs_dirtree, &dnode->tn_dir.tn_dirtree, de);
f5f22af6 632 RB_REMOVE(tmpfs_dirtree_cookie, &dnode->tn_dir.tn_cookietree, de);
7a2de9a4 633 dnode->tn_size -= sizeof(struct tmpfs_dirent);
22d3b394
MD
634 dnode->tn_status |= TMPFS_NODE_ACCESSED | TMPFS_NODE_CHANGED |
635 TMPFS_NODE_MODIFIED;
7a2de9a4 636 TMPFS_NODE_UNLOCK(dnode);
6e0c5aab
MD
637
638 /*
639 * Clean out the tn_parent pointer immediately when removing a
640 * directory.
641 *
642 * Removal of the parent linkage also cleans out the extra tn_links
643 * count we had on both node and dnode.
644 *
645 * node can be NULL (typ during a forced umount), in which case
646 * the mount code is dealing with the linkages from a linked list
647 * scan.
648 */
649 if (node && node->tn_type == VDIR && node->tn_dir.tn_parent) {
650 TMPFS_NODE_LOCK(dnode);
651 TMPFS_NODE_LOCK(node);
652 KKASSERT(node->tn_dir.tn_parent == dnode);
ff837cd5
MD
653 dnode->tn_links--;
654 node->tn_links--;
6e0c5aab
MD
655 node->tn_dir.tn_parent = NULL;
656 TMPFS_NODE_UNLOCK(node);
657 TMPFS_NODE_UNLOCK(dnode);
658 }
7a2de9a4
MD
659}
660
661/* --------------------------------------------------------------------- */
662
663/*
664 * Looks for a directory entry in the directory represented by node.
665 * 'ncp' describes the name of the entry to look for. Note that the .
666 * and .. components are not allowed as they do not physically exist
667 * within directories.
668 *
669 * Returns a pointer to the entry when found, otherwise NULL.
ff837cd5
MD
670 *
671 * Caller must hold the node locked (shared ok)
7a2de9a4
MD
672 */
673struct tmpfs_dirent *
674tmpfs_dir_lookup(struct tmpfs_node *node, struct tmpfs_node *f,
6e0c5aab 675 struct namecache *ncp)
7a2de9a4 676{
7a2de9a4
MD
677 struct tmpfs_dirent *de;
678 int len = ncp->nc_nlen;
29ca4fd6
JH
679 struct tmpfs_dirent wanted;
680
681 wanted.td_namelen = len;
682 wanted.td_name = ncp->nc_name;
7a2de9a4
MD
683
684 TMPFS_VALIDATE_DIR(node);
685
29ca4fd6
JH
686 de = RB_FIND(tmpfs_dirtree, &node->tn_dir.tn_dirtree, &wanted);
687
688 KKASSERT(f == NULL || f == de->td_node);
7a2de9a4 689
9fc94b5f 690 return de;
7a2de9a4
MD
691}
692
693/* --------------------------------------------------------------------- */
694
695/*
696 * Helper function for tmpfs_readdir. Creates a '.' entry for the given
697 * directory and returns it in the uio space. The function returns 0
698 * on success, -1 if there was not enough space in the uio structure to
699 * hold the directory entry or an appropriate error code if another
700 * error happens.
701 */
702int
703tmpfs_dir_getdotdent(struct tmpfs_node *node, struct uio *uio)
704{
705 int error;
7a2de9a4
MD
706
707 TMPFS_VALIDATE_DIR(node);
708 KKASSERT(uio->uio_offset == TMPFS_DIRCOOKIE_DOT);
709
e4f20828
TK
710 if (vop_write_dirent(&error, uio, node->tn_id, DT_DIR, 1, "."))
711 return -1;
712 if (error == 0)
713 uio->uio_offset = TMPFS_DIRCOOKIE_DOTDOT;
7a2de9a4
MD
714 return error;
715}
716
717/* --------------------------------------------------------------------- */
718
719/*
720 * Helper function for tmpfs_readdir. Creates a '..' entry for the given
721 * directory and returns it in the uio space. The function returns 0
722 * on success, -1 if there was not enough space in the uio structure to
723 * hold the directory entry or an appropriate error code if another
724 * error happens.
725 */
726int
22d3b394
MD
727tmpfs_dir_getdotdotdent(struct tmpfs_mount *tmp, struct tmpfs_node *node,
728 struct uio *uio)
7a2de9a4
MD
729{
730 int error;
e4f20828 731 ino_t d_ino;
7a2de9a4
MD
732
733 TMPFS_VALIDATE_DIR(node);
734 KKASSERT(uio->uio_offset == TMPFS_DIRCOOKIE_DOTDOT);
735
22d3b394 736 if (node->tn_dir.tn_parent) {
b08327b7
MD
737 TMPFS_NODE_LOCK(node);
738 if (node->tn_dir.tn_parent)
e4f20828 739 d_ino = node->tn_dir.tn_parent->tn_id;
b08327b7 740 else
e4f20828 741 d_ino = tmp->tm_root->tn_id;
b08327b7 742 TMPFS_NODE_UNLOCK(node);
22d3b394 743 } else {
e4f20828 744 d_ino = tmp->tm_root->tn_id;
7a2de9a4
MD
745 }
746
e4f20828
TK
747 if (vop_write_dirent(&error, uio, d_ino, DT_DIR, 2, ".."))
748 return -1;
749 if (error == 0) {
750 struct tmpfs_dirent *de;
751 de = RB_MIN(tmpfs_dirtree_cookie, &node->tn_dir.tn_cookietree);
752 if (de == NULL)
753 uio->uio_offset = TMPFS_DIRCOOKIE_EOF;
754 else
755 uio->uio_offset = tmpfs_dircookie(de);
7a2de9a4 756 }
7a2de9a4
MD
757 return error;
758}
759
760/* --------------------------------------------------------------------- */
761
762/*
763 * Lookup a directory entry by its associated cookie.
f5f22af6
MD
764 *
765 * Must be called with the directory node locked (shared ok)
7a2de9a4 766 */
f5f22af6
MD
767struct lubycookie_info {
768 off_t cookie;
7a2de9a4 769 struct tmpfs_dirent *de;
f5f22af6 770};
7a2de9a4 771
f5f22af6
MD
772static int
773lubycookie_cmp(struct tmpfs_dirent *de, void *arg)
774{
775 struct lubycookie_info *info = arg;
776 off_t cookie = tmpfs_dircookie(de);
777
778 if (cookie < info->cookie)
779 return(-1);
780 if (cookie > info->cookie)
781 return(1);
782 return(0);
783}
7a2de9a4 784
f5f22af6
MD
785static int
786lubycookie_callback(struct tmpfs_dirent *de, void *arg)
787{
788 struct lubycookie_info *info = arg;
789
790 if (tmpfs_dircookie(de) == info->cookie) {
791 info->de = de;
792 return(-1);
7a2de9a4 793 }
f5f22af6
MD
794 return(0);
795}
7a2de9a4 796
f5f22af6
MD
797struct tmpfs_dirent *
798tmpfs_dir_lookupbycookie(struct tmpfs_node *node, off_t cookie)
799{
800 struct lubycookie_info info;
801
802 info.cookie = cookie;
803 info.de = NULL;
804 RB_SCAN(tmpfs_dirtree_cookie, &node->tn_dir.tn_cookietree,
805 lubycookie_cmp, lubycookie_callback, &info);
806 return (info.de);
7a2de9a4
MD
807}
808
809/* --------------------------------------------------------------------- */
810
811/*
812 * Helper function for tmpfs_readdir. Returns as much directory entries
813 * as can fit in the uio space. The read starts at uio->uio_offset.
814 * The function returns 0 on success, -1 if there was not enough space
815 * in the uio structure to hold the directory entry or an appropriate
816 * error code if another error happens.
833d1222
MD
817 *
818 * Caller must hold the node locked (shared ok)
7a2de9a4
MD
819 */
820int
821tmpfs_dir_getdents(struct tmpfs_node *node, struct uio *uio, off_t *cntp)
822{
823 int error;
824 off_t startcookie;
825 struct tmpfs_dirent *de;
826
827 TMPFS_VALIDATE_DIR(node);
828
833d1222
MD
829 /*
830 * Locate the first directory entry we have to return. We have cached
7a2de9a4 831 * the last readdir in the node, so use those values if appropriate.
833d1222
MD
832 * Otherwise do a linear scan to find the requested entry.
833 */
7a2de9a4
MD
834 startcookie = uio->uio_offset;
835 KKASSERT(startcookie != TMPFS_DIRCOOKIE_DOT);
836 KKASSERT(startcookie != TMPFS_DIRCOOKIE_DOTDOT);
833d1222
MD
837
838 if (startcookie == TMPFS_DIRCOOKIE_EOF)
7a2de9a4 839 return 0;
833d1222
MD
840
841 de = tmpfs_dir_lookupbycookie(node, startcookie);
842 if (de == NULL)
7a2de9a4 843 return EINVAL;
7a2de9a4 844
f5f22af6
MD
845 /*
846 * Read as much entries as possible; i.e., until we reach the end of
847 * the directory or we exhaust uio space.
848 */
7a2de9a4 849 do {
e4f20828
TK
850 ino_t d_ino;
851 uint8_t d_type;
7a2de9a4
MD
852
853 /* Create a dirent structure representing the current
854 * tmpfs_node and fill it. */
e4f20828 855 d_ino = de->td_node->tn_id;
7a2de9a4
MD
856 switch (de->td_node->tn_type) {
857 case VBLK:
e4f20828 858 d_type = DT_BLK;
7a2de9a4
MD
859 break;
860
861 case VCHR:
e4f20828 862 d_type = DT_CHR;
7a2de9a4
MD
863 break;
864
865 case VDIR:
e4f20828 866 d_type = DT_DIR;
7a2de9a4
MD
867 break;
868
869 case VFIFO:
e4f20828 870 d_type = DT_FIFO;
7a2de9a4
MD
871 break;
872
873 case VLNK:
e4f20828 874 d_type = DT_LNK;
7a2de9a4
MD
875 break;
876
877 case VREG:
e4f20828 878 d_type = DT_REG;
7a2de9a4
MD
879 break;
880
881 case VSOCK:
e4f20828 882 d_type = DT_SOCK;
7a2de9a4
MD
883 break;
884
885 default:
886 panic("tmpfs_dir_getdents: type %p %d",
887 de->td_node, (int)de->td_node->tn_type);
888 }
e4f20828
TK
889 KKASSERT(de->td_namelen < 256); /* 255 + 1 */
890
891 if (vop_write_dirent(&error, uio, d_ino, d_type,
892 de->td_namelen, de->td_name)) {
7a2de9a4
MD
893 error = -1;
894 break;
895 }
896
7a2de9a4 897 (*cntp)++;
f5f22af6
MD
898 de = RB_NEXT(tmpfs_dirtree_cookie,
899 node->tn_dir.tn_cookietree, de);
7a2de9a4
MD
900 } while (error == 0 && uio->uio_resid > 0 && de != NULL);
901
902 /* Update the offset and cache. */
903 if (de == NULL) {
904 uio->uio_offset = TMPFS_DIRCOOKIE_EOF;
7a2de9a4 905 } else {
f5f22af6 906 uio->uio_offset = tmpfs_dircookie(de);
7a2de9a4 907 }
7a2de9a4
MD
908
909 return error;
910}
911
912/* --------------------------------------------------------------------- */
913
914/*
915 * Resizes the aobj associated to the regular file pointed to by vp to
916 * the size newsize. 'vp' must point to a vnode that represents a regular
917 * file. 'newsize' must be positive.
918 *
919 * pass trivial as 1 when buf content will be overwritten, otherwise set 0
920 * to be zero filled.
921 *
922 * Returns zero on success or an appropriate error code on failure.
b37a7c00
MD
923 *
924 * Caller must hold the node exclusively locked.
7a2de9a4
MD
925 */
926int
927tmpfs_reg_resize(struct vnode *vp, off_t newsize, int trivial)
928{
929 int error;
29ffeb28 930 vm_pindex_t newpages, oldpages;
7a2de9a4
MD
931 struct tmpfs_mount *tmp;
932 struct tmpfs_node *node;
933 off_t oldsize;
7a2de9a4
MD
934
935#ifdef INVARIANTS
936 KKASSERT(vp->v_type == VREG);
937 KKASSERT(newsize >= 0);
938#endif
939
940 node = VP_TO_TMPFS_NODE(vp);
941 tmp = VFS_TO_TMPFS(vp->v_mount);
942
ff837cd5
MD
943 /*
944 * Convert the old and new sizes to the number of pages needed to
7a2de9a4
MD
945 * store them. It may happen that we do not need to do anything
946 * because the last allocated page can accommodate the change on
ff837cd5
MD
947 * its own.
948 */
7a2de9a4 949 oldsize = node->tn_size;
29ffeb28 950 oldpages = round_page64(oldsize) / PAGE_SIZE;
7a2de9a4 951 KKASSERT(oldpages == node->tn_reg.tn_aobj_pages);
29ffeb28 952 newpages = round_page64(newsize) / PAGE_SIZE;
7a2de9a4
MD
953
954 if (newpages > oldpages &&
29ffeb28 955 tmp->tm_pages_used + newpages - oldpages > tmp->tm_pages_max) {
7a2de9a4
MD
956 error = ENOSPC;
957 goto out;
958 }
ff837cd5
MD
959 node->tn_reg.tn_aobj_pages = newpages;
960 node->tn_size = newsize;
7a2de9a4 961
eae96dbb
MD
962 if (newpages != oldpages)
963 atomic_add_long(&tmp->tm_pages_used, (newpages - oldpages));
7a2de9a4 964
9fc94b5f
MD
965 /*
966 * When adjusting the vnode filesize and its VM object we must
967 * also adjust our backing VM object (aobj). The blocksize
968 * used must match the block sized we use for the buffer cache.
22d3b394
MD
969 *
970 * The backing VM object contains no VM pages, only swap
971 * assignments.
9fc94b5f
MD
972 */
973 if (newsize < oldsize) {
974 vm_pindex_t osize;
22d3b394 975 vm_pindex_t nsize;
9fc94b5f
MD
976 vm_object_t aobj;
977
3c54bb74 978 error = nvtruncbuf(vp, newsize, TMPFS_BLKSIZE, -1, 0);
9fc94b5f
MD
979 aobj = node->tn_reg.tn_aobj;
980 if (aobj) {
981 osize = aobj->size;
22d3b394
MD
982 nsize = vp->v_object->size;
983 if (nsize < osize) {
9fc94b5f 984 aobj->size = osize;
22d3b394
MD
985 swap_pager_freespace(aobj, nsize,
986 osize - nsize);
9fc94b5f
MD
987 }
988 }
989 } else {
990 vm_object_t aobj;
991
3c54bb74
MD
992 error = nvextendbuf(vp, oldsize, newsize,
993 TMPFS_BLKSIZE, TMPFS_BLKSIZE,
7a2de9a4 994 -1, -1, trivial);
9fc94b5f
MD
995 aobj = node->tn_reg.tn_aobj;
996 if (aobj)
997 aobj->size = vp->v_object->size;
998 }
7a2de9a4
MD
999
1000out:
1001 return error;
1002}
1003
1004/* --------------------------------------------------------------------- */
1005
1006/*
1007 * Change flags of the given vnode.
1008 * Caller should execute tmpfs_update on vp after a successful execution.
1009 * The vnode must be locked on entry and remain locked on exit.
1010 */
1011int
80ae59d7 1012tmpfs_chflags(struct vnode *vp, int vaflags, struct ucred *cred)
7a2de9a4
MD
1013{
1014 int error;
1015 struct tmpfs_node *node;
80ae59d7 1016 int flags;
7a2de9a4
MD
1017
1018 KKASSERT(vn_islocked(vp));
1019
1020 node = VP_TO_TMPFS_NODE(vp);
80ae59d7 1021 flags = node->tn_flags;
7a2de9a4
MD
1022
1023 /* Disallow this operation if the file system is mounted read-only. */
1024 if (vp->v_mount->mnt_flag & MNT_RDONLY)
1025 return EROFS;
80ae59d7 1026 error = vop_helper_setattr_flags(&flags, vaflags, node->tn_uid, cred);
7a2de9a4 1027
5339dfe4 1028 /* Actually change the flags on the node itself */
80ae59d7
MD
1029 if (error == 0) {
1030 TMPFS_NODE_LOCK(node);
5339dfe4 1031 node->tn_flags = flags;
80ae59d7
MD
1032 node->tn_status |= TMPFS_NODE_CHANGED;
1033 TMPFS_NODE_UNLOCK(node);
7a2de9a4 1034 }
7a2de9a4
MD
1035
1036 KKASSERT(vn_islocked(vp));
1037
80ae59d7 1038 return error;
7a2de9a4
MD
1039}
1040
1041/* --------------------------------------------------------------------- */
1042
1043/*
1044 * Change access mode on the given vnode.
1045 * Caller should execute tmpfs_update on vp after a successful execution.
1046 * The vnode must be locked on entry and remain locked on exit.
1047 */
1048int
80ae59d7 1049tmpfs_chmod(struct vnode *vp, mode_t vamode, struct ucred *cred)
7a2de9a4 1050{
7a2de9a4 1051 struct tmpfs_node *node;
80ae59d7
MD
1052 mode_t cur_mode;
1053 int error;
7a2de9a4
MD
1054
1055 KKASSERT(vn_islocked(vp));
1056
1057 node = VP_TO_TMPFS_NODE(vp);
1058
1059 /* Disallow this operation if the file system is mounted read-only. */
1060 if (vp->v_mount->mnt_flag & MNT_RDONLY)
1061 return EROFS;
1062
1063 /* Immutable or append-only files cannot be modified, either. */
1064 if (node->tn_flags & (IMMUTABLE | APPEND))
1065 return EPERM;
1066
80ae59d7
MD
1067 cur_mode = node->tn_mode;
1068 error = vop_helper_chmod(vp, vamode, cred, node->tn_uid, node->tn_gid,
1069 &cur_mode);
7a2de9a4 1070
80ae59d7
MD
1071 if (error == 0 &&
1072 (node->tn_mode & ALLPERMS) != (cur_mode & ALLPERMS)) {
1073 TMPFS_NODE_LOCK(node);
1074 node->tn_mode &= ~ALLPERMS;
1075 node->tn_mode |= cur_mode & ALLPERMS;
7a2de9a4 1076
80ae59d7
MD
1077 node->tn_status |= TMPFS_NODE_CHANGED;
1078 TMPFS_NODE_UNLOCK(node);
7a2de9a4
MD
1079 }
1080
7a2de9a4
MD
1081 KKASSERT(vn_islocked(vp));
1082
1083 return 0;
1084}
1085
1086/* --------------------------------------------------------------------- */
1087
1088/*
1089 * Change ownership of the given vnode. At least one of uid or gid must
1090 * be different than VNOVAL. If one is set to that value, the attribute
1091 * is unchanged.
1092 * Caller should execute tmpfs_update on vp after a successful execution.
1093 * The vnode must be locked on entry and remain locked on exit.
1094 */
1095int
1096tmpfs_chown(struct vnode *vp, uid_t uid, gid_t gid, struct ucred *cred)
1097{
bd48f29c
MD
1098 mode_t cur_mode;
1099 uid_t cur_uid;
1100 gid_t cur_gid;
7a2de9a4 1101 struct tmpfs_node *node;
bd48f29c 1102 int error;
7a2de9a4
MD
1103
1104 KKASSERT(vn_islocked(vp));
7a2de9a4
MD
1105 node = VP_TO_TMPFS_NODE(vp);
1106
7a2de9a4
MD
1107 /* Disallow this operation if the file system is mounted read-only. */
1108 if (vp->v_mount->mnt_flag & MNT_RDONLY)
1109 return EROFS;
1110
1111 /* Immutable or append-only files cannot be modified, either. */
1112 if (node->tn_flags & (IMMUTABLE | APPEND))
1113 return EPERM;
1114
bd48f29c
MD
1115 cur_uid = node->tn_uid;
1116 cur_gid = node->tn_gid;
1117 cur_mode = node->tn_mode;
1118 error = vop_helper_chown(vp, uid, gid, cred,
1119 &cur_uid, &cur_gid, &cur_mode);
1120
1121 if (error == 0) {
1122 TMPFS_NODE_LOCK(node);
1123 if (cur_uid != node->tn_uid ||
1124 cur_gid != node->tn_gid ||
1125 cur_mode != node->tn_mode) {
3b9337bb
YT
1126 node->tn_uid = cur_uid;
1127 node->tn_gid = cur_gid;
bd48f29c
MD
1128 node->tn_mode = cur_mode;
1129 node->tn_status |= TMPFS_NODE_CHANGED;
7a2de9a4 1130 }
bd48f29c 1131 TMPFS_NODE_UNLOCK(node);
7a2de9a4 1132 }
7a2de9a4 1133
bd48f29c 1134 return error;
7a2de9a4
MD
1135}
1136
1137/* --------------------------------------------------------------------- */
1138
1139/*
1140 * Change size of the given vnode.
1141 * Caller should execute tmpfs_update on vp after a successful execution.
1142 * The vnode must be locked on entry and remain locked on exit.
1143 */
1144int
1145tmpfs_chsize(struct vnode *vp, u_quad_t size, struct ucred *cred)
1146{
1147 int error;
1148 struct tmpfs_node *node;
1149
1150 KKASSERT(vn_islocked(vp));
1151
1152 node = VP_TO_TMPFS_NODE(vp);
1153
1154 /* Decide whether this is a valid operation based on the file type. */
1155 error = 0;
1156 switch (vp->v_type) {
1157 case VDIR:
1158 return EISDIR;
1159
1160 case VREG:
1161 if (vp->v_mount->mnt_flag & MNT_RDONLY)
1162 return EROFS;
1163 break;
1164
1165 case VBLK:
1166 /* FALLTHROUGH */
1167 case VCHR:
1168 /* FALLTHROUGH */
1169 case VFIFO:
1170 /* Allow modifications of special files even if in the file
1171 * system is mounted read-only (we are not modifying the
1172 * files themselves, but the objects they represent). */
1173 return 0;
1174
1175 default:
1176 /* Anything else is unsupported. */
1177 return EOPNOTSUPP;
1178 }
1179
1180 /* Immutable or append-only files cannot be modified, either. */
1181 if (node->tn_flags & (IMMUTABLE | APPEND))
1182 return EPERM;
1183
1184 error = tmpfs_truncate(vp, size);
1185 /* tmpfs_truncate will raise the NOTE_EXTEND and NOTE_ATTRIB kevents
1186 * for us, as will update tn_status; no need to do that here. */
1187
1188 KKASSERT(vn_islocked(vp));
1189
1190 return error;
1191}
1192
1193/* --------------------------------------------------------------------- */
1194
1195/*
1196 * Change access and modification times of the given vnode.
1197 * Caller should execute tmpfs_update on vp after a successful execution.
1198 * The vnode must be locked on entry and remain locked on exit.
1199 */
1200int
1201tmpfs_chtimes(struct vnode *vp, struct timespec *atime, struct timespec *mtime,
aa1adbf0 1202 int vaflags, struct ucred *cred)
7a2de9a4 1203{
7a2de9a4 1204 struct tmpfs_node *node;
7a2de9a4
MD
1205
1206 KKASSERT(vn_islocked(vp));
1207
1208 node = VP_TO_TMPFS_NODE(vp);
1209
1210 /* Disallow this operation if the file system is mounted read-only. */
1211 if (vp->v_mount->mnt_flag & MNT_RDONLY)
1212 return EROFS;
1213
1214 /* Immutable or append-only files cannot be modified, either. */
1215 if (node->tn_flags & (IMMUTABLE | APPEND))
1216 return EPERM;
1217
7a2de9a4
MD
1218 TMPFS_NODE_LOCK(node);
1219 if (atime->tv_sec != VNOVAL && atime->tv_nsec != VNOVAL)
1220 node->tn_status |= TMPFS_NODE_ACCESSED;
1221
1222 if (mtime->tv_sec != VNOVAL && mtime->tv_nsec != VNOVAL)
1223 node->tn_status |= TMPFS_NODE_MODIFIED;
1224
1225 TMPFS_NODE_UNLOCK(node);
1226
1227 tmpfs_itimes(vp, atime, mtime);
1228
1229 KKASSERT(vn_islocked(vp));
1230
1231 return 0;
1232}
1233
1234/* --------------------------------------------------------------------- */
1235/* Sync timestamps */
1236void
1237tmpfs_itimes(struct vnode *vp, const struct timespec *acc,
aa1adbf0 1238 const struct timespec *mod)
7a2de9a4
MD
1239{
1240 struct tmpfs_node *node;
1241 struct timespec now;
1242
1243 node = VP_TO_TMPFS_NODE(vp);
1244
1245 if ((node->tn_status & (TMPFS_NODE_ACCESSED | TMPFS_NODE_MODIFIED |
1246 TMPFS_NODE_CHANGED)) == 0)
1247 return;
1248
1249 vfs_timestamp(&now);
1250
1251 TMPFS_NODE_LOCK(node);
1252 if (node->tn_status & TMPFS_NODE_ACCESSED) {
1253 if (acc == NULL)
1254 acc = &now;
1255 node->tn_atime = acc->tv_sec;
1256 node->tn_atimensec = acc->tv_nsec;
1257 }
1258 if (node->tn_status & TMPFS_NODE_MODIFIED) {
1259 if (mod == NULL)
1260 mod = &now;
1261 node->tn_mtime = mod->tv_sec;
1262 node->tn_mtimensec = mod->tv_nsec;
1263 }
1264 if (node->tn_status & TMPFS_NODE_CHANGED) {
1265 node->tn_ctime = now.tv_sec;
1266 node->tn_ctimensec = now.tv_nsec;
1267 }
1268 node->tn_status &=
1269 ~(TMPFS_NODE_ACCESSED | TMPFS_NODE_MODIFIED | TMPFS_NODE_CHANGED);
1270 TMPFS_NODE_UNLOCK(node);
1271}
1272
1273/* --------------------------------------------------------------------- */
1274
1275void
1276tmpfs_update(struct vnode *vp)
1277{
7a2de9a4
MD
1278 tmpfs_itimes(vp, NULL, NULL);
1279}
1280
1281/* --------------------------------------------------------------------- */
1282
b37a7c00
MD
1283/*
1284 * Caller must hold an exclusive node lock.
1285 */
7a2de9a4
MD
1286int
1287tmpfs_truncate(struct vnode *vp, off_t length)
1288{
1289 int error;
1290 struct tmpfs_node *node;
1291
1292 node = VP_TO_TMPFS_NODE(vp);
1293
1294 if (length < 0) {
1295 error = EINVAL;
1296 goto out;
1297 }
1298
1299 if (node->tn_size == length) {
1300 error = 0;
1301 goto out;
1302 }
1303
1304 if (length > VFS_TO_TMPFS(vp->v_mount)->tm_maxfilesize)
1305 return (EFBIG);
1306
1307
1308 error = tmpfs_reg_resize(vp, length, 1);
1309
b37a7c00 1310 if (error == 0)
7a2de9a4 1311 node->tn_status |= TMPFS_NODE_CHANGED | TMPFS_NODE_MODIFIED;
7a2de9a4
MD
1312
1313out:
1314 tmpfs_update(vp);
1315
1316 return error;
1317}
1318
1319/* --------------------------------------------------------------------- */
1320
1321static ino_t
f7db522f 1322tmpfs_fetch_ino(struct tmpfs_mount *tmp)
7a2de9a4 1323{
f7db522f 1324 ino_t ret;
7a2de9a4 1325
23db3379 1326 TMPFS_LOCK(tmp);
f7db522f 1327 ret = tmp->tm_ino++;
23db3379 1328 TMPFS_UNLOCK(tmp);
7a2de9a4 1329
f7db522f 1330 return (ret);
7a2de9a4 1331}
29ca4fd6
JH
1332
1333static int
1334tmpfs_dirtree_compare(struct tmpfs_dirent *a, struct tmpfs_dirent *b)
1335{
1336 if (a->td_namelen > b->td_namelen)
1337 return 1;
1338 else if (a->td_namelen < b->td_namelen)
1339 return -1;
1340 else
1341 return strncmp(a->td_name, b->td_name, a->td_namelen);
1342}
f5f22af6
MD
1343
1344static int
1345tmpfs_dirtree_compare_cookie(struct tmpfs_dirent *a, struct tmpfs_dirent *b)
1346{
1347 if (a < b)
1348 return(-1);
1349 if (a > b)
1350 return(1);
1351 return 0;
1352}