ceph: don't include used caps in cap_wanted
[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         if (session->s_cap_iterator == cap) {
930                 /* not yet, we are iterating over this very cap */
931                 dout("__ceph_remove_cap  delaying %p removal from session %p\n",
932                      cap, cap->session);
933         } else {
934                 list_del_init(&cap->session_caps);
935                 session->s_nr_caps--;
936                 cap->session = NULL;
937                 removed = 1;
938         }
939         /* protect backpointer with s_cap_lock: see iterate_session_caps */
940         cap->ci = NULL;
941
942         /*
943          * s_cap_reconnect is protected by s_cap_lock. no one changes
944          * s_cap_gen while session is in the reconnect state.
945          */
946         if (queue_release &&
947             (!session->s_cap_reconnect || cap->cap_gen == session->s_cap_gen)) {
948                 cap->queue_release = 1;
949                 if (removed) {
950                         list_add_tail(&cap->session_caps,
951                                       &session->s_cap_releases);
952                         session->s_num_cap_releases++;
953                         removed = 0;
954                 }
955         } else {
956                 cap->queue_release = 0;
957         }
958         cap->cap_ino = ci->i_vino.ino;
959
960         spin_unlock(&session->s_cap_lock);
961
962         /* remove from inode list */
963         rb_erase(&cap->ci_node, &ci->i_caps);
964         if (ci->i_auth_cap == cap)
965                 ci->i_auth_cap = NULL;
966
967         if (removed)
968                 ceph_put_cap(mdsc, cap);
969
970         /* when reconnect denied, we remove session caps forcibly,
971          * i_wr_ref can be non-zero. If there are ongoing write,
972          * keep i_snap_realm.
973          */
974         if (!__ceph_is_any_caps(ci) && ci->i_wr_ref == 0 && ci->i_snap_realm)
975                 drop_inode_snap_realm(ci);
976
977         if (!__ceph_is_any_real_caps(ci))
978                 __cap_delay_cancel(mdsc, ci);
979 }
980
981 /*
982  * Build and send a cap message to the given MDS.
983  *
984  * Caller should be holding s_mutex.
985  */
986 static int send_cap_msg(struct ceph_mds_session *session,
987                         u64 ino, u64 cid, int op,
988                         int caps, int wanted, int dirty,
989                         u32 seq, u64 flush_tid, u32 issue_seq, u32 mseq,
990                         u64 size, u64 max_size,
991                         struct timespec *mtime, struct timespec *atime,
992                         u64 time_warp_seq,
993                         kuid_t uid, kgid_t gid, umode_t mode,
994                         u64 xattr_version,
995                         struct ceph_buffer *xattrs_buf,
996                         u64 follows, bool inline_data)
997 {
998         struct ceph_mds_caps *fc;
999         struct ceph_msg *msg;
1000         void *p;
1001         size_t extra_len;
1002
1003         dout("send_cap_msg %s %llx %llx caps %s wanted %s dirty %s"
1004              " seq %u/%u mseq %u follows %lld size %llu/%llu"
1005              " xattr_ver %llu xattr_len %d\n", ceph_cap_op_name(op),
1006              cid, ino, ceph_cap_string(caps), ceph_cap_string(wanted),
1007              ceph_cap_string(dirty),
1008              seq, issue_seq, mseq, follows, size, max_size,
1009              xattr_version, xattrs_buf ? (int)xattrs_buf->vec.iov_len : 0);
1010
1011         /* flock buffer size + inline version + inline data size */
1012         extra_len = 4 + 8 + 4;
1013         msg = ceph_msg_new(CEPH_MSG_CLIENT_CAPS, sizeof(*fc) + extra_len,
1014                            GFP_NOFS, false);
1015         if (!msg)
1016                 return -ENOMEM;
1017
1018         msg->hdr.tid = cpu_to_le64(flush_tid);
1019
1020         fc = msg->front.iov_base;
1021         memset(fc, 0, sizeof(*fc));
1022
1023         fc->cap_id = cpu_to_le64(cid);
1024         fc->op = cpu_to_le32(op);
1025         fc->seq = cpu_to_le32(seq);
1026         fc->issue_seq = cpu_to_le32(issue_seq);
1027         fc->migrate_seq = cpu_to_le32(mseq);
1028         fc->caps = cpu_to_le32(caps);
1029         fc->wanted = cpu_to_le32(wanted);
1030         fc->dirty = cpu_to_le32(dirty);
1031         fc->ino = cpu_to_le64(ino);
1032         fc->snap_follows = cpu_to_le64(follows);
1033
1034         fc->size = cpu_to_le64(size);
1035         fc->max_size = cpu_to_le64(max_size);
1036         if (mtime)
1037                 ceph_encode_timespec(&fc->mtime, mtime);
1038         if (atime)
1039                 ceph_encode_timespec(&fc->atime, atime);
1040         fc->time_warp_seq = cpu_to_le32(time_warp_seq);
1041
1042         fc->uid = cpu_to_le32(from_kuid(&init_user_ns, uid));
1043         fc->gid = cpu_to_le32(from_kgid(&init_user_ns, gid));
1044         fc->mode = cpu_to_le32(mode);
1045
1046         p = fc + 1;
1047         /* flock buffer size */
1048         ceph_encode_32(&p, 0);
1049         /* inline version */
1050         ceph_encode_64(&p, inline_data ? 0 : CEPH_INLINE_NONE);
1051         /* inline data size */
1052         ceph_encode_32(&p, 0);
1053
1054         fc->xattr_version = cpu_to_le64(xattr_version);
1055         if (xattrs_buf) {
1056                 msg->middle = ceph_buffer_get(xattrs_buf);
1057                 fc->xattr_len = cpu_to_le32(xattrs_buf->vec.iov_len);
1058                 msg->hdr.middle_len = cpu_to_le32(xattrs_buf->vec.iov_len);
1059         }
1060
1061         ceph_con_send(&session->s_con, msg);
1062         return 0;
1063 }
1064
1065 /*
1066  * Queue cap releases when an inode is dropped from our cache.  Since
1067  * inode is about to be destroyed, there is no need for i_ceph_lock.
1068  */
1069 void ceph_queue_caps_release(struct inode *inode)
1070 {
1071         struct ceph_inode_info *ci = ceph_inode(inode);
1072         struct rb_node *p;
1073
1074         p = rb_first(&ci->i_caps);
1075         while (p) {
1076                 struct ceph_cap *cap = rb_entry(p, struct ceph_cap, ci_node);
1077                 p = rb_next(p);
1078                 __ceph_remove_cap(cap, true);
1079         }
1080 }
1081
1082 /*
1083  * Send a cap msg on the given inode.  Update our caps state, then
1084  * drop i_ceph_lock and send the message.
1085  *
1086  * Make note of max_size reported/requested from mds, revoked caps
1087  * that have now been implemented.
1088  *
1089  * Make half-hearted attempt ot to invalidate page cache if we are
1090  * dropping RDCACHE.  Note that this will leave behind locked pages
1091  * that we'll then need to deal with elsewhere.
1092  *
1093  * Return non-zero if delayed release, or we experienced an error
1094  * such that the caller should requeue + retry later.
1095  *
1096  * called with i_ceph_lock, then drops it.
1097  * caller should hold snap_rwsem (read), s_mutex.
1098  */
1099 static int __send_cap(struct ceph_mds_client *mdsc, struct ceph_cap *cap,
1100                       int op, int used, int want, int retain, int flushing,
1101                       unsigned *pflush_tid)
1102         __releases(cap->ci->i_ceph_lock)
1103 {
1104         struct ceph_inode_info *ci = cap->ci;
1105         struct inode *inode = &ci->vfs_inode;
1106         u64 cap_id = cap->cap_id;
1107         int held, revoking, dropping, keep;
1108         u64 seq, issue_seq, mseq, time_warp_seq, follows;
1109         u64 size, max_size;
1110         struct timespec mtime, atime;
1111         int wake = 0;
1112         umode_t mode;
1113         kuid_t uid;
1114         kgid_t gid;
1115         struct ceph_mds_session *session;
1116         u64 xattr_version = 0;
1117         struct ceph_buffer *xattr_blob = NULL;
1118         int delayed = 0;
1119         u64 flush_tid = 0;
1120         int i;
1121         int ret;
1122         bool inline_data;
1123
1124         held = cap->issued | cap->implemented;
1125         revoking = cap->implemented & ~cap->issued;
1126         retain &= ~revoking;
1127         dropping = cap->issued & ~retain;
1128
1129         dout("__send_cap %p cap %p session %p %s -> %s (revoking %s)\n",
1130              inode, cap, cap->session,
1131              ceph_cap_string(held), ceph_cap_string(held & retain),
1132              ceph_cap_string(revoking));
1133         BUG_ON((retain & CEPH_CAP_PIN) == 0);
1134
1135         session = cap->session;
1136
1137         /* don't release wanted unless we've waited a bit. */
1138         if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0 &&
1139             time_before(jiffies, ci->i_hold_caps_min)) {
1140                 dout(" delaying issued %s -> %s, wanted %s -> %s on send\n",
1141                      ceph_cap_string(cap->issued),
1142                      ceph_cap_string(cap->issued & retain),
1143                      ceph_cap_string(cap->mds_wanted),
1144                      ceph_cap_string(want));
1145                 want |= cap->mds_wanted;
1146                 retain |= cap->issued;
1147                 delayed = 1;
1148         }
1149         ci->i_ceph_flags &= ~(CEPH_I_NODELAY | CEPH_I_FLUSH);
1150
1151         cap->issued &= retain;  /* drop bits we don't want */
1152         if (cap->implemented & ~cap->issued) {
1153                 /*
1154                  * Wake up any waiters on wanted -> needed transition.
1155                  * This is due to the weird transition from buffered
1156                  * to sync IO... we need to flush dirty pages _before_
1157                  * allowing sync writes to avoid reordering.
1158                  */
1159                 wake = 1;
1160         }
1161         cap->implemented &= cap->issued | used;
1162         cap->mds_wanted = want;
1163
1164         if (flushing) {
1165                 /*
1166                  * assign a tid for flush operations so we can avoid
1167                  * flush1 -> dirty1 -> flush2 -> flushack1 -> mark
1168                  * clean type races.  track latest tid for every bit
1169                  * so we can handle flush AxFw, flush Fw, and have the
1170                  * first ack clean Ax.
1171                  */
1172                 flush_tid = ++ci->i_cap_flush_last_tid;
1173                 if (pflush_tid)
1174                         *pflush_tid = flush_tid;
1175                 dout(" cap_flush_tid %d\n", (int)flush_tid);
1176                 for (i = 0; i < CEPH_CAP_BITS; i++)
1177                         if (flushing & (1 << i))
1178                                 ci->i_cap_flush_tid[i] = flush_tid;
1179
1180                 follows = ci->i_head_snapc->seq;
1181         } else {
1182                 follows = 0;
1183         }
1184
1185         keep = cap->implemented;
1186         seq = cap->seq;
1187         issue_seq = cap->issue_seq;
1188         mseq = cap->mseq;
1189         size = inode->i_size;
1190         ci->i_reported_size = size;
1191         max_size = ci->i_wanted_max_size;
1192         ci->i_requested_max_size = max_size;
1193         mtime = inode->i_mtime;
1194         atime = inode->i_atime;
1195         time_warp_seq = ci->i_time_warp_seq;
1196         uid = inode->i_uid;
1197         gid = inode->i_gid;
1198         mode = inode->i_mode;
1199
1200         if (flushing & CEPH_CAP_XATTR_EXCL) {
1201                 __ceph_build_xattrs_blob(ci);
1202                 xattr_blob = ci->i_xattrs.blob;
1203                 xattr_version = ci->i_xattrs.version;
1204         }
1205
1206         inline_data = ci->i_inline_version != CEPH_INLINE_NONE;
1207
1208         spin_unlock(&ci->i_ceph_lock);
1209
1210         ret = send_cap_msg(session, ceph_vino(inode).ino, cap_id,
1211                 op, keep, want, flushing, seq, flush_tid, issue_seq, mseq,
1212                 size, max_size, &mtime, &atime, time_warp_seq,
1213                 uid, gid, mode, xattr_version, xattr_blob,
1214                 follows, inline_data);
1215         if (ret < 0) {
1216                 dout("error sending cap msg, must requeue %p\n", inode);
1217                 delayed = 1;
1218         }
1219
1220         if (wake)
1221                 wake_up_all(&ci->i_cap_wq);
1222
1223         return delayed;
1224 }
1225
1226 /*
1227  * When a snapshot is taken, clients accumulate dirty metadata on
1228  * inodes with capabilities in ceph_cap_snaps to describe the file
1229  * state at the time the snapshot was taken.  This must be flushed
1230  * asynchronously back to the MDS once sync writes complete and dirty
1231  * data is written out.
1232  *
1233  * Unless @kick is true, skip cap_snaps that were already sent to
1234  * the MDS (i.e., during this session).
1235  *
1236  * Called under i_ceph_lock.  Takes s_mutex as needed.
1237  */
1238 void __ceph_flush_snaps(struct ceph_inode_info *ci,
1239                         struct ceph_mds_session **psession,
1240                         int kick)
1241                 __releases(ci->i_ceph_lock)
1242                 __acquires(ci->i_ceph_lock)
1243 {
1244         struct inode *inode = &ci->vfs_inode;
1245         int mds;
1246         struct ceph_cap_snap *capsnap;
1247         u32 mseq;
1248         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
1249         struct ceph_mds_session *session = NULL; /* if session != NULL, we hold
1250                                                     session->s_mutex */
1251         u64 next_follows = 0;  /* keep track of how far we've gotten through the
1252                              i_cap_snaps list, and skip these entries next time
1253                              around to avoid an infinite loop */
1254
1255         if (psession)
1256                 session = *psession;
1257
1258         dout("__flush_snaps %p\n", inode);
1259 retry:
1260         list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
1261                 /* avoid an infiniute loop after retry */
1262                 if (capsnap->follows < next_follows)
1263                         continue;
1264                 /*
1265                  * we need to wait for sync writes to complete and for dirty
1266                  * pages to be written out.
1267                  */
1268                 if (capsnap->dirty_pages || capsnap->writing)
1269                         break;
1270
1271                 /* should be removed by ceph_try_drop_cap_snap() */
1272                 BUG_ON(!capsnap->need_flush);
1273
1274                 /* pick mds, take s_mutex */
1275                 if (ci->i_auth_cap == NULL) {
1276                         dout("no auth cap (migrating?), doing nothing\n");
1277                         goto out;
1278                 }
1279
1280                 /* only flush each capsnap once */
1281                 if (!kick && !list_empty(&capsnap->flushing_item)) {
1282                         dout("already flushed %p, skipping\n", capsnap);
1283                         continue;
1284                 }
1285
1286                 mds = ci->i_auth_cap->session->s_mds;
1287                 mseq = ci->i_auth_cap->mseq;
1288
1289                 if (session && session->s_mds != mds) {
1290                         dout("oops, wrong session %p mutex\n", session);
1291                         if (kick)
1292                                 goto out;
1293
1294                         mutex_unlock(&session->s_mutex);
1295                         ceph_put_mds_session(session);
1296                         session = NULL;
1297                 }
1298                 if (!session) {
1299                         spin_unlock(&ci->i_ceph_lock);
1300                         mutex_lock(&mdsc->mutex);
1301                         session = __ceph_lookup_mds_session(mdsc, mds);
1302                         mutex_unlock(&mdsc->mutex);
1303                         if (session) {
1304                                 dout("inverting session/ino locks on %p\n",
1305                                      session);
1306                                 mutex_lock(&session->s_mutex);
1307                         }
1308                         /*
1309                          * if session == NULL, we raced against a cap
1310                          * deletion or migration.  retry, and we'll
1311                          * get a better @mds value next time.
1312                          */
1313                         spin_lock(&ci->i_ceph_lock);
1314                         goto retry;
1315                 }
1316
1317                 capsnap->flush_tid = ++ci->i_cap_flush_last_tid;
1318                 atomic_inc(&capsnap->nref);
1319                 if (list_empty(&capsnap->flushing_item))
1320                         list_add_tail(&capsnap->flushing_item,
1321                                       &session->s_cap_snaps_flushing);
1322                 spin_unlock(&ci->i_ceph_lock);
1323
1324                 dout("flush_snaps %p cap_snap %p follows %lld tid %llu\n",
1325                      inode, capsnap, capsnap->follows, capsnap->flush_tid);
1326                 send_cap_msg(session, ceph_vino(inode).ino, 0,
1327                              CEPH_CAP_OP_FLUSHSNAP, capsnap->issued, 0,
1328                              capsnap->dirty, 0, capsnap->flush_tid, 0, mseq,
1329                              capsnap->size, 0,
1330                              &capsnap->mtime, &capsnap->atime,
1331                              capsnap->time_warp_seq,
1332                              capsnap->uid, capsnap->gid, capsnap->mode,
1333                              capsnap->xattr_version, capsnap->xattr_blob,
1334                              capsnap->follows, capsnap->inline_data);
1335
1336                 next_follows = capsnap->follows + 1;
1337                 ceph_put_cap_snap(capsnap);
1338
1339                 spin_lock(&ci->i_ceph_lock);
1340                 goto retry;
1341         }
1342
1343         /* we flushed them all; remove this inode from the queue */
1344         spin_lock(&mdsc->snap_flush_lock);
1345         list_del_init(&ci->i_snap_flush_item);
1346         spin_unlock(&mdsc->snap_flush_lock);
1347
1348 out:
1349         if (psession)
1350                 *psession = session;
1351         else if (session) {
1352                 mutex_unlock(&session->s_mutex);
1353                 ceph_put_mds_session(session);
1354         }
1355 }
1356
1357 static void ceph_flush_snaps(struct ceph_inode_info *ci)
1358 {
1359         spin_lock(&ci->i_ceph_lock);
1360         __ceph_flush_snaps(ci, NULL, 0);
1361         spin_unlock(&ci->i_ceph_lock);
1362 }
1363
1364 /*
1365  * Mark caps dirty.  If inode is newly dirty, return the dirty flags.
1366  * Caller is then responsible for calling __mark_inode_dirty with the
1367  * returned flags value.
1368  */
1369 int __ceph_mark_dirty_caps(struct ceph_inode_info *ci, int mask)
1370 {
1371         struct ceph_mds_client *mdsc =
1372                 ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc;
1373         struct inode *inode = &ci->vfs_inode;
1374         int was = ci->i_dirty_caps;
1375         int dirty = 0;
1376
1377         if (!ci->i_auth_cap) {
1378                 pr_warn("__mark_dirty_caps %p %llx mask %s, "
1379                         "but no auth cap (session was closed?)\n",
1380                         inode, ceph_ino(inode), ceph_cap_string(mask));
1381                 return 0;
1382         }
1383
1384         dout("__mark_dirty_caps %p %s dirty %s -> %s\n", &ci->vfs_inode,
1385              ceph_cap_string(mask), ceph_cap_string(was),
1386              ceph_cap_string(was | mask));
1387         ci->i_dirty_caps |= mask;
1388         if (was == 0) {
1389                 if (!ci->i_head_snapc) {
1390                         WARN_ON_ONCE(!rwsem_is_locked(&mdsc->snap_rwsem));
1391                         ci->i_head_snapc = ceph_get_snap_context(
1392                                 ci->i_snap_realm->cached_context);
1393                 }
1394                 dout(" inode %p now dirty snapc %p auth cap %p\n",
1395                      &ci->vfs_inode, ci->i_head_snapc, ci->i_auth_cap);
1396                 BUG_ON(!list_empty(&ci->i_dirty_item));
1397                 spin_lock(&mdsc->cap_dirty_lock);
1398                 list_add(&ci->i_dirty_item, &mdsc->cap_dirty);
1399                 spin_unlock(&mdsc->cap_dirty_lock);
1400                 if (ci->i_flushing_caps == 0) {
1401                         ihold(inode);
1402                         dirty |= I_DIRTY_SYNC;
1403                 }
1404         }
1405         BUG_ON(list_empty(&ci->i_dirty_item));
1406         if (((was | ci->i_flushing_caps) & CEPH_CAP_FILE_BUFFER) &&
1407             (mask & CEPH_CAP_FILE_BUFFER))
1408                 dirty |= I_DIRTY_DATASYNC;
1409         __cap_delay_requeue(mdsc, ci);
1410         return dirty;
1411 }
1412
1413 /*
1414  * Add dirty inode to the flushing list.  Assigned a seq number so we
1415  * can wait for caps to flush without starving.
1416  *
1417  * Called under i_ceph_lock.
1418  */
1419 static int __mark_caps_flushing(struct inode *inode,
1420                                  struct ceph_mds_session *session)
1421 {
1422         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
1423         struct ceph_inode_info *ci = ceph_inode(inode);
1424         int flushing;
1425
1426         BUG_ON(ci->i_dirty_caps == 0);
1427         BUG_ON(list_empty(&ci->i_dirty_item));
1428
1429         flushing = ci->i_dirty_caps;
1430         dout("__mark_caps_flushing flushing %s, flushing_caps %s -> %s\n",
1431              ceph_cap_string(flushing),
1432              ceph_cap_string(ci->i_flushing_caps),
1433              ceph_cap_string(ci->i_flushing_caps | flushing));
1434         ci->i_flushing_caps |= flushing;
1435         ci->i_dirty_caps = 0;
1436         dout(" inode %p now !dirty\n", inode);
1437
1438         spin_lock(&mdsc->cap_dirty_lock);
1439         list_del_init(&ci->i_dirty_item);
1440
1441         if (list_empty(&ci->i_flushing_item)) {
1442                 ci->i_cap_flush_seq = ++mdsc->cap_flush_seq;
1443                 list_add_tail(&ci->i_flushing_item, &session->s_cap_flushing);
1444                 mdsc->num_cap_flushing++;
1445                 dout(" inode %p now flushing seq %lld\n", inode,
1446                      ci->i_cap_flush_seq);
1447         } else {
1448                 list_move_tail(&ci->i_flushing_item, &session->s_cap_flushing);
1449                 dout(" inode %p now flushing (more) seq %lld\n", inode,
1450                      ci->i_cap_flush_seq);
1451         }
1452         spin_unlock(&mdsc->cap_dirty_lock);
1453
1454         return flushing;
1455 }
1456
1457 /*
1458  * try to invalidate mapping pages without blocking.
1459  */
1460 static int try_nonblocking_invalidate(struct inode *inode)
1461 {
1462         struct ceph_inode_info *ci = ceph_inode(inode);
1463         u32 invalidating_gen = ci->i_rdcache_gen;
1464
1465         spin_unlock(&ci->i_ceph_lock);
1466         invalidate_mapping_pages(&inode->i_data, 0, -1);
1467         spin_lock(&ci->i_ceph_lock);
1468
1469         if (inode->i_data.nrpages == 0 &&
1470             invalidating_gen == ci->i_rdcache_gen) {
1471                 /* success. */
1472                 dout("try_nonblocking_invalidate %p success\n", inode);
1473                 /* save any racing async invalidate some trouble */
1474                 ci->i_rdcache_revoking = ci->i_rdcache_gen - 1;
1475                 return 0;
1476         }
1477         dout("try_nonblocking_invalidate %p failed\n", inode);
1478         return -1;
1479 }
1480
1481 /*
1482  * Swiss army knife function to examine currently used and wanted
1483  * versus held caps.  Release, flush, ack revoked caps to mds as
1484  * appropriate.
1485  *
1486  *  CHECK_CAPS_NODELAY - caller is delayed work and we should not delay
1487  *    cap release further.
1488  *  CHECK_CAPS_AUTHONLY - we should only check the auth cap
1489  *  CHECK_CAPS_FLUSH - we should flush any dirty caps immediately, without
1490  *    further delay.
1491  */
1492 void ceph_check_caps(struct ceph_inode_info *ci, int flags,
1493                      struct ceph_mds_session *session)
1494 {
1495         struct ceph_fs_client *fsc = ceph_inode_to_client(&ci->vfs_inode);
1496         struct ceph_mds_client *mdsc = fsc->mdsc;
1497         struct inode *inode = &ci->vfs_inode;
1498         struct ceph_cap *cap;
1499         int file_wanted, used, cap_used;
1500         int took_snap_rwsem = 0;             /* true if mdsc->snap_rwsem held */
1501         int issued, implemented, want, retain, revoking, flushing = 0;
1502         int mds = -1;   /* keep track of how far we've gone through i_caps list
1503                            to avoid an infinite loop on retry */
1504         struct rb_node *p;
1505         int tried_invalidate = 0;
1506         int delayed = 0, sent = 0, force_requeue = 0, num;
1507         int queue_invalidate = 0;
1508         int is_delayed = flags & CHECK_CAPS_NODELAY;
1509
1510         /* if we are unmounting, flush any unused caps immediately. */
1511         if (mdsc->stopping)
1512                 is_delayed = 1;
1513
1514         spin_lock(&ci->i_ceph_lock);
1515
1516         if (ci->i_ceph_flags & CEPH_I_FLUSH)
1517                 flags |= CHECK_CAPS_FLUSH;
1518
1519         /* flush snaps first time around only */
1520         if (!list_empty(&ci->i_cap_snaps))
1521                 __ceph_flush_snaps(ci, &session, 0);
1522         goto retry_locked;
1523 retry:
1524         spin_lock(&ci->i_ceph_lock);
1525 retry_locked:
1526         file_wanted = __ceph_caps_file_wanted(ci);
1527         used = __ceph_caps_used(ci);
1528         issued = __ceph_caps_issued(ci, &implemented);
1529         revoking = implemented & ~issued;
1530
1531         want = file_wanted;
1532         retain = file_wanted | used | CEPH_CAP_PIN;
1533         if (!mdsc->stopping && inode->i_nlink > 0) {
1534                 if (file_wanted) {
1535                         retain |= CEPH_CAP_ANY;       /* be greedy */
1536                 } else if (S_ISDIR(inode->i_mode) &&
1537                            (issued & CEPH_CAP_FILE_SHARED) &&
1538                             __ceph_dir_is_complete(ci)) {
1539                         /*
1540                          * If a directory is complete, we want to keep
1541                          * the exclusive cap. So that MDS does not end up
1542                          * revoking the shared cap on every create/unlink
1543                          * operation.
1544                          */
1545                         want = CEPH_CAP_ANY_SHARED | CEPH_CAP_FILE_EXCL;
1546                         retain |= want;
1547                 } else {
1548
1549                         retain |= CEPH_CAP_ANY_SHARED;
1550                         /*
1551                          * keep RD only if we didn't have the file open RW,
1552                          * because then the mds would revoke it anyway to
1553                          * journal max_size=0.
1554                          */
1555                         if (ci->i_max_size == 0)
1556                                 retain |= CEPH_CAP_ANY_RD;
1557                 }
1558         }
1559
1560         dout("check_caps %p file_want %s used %s dirty %s flushing %s"
1561              " issued %s revoking %s retain %s %s%s%s\n", inode,
1562              ceph_cap_string(file_wanted),
1563              ceph_cap_string(used), ceph_cap_string(ci->i_dirty_caps),
1564              ceph_cap_string(ci->i_flushing_caps),
1565              ceph_cap_string(issued), ceph_cap_string(revoking),
1566              ceph_cap_string(retain),
1567              (flags & CHECK_CAPS_AUTHONLY) ? " AUTHONLY" : "",
1568              (flags & CHECK_CAPS_NODELAY) ? " NODELAY" : "",
1569              (flags & CHECK_CAPS_FLUSH) ? " FLUSH" : "");
1570
1571         /*
1572          * If we no longer need to hold onto old our caps, and we may
1573          * have cached pages, but don't want them, then try to invalidate.
1574          * If we fail, it's because pages are locked.... try again later.
1575          */
1576         if ((!is_delayed || mdsc->stopping) &&
1577             ci->i_wrbuffer_ref == 0 &&               /* no dirty pages... */
1578             inode->i_data.nrpages &&                 /* have cached pages */
1579             (file_wanted == 0 ||                     /* no open files */
1580              (revoking & (CEPH_CAP_FILE_CACHE|
1581                           CEPH_CAP_FILE_LAZYIO))) && /*  or revoking cache */
1582             !tried_invalidate) {
1583                 dout("check_caps trying to invalidate on %p\n", inode);
1584                 if (try_nonblocking_invalidate(inode) < 0) {
1585                         if (revoking & (CEPH_CAP_FILE_CACHE|
1586                                         CEPH_CAP_FILE_LAZYIO)) {
1587                                 dout("check_caps queuing invalidate\n");
1588                                 queue_invalidate = 1;
1589                                 ci->i_rdcache_revoking = ci->i_rdcache_gen;
1590                         } else {
1591                                 dout("check_caps failed to invalidate pages\n");
1592                                 /* we failed to invalidate pages.  check these
1593                                    caps again later. */
1594                                 force_requeue = 1;
1595                                 __cap_set_timeouts(mdsc, ci);
1596                         }
1597                 }
1598                 tried_invalidate = 1;
1599                 goto retry_locked;
1600         }
1601
1602         num = 0;
1603         for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
1604                 cap = rb_entry(p, struct ceph_cap, ci_node);
1605                 num++;
1606
1607                 /* avoid looping forever */
1608                 if (mds >= cap->mds ||
1609                     ((flags & CHECK_CAPS_AUTHONLY) && cap != ci->i_auth_cap))
1610                         continue;
1611
1612                 /* NOTE: no side-effects allowed, until we take s_mutex */
1613
1614                 cap_used = used;
1615                 if (ci->i_auth_cap && cap != ci->i_auth_cap)
1616                         cap_used &= ~ci->i_auth_cap->issued;
1617
1618                 revoking = cap->implemented & ~cap->issued;
1619                 dout(" mds%d cap %p used %s issued %s implemented %s revoking %s\n",
1620                      cap->mds, cap, ceph_cap_string(cap->issued),
1621                      ceph_cap_string(cap_used),
1622                      ceph_cap_string(cap->implemented),
1623                      ceph_cap_string(revoking));
1624
1625                 if (cap == ci->i_auth_cap &&
1626                     (cap->issued & CEPH_CAP_FILE_WR)) {
1627                         /* request larger max_size from MDS? */
1628                         if (ci->i_wanted_max_size > ci->i_max_size &&
1629                             ci->i_wanted_max_size > ci->i_requested_max_size) {
1630                                 dout("requesting new max_size\n");
1631                                 goto ack;
1632                         }
1633
1634                         /* approaching file_max? */
1635                         if ((inode->i_size << 1) >= ci->i_max_size &&
1636                             (ci->i_reported_size << 1) < ci->i_max_size) {
1637                                 dout("i_size approaching max_size\n");
1638                                 goto ack;
1639                         }
1640                 }
1641                 /* flush anything dirty? */
1642                 if (cap == ci->i_auth_cap && (flags & CHECK_CAPS_FLUSH) &&
1643                     ci->i_dirty_caps) {
1644                         dout("flushing dirty caps\n");
1645                         goto ack;
1646                 }
1647
1648                 /* completed revocation? going down and there are no caps? */
1649                 if (revoking && (revoking & cap_used) == 0) {
1650                         dout("completed revocation of %s\n",
1651                              ceph_cap_string(cap->implemented & ~cap->issued));
1652                         goto ack;
1653                 }
1654
1655                 /* want more caps from mds? */
1656                 if (want & ~(cap->mds_wanted | cap->issued))
1657                         goto ack;
1658
1659                 /* things we might delay */
1660                 if ((cap->issued & ~retain) == 0 &&
1661                     cap->mds_wanted == want)
1662                         continue;     /* nope, all good */
1663
1664                 if (is_delayed)
1665                         goto ack;
1666
1667                 /* delay? */
1668                 if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0 &&
1669                     time_before(jiffies, ci->i_hold_caps_max)) {
1670                         dout(" delaying issued %s -> %s, wanted %s -> %s\n",
1671                              ceph_cap_string(cap->issued),
1672                              ceph_cap_string(cap->issued & retain),
1673                              ceph_cap_string(cap->mds_wanted),
1674                              ceph_cap_string(want));
1675                         delayed++;
1676                         continue;
1677                 }
1678
1679 ack:
1680                 if (ci->i_ceph_flags & CEPH_I_NOFLUSH) {
1681                         dout(" skipping %p I_NOFLUSH set\n", inode);
1682                         continue;
1683                 }
1684
1685                 if (session && session != cap->session) {
1686                         dout("oops, wrong session %p mutex\n", session);
1687                         mutex_unlock(&session->s_mutex);
1688                         session = NULL;
1689                 }
1690                 if (!session) {
1691                         session = cap->session;
1692                         if (mutex_trylock(&session->s_mutex) == 0) {
1693                                 dout("inverting session/ino locks on %p\n",
1694                                      session);
1695                                 spin_unlock(&ci->i_ceph_lock);
1696                                 if (took_snap_rwsem) {
1697                                         up_read(&mdsc->snap_rwsem);
1698                                         took_snap_rwsem = 0;
1699                                 }
1700                                 mutex_lock(&session->s_mutex);
1701                                 goto retry;
1702                         }
1703                 }
1704                 /* take snap_rwsem after session mutex */
1705                 if (!took_snap_rwsem) {
1706                         if (down_read_trylock(&mdsc->snap_rwsem) == 0) {
1707                                 dout("inverting snap/in locks on %p\n",
1708                                      inode);
1709                                 spin_unlock(&ci->i_ceph_lock);
1710                                 down_read(&mdsc->snap_rwsem);
1711                                 took_snap_rwsem = 1;
1712                                 goto retry;
1713                         }
1714                         took_snap_rwsem = 1;
1715                 }
1716
1717                 if (cap == ci->i_auth_cap && ci->i_dirty_caps)
1718                         flushing = __mark_caps_flushing(inode, session);
1719                 else
1720                         flushing = 0;
1721
1722                 mds = cap->mds;  /* remember mds, so we don't repeat */
1723                 sent++;
1724
1725                 /* __send_cap drops i_ceph_lock */
1726                 delayed += __send_cap(mdsc, cap, CEPH_CAP_OP_UPDATE, cap_used,
1727                                       want, retain, flushing, NULL);
1728                 goto retry; /* retake i_ceph_lock and restart our cap scan. */
1729         }
1730
1731         /*
1732          * Reschedule delayed caps release if we delayed anything,
1733          * otherwise cancel.
1734          */
1735         if (delayed && is_delayed)
1736                 force_requeue = 1;   /* __send_cap delayed release; requeue */
1737         if (!delayed && !is_delayed)
1738                 __cap_delay_cancel(mdsc, ci);
1739         else if (!is_delayed || force_requeue)
1740                 __cap_delay_requeue(mdsc, ci);
1741
1742         spin_unlock(&ci->i_ceph_lock);
1743
1744         if (queue_invalidate)
1745                 ceph_queue_invalidate(inode);
1746
1747         if (session)
1748                 mutex_unlock(&session->s_mutex);
1749         if (took_snap_rwsem)
1750                 up_read(&mdsc->snap_rwsem);
1751 }
1752
1753 /*
1754  * Try to flush dirty caps back to the auth mds.
1755  */
1756 static int try_flush_caps(struct inode *inode, unsigned *flush_tid)
1757 {
1758         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
1759         struct ceph_inode_info *ci = ceph_inode(inode);
1760         int flushing = 0;
1761         struct ceph_mds_session *session = NULL;
1762
1763 retry:
1764         spin_lock(&ci->i_ceph_lock);
1765         if (ci->i_ceph_flags & CEPH_I_NOFLUSH) {
1766                 dout("try_flush_caps skipping %p I_NOFLUSH set\n", inode);
1767                 goto out;
1768         }
1769         if (ci->i_dirty_caps && ci->i_auth_cap) {
1770                 struct ceph_cap *cap = ci->i_auth_cap;
1771                 int used = __ceph_caps_used(ci);
1772                 int want = __ceph_caps_wanted(ci);
1773                 int delayed;
1774
1775                 if (!session || session != cap->session) {
1776                         spin_unlock(&ci->i_ceph_lock);
1777                         if (session)
1778                                 mutex_unlock(&session->s_mutex);
1779                         session = cap->session;
1780                         mutex_lock(&session->s_mutex);
1781                         goto retry;
1782                 }
1783                 if (cap->session->s_state < CEPH_MDS_SESSION_OPEN)
1784                         goto out;
1785
1786                 flushing = __mark_caps_flushing(inode, session);
1787
1788                 /* __send_cap drops i_ceph_lock */
1789                 delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH, used, want,
1790                                      cap->issued | cap->implemented, flushing,
1791                                      flush_tid);
1792                 if (!delayed)
1793                         goto out_unlocked;
1794
1795                 spin_lock(&ci->i_ceph_lock);
1796                 __cap_delay_requeue(mdsc, ci);
1797         }
1798 out:
1799         spin_unlock(&ci->i_ceph_lock);
1800 out_unlocked:
1801         if (session)
1802                 mutex_unlock(&session->s_mutex);
1803         return flushing;
1804 }
1805
1806 /*
1807  * Return true if we've flushed caps through the given flush_tid.
1808  */
1809 static int caps_are_flushed(struct inode *inode, unsigned tid)
1810 {
1811         struct ceph_inode_info *ci = ceph_inode(inode);
1812         int i, ret = 1;
1813
1814         spin_lock(&ci->i_ceph_lock);
1815         for (i = 0; i < CEPH_CAP_BITS; i++)
1816                 if ((ci->i_flushing_caps & (1 << i)) &&
1817                     ci->i_cap_flush_tid[i] <= tid) {
1818                         /* still flushing this bit */
1819                         ret = 0;
1820                         break;
1821                 }
1822         spin_unlock(&ci->i_ceph_lock);
1823         return ret;
1824 }
1825
1826 /*
1827  * Wait on any unsafe replies for the given inode.  First wait on the
1828  * newest request, and make that the upper bound.  Then, if there are
1829  * more requests, keep waiting on the oldest as long as it is still older
1830  * than the original request.
1831  */
1832 static void sync_write_wait(struct inode *inode)
1833 {
1834         struct ceph_inode_info *ci = ceph_inode(inode);
1835         struct list_head *head = &ci->i_unsafe_writes;
1836         struct ceph_osd_request *req;
1837         u64 last_tid;
1838
1839         spin_lock(&ci->i_unsafe_lock);
1840         if (list_empty(head))
1841                 goto out;
1842
1843         /* set upper bound as _last_ entry in chain */
1844         req = list_entry(head->prev, struct ceph_osd_request,
1845                          r_unsafe_item);
1846         last_tid = req->r_tid;
1847
1848         do {
1849                 ceph_osdc_get_request(req);
1850                 spin_unlock(&ci->i_unsafe_lock);
1851                 dout("sync_write_wait on tid %llu (until %llu)\n",
1852                      req->r_tid, last_tid);
1853                 wait_for_completion(&req->r_safe_completion);
1854                 spin_lock(&ci->i_unsafe_lock);
1855                 ceph_osdc_put_request(req);
1856
1857                 /*
1858                  * from here on look at first entry in chain, since we
1859                  * only want to wait for anything older than last_tid
1860                  */
1861                 if (list_empty(head))
1862                         break;
1863                 req = list_entry(head->next, struct ceph_osd_request,
1864                                  r_unsafe_item);
1865         } while (req->r_tid < last_tid);
1866 out:
1867         spin_unlock(&ci->i_unsafe_lock);
1868 }
1869
1870 int ceph_fsync(struct file *file, loff_t start, loff_t end, int datasync)
1871 {
1872         struct inode *inode = file->f_mapping->host;
1873         struct ceph_inode_info *ci = ceph_inode(inode);
1874         unsigned flush_tid;
1875         int ret;
1876         int dirty;
1877
1878         dout("fsync %p%s\n", inode, datasync ? " datasync" : "");
1879         sync_write_wait(inode);
1880
1881         ret = filemap_write_and_wait_range(inode->i_mapping, start, end);
1882         if (ret < 0)
1883                 return ret;
1884         mutex_lock(&inode->i_mutex);
1885
1886         dirty = try_flush_caps(inode, &flush_tid);
1887         dout("fsync dirty caps are %s\n", ceph_cap_string(dirty));
1888
1889         /*
1890          * only wait on non-file metadata writeback (the mds
1891          * can recover size and mtime, so we don't need to
1892          * wait for that)
1893          */
1894         if (!datasync && (dirty & ~CEPH_CAP_ANY_FILE_WR)) {
1895                 dout("fsync waiting for flush_tid %u\n", flush_tid);
1896                 ret = wait_event_interruptible(ci->i_cap_wq,
1897                                        caps_are_flushed(inode, flush_tid));
1898         }
1899
1900         dout("fsync %p%s done\n", inode, datasync ? " datasync" : "");
1901         mutex_unlock(&inode->i_mutex);
1902         return ret;
1903 }
1904
1905 /*
1906  * Flush any dirty caps back to the mds.  If we aren't asked to wait,
1907  * queue inode for flush but don't do so immediately, because we can
1908  * get by with fewer MDS messages if we wait for data writeback to
1909  * complete first.
1910  */
1911 int ceph_write_inode(struct inode *inode, struct writeback_control *wbc)
1912 {
1913         struct ceph_inode_info *ci = ceph_inode(inode);
1914         unsigned flush_tid;
1915         int err = 0;
1916         int dirty;
1917         int wait = wbc->sync_mode == WB_SYNC_ALL;
1918
1919         dout("write_inode %p wait=%d\n", inode, wait);
1920         if (wait) {
1921                 dirty = try_flush_caps(inode, &flush_tid);
1922                 if (dirty)
1923                         err = wait_event_interruptible(ci->i_cap_wq,
1924                                        caps_are_flushed(inode, flush_tid));
1925         } else {
1926                 struct ceph_mds_client *mdsc =
1927                         ceph_sb_to_client(inode->i_sb)->mdsc;
1928
1929                 spin_lock(&ci->i_ceph_lock);
1930                 if (__ceph_caps_dirty(ci))
1931                         __cap_delay_requeue_front(mdsc, ci);
1932                 spin_unlock(&ci->i_ceph_lock);
1933         }
1934         return err;
1935 }
1936
1937 /*
1938  * After a recovering MDS goes active, we need to resend any caps
1939  * we were flushing.
1940  *
1941  * Caller holds session->s_mutex.
1942  */
1943 static void kick_flushing_capsnaps(struct ceph_mds_client *mdsc,
1944                                    struct ceph_mds_session *session)
1945 {
1946         struct ceph_cap_snap *capsnap;
1947
1948         dout("kick_flushing_capsnaps mds%d\n", session->s_mds);
1949         list_for_each_entry(capsnap, &session->s_cap_snaps_flushing,
1950                             flushing_item) {
1951                 struct ceph_inode_info *ci = capsnap->ci;
1952                 struct inode *inode = &ci->vfs_inode;
1953                 struct ceph_cap *cap;
1954
1955                 spin_lock(&ci->i_ceph_lock);
1956                 cap = ci->i_auth_cap;
1957                 if (cap && cap->session == session) {
1958                         dout("kick_flushing_caps %p cap %p capsnap %p\n", inode,
1959                              cap, capsnap);
1960                         __ceph_flush_snaps(ci, &session, 1);
1961                 } else {
1962                         pr_err("%p auth cap %p not mds%d ???\n", inode,
1963                                cap, session->s_mds);
1964                 }
1965                 spin_unlock(&ci->i_ceph_lock);
1966         }
1967 }
1968
1969 void ceph_kick_flushing_caps(struct ceph_mds_client *mdsc,
1970                              struct ceph_mds_session *session)
1971 {
1972         struct ceph_inode_info *ci;
1973
1974         kick_flushing_capsnaps(mdsc, session);
1975
1976         dout("kick_flushing_caps mds%d\n", session->s_mds);
1977         list_for_each_entry(ci, &session->s_cap_flushing, i_flushing_item) {
1978                 struct inode *inode = &ci->vfs_inode;
1979                 struct ceph_cap *cap;
1980                 int delayed = 0;
1981
1982                 spin_lock(&ci->i_ceph_lock);
1983                 cap = ci->i_auth_cap;
1984                 if (cap && cap->session == session) {
1985                         dout("kick_flushing_caps %p cap %p %s\n", inode,
1986                              cap, ceph_cap_string(ci->i_flushing_caps));
1987                         delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH,
1988                                              __ceph_caps_used(ci),
1989                                              __ceph_caps_wanted(ci),
1990                                              cap->issued | cap->implemented,
1991                                              ci->i_flushing_caps, NULL);
1992                         if (delayed) {
1993                                 spin_lock(&ci->i_ceph_lock);
1994                                 __cap_delay_requeue(mdsc, ci);
1995                                 spin_unlock(&ci->i_ceph_lock);
1996                         }
1997                 } else {
1998                         pr_err("%p auth cap %p not mds%d ???\n", inode,
1999                                cap, session->s_mds);
2000                         spin_unlock(&ci->i_ceph_lock);
2001                 }
2002         }
2003 }
2004
2005 static void kick_flushing_inode_caps(struct ceph_mds_client *mdsc,
2006                                      struct ceph_mds_session *session,
2007                                      struct inode *inode)
2008 {
2009         struct ceph_inode_info *ci = ceph_inode(inode);
2010         struct ceph_cap *cap;
2011         int delayed = 0;
2012
2013         spin_lock(&ci->i_ceph_lock);
2014         cap = ci->i_auth_cap;
2015         dout("kick_flushing_inode_caps %p flushing %s flush_seq %lld\n", inode,
2016              ceph_cap_string(ci->i_flushing_caps), ci->i_cap_flush_seq);
2017
2018         __ceph_flush_snaps(ci, &session, 1);
2019
2020         if (ci->i_flushing_caps) {
2021                 spin_lock(&mdsc->cap_dirty_lock);
2022                 list_move_tail(&ci->i_flushing_item,
2023                                &cap->session->s_cap_flushing);
2024                 spin_unlock(&mdsc->cap_dirty_lock);
2025
2026                 delayed = __send_cap(mdsc, cap, CEPH_CAP_OP_FLUSH,
2027                                      __ceph_caps_used(ci),
2028                                      __ceph_caps_wanted(ci),
2029                                      cap->issued | cap->implemented,
2030                                      ci->i_flushing_caps, NULL);
2031                 if (delayed) {
2032                         spin_lock(&ci->i_ceph_lock);
2033                         __cap_delay_requeue(mdsc, ci);
2034                         spin_unlock(&ci->i_ceph_lock);
2035                 }
2036         } else {
2037                 spin_unlock(&ci->i_ceph_lock);
2038         }
2039 }
2040
2041
2042 /*
2043  * Take references to capabilities we hold, so that we don't release
2044  * them to the MDS prematurely.
2045  *
2046  * Protected by i_ceph_lock.
2047  */
2048 static void __take_cap_refs(struct ceph_inode_info *ci, int got,
2049                             bool snap_rwsem_locked)
2050 {
2051         if (got & CEPH_CAP_PIN)
2052                 ci->i_pin_ref++;
2053         if (got & CEPH_CAP_FILE_RD)
2054                 ci->i_rd_ref++;
2055         if (got & CEPH_CAP_FILE_CACHE)
2056                 ci->i_rdcache_ref++;
2057         if (got & CEPH_CAP_FILE_WR) {
2058                 if (ci->i_wr_ref == 0 && !ci->i_head_snapc) {
2059                         BUG_ON(!snap_rwsem_locked);
2060                         ci->i_head_snapc = ceph_get_snap_context(
2061                                         ci->i_snap_realm->cached_context);
2062                 }
2063                 ci->i_wr_ref++;
2064         }
2065         if (got & CEPH_CAP_FILE_BUFFER) {
2066                 if (ci->i_wb_ref == 0)
2067                         ihold(&ci->vfs_inode);
2068                 ci->i_wb_ref++;
2069                 dout("__take_cap_refs %p wb %d -> %d (?)\n",
2070                      &ci->vfs_inode, ci->i_wb_ref-1, ci->i_wb_ref);
2071         }
2072 }
2073
2074 /*
2075  * Try to grab cap references.  Specify those refs we @want, and the
2076  * minimal set we @need.  Also include the larger offset we are writing
2077  * to (when applicable), and check against max_size here as well.
2078  * Note that caller is responsible for ensuring max_size increases are
2079  * requested from the MDS.
2080  */
2081 static int try_get_cap_refs(struct ceph_inode_info *ci, int need, int want,
2082                             loff_t endoff, bool nonblock, int *got, int *err)
2083 {
2084         struct inode *inode = &ci->vfs_inode;
2085         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
2086         int ret = 0;
2087         int have, implemented;
2088         int file_wanted;
2089         bool snap_rwsem_locked = false;
2090
2091         dout("get_cap_refs %p need %s want %s\n", inode,
2092              ceph_cap_string(need), ceph_cap_string(want));
2093
2094 again:
2095         spin_lock(&ci->i_ceph_lock);
2096
2097         /* make sure file is actually open */
2098         file_wanted = __ceph_caps_file_wanted(ci);
2099         if ((file_wanted & need) == 0) {
2100                 dout("try_get_cap_refs need %s file_wanted %s, EBADF\n",
2101                      ceph_cap_string(need), ceph_cap_string(file_wanted));
2102                 *err = -EBADF;
2103                 ret = 1;
2104                 goto out_unlock;
2105         }
2106
2107         /* finish pending truncate */
2108         while (ci->i_truncate_pending) {
2109                 spin_unlock(&ci->i_ceph_lock);
2110                 if (snap_rwsem_locked) {
2111                         up_read(&mdsc->snap_rwsem);
2112                         snap_rwsem_locked = false;
2113                 }
2114                 __ceph_do_pending_vmtruncate(inode);
2115                 spin_lock(&ci->i_ceph_lock);
2116         }
2117
2118         have = __ceph_caps_issued(ci, &implemented);
2119
2120         if (have & need & CEPH_CAP_FILE_WR) {
2121                 if (endoff >= 0 && endoff > (loff_t)ci->i_max_size) {
2122                         dout("get_cap_refs %p endoff %llu > maxsize %llu\n",
2123                              inode, endoff, ci->i_max_size);
2124                         if (endoff > ci->i_requested_max_size) {
2125                                 *err = -EAGAIN;
2126                                 ret = 1;
2127                         }
2128                         goto out_unlock;
2129                 }
2130                 /*
2131                  * If a sync write is in progress, we must wait, so that we
2132                  * can get a final snapshot value for size+mtime.
2133                  */
2134                 if (__ceph_have_pending_cap_snap(ci)) {
2135                         dout("get_cap_refs %p cap_snap_pending\n", inode);
2136                         goto out_unlock;
2137                 }
2138         }
2139
2140         if ((have & need) == need) {
2141                 /*
2142                  * Look at (implemented & ~have & not) so that we keep waiting
2143                  * on transition from wanted -> needed caps.  This is needed
2144                  * for WRBUFFER|WR -> WR to avoid a new WR sync write from
2145                  * going before a prior buffered writeback happens.
2146                  */
2147                 int not = want & ~(have & need);
2148                 int revoking = implemented & ~have;
2149                 dout("get_cap_refs %p have %s but not %s (revoking %s)\n",
2150                      inode, ceph_cap_string(have), ceph_cap_string(not),
2151                      ceph_cap_string(revoking));
2152                 if ((revoking & not) == 0) {
2153                         if (!snap_rwsem_locked &&
2154                             !ci->i_head_snapc &&
2155                             (need & CEPH_CAP_FILE_WR)) {
2156                                 if (!down_read_trylock(&mdsc->snap_rwsem)) {
2157                                         /*
2158                                          * we can not call down_read() when
2159                                          * task isn't in TASK_RUNNING state
2160                                          */
2161                                         if (nonblock) {
2162                                                 *err = -EAGAIN;
2163                                                 ret = 1;
2164                                                 goto out_unlock;
2165                                         }
2166
2167                                         spin_unlock(&ci->i_ceph_lock);
2168                                         down_read(&mdsc->snap_rwsem);
2169                                         snap_rwsem_locked = true;
2170                                         goto again;
2171                                 }
2172                                 snap_rwsem_locked = true;
2173                         }
2174                         *got = need | (have & want);
2175                         __take_cap_refs(ci, *got, true);
2176                         ret = 1;
2177                 }
2178         } else {
2179                 int session_readonly = false;
2180                 if ((need & CEPH_CAP_FILE_WR) && ci->i_auth_cap) {
2181                         struct ceph_mds_session *s = ci->i_auth_cap->session;
2182                         spin_lock(&s->s_cap_lock);
2183                         session_readonly = s->s_readonly;
2184                         spin_unlock(&s->s_cap_lock);
2185                 }
2186                 if (session_readonly) {
2187                         dout("get_cap_refs %p needed %s but mds%d readonly\n",
2188                              inode, ceph_cap_string(need), ci->i_auth_cap->mds);
2189                         *err = -EROFS;
2190                         ret = 1;
2191                         goto out_unlock;
2192                 }
2193
2194                 dout("get_cap_refs %p have %s needed %s\n", inode,
2195                      ceph_cap_string(have), ceph_cap_string(need));
2196         }
2197 out_unlock:
2198         spin_unlock(&ci->i_ceph_lock);
2199         if (snap_rwsem_locked)
2200                 up_read(&mdsc->snap_rwsem);
2201
2202         dout("get_cap_refs %p ret %d got %s\n", inode,
2203              ret, ceph_cap_string(*got));
2204         return ret;
2205 }
2206
2207 /*
2208  * Check the offset we are writing up to against our current
2209  * max_size.  If necessary, tell the MDS we want to write to
2210  * a larger offset.
2211  */
2212 static void check_max_size(struct inode *inode, loff_t endoff)
2213 {
2214         struct ceph_inode_info *ci = ceph_inode(inode);
2215         int check = 0;
2216
2217         /* do we need to explicitly request a larger max_size? */
2218         spin_lock(&ci->i_ceph_lock);
2219         if (endoff >= ci->i_max_size && endoff > ci->i_wanted_max_size) {
2220                 dout("write %p at large endoff %llu, req max_size\n",
2221                      inode, endoff);
2222                 ci->i_wanted_max_size = endoff;
2223         }
2224         /* duplicate ceph_check_caps()'s logic */
2225         if (ci->i_auth_cap &&
2226             (ci->i_auth_cap->issued & CEPH_CAP_FILE_WR) &&
2227             ci->i_wanted_max_size > ci->i_max_size &&
2228             ci->i_wanted_max_size > ci->i_requested_max_size)
2229                 check = 1;
2230         spin_unlock(&ci->i_ceph_lock);
2231         if (check)
2232                 ceph_check_caps(ci, CHECK_CAPS_AUTHONLY, NULL);
2233 }
2234
2235 /*
2236  * Wait for caps, and take cap references.  If we can't get a WR cap
2237  * due to a small max_size, make sure we check_max_size (and possibly
2238  * ask the mds) so we don't get hung up indefinitely.
2239  */
2240 int ceph_get_caps(struct ceph_inode_info *ci, int need, int want,
2241                   loff_t endoff, int *got, struct page **pinned_page)
2242 {
2243         int _got, ret, err = 0;
2244
2245         ret = ceph_pool_perm_check(ci, need);
2246         if (ret < 0)
2247                 return ret;
2248
2249         while (true) {
2250                 if (endoff > 0)
2251                         check_max_size(&ci->vfs_inode, endoff);
2252
2253                 err = 0;
2254                 _got = 0;
2255                 ret = try_get_cap_refs(ci, need, want, endoff,
2256                                        false, &_got, &err);
2257                 if (ret) {
2258                         if (err == -EAGAIN)
2259                                 continue;
2260                         if (err < 0)
2261                                 return err;
2262                 } else {
2263                         ret = wait_event_interruptible(ci->i_cap_wq,
2264                                         try_get_cap_refs(ci, need, want, endoff,
2265                                                          true, &_got, &err));
2266                         if (err == -EAGAIN)
2267                                 continue;
2268                         if (err < 0)
2269                                 ret = err;
2270                         if (ret < 0)
2271                                 return ret;
2272                 }
2273
2274                 if (ci->i_inline_version != CEPH_INLINE_NONE &&
2275                     (_got & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) &&
2276                     i_size_read(&ci->vfs_inode) > 0) {
2277                         struct page *page =
2278                                 find_get_page(ci->vfs_inode.i_mapping, 0);
2279                         if (page) {
2280                                 if (PageUptodate(page)) {
2281                                         *pinned_page = page;
2282                                         break;
2283                                 }
2284                                 page_cache_release(page);
2285                         }
2286                         /*
2287                          * drop cap refs first because getattr while
2288                          * holding * caps refs can cause deadlock.
2289                          */
2290                         ceph_put_cap_refs(ci, _got);
2291                         _got = 0;
2292
2293                         /*
2294                          * getattr request will bring inline data into
2295                          * page cache
2296                          */
2297                         ret = __ceph_do_getattr(&ci->vfs_inode, NULL,
2298                                                 CEPH_STAT_CAP_INLINE_DATA,
2299                                                 true);
2300                         if (ret < 0)
2301                                 return ret;
2302                         continue;
2303                 }
2304                 break;
2305         }
2306
2307         *got = _got;
2308         return 0;
2309 }
2310
2311 /*
2312  * Take cap refs.  Caller must already know we hold at least one ref
2313  * on the caps in question or we don't know this is safe.
2314  */
2315 void ceph_get_cap_refs(struct ceph_inode_info *ci, int caps)
2316 {
2317         spin_lock(&ci->i_ceph_lock);
2318         __take_cap_refs(ci, caps, false);
2319         spin_unlock(&ci->i_ceph_lock);
2320 }
2321
2322
2323 /*
2324  * drop cap_snap that is not associated with any snapshot.
2325  * we don't need to send FLUSHSNAP message for it.
2326  */
2327 static int ceph_try_drop_cap_snap(struct ceph_cap_snap *capsnap)
2328 {
2329         if (!capsnap->need_flush &&
2330             !capsnap->writing && !capsnap->dirty_pages) {
2331
2332                 dout("dropping cap_snap %p follows %llu\n",
2333                      capsnap, capsnap->follows);
2334                 ceph_put_snap_context(capsnap->context);
2335                 list_del(&capsnap->ci_item);
2336                 list_del(&capsnap->flushing_item);
2337                 ceph_put_cap_snap(capsnap);
2338                 return 1;
2339         }
2340         return 0;
2341 }
2342
2343 /*
2344  * Release cap refs.
2345  *
2346  * If we released the last ref on any given cap, call ceph_check_caps
2347  * to release (or schedule a release).
2348  *
2349  * If we are releasing a WR cap (from a sync write), finalize any affected
2350  * cap_snap, and wake up any waiters.
2351  */
2352 void ceph_put_cap_refs(struct ceph_inode_info *ci, int had)
2353 {
2354         struct inode *inode = &ci->vfs_inode;
2355         int last = 0, put = 0, flushsnaps = 0, wake = 0;
2356
2357         spin_lock(&ci->i_ceph_lock);
2358         if (had & CEPH_CAP_PIN)
2359                 --ci->i_pin_ref;
2360         if (had & CEPH_CAP_FILE_RD)
2361                 if (--ci->i_rd_ref == 0)
2362                         last++;
2363         if (had & CEPH_CAP_FILE_CACHE)
2364                 if (--ci->i_rdcache_ref == 0)
2365                         last++;
2366         if (had & CEPH_CAP_FILE_BUFFER) {
2367                 if (--ci->i_wb_ref == 0) {
2368                         last++;
2369                         put++;
2370                 }
2371                 dout("put_cap_refs %p wb %d -> %d (?)\n",
2372                      inode, ci->i_wb_ref+1, ci->i_wb_ref);
2373         }
2374         if (had & CEPH_CAP_FILE_WR)
2375                 if (--ci->i_wr_ref == 0) {
2376                         last++;
2377                         if (__ceph_have_pending_cap_snap(ci)) {
2378                                 struct ceph_cap_snap *capsnap =
2379                                         list_last_entry(&ci->i_cap_snaps,
2380                                                         struct ceph_cap_snap,
2381                                                         ci_item);
2382                                 capsnap->writing = 0;
2383                                 if (ceph_try_drop_cap_snap(capsnap))
2384                                         put++;
2385                                 else if (__ceph_finish_cap_snap(ci, capsnap))
2386                                         flushsnaps = 1;
2387                                 wake = 1;
2388                         }
2389                         if (ci->i_wrbuffer_ref_head == 0 &&
2390                             ci->i_dirty_caps == 0 &&
2391                             ci->i_flushing_caps == 0) {
2392                                 BUG_ON(!ci->i_head_snapc);
2393                                 ceph_put_snap_context(ci->i_head_snapc);
2394                                 ci->i_head_snapc = NULL;
2395                         }
2396                         /* see comment in __ceph_remove_cap() */
2397                         if (!__ceph_is_any_caps(ci) && ci->i_snap_realm)
2398                                 drop_inode_snap_realm(ci);
2399                 }
2400         spin_unlock(&ci->i_ceph_lock);
2401
2402         dout("put_cap_refs %p had %s%s%s\n", inode, ceph_cap_string(had),
2403              last ? " last" : "", put ? " put" : "");
2404
2405         if (last && !flushsnaps)
2406                 ceph_check_caps(ci, 0, NULL);
2407         else if (flushsnaps)
2408                 ceph_flush_snaps(ci);
2409         if (wake)
2410                 wake_up_all(&ci->i_cap_wq);
2411         while (put-- > 0)
2412                 iput(inode);
2413 }
2414
2415 /*
2416  * Release @nr WRBUFFER refs on dirty pages for the given @snapc snap
2417  * context.  Adjust per-snap dirty page accounting as appropriate.
2418  * Once all dirty data for a cap_snap is flushed, flush snapped file
2419  * metadata back to the MDS.  If we dropped the last ref, call
2420  * ceph_check_caps.
2421  */
2422 void ceph_put_wrbuffer_cap_refs(struct ceph_inode_info *ci, int nr,
2423                                 struct ceph_snap_context *snapc)
2424 {
2425         struct inode *inode = &ci->vfs_inode;
2426         int last = 0;
2427         int complete_capsnap = 0;
2428         int drop_capsnap = 0;
2429         int found = 0;
2430         struct ceph_cap_snap *capsnap = NULL;
2431
2432         spin_lock(&ci->i_ceph_lock);
2433         ci->i_wrbuffer_ref -= nr;
2434         last = !ci->i_wrbuffer_ref;
2435
2436         if (ci->i_head_snapc == snapc) {
2437                 ci->i_wrbuffer_ref_head -= nr;
2438                 if (ci->i_wrbuffer_ref_head == 0 &&
2439                     ci->i_wr_ref == 0 &&
2440                     ci->i_dirty_caps == 0 &&
2441                     ci->i_flushing_caps == 0) {
2442                         BUG_ON(!ci->i_head_snapc);
2443                         ceph_put_snap_context(ci->i_head_snapc);
2444                         ci->i_head_snapc = NULL;
2445                 }
2446                 dout("put_wrbuffer_cap_refs on %p head %d/%d -> %d/%d %s\n",
2447                      inode,
2448                      ci->i_wrbuffer_ref+nr, ci->i_wrbuffer_ref_head+nr,
2449                      ci->i_wrbuffer_ref, ci->i_wrbuffer_ref_head,
2450                      last ? " LAST" : "");
2451         } else {
2452                 list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
2453                         if (capsnap->context == snapc) {
2454                                 found = 1;
2455                                 break;
2456                         }
2457                 }
2458                 BUG_ON(!found);
2459                 capsnap->dirty_pages -= nr;
2460                 if (capsnap->dirty_pages == 0) {
2461                         complete_capsnap = 1;
2462                         drop_capsnap = ceph_try_drop_cap_snap(capsnap);
2463                 }
2464                 dout("put_wrbuffer_cap_refs on %p cap_snap %p "
2465                      " snap %lld %d/%d -> %d/%d %s%s\n",
2466                      inode, capsnap, capsnap->context->seq,
2467                      ci->i_wrbuffer_ref+nr, capsnap->dirty_pages + nr,
2468                      ci->i_wrbuffer_ref, capsnap->dirty_pages,
2469                      last ? " (wrbuffer last)" : "",
2470                      complete_capsnap ? " (complete capsnap)" : "");
2471         }
2472
2473         spin_unlock(&ci->i_ceph_lock);
2474
2475         if (last) {
2476                 ceph_check_caps(ci, CHECK_CAPS_AUTHONLY, NULL);
2477                 iput(inode);
2478         } else if (complete_capsnap) {
2479                 ceph_flush_snaps(ci);
2480                 wake_up_all(&ci->i_cap_wq);
2481         }
2482         if (drop_capsnap)
2483                 iput(inode);
2484 }
2485
2486 /*
2487  * Invalidate unlinked inode's aliases, so we can drop the inode ASAP.
2488  */
2489 static void invalidate_aliases(struct inode *inode)
2490 {
2491         struct dentry *dn, *prev = NULL;
2492
2493         dout("invalidate_aliases inode %p\n", inode);
2494         d_prune_aliases(inode);
2495         /*
2496          * For non-directory inode, d_find_alias() only returns
2497          * hashed dentry. After calling d_invalidate(), the
2498          * dentry becomes unhashed.
2499          *
2500          * For directory inode, d_find_alias() can return
2501          * unhashed dentry. But directory inode should have
2502          * one alias at most.
2503          */
2504         while ((dn = d_find_alias(inode))) {
2505                 if (dn == prev) {
2506                         dput(dn);
2507                         break;
2508                 }
2509                 d_invalidate(dn);
2510                 if (prev)
2511                         dput(prev);
2512                 prev = dn;
2513         }
2514         if (prev)
2515                 dput(prev);
2516 }
2517
2518 /*
2519  * Handle a cap GRANT message from the MDS.  (Note that a GRANT may
2520  * actually be a revocation if it specifies a smaller cap set.)
2521  *
2522  * caller holds s_mutex and i_ceph_lock, we drop both.
2523  */
2524 static void handle_cap_grant(struct ceph_mds_client *mdsc,
2525                              struct inode *inode, struct ceph_mds_caps *grant,
2526                              u64 inline_version,
2527                              void *inline_data, int inline_len,
2528                              struct ceph_buffer *xattr_buf,
2529                              struct ceph_mds_session *session,
2530                              struct ceph_cap *cap, int issued)
2531         __releases(ci->i_ceph_lock)
2532         __releases(mdsc->snap_rwsem)
2533 {
2534         struct ceph_inode_info *ci = ceph_inode(inode);
2535         int mds = session->s_mds;
2536         int seq = le32_to_cpu(grant->seq);
2537         int newcaps = le32_to_cpu(grant->caps);
2538         int used, wanted, dirty;
2539         u64 size = le64_to_cpu(grant->size);
2540         u64 max_size = le64_to_cpu(grant->max_size);
2541         struct timespec mtime, atime, ctime;
2542         int check_caps = 0;
2543         bool wake = false;
2544         bool writeback = false;
2545         bool queue_trunc = false;
2546         bool queue_invalidate = false;
2547         bool queue_revalidate = false;
2548         bool deleted_inode = false;
2549         bool fill_inline = false;
2550
2551         dout("handle_cap_grant inode %p cap %p mds%d seq %d %s\n",
2552              inode, cap, mds, seq, ceph_cap_string(newcaps));
2553         dout(" size %llu max_size %llu, i_size %llu\n", size, max_size,
2554                 inode->i_size);
2555
2556
2557         /*
2558          * auth mds of the inode changed. we received the cap export message,
2559          * but still haven't received the cap import message. handle_cap_export
2560          * updated the new auth MDS' cap.
2561          *
2562          * "ceph_seq_cmp(seq, cap->seq) <= 0" means we are processing a message
2563          * that was sent before the cap import message. So don't remove caps.
2564          */
2565         if (ceph_seq_cmp(seq, cap->seq) <= 0) {
2566                 WARN_ON(cap != ci->i_auth_cap);
2567                 WARN_ON(cap->cap_id != le64_to_cpu(grant->cap_id));
2568                 seq = cap->seq;
2569                 newcaps |= cap->issued;
2570         }
2571
2572         /*
2573          * If CACHE is being revoked, and we have no dirty buffers,
2574          * try to invalidate (once).  (If there are dirty buffers, we
2575          * will invalidate _after_ writeback.)
2576          */
2577         if (((cap->issued & ~newcaps) & CEPH_CAP_FILE_CACHE) &&
2578             (newcaps & CEPH_CAP_FILE_LAZYIO) == 0 &&
2579             !ci->i_wrbuffer_ref) {
2580                 if (try_nonblocking_invalidate(inode)) {
2581                         /* there were locked pages.. invalidate later
2582                            in a separate thread. */
2583                         if (ci->i_rdcache_revoking != ci->i_rdcache_gen) {
2584                                 queue_invalidate = true;
2585                                 ci->i_rdcache_revoking = ci->i_rdcache_gen;
2586                         }
2587                 }
2588
2589                 ceph_fscache_invalidate(inode);
2590         }
2591
2592         /* side effects now are allowed */
2593         cap->cap_gen = session->s_cap_gen;
2594         cap->seq = seq;
2595
2596         __check_cap_issue(ci, cap, newcaps);
2597
2598         if ((newcaps & CEPH_CAP_AUTH_SHARED) &&
2599             (issued & CEPH_CAP_AUTH_EXCL) == 0) {
2600                 inode->i_mode = le32_to_cpu(grant->mode);
2601                 inode->i_uid = make_kuid(&init_user_ns, le32_to_cpu(grant->uid));
2602                 inode->i_gid = make_kgid(&init_user_ns, le32_to_cpu(grant->gid));
2603                 dout("%p mode 0%o uid.gid %d.%d\n", inode, inode->i_mode,
2604                      from_kuid(&init_user_ns, inode->i_uid),
2605                      from_kgid(&init_user_ns, inode->i_gid));
2606         }
2607
2608         if ((newcaps & CEPH_CAP_AUTH_SHARED) &&
2609             (issued & CEPH_CAP_LINK_EXCL) == 0) {
2610                 set_nlink(inode, le32_to_cpu(grant->nlink));
2611                 if (inode->i_nlink == 0 &&
2612                     (newcaps & (CEPH_CAP_LINK_SHARED | CEPH_CAP_LINK_EXCL)))
2613                         deleted_inode = true;
2614         }
2615
2616         if ((issued & CEPH_CAP_XATTR_EXCL) == 0 && grant->xattr_len) {
2617                 int len = le32_to_cpu(grant->xattr_len);
2618                 u64 version = le64_to_cpu(grant->xattr_version);
2619
2620                 if (version > ci->i_xattrs.version) {
2621                         dout(" got new xattrs v%llu on %p len %d\n",
2622                              version, inode, len);
2623                         if (ci->i_xattrs.blob)
2624                                 ceph_buffer_put(ci->i_xattrs.blob);
2625                         ci->i_xattrs.blob = ceph_buffer_get(xattr_buf);
2626                         ci->i_xattrs.version = version;
2627                         ceph_forget_all_cached_acls(inode);
2628                 }
2629         }
2630
2631         /* Do we need to revalidate our fscache cookie. Don't bother on the
2632          * first cache cap as we already validate at cookie creation time. */
2633         if ((issued & CEPH_CAP_FILE_CACHE) && ci->i_rdcache_gen > 1)
2634                 queue_revalidate = true;
2635
2636         if (newcaps & CEPH_CAP_ANY_RD) {
2637                 /* ctime/mtime/atime? */
2638                 ceph_decode_timespec(&mtime, &grant->mtime);
2639                 ceph_decode_timespec(&atime, &grant->atime);
2640                 ceph_decode_timespec(&ctime, &grant->ctime);
2641                 ceph_fill_file_time(inode, issued,
2642                                     le32_to_cpu(grant->time_warp_seq),
2643                                     &ctime, &mtime, &atime);
2644         }
2645
2646         if (newcaps & (CEPH_CAP_ANY_FILE_RD | CEPH_CAP_ANY_FILE_WR)) {
2647                 /* file layout may have changed */
2648                 ci->i_layout = grant->layout;
2649                 /* size/truncate_seq? */
2650                 queue_trunc = ceph_fill_file_size(inode, issued,
2651                                         le32_to_cpu(grant->truncate_seq),
2652                                         le64_to_cpu(grant->truncate_size),
2653                                         size);
2654                 /* max size increase? */
2655                 if (ci->i_auth_cap == cap && max_size != ci->i_max_size) {
2656                         dout("max_size %lld -> %llu\n",
2657                              ci->i_max_size, max_size);
2658                         ci->i_max_size = max_size;
2659                         if (max_size >= ci->i_wanted_max_size) {
2660                                 ci->i_wanted_max_size = 0;  /* reset */
2661                                 ci->i_requested_max_size = 0;
2662                         }
2663                         wake = true;
2664                 }
2665         }
2666
2667         /* check cap bits */
2668         wanted = __ceph_caps_wanted(ci);
2669         used = __ceph_caps_used(ci);
2670         dirty = __ceph_caps_dirty(ci);
2671         dout(" my wanted = %s, used = %s, dirty %s\n",
2672              ceph_cap_string(wanted),
2673              ceph_cap_string(used),
2674              ceph_cap_string(dirty));
2675         if (wanted != le32_to_cpu(grant->wanted)) {
2676                 dout("mds wanted %s -> %s\n",
2677                      ceph_cap_string(le32_to_cpu(grant->wanted)),
2678                      ceph_cap_string(wanted));
2679                 /* imported cap may not have correct mds_wanted */
2680                 if (le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT)
2681                         check_caps = 1;
2682         }
2683
2684         /* revocation, grant, or no-op? */
2685         if (cap->issued & ~newcaps) {
2686                 int revoking = cap->issued & ~newcaps;
2687
2688                 dout("revocation: %s -> %s (revoking %s)\n",
2689                      ceph_cap_string(cap->issued),
2690                      ceph_cap_string(newcaps),
2691                      ceph_cap_string(revoking));
2692                 if (revoking & used & CEPH_CAP_FILE_BUFFER)
2693                         writeback = true;  /* initiate writeback; will delay ack */
2694                 else if (revoking == CEPH_CAP_FILE_CACHE &&
2695                          (newcaps & CEPH_CAP_FILE_LAZYIO) == 0 &&
2696                          queue_invalidate)
2697                         ; /* do nothing yet, invalidation will be queued */
2698                 else if (cap == ci->i_auth_cap)
2699                         check_caps = 1; /* check auth cap only */
2700                 else
2701                         check_caps = 2; /* check all caps */
2702                 cap->issued = newcaps;
2703                 cap->implemented |= newcaps;
2704         } else if (cap->issued == newcaps) {
2705                 dout("caps unchanged: %s -> %s\n",
2706                      ceph_cap_string(cap->issued), ceph_cap_string(newcaps));
2707         } else {
2708                 dout("grant: %s -> %s\n", ceph_cap_string(cap->issued),
2709                      ceph_cap_string(newcaps));
2710                 /* non-auth MDS is revoking the newly grant caps ? */
2711                 if (cap == ci->i_auth_cap &&
2712                     __ceph_caps_revoking_other(ci, cap, newcaps))
2713                     check_caps = 2;
2714
2715                 cap->issued = newcaps;
2716                 cap->implemented |= newcaps; /* add bits only, to
2717                                               * avoid stepping on a
2718                                               * pending revocation */
2719                 wake = true;
2720         }
2721         BUG_ON(cap->issued & ~cap->implemented);
2722
2723         if (inline_version > 0 && inline_version >= ci->i_inline_version) {
2724                 ci->i_inline_version = inline_version;
2725                 if (ci->i_inline_version != CEPH_INLINE_NONE &&
2726                     (newcaps & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)))
2727                         fill_inline = true;
2728         }
2729
2730         spin_unlock(&ci->i_ceph_lock);
2731
2732         if (le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT) {
2733                 kick_flushing_inode_caps(mdsc, session, inode);
2734                 up_read(&mdsc->snap_rwsem);
2735                 if (newcaps & ~issued)
2736                         wake = true;
2737         }
2738
2739         if (fill_inline)
2740                 ceph_fill_inline_data(inode, NULL, inline_data, inline_len);
2741
2742         if (queue_trunc) {
2743                 ceph_queue_vmtruncate(inode);
2744                 ceph_queue_revalidate(inode);
2745         } else if (queue_revalidate)
2746                 ceph_queue_revalidate(inode);
2747
2748         if (writeback)
2749                 /*
2750                  * queue inode for writeback: we can't actually call
2751                  * filemap_write_and_wait, etc. from message handler
2752                  * context.
2753                  */
2754                 ceph_queue_writeback(inode);
2755         if (queue_invalidate)
2756                 ceph_queue_invalidate(inode);
2757         if (deleted_inode)
2758                 invalidate_aliases(inode);
2759         if (wake)
2760                 wake_up_all(&ci->i_cap_wq);
2761
2762         if (check_caps == 1)
2763                 ceph_check_caps(ci, CHECK_CAPS_NODELAY|CHECK_CAPS_AUTHONLY,
2764                                 session);
2765         else if (check_caps == 2)
2766                 ceph_check_caps(ci, CHECK_CAPS_NODELAY, session);
2767         else
2768                 mutex_unlock(&session->s_mutex);
2769 }
2770
2771 /*
2772  * Handle FLUSH_ACK from MDS, indicating that metadata we sent to the
2773  * MDS has been safely committed.
2774  */
2775 static void handle_cap_flush_ack(struct inode *inode, u64 flush_tid,
2776                                  struct ceph_mds_caps *m,
2777                                  struct ceph_mds_session *session,
2778                                  struct ceph_cap *cap)
2779         __releases(ci->i_ceph_lock)
2780 {
2781         struct ceph_inode_info *ci = ceph_inode(inode);
2782         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
2783         unsigned seq = le32_to_cpu(m->seq);
2784         int dirty = le32_to_cpu(m->dirty);
2785         int cleaned = 0;
2786         int drop = 0;
2787         int i;
2788
2789         for (i = 0; i < CEPH_CAP_BITS; i++)
2790                 if ((dirty & (1 << i)) &&
2791                     (u16)flush_tid == ci->i_cap_flush_tid[i])
2792                         cleaned |= 1 << i;
2793
2794         dout("handle_cap_flush_ack inode %p mds%d seq %d on %s cleaned %s,"
2795              " flushing %s -> %s\n",
2796              inode, session->s_mds, seq, ceph_cap_string(dirty),
2797              ceph_cap_string(cleaned), ceph_cap_string(ci->i_flushing_caps),
2798              ceph_cap_string(ci->i_flushing_caps & ~cleaned));
2799
2800         if (ci->i_flushing_caps == (ci->i_flushing_caps & ~cleaned))
2801                 goto out;
2802
2803         ci->i_flushing_caps &= ~cleaned;
2804
2805         spin_lock(&mdsc->cap_dirty_lock);
2806         if (ci->i_flushing_caps == 0) {
2807                 list_del_init(&ci->i_flushing_item);
2808                 if (!list_empty(&session->s_cap_flushing))
2809                         dout(" mds%d still flushing cap on %p\n",
2810                              session->s_mds,
2811                              &list_entry(session->s_cap_flushing.next,
2812                                          struct ceph_inode_info,
2813                                          i_flushing_item)->vfs_inode);
2814                 mdsc->num_cap_flushing--;
2815                 wake_up_all(&mdsc->cap_flushing_wq);
2816                 dout(" inode %p now !flushing\n", inode);
2817
2818                 if (ci->i_dirty_caps == 0) {
2819                         dout(" inode %p now clean\n", inode);
2820                         BUG_ON(!list_empty(&ci->i_dirty_item));
2821                         drop = 1;
2822                         if (ci->i_wr_ref == 0 &&
2823                             ci->i_wrbuffer_ref_head == 0) {
2824                                 BUG_ON(!ci->i_head_snapc);
2825                                 ceph_put_snap_context(ci->i_head_snapc);
2826                                 ci->i_head_snapc = NULL;
2827                         }
2828                 } else {
2829                         BUG_ON(list_empty(&ci->i_dirty_item));
2830                 }
2831         }
2832         spin_unlock(&mdsc->cap_dirty_lock);
2833         wake_up_all(&ci->i_cap_wq);
2834
2835 out:
2836         spin_unlock(&ci->i_ceph_lock);
2837         if (drop)
2838                 iput(inode);
2839 }
2840
2841 /*
2842  * Handle FLUSHSNAP_ACK.  MDS has flushed snap data to disk and we can
2843  * throw away our cap_snap.
2844  *
2845  * Caller hold s_mutex.
2846  */
2847 static void handle_cap_flushsnap_ack(struct inode *inode, u64 flush_tid,
2848                                      struct ceph_mds_caps *m,
2849                                      struct ceph_mds_session *session)
2850 {
2851         struct ceph_inode_info *ci = ceph_inode(inode);
2852         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
2853         u64 follows = le64_to_cpu(m->snap_follows);
2854         struct ceph_cap_snap *capsnap;
2855         int drop = 0;
2856
2857         dout("handle_cap_flushsnap_ack inode %p ci %p mds%d follows %lld\n",
2858              inode, ci, session->s_mds, follows);
2859
2860         spin_lock(&ci->i_ceph_lock);
2861         list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
2862                 if (capsnap->follows == follows) {
2863                         if (capsnap->flush_tid != flush_tid) {
2864                                 dout(" cap_snap %p follows %lld tid %lld !="
2865                                      " %lld\n", capsnap, follows,
2866                                      flush_tid, capsnap->flush_tid);
2867                                 break;
2868                         }
2869                         WARN_ON(capsnap->dirty_pages || capsnap->writing);
2870                         dout(" removing %p cap_snap %p follows %lld\n",
2871                              inode, capsnap, follows);
2872                         ceph_put_snap_context(capsnap->context);
2873                         list_del(&capsnap->ci_item);
2874                         list_del(&capsnap->flushing_item);
2875                         ceph_put_cap_snap(capsnap);
2876                         wake_up_all(&mdsc->cap_flushing_wq);
2877                         drop = 1;
2878                         break;
2879                 } else {
2880                         dout(" skipping cap_snap %p follows %lld\n",
2881                              capsnap, capsnap->follows);
2882                 }
2883         }
2884         spin_unlock(&ci->i_ceph_lock);
2885         if (drop)
2886                 iput(inode);
2887 }
2888
2889 /*
2890  * Handle TRUNC from MDS, indicating file truncation.
2891  *
2892  * caller hold s_mutex.
2893  */
2894 static void handle_cap_trunc(struct inode *inode,
2895                              struct ceph_mds_caps *trunc,
2896                              struct ceph_mds_session *session)
2897         __releases(ci->i_ceph_lock)
2898 {
2899         struct ceph_inode_info *ci = ceph_inode(inode);
2900         int mds = session->s_mds;
2901         int seq = le32_to_cpu(trunc->seq);
2902         u32 truncate_seq = le32_to_cpu(trunc->truncate_seq);
2903         u64 truncate_size = le64_to_cpu(trunc->truncate_size);
2904         u64 size = le64_to_cpu(trunc->size);
2905         int implemented = 0;
2906         int dirty = __ceph_caps_dirty(ci);
2907         int issued = __ceph_caps_issued(ceph_inode(inode), &implemented);
2908         int queue_trunc = 0;
2909
2910         issued |= implemented | dirty;
2911
2912         dout("handle_cap_trunc inode %p mds%d seq %d to %lld seq %d\n",
2913              inode, mds, seq, truncate_size, truncate_seq);
2914         queue_trunc = ceph_fill_file_size(inode, issued,
2915                                           truncate_seq, truncate_size, size);
2916         spin_unlock(&ci->i_ceph_lock);
2917
2918         if (queue_trunc) {
2919                 ceph_queue_vmtruncate(inode);
2920                 ceph_fscache_invalidate(inode);
2921         }
2922 }
2923
2924 /*
2925  * Handle EXPORT from MDS.  Cap is being migrated _from_ this mds to a
2926  * different one.  If we are the most recent migration we've seen (as
2927  * indicated by mseq), make note of the migrating cap bits for the
2928  * duration (until we see the corresponding IMPORT).
2929  *
2930  * caller holds s_mutex
2931  */
2932 static void handle_cap_export(struct inode *inode, struct ceph_mds_caps *ex,
2933                               struct ceph_mds_cap_peer *ph,
2934                               struct ceph_mds_session *session)
2935 {
2936         struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
2937         struct ceph_mds_session *tsession = NULL;
2938         struct ceph_cap *cap, *tcap, *new_cap = NULL;
2939         struct ceph_inode_info *ci = ceph_inode(inode);
2940         u64 t_cap_id;
2941         unsigned mseq = le32_to_cpu(ex->migrate_seq);
2942         unsigned t_seq, t_mseq;
2943         int target, issued;
2944         int mds = session->s_mds;
2945
2946         if (ph) {
2947                 t_cap_id = le64_to_cpu(ph->cap_id);
2948                 t_seq = le32_to_cpu(ph->seq);
2949                 t_mseq = le32_to_cpu(ph->mseq);
2950                 target = le32_to_cpu(ph->mds);
2951         } else {
2952                 t_cap_id = t_seq = t_mseq = 0;
2953                 target = -1;
2954         }
2955
2956         dout("handle_cap_export inode %p ci %p mds%d mseq %d target %d\n",
2957              inode, ci, mds, mseq, target);
2958 retry:
2959         spin_lock(&ci->i_ceph_lock);
2960         cap = __get_cap_for_mds(ci, mds);
2961         if (!cap || cap->cap_id != le64_to_cpu(ex->cap_id))
2962                 goto out_unlock;
2963
2964         if (target < 0) {
2965                 __ceph_remove_cap(cap, false);
2966                 goto out_unlock;
2967         }
2968
2969         /*
2970          * now we know we haven't received the cap import message yet
2971          * because the exported cap still exist.
2972          */
2973
2974         issued = cap->issued;
2975         WARN_ON(issued != cap->implemented);
2976
2977         tcap = __get_cap_for_mds(ci, target);
2978         if (tcap) {
2979                 /* already have caps from the target */
2980                 if (tcap->cap_id != t_cap_id ||
2981                     ceph_seq_cmp(tcap->seq, t_seq) < 0) {
2982                         dout(" updating import cap %p mds%d\n", tcap, target);
2983                         tcap->cap_id = t_cap_id;
2984                         tcap->seq = t_seq - 1;
2985                         tcap->issue_seq = t_seq - 1;
2986                         tcap->mseq = t_mseq;
2987                         tcap->issued |= issued;
2988                         tcap->implemented |= issued;
2989                         if (cap == ci->i_auth_cap)
2990                                 ci->i_auth_cap = tcap;
2991                         if (ci->i_flushing_caps && ci->i_auth_cap == tcap) {
2992                                 spin_lock(&mdsc->cap_dirty_lock);
2993                                 list_move_tail(&ci->i_flushing_item,
2994                                                &tcap->session->s_cap_flushing);
2995                                 spin_unlock(&mdsc->cap_dirty_lock);
2996                         }
2997                 }
2998                 __ceph_remove_cap(cap, false);
2999                 goto out_unlock;
3000         } else if (tsession) {
3001                 /* add placeholder for the export tagert */
3002                 int flag = (cap == ci->i_auth_cap) ? CEPH_CAP_FLAG_AUTH : 0;
3003                 ceph_add_cap(inode, tsession, t_cap_id, -1, issued, 0,
3004                              t_seq - 1, t_mseq, (u64)-1, flag, &new_cap);
3005
3006                 __ceph_remove_cap(cap, false);
3007                 goto out_unlock;
3008         }
3009
3010         spin_unlock(&ci->i_ceph_lock);
3011         mutex_unlock(&session->s_mutex);
3012
3013         /* open target session */
3014         tsession = ceph_mdsc_open_export_target_session(mdsc, target);
3015         if (!IS_ERR(tsession)) {
3016                 if (mds > target) {
3017                         mutex_lock(&session->s_mutex);
3018                         mutex_lock_nested(&tsession->s_mutex,
3019                                           SINGLE_DEPTH_NESTING);
3020                 } else {
3021                         mutex_lock(&tsession->s_mutex);
3022                         mutex_lock_nested(&session->s_mutex,
3023                                           SINGLE_DEPTH_NESTING);
3024                 }
3025                 new_cap = ceph_get_cap(mdsc, NULL);
3026         } else {
3027                 WARN_ON(1);
3028                 tsession = NULL;
3029                 target = -1;
3030         }
3031         goto retry;
3032
3033 out_unlock:
3034         spin_unlock(&ci->i_ceph_lock);
3035         mutex_unlock(&session->s_mutex);
3036         if (tsession) {
3037                 mutex_unlock(&tsession->s_mutex);
3038                 ceph_put_mds_session(tsession);
3039         }
3040         if (new_cap)
3041                 ceph_put_cap(mdsc, new_cap);
3042 }
3043
3044 /*
3045  * Handle cap IMPORT.
3046  *
3047  * caller holds s_mutex. acquires i_ceph_lock
3048  */
3049 static void handle_cap_import(struct ceph_mds_client *mdsc,
3050                               struct inode *inode, struct ceph_mds_caps *im,
3051                               struct ceph_mds_cap_peer *ph,
3052                               struct ceph_mds_session *session,
3053                               struct ceph_cap **target_cap, int *old_issued)
3054         __acquires(ci->i_ceph_lock)
3055 {
3056         struct ceph_inode_info *ci = ceph_inode(inode);
3057         struct ceph_cap *cap, *ocap, *new_cap = NULL;
3058         int mds = session->s_mds;
3059         int issued;
3060         unsigned caps = le32_to_cpu(im->caps);
3061         unsigned wanted = le32_to_cpu(im->wanted);
3062         unsigned seq = le32_to_cpu(im->seq);
3063         unsigned mseq = le32_to_cpu(im->migrate_seq);
3064         u64 realmino = le64_to_cpu(im->realm);
3065         u64 cap_id = le64_to_cpu(im->cap_id);
3066         u64 p_cap_id;
3067         int peer;
3068
3069         if (ph) {
3070                 p_cap_id = le64_to_cpu(ph->cap_id);
3071                 peer = le32_to_cpu(ph->mds);
3072         } else {
3073                 p_cap_id = 0;
3074                 peer = -1;
3075         }
3076
3077         dout("handle_cap_import inode %p ci %p mds%d mseq %d peer %d\n",
3078              inode, ci, mds, mseq, peer);
3079
3080 retry:
3081         spin_lock(&ci->i_ceph_lock);
3082         cap = __get_cap_for_mds(ci, mds);
3083         if (!cap) {
3084                 if (!new_cap) {
3085                         spin_unlock(&ci->i_ceph_lock);
3086                         new_cap = ceph_get_cap(mdsc, NULL);
3087                         goto retry;
3088                 }
3089                 cap = new_cap;
3090         } else {
3091                 if (new_cap) {
3092                         ceph_put_cap(mdsc, new_cap);
3093                         new_cap = NULL;
3094                 }
3095         }
3096
3097         __ceph_caps_issued(ci, &issued);
3098         issued |= __ceph_caps_dirty(ci);
3099
3100         ceph_add_cap(inode, session, cap_id, -1, caps, wanted, seq, mseq,
3101                      realmino, CEPH_CAP_FLAG_AUTH, &new_cap);
3102
3103         ocap = peer >= 0 ? __get_cap_for_mds(ci, peer) : NULL;
3104         if (ocap && ocap->cap_id == p_cap_id) {
3105                 dout(" remove export cap %p mds%d flags %d\n",
3106                      ocap, peer, ph->flags);
3107                 if ((ph->flags & CEPH_CAP_FLAG_AUTH) &&
3108                     (ocap->seq != le32_to_cpu(ph->seq) ||
3109                      ocap->mseq != le32_to_cpu(ph->mseq))) {
3110                         pr_err("handle_cap_import: mismatched seq/mseq: "
3111                                "ino (%llx.%llx) mds%d seq %d mseq %d "
3112                                "importer mds%d has peer seq %d mseq %d\n",
3113                                ceph_vinop(inode), peer, ocap->seq,
3114                                ocap->mseq, mds, le32_to_cpu(ph->seq),
3115                                le32_to_cpu(ph->mseq));
3116                 }
3117                 __ceph_remove_cap(ocap, (ph->flags & CEPH_CAP_FLAG_RELEASE));
3118         }
3119
3120         /* make sure we re-request max_size, if necessary */
3121         ci->i_wanted_max_size = 0;
3122         ci->i_requested_max_size = 0;
3123
3124         *old_issued = issued;
3125         *target_cap = cap;
3126 }
3127
3128 /*
3129  * Handle a caps message from the MDS.
3130  *
3131  * Identify the appropriate session, inode, and call the right handler
3132  * based on the cap op.
3133  */
3134 void ceph_handle_caps(struct ceph_mds_session *session,
3135                       struct ceph_msg *msg)
3136 {
3137         struct ceph_mds_client *mdsc = session->s_mdsc;
3138         struct super_block *sb = mdsc->fsc->sb;
3139         struct inode *inode;
3140         struct ceph_inode_info *ci;
3141         struct ceph_cap *cap;
3142         struct ceph_mds_caps *h;
3143         struct ceph_mds_cap_peer *peer = NULL;
3144         struct ceph_snap_realm *realm;
3145         int mds = session->s_mds;
3146         int op, issued;
3147         u32 seq, mseq;
3148         struct ceph_vino vino;
3149         u64 cap_id;
3150         u64 size, max_size;
3151         u64 tid;
3152         u64 inline_version = 0;
3153         void *inline_data = NULL;
3154         u32  inline_len = 0;
3155         void *snaptrace;
3156         size_t snaptrace_len;
3157         void *p, *end;
3158
3159         dout("handle_caps from mds%d\n", mds);
3160
3161         /* decode */
3162         end = msg->front.iov_base + msg->front.iov_len;
3163         tid = le64_to_cpu(msg->hdr.tid);
3164         if (msg->front.iov_len < sizeof(*h))
3165                 goto bad;
3166         h = msg->front.iov_base;
3167         op = le32_to_cpu(h->op);
3168         vino.ino = le64_to_cpu(h->ino);
3169         vino.snap = CEPH_NOSNAP;
3170         cap_id = le64_to_cpu(h->cap_id);
3171         seq = le32_to_cpu(h->seq);
3172         mseq = le32_to_cpu(h->migrate_seq);
3173         size = le64_to_cpu(h->size);
3174         max_size = le64_to_cpu(h->max_size);
3175
3176         snaptrace = h + 1;
3177         snaptrace_len = le32_to_cpu(h->snap_trace_len);
3178         p = snaptrace + snaptrace_len;
3179
3180         if (le16_to_cpu(msg->hdr.version) >= 2) {
3181                 u32 flock_len;
3182                 ceph_decode_32_safe(&p, end, flock_len, bad);
3183                 if (p + flock_len > end)
3184                         goto bad;
3185                 p += flock_len;
3186         }
3187
3188         if (le16_to_cpu(msg->hdr.version) >= 3) {
3189                 if (op == CEPH_CAP_OP_IMPORT) {
3190                         if (p + sizeof(*peer) > end)
3191                                 goto bad;
3192                         peer = p;
3193                         p += sizeof(*peer);
3194                 } else if (op == CEPH_CAP_OP_EXPORT) {
3195                         /* recorded in unused fields */
3196                         peer = (void *)&h->size;
3197                 }
3198         }
3199
3200         if (le16_to_cpu(msg->hdr.version) >= 4) {
3201                 ceph_decode_64_safe(&p, end, inline_version, bad);
3202                 ceph_decode_32_safe(&p, end, inline_len, bad);
3203                 if (p + inline_len > end)
3204                         goto bad;
3205                 inline_data = p;
3206                 p += inline_len;
3207         }
3208
3209         /* lookup ino */
3210         inode = ceph_find_inode(sb, vino);
3211         ci = ceph_inode(inode);
3212         dout(" op %s ino %llx.%llx inode %p\n", ceph_cap_op_name(op), vino.ino,
3213              vino.snap, inode);
3214
3215         mutex_lock(&session->s_mutex);
3216         session->s_seq++;
3217         dout(" mds%d seq %lld cap seq %u\n", session->s_mds, session->s_seq,
3218              (unsigned)seq);
3219
3220         if (!inode) {
3221                 dout(" i don't have ino %llx\n", vino.ino);
3222
3223                 if (op == CEPH_CAP_OP_IMPORT) {
3224                         cap = ceph_get_cap(mdsc, NULL);
3225                         cap->cap_ino = vino.ino;
3226                         cap->queue_release = 1;
3227                         cap->cap_id = cap_id;
3228                         cap->mseq = mseq;
3229                         cap->seq = seq;
3230                         spin_lock(&session->s_cap_lock);
3231                         list_add_tail(&cap->session_caps,
3232                                         &session->s_cap_releases);
3233                         session->s_num_cap_releases++;
3234                         spin_unlock(&session->s_cap_lock);
3235                 }
3236                 goto flush_cap_releases;
3237         }
3238
3239         /* these will work even if we don't have a cap yet */
3240         switch (op) {
3241         case CEPH_CAP_OP_FLUSHSNAP_ACK:
3242                 handle_cap_flushsnap_ack(inode, tid, h, session);
3243                 goto done;
3244
3245         case CEPH_CAP_OP_EXPORT:
3246                 handle_cap_export(inode, h, peer, session);
3247                 goto done_unlocked;
3248
3249         case CEPH_CAP_OP_IMPORT:
3250                 realm = NULL;
3251                 if (snaptrace_len) {
3252                         down_write(&mdsc->snap_rwsem);
3253                         ceph_update_snap_trace(mdsc, snaptrace,
3254                                                snaptrace + snaptrace_len,
3255                                                false, &realm);
3256                         downgrade_write(&mdsc->snap_rwsem);
3257                 } else {
3258                         down_read(&mdsc->snap_rwsem);
3259                 }
3260                 handle_cap_import(mdsc, inode, h, peer, session,
3261                                   &cap, &issued);
3262                 handle_cap_grant(mdsc, inode, h,
3263                                  inline_version, inline_data, inline_len,
3264                                  msg->middle, session, cap, issued);
3265                 if (realm)
3266                         ceph_put_snap_realm(mdsc, realm);
3267                 goto done_unlocked;
3268         }
3269
3270         /* the rest require a cap */
3271         spin_lock(&ci->i_ceph_lock);
3272         cap = __get_cap_for_mds(ceph_inode(inode), mds);
3273         if (!cap) {
3274                 dout(" no cap on %p ino %llx.%llx from mds%d\n",
3275                      inode, ceph_ino(inode), ceph_snap(inode), mds);
3276                 spin_unlock(&ci->i_ceph_lock);
3277                 goto flush_cap_releases;
3278         }
3279
3280         /* note that each of these drops i_ceph_lock for us */
3281         switch (op) {
3282         case CEPH_CAP_OP_REVOKE:
3283         case CEPH_CAP_OP_GRANT:
3284                 __ceph_caps_issued(ci, &issued);
3285                 issued |= __ceph_caps_dirty(ci);
3286                 handle_cap_grant(mdsc, inode, h,
3287                                  inline_version, inline_data, inline_len,
3288                                  msg->middle, session, cap, issued);
3289                 goto done_unlocked;
3290
3291         case CEPH_CAP_OP_FLUSH_ACK:
3292                 handle_cap_flush_ack(inode, tid, h, session, cap);
3293                 break;
3294
3295         case CEPH_CAP_OP_TRUNC:
3296                 handle_cap_trunc(inode, h, session);
3297                 break;
3298
3299         default:
3300                 spin_unlock(&ci->i_ceph_lock);
3301                 pr_err("ceph_handle_caps: unknown cap op %d %s\n", op,
3302                        ceph_cap_op_name(op));
3303         }
3304
3305         goto done;
3306
3307 flush_cap_releases:
3308         /*
3309          * send any cap release message to try to move things
3310          * along for the mds (who clearly thinks we still have this
3311          * cap).
3312          */
3313         ceph_send_cap_releases(mdsc, session);
3314
3315 done:
3316         mutex_unlock(&session->s_mutex);
3317 done_unlocked:
3318         iput(inode);
3319         return;
3320
3321 bad:
3322         pr_err("ceph_handle_caps: corrupt message\n");
3323         ceph_msg_dump(msg);
3324         return;
3325 }
3326
3327 /*
3328  * Delayed work handler to process end of delayed cap release LRU list.
3329  */
3330 void ceph_check_delayed_caps(struct ceph_mds_client *mdsc)
3331 {
3332         struct ceph_inode_info *ci;
3333         int flags = CHECK_CAPS_NODELAY;
3334
3335         dout("check_delayed_caps\n");
3336         while (1) {
3337                 spin_lock(&mdsc->cap_delay_lock);
3338                 if (list_empty(&mdsc->cap_delay_list))
3339                         break;
3340                 ci = list_first_entry(&mdsc->cap_delay_list,
3341                                       struct ceph_inode_info,
3342                                       i_cap_delay_list);
3343                 if ((ci->i_ceph_flags & CEPH_I_FLUSH) == 0 &&
3344                     time_before(jiffies, ci->i_hold_caps_max))
3345                         break;
3346                 list_del_init(&ci->i_cap_delay_list);
3347                 spin_unlock(&mdsc->cap_delay_lock);
3348                 dout("check_delayed_caps on %p\n", &ci->vfs_inode);
3349                 ceph_check_caps(ci, flags, NULL);
3350         }
3351         spin_unlock(&mdsc->cap_delay_lock);
3352 }
3353
3354 /*
3355  * Flush all dirty caps to the mds
3356  */
3357 void ceph_flush_dirty_caps(struct ceph_mds_client *mdsc)
3358 {
3359         struct ceph_inode_info *ci;
3360         struct inode *inode;
3361
3362         dout("flush_dirty_caps\n");
3363         spin_lock(&mdsc->cap_dirty_lock);
3364         while (!list_empty(&mdsc->cap_dirty)) {
3365                 ci = list_first_entry(&mdsc->cap_dirty, struct ceph_inode_info,
3366                                       i_dirty_item);
3367                 inode = &ci->vfs_inode;
3368                 ihold(inode);
3369                 dout("flush_dirty_caps %p\n", inode);
3370                 spin_unlock(&mdsc->cap_dirty_lock);
3371                 ceph_check_caps(ci, CHECK_CAPS_NODELAY|CHECK_CAPS_FLUSH, NULL);
3372                 iput(inode);
3373                 spin_lock(&mdsc->cap_dirty_lock);
3374         }
3375         spin_unlock(&mdsc->cap_dirty_lock);
3376         dout("flush_dirty_caps done\n");
3377 }
3378
3379 /*
3380  * Drop open file reference.  If we were the last open file,
3381  * we may need to release capabilities to the MDS (or schedule
3382  * their delayed release).
3383  */
3384 void ceph_put_fmode(struct ceph_inode_info *ci, int fmode)
3385 {
3386         struct inode *inode = &ci->vfs_inode;
3387         int last = 0;
3388
3389         spin_lock(&ci->i_ceph_lock);
3390         dout("put_fmode %p fmode %d %d -> %d\n", inode, fmode,
3391              ci->i_nr_by_mode[fmode], ci->i_nr_by_mode[fmode]-1);
3392         BUG_ON(ci->i_nr_by_mode[fmode] == 0);
3393         if (--ci->i_nr_by_mode[fmode] == 0)
3394                 last++;
3395         spin_unlock(&ci->i_ceph_lock);
3396
3397         if (last && ci->i_vino.snap == CEPH_NOSNAP)
3398                 ceph_check_caps(ci, 0, NULL);
3399 }
3400
3401 /*
3402  * Helpers for embedding cap and dentry lease releases into mds
3403  * requests.
3404  *
3405  * @force is used by dentry_release (below) to force inclusion of a
3406  * record for the directory inode, even when there aren't any caps to
3407  * drop.
3408  */
3409 int ceph_encode_inode_release(void **p, struct inode *inode,
3410                               int mds, int drop, int unless, int force)
3411 {
3412         struct ceph_inode_info *ci = ceph_inode(inode);
3413         struct ceph_cap *cap;
3414         struct ceph_mds_request_release *rel = *p;
3415         int used, dirty;
3416         int ret = 0;
3417
3418         spin_lock(&ci->i_ceph_lock);
3419         used = __ceph_caps_used(ci);
3420         dirty = __ceph_caps_dirty(ci);
3421
3422         dout("encode_inode_release %p mds%d used|dirty %s drop %s unless %s\n",
3423              inode, mds, ceph_cap_string(used|dirty), ceph_cap_string(drop),
3424              ceph_cap_string(unless));
3425
3426         /* only drop unused, clean caps */
3427         drop &= ~(used | dirty);
3428
3429         cap = __get_cap_for_mds(ci, mds);
3430         if (cap && __cap_is_valid(cap)) {
3431                 if (force ||
3432                     ((cap->issued & drop) &&
3433                      (cap->issued & unless) == 0)) {
3434                         if ((cap->issued & drop) &&
3435                             (cap->issued & unless) == 0) {
3436                                 int wanted = __ceph_caps_wanted(ci);
3437                                 if ((ci->i_ceph_flags & CEPH_I_NODELAY) == 0)
3438                                         wanted |= cap->mds_wanted;
3439                                 dout("encode_inode_release %p cap %p "
3440                                      "%s -> %s, wanted %s -> %s\n", inode, cap,
3441                                      ceph_cap_string(cap->issued),
3442                                      ceph_cap_string(cap->issued & ~drop),
3443                                      ceph_cap_string(cap->mds_wanted),
3444                                      ceph_cap_string(wanted));
3445
3446                                 cap->issued &= ~drop;
3447                                 cap->implemented &= ~drop;
3448                                 cap->mds_wanted = wanted;
3449                         } else {
3450                                 dout("encode_inode_release %p cap %p %s"
3451                                      " (force)\n", inode, cap,
3452                                      ceph_cap_string(cap->issued));
3453                         }
3454
3455                         rel->ino = cpu_to_le64(ceph_ino(inode));
3456                         rel->cap_id = cpu_to_le64(cap->cap_id);
3457                         rel->seq = cpu_to_le32(cap->seq);
3458                         rel->issue_seq = cpu_to_le32(cap->issue_seq);
3459                         rel->mseq = cpu_to_le32(cap->mseq);
3460                         rel->caps = cpu_to_le32(cap->implemented);
3461                         rel->wanted = cpu_to_le32(cap->mds_wanted);
3462                         rel->dname_len = 0;
3463                         rel->dname_seq = 0;
3464                         *p += sizeof(*rel);
3465                         ret = 1;
3466                 } else {
3467                         dout("encode_inode_release %p cap %p %s\n",
3468                              inode, cap, ceph_cap_string(cap->issued));
3469                 }
3470         }
3471         spin_unlock(&ci->i_ceph_lock);
3472         return ret;
3473 }
3474
3475 int ceph_encode_dentry_release(void **p, struct dentry *dentry,
3476                                int mds, int drop, int unless)
3477 {
3478         struct inode *dir = d_inode(dentry->d_parent);
3479         struct ceph_mds_request_release *rel = *p;
3480         struct ceph_dentry_info *di = ceph_dentry(dentry);
3481         int force = 0;
3482         int ret;
3483
3484         /*
3485          * force an record for the directory caps if we have a dentry lease.
3486          * this is racy (can't take i_ceph_lock and d_lock together), but it
3487          * doesn't have to be perfect; the mds will revoke anything we don't
3488          * release.
3489          */
3490         spin_lock(&dentry->d_lock);
3491         if (di->lease_session && di->lease_session->s_mds == mds)
3492                 force = 1;
3493         spin_unlock(&dentry->d_lock);
3494
3495         ret = ceph_encode_inode_release(p, dir, mds, drop, unless, force);
3496
3497         spin_lock(&dentry->d_lock);
3498         if (ret && di->lease_session && di->lease_session->s_mds == mds) {
3499                 dout("encode_dentry_release %p mds%d seq %d\n",
3500                      dentry, mds, (int)di->lease_seq);
3501                 rel->dname_len = cpu_to_le32(dentry->d_name.len);
3502                 memcpy(*p, dentry->d_name.name, dentry->d_name.len);
3503                 *p += dentry->d_name.len;
3504                 rel->dname_seq = cpu_to_le32(di->lease_seq);
3505                 __ceph_mdsc_drop_dentry_lease(dentry);
3506         }
3507         spin_unlock(&dentry->d_lock);
3508         return ret;
3509 }