fs/nfsd/filecache.c

   1 /*
   2  * Open file cache.
   3  *
   4  * (c) 2015 - Jeff Layton <jeff.layton@primarydata.com>
   5  */
   6
   7 #include <linux/hash.h>
   8 #include <linux/slab.h>
   9 #include <linux/file.h>
  10 #include <linux/sched.h>
  11 #include <linux/list_lru.h>
  12 #include <linux/fsnotify_backend.h>
  13 #include <linux/fsnotify.h>
  14 #include <linux/seq_file.h>
  15
  16 #include "vfs.h"
  17 #include "nfsd.h"
  18 #include "nfsfh.h"
  19 #include "netns.h"
  20 #include "filecache.h"
  21 #include "trace.h"
  22
  23 #define NFSDDBG_FACILITY        NFSDDBG_FH
  24
  25 /* FIXME: dynamically size this for the machine somehow? */
  26 #define NFSD_FILE_HASH_BITS                   12
  27 #define NFSD_FILE_HASH_SIZE                  (1 << NFSD_FILE_HASH_BITS)
  28 #define NFSD_LAUNDRETTE_DELAY                (2 * HZ)
  29
  30 #define NFSD_FILE_SHUTDOWN                   (1)
  31 #define NFSD_FILE_LRU_THRESHOLD              (4096UL)
  32 #define NFSD_FILE_LRU_LIMIT                  (NFSD_FILE_LRU_THRESHOLD << 2)
  33
  34 /* We only care about NFSD_MAY_READ/WRITE for this cache */
  35 #define NFSD_FILE_MAY_MASK      (NFSD_MAY_READ|NFSD_MAY_WRITE)
  36
  37 struct nfsd_fcache_bucket {
  38         struct hlist_head       nfb_head;
  39         spinlock_t              nfb_lock;
  40         unsigned int            nfb_count;
  41         unsigned int            nfb_maxcount;
  42 };
  43
  44 static DEFINE_PER_CPU(unsigned long, nfsd_file_cache_hits);
  45
  46 struct nfsd_fcache_disposal {
  47         struct list_head list;
  48         struct work_struct work;
  49         struct net *net;
  50         spinlock_t lock;
  51         struct list_head freeme;
  52         struct rcu_head rcu;
  53 };
  54
  55 struct workqueue_struct *nfsd_filecache_wq __read_mostly;
  56
  57 static struct kmem_cache                *nfsd_file_slab;
  58 static struct kmem_cache                *nfsd_file_mark_slab;
  59 static struct nfsd_fcache_bucket        *nfsd_file_hashtbl;
  60 static struct list_lru                  nfsd_file_lru;
  61 static long                             nfsd_file_lru_flags;
  62 static struct fsnotify_group            *nfsd_file_fsnotify_group;
  63 static atomic_long_t                    nfsd_filecache_count;
  64 static struct delayed_work              nfsd_filecache_laundrette;
  65 static DEFINE_SPINLOCK(laundrette_lock);
  66 static LIST_HEAD(laundrettes);
  67
  68 static void nfsd_file_gc(void);
  69
  70 static void
  71 nfsd_file_schedule_laundrette(void)
  72 {
  73         long count = atomic_long_read(&nfsd_filecache_count);
  74
  75         if (count == 0 || test_bit(NFSD_FILE_SHUTDOWN, &nfsd_file_lru_flags))
  76                 return;
  77
  78         queue_delayed_work(system_wq, &nfsd_filecache_laundrette,
  79                         NFSD_LAUNDRETTE_DELAY);
  80 }
  81
  82 static void
  83 nfsd_file_slab_free(struct rcu_head *rcu)
  84 {
  85         struct nfsd_file *nf = container_of(rcu, struct nfsd_file, nf_rcu);
  86
  87         put_cred(nf->nf_cred);
  88         kmem_cache_free(nfsd_file_slab, nf);
  89 }
  90
  91 static void
  92 nfsd_file_mark_free(struct fsnotify_mark *mark)
  93 {
  94         struct nfsd_file_mark *nfm = container_of(mark, struct nfsd_file_mark,
  95                                                   nfm_mark);
  96
  97         kmem_cache_free(nfsd_file_mark_slab, nfm);
  98 }
  99
 100 static struct nfsd_file_mark *
 101 nfsd_file_mark_get(struct nfsd_file_mark *nfm)
 102 {
 103         if (!atomic_inc_not_zero(&nfm->nfm_ref))
 104                 return NULL;
 105         return nfm;
 106 }
 107
 108 static void
 109 nfsd_file_mark_put(struct nfsd_file_mark *nfm)
 110 {
 111         if (atomic_dec_and_test(&nfm->nfm_ref)) {
 112
 113                 fsnotify_destroy_mark(&nfm->nfm_mark, nfsd_file_fsnotify_group);
 114                 fsnotify_put_mark(&nfm->nfm_mark);
 115         }
 116 }
 117
 118 static struct nfsd_file_mark *
 119 nfsd_file_mark_find_or_create(struct nfsd_file *nf)
 120 {
 121         int                     err;
 122         struct fsnotify_mark    *mark;
 123         struct nfsd_file_mark   *nfm = NULL, *new;
 124         struct inode *inode = nf->nf_inode;
 125
 126         do {
 127                 mutex_lock(&nfsd_file_fsnotify_group->mark_mutex);
 128                 mark = fsnotify_find_mark(&inode->i_fsnotify_marks,
 129                                 nfsd_file_fsnotify_group);
 130                 if (mark) {
 131                         nfm = nfsd_file_mark_get(container_of(mark,
 132                                                  struct nfsd_file_mark,
 133                                                  nfm_mark));
 134                         mutex_unlock(&nfsd_file_fsnotify_group->mark_mutex);
 135                         fsnotify_put_mark(mark);
 136                         if (likely(nfm))
 137                                 break;
 138                 } else
 139                         mutex_unlock(&nfsd_file_fsnotify_group->mark_mutex);
 140
 141                 /* allocate a new nfm */
 142                 new = kmem_cache_alloc(nfsd_file_mark_slab, GFP_KERNEL);
 143                 if (!new)
 144                         return NULL;
 145                 fsnotify_init_mark(&new->nfm_mark, nfsd_file_fsnotify_group);
 146                 new->nfm_mark.mask = FS_ATTRIB|FS_DELETE_SELF;
 147                 atomic_set(&new->nfm_ref, 1);
 148
 149                 err = fsnotify_add_inode_mark(&new->nfm_mark, inode, 0);
 150
 151                 /*
 152                  * If the add was successful, then return the object.
 153                  * Otherwise, we need to put the reference we hold on the
 154                  * nfm_mark. The fsnotify code will take a reference and put
 155                  * it on failure, so we can't just free it directly. It's also
 156                  * not safe to call fsnotify_destroy_mark on it as the
 157                  * mark->group will be NULL. Thus, we can't let the nfm_ref
 158                  * counter drive the destruction at this point.
 159                  */
 160                 if (likely(!err))
 161                         nfm = new;
 162                 else
 163                         fsnotify_put_mark(&new->nfm_mark);
 164         } while (unlikely(err == -EEXIST));
 165
 166         return nfm;
 167 }
 168
 169 static struct nfsd_file *
 170 nfsd_file_alloc(struct inode *inode, unsigned int may, unsigned int hashval,
 171                 struct net *net)
 172 {
 173         struct nfsd_file *nf;
 174
 175         nf = kmem_cache_alloc(nfsd_file_slab, GFP_KERNEL);
 176         if (nf) {
 177                 INIT_HLIST_NODE(&nf->nf_node);
 178                 INIT_LIST_HEAD(&nf->nf_lru);
 179                 nf->nf_file = NULL;
 180                 nf->nf_cred = get_current_cred();
 181                 nf->nf_net = net;
 182                 nf->nf_flags = 0;
 183                 nf->nf_inode = inode;
 184                 nf->nf_hashval = hashval;
 185                 atomic_set(&nf->nf_ref, 1);
 186                 nf->nf_may = may & NFSD_FILE_MAY_MASK;
 187                 if (may & NFSD_MAY_NOT_BREAK_LEASE) {
 188                         if (may & NFSD_MAY_WRITE)
 189                                 __set_bit(NFSD_FILE_BREAK_WRITE, &nf->nf_flags);
 190                         if (may & NFSD_MAY_READ)
 191                                 __set_bit(NFSD_FILE_BREAK_READ, &nf->nf_flags);
 192                 }
 193                 nf->nf_mark = NULL;
 194                 trace_nfsd_file_alloc(nf);
 195         }
 196         return nf;
 197 }
 198
 199 static bool
 200 nfsd_file_free(struct nfsd_file *nf)
 201 {
 202         bool flush = false;
 203
 204         trace_nfsd_file_put_final(nf);
 205         if (nf->nf_mark)
 206                 nfsd_file_mark_put(nf->nf_mark);
 207         if (nf->nf_file) {
 208                 get_file(nf->nf_file);
 209                 filp_close(nf->nf_file, NULL);
 210                 fput(nf->nf_file);
 211                 flush = true;
 212         }
 213         call_rcu(&nf->nf_rcu, nfsd_file_slab_free);
 214         return flush;
 215 }
 216
 217 static bool
 218 nfsd_file_check_writeback(struct nfsd_file *nf)
 219 {
 220         struct file *file = nf->nf_file;
 221         struct address_space *mapping;
 222
 223         if (!file || !(file->f_mode & FMODE_WRITE))
 224                 return false;
 225         mapping = file->f_mapping;
 226         return mapping_tagged(mapping, PAGECACHE_TAG_DIRTY) ||
 227                 mapping_tagged(mapping, PAGECACHE_TAG_WRITEBACK);
 228 }
 229
 230 static int
 231 nfsd_file_check_write_error(struct nfsd_file *nf)
 232 {
 233         struct file *file = nf->nf_file;
 234
 235         if (!file || !(file->f_mode & FMODE_WRITE))
 236                 return 0;
 237         return filemap_check_wb_err(file->f_mapping, READ_ONCE(file->f_wb_err));
 238 }
 239
 240 static bool
 241 nfsd_file_in_use(struct nfsd_file *nf)
 242 {
 243         return nfsd_file_check_writeback(nf) ||
 244                         nfsd_file_check_write_error(nf);
 245 }
 246
 247 static void
 248 nfsd_file_do_unhash(struct nfsd_file *nf)
 249 {
 250         lockdep_assert_held(&nfsd_file_hashtbl[nf->nf_hashval].nfb_lock);
 251
 252         trace_nfsd_file_unhash(nf);
 253
 254         if (nfsd_file_check_write_error(nf))
 255                 nfsd_reset_boot_verifier(net_generic(nf->nf_net, nfsd_net_id));
 256         --nfsd_file_hashtbl[nf->nf_hashval].nfb_count;
 257         hlist_del_rcu(&nf->nf_node);
 258         atomic_long_dec(&nfsd_filecache_count);
 259 }
 260
 261 static bool
 262 nfsd_file_unhash(struct nfsd_file *nf)
 263 {
 264         if (test_and_clear_bit(NFSD_FILE_HASHED, &nf->nf_flags)) {
 265                 nfsd_file_do_unhash(nf);
 266                 if (!list_empty(&nf->nf_lru))
 267                         list_lru_del(&nfsd_file_lru, &nf->nf_lru);
 268                 return true;
 269         }
 270         return false;
 271 }
 272
 273 /*
 274  * Return true if the file was unhashed.
 275  */
 276 static bool
 277 nfsd_file_unhash_and_release_locked(struct nfsd_file *nf, struct list_head *dispose)
 278 {
 279         lockdep_assert_held(&nfsd_file_hashtbl[nf->nf_hashval].nfb_lock);
 280
 281         trace_nfsd_file_unhash_and_release_locked(nf);
 282         if (!nfsd_file_unhash(nf))
 283                 return false;
 284         /* keep final reference for nfsd_file_lru_dispose */
 285         if (atomic_add_unless(&nf->nf_ref, -1, 1))
 286                 return true;
 287
 288         list_add(&nf->nf_lru, dispose);
 289         return true;
 290 }
 291
 292 static int
 293 nfsd_file_put_noref(struct nfsd_file *nf)
 294 {
 295         int count;
 296         trace_nfsd_file_put(nf);
 297
 298         count = atomic_dec_return(&nf->nf_ref);
 299         if (!count) {
 300                 WARN_ON(test_bit(NFSD_FILE_HASHED, &nf->nf_flags));
 301                 nfsd_file_free(nf);
 302         }
 303         return count;
 304 }
 305
 306 void
 307 nfsd_file_put(struct nfsd_file *nf)
 308 {
 309         bool is_hashed = test_bit(NFSD_FILE_HASHED, &nf->nf_flags) != 0;
 310         bool unused = !nfsd_file_in_use(nf);
 311
 312         set_bit(NFSD_FILE_REFERENCED, &nf->nf_flags);
 313         if (nfsd_file_put_noref(nf) == 1 && is_hashed && unused)
 314                 nfsd_file_schedule_laundrette();
 315         if (atomic_long_read(&nfsd_filecache_count) >= NFSD_FILE_LRU_LIMIT)
 316                 nfsd_file_gc();
 317 }
 318
 319 struct nfsd_file *
 320 nfsd_file_get(struct nfsd_file *nf)
 321 {
 322         if (likely(atomic_inc_not_zero(&nf->nf_ref)))
 323                 return nf;
 324         return NULL;
 325 }
 326
 327 static void
 328 nfsd_file_dispose_list(struct list_head *dispose)
 329 {
 330         struct nfsd_file *nf;
 331
 332         while(!list_empty(dispose)) {
 333                 nf = list_first_entry(dispose, struct nfsd_file, nf_lru);
 334                 list_del(&nf->nf_lru);
 335                 nfsd_file_put_noref(nf);
 336         }
 337 }
 338
 339 static void
 340 nfsd_file_dispose_list_sync(struct list_head *dispose)
 341 {
 342         bool flush = false;
 343         struct nfsd_file *nf;
 344
 345         while(!list_empty(dispose)) {
 346                 nf = list_first_entry(dispose, struct nfsd_file, nf_lru);
 347                 list_del(&nf->nf_lru);
 348                 if (!atomic_dec_and_test(&nf->nf_ref))
 349                         continue;
 350                 if (nfsd_file_free(nf))
 351                         flush = true;
 352         }
 353         if (flush)
 354                 flush_delayed_fput();
 355 }
 356
 357 static void
 358 nfsd_file_list_remove_disposal(struct list_head *dst,
 359                 struct nfsd_fcache_disposal *l)
 360 {
 361         spin_lock(&l->lock);
 362         list_splice_init(&l->freeme, dst);
 363         spin_unlock(&l->lock);
 364 }
 365
 366 static void
 367 nfsd_file_list_add_disposal(struct list_head *files, struct net *net)
 368 {
 369         struct nfsd_fcache_disposal *l;
 370
 371         rcu_read_lock();
 372         list_for_each_entry_rcu(l, &laundrettes, list) {
 373                 if (l->net == net) {
 374                         spin_lock(&l->lock);
 375                         list_splice_tail_init(files, &l->freeme);
 376                         spin_unlock(&l->lock);
 377                         queue_work(nfsd_filecache_wq, &l->work);
 378                         break;
 379                 }
 380         }
 381         rcu_read_unlock();
 382 }
 383
 384 static void
 385 nfsd_file_list_add_pernet(struct list_head *dst, struct list_head *src,
 386                 struct net *net)
 387 {
 388         struct nfsd_file *nf, *tmp;
 389
 390         list_for_each_entry_safe(nf, tmp, src, nf_lru) {
 391                 if (nf->nf_net == net)
 392                         list_move_tail(&nf->nf_lru, dst);
 393         }
 394 }
 395
 396 static void
 397 nfsd_file_dispose_list_delayed(struct list_head *dispose)
 398 {
 399         LIST_HEAD(list);
 400         struct nfsd_file *nf;
 401
 402         while(!list_empty(dispose)) {
 403                 nf = list_first_entry(dispose, struct nfsd_file, nf_lru);
 404                 nfsd_file_list_add_pernet(&list, dispose, nf->nf_net);
 405                 nfsd_file_list_add_disposal(&list, nf->nf_net);
 406         }
 407 }
 408
 409 /*
 410  * Note this can deadlock with nfsd_file_cache_purge.
 411  */
 412 static enum lru_status
 413 nfsd_file_lru_cb(struct list_head *item, struct list_lru_one *lru,
 414                  spinlock_t *lock, void *arg)
 415         __releases(lock)
 416         __acquires(lock)
 417 {
 418         struct list_head *head = arg;
 419         struct nfsd_file *nf = list_entry(item, struct nfsd_file, nf_lru);
 420
 421         /*
 422          * Do a lockless refcount check. The hashtable holds one reference, so
 423          * we look to see if anything else has a reference, or if any have
 424          * been put since the shrinker last ran. Those don't get unhashed and
 425          * released.
 426          *
 427          * Note that in the put path, we set the flag and then decrement the
 428          * counter. Here we check the counter and then test and clear the flag.
 429          * That order is deliberate to ensure that we can do this locklessly.
 430          */
 431         if (atomic_read(&nf->nf_ref) > 1)
 432                 goto out_skip;
 433
 434         /*
 435          * Don't throw out files that are still undergoing I/O or
 436          * that have uncleared errors pending.
 437          */
 438         if (nfsd_file_check_writeback(nf))
 439                 goto out_skip;
 440
 441         if (test_and_clear_bit(NFSD_FILE_REFERENCED, &nf->nf_flags))
 442                 goto out_skip;
 443
 444         if (!test_and_clear_bit(NFSD_FILE_HASHED, &nf->nf_flags))
 445                 goto out_skip;
 446
 447         list_lru_isolate_move(lru, &nf->nf_lru, head);
 448         return LRU_REMOVED;
 449 out_skip:
 450         return LRU_SKIP;
 451 }
 452
 453 static unsigned long
 454 nfsd_file_lru_walk_list(struct shrink_control *sc)
 455 {
 456         LIST_HEAD(head);
 457         struct nfsd_file *nf;
 458         unsigned long ret;
 459
 460         if (sc)
 461                 ret = list_lru_shrink_walk(&nfsd_file_lru, sc,
 462                                 nfsd_file_lru_cb, &head);
 463         else
 464                 ret = list_lru_walk(&nfsd_file_lru,
 465                                 nfsd_file_lru_cb,
 466                                 &head, LONG_MAX);
 467         list_for_each_entry(nf, &head, nf_lru) {
 468                 spin_lock(&nfsd_file_hashtbl[nf->nf_hashval].nfb_lock);
 469                 nfsd_file_do_unhash(nf);
 470                 spin_unlock(&nfsd_file_hashtbl[nf->nf_hashval].nfb_lock);
 471         }
 472         nfsd_file_dispose_list_delayed(&head);
 473         return ret;
 474 }
 475
 476 static void
 477 nfsd_file_gc(void)
 478 {
 479         nfsd_file_lru_walk_list(NULL);
 480 }
 481
 482 static void
 483 nfsd_file_gc_worker(struct work_struct *work)
 484 {
 485         nfsd_file_gc();
 486         nfsd_file_schedule_laundrette();
 487 }
 488
 489 static unsigned long
 490 nfsd_file_lru_count(struct shrinker *s, struct shrink_control *sc)
 491 {
 492         return list_lru_count(&nfsd_file_lru);
 493 }
 494
 495 static unsigned long
 496 nfsd_file_lru_scan(struct shrinker *s, struct shrink_control *sc)
 497 {
 498         return nfsd_file_lru_walk_list(sc);
 499 }
 500
 501 static struct shrinker  nfsd_file_shrinker = {
 502         .scan_objects = nfsd_file_lru_scan,
 503         .count_objects = nfsd_file_lru_count,
 504         .seeks = 1,
 505 };
 506
 507 static void
 508 __nfsd_file_close_inode(struct inode *inode, unsigned int hashval,
 509                         struct list_head *dispose)
 510 {
 511         struct nfsd_file        *nf;
 512         struct hlist_node       *tmp;
 513
 514         spin_lock(&nfsd_file_hashtbl[hashval].nfb_lock);
 515         hlist_for_each_entry_safe(nf, tmp, &nfsd_file_hashtbl[hashval].nfb_head, nf_node) {
 516                 if (inode == nf->nf_inode)
 517                         nfsd_file_unhash_and_release_locked(nf, dispose);
 518         }
 519         spin_unlock(&nfsd_file_hashtbl[hashval].nfb_lock);
 520 }
 521
 522 /**
 523  * nfsd_file_close_inode_sync - attempt to forcibly close a nfsd_file
 524  * @inode: inode of the file to attempt to remove
 525  *
 526  * Walk the whole hash bucket, looking for any files that correspond to "inode".
 527  * If any do, then unhash them and put the hashtable reference to them and
 528  * destroy any that had their last reference put. Also ensure that any of the
 529  * fputs also have their final __fput done as well.
 530  */
 531 void
 532 nfsd_file_close_inode_sync(struct inode *inode)
 533 {
 534         unsigned int            hashval = (unsigned int)hash_long(inode->i_ino,
 535                                                 NFSD_FILE_HASH_BITS);
 536         LIST_HEAD(dispose);
 537
 538         __nfsd_file_close_inode(inode, hashval, &dispose);
 539         trace_nfsd_file_close_inode_sync(inode, hashval, !list_empty(&dispose));
 540         nfsd_file_dispose_list_sync(&dispose);
 541 }
 542
 543 /**
 544  * nfsd_file_close_inode_sync - attempt to forcibly close a nfsd_file
 545  * @inode: inode of the file to attempt to remove
 546  *
 547  * Walk the whole hash bucket, looking for any files that correspond to "inode".
 548  * If any do, then unhash them and put the hashtable reference to them and
 549  * destroy any that had their last reference put.
 550  */
 551 static void
 552 nfsd_file_close_inode(struct inode *inode)
 553 {
 554         unsigned int            hashval = (unsigned int)hash_long(inode->i_ino,
 555                                                 NFSD_FILE_HASH_BITS);
 556         LIST_HEAD(dispose);
 557
 558         __nfsd_file_close_inode(inode, hashval, &dispose);
 559         trace_nfsd_file_close_inode(inode, hashval, !list_empty(&dispose));
 560         nfsd_file_dispose_list_delayed(&dispose);
 561 }
 562
 563 /**
 564  * nfsd_file_delayed_close - close unused nfsd_files
 565  * @work: dummy
 566  *
 567  * Walk the LRU list and close any entries that have not been used since
 568  * the last scan.
 569  *
 570  * Note this can deadlock with nfsd_file_cache_purge.
 571  */
 572 static void
 573 nfsd_file_delayed_close(struct work_struct *work)
 574 {
 575         LIST_HEAD(head);
 576         struct nfsd_fcache_disposal *l = container_of(work,
 577                         struct nfsd_fcache_disposal, work);
 578
 579         nfsd_file_list_remove_disposal(&head, l);
 580         nfsd_file_dispose_list(&head);
 581 }
 582
 583 static int
 584 nfsd_file_lease_notifier_call(struct notifier_block *nb, unsigned long arg,
 585                             void *data)
 586 {
 587         struct file_lock *fl = data;
 588
 589         /* Only close files for F_SETLEASE leases */
 590         if (fl->fl_flags & FL_LEASE)
 591                 nfsd_file_close_inode_sync(file_inode(fl->fl_file));
 592         return 0;
 593 }
 594
 595 static struct notifier_block nfsd_file_lease_notifier = {
 596         .notifier_call = nfsd_file_lease_notifier_call,
 597 };
 598
 599 static int
 600 nfsd_file_fsnotify_handle_event(struct fsnotify_group *group,
 601                                 struct inode *inode,
 602                                 u32 mask, const void *data, int data_type,
 603                                 const struct qstr *file_name, u32 cookie,
 604                                 struct fsnotify_iter_info *iter_info)
 605 {
 606         trace_nfsd_file_fsnotify_handle_event(inode, mask);
 607
 608         /* Should be no marks on non-regular files */
 609         if (!S_ISREG(inode->i_mode)) {
 610                 WARN_ON_ONCE(1);
 611                 return 0;
 612         }
 613
 614         /* don't close files if this was not the last link */
 615         if (mask & FS_ATTRIB) {
 616                 if (inode->i_nlink)
 617                         return 0;
 618         }
 619
 620         nfsd_file_close_inode(inode);
 621         return 0;
 622 }
 623
 624
 625 static const struct fsnotify_ops nfsd_file_fsnotify_ops = {
 626         .handle_event = nfsd_file_fsnotify_handle_event,
 627         .free_mark = nfsd_file_mark_free,
 628 };
 629
 630 int
 631 nfsd_file_cache_init(void)
 632 {
 633         int             ret = -ENOMEM;
 634         unsigned int    i;
 635
 636         clear_bit(NFSD_FILE_SHUTDOWN, &nfsd_file_lru_flags);
 637
 638         if (nfsd_file_hashtbl)
 639                 return 0;
 640
 641         nfsd_filecache_wq = alloc_workqueue("nfsd_filecache", 0, 0);
 642         if (!nfsd_filecache_wq)
 643                 goto out;
 644
 645         nfsd_file_hashtbl = kcalloc(NFSD_FILE_HASH_SIZE,
 646                                 sizeof(*nfsd_file_hashtbl), GFP_KERNEL);
 647         if (!nfsd_file_hashtbl) {
 648                 pr_err("nfsd: unable to allocate nfsd_file_hashtbl\n");
 649                 goto out_err;
 650         }
 651
 652         nfsd_file_slab = kmem_cache_create("nfsd_file",
 653                                 sizeof(struct nfsd_file), 0, 0, NULL);
 654         if (!nfsd_file_slab) {
 655                 pr_err("nfsd: unable to create nfsd_file_slab\n");
 656                 goto out_err;
 657         }
 658
 659         nfsd_file_mark_slab = kmem_cache_create("nfsd_file_mark",
 660                                         sizeof(struct nfsd_file_mark), 0, 0, NULL);
 661         if (!nfsd_file_mark_slab) {
 662                 pr_err("nfsd: unable to create nfsd_file_mark_slab\n");
 663                 goto out_err;
 664         }
 665
 666
 667         ret = list_lru_init(&nfsd_file_lru);
 668         if (ret) {
 669                 pr_err("nfsd: failed to init nfsd_file_lru: %d\n", ret);
 670                 goto out_err;
 671         }
 672
 673         ret = register_shrinker(&nfsd_file_shrinker);
 674         if (ret) {
 675                 pr_err("nfsd: failed to register nfsd_file_shrinker: %d\n", ret);
 676                 goto out_lru;
 677         }
 678
 679         ret = lease_register_notifier(&nfsd_file_lease_notifier);
 680         if (ret) {
 681                 pr_err("nfsd: unable to register lease notifier: %d\n", ret);
 682                 goto out_shrinker;
 683         }
 684
 685         nfsd_file_fsnotify_group = fsnotify_alloc_group(&nfsd_file_fsnotify_ops);
 686         if (IS_ERR(nfsd_file_fsnotify_group)) {
 687                 pr_err("nfsd: unable to create fsnotify group: %ld\n",
 688                         PTR_ERR(nfsd_file_fsnotify_group));
 689                 nfsd_file_fsnotify_group = NULL;
 690                 goto out_notifier;
 691         }
 692
 693         for (i = 0; i < NFSD_FILE_HASH_SIZE; i++) {
 694                 INIT_HLIST_HEAD(&nfsd_file_hashtbl[i].nfb_head);
 695                 spin_lock_init(&nfsd_file_hashtbl[i].nfb_lock);
 696         }
 697
 698         INIT_DELAYED_WORK(&nfsd_filecache_laundrette, nfsd_file_gc_worker);
 699 out:
 700         return ret;
 701 out_notifier:
 702         lease_unregister_notifier(&nfsd_file_lease_notifier);
 703 out_shrinker:
 704         unregister_shrinker(&nfsd_file_shrinker);
 705 out_lru:
 706         list_lru_destroy(&nfsd_file_lru);
 707 out_err:
 708         kmem_cache_destroy(nfsd_file_slab);
 709         nfsd_file_slab = NULL;
 710         kmem_cache_destroy(nfsd_file_mark_slab);
 711         nfsd_file_mark_slab = NULL;
 712         kfree(nfsd_file_hashtbl);
 713         nfsd_file_hashtbl = NULL;
 714         destroy_workqueue(nfsd_filecache_wq);
 715         nfsd_filecache_wq = NULL;
 716         goto out;
 717 }
 718
 719 /*
 720  * Note this can deadlock with nfsd_file_lru_cb.
 721  */
 722 void
 723 nfsd_file_cache_purge(struct net *net)
 724 {
 725         unsigned int            i;
 726         struct nfsd_file        *nf;
 727         struct hlist_node       *next;
 728         LIST_HEAD(dispose);
 729         bool del;
 730
 731         if (!nfsd_file_hashtbl)
 732                 return;
 733
 734         for (i = 0; i < NFSD_FILE_HASH_SIZE; i++) {
 735                 struct nfsd_fcache_bucket *nfb = &nfsd_file_hashtbl[i];
 736
 737                 spin_lock(&nfb->nfb_lock);
 738                 hlist_for_each_entry_safe(nf, next, &nfb->nfb_head, nf_node) {
 739                         if (net && nf->nf_net != net)
 740                                 continue;
 741                         del = nfsd_file_unhash_and_release_locked(nf, &dispose);
 742
 743                         /*
 744                          * Deadlock detected! Something marked this entry as
 745                          * unhased, but hasn't removed it from the hash list.
 746                          */
 747                         WARN_ON_ONCE(!del);
 748                 }
 749                 spin_unlock(&nfb->nfb_lock);
 750                 nfsd_file_dispose_list(&dispose);
 751         }
 752 }
 753
 754 static struct nfsd_fcache_disposal *
 755 nfsd_alloc_fcache_disposal(struct net *net)
 756 {
 757         struct nfsd_fcache_disposal *l;
 758
 759         l = kmalloc(sizeof(*l), GFP_KERNEL);
 760         if (!l)
 761                 return NULL;
 762         INIT_WORK(&l->work, nfsd_file_delayed_close);
 763         l->net = net;
 764         spin_lock_init(&l->lock);
 765         INIT_LIST_HEAD(&l->freeme);
 766         return l;
 767 }
 768
 769 static void
 770 nfsd_free_fcache_disposal(struct nfsd_fcache_disposal *l)
 771 {
 772         rcu_assign_pointer(l->net, NULL);
 773         cancel_work_sync(&l->work);
 774         nfsd_file_dispose_list(&l->freeme);
 775         kfree_rcu(l, rcu);
 776 }
 777
 778 static void
 779 nfsd_add_fcache_disposal(struct nfsd_fcache_disposal *l)
 780 {
 781         spin_lock(&laundrette_lock);
 782         list_add_tail_rcu(&l->list, &laundrettes);
 783         spin_unlock(&laundrette_lock);
 784 }
 785
 786 static void
 787 nfsd_del_fcache_disposal(struct nfsd_fcache_disposal *l)
 788 {
 789         spin_lock(&laundrette_lock);
 790         list_del_rcu(&l->list);
 791         spin_unlock(&laundrette_lock);
 792 }
 793
 794 static int
 795 nfsd_alloc_fcache_disposal_net(struct net *net)
 796 {
 797         struct nfsd_fcache_disposal *l;
 798
 799         l = nfsd_alloc_fcache_disposal(net);
 800         if (!l)
 801                 return -ENOMEM;
 802         nfsd_add_fcache_disposal(l);
 803         return 0;
 804 }
 805
 806 static void
 807 nfsd_free_fcache_disposal_net(struct net *net)
 808 {
 809         struct nfsd_fcache_disposal *l;
 810
 811         rcu_read_lock();
 812         list_for_each_entry_rcu(l, &laundrettes, list) {
 813                 if (l->net != net)
 814                         continue;
 815                 nfsd_del_fcache_disposal(l);
 816                 rcu_read_unlock();
 817                 nfsd_free_fcache_disposal(l);
 818                 return;
 819         }
 820         rcu_read_unlock();
 821 }
 822
 823 int
 824 nfsd_file_cache_start_net(struct net *net)
 825 {
 826         return nfsd_alloc_fcache_disposal_net(net);
 827 }
 828
 829 void
 830 nfsd_file_cache_shutdown_net(struct net *net)
 831 {
 832         nfsd_file_cache_purge(net);
 833         nfsd_free_fcache_disposal_net(net);
 834 }
 835
 836 void
 837 nfsd_file_cache_shutdown(void)
 838 {
 839         set_bit(NFSD_FILE_SHUTDOWN, &nfsd_file_lru_flags);
 840
 841         lease_unregister_notifier(&nfsd_file_lease_notifier);
 842         unregister_shrinker(&nfsd_file_shrinker);
 843         /*
 844          * make sure all callers of nfsd_file_lru_cb are done before
 845          * calling nfsd_file_cache_purge
 846          */
 847         cancel_delayed_work_sync(&nfsd_filecache_laundrette);
 848         nfsd_file_cache_purge(NULL);
 849         list_lru_destroy(&nfsd_file_lru);
 850         rcu_barrier();
 851         fsnotify_put_group(nfsd_file_fsnotify_group);
 852         nfsd_file_fsnotify_group = NULL;
 853         kmem_cache_destroy(nfsd_file_slab);
 854         nfsd_file_slab = NULL;
 855         fsnotify_wait_marks_destroyed();
 856         kmem_cache_destroy(nfsd_file_mark_slab);
 857         nfsd_file_mark_slab = NULL;
 858         kfree(nfsd_file_hashtbl);
 859         nfsd_file_hashtbl = NULL;
 860         destroy_workqueue(nfsd_filecache_wq);
 861         nfsd_filecache_wq = NULL;
 862 }
 863
 864 static bool
 865 nfsd_match_cred(const struct cred *c1, const struct cred *c2)
 866 {
 867         int i;
 868
 869         if (!uid_eq(c1->fsuid, c2->fsuid))
 870                 return false;
 871         if (!gid_eq(c1->fsgid, c2->fsgid))
 872                 return false;
 873         if (c1->group_info == NULL || c2->group_info == NULL)
 874                 return c1->group_info == c2->group_info;
 875         if (c1->group_info->ngroups != c2->group_info->ngroups)
 876                 return false;
 877         for (i = 0; i < c1->group_info->ngroups; i++) {
 878                 if (!gid_eq(c1->group_info->gid[i], c2->group_info->gid[i]))
 879                         return false;
 880         }
 881         return true;
 882 }
 883
 884 static struct nfsd_file *
 885 nfsd_file_find_locked(struct inode *inode, unsigned int may_flags,
 886                         unsigned int hashval, struct net *net)
 887 {
 888         struct nfsd_file *nf;
 889         unsigned char need = may_flags & NFSD_FILE_MAY_MASK;
 890
 891         hlist_for_each_entry_rcu(nf, &nfsd_file_hashtbl[hashval].nfb_head,
 892                                  nf_node) {
 893                 if ((need & nf->nf_may) != need)
 894                         continue;
 895                 if (nf->nf_inode != inode)
 896                         continue;
 897                 if (nf->nf_net != net)
 898                         continue;
 899                 if (!nfsd_match_cred(nf->nf_cred, current_cred()))
 900                         continue;
 901                 if (nfsd_file_get(nf) != NULL)
 902                         return nf;
 903         }
 904         return NULL;
 905 }
 906
 907 /**
 908  * nfsd_file_is_cached - are there any cached open files for this fh?
 909  * @inode: inode of the file to check
 910  *
 911  * Scan the hashtable for open files that match this fh. Returns true if there
 912  * are any, and false if not.
 913  */
 914 bool
 915 nfsd_file_is_cached(struct inode *inode)
 916 {
 917         bool                    ret = false;
 918         struct nfsd_file        *nf;
 919         unsigned int            hashval;
 920
 921         hashval = (unsigned int)hash_long(inode->i_ino, NFSD_FILE_HASH_BITS);
 922
 923         rcu_read_lock();
 924         hlist_for_each_entry_rcu(nf, &nfsd_file_hashtbl[hashval].nfb_head,
 925                                  nf_node) {
 926                 if (inode == nf->nf_inode) {
 927                         ret = true;
 928                         break;
 929                 }
 930         }
 931         rcu_read_unlock();
 932         trace_nfsd_file_is_cached(inode, hashval, (int)ret);
 933         return ret;
 934 }
 935
 936 __be32
 937 nfsd_file_acquire(struct svc_rqst *rqstp, struct svc_fh *fhp,
 938                   unsigned int may_flags, struct nfsd_file **pnf)
 939 {
 940         __be32  status;
 941         struct net *net = SVC_NET(rqstp);
 942         struct nfsd_file *nf, *new;
 943         struct inode *inode;
 944         unsigned int hashval;
 945         bool retry = true;
 946
 947         /* FIXME: skip this if fh_dentry is already set? */
 948         status = fh_verify(rqstp, fhp, S_IFREG,
 949                                 may_flags|NFSD_MAY_OWNER_OVERRIDE);
 950         if (status != nfs_ok)
 951                 return status;
 952
 953         inode = d_inode(fhp->fh_dentry);
 954         hashval = (unsigned int)hash_long(inode->i_ino, NFSD_FILE_HASH_BITS);
 955 retry:
 956         rcu_read_lock();
 957         nf = nfsd_file_find_locked(inode, may_flags, hashval, net);
 958         rcu_read_unlock();
 959         if (nf)
 960                 goto wait_for_construction;
 961
 962         new = nfsd_file_alloc(inode, may_flags, hashval, net);
 963         if (!new) {
 964                 trace_nfsd_file_acquire(rqstp, hashval, inode, may_flags,
 965                                         NULL, nfserr_jukebox);
 966                 return nfserr_jukebox;
 967         }
 968
 969         spin_lock(&nfsd_file_hashtbl[hashval].nfb_lock);
 970         nf = nfsd_file_find_locked(inode, may_flags, hashval, net);
 971         if (nf == NULL)
 972                 goto open_file;
 973         spin_unlock(&nfsd_file_hashtbl[hashval].nfb_lock);
 974         nfsd_file_slab_free(&new->nf_rcu);
 975
 976 wait_for_construction:
 977         wait_on_bit(&nf->nf_flags, NFSD_FILE_PENDING, TASK_UNINTERRUPTIBLE);
 978
 979         /* Did construction of this file fail? */
 980         if (!test_bit(NFSD_FILE_HASHED, &nf->nf_flags)) {
 981                 if (!retry) {
 982                         status = nfserr_jukebox;
 983                         goto out;
 984                 }
 985                 retry = false;
 986                 nfsd_file_put_noref(nf);
 987                 goto retry;
 988         }
 989
 990         this_cpu_inc(nfsd_file_cache_hits);
 991
 992         if (!(may_flags & NFSD_MAY_NOT_BREAK_LEASE)) {
 993                 bool write = (may_flags & NFSD_MAY_WRITE);
 994
 995                 if (test_bit(NFSD_FILE_BREAK_READ, &nf->nf_flags) ||
 996                     (test_bit(NFSD_FILE_BREAK_WRITE, &nf->nf_flags) && write)) {
 997                         status = nfserrno(nfsd_open_break_lease(
 998                                         file_inode(nf->nf_file), may_flags));
 999                         if (status == nfs_ok) {
1000                                 clear_bit(NFSD_FILE_BREAK_READ, &nf->nf_flags);
1001                                 if (write)
1002                                         clear_bit(NFSD_FILE_BREAK_WRITE,
1003                                                   &nf->nf_flags);
1004                         }
1005                 }
1006         }
1007 out:
1008         if (status == nfs_ok) {
1009                 *pnf = nf;
1010         } else {
1011                 nfsd_file_put(nf);
1012                 nf = NULL;
1013         }
1014
1015         trace_nfsd_file_acquire(rqstp, hashval, inode, may_flags, nf, status);
1016         return status;
1017 open_file:
1018         nf = new;
1019         /* Take reference for the hashtable */
1020         atomic_inc(&nf->nf_ref);
1021         __set_bit(NFSD_FILE_HASHED, &nf->nf_flags);
1022         __set_bit(NFSD_FILE_PENDING, &nf->nf_flags);
1023         list_lru_add(&nfsd_file_lru, &nf->nf_lru);
1024         hlist_add_head_rcu(&nf->nf_node, &nfsd_file_hashtbl[hashval].nfb_head);
1025         ++nfsd_file_hashtbl[hashval].nfb_count;
1026         nfsd_file_hashtbl[hashval].nfb_maxcount = max(nfsd_file_hashtbl[hashval].nfb_maxcount,
1027                         nfsd_file_hashtbl[hashval].nfb_count);
1028         spin_unlock(&nfsd_file_hashtbl[hashval].nfb_lock);
1029         if (atomic_long_inc_return(&nfsd_filecache_count) >= NFSD_FILE_LRU_THRESHOLD)
1030                 nfsd_file_gc();
1031
1032         nf->nf_mark = nfsd_file_mark_find_or_create(nf);
1033         if (nf->nf_mark)
1034                 status = nfsd_open_verified(rqstp, fhp, S_IFREG,
1035                                 may_flags, &nf->nf_file);
1036         else
1037                 status = nfserr_jukebox;
1038         /*
1039          * If construction failed, or we raced with a call to unlink()
1040          * then unhash.
1041          */
1042         if (status != nfs_ok || inode->i_nlink == 0) {
1043                 bool do_free;
1044                 spin_lock(&nfsd_file_hashtbl[hashval].nfb_lock);
1045                 do_free = nfsd_file_unhash(nf);
1046                 spin_unlock(&nfsd_file_hashtbl[hashval].nfb_lock);
1047                 if (do_free)
1048                         nfsd_file_put_noref(nf);
1049         }
1050         clear_bit_unlock(NFSD_FILE_PENDING, &nf->nf_flags);
1051         smp_mb__after_atomic();
1052         wake_up_bit(&nf->nf_flags, NFSD_FILE_PENDING);
1053         goto out;
1054 }
1055
1056 /*
1057  * Note that fields may be added, removed or reordered in the future. Programs
1058  * scraping this file for info should test the labels to ensure they're
1059  * getting the correct field.
1060  */
1061 static int nfsd_file_cache_stats_show(struct seq_file *m, void *v)
1062 {
1063         unsigned int i, count = 0, longest = 0;
1064         unsigned long hits = 0;
1065
1066         /*
1067          * No need for spinlocks here since we're not terribly interested in
1068          * accuracy. We do take the nfsd_mutex simply to ensure that we
1069          * don't end up racing with server shutdown
1070          */
1071         mutex_lock(&nfsd_mutex);
1072         if (nfsd_file_hashtbl) {
1073                 for (i = 0; i < NFSD_FILE_HASH_SIZE; i++) {
1074                         count += nfsd_file_hashtbl[i].nfb_count;
1075                         longest = max(longest, nfsd_file_hashtbl[i].nfb_count);
1076                 }
1077         }
1078         mutex_unlock(&nfsd_mutex);
1079
1080         for_each_possible_cpu(i)
1081                 hits += per_cpu(nfsd_file_cache_hits, i);
1082
1083         seq_printf(m, "total entries: %u\n", count);
1084         seq_printf(m, "longest chain: %u\n", longest);
1085         seq_printf(m, "cache hits:    %lu\n", hits);
1086         return 0;
1087 }
1088
1089 int nfsd_file_cache_stats_open(struct inode *inode, struct file *file)
1090 {
1091         return single_open(file, nfsd_file_cache_stats_show, NULL);
1092 }