cddl/usr.sbin/zfsd/vdev_iterator.cc

   1 /*-
   2  * Copyright (c) 2011, 2012, 2013 Spectra Logic Corporation
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions, and the following disclaimer,
  10  *    without modification.
  11  * 2. Redistributions in binary form must reproduce at minimum a disclaimer
  12  *    substantially similar to the "NO WARRANTY" disclaimer below
  13  *    ("Disclaimer") and any redistribution must be conditioned upon
  14  *    including a substantially similar Disclaimer requirement for further
  15  *    binary redistribution.
  16  *
  17  * NO WARRANTY
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  26  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
  27  * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28  * POSSIBILITY OF SUCH DAMAGES.
  29  *
  30  * Authors: Justin T. Gibbs     (Spectra Logic Corporation)
  31  *
  32  * $FreeBSD$
  33  */
  34
  35 /**
  36  * \file vdev_iterator.cc
  37  *
  38  * Implementation of the VdevIterator class.
  39  */
  40 #include <sys/cdefs.h>
  41 #include <sys/byteorder.h>
  42 #include <sys/fs/zfs.h>
  43
  44 #include <stdint.h>
  45 #include <syslog.h>
  46
  47 #include <libzfs.h>
  48
  49 #include <list>
  50 #include <string>
  51
  52 #include <devdctl/exception.h>
  53 #include <devdctl/guid.h>
  54
  55 #include "vdev.h"
  56 #include "vdev_iterator.h"
  57 #include "zfsd_exception.h"
  58
  59 /*============================ Namespace Control =============================*/
  60 using DevdCtl::Guid;
  61
  62 /*=========================== Class Implementations ==========================*/
  63 /*------------------------------- VdevIterator -------------------------------*/
  64 VdevIterator::VdevIterator(zpool_handle_t *pool)
  65  : m_poolConfig(zpool_get_config(pool, NULL))
  66 {
  67         Reset();
  68 }
  69
  70 VdevIterator::VdevIterator(nvlist_t *poolConfig)
  71  : m_poolConfig(poolConfig)
  72 {
  73         Reset();
  74 }
  75
  76 void
  77 VdevIterator::Reset()
  78 {
  79         nvlist_t  *rootVdev;
  80         nvlist    **cache_child;
  81         int        result;
  82         uint_t   cache_children;
  83
  84         result = nvlist_lookup_nvlist(m_poolConfig,
  85                                       ZPOOL_CONFIG_VDEV_TREE,
  86                                       &rootVdev);
  87         if (result != 0)
  88                 throw ZfsdException(m_poolConfig, "Unable to extract "
  89                                     "ZPOOL_CONFIG_VDEV_TREE from pool.");
  90         m_vdevQueue.assign(1, rootVdev);
  91         result = nvlist_lookup_nvlist_array(rootVdev,
  92                                             ZPOOL_CONFIG_L2CACHE,
  93                                             &cache_child,
  94                                             &cache_children);
  95         if (result == 0)
  96                 for (uint_t c = 0; c < cache_children; c++)
  97                         m_vdevQueue.push_back(cache_child[c]);
  98 }
  99
 100 nvlist_t *
 101 VdevIterator::Next()
 102 {
 103         nvlist_t *vdevConfig;
 104
 105         if (m_vdevQueue.empty())
 106                 return (NULL);
 107
 108         for (;;) {
 109                 nvlist_t **vdevChildren;
 110                 int        result;
 111                 u_int      numChildren;
 112
 113                 vdevConfig = m_vdevQueue.front();
 114                 m_vdevQueue.pop_front();
 115
 116                 /* Expand non-leaf vdevs. */
 117                 result = nvlist_lookup_nvlist_array(vdevConfig,
 118                                                     ZPOOL_CONFIG_CHILDREN,
 119                                                    &vdevChildren, &numChildren);
 120                 if (result != 0) {
 121                         /* leaf vdev */
 122                         break;
 123                 }
 124
 125                 /*
 126                  * Insert children at the head of the queue to effect a
 127                  * depth first traversal of the tree.
 128                  */
 129                 m_vdevQueue.insert(m_vdevQueue.begin(), vdevChildren,
 130                                    vdevChildren + numChildren);
 131         }
 132
 133         return (vdevConfig);
 134 }
 135
 136 void
 137 VdevIterator::Each(VdevCallback_t *callBack, void *callBackArg)
 138 {
 139         nvlist_t *vdevConfig;
 140
 141         Reset();
 142         while ((vdevConfig = Next()) != NULL) {
 143                 Vdev vdev(m_poolConfig, vdevConfig);
 144
 145                 if (callBack(vdev, callBackArg))
 146                         break;
 147         }
 148 }
 149
 150 nvlist_t *
 151 VdevIterator::Find(Guid vdevGUID)
 152 {
 153         nvlist_t *vdevConfig;
 154
 155         Reset();
 156         while ((vdevConfig = Next()) != NULL) {
 157                 Vdev vdev(m_poolConfig, vdevConfig);
 158
 159                 if (vdev.GUID() == vdevGUID)
 160                         return (vdevConfig);
 161         }
 162         return (NULL);
 163 }