sys/cddl/contrib/opensolaris/uts/common/fs/zfs/ddt_zap.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 /*
  23  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
  24  */
  25
  26 #include <sys/zfs_context.h>
  27 #include <sys/spa.h>
  28 #include <sys/zio.h>
  29 #include <sys/ddt.h>
  30 #include <sys/zap.h>
  31 #include <sys/dmu_tx.h>
  32
  33 int ddt_zap_leaf_blockshift = 12;
  34 int ddt_zap_indirect_blockshift = 12;
  35
  36 static int
  37 ddt_zap_create(objset_t *os, uint64_t *objectp, dmu_tx_t *tx, boolean_t prehash)
  38 {
  39         zap_flags_t flags = ZAP_FLAG_HASH64 | ZAP_FLAG_UINT64_KEY;
  40
  41         if (prehash)
  42                 flags |= ZAP_FLAG_PRE_HASHED_KEY;
  43
  44         *objectp = zap_create_flags(os, 0, flags, DMU_OT_DDT_ZAP,
  45             ddt_zap_leaf_blockshift, ddt_zap_indirect_blockshift,
  46             DMU_OT_NONE, 0, tx);
  47
  48         return (*objectp == 0 ? ENOTSUP : 0);
  49 }
  50
  51 static int
  52 ddt_zap_destroy(objset_t *os, uint64_t object, dmu_tx_t *tx)
  53 {
  54         return (zap_destroy(os, object, tx));
  55 }
  56
  57 static int
  58 ddt_zap_lookup(objset_t *os, uint64_t object, ddt_entry_t *dde)
  59 {
  60         uchar_t cbuf[sizeof (dde->dde_phys) + 1];
  61         uint64_t one, csize;
  62         int error;
  63
  64         error = zap_length_uint64(os, object, (uint64_t *)&dde->dde_key,
  65             DDT_KEY_WORDS, &one, &csize);
  66         if (error)
  67                 return (error);
  68
  69         ASSERT(one == 1);
  70         ASSERT(csize <= sizeof (cbuf));
  71
  72         error = zap_lookup_uint64(os, object, (uint64_t *)&dde->dde_key,
  73             DDT_KEY_WORDS, 1, csize, cbuf);
  74         if (error)
  75                 return (error);
  76
  77         ddt_decompress(cbuf, dde->dde_phys, csize, sizeof (dde->dde_phys));
  78
  79         return (0);
  80 }
  81
  82 static void
  83 ddt_zap_prefetch(objset_t *os, uint64_t object, ddt_entry_t *dde)
  84 {
  85         (void) zap_prefetch_uint64(os, object, (uint64_t *)&dde->dde_key,
  86             DDT_KEY_WORDS);
  87 }
  88
  89 static int
  90 ddt_zap_update(objset_t *os, uint64_t object, ddt_entry_t *dde, dmu_tx_t *tx)
  91 {
  92         uchar_t cbuf[sizeof (dde->dde_phys) + 1];
  93         uint64_t csize;
  94
  95         csize = ddt_compress(dde->dde_phys, cbuf,
  96             sizeof (dde->dde_phys), sizeof (cbuf));
  97
  98         return (zap_update_uint64(os, object, (uint64_t *)&dde->dde_key,
  99             DDT_KEY_WORDS, 1, csize, cbuf, tx));
 100 }
 101
 102 static int
 103 ddt_zap_remove(objset_t *os, uint64_t object, ddt_entry_t *dde, dmu_tx_t *tx)
 104 {
 105         return (zap_remove_uint64(os, object, (uint64_t *)&dde->dde_key,
 106             DDT_KEY_WORDS, tx));
 107 }
 108
 109 static int
 110 ddt_zap_walk(objset_t *os, uint64_t object, ddt_entry_t *dde, uint64_t *walk)
 111 {
 112         zap_cursor_t zc;
 113         zap_attribute_t za;
 114         int error;
 115
 116         zap_cursor_init_serialized(&zc, os, object, *walk);
 117         if ((error = zap_cursor_retrieve(&zc, &za)) == 0) {
 118                 uchar_t cbuf[sizeof (dde->dde_phys) + 1];
 119                 uint64_t csize = za.za_num_integers;
 120                 ASSERT(za.za_integer_length == 1);
 121                 error = zap_lookup_uint64(os, object, (uint64_t *)za.za_name,
 122                     DDT_KEY_WORDS, 1, csize, cbuf);
 123                 ASSERT(error == 0);
 124                 if (error == 0) {
 125                         ddt_decompress(cbuf, dde->dde_phys, csize,
 126                             sizeof (dde->dde_phys));
 127                         dde->dde_key = *(ddt_key_t *)za.za_name;
 128                 }
 129                 zap_cursor_advance(&zc);
 130                 *walk = zap_cursor_serialize(&zc);
 131         }
 132         zap_cursor_fini(&zc);
 133         return (error);
 134 }
 135
 136 static int
 137 ddt_zap_count(objset_t *os, uint64_t object, uint64_t *count)
 138 {
 139
 140         return (zap_count(os, object, count));
 141 }
 142
 143 const ddt_ops_t ddt_zap_ops = {
 144         "zap",
 145         ddt_zap_create,
 146         ddt_zap_destroy,
 147         ddt_zap_lookup,
 148         ddt_zap_prefetch,
 149         ddt_zap_update,
 150         ddt_zap_remove,
 151         ddt_zap_walk,
 152         ddt_zap_count,
 153 };