kernel - Rewrite vnode ref-counting code to improve performance
[dragonfly.git] / sys / vfs / tmpfs / tmpfs_vfsops.c
1 /*      $NetBSD: tmpfs_vfsops.c,v 1.10 2005/12/11 12:24:29 christos 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.
35  *
36  * tmpfs is a file system that uses virtual memory to store file data and
37  * metadata efficiently. It does not follow the structure of an on-disk
38  * file system because it simply does not need to. Instead, it uses
39  * memory-specific data structures and algorithms to automatically
40  * allocate and release resources.
41  */
42
43 #include <sys/conf.h>
44 #include <sys/param.h>
45 #include <sys/limits.h>
46 #include <sys/lock.h>
47 #include <sys/mutex.h>
48 #include <sys/kernel.h>
49 #include <sys/stat.h>
50 #include <sys/systm.h>
51 #include <sys/sysctl.h>
52 #include <sys/objcache.h>
53
54 #include <vm/vm.h>
55 #include <vm/vm_object.h>
56 #include <vm/vm_param.h>
57
58 #if 0
59 #include <vfs/tmpfs/tmpfs.h>
60 #endif
61 #include "tmpfs.h"
62 #include <vfs/tmpfs/tmpfs_vnops.h>
63 #include <vfs/tmpfs/tmpfs_args.h>
64
65 /*
66  * Default permission for root node
67  */
68 #define TMPFS_DEFAULT_ROOT_MODE (S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH)
69
70 MALLOC_DEFINE(M_TMPFSMNT, "tmpfs mount", "tmpfs mount structures");
71
72 /* --------------------------------------------------------------------- */
73
74 static int      tmpfs_mount(struct mount *, char *, caddr_t, struct ucred *);
75 static int      tmpfs_unmount(struct mount *, int);
76 static int      tmpfs_root(struct mount *, struct vnode **);
77 static int      tmpfs_fhtovp(struct mount *, struct vnode *, struct fid *, struct vnode **);
78 static int      tmpfs_statfs(struct mount *, struct statfs *, struct ucred *cred);
79
80 /* --------------------------------------------------------------------- */
81 int
82 tmpfs_node_ctor(void *obj, void *privdata, int flags)
83 {
84         struct tmpfs_node *node = obj;
85
86         node->tn_gen++;
87         node->tn_size = 0;
88         node->tn_status = 0;
89         node->tn_flags = 0;
90         node->tn_links = 0;
91         node->tn_vnode = NULL;
92         node->tn_vpstate = TMPFS_VNODE_WANT;
93         bzero(&node->tn_spec, sizeof(node->tn_spec));
94
95         return (1);
96 }
97
98 static void
99 tmpfs_node_dtor(void *obj, void *privdata)
100 {
101         struct tmpfs_node *node = (struct tmpfs_node *)obj;
102         node->tn_type = VNON;
103         node->tn_vpstate = TMPFS_VNODE_DOOMED;
104 }
105
106 static void *
107 tmpfs_node_init(void *args, int flags)
108 {
109         struct tmpfs_node *node = objcache_malloc_alloc(args, flags);
110         if (node == NULL)
111                 return (NULL);
112         node->tn_id = 0;
113
114         lockinit(&node->tn_interlock, "tmpfs node interlock", 0, LK_CANRECURSE);
115         node->tn_gen = karc4random();
116
117         return node;
118 }
119
120 static void
121 tmpfs_node_fini(void *obj, void *args)
122 {
123         struct tmpfs_node *node = (struct tmpfs_node *)obj;
124         lockuninit(&node->tn_interlock);
125         objcache_malloc_free(obj, args);
126 }
127
128 static int
129 tmpfs_mount(struct mount *mp, char *path, caddr_t data, struct ucred *cred)
130 {
131         struct tmpfs_mount *tmp;
132         struct tmpfs_node *root;
133         struct tmpfs_args args;
134         vm_pindex_t pages;
135         vm_pindex_t pages_limit;
136         ino_t nodes;
137         u_int64_t       maxfsize;
138         int error;
139         /* Size counters. */
140         ino_t   nodes_max;
141         off_t   size_max;
142         size_t  maxfsize_max;
143         size_t  size;
144
145         /* Root node attributes. */
146         uid_t   root_uid = cred->cr_uid;
147         gid_t   root_gid = cred->cr_gid;
148         mode_t  root_mode = (VREAD | VWRITE);
149
150         if (mp->mnt_flag & MNT_UPDATE) {
151                 /* XXX: There is no support yet to update file system
152                  * settings.  Should be added. */
153
154                 return EOPNOTSUPP;
155         }
156
157         /*
158          * mount info
159          */
160         bzero(&args, sizeof(args));
161         size_max  = 0;
162         nodes_max = 0;
163         maxfsize_max = 0;
164
165         if (path) {
166                 if (data) {
167                         error = copyin(data, &args, sizeof(args));
168                         if (error)
169                                 return (error);
170                 }
171                 size_max = args.ta_size_max;
172                 nodes_max = args.ta_nodes_max;
173                 maxfsize_max = args.ta_maxfsize_max;
174                 root_uid = args.ta_root_uid;
175                 root_gid = args.ta_root_gid;
176                 root_mode = args.ta_root_mode;
177         }
178
179         /*
180          * If mount by non-root, then verify that user has necessary
181          * permissions on the device.
182          */
183         if (cred->cr_uid != 0) {
184                 root_mode = VREAD;
185                 if ((mp->mnt_flag & MNT_RDONLY) == 0)
186                         root_mode |= VWRITE;
187         }
188
189         pages_limit = vm_swap_max + vmstats.v_page_count / 2;
190
191         if (size_max == 0) {
192                 pages = pages_limit / 2;
193         } else if (size_max < PAGE_SIZE) {
194                 pages = 1;
195         } else if (OFF_TO_IDX(size_max) > pages_limit) {
196                 /*
197                  * do not force pages = pages_limit for this case, otherwise
198                  * we might not honor tmpfs size requests from /etc/fstab
199                  * during boot because they are mounted prior to swap being
200                  * turned on.
201                  */
202                 pages = OFF_TO_IDX(size_max);
203         } else {
204                 pages = OFF_TO_IDX(size_max);
205         }
206
207         if (nodes_max == 0)
208                 nodes = 3 + pages * PAGE_SIZE / 1024;
209         else if (nodes_max < 3)
210                 nodes = 3;
211         else if (nodes_max > pages)
212                 nodes = pages;
213         else
214                 nodes = nodes_max;
215
216         maxfsize = IDX_TO_OFF(pages_limit);
217         if (maxfsize_max != 0 && maxfsize > maxfsize_max)
218                 maxfsize = maxfsize_max;
219
220         /* Allocate the tmpfs mount structure and fill it. */
221         tmp = kmalloc(sizeof(*tmp), M_TMPFSMNT, M_WAITOK | M_ZERO);
222
223         tmp->tm_mount = mp;
224         tmp->tm_nodes_max = nodes;
225         tmp->tm_nodes_inuse = 0;
226         tmp->tm_maxfilesize = maxfsize;
227         LIST_INIT(&tmp->tm_nodes_used);
228
229         tmp->tm_pages_max = pages;
230         tmp->tm_pages_used = 0;
231
232         kmalloc_create(&tmp->tm_node_zone, "tmpfs node");
233         kmalloc_create(&tmp->tm_dirent_zone, "tmpfs dirent");
234         kmalloc_create(&tmp->tm_name_zone, "tmpfs name zone");
235
236         kmalloc_raise_limit(tmp->tm_node_zone, sizeof(struct tmpfs_node) *
237                             tmp->tm_nodes_max);
238
239         tmp->tm_node_zone_malloc_args.objsize = sizeof(struct tmpfs_node);
240         tmp->tm_node_zone_malloc_args.mtype = tmp->tm_node_zone;
241
242         tmp->tm_dirent_zone_malloc_args.objsize = sizeof(struct tmpfs_dirent);
243         tmp->tm_dirent_zone_malloc_args.mtype = tmp->tm_dirent_zone;
244
245         tmp->tm_dirent_pool =  objcache_create( "tmpfs dirent cache",
246             0, 0,
247             NULL, NULL, NULL,
248             objcache_malloc_alloc, objcache_malloc_free,
249             &tmp->tm_dirent_zone_malloc_args);
250         tmp->tm_node_pool = objcache_create( "tmpfs node cache",
251             0, 0,
252             tmpfs_node_ctor, tmpfs_node_dtor, NULL,
253             tmpfs_node_init, tmpfs_node_fini,
254             &tmp->tm_node_zone_malloc_args);
255
256         tmp->tm_ino = 2;
257
258         /* Allocate the root node. */
259         error = tmpfs_alloc_node(tmp, VDIR, root_uid, root_gid,
260                                  root_mode & ALLPERMS, NULL,
261                                  VNOVAL, VNOVAL, &root);
262
263         /*
264          * We are backed by swap, set snocache chflags flag so we
265          * don't trip over swapcache.
266          */
267         root->tn_flags = SF_NOCACHE;
268
269         if (error != 0 || root == NULL) {
270             objcache_destroy(tmp->tm_node_pool);
271             objcache_destroy(tmp->tm_dirent_pool);
272             kfree(tmp, M_TMPFSMNT);
273             return error;
274         }
275         KASSERT(root->tn_id >= 0,
276                 ("tmpfs root with invalid ino: %d", (int)root->tn_id));
277
278         ++root->tn_links;       /* prevent destruction */
279         tmp->tm_root = root;
280
281         mp->mnt_flag |= MNT_LOCAL;
282         mp->mnt_kern_flag |= MNTK_ALL_MPSAFE;
283         mp->mnt_kern_flag |= MNTK_NOMSYNC;
284         mp->mnt_kern_flag |= MNTK_THR_SYNC;     /* new vsyncscan semantics */
285         mp->mnt_data = (qaddr_t)tmp;
286         vfs_getnewfsid(mp);
287
288         vfs_add_vnodeops(mp, &tmpfs_vnode_vops, &mp->mnt_vn_norm_ops);
289         vfs_add_vnodeops(mp, &tmpfs_fifo_vops, &mp->mnt_vn_fifo_ops);
290
291         copystr("tmpfs", mp->mnt_stat.f_mntfromname, MNAMELEN - 1, &size);
292         bzero(mp->mnt_stat.f_mntfromname +size, MNAMELEN - size);
293         bzero(mp->mnt_stat.f_mntonname, sizeof(mp->mnt_stat.f_mntonname));
294         copyinstr(path, mp->mnt_stat.f_mntonname,
295                   sizeof(mp->mnt_stat.f_mntonname) -1,
296                   &size);
297
298         tmpfs_statfs(mp, &mp->mnt_stat, cred);
299
300         return 0;
301 }
302
303 /* --------------------------------------------------------------------- */
304
305 /* ARGSUSED2 */
306 static int
307 tmpfs_unmount(struct mount *mp, int mntflags)
308 {
309         int error;
310         int flags = 0;
311         struct tmpfs_mount *tmp;
312         struct tmpfs_node *node;
313         struct vnode *vp;
314         int isok;
315
316         tmp = VFS_TO_TMPFS(mp);
317         TMPFS_LOCK(tmp);
318
319         /* Handle forced unmounts. */
320         if (mntflags & MNT_FORCE)
321                 flags |= FORCECLOSE;
322
323         /*
324          * Finalize all pending I/O.  In the case of tmpfs we want
325          * to throw all the data away so clean out the buffer cache
326          * and vm objects before calling vflush().
327          */
328         LIST_FOREACH(node, &tmp->tm_nodes_used, tn_entries) {
329                 /*
330                  * tn_links is mnt_token protected
331                  */
332                 ++node->tn_links;
333                 while (node->tn_type == VREG && node->tn_vnode) {
334                         vp = node->tn_vnode;
335                         vhold(vp);
336                         lwkt_yield();
337
338                         /*
339                          * vx_get/vx_put and tmpfs_truncate may block,
340                          * releasing the tmpfs mountpoint token.
341                          *
342                          * Make sure the lock order is correct.
343                          */
344                         vx_get(vp);             /* held vnode */
345                         TMPFS_NODE_LOCK(node);
346                         if (node->tn_vnode == vp) {
347                                 tmpfs_truncate(vp, 0);
348                                 isok = 1;
349                         } else {
350                                 isok = 0;
351                         }
352                         TMPFS_NODE_UNLOCK(node);
353                         vx_put(vp);
354                         vdrop(vp);
355                         if (isok)
356                                 break;
357                         /* retry */
358                 }
359                 --node->tn_links;
360         }
361
362         /*
363          * Flush all vnodes on the mount.
364          *
365          * If we fail to flush, we cannot unmount, but all the nodes have
366          * already been truncated. Erroring out is the best we can do.
367          */
368         error = vflush(mp, 0, flags);
369         if (error != 0) {
370                 TMPFS_UNLOCK(tmp);
371                 return (error);
372         }
373
374         /*
375          * First pass get rid of all the directory entries and
376          * vnode associations.  This will also destroy the
377          * directory topology and should drop all link counts
378          * to 0 except for the root.
379          *
380          * No vnodes should remain after the vflush above.
381          */
382         LIST_FOREACH(node, &tmp->tm_nodes_used, tn_entries) {
383                 ++node->tn_links;       /* mnt_token protected */
384                 lwkt_yield();
385                 TMPFS_NODE_LOCK(node);
386
387                 if (node->tn_type == VDIR) {
388                         struct tmpfs_dirent *de;
389
390                         while ((de = RB_ROOT(&node->tn_dir.tn_dirtree)) != NULL)                        {
391                                 tmpfs_dir_detach(node, de);
392                                 tmpfs_free_dirent(tmp, de);
393                         }
394                 }
395                 KKASSERT(node->tn_vnode == NULL);
396
397                 TMPFS_NODE_UNLOCK(node);
398                 --node->tn_links;       /* mnt_token protected */
399         }
400
401         /*
402          * Allow the root node to be destroyed by dropping the link count
403          * we bumped in the mount code.
404          */
405         KKASSERT(tmp->tm_root);
406         --tmp->tm_root->tn_links;       /* mnt_token protected */
407
408         /*
409          * At this point all nodes, including the root node, should have a
410          * link count of 0.  The root is not necessarily going to be last.
411          */
412         while ((node = LIST_FIRST(&tmp->tm_nodes_used)) != NULL) {
413                 if (node->tn_links)
414                         panic("tmpfs: Dangling nodes during umount (%p)!\n",
415                               node);
416
417                 TMPFS_NODE_LOCK(node);
418                 tmpfs_free_node(tmp, node);
419                 /* eats lock */
420                 lwkt_yield();
421         }
422         KKASSERT(tmp->tm_root == NULL);
423
424         objcache_destroy(tmp->tm_dirent_pool);
425         objcache_destroy(tmp->tm_node_pool);
426
427         kmalloc_destroy(&tmp->tm_name_zone);
428         kmalloc_destroy(&tmp->tm_dirent_zone);
429         kmalloc_destroy(&tmp->tm_node_zone);
430
431         tmp->tm_node_zone = tmp->tm_dirent_zone = NULL;
432
433         KKASSERT(tmp->tm_pages_used == 0);
434         KKASSERT(tmp->tm_nodes_inuse == 0);
435
436         TMPFS_UNLOCK(tmp);
437
438         /* Throw away the tmpfs_mount structure. */
439         kfree(tmp, M_TMPFSMNT);
440         mp->mnt_data = NULL;
441
442         mp->mnt_flag &= ~MNT_LOCAL;
443         return 0;
444 }
445
446 /* --------------------------------------------------------------------- */
447
448 static int
449 tmpfs_root(struct mount *mp, struct vnode **vpp)
450 {
451         struct tmpfs_mount *tmp;
452         int error;
453
454         tmp = VFS_TO_TMPFS(mp);
455         if (tmp->tm_root == NULL) {
456                 kprintf("tmpfs_root: called without root node %p\n", mp);
457                 print_backtrace(-1);
458                 *vpp = NULL;
459                 error = EINVAL;
460         } else {
461                 error = tmpfs_alloc_vp(mp, tmp->tm_root, LK_EXCLUSIVE, vpp);
462                 (*vpp)->v_flag |= VROOT;
463                 (*vpp)->v_type = VDIR;
464         }
465         return error;
466 }
467
468 /* --------------------------------------------------------------------- */
469
470 static int
471 tmpfs_fhtovp(struct mount *mp, struct vnode *rootvp, struct fid *fhp,
472              struct vnode **vpp)
473 {
474         boolean_t found;
475         struct tmpfs_fid *tfhp;
476         struct tmpfs_mount *tmp;
477         struct tmpfs_node *node;
478         int rc;
479
480         tmp = VFS_TO_TMPFS(mp);
481
482         tfhp = (struct tmpfs_fid *) fhp;
483         if (tfhp->tf_len != sizeof(struct tmpfs_fid))
484                 return EINVAL;
485
486         if (tfhp->tf_id >= tmp->tm_nodes_max)
487                 return EINVAL;
488
489         rc = EINVAL;
490         found = FALSE;
491
492         TMPFS_LOCK(tmp);
493         LIST_FOREACH(node, &tmp->tm_nodes_used, tn_entries) {
494                 if (node->tn_id == tfhp->tf_id &&
495                     node->tn_gen == tfhp->tf_gen) {
496                         found = TRUE;
497                         break;
498                 }
499         }
500
501         if (found)
502                 rc = tmpfs_alloc_vp(mp, node, LK_EXCLUSIVE, vpp);
503
504         TMPFS_UNLOCK(tmp);
505
506         return (rc);
507 }
508
509 /* --------------------------------------------------------------------- */
510
511 /* ARGSUSED2 */
512 static int
513 tmpfs_statfs(struct mount *mp, struct statfs *sbp, struct ucred *cred)
514 {
515         fsfilcnt_t freenodes;
516         struct tmpfs_mount *tmp;
517
518         tmp = VFS_TO_TMPFS(mp);
519
520         TMPFS_LOCK(tmp);
521         sbp->f_iosize = PAGE_SIZE;
522         sbp->f_bsize = PAGE_SIZE;
523
524         sbp->f_blocks = tmp->tm_pages_max;
525         sbp->f_bavail = tmp->tm_pages_max - tmp->tm_pages_used;
526         sbp->f_bfree = sbp->f_bavail;
527
528         freenodes = tmp->tm_nodes_max - tmp->tm_nodes_inuse;
529
530         sbp->f_files = freenodes + tmp->tm_nodes_inuse;
531         sbp->f_ffree = freenodes;
532         sbp->f_owner = tmp->tm_root->tn_uid;
533
534         TMPFS_UNLOCK(tmp);
535
536         return 0;
537 }
538
539 /* --------------------------------------------------------------------- */ 
540
541 static int
542 tmpfs_vptofh(struct vnode *vp, struct fid *fhp)
543 {
544         struct tmpfs_node *node;
545         struct tmpfs_fid tfh;
546         node = VP_TO_TMPFS_NODE(vp);
547         memset(&tfh, 0, sizeof(tfh));
548         tfh.tf_len = sizeof(struct tmpfs_fid);
549         tfh.tf_gen = node->tn_gen;
550         tfh.tf_id = node->tn_id;
551         memcpy(fhp, &tfh, sizeof(tfh));
552         return (0);
553 }
554
555 /* --------------------------------------------------------------------- */
556
557 static int 
558 tmpfs_checkexp(struct mount *mp, struct sockaddr *nam, int *exflagsp, 
559                struct ucred **credanonp) 
560
561         struct tmpfs_mount *tmp; 
562         struct netcred *nc; 
563  
564         tmp = (struct tmpfs_mount *) mp->mnt_data;
565         nc = vfs_export_lookup(mp, &tmp->tm_export, nam); 
566         if (nc == NULL) 
567                 return (EACCES); 
568
569         *exflagsp = nc->netc_exflags; 
570         *credanonp = &nc->netc_anon; 
571  
572         return (0); 
573
574
575 /* --------------------------------------------------------------------- */ 
576
577 /*
578  * tmpfs vfs operations.
579  */
580
581 static struct vfsops tmpfs_vfsops = {
582         .vfs_mount =                    tmpfs_mount,
583         .vfs_unmount =                  tmpfs_unmount,
584         .vfs_root =                     tmpfs_root,
585         .vfs_statfs =                   tmpfs_statfs,
586         .vfs_fhtovp =                   tmpfs_fhtovp,
587         .vfs_vptofh =                   tmpfs_vptofh, 
588         .vfs_sync =                     vfs_stdsync,
589         .vfs_checkexp =                 tmpfs_checkexp,
590 };
591
592 VFS_SET(tmpfs_vfsops, tmpfs, 0);
593 MODULE_VERSION(tmpfs, 1);