ceph: avoid sending unnessesary FLUSHSNAP message
[firefly-linux-kernel-4.4.55.git] / fs / ceph / caps.c
1 #include <linux/ceph/ceph_debug.h>
2
3 #include <linux/fs.h>
4 #include <linux/kernel.h>
5 #include <linux/sched.h>
6 #include <linux/slab.h>
7 #include <linux/vmalloc.h>
8 #include <linux/wait.h>
9 #include <linux/writeback.h>
10
11 #include "super.h"
12 #include "mds_client.h"
13 #include "cache.h"
14 #include <linux/ceph/decode.h>
15 #include <linux/ceph/messenger.h>
16
17 /*
18  * Capability management
19  *
20  * The Ceph metadata servers control client access to inode metadata
21  * and file data by issuing capabilities, granting clients permission
22  * to read and/or write both inode field and file data to OSDs
23  * (storage nodes).  Each capability consists of a set of bits
24  * indicating which operations are allowed.
25  *
26  * If the client holds a *_SHARED cap, the client has a coherent value
27  * that can be safely read from the cached inode.
28  *
29  * In the case of a *_EXCL (exclusive) or FILE_WR capabilities, the
30  * client is allowed to change inode attributes (e.g., file size,
31  * mtime), note its dirty state in the ceph_cap, and asynchronously
32  * flush that metadata change to the MDS.
33  *
34  * In the event of a conflicting operation (perhaps by another
35  * client), the MDS will revoke the conflicting client capabilities.
36  *
37  * In order for a client to cache an inode, it must hold a capability
38  * with at least one MDS server.  When inodes are released, release
39  * notifications are batched and periodically sent en masse to the MDS
40  * cluster to release server state.
41  */
42
43
44 /*
45  * Generate readable cap strings for debugging output.
46  */
47 #define MAX_CAP_STR 20
48 static char cap_str[MAX_CAP_STR][40];
49 static DEFINE_SPINLOCK(cap_str_lock);
50 static int last_cap_str;
51
52 static char *gcap_string(char *s, int c)
53 {
54         if (c & CEPH_CAP_GSHARED)
55                 *s++ = 's';
56         if (c & CEPH_CAP_GEXCL)
57                 *s++ = 'x';
58         if (c & CEPH_CAP_GCACHE)
59                 *s++ = 'c';
60         if (c & CEPH_CAP_GRD)
61                 *s++ = 'r';
62         if (c & CEPH_CAP_GWR)
63                 *s++ = 'w';
64         if (c & CEPH_CAP_GBUFFER)
65                 *s++ = 'b';
66         if (c & CEPH_CAP_GLAZYIO)
67                 *s++ = 'l';
68         return s;
69 }
70
71 const char *ceph_cap_string(int caps)
72 {
73         int i;
74         char *s;
75         int c;
76
77         spin_lock(&cap_str_lock);
78         i = last_cap_str++;
79         if (last_cap_str == MAX_CAP_STR)
80                 last_cap_str = 0;
81         spin_unlock(&cap_str_lock);
82
83         s = cap_str[i];
84
85         if (caps & CEPH_CAP_PIN)
86                 *s++ = 'p';
87
88         c = (caps >> CEPH_CAP_SAUTH) & 3;
89         if (c) {
90                 *s++ = 'A';
91                 s = gcap_string(s, c);
92         }
93
94         c = (caps >> CEPH_CAP_SLINK) & 3;
95         if (c) {
96                 *s++ = 'L';
97                 s = gcap_string(s, c);
98         }
99
100         c = (caps >> CEPH_CAP_SXATTR) & 3;
101         if (c) {
102                 *s++ = 'X';
103                 s = gcap_string(s, c);
104         }
105
106         c = caps >> CEPH_CAP_SFILE;
107         if (c) {
108                 *s++ = 'F';
109                 s = gcap_string(s, c);
110         }
111
112         if (s == cap_str[i])
113                 *s++ = '-';
114         *s = 0;
115         return cap_str[i];
116 }
117
118 void ceph_caps_init(struct ceph_mds_client *mdsc)
119 {
120         INIT_LIST_HEAD(&mdsc->caps_list);
121         spin_lock_init(&mdsc->caps_list_lock);
122 }
123
124 void ceph_caps_finalize(struct ceph_mds_client *mdsc)
125 {
126         struct ceph_cap *cap;
127
128         spin_lock(&mdsc->caps_list_lock);
129         while (!list_empty(&mdsc->caps_list)) {
130                 cap = list_first_entry(&mdsc->caps_list,
131                                        struct ceph_cap, caps_item);
132                 list_del(&cap->caps_item);
133                 kmem_cache_free(ceph_cap_cachep, cap);
134         }
135         mdsc->caps_total_count = 0;
136         mdsc->caps_avail_count = 0;
137         mdsc->caps_use_count = 0;
138         mdsc->caps_reserve_count = 0;
139         mdsc->caps_min_count = 0;
140         spin_unlock(&mdsc->caps_list_lock);
141 }
142
143 void ceph_adjust_min_caps(struct ceph_mds_client *mdsc, int delta)
144 {
145         spin_lock(&mdsc->caps_list_lock);
146         mdsc->caps_min_count += delta;
147         BUG_ON(mdsc->caps_min_count < 0);
148         spin_unlock(&mdsc->caps_list_lock);
149 }
150
151 void ceph_reserve_caps(struct ceph_mds_client *mdsc,
152                       struct ceph_cap_reservation *ctx, int need)
153 {
154         int i;
155         struct ceph_cap *cap;
156         int have;
157         int alloc = 0;
158         LIST_HEAD(newcaps);
159
160         dout("reserve caps ctx=%p need=%d\n", ctx, need);
161
162         /* first reserve any caps that are already allocated */
163         spin_lock(&mdsc->caps_list_lock);
164         if (mdsc->caps_avail_count >= need)
165                 have = need;
166         else
167                 have = mdsc->caps_avail_count;
168         mdsc->caps_avail_count -= have;
169         mdsc->caps_reserve_count += have;
170         BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
171                                          mdsc->caps_reserve_count +
172                                          mdsc->caps_avail_count);
173         spin_unlock(&mdsc->caps_list_lock);
174
175         for (i = have; i < need; i++) {
176                 cap = kmem_cache_alloc(ceph_cap_cachep, GFP_NOFS);
177                 if (!cap)
178                         break;
179                 list_add(&cap->caps_item, &newcaps);
180                 alloc++;
181         }
182         /* we didn't manage to reserve as much as we needed */
183         if (have + alloc != need)
184                 pr_warn("reserve caps ctx=%p ENOMEM need=%d got=%d\n",
185                         ctx, need, have + alloc);
186
187         spin_lock(&mdsc->caps_list_lock);
188         mdsc->caps_total_count += alloc;
189         mdsc->caps_reserve_count += alloc;
190         list_splice(&newcaps, &mdsc->caps_list);
191
192         BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
193                                          mdsc->caps_reserve_count +
194                                          mdsc->caps_avail_count);
195         spin_unlock(&mdsc->caps_list_lock);
196
197         ctx->count = need;
198         dout("reserve caps ctx=%p %d = %d used + %d resv + %d avail\n",
199              ctx, mdsc->caps_total_count, mdsc->caps_use_count,
200              mdsc->caps_reserve_count, mdsc->caps_avail_count);
201 }
202
203 int ceph_unreserve_caps(struct ceph_mds_client *mdsc,
204                         struct ceph_cap_reservation *ctx)
205 {
206         dout("unreserve caps ctx=%p count=%d\n", ctx, ctx->count);
207         if (ctx->count) {
208                 spin_lock(&mdsc->caps_list_lock);
209                 BUG_ON(mdsc->caps_reserve_count < ctx->count);
210                 mdsc->caps_reserve_count -= ctx->count;
211                 mdsc->caps_avail_count += ctx->count;
212                 ctx->count = 0;
213                 dout("unreserve caps %d = %d used + %d resv + %d avail\n",
214                      mdsc->caps_total_count, mdsc->caps_use_count,
215                      mdsc->caps_reserve_count, mdsc->caps_avail_count);
216                 BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
217                                                  mdsc->caps_reserve_count +
218                                                  mdsc->caps_avail_count);
219                 spin_unlock(&mdsc->caps_list_lock);
220         }
221         return 0;
222 }
223
224 struct ceph_cap *ceph_get_cap(struct ceph_mds_client *mdsc,
225                               struct ceph_cap_reservation *ctx)
226 {
227         struct ceph_cap *cap = NULL;
228
229         /* temporary, until we do something about cap import/export */
230         if (!ctx) {
231                 cap = kmem_cache_alloc(ceph_cap_cachep, GFP_NOFS);
232                 if (cap) {
233                         spin_lock(&mdsc->caps_list_lock);
234                         mdsc->caps_use_count++;
235                         mdsc->caps_total_count++;
236                         spin_unlock(&mdsc->caps_list_lock);
237                 }
238                 return cap;
239         }
240
241         spin_lock(&mdsc->caps_list_lock);
242         dout("get_cap ctx=%p (%d) %d = %d used + %d resv + %d avail\n",
243              ctx, ctx->count, mdsc->caps_total_count, mdsc->caps_use_count,
244              mdsc->caps_reserve_count, mdsc->caps_avail_count);
245         BUG_ON(!ctx->count);
246         BUG_ON(ctx->count > mdsc->caps_reserve_count);
247         BUG_ON(list_empty(&mdsc->caps_list));
248
249         ctx->count--;
250         mdsc->caps_reserve_count--;
251         mdsc->caps_use_count++;
252
253         cap = list_first_entry(&mdsc->caps_list, struct ceph_cap, caps_item);
254         list_del(&cap->caps_item);
255
256         BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
257                mdsc->caps_reserve_count + mdsc->caps_avail_count);
258         spin_unlock(&mdsc->caps_list_lock);
259         return cap;
260 }
261
262 void ceph_put_cap(struct ceph_mds_client *mdsc, struct ceph_cap *cap)
263 {
264         spin_lock(&mdsc->caps_list_lock);
265         dout("put_cap %p %d = %d used + %d resv + %d avail\n",
266              cap, mdsc->caps_total_count, mdsc->caps_use_count,
267              mdsc->caps_reserve_count, mdsc->caps_avail_count);
268         mdsc->caps_use_count--;
269         /*
270          * Keep some preallocated caps around (ceph_min_count), to
271          * avoid lots of free/alloc churn.
272          */
273         if (mdsc->caps_avail_count >= mdsc->caps_reserve_count +
274                                       mdsc->caps_min_count) {
275                 mdsc->caps_total_count--;
276                 kmem_cache_free(ceph_cap_cachep, cap);
277         } else {
278                 mdsc->caps_avail_count++;
279                 list_add(&cap->caps_item, &mdsc->caps_list);
280         }
281
282         BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
283                mdsc->caps_reserve_count + mdsc->caps_avail_count);
284         spin_unlock(&mdsc->caps_list_lock);
285 }
286
287 void ceph_reservation_status(struct ceph_fs_client *fsc,
288                              int *total, int *avail, int *used, int *reserved,
289                              int *min)
290 {
291         struct ceph_mds_client *mdsc = fsc->mdsc;
292
293         if (total)
294                 *total = mdsc->caps_total_count;
295         if (avail)
296                 *avail = mdsc->caps_avail_count;
297         if (used)
298                 *used = mdsc->caps_use_count;
299         if (reserved)
300                 *reserved = mdsc->caps_reserve_count;
301         if (min)
302                 *min = mdsc->caps_min_count;
303 }
304
305 /*
306  * Find ceph_cap for given mds, if any.
307  *
308  * Called with i_ceph_lock held.
309  */
310 static struct ceph_cap *__get_cap_for_mds(struct ceph_inode_info *ci, int mds)
311 {
312         struct ceph_cap *cap;
313         struct rb_node *n = ci->i_caps.rb_node;
314
315         while (n) {
316                 cap = rb_entry(n, struct ceph_cap, ci_node);
317                 if (mds < cap->mds)
318                         n = n->rb_left;
319                 else if (mds > cap->mds)
320                         n = n->rb_right;
321                 else
322                         return cap;
323         }
324         return NULL;
325 }
326
327 struct ceph_cap *ceph_get_cap_for_mds(struct ceph_inode_info *ci, int mds)
328 {
329         struct ceph_cap *cap;
330
331         spin_lock(&ci->i_ceph_lock);
332         cap = __get_cap_for_mds(ci, mds);
333         spin_unlock(&ci->i_ceph_lock);
334         return cap;
335 }
336
337 /*
338  * Return id of any MDS with a cap, preferably FILE_WR|BUFFER|EXCL, else -1.
339  */
340 static int __ceph_get_cap_mds(struct ceph_inode_info *ci)
341 {
342         struct ceph_cap *cap;
343         int mds = -1;
344         struct rb_node *p;
345
346         /* prefer mds with WR|BUFFER|EXCL caps */
347         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
348                 cap = rb_entry(p, struct ceph_cap, ci_node);
349                 mds = cap->mds;
350                 if (cap->issued & (CEPH_CAP_FILE_WR |
351                                    CEPH_CAP_FILE_BUFFER |
352                                    CEPH_CAP_FILE_EXCL))
353                         break;
354         }
355         return mds;
356 }
357
358 int ceph_get_cap_mds(struct inode *inode)
359 {
360         struct ceph_inode_info *ci = ceph_inode(inode);
361         int mds;
362         spin_lock(&ci->i_ceph_lock);
363         mds = __ceph_get_cap_mds(ceph_inode(inode));
364         spin_unlock(&ci->i_ceph_lock);
365         return mds;
366 }
367
368 /*
369  * Called under i_ceph_lock.
370  */
371 static void __insert_cap_node(struct ceph_inode_info *ci,
372                               struct ceph_cap *new)
373 {
374         struct rb_node **p = &ci->i_caps.rb_node;
375         struct rb_node *parent = NULL;
376         struct ceph_cap *cap = NULL;
377
378         while (*p) {
379                 parent = *p;
380                 cap = rb_entry(parent, struct ceph_cap, ci_node);
381                 if (new->mds < cap->mds)
382                         p = &(*p)->rb_left;
383                 else if (new->mds > cap->mds)
384                         p = &(*p)->rb_right;
385                 else
386                         BUG();
387         }
388
389         rb_link_node(&new->ci_node, parent, p);
390         rb_insert_color(&new->ci_node, &ci->i_caps);
391 }
392
393 /*
394  * (re)set cap hold timeouts, which control the delayed release
395  * of unused caps back to the MDS.  Should be called on cap use.
396  */
397 static void __cap_set_timeouts(struct ceph_mds_client *mdsc,
398                                struct ceph_inode_info *ci)
399 {
400         struct ceph_mount_options *ma = mdsc->fsc->mount_options;
401
402         ci->i_hold_caps_min = round_jiffies(jiffies +
403                                             ma->caps_wanted_delay_min * HZ);
404         ci->i_hold_caps_max = round_jiffies(jiffies +
405                                             ma->caps_wanted_delay_max * HZ);
406         dout("__cap_set_timeouts %p min %lu max %lu\n", &ci->vfs_inode,
407              ci->i_hold_caps_min - jiffies, ci->i_hold_caps_max - jiffies);
408 }
409
410 /*
411  * (Re)queue cap at the end of the delayed cap release list.
412  *
413  * If I_FLUSH is set, leave the inode at the front of the list.
414  *
415  * Caller holds i_ceph_lock
416  *    -> we take mdsc->cap_delay_lock
417  */
418 static void __cap_delay_requeue(struct ceph_mds_client *mdsc,
419                                 struct ceph_inode_info *ci)
420 {
421         __cap_set_timeouts(mdsc, ci);
422         dout("__cap_delay_requeue %p flags %d at %lu\n", &ci->vfs_inode,
423              ci->i_ceph_flags, ci->i_hold_caps_max);
424         if (!mdsc->stopping) {
425                 spin_lock(&mdsc->cap_delay_lock);
426                 if (!list_empty(&ci->i_cap_delay_list)) {
427                         if (ci->i_ceph_flags & CEPH_I_FLUSH)
428                                 goto no_change;
429                         list_del_init(&ci->i_cap_delay_list);
430                 }
431                 list_add_tail(&ci->i_cap_delay_list, &mdsc->cap_delay_list);
432 no_change:
433                 spin_unlock(&mdsc->cap_delay_lock);
434         }
435 }
436
437 /*
438  * Queue an inode for immediate writeback.  Mark inode with I_FLUSH,
439  * indicating we should send a cap message to flush dirty metadata
440  * asap, and move to the front of the delayed cap list.
441  */
442 static void __cap_delay_requeue_front(struct ceph_mds_client *mdsc,
443                                       struct ceph_inode_info *ci)
444 {
445         dout("__cap_delay_requeue_front %p\n", &ci->vfs_inode);
446         spin_lock(&mdsc->cap_delay_lock);
447         ci->i_ceph_flags |= CEPH_I_FLUSH;
448         if (!list_empty(&ci->i_cap_delay_list))
449                 list_del_init(&ci->i_cap_delay_list);
450         list_add(&ci->i_cap_delay_list, &mdsc->cap_delay_list);
451         spin_unlock(&mdsc->cap_delay_lock);
452 }
453
454 /*
455  * Cancel delayed work on cap.
456  *
457  * Caller must hold i_ceph_lock.
458  */
459 static void __cap_delay_cancel(struct ceph_mds_client *mdsc,
460                                struct ceph_inode_info *ci)
461 {
462         dout("__cap_delay_cancel %p\n", &ci->vfs_inode);
463         if (list_empty(&ci->i_cap_delay_list))
464                 return;
465         spin_lock(&mdsc->cap_delay_lock);
466         list_del_init(&ci->i_cap_delay_list);
467         spin_unlock(&mdsc->cap_delay_lock);
468 }
469
470 /*
471  * Common issue checks for add_cap, handle_cap_grant.
472  */
473 static void __check_cap_issue(struct ceph_inode_info *ci, struct ceph_cap *cap,
474                               unsigned issued)
475 {
476         unsigned had = __ceph_caps_issued(ci, NULL);
477
478         /*
479          * Each time we receive FILE_CACHE anew, we increment
480          * i_rdcache_gen.
481          */
482         if ((issued & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) &&
483             (had & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) == 0) {
484                 ci->i_rdcache_gen++;
485         }
486
487         /*
488          * if we are newly issued FILE_SHARED, mark dir not complete; we
489          * don't know what happened to this directory while we didn't
490          * have the cap.
491          */
492         if ((issued & CEPH_CAP_FILE_SHARED) &&
493             (had & CEPH_CAP_FILE_SHARED) == 0) {
494                 ci->i_shared_gen++;
495                 if (S_ISDIR(ci->vfs_inode.i_mode)) {
496                         dout(" marking %p NOT complete\n", &ci->vfs_inode);
497                         __ceph_dir_clear_complete(ci);
498                 }
499         }
500 }
501
502 /*
503  * Add a capability under the given MDS session.
504  *
505  * Caller should hold session snap_rwsem (read) and s_mutex.
506  *
507  * @fmode is the open file mode, if we are opening a file, otherwise
508  * it is < 0.  (This is so we can atomically add the cap and add an
509  * open file reference to it.)
510  */
511 void ceph_add_cap(struct inode *inode,
512                   struct ceph_mds_session *session, u64 cap_id,
513                   int fmode, unsigned issued, unsigned wanted,
514                   unsigned seq, unsigned mseq, u64 realmino, int flags,
515                   struct ceph_cap **new_cap)
516 {
517         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
518         struct ceph_inode_info *ci = ceph_inode(inode);
519         struct ceph_cap *cap;
520         int mds = session->s_mds;
521         int actual_wanted;
522
523         dout("add_cap %p mds%d cap %llx %s seq %d\n", inode,
524              session->s_mds, cap_id, ceph_cap_string(issued), seq);
525
526         /*
527          * If we are opening the file, include file mode wanted bits
528          * in wanted.
529          */
530         if (fmode >= 0)
531                 wanted |= ceph_caps_for_mode(fmode);
532
533         cap = __get_cap_for_mds(ci, mds);
534         if (!cap) {
535                 cap = *new_cap;
536                 *new_cap = NULL;
537
538                 cap->issued = 0;
539                 cap->implemented = 0;
540                 cap->mds = mds;
541                 cap->mds_wanted = 0;
542                 cap->mseq = 0;
543
544                 cap->ci = ci;
545                 __insert_cap_node(ci, cap);
546
547                 /* add to session cap list */
548                 cap->session = session;
549                 spin_lock(&session->s_cap_lock);
550                 list_add_tail(&cap->session_caps, &session->s_caps);
551                 session->s_nr_caps++;
552                 spin_unlock(&session->s_cap_lock);
553         } else {
554                 /*
555                  * auth mds of the inode changed. we received the cap export
556                  * message, but still haven't received the cap import message.
557                  * handle_cap_export() updated the new auth MDS' cap.
558                  *
559                  * "ceph_seq_cmp(seq, cap->seq) <= 0" means we are processing
560                  * a message that was send before the cap import message. So
561                  * don't remove caps.
562                  */
563                 if (ceph_seq_cmp(seq, cap->seq) <= 0) {
564                         WARN_ON(cap != ci->i_auth_cap);
565                         WARN_ON(cap->cap_id != cap_id);
566                         seq = cap->seq;
567                         mseq = cap->mseq;
568                         issued |= cap->issued;
569                         flags |= CEPH_CAP_FLAG_AUTH;
570                 }
571         }
572
573         if (!ci->i_snap_realm) {
574                 /*
575                  * add this inode to the appropriate snap realm
576                  */
577                 struct ceph_snap_realm *realm = ceph_lookup_snap_realm(mdsc,
578                                                                realmino);
579                 if (realm) {
580                         spin_lock(&realm->inodes_with_caps_lock);
581                         ci->i_snap_realm = realm;
582                         list_add(&ci->i_snap_realm_item,
583                                  &realm->inodes_with_caps);
584                         spin_unlock(&realm->inodes_with_caps_lock);
585                 } else {
586                         pr_err("ceph_add_cap: couldn't find snap realm %llx\n",
587                                realmino);
588                         WARN_ON(!realm);
589                 }
590         }
591
592         __check_cap_issue(ci, cap, issued);
593
594         /*
595          * If we are issued caps we don't want, or the mds' wanted
596          * value appears to be off, queue a check so we'll release
597          * later and/or update the mds wanted value.
598          */
599         actual_wanted = __ceph_caps_wanted(ci);
600         if ((wanted & ~actual_wanted) ||
601             (issued & ~actual_wanted & CEPH_CAP_ANY_WR)) {
602                 dout(" issued %s, mds wanted %s, actual %s, queueing\n",
603                      ceph_cap_string(issued), ceph_cap_string(wanted),
604                      ceph_cap_string(actual_wanted));
605                 __cap_delay_requeue(mdsc, ci);
606         }
607
608         if (flags & CEPH_CAP_FLAG_AUTH) {
609                 if (ci->i_auth_cap == NULL ||
610                     ceph_seq_cmp(ci->i_auth_cap->mseq, mseq) < 0) {
611                         ci->i_auth_cap = cap;
612                         cap->mds_wanted = wanted;
613                 }
614         } else {
615                 WARN_ON(ci->i_auth_cap == cap);
616         }
617
618         dout("add_cap inode %p (%llx.%llx) cap %p %s now %s seq %d mds%d\n",
619              inode, ceph_vinop(inode), cap, ceph_cap_string(issued),
620              ceph_cap_string(issued|cap->issued), seq, mds);
621         cap->cap_id = cap_id;
622         cap->issued = issued;
623         cap->implemented |= issued;
624         if (ceph_seq_cmp(mseq, cap->mseq) > 0)
625                 cap->mds_wanted = wanted;
626         else
627                 cap->mds_wanted |= wanted;
628         cap->seq = seq;
629         cap->issue_seq = seq;
630         cap->mseq = mseq;
631         cap->cap_gen = session->s_cap_gen;
632
633         if (fmode >= 0)
634                 __ceph_get_fmode(ci, fmode);
635 }
636
637 /*
638  * Return true if cap has not timed out and belongs to the current
639  * generation of the MDS session (i.e. has not gone 'stale' due to
640  * us losing touch with the mds).
641  */
642 static int __cap_is_valid(struct ceph_cap *cap)
643 {
644         unsigned long ttl;
645         u32 gen;
646
647         spin_lock(&cap->session->s_gen_ttl_lock);
648         gen = cap->session->s_cap_gen;
649         ttl = cap->session->s_cap_ttl;
650         spin_unlock(&cap->session->s_gen_ttl_lock);
651
652         if (cap->cap_gen < gen || time_after_eq(jiffies, ttl)) {
653                 dout("__cap_is_valid %p cap %p issued %s "
654                      "but STALE (gen %u vs %u)\n", &cap->ci->vfs_inode,
655                      cap, ceph_cap_string(cap->issued), cap->cap_gen, gen);
656                 return 0;
657         }
658
659         return 1;
660 }
661
662 /*
663  * Return set of valid cap bits issued to us.  Note that caps time
664  * out, and may be invalidated in bulk if the client session times out
665  * and session->s_cap_gen is bumped.
666  */
667 int __ceph_caps_issued(struct ceph_inode_info *ci, int *implemented)
668 {
669         int have = ci->i_snap_caps;
670         struct ceph_cap *cap;
671         struct rb_node *p;
672
673         if (implemented)
674                 *implemented = 0;
675         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
676                 cap = rb_entry(p, struct ceph_cap, ci_node);
677                 if (!__cap_is_valid(cap))
678                         continue;
679                 dout("__ceph_caps_issued %p cap %p issued %s\n",
680                      &ci->vfs_inode, cap, ceph_cap_string(cap->issued));
681                 have |= cap->issued;
682                 if (implemented)
683                         *implemented |= cap->implemented;
684         }
685         /*
686          * exclude caps issued by non-auth MDS, but are been revoking
687          * by the auth MDS. The non-auth MDS should be revoking/exporting
688          * these caps, but the message is delayed.
689          */
690         if (ci->i_auth_cap) {
691                 cap = ci->i_auth_cap;
692                 have &= ~cap->implemented | cap->issued;
693         }
694         return have;
695 }
696
697 /*
698  * Get cap bits issued by caps other than @ocap
699  */
700 int __ceph_caps_issued_other(struct ceph_inode_info *ci, struct ceph_cap *ocap)
701 {
702         int have = ci->i_snap_caps;
703         struct ceph_cap *cap;
704         struct rb_node *p;
705
706         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
707                 cap = rb_entry(p, struct ceph_cap, ci_node);
708                 if (cap == ocap)
709                         continue;
710                 if (!__cap_is_valid(cap))
711                         continue;
712                 have |= cap->issued;
713         }
714         return have;
715 }
716
717 /*
718  * Move a cap to the end of the LRU (oldest caps at list head, newest
719  * at list tail).
720  */
721 static void __touch_cap(struct ceph_cap *cap)
722 {
723         struct ceph_mds_session *s = cap->session;
724
725         spin_lock(&s->s_cap_lock);
726         if (s->s_cap_iterator == NULL) {
727                 dout("__touch_cap %p cap %p mds%d\n", &cap->ci->vfs_inode, cap,
728                      s->s_mds);
729                 list_move_tail(&cap->session_caps, &s->s_caps);
730         } else {
731                 dout("__touch_cap %p cap %p mds%d NOP, iterating over caps\n",
732                      &cap->ci->vfs_inode, cap, s->s_mds);
733         }
734         spin_unlock(&s->s_cap_lock);
735 }
736
737 /*
738  * Check if we hold the given mask.  If so, move the cap(s) to the
739  * front of their respective LRUs.  (This is the preferred way for
740  * callers to check for caps they want.)
741  */
742 int __ceph_caps_issued_mask(struct ceph_inode_info *ci, int mask, int touch)
743 {
744         struct ceph_cap *cap;
745         struct rb_node *p;
746         int have = ci->i_snap_caps;
747
748         if ((have & mask) == mask) {
749                 dout("__ceph_caps_issued_mask %p snap issued %s"
750                      " (mask %s)\n", &ci->vfs_inode,
751                      ceph_cap_string(have),
752                      ceph_cap_string(mask));
753                 return 1;
754         }
755
756         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
757                 cap = rb_entry(p, struct ceph_cap, ci_node);
758                 if (!__cap_is_valid(cap))
759                         continue;
760                 if ((cap->issued & mask) == mask) {
761                         dout("__ceph_caps_issued_mask %p cap %p issued %s"
762                              " (mask %s)\n", &ci->vfs_inode, cap,
763                              ceph_cap_string(cap->issued),
764                              ceph_cap_string(mask));
765                         if (touch)
766                                 __touch_cap(cap);
767                         return 1;
768                 }
769
770                 /* does a combination of caps satisfy mask? */
771                 have |= cap->issued;
772                 if ((have & mask) == mask) {
773                         dout("__ceph_caps_issued_mask %p combo issued %s"
774                              " (mask %s)\n", &ci->vfs_inode,
775                              ceph_cap_string(cap->issued),
776                              ceph_cap_string(mask));
777                         if (touch) {
778                                 struct rb_node *q;
779
780                                 /* touch this + preceding caps */
781                                 __touch_cap(cap);
782                                 for (q = rb_first(&ci->i_caps); q != p;
783                                      q = rb_next(q)) {
784                                         cap = rb_entry(q, struct ceph_cap,
785                                                        ci_node);
786                                         if (!__cap_is_valid(cap))
787                                                 continue;
788                                         __touch_cap(cap);
789                                 }
790                         }
791                         return 1;
792                 }
793         }
794
795         return 0;
796 }
797
798 /*
799  * Return true if mask caps are currently being revoked by an MDS.
800  */
801 int __ceph_caps_revoking_other(struct ceph_inode_info *ci,
802                                struct ceph_cap *ocap, int mask)
803 {
804         struct ceph_cap *cap;
805         struct rb_node *p;
806
807         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
808                 cap = rb_entry(p, struct ceph_cap, ci_node);
809                 if (cap != ocap &&
810                     (cap->implemented & ~cap->issued & mask))
811                         return 1;
812         }
813         return 0;
814 }
815
816 int ceph_caps_revoking(struct ceph_inode_info *ci, int mask)
817 {
818         struct inode *inode = &ci->vfs_inode;
819         int ret;
820
821         spin_lock(&ci->i_ceph_lock);
822         ret = __ceph_caps_revoking_other(ci, NULL, mask);
823         spin_unlock(&ci->i_ceph_lock);
824         dout("ceph_caps_revoking %p %s = %d\n", inode,
825              ceph_cap_string(mask), ret);
826         return ret;
827 }
828
829 int __ceph_caps_used(struct ceph_inode_info *ci)
830 {
831         int used = 0;
832         if (ci->i_pin_ref)
833                 used |= CEPH_CAP_PIN;
834         if (ci->i_rd_ref)
835                 used |= CEPH_CAP_FILE_RD;
836         if (ci->i_rdcache_ref || ci->vfs_inode.i_data.nrpages)
837                 used |= CEPH_CAP_FILE_CACHE;
838         if (ci->i_wr_ref)
839                 used |= CEPH_CAP_FILE_WR;
840         if (ci->i_wb_ref || ci->i_wrbuffer_ref)
841                 used |= CEPH_CAP_FILE_BUFFER;
842         return used;
843 }
844
845 /*
846  * wanted, by virtue of open file modes
847  */
848 int __ceph_caps_file_wanted(struct ceph_inode_info *ci)
849 {
850         int want = 0;
851         int mode;
852         for (mode = 0; mode < CEPH_FILE_MODE_NUM; mode++)
853                 if (ci->i_nr_by_mode[mode])
854                         want |= ceph_caps_for_mode(mode);
855         return want;
856 }
857
858 /*
859  * Return caps we have registered with the MDS(s) as 'wanted'.
860  */
861 int __ceph_caps_mds_wanted(struct ceph_inode_info *ci)
862 {
863         struct ceph_cap *cap;
864         struct rb_node *p;
865         int mds_wanted = 0;
866
867         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
868                 cap = rb_entry(p, struct ceph_cap, ci_node);
869                 if (!__cap_is_valid(cap))
870                         continue;
871                 if (cap == ci->i_auth_cap)
872                         mds_wanted |= cap->mds_wanted;
873                 else
874                         mds_wanted |= (cap->mds_wanted & ~CEPH_CAP_ANY_FILE_WR);
875         }
876         return mds_wanted;
877 }
878
879 /*
880  * called under i_ceph_lock
881  */
882 static int __ceph_is_any_caps(struct ceph_inode_info *ci)
883 {
884         return !RB_EMPTY_ROOT(&ci->i_caps);
885 }
886
887 int ceph_is_any_caps(struct inode *inode)
888 {
889         struct ceph_inode_info *ci = ceph_inode(inode);
890         int ret;
891
892         spin_lock(&ci->i_ceph_lock);
893         ret = __ceph_is_any_caps(ci);
894         spin_unlock(&ci->i_ceph_lock);
895
896         return ret;
897 }
898
899 static void drop_inode_snap_realm(struct ceph_inode_info *ci)
900 {
901         struct ceph_snap_realm *realm = ci->i_snap_realm;
902         spin_lock(&realm->inodes_with_caps_lock);
903         list_del_init(&ci->i_snap_realm_item);
904         ci->i_snap_realm_counter++;
905         ci->i_snap_realm = NULL;
906         spin_unlock(&realm->inodes_with_caps_lock);
907         ceph_put_snap_realm(ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc,
908                             realm);
909 }
910
911 /*
912  * Remove a cap.  Take steps to deal with a racing iterate_session_caps.
913  *
914  * caller should hold i_ceph_lock.
915  * caller will not hold session s_mutex if called from destroy_inode.
916  */
917 void __ceph_remove_cap(struct ceph_cap *cap, bool queue_release)
918 {
919         struct ceph_mds_session *session = cap->session;
920         struct ceph_inode_info *ci = cap->ci;
921         struct ceph_mds_client *mdsc =
922                 ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc;
923         int removed = 0;
924
925         dout("__ceph_remove_cap %p from %p\n", cap, &ci->vfs_inode);
926
927         /* remove from session list */
928         spin_lock(&session->s_cap_lock);
929         /*
930          * s_cap_reconnect is protected by s_cap_lock. no one changes
931          * s_cap_gen while session is in the reconnect state.
932          */
933         if (queue_release &&
934             (!session->s_cap_reconnect ||
935              cap->cap_gen == session->s_cap_gen))
936                 __queue_cap_release(session, ci->i_vino.ino, cap->cap_id,
937                                     cap->mseq, cap->issue_seq);
938
939         if (session->s_cap_iterator == cap) {
940                 /* not yet, we are iterating over this very cap */
941                 dout("__ceph_remove_cap  delaying %p removal from session %p\n",
942                      cap, cap->session);
943         } else {
944                 list_del_init(&cap->session_caps);
945                 session->s_nr_caps--;
946                 cap->session = NULL;
947                 removed = 1;
948         }
949         /* protect backpointer with s_cap_lock: see iterate_session_caps */
950         cap->ci = NULL;
951         spin_unlock(&session->s_cap_lock);
952
953         /* remove from inode list */
954         rb_erase(&cap->ci_node, &ci->i_caps);
955         if (ci->i_auth_cap == cap)
956                 ci->i_auth_cap = NULL;
957
958         if (removed)
959                 ceph_put_cap(mdsc, cap);
960
961         /* when reconnect denied, we remove session caps forcibly,
962          * i_wr_ref can be non-zero. If there are ongoing write,
963          * keep i_snap_realm.
964          */
965         if (!__ceph_is_any_caps(ci) && ci->i_wr_ref == 0 && ci->i_snap_realm)
966                 drop_inode_snap_realm(ci);
967
968         if (!__ceph_is_any_real_caps(ci))
969                 __cap_delay_cancel(mdsc, ci);
970 }
971
972 /*
973  * Build and send a cap message to the given MDS.
974  *
975  * Caller should be holding s_mutex.
976  */
977 static int send_cap_msg(struct ceph_mds_session *session,
978                         u64 ino, u64 cid, int op,
979                         int caps, int wanted, int dirty,
980                         u32 seq, u64 flush_tid, u32 issue_seq, u32 mseq,
981                         u64 size, u64 max_size,
982                         struct timespec *mtime, struct timespec *atime,
983                         u64 time_warp_seq,
984                         kuid_t uid, kgid_t gid, umode_t mode,
985                         u64 xattr_version,
986                         struct ceph_buffer *xattrs_buf,
987                         u64 follows, bool inline_data)
988 {
989         struct ceph_mds_caps *fc;
990         struct ceph_msg *msg;
991         void *p;
992         size_t extra_len;
993
994         dout("send_cap_msg %s %llx %llx caps %s wanted %s dirty %s"
995              " seq %u/%u mseq %u follows %lld size %llu/%llu"
996              " xattr_ver %llu xattr_len %d\n", ceph_cap_op_name(op),
997              cid, ino, ceph_cap_string(caps), ceph_cap_string(wanted),
998              ceph_cap_string(dirty),
999              seq, issue_seq, mseq, follows, size, max_size,
1000              xattr_version, xattrs_buf ? (int)xattrs_buf->vec.iov_len : 0);
1001
1002         /* flock buffer size + inline version + inline data size */
1003         extra_len = 4 + 8 + 4;
1004         msg = ceph_msg_new(CEPH_MSG_CLIENT_CAPS, sizeof(*fc) + extra_len,
1005                            GFP_NOFS, false);
1006         if (!msg)
1007                 return -ENOMEM;
1008
1009         msg->hdr.tid = cpu_to_le64(flush_tid);
1010
1011         fc = msg->front.iov_base;
1012         memset(fc, 0, sizeof(*fc));
1013
1014         fc->cap_id = cpu_to_le64(cid);
1015         fc->op = cpu_to_le32(op);
1016         fc->seq = cpu_to_le32(seq);
1017         fc->issue_seq = cpu_to_le32(issue_seq);
1018         fc->migrate_seq = cpu_to_le32(mseq);
1019         fc->caps = cpu_to_le32(caps);
1020         fc->wanted = cpu_to_le32(wanted);
1021         fc->dirty = cpu_to_le32(dirty);
1022         fc->ino = cpu_to_le64(ino);
1023         fc->snap_follows = cpu_to_le64(follows);
1024
1025         fc->size = cpu_to_le64(size);
1026         fc->max_size = cpu_to_le64(max_size);
1027         if (mtime)
1028                 ceph_encode_timespec(&fc->mtime, mtime);
1029         if (atime)
1030                 ceph_encode_timespec(&fc->atime, atime);
1031         fc->time_warp_seq = cpu_to_le32(time_warp_seq);
1032
1033         fc->uid = cpu_to_le32(from_kuid(&init_user_ns, uid));
1034         fc->gid = cpu_to_le32(from_kgid(&init_user_ns, gid));
1035         fc->mode = cpu_to_le32(mode);
1036
1037         p = fc + 1;
1038         /* flock buffer size */
1039         ceph_encode_32(&p, 0);
1040         /* inline version */
1041         ceph_encode_64(&p, inline_data ? 0 : CEPH_INLINE_NONE);
1042         /* inline data size */
1043         ceph_encode_32(&p, 0);
1044
1045         fc->xattr_version = cpu_to_le64(xattr_version);
1046         if (xattrs_buf) {
1047                 msg->middle = ceph_buffer_get(xattrs_buf);
1048                 fc->xattr_len = cpu_to_le32(xattrs_buf->vec.iov_len);
1049                 msg->hdr.middle_len = cpu_to_le32(xattrs_buf->vec.iov_len);
1050         }
1051
1052         ceph_con_send(&session->s_con, msg);
1053         return 0;
1054 }
1055
1056 void __queue_cap_release(struct ceph_mds_session *session,
1057                          u64 ino, u64 cap_id, u32 migrate_seq,
1058                          u32 issue_seq)
1059 {
1060         struct ceph_msg *msg;
1061         struct ceph_mds_cap_release *head;
1062         struct ceph_mds_cap_item *item;
1063
1064         BUG_ON(!session->s_num_cap_releases);
1065         msg = list_first_entry(&session->s_cap_releases,
1066                                struct ceph_msg, list_head);
1067
1068         dout(" adding %llx release to mds%d msg %p (%d left)\n",
1069              ino, session->s_mds, msg, session->s_num_cap_releases);
1070
1071         BUG_ON(msg->front.iov_len + sizeof(*item) > PAGE_CACHE_SIZE);
1072         head = msg->front.iov_base;
1073         le32_add_cpu(&head->num, 1);
1074         item = msg->front.iov_base + msg->front.iov_len;
1075         item->ino = cpu_to_le64(ino);
1076         item->cap_id = cpu_to_le64(cap_id);
1077         item->migrate_seq = cpu_to_le32(migrate_seq);
1078         item->seq = cpu_to_le32(issue_seq);
1079
1080         session->s_num_cap_releases--;
1081
1082         msg->front.iov_len += sizeof(*item);
1083         if (le32_to_cpu(head->num) == CEPH_CAPS_PER_RELEASE) {
1084                 dout(" release msg %p full\n", msg);
1085                 list_move_tail(&msg->list_head, &session->s_cap_releases_done);
1086         } else {
1087                 dout(" release msg %p at %d/%d (%d)\n", msg,
1088                      (int)le32_to_cpu(head->num),
1089                      (int)CEPH_CAPS_PER_RELEASE,
1090                      (int)msg->front.iov_len);
1091         }
1092 }
1093
1094 /*
1095  * Queue cap releases when an inode is dropped from our cache.  Since
1096  * inode is about to be destroyed, there is no need for i_ceph_lock.
1097  */
1098 void ceph_queue_caps_release(struct inode *inode)
1099 {
1100         struct ceph_inode_info *ci = ceph_inode(inode);
1101         struct rb_node *p;
1102
1103         p = rb_first(&ci->i_caps);
1104         while (p) {
1105                 struct ceph_cap *cap = rb_entry(p, struct ceph_cap, ci_node);
1106                 p = rb_next(p);
1107                 __ceph_remove_cap(cap, true);
1108         }
1109 }
1110
1111 /*
1112  * Send a cap msg on the given inode.  Update our caps state, then
1113  * drop i_ceph_lock and send the message.
1114  *
1115  * Make note of max_size reported/requested from mds, revoked caps
1116  * that have now been implemented.
1117  *
1118  * Make half-hearted attempt ot to invalidate page cache if we are
1119  * dropping RDCACHE.  Note that this will leave behind locked pages
1120  * that we'll then need to deal with elsewhere.
1121  *
1122  * Return non-zero if delayed release, or we experienced an error
1123  * such that the caller should requeue + retry later.
1124  *
1125  * called with i_ceph_lock, then drops it.
1126  * caller should hold snap_rwsem (read), s_mutex.
1127  */
1128 static int __send_cap(struct ceph_mds_client *mdsc, struct ceph_cap *cap,
1129                       int op, int used, int want, int retain, int flushing,
1130                       unsigned *pflush_tid)
1131         __releases(cap->ci->i_ceph_lock)
1132 {
1133         struct ceph_inode_info *ci = cap->ci;
1134         struct inode *inode = &ci->vfs_inode;
1135         u64 cap_id = cap->cap_id;
1136         int held, revoking, dropping, keep;
1137         u64 seq, issue_seq, mseq, time_warp_seq, follows;
1138         u64 size, max_size;
1139         struct timespec mtime, atime;
1140         int wake = 0;
1141         umode_t mode;
1142         kuid_t uid;
1143         kgid_t gid;
1144         struct ceph_mds_session *session;
1145         u64 xattr_version = 0;
1146         struct ceph_buffer *xattr_blob = NULL;
1147         int delayed = 0;
1148         u64 flush_tid = 0;
1149         int i;
1150         int ret;
1151         bool inline_data;
1152
1153         held = cap->issued | cap->implemented;
1154         revoking = cap->implemented & ~cap->issued;
1155         retain &= ~revoking;
1156         dropping = cap->issued & ~retain;
1157
1158         dout("__send_cap %p cap %p session %p %s -> %s (revoking %s)\n",
1159              inode, cap, cap->session,
1160              ceph_cap_string(held), ceph_cap_string(held & retain),
1161              ceph_cap_string(revoking));
1162         BUG_ON((retain & CEPH_CAP_PIN) == 0);
1163
1164         session = cap->session;
1165
1166         /* don't release wanted unless we've waited a bit. */
1167         if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0 &&
1168             time_before(jiffies, ci->i_hold_caps_min)) {
1169                 dout(" delaying issued %s -> %s, wanted %s -> %s on send\n",
1170                      ceph_cap_string(cap->issued),
1171                      ceph_cap_string(cap->issued & retain),
1172                      ceph_cap_string(cap->mds_wanted),
1173                      ceph_cap_string(want));
1174                 want |= cap->mds_wanted;
1175                 retain |= cap->issued;
1176                 delayed = 1;
1177         }
1178         ci->i_ceph_flags &= ~(CEPH_I_NODELAY | CEPH_I_FLUSH);
1179
1180         cap->issued &= retain;  /* drop bits we don't want */
1181         if (cap->implemented & ~cap->issued) {
1182                 /*
1183                  * Wake up any waiters on wanted -> needed transition.
1184                  * This is due to the weird transition from buffered
1185                  * to sync IO... we need to flush dirty pages _before_
1186                  * allowing sync writes to avoid reordering.
1187                  */
1188                 wake = 1;
1189         }
1190         cap->implemented &= cap->issued | used;
1191         cap->mds_wanted = want;
1192
1193         if (flushing) {
1194                 /*
1195                  * assign a tid for flush operations so we can avoid
1196                  * flush1 -> dirty1 -> flush2 -> flushack1 -> mark
1197                  * clean type races.  track latest tid for every bit
1198                  * so we can handle flush AxFw, flush Fw, and have the
1199                  * first ack clean Ax.
1200                  */
1201                 flush_tid = ++ci->i_cap_flush_last_tid;
1202                 if (pflush_tid)
1203                         *pflush_tid = flush_tid;
1204                 dout(" cap_flush_tid %d\n", (int)flush_tid);
1205                 for (i = 0; i < CEPH_CAP_BITS; i++)
1206                         if (flushing & (1 << i))
1207                                 ci->i_cap_flush_tid[i] = flush_tid;
1208
1209                 follows = ci->i_head_snapc->seq;
1210         } else {
1211                 follows = 0;
1212         }
1213
1214         keep = cap->implemented;
1215         seq = cap->seq;
1216         issue_seq = cap->issue_seq;
1217         mseq = cap->mseq;
1218         size = inode->i_size;
1219         ci->i_reported_size = size;
1220         max_size = ci->i_wanted_max_size;
1221         ci->i_requested_max_size = max_size;
1222         mtime = inode->i_mtime;
1223         atime = inode->i_atime;
1224         time_warp_seq = ci->i_time_warp_seq;
1225         uid = inode->i_uid;
1226         gid = inode->i_gid;
1227         mode = inode->i_mode;
1228
1229         if (flushing & CEPH_CAP_XATTR_EXCL) {
1230                 __ceph_build_xattrs_blob(ci);
1231                 xattr_blob = ci->i_xattrs.blob;
1232                 xattr_version = ci->i_xattrs.version;
1233         }
1234
1235         inline_data = ci->i_inline_version != CEPH_INLINE_NONE;
1236
1237         spin_unlock(&ci->i_ceph_lock);
1238
1239         ret = send_cap_msg(session, ceph_vino(inode).ino, cap_id,
1240                 op, keep, want, flushing, seq, flush_tid, issue_seq, mseq,
1241                 size, max_size, &mtime, &atime, time_warp_seq,
1242                 uid, gid, mode, xattr_version, xattr_blob,
1243                 follows, inline_data);
1244         if (ret < 0) {
1245                 dout("error sending cap msg, must requeue %p\n", inode);
1246                 delayed = 1;
1247         }
1248
1249         if (wake)
1250                 wake_up_all(&ci->i_cap_wq);
1251
1252         return delayed;
1253 }
1254
1255 /*
1256  * When a snapshot is taken, clients accumulate dirty metadata on
1257  * inodes with capabilities in ceph_cap_snaps to describe the file
1258  * state at the time the snapshot was taken.  This must be flushed
1259  * asynchronously back to the MDS once sync writes complete and dirty
1260  * data is written out.
1261  *
1262  * Unless @again is true, skip cap_snaps that were already sent to
1263  * the MDS (i.e., during this session).
1264  *
1265  * Called under i_ceph_lock.  Takes s_mutex as needed.
1266  */
1267 void __ceph_flush_snaps(struct ceph_inode_info *ci,
1268                         struct ceph_mds_session **psession,
1269                         int again)
1270                 __releases(ci->i_ceph_lock)
1271                 __acquires(ci->i_ceph_lock)
1272 {
1273         struct inode *inode = &ci->vfs_inode;
1274         int mds;
1275         struct ceph_cap_snap *capsnap;
1276         u32 mseq;
1277         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
1278         struct ceph_mds_session *session = NULL; /* if session != NULL, we hold
1279                                                     session->s_mutex */
1280         u64 next_follows = 0;  /* keep track of how far we've gotten through the
1281                              i_cap_snaps list, and skip these entries next time
1282                              around to avoid an infinite loop */
1283
1284         if (psession)
1285                 session = *psession;
1286
1287         dout("__flush_snaps %p\n", inode);
1288 retry:
1289         list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
1290                 /* avoid an infiniute loop after retry */
1291                 if (capsnap->follows < next_follows)
1292                         continue;
1293                 /*
1294                  * we need to wait for sync writes to complete and for dirty
1295                  * pages to be written out.
1296                  */
1297                 if (capsnap->dirty_pages || capsnap->writing)
1298                         break;
1299
1300                 /* should be removed by ceph_try_drop_cap_snap() */
1301                 BUG_ON(!capsnap->need_flush);
1302
1303                 /* pick mds, take s_mutex */
1304                 if (ci->i_auth_cap == NULL) {
1305                         dout("no auth cap (migrating?), doing nothing\n");
1306                         goto out;
1307                 }
1308
1309                 /* only flush each capsnap once */
1310                 if (!again && !list_empty(&capsnap->flushing_item)) {
1311                         dout("already flushed %p, skipping\n", capsnap);
1312                         continue;
1313                 }
1314
1315                 mds = ci->i_auth_cap->session->s_mds;
1316                 mseq = ci->i_auth_cap->mseq;
1317
1318                 if (session && session->s_mds != mds) {
1319                         dout("oops, wrong session %p mutex\n", session);
1320                         mutex_unlock(&session->s_mutex);
1321                         ceph_put_mds_session(session);
1322                         session = NULL;
1323                 }
1324                 if (!session) {
1325                         spin_unlock(&ci->i_ceph_lock);
1326                         mutex_lock(&mdsc->mutex);
1327                         session = __ceph_lookup_mds_session(mdsc, mds);
1328                         mutex_unlock(&mdsc->mutex);
1329                         if (session) {
1330                                 dout("inverting session/ino locks on %p\n",
1331                                      session);
1332                                 mutex_lock(&session->s_mutex);
1333                         }
1334                         /*
1335                          * if session == NULL, we raced against a cap
1336                          * deletion or migration.  retry, and we'll
1337                          * get a better @mds value next time.
1338                          */
1339                         spin_lock(&ci->i_ceph_lock);
1340                         goto retry;
1341                 }
1342
1343                 capsnap->flush_tid = ++ci->i_cap_flush_last_tid;
1344                 atomic_inc(&capsnap->nref);
1345                 if (!list_empty(&capsnap->flushing_item))
1346                         list_del_init(&capsnap->flushing_item);
1347                 list_add_tail(&capsnap->flushing_item,
1348                               &session->s_cap_snaps_flushing);
1349                 spin_unlock(&ci->i_ceph_lock);
1350
1351                 dout("flush_snaps %p cap_snap %p follows %lld tid %llu\n",
1352                      inode, capsnap, capsnap->follows, capsnap->flush_tid);
1353                 send_cap_msg(session, ceph_vino(inode).ino, 0,
1354                              CEPH_CAP_OP_FLUSHSNAP, capsnap->issued, 0,
1355                              capsnap->dirty, 0, capsnap->flush_tid, 0, mseq,
1356                              capsnap->size, 0,
1357                              &capsnap->mtime, &capsnap->atime,
1358                              capsnap->time_warp_seq,
1359                              capsnap->uid, capsnap->gid, capsnap->mode,
1360                              capsnap->xattr_version, capsnap->xattr_blob,
1361                              capsnap->follows, capsnap->inline_data);
1362
1363                 next_follows = capsnap->follows + 1;
1364                 ceph_put_cap_snap(capsnap);
1365
1366                 spin_lock(&ci->i_ceph_lock);
1367                 goto retry;
1368         }
1369
1370         /* we flushed them all; remove this inode from the queue */
1371         spin_lock(&mdsc->snap_flush_lock);
1372         list_del_init(&ci->i_snap_flush_item);
1373         spin_unlock(&mdsc->snap_flush_lock);
1374
1375 out:
1376         if (psession)
1377                 *psession = session;
1378         else if (session) {
1379                 mutex_unlock(&session->s_mutex);
1380                 ceph_put_mds_session(session);
1381         }
1382 }
1383
1384 static void ceph_flush_snaps(struct ceph_inode_info *ci)
1385 {
1386         spin_lock(&ci->i_ceph_lock);
1387         __ceph_flush_snaps(ci, NULL, 0);
1388         spin_unlock(&ci->i_ceph_lock);
1389 }
1390
1391 /*
1392  * Mark caps dirty.  If inode is newly dirty, return the dirty flags.
1393  * Caller is then responsible for calling __mark_inode_dirty with the
1394  * returned flags value.
1395  */
1396 int __ceph_mark_dirty_caps(struct ceph_inode_info *ci, int mask)
1397 {
1398         struct ceph_mds_client *mdsc =
1399                 ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc;
1400         struct inode *inode = &ci->vfs_inode;
1401         int was = ci->i_dirty_caps;
1402         int dirty = 0;
1403
1404         if (!ci->i_auth_cap) {
1405                 pr_warn("__mark_dirty_caps %p %llx mask %s, "
1406                         "but no auth cap (session was closed?)\n",
1407                         inode, ceph_ino(inode), ceph_cap_string(mask));
1408                 return 0;
1409         }
1410
1411         dout("__mark_dirty_caps %p %s dirty %s -> %s\n", &ci->vfs_inode,
1412              ceph_cap_string(mask), ceph_cap_string(was),
1413              ceph_cap_string(was | mask));
1414         ci->i_dirty_caps |= mask;
1415         if (was == 0) {
1416                 if (!ci->i_head_snapc)
1417                         ci->i_head_snapc = ceph_get_snap_context(
1418                                 ci->i_snap_realm->cached_context);
1419                 dout(" inode %p now dirty snapc %p auth cap %p\n",
1420                      &ci->vfs_inode, ci->i_head_snapc, ci->i_auth_cap);
1421                 BUG_ON(!list_empty(&ci->i_dirty_item));
1422                 spin_lock(&mdsc->cap_dirty_lock);
1423                 list_add(&ci->i_dirty_item, &mdsc->cap_dirty);
1424                 spin_unlock(&mdsc->cap_dirty_lock);
1425                 if (ci->i_flushing_caps == 0) {
1426                         ihold(inode);
1427                         dirty |= I_DIRTY_SYNC;
1428                 }
1429         }
1430         BUG_ON(list_empty(&ci->i_dirty_item));
1431         if (((was | ci->i_flushing_caps) & CEPH_CAP_FILE_BUFFER) &&
1432             (mask & CEPH_CAP_FILE_BUFFER))
1433                 dirty |= I_DIRTY_DATASYNC;
1434         __cap_delay_requeue(mdsc, ci);
1435         return dirty;
1436 }
1437
1438 /*
1439  * Add dirty inode to the flushing list.  Assigned a seq number so we
1440  * can wait for caps to flush without starving.
1441  *
1442  * Called under i_ceph_lock.
1443  */
1444 static int __mark_caps_flushing(struct inode *inode,
1445                                  struct ceph_mds_session *session)
1446 {
1447         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
1448         struct ceph_inode_info *ci = ceph_inode(inode);
1449         int flushing;
1450
1451         BUG_ON(ci->i_dirty_caps == 0);
1452         BUG_ON(list_empty(&ci->i_dirty_item));
1453
1454         flushing = ci->i_dirty_caps;
1455         dout("__mark_caps_flushing flushing %s, flushing_caps %s -> %s\n",
1456              ceph_cap_string(flushing),
1457              ceph_cap_string(ci->i_flushing_caps),
1458              ceph_cap_string(ci->i_flushing_caps | flushing));
1459         ci->i_flushing_caps |= flushing;
1460         ci->i_dirty_caps = 0;
1461         dout(" inode %p now !dirty\n", inode);
1462
1463         spin_lock(&mdsc->cap_dirty_lock);
1464         list_del_init(&ci->i_dirty_item);
1465
1466         if (list_empty(&ci->i_flushing_item)) {
1467                 ci->i_cap_flush_seq = ++mdsc->cap_flush_seq;
1468                 list_add_tail(&ci->i_flushing_item, &session->s_cap_flushing);
1469                 mdsc->num_cap_flushing++;
1470                 dout(" inode %p now flushing seq %lld\n", inode,
1471                      ci->i_cap_flush_seq);
1472         } else {
1473                 list_move_tail(&ci->i_flushing_item, &session->s_cap_flushing);
1474                 dout(" inode %p now flushing (more) seq %lld\n", inode,
1475                      ci->i_cap_flush_seq);
1476         }
1477         spin_unlock(&mdsc->cap_dirty_lock);
1478
1479         return flushing;
1480 }
1481
1482 /*
1483  * try to invalidate mapping pages without blocking.
1484  */
1485 static int try_nonblocking_invalidate(struct inode *inode)
1486 {
1487         struct ceph_inode_info *ci = ceph_inode(inode);
1488         u32 invalidating_gen = ci->i_rdcache_gen;
1489
1490         spin_unlock(&ci->i_ceph_lock);
1491         invalidate_mapping_pages(&inode->i_data, 0, -1);
1492         spin_lock(&ci->i_ceph_lock);
1493
1494         if (inode->i_data.nrpages == 0 &&
1495             invalidating_gen == ci->i_rdcache_gen) {
1496                 /* success. */
1497                 dout("try_nonblocking_invalidate %p success\n", inode);
1498                 /* save any racing async invalidate some trouble */
1499                 ci->i_rdcache_revoking = ci->i_rdcache_gen - 1;
1500                 return 0;
1501         }
1502         dout("try_nonblocking_invalidate %p failed\n", inode);
1503         return -1;
1504 }
1505
1506 /*
1507  * Swiss army knife function to examine currently used and wanted
1508  * versus held caps.  Release, flush, ack revoked caps to mds as
1509  * appropriate.
1510  *
1511  *  CHECK_CAPS_NODELAY - caller is delayed work and we should not delay
1512  *    cap release further.
1513  *  CHECK_CAPS_AUTHONLY - we should only check the auth cap
1514  *  CHECK_CAPS_FLUSH - we should flush any dirty caps immediately, without
1515  *    further delay.
1516  */
1517 void ceph_check_caps(struct ceph_inode_info *ci, int flags,
1518                      struct ceph_mds_session *session)
1519 {
1520         struct ceph_fs_client *fsc = ceph_inode_to_client(&ci->vfs_inode);
1521         struct ceph_mds_client *mdsc = fsc->mdsc;
1522         struct inode *inode = &ci->vfs_inode;
1523         struct ceph_cap *cap;
1524         int file_wanted, used, cap_used;
1525         int took_snap_rwsem = 0;             /* true if mdsc->snap_rwsem held */
1526         int issued, implemented, want, retain, revoking, flushing = 0;
1527         int mds = -1;   /* keep track of how far we've gone through i_caps list
1528                            to avoid an infinite loop on retry */
1529         struct rb_node *p;
1530         int tried_invalidate = 0;
1531         int delayed = 0, sent = 0, force_requeue = 0, num;
1532         int queue_invalidate = 0;
1533         int is_delayed = flags & CHECK_CAPS_NODELAY;
1534
1535         /* if we are unmounting, flush any unused caps immediately. */
1536         if (mdsc->stopping)
1537                 is_delayed = 1;
1538
1539         spin_lock(&ci->i_ceph_lock);
1540
1541         if (ci->i_ceph_flags & CEPH_I_FLUSH)
1542                 flags |= CHECK_CAPS_FLUSH;
1543
1544         /* flush snaps first time around only */
1545         if (!list_empty(&ci->i_cap_snaps))
1546                 __ceph_flush_snaps(ci, &session, 0);
1547         goto retry_locked;
1548 retry:
1549         spin_lock(&ci->i_ceph_lock);
1550 retry_locked:
1551         file_wanted = __ceph_caps_file_wanted(ci);
1552         used = __ceph_caps_used(ci);
1553         want = file_wanted | used;
1554         issued = __ceph_caps_issued(ci, &implemented);
1555         revoking = implemented & ~issued;
1556
1557         retain = want | CEPH_CAP_PIN;
1558         if (!mdsc->stopping && inode->i_nlink > 0) {
1559                 if (want) {
1560                         retain |= CEPH_CAP_ANY;       /* be greedy */
1561                 } else if (S_ISDIR(inode->i_mode) &&
1562                            (issued & CEPH_CAP_FILE_SHARED) &&
1563                             __ceph_dir_is_complete(ci)) {
1564                         /*
1565                          * If a directory is complete, we want to keep
1566                          * the exclusive cap. So that MDS does not end up
1567                          * revoking the shared cap on every create/unlink
1568                          * operation.
1569                          */
1570                         want = CEPH_CAP_ANY_SHARED | CEPH_CAP_FILE_EXCL;
1571                         retain |= want;
1572                 } else {
1573
1574                         retain |= CEPH_CAP_ANY_SHARED;
1575                         /*
1576                          * keep RD only if we didn't have the file open RW,
1577                          * because then the mds would revoke it anyway to
1578                          * journal max_size=0.
1579                          */
1580                         if (ci->i_max_size == 0)
1581                                 retain |= CEPH_CAP_ANY_RD;
1582                 }
1583         }
1584
1585         dout("check_caps %p file_want %s used %s dirty %s flushing %s"
1586              " issued %s revoking %s retain %s %s%s%s\n", inode,
1587              ceph_cap_string(file_wanted),
1588              ceph_cap_string(used), ceph_cap_string(ci->i_dirty_caps),
1589              ceph_cap_string(ci->i_flushing_caps),
1590              ceph_cap_string(issued), ceph_cap_string(revoking),
1591              ceph_cap_string(retain),
1592              (flags & CHECK_CAPS_AUTHONLY) ? " AUTHONLY" : "",
1593              (flags & CHECK_CAPS_NODELAY) ? " NODELAY" : "",
1594              (flags & CHECK_CAPS_FLUSH) ? " FLUSH" : "");
1595
1596         /*
1597          * If we no longer need to hold onto old our caps, and we may
1598          * have cached pages, but don't want them, then try to invalidate.
1599          * If we fail, it's because pages are locked.... try again later.
1600          */
1601         if ((!is_delayed || mdsc->stopping) &&
1602             ci->i_wrbuffer_ref == 0 &&               /* no dirty pages... */
1603             inode->i_data.nrpages &&                 /* have cached pages */
1604             (file_wanted == 0 ||                     /* no open files */
1605              (revoking & (CEPH_CAP_FILE_CACHE|
1606                           CEPH_CAP_FILE_LAZYIO))) && /*  or revoking cache */
1607             !tried_invalidate) {
1608                 dout("check_caps trying to invalidate on %p\n", inode);
1609                 if (try_nonblocking_invalidate(inode) < 0) {
1610                         if (revoking & (CEPH_CAP_FILE_CACHE|
1611                                         CEPH_CAP_FILE_LAZYIO)) {
1612                                 dout("check_caps queuing invalidate\n");
1613                                 queue_invalidate = 1;
1614                                 ci->i_rdcache_revoking = ci->i_rdcache_gen;
1615                         } else {
1616                                 dout("check_caps failed to invalidate pages\n");
1617                                 /* we failed to invalidate pages.  check these
1618                                    caps again later. */
1619                                 force_requeue = 1;
1620                                 __cap_set_timeouts(mdsc, ci);
1621                         }
1622                 }
1623                 tried_invalidate = 1;
1624                 goto retry_locked;
1625         }
1626
1627         num = 0;
1628         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
1629                 cap = rb_entry(p, struct ceph_cap, ci_node);
1630                 num++;
1631
1632                 /* avoid looping forever */
1633                 if (mds >= cap->mds ||
1634                     ((flags & CHECK_CAPS_AUTHONLY) && cap != ci->i_auth_cap))
1635                         continue;
1636
1637                 /* NOTE: no side-effects allowed, until we take s_mutex */
1638
1639                 cap_used = used;
1640                 if (ci->i_auth_cap && cap != ci->i_auth_cap)
1641                         cap_used &= ~ci->i_auth_cap->issued;
1642
1643                 revoking = cap->implemented & ~cap->issued;
1644                 dout(" mds%d cap %p used %s issued %s implemented %s revoking %s\n",
1645                      cap->mds, cap, ceph_cap_string(cap->issued),
1646                      ceph_cap_string(cap_used),
1647                      ceph_cap_string(cap->implemented),
1648                      ceph_cap_string(revoking));
1649
1650                 if (cap == ci->i_auth_cap &&
1651                     (cap->issued & CEPH_CAP_FILE_WR)) {
1652                         /* request larger max_size from MDS? */
1653                         if (ci->i_wanted_max_size > ci->i_max_size &&
1654                             ci->i_wanted_max_size > ci->i_requested_max_size) {
1655                                 dout("requesting new max_size\n");
1656                                 goto ack;
1657                         }
1658
1659                         /* approaching file_max? */
1660                         if ((inode->i_size << 1) >= ci->i_max_size &&
1661                             (ci->i_reported_size << 1) < ci->i_max_size) {
1662                                 dout("i_size approaching max_size\n");
1663                                 goto ack;
1664                         }
1665                 }
1666                 /* flush anything dirty? */
1667                 if (cap == ci->i_auth_cap && (flags & CHECK_CAPS_FLUSH) &&
1668                     ci->i_dirty_caps) {
1669                         dout("flushing dirty caps\n");
1670                         goto ack;
1671                 }
1672
1673                 /* completed revocation? going down and there are no caps? */
1674                 if (revoking && (revoking & cap_used) == 0) {
1675                         dout("completed revocation of %s\n",
1676                              ceph_cap_string(cap->implemented & ~cap->issued));
1677                         goto ack;
1678                 }
1679
1680                 /* want more caps from mds? */
1681                 if (want & ~(cap->mds_wanted | cap->issued))
1682                         goto ack;
1683
1684                 /* things we might delay */
1685                 if ((cap->issued & ~retain) == 0 &&
1686                     cap->mds_wanted == want)
1687                         continue;     /* nope, all good */
1688
1689                 if (is_delayed)
1690                         goto ack;
1691
1692                 /* delay? */
1693                 if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0 &&
1694                     time_before(jiffies, ci->i_hold_caps_max)) {
1695                         dout(" delaying issued %s -> %s, wanted %s -> %s\n",
1696                              ceph_cap_string(cap->issued),
1697                              ceph_cap_string(cap->issued & retain),
1698                              ceph_cap_string(cap->mds_wanted),
1699                              ceph_cap_string(want));
1700                         delayed++;
1701                         continue;
1702                 }
1703
1704 ack:
1705                 if (ci->i_ceph_flags & CEPH_I_NOFLUSH) {
1706                         dout(" skipping %p I_NOFLUSH set\n", inode);
1707                         continue;
1708                 }
1709
1710                 if (session && session != cap->session) {
1711                         dout("oops, wrong session %p mutex\n", session);
1712                         mutex_unlock(&session->s_mutex);
1713                         session = NULL;
1714                 }
1715                 if (!session) {
1716                         session = cap->session;
1717                         if (mutex_trylock(&session->s_mutex) == 0) {
1718                                 dout("inverting session/ino locks on %p\n",
1719                                      session);
1720                                 spin_unlock(&ci->i_ceph_lock);
1721                                 if (took_snap_rwsem) {
1722                                         up_read(&mdsc->snap_rwsem);
1723                                         took_snap_rwsem = 0;
1724                                 }
1725                                 mutex_lock(&session->s_mutex);
1726                                 goto retry;
1727                         }
1728                 }
1729                 /* take snap_rwsem after session mutex */
1730                 if (!took_snap_rwsem) {
1731                         if (down_read_trylock(&mdsc->snap_rwsem) == 0) {
1732                                 dout("inverting snap/in locks on %p\n",
1733                                      inode);
1734                                 spin_unlock(&ci->i_ceph_lock);
1735                                 down_read(&mdsc->snap_rwsem);
1736                                 took_snap_rwsem = 1;
1737                                 goto retry;
1738                         }
1739                         took_snap_rwsem = 1;
1740                 }
1741
1742                 if (cap == ci->i_auth_cap && ci->i_dirty_caps)
1743                         flushing = __mark_caps_flushing(inode, session);
1744                 else
1745                         flushing = 0;
1746
1747                 mds = cap->mds;  /* remember mds, so we don't repeat */
1748                 sent++;
1749
1750                 /* __send_cap drops i_ceph_lock */
1751                 delayed += __send_cap(mdsc, cap, CEPH_CAP_OP_UPDATE, cap_used,
1752                                       want, retain, flushing, NULL);
1753                 goto retry; /* retake i_ceph_lock and restart our cap scan. */
1754         }
1755
1756         /*
1757          * Reschedule delayed caps release if we delayed anything,
1758          * otherwise cancel.
1759          */
1760         if (delayed && is_delayed)
1761                 force_requeue = 1;   /* __send_cap delayed release; requeue */
1762         if (!delayed && !is_delayed)
1763                 __cap_delay_cancel(mdsc, ci);
1764         else if (!is_delayed || force_requeue)
1765                 __cap_delay_requeue(mdsc, ci);
1766
1767         spin_unlock(&ci->i_ceph_lock);
1768
1769         if (queue_invalidate)
1770                 ceph_queue_invalidate(inode);
1771
1772         if (session)
1773                 mutex_unlock(&session->s_mutex);
1774         if (took_snap_rwsem)
1775                 up_read(&mdsc->snap_rwsem);
1776 }
1777
1778 /*
1779  * Try to flush dirty caps back to the auth mds.
1780  */
1781 static int try_flush_caps(struct inode *inode, unsigned *flush_tid)
1782 {
1783         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
1784         struct ceph_inode_info *ci = ceph_inode(inode);
1785         int flushing = 0;
1786         struct ceph_mds_session *session = NULL;
1787
1788 retry:
1789         spin_lock(&ci->i_ceph_lock);
1790         if (ci->i_ceph_flags & CEPH_I_NOFLUSH) {
1791                 dout("try_flush_caps skipping %p I_NOFLUSH set\n", inode);
1792                 goto out;
1793         }
1794         if (ci->i_dirty_caps && ci->i_auth_cap) {
1795                 struct ceph_cap *cap = ci->i_auth_cap;
1796                 int used = __ceph_caps_used(ci);
1797                 int want = __ceph_caps_wanted(ci);
1798                 int delayed;
1799
1800                 if (!session || session != cap->session) {
1801                         spin_unlock(&ci->i_ceph_lock);
1802                         if (session)
1803                                 mutex_unlock(&session->s_mutex);
1804                         session = cap->session;
1805                         mutex_lock(&session->s_mutex);
1806                         goto retry;
1807                 }
1808                 if (cap->session->s_state < CEPH_MDS_SESSION_OPEN)
1809                         goto out;
1810
1811                 flushing = __mark_caps_flushing(inode, session);
1812
1813                 /* __send_cap drops i_ceph_lock */
1814                 delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH, used, want,
1815                                      cap->issued | cap->implemented, flushing,
1816                                      flush_tid);
1817                 if (!delayed)
1818                         goto out_unlocked;
1819
1820                 spin_lock(&ci->i_ceph_lock);
1821                 __cap_delay_requeue(mdsc, ci);
1822         }
1823 out:
1824         spin_unlock(&ci->i_ceph_lock);
1825 out_unlocked:
1826         if (session)
1827                 mutex_unlock(&session->s_mutex);
1828         return flushing;
1829 }
1830
1831 /*
1832  * Return true if we've flushed caps through the given flush_tid.
1833  */
1834 static int caps_are_flushed(struct inode *inode, unsigned tid)
1835 {
1836         struct ceph_inode_info *ci = ceph_inode(inode);
1837         int i, ret = 1;
1838
1839         spin_lock(&ci->i_ceph_lock);
1840         for (i = 0; i < CEPH_CAP_BITS; i++)
1841                 if ((ci->i_flushing_caps & (1 << i)) &&
1842                     ci->i_cap_flush_tid[i] <= tid) {
1843                         /* still flushing this bit */
1844                         ret = 0;
1845                         break;
1846                 }
1847         spin_unlock(&ci->i_ceph_lock);
1848         return ret;
1849 }
1850
1851 /*
1852  * Wait on any unsafe replies for the given inode.  First wait on the
1853  * newest request, and make that the upper bound.  Then, if there are
1854  * more requests, keep waiting on the oldest as long as it is still older
1855  * than the original request.
1856  */
1857 static void sync_write_wait(struct inode *inode)
1858 {
1859         struct ceph_inode_info *ci = ceph_inode(inode);
1860         struct list_head *head = &ci->i_unsafe_writes;
1861         struct ceph_osd_request *req;
1862         u64 last_tid;
1863
1864         spin_lock(&ci->i_unsafe_lock);
1865         if (list_empty(head))
1866                 goto out;
1867
1868         /* set upper bound as _last_ entry in chain */
1869         req = list_entry(head->prev, struct ceph_osd_request,
1870                          r_unsafe_item);
1871         last_tid = req->r_tid;
1872
1873         do {
1874                 ceph_osdc_get_request(req);
1875                 spin_unlock(&ci->i_unsafe_lock);
1876                 dout("sync_write_wait on tid %llu (until %llu)\n",
1877                      req->r_tid, last_tid);
1878                 wait_for_completion(&req->r_safe_completion);
1879                 spin_lock(&ci->i_unsafe_lock);
1880                 ceph_osdc_put_request(req);
1881
1882                 /*
1883                  * from here on look at first entry in chain, since we
1884                  * only want to wait for anything older than last_tid
1885                  */
1886                 if (list_empty(head))
1887                         break;
1888                 req = list_entry(head->next, struct ceph_osd_request,
1889                                  r_unsafe_item);
1890         } while (req->r_tid < last_tid);
1891 out:
1892         spin_unlock(&ci->i_unsafe_lock);
1893 }
1894
1895 int ceph_fsync(struct file *file, loff_t start, loff_t end, int datasync)
1896 {
1897         struct inode *inode = file->f_mapping->host;
1898         struct ceph_inode_info *ci = ceph_inode(inode);
1899         unsigned flush_tid;
1900         int ret;
1901         int dirty;
1902
1903         dout("fsync %p%s\n", inode, datasync ? " datasync" : "");
1904         sync_write_wait(inode);
1905
1906         ret = filemap_write_and_wait_range(inode->i_mapping, start, end);
1907         if (ret < 0)
1908                 return ret;
1909         mutex_lock(&inode->i_mutex);
1910
1911         dirty = try_flush_caps(inode, &flush_tid);
1912         dout("fsync dirty caps are %s\n", ceph_cap_string(dirty));
1913
1914         /*
1915          * only wait on non-file metadata writeback (the mds
1916          * can recover size and mtime, so we don't need to
1917          * wait for that)
1918          */
1919         if (!datasync && (dirty & ~CEPH_CAP_ANY_FILE_WR)) {
1920                 dout("fsync waiting for flush_tid %u\n", flush_tid);
1921                 ret = wait_event_interruptible(ci->i_cap_wq,
1922                                        caps_are_flushed(inode, flush_tid));
1923         }
1924
1925         dout("fsync %p%s done\n", inode, datasync ? " datasync" : "");
1926         mutex_unlock(&inode->i_mutex);
1927         return ret;
1928 }
1929
1930 /*
1931  * Flush any dirty caps back to the mds.  If we aren't asked to wait,
1932  * queue inode for flush but don't do so immediately, because we can
1933  * get by with fewer MDS messages if we wait for data writeback to
1934  * complete first.
1935  */
1936 int ceph_write_inode(struct inode *inode, struct writeback_control *wbc)
1937 {
1938         struct ceph_inode_info *ci = ceph_inode(inode);
1939         unsigned flush_tid;
1940         int err = 0;
1941         int dirty;
1942         int wait = wbc->sync_mode == WB_SYNC_ALL;
1943
1944         dout("write_inode %p wait=%d\n", inode, wait);
1945         if (wait) {
1946                 dirty = try_flush_caps(inode, &flush_tid);
1947                 if (dirty)
1948                         err = wait_event_interruptible(ci->i_cap_wq,
1949                                        caps_are_flushed(inode, flush_tid));
1950         } else {
1951                 struct ceph_mds_client *mdsc =
1952                         ceph_sb_to_client(inode->i_sb)->mdsc;
1953
1954                 spin_lock(&ci->i_ceph_lock);
1955                 if (__ceph_caps_dirty(ci))
1956                         __cap_delay_requeue_front(mdsc, ci);
1957                 spin_unlock(&ci->i_ceph_lock);
1958         }
1959         return err;
1960 }
1961
1962 /*
1963  * After a recovering MDS goes active, we need to resend any caps
1964  * we were flushing.
1965  *
1966  * Caller holds session->s_mutex.
1967  */
1968 static void kick_flushing_capsnaps(struct ceph_mds_client *mdsc,
1969                                    struct ceph_mds_session *session)
1970 {
1971         struct ceph_cap_snap *capsnap;
1972
1973         dout("kick_flushing_capsnaps mds%d\n", session->s_mds);
1974         list_for_each_entry(capsnap, &session->s_cap_snaps_flushing,
1975                             flushing_item) {
1976                 struct ceph_inode_info *ci = capsnap->ci;
1977                 struct inode *inode = &ci->vfs_inode;
1978                 struct ceph_cap *cap;
1979
1980                 spin_lock(&ci->i_ceph_lock);
1981                 cap = ci->i_auth_cap;
1982                 if (cap && cap->session == session) {
1983                         dout("kick_flushing_caps %p cap %p capsnap %p\n", inode,
1984                              cap, capsnap);
1985                         __ceph_flush_snaps(ci, &session, 1);
1986                 } else {
1987                         pr_err("%p auth cap %p not mds%d ???\n", inode,
1988                                cap, session->s_mds);
1989                 }
1990                 spin_unlock(&ci->i_ceph_lock);
1991         }
1992 }
1993
1994 void ceph_kick_flushing_caps(struct ceph_mds_client *mdsc,
1995                              struct ceph_mds_session *session)
1996 {
1997         struct ceph_inode_info *ci;
1998
1999         kick_flushing_capsnaps(mdsc, session);
2000
2001         dout("kick_flushing_caps mds%d\n", session->s_mds);
2002         list_for_each_entry(ci, &session->s_cap_flushing, i_flushing_item) {
2003                 struct inode *inode = &ci->vfs_inode;
2004                 struct ceph_cap *cap;
2005                 int delayed = 0;
2006
2007                 spin_lock(&ci->i_ceph_lock);
2008                 cap = ci->i_auth_cap;
2009                 if (cap && cap->session == session) {
2010                         dout("kick_flushing_caps %p cap %p %s\n", inode,
2011                              cap, ceph_cap_string(ci->i_flushing_caps));
2012                         delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH,
2013                                              __ceph_caps_used(ci),
2014                                              __ceph_caps_wanted(ci),
2015                                              cap->issued | cap->implemented,
2016                                              ci->i_flushing_caps, NULL);
2017                         if (delayed) {
2018                                 spin_lock(&ci->i_ceph_lock);
2019                                 __cap_delay_requeue(mdsc, ci);
2020                                 spin_unlock(&ci->i_ceph_lock);
2021                         }
2022                 } else {
2023                         pr_err("%p auth cap %p not mds%d ???\n", inode,
2024                                cap, session->s_mds);
2025                         spin_unlock(&ci->i_ceph_lock);
2026                 }
2027         }
2028 }
2029
2030 static void kick_flushing_inode_caps(struct ceph_mds_client *mdsc,
2031                                      struct ceph_mds_session *session,
2032                                      struct inode *inode)
2033 {
2034         struct ceph_inode_info *ci = ceph_inode(inode);
2035         struct ceph_cap *cap;
2036         int delayed = 0;
2037
2038         spin_lock(&ci->i_ceph_lock);
2039         cap = ci->i_auth_cap;
2040         dout("kick_flushing_inode_caps %p flushing %s flush_seq %lld\n", inode,
2041              ceph_cap_string(ci->i_flushing_caps), ci->i_cap_flush_seq);
2042
2043         __ceph_flush_snaps(ci, &session, 1);
2044
2045         if (ci->i_flushing_caps) {
2046                 spin_lock(&mdsc->cap_dirty_lock);
2047                 list_move_tail(&ci->i_flushing_item,
2048                                &cap->session->s_cap_flushing);
2049                 spin_unlock(&mdsc->cap_dirty_lock);
2050
2051                 delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH,
2052                                      __ceph_caps_used(ci),
2053                                      __ceph_caps_wanted(ci),
2054                                      cap->issued | cap->implemented,
2055                                      ci->i_flushing_caps, NULL);
2056                 if (delayed) {
2057                         spin_lock(&ci->i_ceph_lock);
2058                         __cap_delay_requeue(mdsc, ci);
2059                         spin_unlock(&ci->i_ceph_lock);
2060                 }
2061         } else {
2062                 spin_unlock(&ci->i_ceph_lock);
2063         }
2064 }
2065
2066
2067 /*
2068  * Take references to capabilities we hold, so that we don't release
2069  * them to the MDS prematurely.
2070  *
2071  * Protected by i_ceph_lock.
2072  */
2073 static void __take_cap_refs(struct ceph_inode_info *ci, int got,
2074                             bool snap_rwsem_locked)
2075 {
2076         if (got & CEPH_CAP_PIN)
2077                 ci->i_pin_ref++;
2078         if (got & CEPH_CAP_FILE_RD)
2079                 ci->i_rd_ref++;
2080         if (got & CEPH_CAP_FILE_CACHE)
2081                 ci->i_rdcache_ref++;
2082         if (got & CEPH_CAP_FILE_WR) {
2083                 if (ci->i_wr_ref == 0 && !ci->i_head_snapc) {
2084                         BUG_ON(!snap_rwsem_locked);
2085                         ci->i_head_snapc = ceph_get_snap_context(
2086                                         ci->i_snap_realm->cached_context);
2087                 }
2088                 ci->i_wr_ref++;
2089         }
2090         if (got & CEPH_CAP_FILE_BUFFER) {
2091                 if (ci->i_wb_ref == 0)
2092                         ihold(&ci->vfs_inode);
2093                 ci->i_wb_ref++;
2094                 dout("__take_cap_refs %p wb %d -> %d (?)\n",
2095                      &ci->vfs_inode, ci->i_wb_ref-1, ci->i_wb_ref);
2096         }
2097 }
2098
2099 /*
2100  * Try to grab cap references.  Specify those refs we @want, and the
2101  * minimal set we @need.  Also include the larger offset we are writing
2102  * to (when applicable), and check against max_size here as well.
2103  * Note that caller is responsible for ensuring max_size increases are
2104  * requested from the MDS.
2105  */
2106 static int try_get_cap_refs(struct ceph_inode_info *ci, int need, int want,
2107                             loff_t endoff, bool nonblock, int *got, int *err)
2108 {
2109         struct inode *inode = &ci->vfs_inode;
2110         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
2111         int ret = 0;
2112         int have, implemented;
2113         int file_wanted;
2114         bool snap_rwsem_locked = false;
2115
2116         dout("get_cap_refs %p need %s want %s\n", inode,
2117              ceph_cap_string(need), ceph_cap_string(want));
2118
2119 again:
2120         spin_lock(&ci->i_ceph_lock);
2121
2122         /* make sure file is actually open */
2123         file_wanted = __ceph_caps_file_wanted(ci);
2124         if ((file_wanted & need) == 0) {
2125                 dout("try_get_cap_refs need %s file_wanted %s, EBADF\n",
2126                      ceph_cap_string(need), ceph_cap_string(file_wanted));
2127                 *err = -EBADF;
2128                 ret = 1;
2129                 goto out_unlock;
2130         }
2131
2132         /* finish pending truncate */
2133         while (ci->i_truncate_pending) {
2134                 spin_unlock(&ci->i_ceph_lock);
2135                 if (snap_rwsem_locked) {
2136                         up_read(&mdsc->snap_rwsem);
2137                         snap_rwsem_locked = false;
2138                 }
2139                 __ceph_do_pending_vmtruncate(inode);
2140                 spin_lock(&ci->i_ceph_lock);
2141         }
2142
2143         have = __ceph_caps_issued(ci, &implemented);
2144
2145         if (have & need & CEPH_CAP_FILE_WR) {
2146                 if (endoff >= 0 && endoff > (loff_t)ci->i_max_size) {
2147                         dout("get_cap_refs %p endoff %llu > maxsize %llu\n",
2148                              inode, endoff, ci->i_max_size);
2149                         if (endoff > ci->i_requested_max_size) {
2150                                 *err = -EAGAIN;
2151                                 ret = 1;
2152                         }
2153                         goto out_unlock;
2154                 }
2155                 /*
2156                  * If a sync write is in progress, we must wait, so that we
2157                  * can get a final snapshot value for size+mtime.
2158                  */
2159                 if (__ceph_have_pending_cap_snap(ci)) {
2160                         dout("get_cap_refs %p cap_snap_pending\n", inode);
2161                         goto out_unlock;
2162                 }
2163         }
2164
2165         if ((have & need) == need) {
2166                 /*
2167                  * Look at (implemented & ~have & not) so that we keep waiting
2168                  * on transition from wanted -> needed caps.  This is needed
2169                  * for WRBUFFER|WR -> WR to avoid a new WR sync write from
2170                  * going before a prior buffered writeback happens.
2171                  */
2172                 int not = want & ~(have & need);
2173                 int revoking = implemented & ~have;
2174                 dout("get_cap_refs %p have %s but not %s (revoking %s)\n",
2175                      inode, ceph_cap_string(have), ceph_cap_string(not),
2176                      ceph_cap_string(revoking));
2177                 if ((revoking & not) == 0) {
2178                         if (!snap_rwsem_locked &&
2179                             !ci->i_head_snapc &&
2180                             (need & CEPH_CAP_FILE_WR)) {
2181                                 if (!down_read_trylock(&mdsc->snap_rwsem)) {
2182                                         /*
2183                                          * we can not call down_read() when
2184                                          * task isn't in TASK_RUNNING state
2185                                          */
2186                                         if (nonblock) {
2187                                                 *err = -EAGAIN;
2188                                                 ret = 1;
2189                                                 goto out_unlock;
2190                                         }
2191
2192                                         spin_unlock(&ci->i_ceph_lock);
2193                                         down_read(&mdsc->snap_rwsem);
2194                                         snap_rwsem_locked = true;
2195                                         goto again;
2196                                 }
2197                                 snap_rwsem_locked = true;
2198                         }
2199                         *got = need | (have & want);
2200                         __take_cap_refs(ci, *got, true);
2201                         ret = 1;
2202                 }
2203         } else {
2204                 int session_readonly = false;
2205                 if ((need & CEPH_CAP_FILE_WR) && ci->i_auth_cap) {
2206                         struct ceph_mds_session *s = ci->i_auth_cap->session;
2207                         spin_lock(&s->s_cap_lock);
2208                         session_readonly = s->s_readonly;
2209                         spin_unlock(&s->s_cap_lock);
2210                 }
2211                 if (session_readonly) {
2212                         dout("get_cap_refs %p needed %s but mds%d readonly\n",
2213                              inode, ceph_cap_string(need), ci->i_auth_cap->mds);
2214                         *err = -EROFS;
2215                         ret = 1;
2216                         goto out_unlock;
2217                 }
2218
2219                 dout("get_cap_refs %p have %s needed %s\n", inode,
2220                      ceph_cap_string(have), ceph_cap_string(need));
2221         }
2222 out_unlock:
2223         spin_unlock(&ci->i_ceph_lock);
2224         if (snap_rwsem_locked)
2225                 up_read(&mdsc->snap_rwsem);
2226
2227         dout("get_cap_refs %p ret %d got %s\n", inode,
2228              ret, ceph_cap_string(*got));
2229         return ret;
2230 }
2231
2232 /*
2233  * Check the offset we are writing up to against our current
2234  * max_size.  If necessary, tell the MDS we want to write to
2235  * a larger offset.
2236  */
2237 static void check_max_size(struct inode *inode, loff_t endoff)
2238 {
2239         struct ceph_inode_info *ci = ceph_inode(inode);
2240         int check = 0;
2241
2242         /* do we need to explicitly request a larger max_size? */
2243         spin_lock(&ci->i_ceph_lock);
2244         if (endoff >= ci->i_max_size && endoff > ci->i_wanted_max_size) {
2245                 dout("write %p at large endoff %llu, req max_size\n",
2246                      inode, endoff);
2247                 ci->i_wanted_max_size = endoff;
2248         }
2249         /* duplicate ceph_check_caps()'s logic */
2250         if (ci->i_auth_cap &&
2251             (ci->i_auth_cap->issued & CEPH_CAP_FILE_WR) &&
2252             ci->i_wanted_max_size > ci->i_max_size &&
2253             ci->i_wanted_max_size > ci->i_requested_max_size)
2254                 check = 1;
2255         spin_unlock(&ci->i_ceph_lock);
2256         if (check)
2257                 ceph_check_caps(ci, CHECK_CAPS_AUTHONLY, NULL);
2258 }
2259
2260 /*
2261  * Wait for caps, and take cap references.  If we can't get a WR cap
2262  * due to a small max_size, make sure we check_max_size (and possibly
2263  * ask the mds) so we don't get hung up indefinitely.
2264  */
2265 int ceph_get_caps(struct ceph_inode_info *ci, int need, int want,
2266                   loff_t endoff, int *got, struct page **pinned_page)
2267 {
2268         int _got, ret, err = 0;
2269
2270         ret = ceph_pool_perm_check(ci, need);
2271         if (ret < 0)
2272                 return ret;
2273
2274         while (true) {
2275                 if (endoff > 0)
2276                         check_max_size(&ci->vfs_inode, endoff);
2277
2278                 err = 0;
2279                 _got = 0;
2280                 ret = try_get_cap_refs(ci, need, want, endoff,
2281                                        false, &_got, &err);
2282                 if (ret) {
2283                         if (err == -EAGAIN)
2284                                 continue;
2285                         if (err < 0)
2286                                 return err;
2287                 } else {
2288                         ret = wait_event_interruptible(ci->i_cap_wq,
2289                                         try_get_cap_refs(ci, need, want, endoff,
2290                                                          true, &_got, &err));
2291                         if (err == -EAGAIN)
2292                                 continue;
2293                         if (err < 0)
2294                                 ret = err;
2295                         if (ret < 0)
2296                                 return ret;
2297                 }
2298
2299                 if (ci->i_inline_version != CEPH_INLINE_NONE &&
2300                     (_got & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) &&
2301                     i_size_read(&ci->vfs_inode) > 0) {
2302                         struct page *page =
2303                                 find_get_page(ci->vfs_inode.i_mapping, 0);
2304                         if (page) {
2305                                 if (PageUptodate(page)) {
2306                                         *pinned_page = page;
2307                                         break;
2308                                 }
2309                                 page_cache_release(page);
2310                         }
2311                         /*
2312                          * drop cap refs first because getattr while
2313                          * holding * caps refs can cause deadlock.
2314                          */
2315                         ceph_put_cap_refs(ci, _got);
2316                         _got = 0;
2317
2318                         /*
2319                          * getattr request will bring inline data into
2320                          * page cache
2321                          */
2322                         ret = __ceph_do_getattr(&ci->vfs_inode, NULL,
2323                                                 CEPH_STAT_CAP_INLINE_DATA,
2324                                                 true);
2325                         if (ret < 0)
2326                                 return ret;
2327                         continue;
2328                 }
2329                 break;
2330         }
2331
2332         *got = _got;
2333         return 0;
2334 }
2335
2336 /*
2337  * Take cap refs.  Caller must already know we hold at least one ref
2338  * on the caps in question or we don't know this is safe.
2339  */
2340 void ceph_get_cap_refs(struct ceph_inode_info *ci, int caps)
2341 {
2342         spin_lock(&ci->i_ceph_lock);
2343         __take_cap_refs(ci, caps, false);
2344         spin_unlock(&ci->i_ceph_lock);
2345 }
2346
2347
2348 /*
2349  * drop cap_snap that is not associated with any snapshot.
2350  * we don't need to send FLUSHSNAP message for it.
2351  */
2352 static int ceph_try_drop_cap_snap(struct ceph_cap_snap *capsnap)
2353 {
2354         if (!capsnap->need_flush &&
2355             !capsnap->writing && !capsnap->dirty_pages) {
2356
2357                 dout("dropping cap_snap %p follows %llu\n",
2358                      capsnap, capsnap->follows);
2359                 ceph_put_snap_context(capsnap->context);
2360                 list_del(&capsnap->ci_item);
2361                 list_del(&capsnap->flushing_item);
2362                 ceph_put_cap_snap(capsnap);
2363                 return 1;
2364         }
2365         return 0;
2366 }
2367
2368 /*
2369  * Release cap refs.
2370  *
2371  * If we released the last ref on any given cap, call ceph_check_caps
2372  * to release (or schedule a release).
2373  *
2374  * If we are releasing a WR cap (from a sync write), finalize any affected
2375  * cap_snap, and wake up any waiters.
2376  */
2377 void ceph_put_cap_refs(struct ceph_inode_info *ci, int had)
2378 {
2379         struct inode *inode = &ci->vfs_inode;
2380         int last = 0, put = 0, flushsnaps = 0, wake = 0;
2381
2382         spin_lock(&ci->i_ceph_lock);
2383         if (had & CEPH_CAP_PIN)
2384                 --ci->i_pin_ref;
2385         if (had & CEPH_CAP_FILE_RD)
2386                 if (--ci->i_rd_ref == 0)
2387                         last++;
2388         if (had & CEPH_CAP_FILE_CACHE)
2389                 if (--ci->i_rdcache_ref == 0)
2390                         last++;
2391         if (had & CEPH_CAP_FILE_BUFFER) {
2392                 if (--ci->i_wb_ref == 0) {
2393                         last++;
2394                         put++;
2395                 }
2396                 dout("put_cap_refs %p wb %d -> %d (?)\n",
2397                      inode, ci->i_wb_ref+1, ci->i_wb_ref);
2398         }
2399         if (had & CEPH_CAP_FILE_WR)
2400                 if (--ci->i_wr_ref == 0) {
2401                         last++;
2402                         if (__ceph_have_pending_cap_snap(ci)) {
2403                                 struct ceph_cap_snap *capsnap =
2404                                         list_last_entry(&ci->i_cap_snaps,
2405                                                         struct ceph_cap_snap,
2406                                                         ci_item);
2407                                 capsnap->writing = 0;
2408                                 if (ceph_try_drop_cap_snap(capsnap))
2409                                         put++;
2410                                 else if (__ceph_finish_cap_snap(ci, capsnap))
2411                                         flushsnaps = 1;
2412                                 wake = 1;
2413                         }
2414                         if (ci->i_wrbuffer_ref_head == 0 &&
2415                             ci->i_dirty_caps == 0 &&
2416                             ci->i_flushing_caps == 0) {
2417                                 BUG_ON(!ci->i_head_snapc);
2418                                 ceph_put_snap_context(ci->i_head_snapc);
2419                                 ci->i_head_snapc = NULL;
2420                         }
2421                         /* see comment in __ceph_remove_cap() */
2422                         if (!__ceph_is_any_caps(ci) && ci->i_snap_realm)
2423                                 drop_inode_snap_realm(ci);
2424                 }
2425         spin_unlock(&ci->i_ceph_lock);
2426
2427         dout("put_cap_refs %p had %s%s%s\n", inode, ceph_cap_string(had),
2428              last ? " last" : "", put ? " put" : "");
2429
2430         if (last && !flushsnaps)
2431                 ceph_check_caps(ci, 0, NULL);
2432         else if (flushsnaps)
2433                 ceph_flush_snaps(ci);
2434         if (wake)
2435                 wake_up_all(&ci->i_cap_wq);
2436         while (put-- > 0)
2437                 iput(inode);
2438 }
2439
2440 /*
2441  * Release @nr WRBUFFER refs on dirty pages for the given @snapc snap
2442  * context.  Adjust per-snap dirty page accounting as appropriate.
2443  * Once all dirty data for a cap_snap is flushed, flush snapped file
2444  * metadata back to the MDS.  If we dropped the last ref, call
2445  * ceph_check_caps.
2446  */
2447 void ceph_put_wrbuffer_cap_refs(struct ceph_inode_info *ci, int nr,
2448                                 struct ceph_snap_context *snapc)
2449 {
2450         struct inode *inode = &ci->vfs_inode;
2451         int last = 0;
2452         int complete_capsnap = 0;
2453         int drop_capsnap = 0;
2454         int found = 0;
2455         struct ceph_cap_snap *capsnap = NULL;
2456
2457         spin_lock(&ci->i_ceph_lock);
2458         ci->i_wrbuffer_ref -= nr;
2459         last = !ci->i_wrbuffer_ref;
2460
2461         if (ci->i_head_snapc == snapc) {
2462                 ci->i_wrbuffer_ref_head -= nr;
2463                 if (ci->i_wrbuffer_ref_head == 0 &&
2464                     ci->i_wr_ref == 0 &&
2465                     ci->i_dirty_caps == 0 &&
2466                     ci->i_flushing_caps == 0) {
2467                         BUG_ON(!ci->i_head_snapc);
2468                         ceph_put_snap_context(ci->i_head_snapc);
2469                         ci->i_head_snapc = NULL;
2470                 }
2471                 dout("put_wrbuffer_cap_refs on %p head %d/%d -> %d/%d %s\n",
2472                      inode,
2473                      ci->i_wrbuffer_ref+nr, ci->i_wrbuffer_ref_head+nr,
2474                      ci->i_wrbuffer_ref, ci->i_wrbuffer_ref_head,
2475                      last ? " LAST" : "");
2476         } else {
2477                 list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
2478                         if (capsnap->context == snapc) {
2479                                 found = 1;
2480                                 break;
2481                         }
2482                 }
2483                 BUG_ON(!found);
2484                 capsnap->dirty_pages -= nr;
2485                 if (capsnap->dirty_pages == 0) {
2486                         complete_capsnap = 1;
2487                         drop_capsnap = ceph_try_drop_cap_snap(capsnap);
2488                 }
2489                 dout("put_wrbuffer_cap_refs on %p cap_snap %p "
2490                      " snap %lld %d/%d -> %d/%d %s%s\n",
2491                      inode, capsnap, capsnap->context->seq,
2492                      ci->i_wrbuffer_ref+nr, capsnap->dirty_pages + nr,
2493                      ci->i_wrbuffer_ref, capsnap->dirty_pages,
2494                      last ? " (wrbuffer last)" : "",
2495                      complete_capsnap ? " (complete capsnap)" : "");
2496         }
2497
2498         spin_unlock(&ci->i_ceph_lock);
2499
2500         if (last) {
2501                 ceph_check_caps(ci, CHECK_CAPS_AUTHONLY, NULL);
2502                 iput(inode);
2503         } else if (complete_capsnap) {
2504                 ceph_flush_snaps(ci);
2505                 wake_up_all(&ci->i_cap_wq);
2506         }
2507         if (drop_capsnap)
2508                 iput(inode);
2509 }
2510
2511 /*
2512  * Invalidate unlinked inode's aliases, so we can drop the inode ASAP.
2513  */
2514 static void invalidate_aliases(struct inode *inode)
2515 {
2516         struct dentry *dn, *prev = NULL;
2517
2518         dout("invalidate_aliases inode %p\n", inode);
2519         d_prune_aliases(inode);
2520         /*
2521          * For non-directory inode, d_find_alias() only returns
2522          * hashed dentry. After calling d_invalidate(), the
2523          * dentry becomes unhashed.
2524          *
2525          * For directory inode, d_find_alias() can return
2526          * unhashed dentry. But directory inode should have
2527          * one alias at most.
2528          */
2529         while ((dn = d_find_alias(inode))) {
2530                 if (dn == prev) {
2531                         dput(dn);
2532                         break;
2533                 }
2534                 d_invalidate(dn);
2535                 if (prev)
2536                         dput(prev);
2537                 prev = dn;
2538         }
2539         if (prev)
2540                 dput(prev);
2541 }
2542
2543 /*
2544  * Handle a cap GRANT message from the MDS.  (Note that a GRANT may
2545  * actually be a revocation if it specifies a smaller cap set.)
2546  *
2547  * caller holds s_mutex and i_ceph_lock, we drop both.
2548  */
2549 static void handle_cap_grant(struct ceph_mds_client *mdsc,
2550                              struct inode *inode, struct ceph_mds_caps *grant,
2551                              u64 inline_version,
2552                              void *inline_data, int inline_len,
2553                              struct ceph_buffer *xattr_buf,
2554                              struct ceph_mds_session *session,
2555                              struct ceph_cap *cap, int issued)
2556         __releases(ci->i_ceph_lock)
2557         __releases(mdsc->snap_rwsem)
2558 {
2559         struct ceph_inode_info *ci = ceph_inode(inode);
2560         int mds = session->s_mds;
2561         int seq = le32_to_cpu(grant->seq);
2562         int newcaps = le32_to_cpu(grant->caps);
2563         int used, wanted, dirty;
2564         u64 size = le64_to_cpu(grant->size);
2565         u64 max_size = le64_to_cpu(grant->max_size);
2566         struct timespec mtime, atime, ctime;
2567         int check_caps = 0;
2568         bool wake = false;
2569         bool writeback = false;
2570         bool queue_trunc = false;
2571         bool queue_invalidate = false;
2572         bool queue_revalidate = false;
2573         bool deleted_inode = false;
2574         bool fill_inline = false;
2575
2576         dout("handle_cap_grant inode %p cap %p mds%d seq %d %s\n",
2577              inode, cap, mds, seq, ceph_cap_string(newcaps));
2578         dout(" size %llu max_size %llu, i_size %llu\n", size, max_size,
2579                 inode->i_size);
2580
2581
2582         /*
2583          * auth mds of the inode changed. we received the cap export message,
2584          * but still haven't received the cap import message. handle_cap_export
2585          * updated the new auth MDS' cap.
2586          *
2587          * "ceph_seq_cmp(seq, cap->seq) <= 0" means we are processing a message
2588          * that was sent before the cap import message. So don't remove caps.
2589          */
2590         if (ceph_seq_cmp(seq, cap->seq) <= 0) {
2591                 WARN_ON(cap != ci->i_auth_cap);
2592                 WARN_ON(cap->cap_id != le64_to_cpu(grant->cap_id));
2593                 seq = cap->seq;
2594                 newcaps |= cap->issued;
2595         }
2596
2597         /*
2598          * If CACHE is being revoked, and we have no dirty buffers,
2599          * try to invalidate (once).  (If there are dirty buffers, we
2600          * will invalidate _after_ writeback.)
2601          */
2602         if (((cap->issued & ~newcaps) & CEPH_CAP_FILE_CACHE) &&
2603             (newcaps & CEPH_CAP_FILE_LAZYIO) == 0 &&
2604             !ci->i_wrbuffer_ref) {
2605                 if (try_nonblocking_invalidate(inode)) {
2606                         /* there were locked pages.. invalidate later
2607                            in a separate thread. */
2608                         if (ci->i_rdcache_revoking != ci->i_rdcache_gen) {
2609                                 queue_invalidate = true;
2610                                 ci->i_rdcache_revoking = ci->i_rdcache_gen;
2611                         }
2612                 }
2613
2614                 ceph_fscache_invalidate(inode);
2615         }
2616
2617         /* side effects now are allowed */
2618         cap->cap_gen = session->s_cap_gen;
2619         cap->seq = seq;
2620
2621         __check_cap_issue(ci, cap, newcaps);
2622
2623         if ((newcaps & CEPH_CAP_AUTH_SHARED) &&
2624             (issued & CEPH_CAP_AUTH_EXCL) == 0) {
2625                 inode->i_mode = le32_to_cpu(grant->mode);
2626                 inode->i_uid = make_kuid(&init_user_ns, le32_to_cpu(grant->uid));
2627                 inode->i_gid = make_kgid(&init_user_ns, le32_to_cpu(grant->gid));
2628                 dout("%p mode 0%o uid.gid %d.%d\n", inode, inode->i_mode,
2629                      from_kuid(&init_user_ns, inode->i_uid),
2630                      from_kgid(&init_user_ns, inode->i_gid));
2631         }
2632
2633         if ((newcaps & CEPH_CAP_AUTH_SHARED) &&
2634             (issued & CEPH_CAP_LINK_EXCL) == 0) {
2635                 set_nlink(inode, le32_to_cpu(grant->nlink));
2636                 if (inode->i_nlink == 0 &&
2637                     (newcaps & (CEPH_CAP_LINK_SHARED | CEPH_CAP_LINK_EXCL)))
2638                         deleted_inode = true;
2639         }
2640
2641         if ((issued & CEPH_CAP_XATTR_EXCL) == 0 && grant->xattr_len) {
2642                 int len = le32_to_cpu(grant->xattr_len);
2643                 u64 version = le64_to_cpu(grant->xattr_version);
2644
2645                 if (version > ci->i_xattrs.version) {
2646                         dout(" got new xattrs v%llu on %p len %d\n",
2647                              version, inode, len);
2648                         if (ci->i_xattrs.blob)
2649                                 ceph_buffer_put(ci->i_xattrs.blob);
2650                         ci->i_xattrs.blob = ceph_buffer_get(xattr_buf);
2651                         ci->i_xattrs.version = version;
2652                         ceph_forget_all_cached_acls(inode);
2653                 }
2654         }
2655
2656         /* Do we need to revalidate our fscache cookie. Don't bother on the
2657          * first cache cap as we already validate at cookie creation time. */
2658         if ((issued & CEPH_CAP_FILE_CACHE) && ci->i_rdcache_gen > 1)
2659                 queue_revalidate = true;
2660
2661         if (newcaps & CEPH_CAP_ANY_RD) {
2662                 /* ctime/mtime/atime? */
2663                 ceph_decode_timespec(&mtime, &grant->mtime);
2664                 ceph_decode_timespec(&atime, &grant->atime);
2665                 ceph_decode_timespec(&ctime, &grant->ctime);
2666                 ceph_fill_file_time(inode, issued,
2667                                     le32_to_cpu(grant->time_warp_seq),
2668                                     &ctime, &mtime, &atime);
2669         }
2670
2671         if (newcaps & (CEPH_CAP_ANY_FILE_RD | CEPH_CAP_ANY_FILE_WR)) {
2672                 /* file layout may have changed */
2673                 ci->i_layout = grant->layout;
2674                 /* size/truncate_seq? */
2675                 queue_trunc = ceph_fill_file_size(inode, issued,
2676                                         le32_to_cpu(grant->truncate_seq),
2677                                         le64_to_cpu(grant->truncate_size),
2678                                         size);
2679                 /* max size increase? */
2680                 if (ci->i_auth_cap == cap && max_size != ci->i_max_size) {
2681                         dout("max_size %lld -> %llu\n",
2682                              ci->i_max_size, max_size);
2683                         ci->i_max_size = max_size;
2684                         if (max_size >= ci->i_wanted_max_size) {
2685                                 ci->i_wanted_max_size = 0;  /* reset */
2686                                 ci->i_requested_max_size = 0;
2687                         }
2688                         wake = true;
2689                 }
2690         }
2691
2692         /* check cap bits */
2693         wanted = __ceph_caps_wanted(ci);
2694         used = __ceph_caps_used(ci);
2695         dirty = __ceph_caps_dirty(ci);
2696         dout(" my wanted = %s, used = %s, dirty %s\n",
2697              ceph_cap_string(wanted),
2698              ceph_cap_string(used),
2699              ceph_cap_string(dirty));
2700         if (wanted != le32_to_cpu(grant->wanted)) {
2701                 dout("mds wanted %s -> %s\n",
2702                      ceph_cap_string(le32_to_cpu(grant->wanted)),
2703                      ceph_cap_string(wanted));
2704                 /* imported cap may not have correct mds_wanted */
2705                 if (le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT)
2706                         check_caps = 1;
2707         }
2708
2709         /* revocation, grant, or no-op? */
2710         if (cap->issued & ~newcaps) {
2711                 int revoking = cap->issued & ~newcaps;
2712
2713                 dout("revocation: %s -> %s (revoking %s)\n",
2714                      ceph_cap_string(cap->issued),
2715                      ceph_cap_string(newcaps),
2716                      ceph_cap_string(revoking));
2717                 if (revoking & used & CEPH_CAP_FILE_BUFFER)
2718                         writeback = true;  /* initiate writeback; will delay ack */
2719                 else if (revoking == CEPH_CAP_FILE_CACHE &&
2720                          (newcaps & CEPH_CAP_FILE_LAZYIO) == 0 &&
2721                          queue_invalidate)
2722                         ; /* do nothing yet, invalidation will be queued */
2723                 else if (cap == ci->i_auth_cap)
2724                         check_caps = 1; /* check auth cap only */
2725                 else
2726                         check_caps = 2; /* check all caps */
2727                 cap->issued = newcaps;
2728                 cap->implemented |= newcaps;
2729         } else if (cap->issued == newcaps) {
2730                 dout("caps unchanged: %s -> %s\n",
2731                      ceph_cap_string(cap->issued), ceph_cap_string(newcaps));
2732         } else {
2733                 dout("grant: %s -> %s\n", ceph_cap_string(cap->issued),
2734                      ceph_cap_string(newcaps));
2735                 /* non-auth MDS is revoking the newly grant caps ? */
2736                 if (cap == ci->i_auth_cap &&
2737                     __ceph_caps_revoking_other(ci, cap, newcaps))
2738                     check_caps = 2;
2739
2740                 cap->issued = newcaps;
2741                 cap->implemented |= newcaps; /* add bits only, to
2742                                               * avoid stepping on a
2743                                               * pending revocation */
2744                 wake = true;
2745         }
2746         BUG_ON(cap->issued & ~cap->implemented);
2747
2748         if (inline_version > 0 && inline_version >= ci->i_inline_version) {
2749                 ci->i_inline_version = inline_version;
2750                 if (ci->i_inline_version != CEPH_INLINE_NONE &&
2751                     (newcaps & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)))
2752                         fill_inline = true;
2753         }
2754
2755         spin_unlock(&ci->i_ceph_lock);
2756
2757         if (le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT) {
2758                 kick_flushing_inode_caps(mdsc, session, inode);
2759                 up_read(&mdsc->snap_rwsem);
2760                 if (newcaps & ~issued)
2761                         wake = true;
2762         }
2763
2764         if (fill_inline)
2765                 ceph_fill_inline_data(inode, NULL, inline_data, inline_len);
2766
2767         if (queue_trunc) {
2768                 ceph_queue_vmtruncate(inode);
2769                 ceph_queue_revalidate(inode);
2770         } else if (queue_revalidate)
2771                 ceph_queue_revalidate(inode);
2772
2773         if (writeback)
2774                 /*
2775                  * queue inode for writeback: we can't actually call
2776                  * filemap_write_and_wait, etc. from message handler
2777                  * context.
2778                  */
2779                 ceph_queue_writeback(inode);
2780         if (queue_invalidate)
2781                 ceph_queue_invalidate(inode);
2782         if (deleted_inode)
2783                 invalidate_aliases(inode);
2784         if (wake)
2785                 wake_up_all(&ci->i_cap_wq);
2786
2787         if (check_caps == 1)
2788                 ceph_check_caps(ci, CHECK_CAPS_NODELAY|CHECK_CAPS_AUTHONLY,
2789                                 session);
2790         else if (check_caps == 2)
2791                 ceph_check_caps(ci, CHECK_CAPS_NODELAY, session);
2792         else
2793                 mutex_unlock(&session->s_mutex);
2794 }
2795
2796 /*
2797  * Handle FLUSH_ACK from MDS, indicating that metadata we sent to the
2798  * MDS has been safely committed.
2799  */
2800 static void handle_cap_flush_ack(struct inode *inode, u64 flush_tid,
2801                                  struct ceph_mds_caps *m,
2802                                  struct ceph_mds_session *session,
2803                                  struct ceph_cap *cap)
2804         __releases(ci->i_ceph_lock)
2805 {
2806         struct ceph_inode_info *ci = ceph_inode(inode);
2807         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
2808         unsigned seq = le32_to_cpu(m->seq);
2809         int dirty = le32_to_cpu(m->dirty);
2810         int cleaned = 0;
2811         int drop = 0;
2812         int i;
2813
2814         for (i = 0; i < CEPH_CAP_BITS; i++)
2815                 if ((dirty & (1 << i)) &&
2816                     (u16)flush_tid == ci->i_cap_flush_tid[i])
2817                         cleaned |= 1 << i;
2818
2819         dout("handle_cap_flush_ack inode %p mds%d seq %d on %s cleaned %s,"
2820              " flushing %s -> %s\n",
2821              inode, session->s_mds, seq, ceph_cap_string(dirty),
2822              ceph_cap_string(cleaned), ceph_cap_string(ci->i_flushing_caps),
2823              ceph_cap_string(ci->i_flushing_caps & ~cleaned));
2824
2825         if (ci->i_flushing_caps == (ci->i_flushing_caps & ~cleaned))
2826                 goto out;
2827
2828         ci->i_flushing_caps &= ~cleaned;
2829
2830         spin_lock(&mdsc->cap_dirty_lock);
2831         if (ci->i_flushing_caps == 0) {
2832                 list_del_init(&ci->i_flushing_item);
2833                 if (!list_empty(&session->s_cap_flushing))
2834                         dout(" mds%d still flushing cap on %p\n",
2835                              session->s_mds,
2836                              &list_entry(session->s_cap_flushing.next,
2837                                          struct ceph_inode_info,
2838                                          i_flushing_item)->vfs_inode);
2839                 mdsc->num_cap_flushing--;
2840                 wake_up_all(&mdsc->cap_flushing_wq);
2841                 dout(" inode %p now !flushing\n", inode);
2842
2843                 if (ci->i_dirty_caps == 0) {
2844                         dout(" inode %p now clean\n", inode);
2845                         BUG_ON(!list_empty(&ci->i_dirty_item));
2846                         drop = 1;
2847                         if (ci->i_wr_ref == 0 &&
2848                             ci->i_wrbuffer_ref_head == 0) {
2849                                 BUG_ON(!ci->i_head_snapc);
2850                                 ceph_put_snap_context(ci->i_head_snapc);
2851                                 ci->i_head_snapc = NULL;
2852                         }
2853                 } else {
2854                         BUG_ON(list_empty(&ci->i_dirty_item));
2855                 }
2856         }
2857         spin_unlock(&mdsc->cap_dirty_lock);
2858         wake_up_all(&ci->i_cap_wq);
2859
2860 out:
2861         spin_unlock(&ci->i_ceph_lock);
2862         if (drop)
2863                 iput(inode);
2864 }
2865
2866 /*
2867  * Handle FLUSHSNAP_ACK.  MDS has flushed snap data to disk and we can
2868  * throw away our cap_snap.
2869  *
2870  * Caller hold s_mutex.
2871  */
2872 static void handle_cap_flushsnap_ack(struct inode *inode, u64 flush_tid,
2873                                      struct ceph_mds_caps *m,
2874                                      struct ceph_mds_session *session)
2875 {
2876         struct ceph_inode_info *ci = ceph_inode(inode);
2877         u64 follows = le64_to_cpu(m->snap_follows);
2878         struct ceph_cap_snap *capsnap;
2879         int drop = 0;
2880
2881         dout("handle_cap_flushsnap_ack inode %p ci %p mds%d follows %lld\n",
2882              inode, ci, session->s_mds, follows);
2883
2884         spin_lock(&ci->i_ceph_lock);
2885         list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
2886                 if (capsnap->follows == follows) {
2887                         if (capsnap->flush_tid != flush_tid) {
2888                                 dout(" cap_snap %p follows %lld tid %lld !="
2889                                      " %lld\n", capsnap, follows,
2890                                      flush_tid, capsnap->flush_tid);
2891                                 break;
2892                         }
2893                         WARN_ON(capsnap->dirty_pages || capsnap->writing);
2894                         dout(" removing %p cap_snap %p follows %lld\n",
2895                              inode, capsnap, follows);
2896                         ceph_put_snap_context(capsnap->context);
2897                         list_del(&capsnap->ci_item);
2898                         list_del(&capsnap->flushing_item);
2899                         ceph_put_cap_snap(capsnap);
2900                         drop = 1;
2901                         break;
2902                 } else {
2903                         dout(" skipping cap_snap %p follows %lld\n",
2904                              capsnap, capsnap->follows);
2905                 }
2906         }
2907         spin_unlock(&ci->i_ceph_lock);
2908         if (drop)
2909                 iput(inode);
2910 }
2911
2912 /*
2913  * Handle TRUNC from MDS, indicating file truncation.
2914  *
2915  * caller hold s_mutex.
2916  */
2917 static void handle_cap_trunc(struct inode *inode,
2918                              struct ceph_mds_caps *trunc,
2919                              struct ceph_mds_session *session)
2920         __releases(ci->i_ceph_lock)
2921 {
2922         struct ceph_inode_info *ci = ceph_inode(inode);
2923         int mds = session->s_mds;
2924         int seq = le32_to_cpu(trunc->seq);
2925         u32 truncate_seq = le32_to_cpu(trunc->truncate_seq);
2926         u64 truncate_size = le64_to_cpu(trunc->truncate_size);
2927         u64 size = le64_to_cpu(trunc->size);
2928         int implemented = 0;
2929         int dirty = __ceph_caps_dirty(ci);
2930         int issued = __ceph_caps_issued(ceph_inode(inode), &implemented);
2931         int queue_trunc = 0;
2932
2933         issued |= implemented | dirty;
2934
2935         dout("handle_cap_trunc inode %p mds%d seq %d to %lld seq %d\n",
2936              inode, mds, seq, truncate_size, truncate_seq);
2937         queue_trunc = ceph_fill_file_size(inode, issued,
2938                                           truncate_seq, truncate_size, size);
2939         spin_unlock(&ci->i_ceph_lock);
2940
2941         if (queue_trunc) {
2942                 ceph_queue_vmtruncate(inode);
2943                 ceph_fscache_invalidate(inode);
2944         }
2945 }
2946
2947 /*
2948  * Handle EXPORT from MDS.  Cap is being migrated _from_ this mds to a
2949  * different one.  If we are the most recent migration we've seen (as
2950  * indicated by mseq), make note of the migrating cap bits for the
2951  * duration (until we see the corresponding IMPORT).
2952  *
2953  * caller holds s_mutex
2954  */
2955 static void handle_cap_export(struct inode *inode, struct ceph_mds_caps *ex,
2956                               struct ceph_mds_cap_peer *ph,
2957                               struct ceph_mds_session *session)
2958 {
2959         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
2960         struct ceph_mds_session *tsession = NULL;
2961         struct ceph_cap *cap, *tcap, *new_cap = NULL;
2962         struct ceph_inode_info *ci = ceph_inode(inode);
2963         u64 t_cap_id;
2964         unsigned mseq = le32_to_cpu(ex->migrate_seq);
2965         unsigned t_seq, t_mseq;
2966         int target, issued;
2967         int mds = session->s_mds;
2968
2969         if (ph) {
2970                 t_cap_id = le64_to_cpu(ph->cap_id);
2971                 t_seq = le32_to_cpu(ph->seq);
2972                 t_mseq = le32_to_cpu(ph->mseq);
2973                 target = le32_to_cpu(ph->mds);
2974         } else {
2975                 t_cap_id = t_seq = t_mseq = 0;
2976                 target = -1;
2977         }
2978
2979         dout("handle_cap_export inode %p ci %p mds%d mseq %d target %d\n",
2980              inode, ci, mds, mseq, target);
2981 retry:
2982         spin_lock(&ci->i_ceph_lock);
2983         cap = __get_cap_for_mds(ci, mds);
2984         if (!cap || cap->cap_id != le64_to_cpu(ex->cap_id))
2985                 goto out_unlock;
2986
2987         if (target < 0) {
2988                 __ceph_remove_cap(cap, false);
2989                 goto out_unlock;
2990         }
2991
2992         /*
2993          * now we know we haven't received the cap import message yet
2994          * because the exported cap still exist.
2995          */
2996
2997         issued = cap->issued;
2998         WARN_ON(issued != cap->implemented);
2999
3000         tcap = __get_cap_for_mds(ci, target);
3001         if (tcap) {
3002                 /* already have caps from the target */
3003                 if (tcap->cap_id != t_cap_id ||
3004                     ceph_seq_cmp(tcap->seq, t_seq) < 0) {
3005                         dout(" updating import cap %p mds%d\n", tcap, target);
3006                         tcap->cap_id = t_cap_id;
3007                         tcap->seq = t_seq - 1;
3008                         tcap->issue_seq = t_seq - 1;
3009                         tcap->mseq = t_mseq;
3010                         tcap->issued |= issued;
3011                         tcap->implemented |= issued;
3012                         if (cap == ci->i_auth_cap)
3013                                 ci->i_auth_cap = tcap;
3014                         if (ci->i_flushing_caps && ci->i_auth_cap == tcap) {
3015                                 spin_lock(&mdsc->cap_dirty_lock);
3016                                 list_move_tail(&ci->i_flushing_item,
3017                                                &tcap->session->s_cap_flushing);
3018                                 spin_unlock(&mdsc->cap_dirty_lock);
3019                         }
3020                 }
3021                 __ceph_remove_cap(cap, false);
3022                 goto out_unlock;
3023         } else if (tsession) {
3024                 /* add placeholder for the export tagert */
3025                 int flag = (cap == ci->i_auth_cap) ? CEPH_CAP_FLAG_AUTH : 0;
3026                 ceph_add_cap(inode, tsession, t_cap_id, -1, issued, 0,
3027                              t_seq - 1, t_mseq, (u64)-1, flag, &new_cap);
3028
3029                 __ceph_remove_cap(cap, false);
3030                 goto out_unlock;
3031         }
3032
3033         spin_unlock(&ci->i_ceph_lock);
3034         mutex_unlock(&session->s_mutex);
3035
3036         /* open target session */
3037         tsession = ceph_mdsc_open_export_target_session(mdsc, target);
3038         if (!IS_ERR(tsession)) {
3039                 if (mds > target) {
3040                         mutex_lock(&session->s_mutex);
3041                         mutex_lock_nested(&tsession->s_mutex,
3042                                           SINGLE_DEPTH_NESTING);
3043                 } else {
3044                         mutex_lock(&tsession->s_mutex);
3045                         mutex_lock_nested(&session->s_mutex,
3046                                           SINGLE_DEPTH_NESTING);
3047                 }
3048                 ceph_add_cap_releases(mdsc, tsession);
3049                 new_cap = ceph_get_cap(mdsc, NULL);
3050         } else {
3051                 WARN_ON(1);
3052                 tsession = NULL;
3053                 target = -1;
3054         }
3055         goto retry;
3056
3057 out_unlock:
3058         spin_unlock(&ci->i_ceph_lock);
3059         mutex_unlock(&session->s_mutex);
3060         if (tsession) {
3061                 mutex_unlock(&tsession->s_mutex);
3062                 ceph_put_mds_session(tsession);
3063         }
3064         if (new_cap)
3065                 ceph_put_cap(mdsc, new_cap);
3066 }
3067
3068 /*
3069  * Handle cap IMPORT.
3070  *
3071  * caller holds s_mutex. acquires i_ceph_lock
3072  */
3073 static void handle_cap_import(struct ceph_mds_client *mdsc,
3074                               struct inode *inode, struct ceph_mds_caps *im,
3075                               struct ceph_mds_cap_peer *ph,
3076                               struct ceph_mds_session *session,
3077                               struct ceph_cap **target_cap, int *old_issued)
3078         __acquires(ci->i_ceph_lock)
3079 {
3080         struct ceph_inode_info *ci = ceph_inode(inode);
3081         struct ceph_cap *cap, *ocap, *new_cap = NULL;
3082         int mds = session->s_mds;
3083         int issued;
3084         unsigned caps = le32_to_cpu(im->caps);
3085         unsigned wanted = le32_to_cpu(im->wanted);
3086         unsigned seq = le32_to_cpu(im->seq);
3087         unsigned mseq = le32_to_cpu(im->migrate_seq);
3088         u64 realmino = le64_to_cpu(im->realm);
3089         u64 cap_id = le64_to_cpu(im->cap_id);
3090         u64 p_cap_id;
3091         int peer;
3092
3093         if (ph) {
3094                 p_cap_id = le64_to_cpu(ph->cap_id);
3095                 peer = le32_to_cpu(ph->mds);
3096         } else {
3097                 p_cap_id = 0;
3098                 peer = -1;
3099         }
3100
3101         dout("handle_cap_import inode %p ci %p mds%d mseq %d peer %d\n",
3102              inode, ci, mds, mseq, peer);
3103
3104 retry:
3105         spin_lock(&ci->i_ceph_lock);
3106         cap = __get_cap_for_mds(ci, mds);
3107         if (!cap) {
3108                 if (!new_cap) {
3109                         spin_unlock(&ci->i_ceph_lock);
3110                         new_cap = ceph_get_cap(mdsc, NULL);
3111                         goto retry;
3112                 }
3113                 cap = new_cap;
3114         } else {
3115                 if (new_cap) {
3116                         ceph_put_cap(mdsc, new_cap);
3117                         new_cap = NULL;
3118                 }
3119         }
3120
3121         __ceph_caps_issued(ci, &issued);
3122         issued |= __ceph_caps_dirty(ci);
3123
3124         ceph_add_cap(inode, session, cap_id, -1, caps, wanted, seq, mseq,
3125                      realmino, CEPH_CAP_FLAG_AUTH, &new_cap);
3126
3127         ocap = peer >= 0 ? __get_cap_for_mds(ci, peer) : NULL;
3128         if (ocap && ocap->cap_id == p_cap_id) {
3129                 dout(" remove export cap %p mds%d flags %d\n",
3130                      ocap, peer, ph->flags);
3131                 if ((ph->flags & CEPH_CAP_FLAG_AUTH) &&
3132                     (ocap->seq != le32_to_cpu(ph->seq) ||
3133                      ocap->mseq != le32_to_cpu(ph->mseq))) {
3134                         pr_err("handle_cap_import: mismatched seq/mseq: "
3135                                "ino (%llx.%llx) mds%d seq %d mseq %d "
3136                                "importer mds%d has peer seq %d mseq %d\n",
3137                                ceph_vinop(inode), peer, ocap->seq,
3138                                ocap->mseq, mds, le32_to_cpu(ph->seq),
3139                                le32_to_cpu(ph->mseq));
3140                 }
3141                 __ceph_remove_cap(ocap, (ph->flags & CEPH_CAP_FLAG_RELEASE));
3142         }
3143
3144         /* make sure we re-request max_size, if necessary */
3145         ci->i_wanted_max_size = 0;
3146         ci->i_requested_max_size = 0;
3147
3148         *old_issued = issued;
3149         *target_cap = cap;
3150 }
3151
3152 /*
3153  * Handle a caps message from the MDS.
3154  *
3155  * Identify the appropriate session, inode, and call the right handler
3156  * based on the cap op.
3157  */
3158 void ceph_handle_caps(struct ceph_mds_session *session,
3159                       struct ceph_msg *msg)
3160 {
3161         struct ceph_mds_client *mdsc = session->s_mdsc;
3162         struct super_block *sb = mdsc->fsc->sb;
3163         struct inode *inode;
3164         struct ceph_inode_info *ci;
3165         struct ceph_cap *cap;
3166         struct ceph_mds_caps *h;
3167         struct ceph_mds_cap_peer *peer = NULL;
3168         struct ceph_snap_realm *realm;
3169         int mds = session->s_mds;
3170         int op, issued;
3171         u32 seq, mseq;
3172         struct ceph_vino vino;
3173         u64 cap_id;
3174         u64 size, max_size;
3175         u64 tid;
3176         u64 inline_version = 0;
3177         void *inline_data = NULL;
3178         u32  inline_len = 0;
3179         void *snaptrace;
3180         size_t snaptrace_len;
3181         void *p, *end;
3182
3183         dout("handle_caps from mds%d\n", mds);
3184
3185         /* decode */
3186         end = msg->front.iov_base + msg->front.iov_len;
3187         tid = le64_to_cpu(msg->hdr.tid);
3188         if (msg->front.iov_len < sizeof(*h))
3189                 goto bad;
3190         h = msg->front.iov_base;
3191         op = le32_to_cpu(h->op);
3192         vino.ino = le64_to_cpu(h->ino);
3193         vino.snap = CEPH_NOSNAP;
3194         cap_id = le64_to_cpu(h->cap_id);
3195         seq = le32_to_cpu(h->seq);
3196         mseq = le32_to_cpu(h->migrate_seq);
3197         size = le64_to_cpu(h->size);
3198         max_size = le64_to_cpu(h->max_size);
3199
3200         snaptrace = h + 1;
3201         snaptrace_len = le32_to_cpu(h->snap_trace_len);
3202         p = snaptrace + snaptrace_len;
3203
3204         if (le16_to_cpu(msg->hdr.version) >= 2) {
3205                 u32 flock_len;
3206                 ceph_decode_32_safe(&p, end, flock_len, bad);
3207                 if (p + flock_len > end)
3208                         goto bad;
3209                 p += flock_len;
3210         }
3211
3212         if (le16_to_cpu(msg->hdr.version) >= 3) {
3213                 if (op == CEPH_CAP_OP_IMPORT) {
3214                         if (p + sizeof(*peer) > end)
3215                                 goto bad;
3216                         peer = p;
3217                         p += sizeof(*peer);
3218                 } else if (op == CEPH_CAP_OP_EXPORT) {
3219                         /* recorded in unused fields */
3220                         peer = (void *)&h->size;
3221                 }
3222         }
3223
3224         if (le16_to_cpu(msg->hdr.version) >= 4) {
3225                 ceph_decode_64_safe(&p, end, inline_version, bad);
3226                 ceph_decode_32_safe(&p, end, inline_len, bad);
3227                 if (p + inline_len > end)
3228                         goto bad;
3229                 inline_data = p;
3230                 p += inline_len;
3231         }
3232
3233         /* lookup ino */
3234         inode = ceph_find_inode(sb, vino);
3235         ci = ceph_inode(inode);
3236         dout(" op %s ino %llx.%llx inode %p\n", ceph_cap_op_name(op), vino.ino,
3237              vino.snap, inode);
3238
3239         mutex_lock(&session->s_mutex);
3240         session->s_seq++;
3241         dout(" mds%d seq %lld cap seq %u\n", session->s_mds, session->s_seq,
3242              (unsigned)seq);
3243
3244         if (op == CEPH_CAP_OP_IMPORT)
3245                 ceph_add_cap_releases(mdsc, session);
3246
3247         if (!inode) {
3248                 dout(" i don't have ino %llx\n", vino.ino);
3249
3250                 if (op == CEPH_CAP_OP_IMPORT) {
3251                         spin_lock(&session->s_cap_lock);
3252                         __queue_cap_release(session, vino.ino, cap_id,
3253                                             mseq, seq);
3254                         spin_unlock(&session->s_cap_lock);
3255                 }
3256                 goto flush_cap_releases;
3257         }
3258
3259         /* these will work even if we don't have a cap yet */
3260         switch (op) {
3261         case CEPH_CAP_OP_FLUSHSNAP_ACK:
3262                 handle_cap_flushsnap_ack(inode, tid, h, session);
3263                 goto done;
3264
3265         case CEPH_CAP_OP_EXPORT:
3266                 handle_cap_export(inode, h, peer, session);
3267                 goto done_unlocked;
3268
3269         case CEPH_CAP_OP_IMPORT:
3270                 realm = NULL;
3271                 if (snaptrace_len) {
3272                         down_write(&mdsc->snap_rwsem);
3273                         ceph_update_snap_trace(mdsc, snaptrace,
3274                                                snaptrace + snaptrace_len,
3275                                                false, &realm);
3276                         downgrade_write(&mdsc->snap_rwsem);
3277                 } else {
3278                         down_read(&mdsc->snap_rwsem);
3279                 }
3280                 handle_cap_import(mdsc, inode, h, peer, session,
3281                                   &cap, &issued);
3282                 handle_cap_grant(mdsc, inode, h,
3283                                  inline_version, inline_data, inline_len,
3284                                  msg->middle, session, cap, issued);
3285                 if (realm)
3286                         ceph_put_snap_realm(mdsc, realm);
3287                 goto done_unlocked;
3288         }
3289
3290         /* the rest require a cap */
3291         spin_lock(&ci->i_ceph_lock);
3292         cap = __get_cap_for_mds(ceph_inode(inode), mds);
3293         if (!cap) {
3294                 dout(" no cap on %p ino %llx.%llx from mds%d\n",
3295                      inode, ceph_ino(inode), ceph_snap(inode), mds);
3296                 spin_unlock(&ci->i_ceph_lock);
3297                 goto flush_cap_releases;
3298         }
3299
3300         /* note that each of these drops i_ceph_lock for us */
3301         switch (op) {
3302         case CEPH_CAP_OP_REVOKE:
3303         case CEPH_CAP_OP_GRANT:
3304                 __ceph_caps_issued(ci, &issued);
3305                 issued |= __ceph_caps_dirty(ci);
3306                 handle_cap_grant(mdsc, inode, h,
3307                                  inline_version, inline_data, inline_len,
3308                                  msg->middle, session, cap, issued);
3309                 goto done_unlocked;
3310
3311         case CEPH_CAP_OP_FLUSH_ACK:
3312                 handle_cap_flush_ack(inode, tid, h, session, cap);
3313                 break;
3314
3315         case CEPH_CAP_OP_TRUNC:
3316                 handle_cap_trunc(inode, h, session);
3317                 break;
3318
3319         default:
3320                 spin_unlock(&ci->i_ceph_lock);
3321                 pr_err("ceph_handle_caps: unknown cap op %d %s\n", op,
3322                        ceph_cap_op_name(op));
3323         }
3324
3325         goto done;
3326
3327 flush_cap_releases:
3328         /*
3329          * send any full release message to try to move things
3330          * along for the mds (who clearly thinks we still have this
3331          * cap).
3332          */
3333         ceph_add_cap_releases(mdsc, session);
3334         ceph_send_cap_releases(mdsc, session);
3335
3336 done:
3337         mutex_unlock(&session->s_mutex);
3338 done_unlocked:
3339         iput(inode);
3340         return;
3341
3342 bad:
3343         pr_err("ceph_handle_caps: corrupt message\n");
3344         ceph_msg_dump(msg);
3345         return;
3346 }
3347
3348 /*
3349  * Delayed work handler to process end of delayed cap release LRU list.
3350  */
3351 void ceph_check_delayed_caps(struct ceph_mds_client *mdsc)
3352 {
3353         struct ceph_inode_info *ci;
3354         int flags = CHECK_CAPS_NODELAY;
3355
3356         dout("check_delayed_caps\n");
3357         while (1) {
3358                 spin_lock(&mdsc->cap_delay_lock);
3359                 if (list_empty(&mdsc->cap_delay_list))
3360                         break;
3361                 ci = list_first_entry(&mdsc->cap_delay_list,
3362                                       struct ceph_inode_info,
3363                                       i_cap_delay_list);
3364                 if ((ci->i_ceph_flags & CEPH_I_FLUSH) == 0 &&
3365                     time_before(jiffies, ci->i_hold_caps_max))
3366                         break;
3367                 list_del_init(&ci->i_cap_delay_list);
3368                 spin_unlock(&mdsc->cap_delay_lock);
3369                 dout("check_delayed_caps on %p\n", &ci->vfs_inode);
3370                 ceph_check_caps(ci, flags, NULL);
3371         }
3372         spin_unlock(&mdsc->cap_delay_lock);
3373 }
3374
3375 /*
3376  * Flush all dirty caps to the mds
3377  */
3378 void ceph_flush_dirty_caps(struct ceph_mds_client *mdsc)
3379 {
3380         struct ceph_inode_info *ci;
3381         struct inode *inode;
3382
3383         dout("flush_dirty_caps\n");
3384         spin_lock(&mdsc->cap_dirty_lock);
3385         while (!list_empty(&mdsc->cap_dirty)) {
3386                 ci = list_first_entry(&mdsc->cap_dirty, struct ceph_inode_info,
3387                                       i_dirty_item);
3388                 inode = &ci->vfs_inode;
3389                 ihold(inode);
3390                 dout("flush_dirty_caps %p\n", inode);
3391                 spin_unlock(&mdsc->cap_dirty_lock);
3392                 ceph_check_caps(ci, CHECK_CAPS_NODELAY|CHECK_CAPS_FLUSH, NULL);
3393                 iput(inode);
3394                 spin_lock(&mdsc->cap_dirty_lock);
3395         }
3396         spin_unlock(&mdsc->cap_dirty_lock);
3397         dout("flush_dirty_caps done\n");
3398 }
3399
3400 /*
3401  * Drop open file reference.  If we were the last open file,
3402  * we may need to release capabilities to the MDS (or schedule
3403  * their delayed release).
3404  */
3405 void ceph_put_fmode(struct ceph_inode_info *ci, int fmode)
3406 {
3407         struct inode *inode = &ci->vfs_inode;
3408         int last = 0;
3409
3410         spin_lock(&ci->i_ceph_lock);
3411         dout("put_fmode %p fmode %d %d -> %d\n", inode, fmode,
3412              ci->i_nr_by_mode[fmode], ci->i_nr_by_mode[fmode]-1);
3413         BUG_ON(ci->i_nr_by_mode[fmode] == 0);
3414         if (--ci->i_nr_by_mode[fmode] == 0)
3415                 last++;
3416         spin_unlock(&ci->i_ceph_lock);
3417
3418         if (last && ci->i_vino.snap == CEPH_NOSNAP)
3419                 ceph_check_caps(ci, 0, NULL);
3420 }
3421
3422 /*
3423  * Helpers for embedding cap and dentry lease releases into mds
3424  * requests.
3425  *
3426  * @force is used by dentry_release (below) to force inclusion of a
3427  * record for the directory inode, even when there aren't any caps to
3428  * drop.
3429  */
3430 int ceph_encode_inode_release(void **p, struct inode *inode,
3431                               int mds, int drop, int unless, int force)
3432 {
3433         struct ceph_inode_info *ci = ceph_inode(inode);
3434         struct ceph_cap *cap;
3435         struct ceph_mds_request_release *rel = *p;
3436         int used, dirty;
3437         int ret = 0;
3438
3439         spin_lock(&ci->i_ceph_lock);
3440         used = __ceph_caps_used(ci);
3441         dirty = __ceph_caps_dirty(ci);
3442
3443         dout("encode_inode_release %p mds%d used|dirty %s drop %s unless %s\n",
3444              inode, mds, ceph_cap_string(used|dirty), ceph_cap_string(drop),
3445              ceph_cap_string(unless));
3446
3447         /* only drop unused, clean caps */
3448         drop &= ~(used | dirty);
3449
3450         cap = __get_cap_for_mds(ci, mds);
3451         if (cap && __cap_is_valid(cap)) {
3452                 if (force ||
3453                     ((cap->issued & drop) &&
3454                      (cap->issued & unless) == 0)) {
3455                         if ((cap->issued & drop) &&
3456                             (cap->issued & unless) == 0) {
3457                                 int wanted = __ceph_caps_wanted(ci);
3458                                 if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0)
3459                                         wanted |= cap->mds_wanted;
3460                                 dout("encode_inode_release %p cap %p "
3461                                      "%s -> %s, wanted %s -> %s\n", inode, cap,
3462                                      ceph_cap_string(cap->issued),
3463                                      ceph_cap_string(cap->issued & ~drop),
3464                                      ceph_cap_string(cap->mds_wanted),
3465                                      ceph_cap_string(wanted));
3466
3467                                 cap->issued &= ~drop;
3468                                 cap->implemented &= ~drop;
3469                                 cap->mds_wanted = wanted;
3470                         } else {
3471                                 dout("encode_inode_release %p cap %p %s"
3472                                      " (force)\n", inode, cap,
3473                                      ceph_cap_string(cap->issued));
3474                         }
3475
3476                         rel->ino = cpu_to_le64(ceph_ino(inode));
3477                         rel->cap_id = cpu_to_le64(cap->cap_id);
3478                         rel->seq = cpu_to_le32(cap->seq);
3479                         rel->issue_seq = cpu_to_le32(cap->issue_seq);
3480                         rel->mseq = cpu_to_le32(cap->mseq);
3481                         rel->caps = cpu_to_le32(cap->implemented);
3482                         rel->wanted = cpu_to_le32(cap->mds_wanted);
3483                         rel->dname_len = 0;
3484                         rel->dname_seq = 0;
3485                         *p += sizeof(*rel);
3486                         ret = 1;
3487                 } else {
3488                         dout("encode_inode_release %p cap %p %s\n",
3489                              inode, cap, ceph_cap_string(cap->issued));
3490                 }
3491         }
3492         spin_unlock(&ci->i_ceph_lock);
3493         return ret;
3494 }
3495
3496 int ceph_encode_dentry_release(void **p, struct dentry *dentry,
3497                                int mds, int drop, int unless)
3498 {
3499         struct inode *dir = d_inode(dentry->d_parent);
3500         struct ceph_mds_request_release *rel = *p;
3501         struct ceph_dentry_info *di = ceph_dentry(dentry);
3502         int force = 0;
3503         int ret;
3504
3505         /*
3506          * force an record for the directory caps if we have a dentry lease.
3507          * this is racy (can't take i_ceph_lock and d_lock together), but it
3508          * doesn't have to be perfect; the mds will revoke anything we don't
3509          * release.
3510          */
3511         spin_lock(&dentry->d_lock);
3512         if (di->lease_session && di->lease_session->s_mds == mds)
3513                 force = 1;
3514         spin_unlock(&dentry->d_lock);
3515
3516         ret = ceph_encode_inode_release(p, dir, mds, drop, unless, force);
3517
3518         spin_lock(&dentry->d_lock);
3519         if (ret && di->lease_session && di->lease_session->s_mds == mds) {
3520                 dout("encode_dentry_release %p mds%d seq %d\n",
3521                      dentry, mds, (int)di->lease_seq);
3522                 rel->dname_len = cpu_to_le32(dentry->d_name.len);
3523                 memcpy(*p, dentry->d_name.name, dentry->d_name.len);
3524                 *p += dentry->d_name.len;
3525                 rel->dname_seq = cpu_to_le32(di->lease_seq);
3526                 __ceph_mdsc_drop_dentry_lease(dentry);
3527         }
3528         spin_unlock(&dentry->d_lock);
3529         return ret;
3530 }