GNU Linux-libre 4.4.283-gnu1
[releases.git] / fs / ceph / snap.c
1 #include <linux/ceph/ceph_debug.h>
2
3 #include <linux/sort.h>
4 #include <linux/slab.h>
5
6 #include "super.h"
7 #include "mds_client.h"
8
9 #include <linux/ceph/decode.h>
10
11 /*
12  * Snapshots in ceph are driven in large part by cooperation from the
13  * client.  In contrast to local file systems or file servers that
14  * implement snapshots at a single point in the system, ceph's
15  * distributed access to storage requires clients to help decide
16  * whether a write logically occurs before or after a recently created
17  * snapshot.
18  *
19  * This provides a perfect instantanous client-wide snapshot.  Between
20  * clients, however, snapshots may appear to be applied at slightly
21  * different points in time, depending on delays in delivering the
22  * snapshot notification.
23  *
24  * Snapshots are _not_ file system-wide.  Instead, each snapshot
25  * applies to the subdirectory nested beneath some directory.  This
26  * effectively divides the hierarchy into multiple "realms," where all
27  * of the files contained by each realm share the same set of
28  * snapshots.  An individual realm's snap set contains snapshots
29  * explicitly created on that realm, as well as any snaps in its
30  * parent's snap set _after_ the point at which the parent became it's
31  * parent (due to, say, a rename).  Similarly, snaps from prior parents
32  * during the time intervals during which they were the parent are included.
33  *
34  * The client is spared most of this detail, fortunately... it must only
35  * maintains a hierarchy of realms reflecting the current parent/child
36  * realm relationship, and for each realm has an explicit list of snaps
37  * inherited from prior parents.
38  *
39  * A snap_realm struct is maintained for realms containing every inode
40  * with an open cap in the system.  (The needed snap realm information is
41  * provided by the MDS whenever a cap is issued, i.e., on open.)  A 'seq'
42  * version number is used to ensure that as realm parameters change (new
43  * snapshot, new parent, etc.) the client's realm hierarchy is updated.
44  *
45  * The realm hierarchy drives the generation of a 'snap context' for each
46  * realm, which simply lists the resulting set of snaps for the realm.  This
47  * is attached to any writes sent to OSDs.
48  */
49 /*
50  * Unfortunately error handling is a bit mixed here.  If we get a snap
51  * update, but don't have enough memory to update our realm hierarchy,
52  * it's not clear what we can do about it (besides complaining to the
53  * console).
54  */
55
56
57 /*
58  * increase ref count for the realm
59  *
60  * caller must hold snap_rwsem for write.
61  */
62 void ceph_get_snap_realm(struct ceph_mds_client *mdsc,
63                          struct ceph_snap_realm *realm)
64 {
65         dout("get_realm %p %d -> %d\n", realm,
66              atomic_read(&realm->nref), atomic_read(&realm->nref)+1);
67         /*
68          * since we _only_ increment realm refs or empty the empty
69          * list with snap_rwsem held, adjusting the empty list here is
70          * safe.  we do need to protect against concurrent empty list
71          * additions, however.
72          */
73         if (atomic_inc_return(&realm->nref) == 1) {
74                 spin_lock(&mdsc->snap_empty_lock);
75                 list_del_init(&realm->empty_item);
76                 spin_unlock(&mdsc->snap_empty_lock);
77         }
78 }
79
80 static void __insert_snap_realm(struct rb_root *root,
81                                 struct ceph_snap_realm *new)
82 {
83         struct rb_node **p = &root->rb_node;
84         struct rb_node *parent = NULL;
85         struct ceph_snap_realm *r = NULL;
86
87         while (*p) {
88                 parent = *p;
89                 r = rb_entry(parent, struct ceph_snap_realm, node);
90                 if (new->ino < r->ino)
91                         p = &(*p)->rb_left;
92                 else if (new->ino > r->ino)
93                         p = &(*p)->rb_right;
94                 else
95                         BUG();
96         }
97
98         rb_link_node(&new->node, parent, p);
99         rb_insert_color(&new->node, root);
100 }
101
102 /*
103  * create and get the realm rooted at @ino and bump its ref count.
104  *
105  * caller must hold snap_rwsem for write.
106  */
107 static struct ceph_snap_realm *ceph_create_snap_realm(
108         struct ceph_mds_client *mdsc,
109         u64 ino)
110 {
111         struct ceph_snap_realm *realm;
112
113         realm = kzalloc(sizeof(*realm), GFP_NOFS);
114         if (!realm)
115                 return ERR_PTR(-ENOMEM);
116
117         atomic_set(&realm->nref, 1);    /* for caller */
118         realm->ino = ino;
119         INIT_LIST_HEAD(&realm->children);
120         INIT_LIST_HEAD(&realm->child_item);
121         INIT_LIST_HEAD(&realm->empty_item);
122         INIT_LIST_HEAD(&realm->dirty_item);
123         INIT_LIST_HEAD(&realm->inodes_with_caps);
124         spin_lock_init(&realm->inodes_with_caps_lock);
125         __insert_snap_realm(&mdsc->snap_realms, realm);
126         dout("create_snap_realm %llx %p\n", realm->ino, realm);
127         return realm;
128 }
129
130 /*
131  * lookup the realm rooted at @ino.
132  *
133  * caller must hold snap_rwsem for write.
134  */
135 static struct ceph_snap_realm *__lookup_snap_realm(struct ceph_mds_client *mdsc,
136                                                    u64 ino)
137 {
138         struct rb_node *n = mdsc->snap_realms.rb_node;
139         struct ceph_snap_realm *r;
140
141         while (n) {
142                 r = rb_entry(n, struct ceph_snap_realm, node);
143                 if (ino < r->ino)
144                         n = n->rb_left;
145                 else if (ino > r->ino)
146                         n = n->rb_right;
147                 else {
148                         dout("lookup_snap_realm %llx %p\n", r->ino, r);
149                         return r;
150                 }
151         }
152         return NULL;
153 }
154
155 struct ceph_snap_realm *ceph_lookup_snap_realm(struct ceph_mds_client *mdsc,
156                                                u64 ino)
157 {
158         struct ceph_snap_realm *r;
159         r = __lookup_snap_realm(mdsc, ino);
160         if (r)
161                 ceph_get_snap_realm(mdsc, r);
162         return r;
163 }
164
165 static void __put_snap_realm(struct ceph_mds_client *mdsc,
166                              struct ceph_snap_realm *realm);
167
168 /*
169  * called with snap_rwsem (write)
170  */
171 static void __destroy_snap_realm(struct ceph_mds_client *mdsc,
172                                  struct ceph_snap_realm *realm)
173 {
174         dout("__destroy_snap_realm %p %llx\n", realm, realm->ino);
175
176         rb_erase(&realm->node, &mdsc->snap_realms);
177
178         if (realm->parent) {
179                 list_del_init(&realm->child_item);
180                 __put_snap_realm(mdsc, realm->parent);
181         }
182
183         kfree(realm->prior_parent_snaps);
184         kfree(realm->snaps);
185         ceph_put_snap_context(realm->cached_context);
186         kfree(realm);
187 }
188
189 /*
190  * caller holds snap_rwsem (write)
191  */
192 static void __put_snap_realm(struct ceph_mds_client *mdsc,
193                              struct ceph_snap_realm *realm)
194 {
195         dout("__put_snap_realm %llx %p %d -> %d\n", realm->ino, realm,
196              atomic_read(&realm->nref), atomic_read(&realm->nref)-1);
197         if (atomic_dec_and_test(&realm->nref))
198                 __destroy_snap_realm(mdsc, realm);
199 }
200
201 /*
202  * caller needn't hold any locks
203  */
204 void ceph_put_snap_realm(struct ceph_mds_client *mdsc,
205                          struct ceph_snap_realm *realm)
206 {
207         dout("put_snap_realm %llx %p %d -> %d\n", realm->ino, realm,
208              atomic_read(&realm->nref), atomic_read(&realm->nref)-1);
209         if (!atomic_dec_and_test(&realm->nref))
210                 return;
211
212         if (down_write_trylock(&mdsc->snap_rwsem)) {
213                 __destroy_snap_realm(mdsc, realm);
214                 up_write(&mdsc->snap_rwsem);
215         } else {
216                 spin_lock(&mdsc->snap_empty_lock);
217                 list_add(&realm->empty_item, &mdsc->snap_empty);
218                 spin_unlock(&mdsc->snap_empty_lock);
219         }
220 }
221
222 /*
223  * Clean up any realms whose ref counts have dropped to zero.  Note
224  * that this does not include realms who were created but not yet
225  * used.
226  *
227  * Called under snap_rwsem (write)
228  */
229 static void __cleanup_empty_realms(struct ceph_mds_client *mdsc)
230 {
231         struct ceph_snap_realm *realm;
232
233         spin_lock(&mdsc->snap_empty_lock);
234         while (!list_empty(&mdsc->snap_empty)) {
235                 realm = list_first_entry(&mdsc->snap_empty,
236                                    struct ceph_snap_realm, empty_item);
237                 list_del(&realm->empty_item);
238                 spin_unlock(&mdsc->snap_empty_lock);
239                 __destroy_snap_realm(mdsc, realm);
240                 spin_lock(&mdsc->snap_empty_lock);
241         }
242         spin_unlock(&mdsc->snap_empty_lock);
243 }
244
245 void ceph_cleanup_empty_realms(struct ceph_mds_client *mdsc)
246 {
247         down_write(&mdsc->snap_rwsem);
248         __cleanup_empty_realms(mdsc);
249         up_write(&mdsc->snap_rwsem);
250 }
251
252 /*
253  * adjust the parent realm of a given @realm.  adjust child list, and parent
254  * pointers, and ref counts appropriately.
255  *
256  * return true if parent was changed, 0 if unchanged, <0 on error.
257  *
258  * caller must hold snap_rwsem for write.
259  */
260 static int adjust_snap_realm_parent(struct ceph_mds_client *mdsc,
261                                     struct ceph_snap_realm *realm,
262                                     u64 parentino)
263 {
264         struct ceph_snap_realm *parent;
265
266         if (realm->parent_ino == parentino)
267                 return 0;
268
269         parent = ceph_lookup_snap_realm(mdsc, parentino);
270         if (!parent) {
271                 parent = ceph_create_snap_realm(mdsc, parentino);
272                 if (IS_ERR(parent))
273                         return PTR_ERR(parent);
274         }
275         dout("adjust_snap_realm_parent %llx %p: %llx %p -> %llx %p\n",
276              realm->ino, realm, realm->parent_ino, realm->parent,
277              parentino, parent);
278         if (realm->parent) {
279                 list_del_init(&realm->child_item);
280                 ceph_put_snap_realm(mdsc, realm->parent);
281         }
282         realm->parent_ino = parentino;
283         realm->parent = parent;
284         list_add(&realm->child_item, &parent->children);
285         return 1;
286 }
287
288
289 static int cmpu64_rev(const void *a, const void *b)
290 {
291         if (*(u64 *)a < *(u64 *)b)
292                 return 1;
293         if (*(u64 *)a > *(u64 *)b)
294                 return -1;
295         return 0;
296 }
297
298
299 struct ceph_snap_context *ceph_empty_snapc;
300
301 /*
302  * build the snap context for a given realm.
303  */
304 static int build_snap_context(struct ceph_snap_realm *realm)
305 {
306         struct ceph_snap_realm *parent = realm->parent;
307         struct ceph_snap_context *snapc;
308         int err = 0;
309         u32 num = realm->num_prior_parent_snaps + realm->num_snaps;
310
311         /*
312          * build parent context, if it hasn't been built.
313          * conservatively estimate that all parent snaps might be
314          * included by us.
315          */
316         if (parent) {
317                 if (!parent->cached_context) {
318                         err = build_snap_context(parent);
319                         if (err)
320                                 goto fail;
321                 }
322                 num += parent->cached_context->num_snaps;
323         }
324
325         /* do i actually need to update?  not if my context seq
326            matches realm seq, and my parents' does to.  (this works
327            because we rebuild_snap_realms() works _downward_ in
328            hierarchy after each update.) */
329         if (realm->cached_context &&
330             realm->cached_context->seq == realm->seq &&
331             (!parent ||
332              realm->cached_context->seq >= parent->cached_context->seq)) {
333                 dout("build_snap_context %llx %p: %p seq %lld (%u snaps)"
334                      " (unchanged)\n",
335                      realm->ino, realm, realm->cached_context,
336                      realm->cached_context->seq,
337                      (unsigned int) realm->cached_context->num_snaps);
338                 return 0;
339         }
340
341         /* alloc new snap context */
342         err = -ENOMEM;
343         if (num > (SIZE_MAX - sizeof(*snapc)) / sizeof(u64))
344                 goto fail;
345         snapc = ceph_create_snap_context(num, GFP_NOFS);
346         if (!snapc)
347                 goto fail;
348
349         /* build (reverse sorted) snap vector */
350         num = 0;
351         snapc->seq = realm->seq;
352         if (parent) {
353                 u32 i;
354
355                 /* include any of parent's snaps occurring _after_ my
356                    parent became my parent */
357                 for (i = 0; i < parent->cached_context->num_snaps; i++)
358                         if (parent->cached_context->snaps[i] >=
359                             realm->parent_since)
360                                 snapc->snaps[num++] =
361                                         parent->cached_context->snaps[i];
362                 if (parent->cached_context->seq > snapc->seq)
363                         snapc->seq = parent->cached_context->seq;
364         }
365         memcpy(snapc->snaps + num, realm->snaps,
366                sizeof(u64)*realm->num_snaps);
367         num += realm->num_snaps;
368         memcpy(snapc->snaps + num, realm->prior_parent_snaps,
369                sizeof(u64)*realm->num_prior_parent_snaps);
370         num += realm->num_prior_parent_snaps;
371
372         sort(snapc->snaps, num, sizeof(u64), cmpu64_rev, NULL);
373         snapc->num_snaps = num;
374         dout("build_snap_context %llx %p: %p seq %lld (%u snaps)\n",
375              realm->ino, realm, snapc, snapc->seq,
376              (unsigned int) snapc->num_snaps);
377
378         ceph_put_snap_context(realm->cached_context);
379         realm->cached_context = snapc;
380         return 0;
381
382 fail:
383         /*
384          * if we fail, clear old (incorrect) cached_context... hopefully
385          * we'll have better luck building it later
386          */
387         if (realm->cached_context) {
388                 ceph_put_snap_context(realm->cached_context);
389                 realm->cached_context = NULL;
390         }
391         pr_err("build_snap_context %llx %p fail %d\n", realm->ino,
392                realm, err);
393         return err;
394 }
395
396 /*
397  * rebuild snap context for the given realm and all of its children.
398  */
399 static void rebuild_snap_realms(struct ceph_snap_realm *realm)
400 {
401         struct ceph_snap_realm *child;
402
403         dout("rebuild_snap_realms %llx %p\n", realm->ino, realm);
404         build_snap_context(realm);
405
406         list_for_each_entry(child, &realm->children, child_item)
407                 rebuild_snap_realms(child);
408 }
409
410
411 /*
412  * helper to allocate and decode an array of snapids.  free prior
413  * instance, if any.
414  */
415 static int dup_array(u64 **dst, __le64 *src, u32 num)
416 {
417         u32 i;
418
419         kfree(*dst);
420         if (num) {
421                 *dst = kcalloc(num, sizeof(u64), GFP_NOFS);
422                 if (!*dst)
423                         return -ENOMEM;
424                 for (i = 0; i < num; i++)
425                         (*dst)[i] = get_unaligned_le64(src + i);
426         } else {
427                 *dst = NULL;
428         }
429         return 0;
430 }
431
432 static bool has_new_snaps(struct ceph_snap_context *o,
433                           struct ceph_snap_context *n)
434 {
435         if (n->num_snaps == 0)
436                 return false;
437         /* snaps are in descending order */
438         return n->snaps[0] > o->seq;
439 }
440
441 /*
442  * When a snapshot is applied, the size/mtime inode metadata is queued
443  * in a ceph_cap_snap (one for each snapshot) until writeback
444  * completes and the metadata can be flushed back to the MDS.
445  *
446  * However, if a (sync) write is currently in-progress when we apply
447  * the snapshot, we have to wait until the write succeeds or fails
448  * (and a final size/mtime is known).  In this case the
449  * cap_snap->writing = 1, and is said to be "pending."  When the write
450  * finishes, we __ceph_finish_cap_snap().
451  *
452  * Caller must hold snap_rwsem for read (i.e., the realm topology won't
453  * change).
454  */
455 void ceph_queue_cap_snap(struct ceph_inode_info *ci)
456 {
457         struct inode *inode = &ci->vfs_inode;
458         struct ceph_cap_snap *capsnap;
459         struct ceph_snap_context *old_snapc, *new_snapc;
460         int used, dirty;
461
462         capsnap = kzalloc(sizeof(*capsnap), GFP_NOFS);
463         if (!capsnap) {
464                 pr_err("ENOMEM allocating ceph_cap_snap on %p\n", inode);
465                 return;
466         }
467
468         spin_lock(&ci->i_ceph_lock);
469         used = __ceph_caps_used(ci);
470         dirty = __ceph_caps_dirty(ci);
471
472         old_snapc = ci->i_head_snapc;
473         new_snapc = ci->i_snap_realm->cached_context;
474
475         /*
476          * If there is a write in progress, treat that as a dirty Fw,
477          * even though it hasn't completed yet; by the time we finish
478          * up this capsnap it will be.
479          */
480         if (used & CEPH_CAP_FILE_WR)
481                 dirty |= CEPH_CAP_FILE_WR;
482
483         if (__ceph_have_pending_cap_snap(ci)) {
484                 /* there is no point in queuing multiple "pending" cap_snaps,
485                    as no new writes are allowed to start when pending, so any
486                    writes in progress now were started before the previous
487                    cap_snap.  lucky us. */
488                 dout("queue_cap_snap %p already pending\n", inode);
489                 goto update_snapc;
490         }
491         if (ci->i_wrbuffer_ref_head == 0 &&
492             !(dirty & (CEPH_CAP_ANY_EXCL|CEPH_CAP_FILE_WR))) {
493                 dout("queue_cap_snap %p nothing dirty|writing\n", inode);
494                 goto update_snapc;
495         }
496
497         BUG_ON(!old_snapc);
498
499         /*
500          * There is no need to send FLUSHSNAP message to MDS if there is
501          * no new snapshot. But when there is dirty pages or on-going
502          * writes, we still need to create cap_snap. cap_snap is needed
503          * by the write path and page writeback path.
504          *
505          * also see ceph_try_drop_cap_snap()
506          */
507         if (has_new_snaps(old_snapc, new_snapc)) {
508                 if (dirty & (CEPH_CAP_ANY_EXCL|CEPH_CAP_FILE_WR))
509                         capsnap->need_flush = true;
510         } else {
511                 if (!(used & CEPH_CAP_FILE_WR) &&
512                     ci->i_wrbuffer_ref_head == 0) {
513                         dout("queue_cap_snap %p "
514                              "no new_snap|dirty_page|writing\n", inode);
515                         goto update_snapc;
516                 }
517         }
518
519         dout("queue_cap_snap %p cap_snap %p queuing under %p %s %s\n",
520              inode, capsnap, old_snapc, ceph_cap_string(dirty),
521              capsnap->need_flush ? "" : "no_flush");
522         ihold(inode);
523
524         atomic_set(&capsnap->nref, 1);
525         capsnap->ci = ci;
526         INIT_LIST_HEAD(&capsnap->ci_item);
527         INIT_LIST_HEAD(&capsnap->flushing_item);
528
529         capsnap->follows = old_snapc->seq;
530         capsnap->issued = __ceph_caps_issued(ci, NULL);
531         capsnap->dirty = dirty;
532
533         capsnap->mode = inode->i_mode;
534         capsnap->uid = inode->i_uid;
535         capsnap->gid = inode->i_gid;
536
537         if (dirty & CEPH_CAP_XATTR_EXCL) {
538                 __ceph_build_xattrs_blob(ci);
539                 capsnap->xattr_blob =
540                         ceph_buffer_get(ci->i_xattrs.blob);
541                 capsnap->xattr_version = ci->i_xattrs.version;
542         } else {
543                 capsnap->xattr_blob = NULL;
544                 capsnap->xattr_version = 0;
545         }
546
547         capsnap->inline_data = ci->i_inline_version != CEPH_INLINE_NONE;
548
549         /* dirty page count moved from _head to this cap_snap;
550            all subsequent writes page dirties occur _after_ this
551            snapshot. */
552         capsnap->dirty_pages = ci->i_wrbuffer_ref_head;
553         ci->i_wrbuffer_ref_head = 0;
554         capsnap->context = old_snapc;
555         list_add_tail(&capsnap->ci_item, &ci->i_cap_snaps);
556         old_snapc = NULL;
557
558         if (used & CEPH_CAP_FILE_WR) {
559                 dout("queue_cap_snap %p cap_snap %p snapc %p"
560                      " seq %llu used WR, now pending\n", inode,
561                      capsnap, old_snapc, old_snapc->seq);
562                 capsnap->writing = 1;
563         } else {
564                 /* note mtime, size NOW. */
565                 __ceph_finish_cap_snap(ci, capsnap);
566         }
567         capsnap = NULL;
568
569 update_snapc:
570        if (ci->i_wrbuffer_ref_head == 0 &&
571            ci->i_wr_ref == 0 &&
572            ci->i_dirty_caps == 0 &&
573            ci->i_flushing_caps == 0) {
574                ci->i_head_snapc = NULL;
575        } else {
576                 ci->i_head_snapc = ceph_get_snap_context(new_snapc);
577                 dout(" new snapc is %p\n", new_snapc);
578         }
579         spin_unlock(&ci->i_ceph_lock);
580
581         kfree(capsnap);
582         ceph_put_snap_context(old_snapc);
583 }
584
585 /*
586  * Finalize the size, mtime for a cap_snap.. that is, settle on final values
587  * to be used for the snapshot, to be flushed back to the mds.
588  *
589  * If capsnap can now be flushed, add to snap_flush list, and return 1.
590  *
591  * Caller must hold i_ceph_lock.
592  */
593 int __ceph_finish_cap_snap(struct ceph_inode_info *ci,
594                             struct ceph_cap_snap *capsnap)
595 {
596         struct inode *inode = &ci->vfs_inode;
597         struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
598
599         BUG_ON(capsnap->writing);
600         capsnap->size = inode->i_size;
601         capsnap->mtime = inode->i_mtime;
602         capsnap->atime = inode->i_atime;
603         capsnap->ctime = inode->i_ctime;
604         capsnap->time_warp_seq = ci->i_time_warp_seq;
605         if (capsnap->dirty_pages) {
606                 dout("finish_cap_snap %p cap_snap %p snapc %p %llu %s s=%llu "
607                      "still has %d dirty pages\n", inode, capsnap,
608                      capsnap->context, capsnap->context->seq,
609                      ceph_cap_string(capsnap->dirty), capsnap->size,
610                      capsnap->dirty_pages);
611                 return 0;
612         }
613         dout("finish_cap_snap %p cap_snap %p snapc %p %llu %s s=%llu\n",
614              inode, capsnap, capsnap->context,
615              capsnap->context->seq, ceph_cap_string(capsnap->dirty),
616              capsnap->size);
617
618         spin_lock(&mdsc->snap_flush_lock);
619         if (list_empty(&ci->i_snap_flush_item))
620                 list_add_tail(&ci->i_snap_flush_item, &mdsc->snap_flush_list);
621         spin_unlock(&mdsc->snap_flush_lock);
622         return 1;  /* caller may want to ceph_flush_snaps */
623 }
624
625 /*
626  * Queue cap_snaps for snap writeback for this realm and its children.
627  * Called under snap_rwsem, so realm topology won't change.
628  */
629 static void queue_realm_cap_snaps(struct ceph_snap_realm *realm)
630 {
631         struct ceph_inode_info *ci;
632         struct inode *lastinode = NULL;
633         struct ceph_snap_realm *child;
634
635         dout("queue_realm_cap_snaps %p %llx inodes\n", realm, realm->ino);
636
637         spin_lock(&realm->inodes_with_caps_lock);
638         list_for_each_entry(ci, &realm->inodes_with_caps,
639                             i_snap_realm_item) {
640                 struct inode *inode = igrab(&ci->vfs_inode);
641                 if (!inode)
642                         continue;
643                 spin_unlock(&realm->inodes_with_caps_lock);
644                 iput(lastinode);
645                 lastinode = inode;
646                 ceph_queue_cap_snap(ci);
647                 spin_lock(&realm->inodes_with_caps_lock);
648         }
649         spin_unlock(&realm->inodes_with_caps_lock);
650         iput(lastinode);
651
652         list_for_each_entry(child, &realm->children, child_item) {
653                 dout("queue_realm_cap_snaps %p %llx queue child %p %llx\n",
654                      realm, realm->ino, child, child->ino);
655                 list_del_init(&child->dirty_item);
656                 list_add(&child->dirty_item, &realm->dirty_item);
657         }
658
659         list_del_init(&realm->dirty_item);
660         dout("queue_realm_cap_snaps %p %llx done\n", realm, realm->ino);
661 }
662
663 /*
664  * Parse and apply a snapblob "snap trace" from the MDS.  This specifies
665  * the snap realm parameters from a given realm and all of its ancestors,
666  * up to the root.
667  *
668  * Caller must hold snap_rwsem for write.
669  */
670 int ceph_update_snap_trace(struct ceph_mds_client *mdsc,
671                            void *p, void *e, bool deletion,
672                            struct ceph_snap_realm **realm_ret)
673 {
674         struct ceph_mds_snap_realm *ri;    /* encoded */
675         __le64 *snaps;                     /* encoded */
676         __le64 *prior_parent_snaps;        /* encoded */
677         struct ceph_snap_realm *realm = NULL;
678         struct ceph_snap_realm *first_realm = NULL;
679         int invalidate = 0;
680         int err = -ENOMEM;
681         LIST_HEAD(dirty_realms);
682
683         dout("update_snap_trace deletion=%d\n", deletion);
684 more:
685         ceph_decode_need(&p, e, sizeof(*ri), bad);
686         ri = p;
687         p += sizeof(*ri);
688         ceph_decode_need(&p, e, sizeof(u64)*(le32_to_cpu(ri->num_snaps) +
689                             le32_to_cpu(ri->num_prior_parent_snaps)), bad);
690         snaps = p;
691         p += sizeof(u64) * le32_to_cpu(ri->num_snaps);
692         prior_parent_snaps = p;
693         p += sizeof(u64) * le32_to_cpu(ri->num_prior_parent_snaps);
694
695         realm = ceph_lookup_snap_realm(mdsc, le64_to_cpu(ri->ino));
696         if (!realm) {
697                 realm = ceph_create_snap_realm(mdsc, le64_to_cpu(ri->ino));
698                 if (IS_ERR(realm)) {
699                         err = PTR_ERR(realm);
700                         goto fail;
701                 }
702         }
703
704         /* ensure the parent is correct */
705         err = adjust_snap_realm_parent(mdsc, realm, le64_to_cpu(ri->parent));
706         if (err < 0)
707                 goto fail;
708         invalidate += err;
709
710         if (le64_to_cpu(ri->seq) > realm->seq) {
711                 dout("update_snap_trace updating %llx %p %lld -> %lld\n",
712                      realm->ino, realm, realm->seq, le64_to_cpu(ri->seq));
713                 /* update realm parameters, snap lists */
714                 realm->seq = le64_to_cpu(ri->seq);
715                 realm->created = le64_to_cpu(ri->created);
716                 realm->parent_since = le64_to_cpu(ri->parent_since);
717
718                 realm->num_snaps = le32_to_cpu(ri->num_snaps);
719                 err = dup_array(&realm->snaps, snaps, realm->num_snaps);
720                 if (err < 0)
721                         goto fail;
722
723                 realm->num_prior_parent_snaps =
724                         le32_to_cpu(ri->num_prior_parent_snaps);
725                 err = dup_array(&realm->prior_parent_snaps, prior_parent_snaps,
726                                 realm->num_prior_parent_snaps);
727                 if (err < 0)
728                         goto fail;
729
730                 /* queue realm for cap_snap creation */
731                 list_add(&realm->dirty_item, &dirty_realms);
732                 if (realm->seq > mdsc->last_snap_seq)
733                         mdsc->last_snap_seq = realm->seq;
734
735                 invalidate = 1;
736         } else if (!realm->cached_context) {
737                 dout("update_snap_trace %llx %p seq %lld new\n",
738                      realm->ino, realm, realm->seq);
739                 invalidate = 1;
740         } else {
741                 dout("update_snap_trace %llx %p seq %lld unchanged\n",
742                      realm->ino, realm, realm->seq);
743         }
744
745         dout("done with %llx %p, invalidated=%d, %p %p\n", realm->ino,
746              realm, invalidate, p, e);
747
748         /* invalidate when we reach the _end_ (root) of the trace */
749         if (invalidate && p >= e)
750                 rebuild_snap_realms(realm);
751
752         if (!first_realm)
753                 first_realm = realm;
754         else
755                 ceph_put_snap_realm(mdsc, realm);
756
757         if (p < e)
758                 goto more;
759
760         /*
761          * queue cap snaps _after_ we've built the new snap contexts,
762          * so that i_head_snapc can be set appropriately.
763          */
764         while (!list_empty(&dirty_realms)) {
765                 realm = list_first_entry(&dirty_realms, struct ceph_snap_realm,
766                                          dirty_item);
767                 queue_realm_cap_snaps(realm);
768         }
769
770         if (realm_ret)
771                 *realm_ret = first_realm;
772         else
773                 ceph_put_snap_realm(mdsc, first_realm);
774
775         __cleanup_empty_realms(mdsc);
776         return 0;
777
778 bad:
779         err = -EINVAL;
780 fail:
781         if (realm && !IS_ERR(realm))
782                 ceph_put_snap_realm(mdsc, realm);
783         if (first_realm)
784                 ceph_put_snap_realm(mdsc, first_realm);
785         pr_err("update_snap_trace error %d\n", err);
786         return err;
787 }
788
789
790 /*
791  * Send any cap_snaps that are queued for flush.  Try to carry
792  * s_mutex across multiple snap flushes to avoid locking overhead.
793  *
794  * Caller holds no locks.
795  */
796 static void flush_snaps(struct ceph_mds_client *mdsc)
797 {
798         struct ceph_inode_info *ci;
799         struct inode *inode;
800         struct ceph_mds_session *session = NULL;
801
802         dout("flush_snaps\n");
803         spin_lock(&mdsc->snap_flush_lock);
804         while (!list_empty(&mdsc->snap_flush_list)) {
805                 ci = list_first_entry(&mdsc->snap_flush_list,
806                                 struct ceph_inode_info, i_snap_flush_item);
807                 inode = &ci->vfs_inode;
808                 ihold(inode);
809                 spin_unlock(&mdsc->snap_flush_lock);
810                 spin_lock(&ci->i_ceph_lock);
811                 __ceph_flush_snaps(ci, &session, 0);
812                 spin_unlock(&ci->i_ceph_lock);
813                 iput(inode);
814                 spin_lock(&mdsc->snap_flush_lock);
815         }
816         spin_unlock(&mdsc->snap_flush_lock);
817
818         if (session) {
819                 mutex_unlock(&session->s_mutex);
820                 ceph_put_mds_session(session);
821         }
822         dout("flush_snaps done\n");
823 }
824
825
826 /*
827  * Handle a snap notification from the MDS.
828  *
829  * This can take two basic forms: the simplest is just a snap creation
830  * or deletion notification on an existing realm.  This should update the
831  * realm and its children.
832  *
833  * The more difficult case is realm creation, due to snap creation at a
834  * new point in the file hierarchy, or due to a rename that moves a file or
835  * directory into another realm.
836  */
837 void ceph_handle_snap(struct ceph_mds_client *mdsc,
838                       struct ceph_mds_session *session,
839                       struct ceph_msg *msg)
840 {
841         struct super_block *sb = mdsc->fsc->sb;
842         int mds = session->s_mds;
843         u64 split;
844         int op;
845         int trace_len;
846         struct ceph_snap_realm *realm = NULL;
847         void *p = msg->front.iov_base;
848         void *e = p + msg->front.iov_len;
849         struct ceph_mds_snap_head *h;
850         int num_split_inos, num_split_realms;
851         __le64 *split_inos = NULL, *split_realms = NULL;
852         int i;
853         int locked_rwsem = 0;
854
855         /* decode */
856         if (msg->front.iov_len < sizeof(*h))
857                 goto bad;
858         h = p;
859         op = le32_to_cpu(h->op);
860         split = le64_to_cpu(h->split);   /* non-zero if we are splitting an
861                                           * existing realm */
862         num_split_inos = le32_to_cpu(h->num_split_inos);
863         num_split_realms = le32_to_cpu(h->num_split_realms);
864         trace_len = le32_to_cpu(h->trace_len);
865         p += sizeof(*h);
866
867         dout("handle_snap from mds%d op %s split %llx tracelen %d\n", mds,
868              ceph_snap_op_name(op), split, trace_len);
869
870         mutex_lock(&session->s_mutex);
871         session->s_seq++;
872         mutex_unlock(&session->s_mutex);
873
874         down_write(&mdsc->snap_rwsem);
875         locked_rwsem = 1;
876
877         if (op == CEPH_SNAP_OP_SPLIT) {
878                 struct ceph_mds_snap_realm *ri;
879
880                 /*
881                  * A "split" breaks part of an existing realm off into
882                  * a new realm.  The MDS provides a list of inodes
883                  * (with caps) and child realms that belong to the new
884                  * child.
885                  */
886                 split_inos = p;
887                 p += sizeof(u64) * num_split_inos;
888                 split_realms = p;
889                 p += sizeof(u64) * num_split_realms;
890                 ceph_decode_need(&p, e, sizeof(*ri), bad);
891                 /* we will peek at realm info here, but will _not_
892                  * advance p, as the realm update will occur below in
893                  * ceph_update_snap_trace. */
894                 ri = p;
895
896                 realm = ceph_lookup_snap_realm(mdsc, split);
897                 if (!realm) {
898                         realm = ceph_create_snap_realm(mdsc, split);
899                         if (IS_ERR(realm))
900                                 goto out;
901                 }
902
903                 dout("splitting snap_realm %llx %p\n", realm->ino, realm);
904                 for (i = 0; i < num_split_inos; i++) {
905                         struct ceph_vino vino = {
906                                 .ino = le64_to_cpu(split_inos[i]),
907                                 .snap = CEPH_NOSNAP,
908                         };
909                         struct inode *inode = ceph_find_inode(sb, vino);
910                         struct ceph_inode_info *ci;
911                         struct ceph_snap_realm *oldrealm;
912
913                         if (!inode)
914                                 continue;
915                         ci = ceph_inode(inode);
916
917                         spin_lock(&ci->i_ceph_lock);
918                         if (!ci->i_snap_realm)
919                                 goto skip_inode;
920                         /*
921                          * If this inode belongs to a realm that was
922                          * created after our new realm, we experienced
923                          * a race (due to another split notifications
924                          * arriving from a different MDS).  So skip
925                          * this inode.
926                          */
927                         if (ci->i_snap_realm->created >
928                             le64_to_cpu(ri->created)) {
929                                 dout(" leaving %p in newer realm %llx %p\n",
930                                      inode, ci->i_snap_realm->ino,
931                                      ci->i_snap_realm);
932                                 goto skip_inode;
933                         }
934                         dout(" will move %p to split realm %llx %p\n",
935                              inode, realm->ino, realm);
936                         /*
937                          * Move the inode to the new realm
938                          */
939                         spin_lock(&realm->inodes_with_caps_lock);
940                         list_del_init(&ci->i_snap_realm_item);
941                         list_add(&ci->i_snap_realm_item,
942                                  &realm->inodes_with_caps);
943                         oldrealm = ci->i_snap_realm;
944                         ci->i_snap_realm = realm;
945                         spin_unlock(&realm->inodes_with_caps_lock);
946                         spin_unlock(&ci->i_ceph_lock);
947
948                         ceph_get_snap_realm(mdsc, realm);
949                         ceph_put_snap_realm(mdsc, oldrealm);
950
951                         iput(inode);
952                         continue;
953
954 skip_inode:
955                         spin_unlock(&ci->i_ceph_lock);
956                         iput(inode);
957                 }
958
959                 /* we may have taken some of the old realm's children. */
960                 for (i = 0; i < num_split_realms; i++) {
961                         struct ceph_snap_realm *child =
962                                 __lookup_snap_realm(mdsc,
963                                            le64_to_cpu(split_realms[i]));
964                         if (!child)
965                                 continue;
966                         adjust_snap_realm_parent(mdsc, child, realm->ino);
967                 }
968         }
969
970         /*
971          * update using the provided snap trace. if we are deleting a
972          * snap, we can avoid queueing cap_snaps.
973          */
974         ceph_update_snap_trace(mdsc, p, e,
975                                op == CEPH_SNAP_OP_DESTROY, NULL);
976
977         if (op == CEPH_SNAP_OP_SPLIT)
978                 /* we took a reference when we created the realm, above */
979                 ceph_put_snap_realm(mdsc, realm);
980
981         __cleanup_empty_realms(mdsc);
982
983         up_write(&mdsc->snap_rwsem);
984
985         flush_snaps(mdsc);
986         return;
987
988 bad:
989         pr_err("corrupt snap message from mds%d\n", mds);
990         ceph_msg_dump(msg);
991 out:
992         if (locked_rwsem)
993                 up_write(&mdsc->snap_rwsem);
994         return;
995 }
996
997 int __init ceph_snap_init(void)
998 {
999         ceph_empty_snapc = ceph_create_snap_context(0, GFP_NOFS);
1000         if (!ceph_empty_snapc)
1001                 return -ENOMEM;
1002         ceph_empty_snapc->seq = 1;
1003         return 0;
1004 }
1005
1006 void ceph_snap_exit(void)
1007 {
1008         ceph_put_snap_context(ceph_empty_snapc);
1009 }