sys/cddl/contrib/opensolaris/uts/common/fs/zfs/space_reftree.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21 /*
  22  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
  23  * Use is subject to license terms.
  24  */
  25 /*
  26  * Copyright (c) 2013 by Delphix. All rights reserved.
  27  */
  28
  29 #include <sys/zfs_context.h>
  30 #include <sys/range_tree.h>
  31 #include <sys/space_reftree.h>
  32
  33 /*
  34  * Space reference trees.
  35  *
  36  * A range tree is a collection of integers.  Every integer is either
  37  * in the tree, or it's not.  A space reference tree generalizes
  38  * the idea: it allows its members to have arbitrary reference counts,
  39  * as opposed to the implicit reference count of 0 or 1 in a range tree.
  40  * This representation comes in handy when computing the union or
  41  * intersection of multiple space maps.  For example, the union of
  42  * N range trees is the subset of the reference tree with refcnt >= 1.
  43  * The intersection of N range trees is the subset with refcnt >= N.
  44  *
  45  * [It's very much like a Fourier transform.  Unions and intersections
  46  * are hard to perform in the 'range tree domain', so we convert the trees
  47  * into the 'reference count domain', where it's trivial, then invert.]
  48  *
  49  * vdev_dtl_reassess() uses computations of this form to determine
  50  * DTL_MISSING and DTL_OUTAGE for interior vdevs -- e.g. a RAID-Z vdev
  51  * has an outage wherever refcnt >= vdev_nparity + 1, and a mirror vdev
  52  * has an outage wherever refcnt >= vdev_children.
  53  */
  54 static int
  55 space_reftree_compare(const void *x1, const void *x2)
  56 {
  57         const space_ref_t *sr1 = x1;
  58         const space_ref_t *sr2 = x2;
  59
  60         if (sr1->sr_offset < sr2->sr_offset)
  61                 return (-1);
  62         if (sr1->sr_offset > sr2->sr_offset)
  63                 return (1);
  64
  65         if (sr1 < sr2)
  66                 return (-1);
  67         if (sr1 > sr2)
  68                 return (1);
  69
  70         return (0);
  71 }
  72
  73 void
  74 space_reftree_create(avl_tree_t *t)
  75 {
  76         avl_create(t, space_reftree_compare,
  77             sizeof (space_ref_t), offsetof(space_ref_t, sr_node));
  78 }
  79
  80 void
  81 space_reftree_destroy(avl_tree_t *t)
  82 {
  83         space_ref_t *sr;
  84         void *cookie = NULL;
  85
  86         while ((sr = avl_destroy_nodes(t, &cookie)) != NULL)
  87                 kmem_free(sr, sizeof (*sr));
  88
  89         avl_destroy(t);
  90 }
  91
  92 static void
  93 space_reftree_add_node(avl_tree_t *t, uint64_t offset, int64_t refcnt)
  94 {
  95         space_ref_t *sr;
  96
  97         sr = kmem_alloc(sizeof (*sr), KM_SLEEP);
  98         sr->sr_offset = offset;
  99         sr->sr_refcnt = refcnt;
 100
 101         avl_add(t, sr);
 102 }
 103
 104 void
 105 space_reftree_add_seg(avl_tree_t *t, uint64_t start, uint64_t end,
 106         int64_t refcnt)
 107 {
 108         space_reftree_add_node(t, start, refcnt);
 109         space_reftree_add_node(t, end, -refcnt);
 110 }
 111
 112 /*
 113  * Convert (or add) a range tree into a reference tree.
 114  */
 115 void
 116 space_reftree_add_map(avl_tree_t *t, range_tree_t *rt, int64_t refcnt)
 117 {
 118         range_seg_t *rs;
 119
 120         ASSERT(MUTEX_HELD(rt->rt_lock));
 121
 122         for (rs = avl_first(&rt->rt_root); rs; rs = AVL_NEXT(&rt->rt_root, rs))
 123                 space_reftree_add_seg(t, rs->rs_start, rs->rs_end, refcnt);
 124 }
 125
 126 /*
 127  * Convert a reference tree into a range tree.  The range tree will contain
 128  * all members of the reference tree for which refcnt >= minref.
 129  */
 130 void
 131 space_reftree_generate_map(avl_tree_t *t, range_tree_t *rt, int64_t minref)
 132 {
 133         uint64_t start = -1ULL;
 134         int64_t refcnt = 0;
 135         space_ref_t *sr;
 136
 137         ASSERT(MUTEX_HELD(rt->rt_lock));
 138
 139         range_tree_vacate(rt, NULL, NULL);
 140
 141         for (sr = avl_first(t); sr != NULL; sr = AVL_NEXT(t, sr)) {
 142                 refcnt += sr->sr_refcnt;
 143                 if (refcnt >= minref) {
 144                         if (start == -1ULL) {
 145                                 start = sr->sr_offset;
 146                         }
 147                 } else {
 148                         if (start != -1ULL) {
 149                                 uint64_t end = sr->sr_offset;
 150                                 ASSERT(start <= end);
 151                                 if (end > start)
 152                                         range_tree_add(rt, start, end - start);
 153                                 start = -1ULL;
 154                         }
 155                 }
 156         }
 157         ASSERT(refcnt == 0);
 158         ASSERT(start == -1ULL);
 159 }