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