sys/cddl/contrib/opensolaris/uts/common/fs/zfs/sa.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) 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Portions Copyright 2011 iXsystems, Inc
  25  * Copyright (c) 2012 by Delphix. All rights reserved.
  26  */
  27
  28 #include <sys/zfs_context.h>
  29 #include <sys/types.h>
  30 #include <sys/param.h>
  31 #include <sys/systm.h>
  32 #include <sys/sysmacros.h>
  33 #include <sys/dmu.h>
  34 #include <sys/dmu_impl.h>
  35 #include <sys/dmu_objset.h>
  36 #include <sys/dbuf.h>
  37 #include <sys/dnode.h>
  38 #include <sys/zap.h>
  39 #include <sys/sa.h>
  40 #include <sys/sunddi.h>
  41 #include <sys/sa_impl.h>
  42 #include <sys/dnode.h>
  43 #include <sys/errno.h>
  44 #include <sys/zfs_context.h>
  45
  46 /*
  47  * ZFS System attributes:
  48  *
  49  * A generic mechanism to allow for arbitrary attributes
  50  * to be stored in a dnode.  The data will be stored in the bonus buffer of
  51  * the dnode and if necessary a special "spill" block will be used to handle
  52  * overflow situations.  The spill block will be sized to fit the data
  53  * from 512 - 128K.  When a spill block is used the BP (blkptr_t) for the
  54  * spill block is stored at the end of the current bonus buffer.  Any
  55  * attributes that would be in the way of the blkptr_t will be relocated
  56  * into the spill block.
  57  *
  58  * Attribute registration:
  59  *
  60  * Stored persistently on a per dataset basis
  61  * a mapping between attribute "string" names and their actual attribute
  62  * numeric values, length, and byteswap function.  The names are only used
  63  * during registration.  All  attributes are known by their unique attribute
  64  * id value.  If an attribute can have a variable size then the value
  65  * 0 will be used to indicate this.
  66  *
  67  * Attribute Layout:
  68  *
  69  * Attribute layouts are a way to compactly store multiple attributes, but
  70  * without taking the overhead associated with managing each attribute
  71  * individually.  Since you will typically have the same set of attributes
  72  * stored in the same order a single table will be used to represent that
  73  * layout.  The ZPL for example will usually have only about 10 different
  74  * layouts (regular files, device files, symlinks,
  75  * regular files + scanstamp, files/dir with extended attributes, and then
  76  * you have the possibility of all of those minus ACL, because it would
  77  * be kicked out into the spill block)
  78  *
  79  * Layouts are simply an array of the attributes and their
  80  * ordering i.e. [0, 1, 4, 5, 2]
  81  *
  82  * Each distinct layout is given a unique layout number and that is whats
  83  * stored in the header at the beginning of the SA data buffer.
  84  *
  85  * A layout only covers a single dbuf (bonus or spill).  If a set of
  86  * attributes is split up between the bonus buffer and a spill buffer then
  87  * two different layouts will be used.  This allows us to byteswap the
  88  * spill without looking at the bonus buffer and keeps the on disk format of
  89  * the bonus and spill buffer the same.
  90  *
  91  * Adding a single attribute will cause the entire set of attributes to
  92  * be rewritten and could result in a new layout number being constructed
  93  * as part of the rewrite if no such layout exists for the new set of
  94  * attribues.  The new attribute will be appended to the end of the already
  95  * existing attributes.
  96  *
  97  * Both the attribute registration and attribute layout information are
  98  * stored in normal ZAP attributes.  Their should be a small number of
  99  * known layouts and the set of attributes is assumed to typically be quite
 100  * small.
 101  *
 102  * The registered attributes and layout "table" information is maintained
 103  * in core and a special "sa_os_t" is attached to the objset_t.
 104  *
 105  * A special interface is provided to allow for quickly applying
 106  * a large set of attributes at once.  sa_replace_all_by_template() is
 107  * used to set an array of attributes.  This is used by the ZPL when
 108  * creating a brand new file.  The template that is passed into the function
 109  * specifies the attribute, size for variable length attributes, location of
 110  * data and special "data locator" function if the data isn't in a contiguous
 111  * location.
 112  *
 113  * Byteswap implications:
 114  * Since the SA attributes are not entirely self describing we can't do
 115  * the normal byteswap processing.  The special ZAP layout attribute and
 116  * attribute registration attributes define the byteswap function and the
 117  * size of the attributes, unless it is variable sized.
 118  * The normal ZFS byteswapping infrastructure assumes you don't need
 119  * to read any objects in order to do the necessary byteswapping.  Whereas
 120  * SA attributes can only be properly byteswapped if the dataset is opened
 121  * and the layout/attribute ZAP attributes are available.  Because of this
 122  * the SA attributes will be byteswapped when they are first accessed by
 123  * the SA code that will read the SA data.
 124  */
 125
 126 typedef void (sa_iterfunc_t)(void *hdr, void *addr, sa_attr_type_t,
 127     uint16_t length, int length_idx, boolean_t, void *userp);
 128
 129 static int sa_build_index(sa_handle_t *hdl, sa_buf_type_t buftype);
 130 static void sa_idx_tab_hold(objset_t *os, sa_idx_tab_t *idx_tab);
 131 static void *sa_find_idx_tab(objset_t *os, dmu_object_type_t bonustype,
 132     void *data);
 133 static void sa_idx_tab_rele(objset_t *os, void *arg);
 134 static void sa_copy_data(sa_data_locator_t *func, void *start, void *target,
 135     int buflen);
 136 static int sa_modify_attrs(sa_handle_t *hdl, sa_attr_type_t newattr,
 137     sa_data_op_t action, sa_data_locator_t *locator, void *datastart,
 138     uint16_t buflen, dmu_tx_t *tx);
 139
 140 arc_byteswap_func_t *sa_bswap_table[] = {
 141         byteswap_uint64_array,
 142         byteswap_uint32_array,
 143         byteswap_uint16_array,
 144         byteswap_uint8_array,
 145         zfs_acl_byteswap,
 146 };
 147
 148 #define SA_COPY_DATA(f, s, t, l) \
 149         { \
 150                 if (f == NULL) { \
 151                         if (l == 8) { \
 152                                 *(uint64_t *)t = *(uint64_t *)s; \
 153                         } else if (l == 16) { \
 154                                 *(uint64_t *)t = *(uint64_t *)s; \
 155                                 *(uint64_t *)((uintptr_t)t + 8) = \
 156                                     *(uint64_t *)((uintptr_t)s + 8); \
 157                         } else { \
 158                                 bcopy(s, t, l); \
 159                         } \
 160                 } else \
 161                         sa_copy_data(f, s, t, l); \
 162         }
 163
 164 /*
 165  * This table is fixed and cannot be changed.  Its purpose is to
 166  * allow the SA code to work with both old/new ZPL file systems.
 167  * It contains the list of legacy attributes.  These attributes aren't
 168  * stored in the "attribute" registry zap objects, since older ZPL file systems
 169  * won't have the registry.  Only objsets of type ZFS_TYPE_FILESYSTEM will
 170  * use this static table.
 171  */
 172 sa_attr_reg_t sa_legacy_attrs[] = {
 173         {"ZPL_ATIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 0},
 174         {"ZPL_MTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 1},
 175         {"ZPL_CTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 2},
 176         {"ZPL_CRTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 3},
 177         {"ZPL_GEN", sizeof (uint64_t), SA_UINT64_ARRAY, 4},
 178         {"ZPL_MODE", sizeof (uint64_t), SA_UINT64_ARRAY, 5},
 179         {"ZPL_SIZE", sizeof (uint64_t), SA_UINT64_ARRAY, 6},
 180         {"ZPL_PARENT", sizeof (uint64_t), SA_UINT64_ARRAY, 7},
 181         {"ZPL_LINKS", sizeof (uint64_t), SA_UINT64_ARRAY, 8},
 182         {"ZPL_XATTR", sizeof (uint64_t), SA_UINT64_ARRAY, 9},
 183         {"ZPL_RDEV", sizeof (uint64_t), SA_UINT64_ARRAY, 10},
 184         {"ZPL_FLAGS", sizeof (uint64_t), SA_UINT64_ARRAY, 11},
 185         {"ZPL_UID", sizeof (uint64_t), SA_UINT64_ARRAY, 12},
 186         {"ZPL_GID", sizeof (uint64_t), SA_UINT64_ARRAY, 13},
 187         {"ZPL_PAD", sizeof (uint64_t) * 4, SA_UINT64_ARRAY, 14},
 188         {"ZPL_ZNODE_ACL", 88, SA_UINT8_ARRAY, 15},
 189 };
 190
 191 /*
 192  * ZPL legacy layout
 193  * This is only used for objects of type DMU_OT_ZNODE
 194  */
 195 sa_attr_type_t sa_legacy_zpl_layout[] = {
 196     0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15
 197 };
 198
 199 /*
 200  * Special dummy layout used for buffers with no attributes.
 201  */
 202
 203 sa_attr_type_t sa_dummy_zpl_layout[] = { 0 };
 204
 205 static int sa_legacy_attr_count = 16;
 206 static kmem_cache_t *sa_cache = NULL;
 207
 208 /*ARGSUSED*/
 209 static int
 210 sa_cache_constructor(void *buf, void *unused, int kmflag)
 211 {
 212         sa_handle_t *hdl = buf;
 213
 214         hdl->sa_bonus_tab = NULL;
 215         hdl->sa_spill_tab = NULL;
 216         hdl->sa_os = NULL;
 217         hdl->sa_userp = NULL;
 218         hdl->sa_bonus = NULL;
 219         hdl->sa_spill = NULL;
 220         mutex_init(&hdl->sa_lock, NULL, MUTEX_DEFAULT, NULL);
 221         return (0);
 222 }
 223
 224 /*ARGSUSED*/
 225 static void
 226 sa_cache_destructor(void *buf, void *unused)
 227 {
 228         sa_handle_t *hdl = buf;
 229         mutex_destroy(&hdl->sa_lock);
 230 }
 231
 232 void
 233 sa_cache_init(void)
 234 {
 235         sa_cache = kmem_cache_create("sa_cache",
 236             sizeof (sa_handle_t), 0, sa_cache_constructor,
 237             sa_cache_destructor, NULL, NULL, NULL, 0);
 238 }
 239
 240 void
 241 sa_cache_fini(void)
 242 {
 243         if (sa_cache)
 244                 kmem_cache_destroy(sa_cache);
 245 }
 246
 247 static int
 248 layout_num_compare(const void *arg1, const void *arg2)
 249 {
 250         const sa_lot_t *node1 = arg1;
 251         const sa_lot_t *node2 = arg2;
 252
 253         if (node1->lot_num > node2->lot_num)
 254                 return (1);
 255         else if (node1->lot_num < node2->lot_num)
 256                 return (-1);
 257         return (0);
 258 }
 259
 260 static int
 261 layout_hash_compare(const void *arg1, const void *arg2)
 262 {
 263         const sa_lot_t *node1 = arg1;
 264         const sa_lot_t *node2 = arg2;
 265
 266         if (node1->lot_hash > node2->lot_hash)
 267                 return (1);
 268         if (node1->lot_hash < node2->lot_hash)
 269                 return (-1);
 270         if (node1->lot_instance > node2->lot_instance)
 271                 return (1);
 272         if (node1->lot_instance < node2->lot_instance)
 273                 return (-1);
 274         return (0);
 275 }
 276
 277 boolean_t
 278 sa_layout_equal(sa_lot_t *tbf, sa_attr_type_t *attrs, int count)
 279 {
 280         int i;
 281
 282         if (count != tbf->lot_attr_count)
 283                 return (1);
 284
 285         for (i = 0; i != count; i++) {
 286                 if (attrs[i] != tbf->lot_attrs[i])
 287                         return (1);
 288         }
 289         return (0);
 290 }
 291
 292 #define SA_ATTR_HASH(attr) (zfs_crc64_table[(-1ULL ^ attr) & 0xFF])
 293
 294 static uint64_t
 295 sa_layout_info_hash(sa_attr_type_t *attrs, int attr_count)
 296 {
 297         int i;
 298         uint64_t crc = -1ULL;
 299
 300         for (i = 0; i != attr_count; i++)
 301                 crc ^= SA_ATTR_HASH(attrs[i]);
 302
 303         return (crc);
 304 }
 305
 306 static int
 307 sa_get_spill(sa_handle_t *hdl)
 308 {
 309         int rc;
 310         if (hdl->sa_spill == NULL) {
 311                 if ((rc = dmu_spill_hold_existing(hdl->sa_bonus, NULL,
 312                     &hdl->sa_spill)) == 0)
 313                         VERIFY(0 == sa_build_index(hdl, SA_SPILL));
 314         } else {
 315                 rc = 0;
 316         }
 317
 318         return (rc);
 319 }
 320
 321 /*
 322  * Main attribute lookup/update function
 323  * returns 0 for success or non zero for failures
 324  *
 325  * Operates on bulk array, first failure will abort further processing
 326  */
 327 int
 328 sa_attr_op(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count,
 329     sa_data_op_t data_op, dmu_tx_t *tx)
 330 {
 331         sa_os_t *sa = hdl->sa_os->os_sa;
 332         int i;
 333         int error = 0;
 334         sa_buf_type_t buftypes;
 335
 336         buftypes = 0;
 337
 338         ASSERT(count > 0);
 339         for (i = 0; i != count; i++) {
 340                 ASSERT(bulk[i].sa_attr <= hdl->sa_os->os_sa->sa_num_attrs);
 341
 342                 bulk[i].sa_addr = NULL;
 343                 /* First check the bonus buffer */
 344
 345                 if (hdl->sa_bonus_tab && TOC_ATTR_PRESENT(
 346                     hdl->sa_bonus_tab->sa_idx_tab[bulk[i].sa_attr])) {
 347                         SA_ATTR_INFO(sa, hdl->sa_bonus_tab,
 348                             SA_GET_HDR(hdl, SA_BONUS),
 349                             bulk[i].sa_attr, bulk[i], SA_BONUS, hdl);
 350                         if (tx && !(buftypes & SA_BONUS)) {
 351                                 dmu_buf_will_dirty(hdl->sa_bonus, tx);
 352                                 buftypes |= SA_BONUS;
 353                         }
 354                 }
 355                 if (bulk[i].sa_addr == NULL &&
 356                     ((error = sa_get_spill(hdl)) == 0)) {
 357                         if (TOC_ATTR_PRESENT(
 358                             hdl->sa_spill_tab->sa_idx_tab[bulk[i].sa_attr])) {
 359                                 SA_ATTR_INFO(sa, hdl->sa_spill_tab,
 360                                     SA_GET_HDR(hdl, SA_SPILL),
 361                                     bulk[i].sa_attr, bulk[i], SA_SPILL, hdl);
 362                                 if (tx && !(buftypes & SA_SPILL) &&
 363                                     bulk[i].sa_size == bulk[i].sa_length) {
 364                                         dmu_buf_will_dirty(hdl->sa_spill, tx);
 365                                         buftypes |= SA_SPILL;
 366                                 }
 367                         }
 368                 }
 369                 if (error && error != ENOENT) {
 370                         return ((error == ECKSUM) ? EIO : error);
 371                 }
 372
 373                 switch (data_op) {
 374                 case SA_LOOKUP:
 375                         if (bulk[i].sa_addr == NULL)
 376                                 return (ENOENT);
 377                         if (bulk[i].sa_data) {
 378                                 SA_COPY_DATA(bulk[i].sa_data_func,
 379                                     bulk[i].sa_addr, bulk[i].sa_data,
 380                                     bulk[i].sa_size);
 381                         }
 382                         continue;
 383
 384                 case SA_UPDATE:
 385                         /* existing rewrite of attr */
 386                         if (bulk[i].sa_addr &&
 387                             bulk[i].sa_size == bulk[i].sa_length) {
 388                                 SA_COPY_DATA(bulk[i].sa_data_func,
 389                                     bulk[i].sa_data, bulk[i].sa_addr,
 390                                     bulk[i].sa_length);
 391                                 continue;
 392                         } else if (bulk[i].sa_addr) { /* attr size change */
 393                                 error = sa_modify_attrs(hdl, bulk[i].sa_attr,
 394                                     SA_REPLACE, bulk[i].sa_data_func,
 395                                     bulk[i].sa_data, bulk[i].sa_length, tx);
 396                         } else { /* adding new attribute */
 397                                 error = sa_modify_attrs(hdl, bulk[i].sa_attr,
 398                                     SA_ADD, bulk[i].sa_data_func,
 399                                     bulk[i].sa_data, bulk[i].sa_length, tx);
 400                         }
 401                         if (error)
 402                                 return (error);
 403                         break;
 404                 }
 405         }
 406         return (error);
 407 }
 408
 409 static sa_lot_t *
 410 sa_add_layout_entry(objset_t *os, sa_attr_type_t *attrs, int attr_count,
 411     uint64_t lot_num, uint64_t hash, boolean_t zapadd, dmu_tx_t *tx)
 412 {
 413         sa_os_t *sa = os->os_sa;
 414         sa_lot_t *tb, *findtb;
 415         int i;
 416         avl_index_t loc;
 417
 418         ASSERT(MUTEX_HELD(&sa->sa_lock));
 419         tb = kmem_zalloc(sizeof (sa_lot_t), KM_SLEEP);
 420         tb->lot_attr_count = attr_count;
 421         tb->lot_attrs = kmem_alloc(sizeof (sa_attr_type_t) * attr_count,
 422             KM_SLEEP);
 423         bcopy(attrs, tb->lot_attrs, sizeof (sa_attr_type_t) * attr_count);
 424         tb->lot_num = lot_num;
 425         tb->lot_hash = hash;
 426         tb->lot_instance = 0;
 427
 428         if (zapadd) {
 429                 char attr_name[8];
 430
 431                 if (sa->sa_layout_attr_obj == 0) {
 432                         sa->sa_layout_attr_obj = zap_create_link(os,
 433                             DMU_OT_SA_ATTR_LAYOUTS,
 434                             sa->sa_master_obj, SA_LAYOUTS, tx);
 435                 }
 436
 437                 (void) snprintf(attr_name, sizeof (attr_name),
 438                     "%d", (int)lot_num);
 439                 VERIFY(0 == zap_update(os, os->os_sa->sa_layout_attr_obj,
 440                     attr_name, 2, attr_count, attrs, tx));
 441         }
 442
 443         list_create(&tb->lot_idx_tab, sizeof (sa_idx_tab_t),
 444             offsetof(sa_idx_tab_t, sa_next));
 445
 446         for (i = 0; i != attr_count; i++) {
 447                 if (sa->sa_attr_table[tb->lot_attrs[i]].sa_length == 0)
 448                         tb->lot_var_sizes++;
 449         }
 450
 451         avl_add(&sa->sa_layout_num_tree, tb);
 452
 453         /* verify we don't have a hash collision */
 454         if ((findtb = avl_find(&sa->sa_layout_hash_tree, tb, &loc)) != NULL) {
 455                 for (; findtb && findtb->lot_hash == hash;
 456                     findtb = AVL_NEXT(&sa->sa_layout_hash_tree, findtb)) {
 457                         if (findtb->lot_instance != tb->lot_instance)
 458                                 break;
 459                         tb->lot_instance++;
 460                 }
 461         }
 462         avl_add(&sa->sa_layout_hash_tree, tb);
 463         return (tb);
 464 }
 465
 466 static void
 467 sa_find_layout(objset_t *os, uint64_t hash, sa_attr_type_t *attrs,
 468     int count, dmu_tx_t *tx, sa_lot_t **lot)
 469 {
 470         sa_lot_t *tb, tbsearch;
 471         avl_index_t loc;
 472         sa_os_t *sa = os->os_sa;
 473         boolean_t found = B_FALSE;
 474
 475         mutex_enter(&sa->sa_lock);
 476         tbsearch.lot_hash = hash;
 477         tbsearch.lot_instance = 0;
 478         tb = avl_find(&sa->sa_layout_hash_tree, &tbsearch, &loc);
 479         if (tb) {
 480                 for (; tb && tb->lot_hash == hash;
 481                     tb = AVL_NEXT(&sa->sa_layout_hash_tree, tb)) {
 482                         if (sa_layout_equal(tb, attrs, count) == 0) {
 483                                 found = B_TRUE;
 484                                 break;
 485                         }
 486                 }
 487         }
 488         if (!found) {
 489                 tb = sa_add_layout_entry(os, attrs, count,
 490                     avl_numnodes(&sa->sa_layout_num_tree), hash, B_TRUE, tx);
 491         }
 492         mutex_exit(&sa->sa_lock);
 493         *lot = tb;
 494 }
 495
 496 static int
 497 sa_resize_spill(sa_handle_t *hdl, uint32_t size, dmu_tx_t *tx)
 498 {
 499         int error;
 500         uint32_t blocksize;
 501
 502         if (size == 0) {
 503                 blocksize = SPA_MINBLOCKSIZE;
 504         } else if (size > SPA_MAXBLOCKSIZE) {
 505                 ASSERT(0);
 506                 return (EFBIG);
 507         } else {
 508                 blocksize = P2ROUNDUP_TYPED(size, SPA_MINBLOCKSIZE, uint32_t);
 509         }
 510
 511         error = dbuf_spill_set_blksz(hdl->sa_spill, blocksize, tx);
 512         ASSERT(error == 0);
 513         return (error);
 514 }
 515
 516 static void
 517 sa_copy_data(sa_data_locator_t *func, void *datastart, void *target, int buflen)
 518 {
 519         if (func == NULL) {
 520                 bcopy(datastart, target, buflen);
 521         } else {
 522                 boolean_t start;
 523                 int bytes;
 524                 void *dataptr;
 525                 void *saptr = target;
 526                 uint32_t length;
 527
 528                 start = B_TRUE;
 529                 bytes = 0;
 530                 while (bytes < buflen) {
 531                         func(&dataptr, &length, buflen, start, datastart);
 532                         bcopy(dataptr, saptr, length);
 533                         saptr = (void *)((caddr_t)saptr + length);
 534                         bytes += length;
 535                         start = B_FALSE;
 536                 }
 537         }
 538 }
 539
 540 /*
 541  * Determine several different sizes
 542  * first the sa header size
 543  * the number of bytes to be stored
 544  * if spill would occur the index in the attribute array is returned
 545  *
 546  * the boolean will_spill will be set when spilling is necessary.  It
 547  * is only set when the buftype is SA_BONUS
 548  */
 549 static int
 550 sa_find_sizes(sa_os_t *sa, sa_bulk_attr_t *attr_desc, int attr_count,
 551     dmu_buf_t *db, sa_buf_type_t buftype, int *index, int *total,
 552     boolean_t *will_spill)
 553 {
 554         int var_size = 0;
 555         int i;
 556         int full_space;
 557         int hdrsize;
 558         boolean_t done = B_FALSE;
 559
 560         if (buftype == SA_BONUS && sa->sa_force_spill) {
 561                 *total = 0;
 562                 *index = 0;
 563                 *will_spill = B_TRUE;
 564                 return (0);
 565         }
 566
 567         *index = -1;
 568         *total = 0;
 569
 570         if (buftype == SA_BONUS)
 571                 *will_spill = B_FALSE;
 572
 573         hdrsize = (SA_BONUSTYPE_FROM_DB(db) == DMU_OT_ZNODE) ? 0 :
 574             sizeof (sa_hdr_phys_t);
 575
 576         full_space = (buftype == SA_BONUS) ? DN_MAX_BONUSLEN : db->db_size;
 577
 578         for (i = 0; i != attr_count; i++) {
 579                 boolean_t is_var_sz;
 580
 581                 *total += P2ROUNDUP(attr_desc[i].sa_length, 8);
 582                 if (done)
 583                         goto next;
 584
 585                 is_var_sz = (SA_REGISTERED_LEN(sa, attr_desc[i].sa_attr) == 0);
 586                 if (is_var_sz) {
 587                         var_size++;
 588                 }
 589
 590                 if (is_var_sz && var_size > 1) {
 591                         if (P2ROUNDUP(hdrsize + sizeof (uint16_t), 8) +
 592                             *total < full_space) {
 593                                 hdrsize += sizeof (uint16_t);
 594                         } else {
 595                                 done = B_TRUE;
 596                                 *index = i;
 597                                 if (buftype == SA_BONUS)
 598                                         *will_spill = B_TRUE;
 599                                 continue;
 600                         }
 601                 }
 602
 603                 /*
 604                  * find index of where spill *could* occur.
 605                  * Then continue to count of remainder attribute
 606                  * space.  The sum is used later for sizing bonus
 607                  * and spill buffer.
 608                  */
 609                 if (buftype == SA_BONUS && *index == -1 &&
 610                     (*total + P2ROUNDUP(hdrsize, 8)) >
 611                     (full_space - sizeof (blkptr_t))) {
 612                         *index = i;
 613                         done = B_TRUE;
 614                 }
 615
 616 next:
 617                 if ((*total + P2ROUNDUP(hdrsize, 8)) > full_space &&
 618                     buftype == SA_BONUS)
 619                         *will_spill = B_TRUE;
 620         }
 621
 622         hdrsize = P2ROUNDUP(hdrsize, 8);
 623         return (hdrsize);
 624 }
 625
 626 #define BUF_SPACE_NEEDED(total, header) (total + header)
 627
 628 /*
 629  * Find layout that corresponds to ordering of attributes
 630  * If not found a new layout number is created and added to
 631  * persistent layout tables.
 632  */
 633 static int
 634 sa_build_layouts(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc, int attr_count,
 635     dmu_tx_t *tx)
 636 {
 637         sa_os_t *sa = hdl->sa_os->os_sa;
 638         uint64_t hash;
 639         sa_buf_type_t buftype;
 640         sa_hdr_phys_t *sahdr;
 641         void *data_start;
 642         int buf_space;
 643         sa_attr_type_t *attrs, *attrs_start;
 644         int i, lot_count;
 645         int hdrsize, spillhdrsize;
 646         int used;
 647         dmu_object_type_t bonustype;
 648         sa_lot_t *lot;
 649         int len_idx;
 650         int spill_used;
 651         boolean_t spilling;
 652
 653         dmu_buf_will_dirty(hdl->sa_bonus, tx);
 654         bonustype = SA_BONUSTYPE_FROM_DB(hdl->sa_bonus);
 655
 656         /* first determine bonus header size and sum of all attributes */
 657         hdrsize = sa_find_sizes(sa, attr_desc, attr_count, hdl->sa_bonus,
 658             SA_BONUS, &i, &used, &spilling);
 659
 660         if (used > SPA_MAXBLOCKSIZE)
 661                 return (EFBIG);
 662
 663         VERIFY(0 == dmu_set_bonus(hdl->sa_bonus, spilling ?
 664             MIN(DN_MAX_BONUSLEN - sizeof (blkptr_t), used + hdrsize) :
 665             used + hdrsize, tx));
 666
 667         ASSERT((bonustype == DMU_OT_ZNODE && spilling == 0) ||
 668             bonustype == DMU_OT_SA);
 669
 670         /* setup and size spill buffer when needed */
 671         if (spilling) {
 672                 boolean_t dummy;
 673
 674                 if (hdl->sa_spill == NULL) {
 675                         VERIFY(dmu_spill_hold_by_bonus(hdl->sa_bonus, NULL,
 676                             &hdl->sa_spill) == 0);
 677                 }
 678                 dmu_buf_will_dirty(hdl->sa_spill, tx);
 679
 680                 spillhdrsize = sa_find_sizes(sa, &attr_desc[i],
 681                     attr_count - i, hdl->sa_spill, SA_SPILL, &i,
 682                     &spill_used, &dummy);
 683
 684                 if (spill_used > SPA_MAXBLOCKSIZE)
 685                         return (EFBIG);
 686
 687                 buf_space = hdl->sa_spill->db_size - spillhdrsize;
 688                 if (BUF_SPACE_NEEDED(spill_used, spillhdrsize) >
 689                     hdl->sa_spill->db_size)
 690                         VERIFY(0 == sa_resize_spill(hdl,
 691                             BUF_SPACE_NEEDED(spill_used, spillhdrsize), tx));
 692         }
 693
 694         /* setup starting pointers to lay down data */
 695         data_start = (void *)((uintptr_t)hdl->sa_bonus->db_data + hdrsize);
 696         sahdr = (sa_hdr_phys_t *)hdl->sa_bonus->db_data;
 697         buftype = SA_BONUS;
 698
 699         if (spilling)
 700                 buf_space = (sa->sa_force_spill) ?
 701                     0 : SA_BLKPTR_SPACE - hdrsize;
 702         else
 703                 buf_space = hdl->sa_bonus->db_size - hdrsize;
 704
 705         attrs_start = attrs = kmem_alloc(sizeof (sa_attr_type_t) * attr_count,
 706             KM_SLEEP);
 707         lot_count = 0;
 708
 709         for (i = 0, len_idx = 0, hash = -1ULL; i != attr_count; i++) {
 710                 uint16_t length;
 711
 712                 attrs[i] = attr_desc[i].sa_attr;
 713                 length = SA_REGISTERED_LEN(sa, attrs[i]);
 714                 if (length == 0)
 715                         length = attr_desc[i].sa_length;
 716                 else
 717                         VERIFY(length == attr_desc[i].sa_length);
 718
 719                 if (buf_space < length) {  /* switch to spill buffer */
 720                         VERIFY(bonustype == DMU_OT_SA);
 721                         if (buftype == SA_BONUS && !sa->sa_force_spill) {
 722                                 sa_find_layout(hdl->sa_os, hash, attrs_start,
 723                                     lot_count, tx, &lot);
 724                                 SA_SET_HDR(sahdr, lot->lot_num, hdrsize);
 725                         }
 726
 727                         buftype = SA_SPILL;
 728                         hash = -1ULL;
 729                         len_idx = 0;
 730
 731                         sahdr = (sa_hdr_phys_t *)hdl->sa_spill->db_data;
 732                         sahdr->sa_magic = SA_MAGIC;
 733                         data_start = (void *)((uintptr_t)sahdr +
 734                             spillhdrsize);
 735                         attrs_start = &attrs[i];
 736                         buf_space = hdl->sa_spill->db_size - spillhdrsize;
 737                         lot_count = 0;
 738                 }
 739                 hash ^= SA_ATTR_HASH(attrs[i]);
 740                 attr_desc[i].sa_addr = data_start;
 741                 attr_desc[i].sa_size = length;
 742                 SA_COPY_DATA(attr_desc[i].sa_data_func, attr_desc[i].sa_data,
 743                     data_start, length);
 744                 if (sa->sa_attr_table[attrs[i]].sa_length == 0) {
 745                         sahdr->sa_lengths[len_idx++] = length;
 746                 }
 747                 VERIFY((uintptr_t)data_start % 8 == 0);
 748                 data_start = (void *)P2ROUNDUP(((uintptr_t)data_start +
 749                     length), 8);
 750                 buf_space -= P2ROUNDUP(length, 8);
 751                 lot_count++;
 752         }
 753
 754         sa_find_layout(hdl->sa_os, hash, attrs_start, lot_count, tx, &lot);
 755
 756         /*
 757          * Verify that old znodes always have layout number 0.
 758          * Must be DMU_OT_SA for arbitrary layouts
 759          */
 760         VERIFY((bonustype == DMU_OT_ZNODE && lot->lot_num == 0) ||
 761             (bonustype == DMU_OT_SA && lot->lot_num > 1));
 762
 763         if (bonustype == DMU_OT_SA) {
 764                 SA_SET_HDR(sahdr, lot->lot_num,
 765                     buftype == SA_BONUS ? hdrsize : spillhdrsize);
 766         }
 767
 768         kmem_free(attrs, sizeof (sa_attr_type_t) * attr_count);
 769         if (hdl->sa_bonus_tab) {
 770                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_bonus_tab);
 771                 hdl->sa_bonus_tab = NULL;
 772         }
 773         if (!sa->sa_force_spill)
 774                 VERIFY(0 == sa_build_index(hdl, SA_BONUS));
 775         if (hdl->sa_spill) {
 776                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
 777                 if (!spilling) {
 778                         /*
 779                          * remove spill block that is no longer needed.
 780                          */
 781                         dmu_buf_rele(hdl->sa_spill, NULL);
 782                         hdl->sa_spill = NULL;
 783                         hdl->sa_spill_tab = NULL;
 784                         VERIFY(0 == dmu_rm_spill(hdl->sa_os,
 785                             sa_handle_object(hdl), tx));
 786                 } else {
 787                         VERIFY(0 == sa_build_index(hdl, SA_SPILL));
 788                 }
 789         }
 790
 791         return (0);
 792 }
 793
 794 static void
 795 sa_free_attr_table(sa_os_t *sa)
 796 {
 797         int i;
 798
 799         if (sa->sa_attr_table == NULL)
 800                 return;
 801
 802         for (i = 0; i != sa->sa_num_attrs; i++) {
 803                 if (sa->sa_attr_table[i].sa_name)
 804                         kmem_free(sa->sa_attr_table[i].sa_name,
 805                             strlen(sa->sa_attr_table[i].sa_name) + 1);
 806         }
 807
 808         kmem_free(sa->sa_attr_table,
 809             sizeof (sa_attr_table_t) * sa->sa_num_attrs);
 810
 811         sa->sa_attr_table = NULL;
 812 }
 813
 814 static int
 815 sa_attr_table_setup(objset_t *os, sa_attr_reg_t *reg_attrs, int count)
 816 {
 817         sa_os_t *sa = os->os_sa;
 818         uint64_t sa_attr_count = 0;
 819         uint64_t sa_reg_count;
 820         int error = 0;
 821         uint64_t attr_value;
 822         sa_attr_table_t *tb;
 823         zap_cursor_t zc;
 824         zap_attribute_t za;
 825         int registered_count = 0;
 826         int i;
 827         dmu_objset_type_t ostype = dmu_objset_type(os);
 828
 829         sa->sa_user_table =
 830             kmem_zalloc(count * sizeof (sa_attr_type_t), KM_SLEEP);
 831         sa->sa_user_table_sz = count * sizeof (sa_attr_type_t);
 832
 833         if (sa->sa_reg_attr_obj != 0) {
 834                 error = zap_count(os, sa->sa_reg_attr_obj,
 835                     &sa_attr_count);
 836
 837                 /*
 838                  * Make sure we retrieved a count and that it isn't zero
 839                  */
 840                 if (error || (error == 0 && sa_attr_count == 0)) {
 841                         if (error == 0)
 842                                 error = EINVAL;
 843                         goto bail;
 844                 }
 845                 sa_reg_count = sa_attr_count;
 846         }
 847
 848         if (ostype == DMU_OST_ZFS && sa_attr_count == 0)
 849                 sa_attr_count += sa_legacy_attr_count;
 850
 851         /* Allocate attribute numbers for attributes that aren't registered */
 852         for (i = 0; i != count; i++) {
 853                 boolean_t found = B_FALSE;
 854                 int j;
 855
 856                 if (ostype == DMU_OST_ZFS) {
 857                         for (j = 0; j != sa_legacy_attr_count; j++) {
 858                                 if (strcmp(reg_attrs[i].sa_name,
 859                                     sa_legacy_attrs[j].sa_name) == 0) {
 860                                         sa->sa_user_table[i] =
 861                                             sa_legacy_attrs[j].sa_attr;
 862                                         found = B_TRUE;
 863                                 }
 864                         }
 865                 }
 866                 if (found)
 867                         continue;
 868
 869                 if (sa->sa_reg_attr_obj)
 870                         error = zap_lookup(os, sa->sa_reg_attr_obj,
 871                             reg_attrs[i].sa_name, 8, 1, &attr_value);
 872                 else
 873                         error = ENOENT;
 874                 switch (error) {
 875                 case ENOENT:
 876                         sa->sa_user_table[i] = (sa_attr_type_t)sa_attr_count;
 877                         sa_attr_count++;
 878                         break;
 879                 case 0:
 880                         sa->sa_user_table[i] = ATTR_NUM(attr_value);
 881                         break;
 882                 default:
 883                         goto bail;
 884                 }
 885         }
 886
 887         sa->sa_num_attrs = sa_attr_count;
 888         tb = sa->sa_attr_table =
 889             kmem_zalloc(sizeof (sa_attr_table_t) * sa_attr_count, KM_SLEEP);
 890
 891         /*
 892          * Attribute table is constructed from requested attribute list,
 893          * previously foreign registered attributes, and also the legacy
 894          * ZPL set of attributes.
 895          */
 896
 897         if (sa->sa_reg_attr_obj) {
 898                 for (zap_cursor_init(&zc, os, sa->sa_reg_attr_obj);
 899                     (error = zap_cursor_retrieve(&zc, &za)) == 0;
 900                     zap_cursor_advance(&zc)) {
 901                         uint64_t value;
 902                         value  = za.za_first_integer;
 903
 904                         registered_count++;
 905                         tb[ATTR_NUM(value)].sa_attr = ATTR_NUM(value);
 906                         tb[ATTR_NUM(value)].sa_length = ATTR_LENGTH(value);
 907                         tb[ATTR_NUM(value)].sa_byteswap = ATTR_BSWAP(value);
 908                         tb[ATTR_NUM(value)].sa_registered = B_TRUE;
 909
 910                         if (tb[ATTR_NUM(value)].sa_name) {
 911                                 continue;
 912                         }
 913                         tb[ATTR_NUM(value)].sa_name =
 914                             kmem_zalloc(strlen(za.za_name) +1, KM_SLEEP);
 915                         (void) strlcpy(tb[ATTR_NUM(value)].sa_name, za.za_name,
 916                             strlen(za.za_name) +1);
 917                 }
 918                 zap_cursor_fini(&zc);
 919                 /*
 920                  * Make sure we processed the correct number of registered
 921                  * attributes
 922                  */
 923                 if (registered_count != sa_reg_count) {
 924                         ASSERT(error != 0);
 925                         goto bail;
 926                 }
 927
 928         }
 929
 930         if (ostype == DMU_OST_ZFS) {
 931                 for (i = 0; i != sa_legacy_attr_count; i++) {
 932                         if (tb[i].sa_name)
 933                                 continue;
 934                         tb[i].sa_attr = sa_legacy_attrs[i].sa_attr;
 935                         tb[i].sa_length = sa_legacy_attrs[i].sa_length;
 936                         tb[i].sa_byteswap = sa_legacy_attrs[i].sa_byteswap;
 937                         tb[i].sa_registered = B_FALSE;
 938                         tb[i].sa_name =
 939                             kmem_zalloc(strlen(sa_legacy_attrs[i].sa_name) +1,
 940                             KM_SLEEP);
 941                         (void) strlcpy(tb[i].sa_name,
 942                             sa_legacy_attrs[i].sa_name,
 943                             strlen(sa_legacy_attrs[i].sa_name) + 1);
 944                 }
 945         }
 946
 947         for (i = 0; i != count; i++) {
 948                 sa_attr_type_t attr_id;
 949
 950                 attr_id = sa->sa_user_table[i];
 951                 if (tb[attr_id].sa_name)
 952                         continue;
 953
 954                 tb[attr_id].sa_length = reg_attrs[i].sa_length;
 955                 tb[attr_id].sa_byteswap = reg_attrs[i].sa_byteswap;
 956                 tb[attr_id].sa_attr = attr_id;
 957                 tb[attr_id].sa_name =
 958                     kmem_zalloc(strlen(reg_attrs[i].sa_name) + 1, KM_SLEEP);
 959                 (void) strlcpy(tb[attr_id].sa_name, reg_attrs[i].sa_name,
 960                     strlen(reg_attrs[i].sa_name) + 1);
 961         }
 962
 963         sa->sa_need_attr_registration =
 964             (sa_attr_count != registered_count);
 965
 966         return (0);
 967 bail:
 968         kmem_free(sa->sa_user_table, count * sizeof (sa_attr_type_t));
 969         sa->sa_user_table = NULL;
 970         sa_free_attr_table(sa);
 971         return ((error != 0) ? error : EINVAL);
 972 }
 973
 974 int
 975 sa_setup(objset_t *os, uint64_t sa_obj, sa_attr_reg_t *reg_attrs, int count,
 976     sa_attr_type_t **user_table)
 977 {
 978         zap_cursor_t zc;
 979         zap_attribute_t za;
 980         sa_os_t *sa;
 981         dmu_objset_type_t ostype = dmu_objset_type(os);
 982         sa_attr_type_t *tb;
 983         int error;
 984
 985         mutex_enter(&os->os_lock);
 986         if (os->os_sa) {
 987                 mutex_enter(&os->os_sa->sa_lock);
 988                 mutex_exit(&os->os_lock);
 989                 tb = os->os_sa->sa_user_table;
 990                 mutex_exit(&os->os_sa->sa_lock);
 991                 *user_table = tb;
 992                 return (0);
 993         }
 994
 995         sa = kmem_zalloc(sizeof (sa_os_t), KM_SLEEP);
 996         mutex_init(&sa->sa_lock, NULL, MUTEX_DEFAULT, NULL);
 997         sa->sa_master_obj = sa_obj;
 998
 999         os->os_sa = sa;
1000         mutex_enter(&sa->sa_lock);
1001         mutex_exit(&os->os_lock);
1002         avl_create(&sa->sa_layout_num_tree, layout_num_compare,
1003             sizeof (sa_lot_t), offsetof(sa_lot_t, lot_num_node));
1004         avl_create(&sa->sa_layout_hash_tree, layout_hash_compare,
1005             sizeof (sa_lot_t), offsetof(sa_lot_t, lot_hash_node));
1006
1007         if (sa_obj) {
1008                 error = zap_lookup(os, sa_obj, SA_LAYOUTS,
1009                     8, 1, &sa->sa_layout_attr_obj);
1010                 if (error != 0 && error != ENOENT)
1011                         goto fail;
1012                 error = zap_lookup(os, sa_obj, SA_REGISTRY,
1013                     8, 1, &sa->sa_reg_attr_obj);
1014                 if (error != 0 && error != ENOENT)
1015                         goto fail;
1016         }
1017
1018         if ((error = sa_attr_table_setup(os, reg_attrs, count)) != 0)
1019                 goto fail;
1020
1021         if (sa->sa_layout_attr_obj != 0) {
1022                 uint64_t layout_count;
1023
1024                 error = zap_count(os, sa->sa_layout_attr_obj,
1025                     &layout_count);
1026
1027                 /*
1028                  * Layout number count should be > 0
1029                  */
1030                 if (error || (error == 0 && layout_count == 0)) {
1031                         if (error == 0)
1032                                 error = EINVAL;
1033                         goto fail;
1034                 }
1035
1036                 for (zap_cursor_init(&zc, os, sa->sa_layout_attr_obj);
1037                     (error = zap_cursor_retrieve(&zc, &za)) == 0;
1038                     zap_cursor_advance(&zc)) {
1039                         sa_attr_type_t *lot_attrs;
1040                         uint64_t lot_num;
1041
1042                         lot_attrs = kmem_zalloc(sizeof (sa_attr_type_t) *
1043                             za.za_num_integers, KM_SLEEP);
1044
1045                         if ((error = (zap_lookup(os, sa->sa_layout_attr_obj,
1046                             za.za_name, 2, za.za_num_integers,
1047                             lot_attrs))) != 0) {
1048                                 kmem_free(lot_attrs, sizeof (sa_attr_type_t) *
1049                                     za.za_num_integers);
1050                                 break;
1051                         }
1052                         VERIFY(ddi_strtoull(za.za_name, NULL, 10,
1053                             (unsigned long long *)&lot_num) == 0);
1054
1055                         (void) sa_add_layout_entry(os, lot_attrs,
1056                             za.za_num_integers, lot_num,
1057                             sa_layout_info_hash(lot_attrs,
1058                             za.za_num_integers), B_FALSE, NULL);
1059                         kmem_free(lot_attrs, sizeof (sa_attr_type_t) *
1060                             za.za_num_integers);
1061                 }
1062                 zap_cursor_fini(&zc);
1063
1064                 /*
1065                  * Make sure layout count matches number of entries added
1066                  * to AVL tree
1067                  */
1068                 if (avl_numnodes(&sa->sa_layout_num_tree) != layout_count) {
1069                         ASSERT(error != 0);
1070                         goto fail;
1071                 }
1072         }
1073
1074         /* Add special layout number for old ZNODES */
1075         if (ostype == DMU_OST_ZFS) {
1076                 (void) sa_add_layout_entry(os, sa_legacy_zpl_layout,
1077                     sa_legacy_attr_count, 0,
1078                     sa_layout_info_hash(sa_legacy_zpl_layout,
1079                     sa_legacy_attr_count), B_FALSE, NULL);
1080
1081                 (void) sa_add_layout_entry(os, sa_dummy_zpl_layout, 0, 1,
1082                     0, B_FALSE, NULL);
1083         }
1084         *user_table = os->os_sa->sa_user_table;
1085         mutex_exit(&sa->sa_lock);
1086         return (0);
1087 fail:
1088         os->os_sa = NULL;
1089         sa_free_attr_table(sa);
1090         if (sa->sa_user_table)
1091                 kmem_free(sa->sa_user_table, sa->sa_user_table_sz);
1092         mutex_exit(&sa->sa_lock);
1093         kmem_free(sa, sizeof (sa_os_t));
1094         return ((error == ECKSUM) ? EIO : error);
1095 }
1096
1097 void
1098 sa_tear_down(objset_t *os)
1099 {
1100         sa_os_t *sa = os->os_sa;
1101         sa_lot_t *layout;
1102         void *cookie;
1103
1104         kmem_free(sa->sa_user_table, sa->sa_user_table_sz);
1105
1106         /* Free up attr table */
1107
1108         sa_free_attr_table(sa);
1109
1110         cookie = NULL;
1111         while (layout = avl_destroy_nodes(&sa->sa_layout_hash_tree, &cookie)) {
1112                 sa_idx_tab_t *tab;
1113                 while (tab = list_head(&layout->lot_idx_tab)) {
1114                         ASSERT(refcount_count(&tab->sa_refcount));
1115                         sa_idx_tab_rele(os, tab);
1116                 }
1117         }
1118
1119         cookie = NULL;
1120         while (layout = avl_destroy_nodes(&sa->sa_layout_num_tree, &cookie)) {
1121                 kmem_free(layout->lot_attrs,
1122                     sizeof (sa_attr_type_t) * layout->lot_attr_count);
1123                 kmem_free(layout, sizeof (sa_lot_t));
1124         }
1125
1126         avl_destroy(&sa->sa_layout_hash_tree);
1127         avl_destroy(&sa->sa_layout_num_tree);
1128
1129         kmem_free(sa, sizeof (sa_os_t));
1130         os->os_sa = NULL;
1131 }
1132
1133 void
1134 sa_build_idx_tab(void *hdr, void *attr_addr, sa_attr_type_t attr,
1135     uint16_t length, int length_idx, boolean_t var_length, void *userp)
1136 {
1137         sa_idx_tab_t *idx_tab = userp;
1138
1139         if (var_length) {
1140                 ASSERT(idx_tab->sa_variable_lengths);
1141                 idx_tab->sa_variable_lengths[length_idx] = length;
1142         }
1143         TOC_ATTR_ENCODE(idx_tab->sa_idx_tab[attr], length_idx,
1144             (uint32_t)((uintptr_t)attr_addr - (uintptr_t)hdr));
1145 }
1146
1147 static void
1148 sa_attr_iter(objset_t *os, sa_hdr_phys_t *hdr, dmu_object_type_t type,
1149     sa_iterfunc_t func, sa_lot_t *tab, void *userp)
1150 {
1151         void *data_start;
1152         sa_lot_t *tb = tab;
1153         sa_lot_t search;
1154         avl_index_t loc;
1155         sa_os_t *sa = os->os_sa;
1156         int i;
1157         uint16_t *length_start = NULL;
1158         uint8_t length_idx = 0;
1159
1160         if (tab == NULL) {
1161                 search.lot_num = SA_LAYOUT_NUM(hdr, type);
1162                 tb = avl_find(&sa->sa_layout_num_tree, &search, &loc);
1163                 ASSERT(tb);
1164         }
1165
1166         if (IS_SA_BONUSTYPE(type)) {
1167                 data_start = (void *)P2ROUNDUP(((uintptr_t)hdr +
1168                     offsetof(sa_hdr_phys_t, sa_lengths) +
1169                     (sizeof (uint16_t) * tb->lot_var_sizes)), 8);
1170                 length_start = hdr->sa_lengths;
1171         } else {
1172                 data_start = hdr;
1173         }
1174
1175         for (i = 0; i != tb->lot_attr_count; i++) {
1176                 int attr_length, reg_length;
1177                 uint8_t idx_len;
1178
1179                 reg_length = sa->sa_attr_table[tb->lot_attrs[i]].sa_length;
1180                 if (reg_length) {
1181                         attr_length = reg_length;
1182                         idx_len = 0;
1183                 } else {
1184                         attr_length = length_start[length_idx];
1185                         idx_len = length_idx++;
1186                 }
1187
1188                 func(hdr, data_start, tb->lot_attrs[i], attr_length,
1189                     idx_len, reg_length == 0 ? B_TRUE : B_FALSE, userp);
1190
1191                 data_start = (void *)P2ROUNDUP(((uintptr_t)data_start +
1192                     attr_length), 8);
1193         }
1194 }
1195
1196 /*ARGSUSED*/
1197 void
1198 sa_byteswap_cb(void *hdr, void *attr_addr, sa_attr_type_t attr,
1199     uint16_t length, int length_idx, boolean_t variable_length, void *userp)
1200 {
1201         sa_handle_t *hdl = userp;
1202         sa_os_t *sa = hdl->sa_os->os_sa;
1203
1204         sa_bswap_table[sa->sa_attr_table[attr].sa_byteswap](attr_addr, length);
1205 }
1206
1207 void
1208 sa_byteswap(sa_handle_t *hdl, sa_buf_type_t buftype)
1209 {
1210         sa_hdr_phys_t *sa_hdr_phys = SA_GET_HDR(hdl, buftype);
1211         dmu_buf_impl_t *db;
1212         sa_os_t *sa = hdl->sa_os->os_sa;
1213         int num_lengths = 1;
1214         int i;
1215
1216         ASSERT(MUTEX_HELD(&sa->sa_lock));
1217         if (sa_hdr_phys->sa_magic == SA_MAGIC)
1218                 return;
1219
1220         db = SA_GET_DB(hdl, buftype);
1221
1222         if (buftype == SA_SPILL) {
1223                 arc_release(db->db_buf, NULL);
1224                 arc_buf_thaw(db->db_buf);
1225         }
1226
1227         sa_hdr_phys->sa_magic = BSWAP_32(sa_hdr_phys->sa_magic);
1228         sa_hdr_phys->sa_layout_info = BSWAP_16(sa_hdr_phys->sa_layout_info);
1229
1230         /*
1231          * Determine number of variable lenghts in header
1232          * The standard 8 byte header has one for free and a
1233          * 16 byte header would have 4 + 1;
1234          */
1235         if (SA_HDR_SIZE(sa_hdr_phys) > 8)
1236                 num_lengths += (SA_HDR_SIZE(sa_hdr_phys) - 8) >> 1;
1237         for (i = 0; i != num_lengths; i++)
1238                 sa_hdr_phys->sa_lengths[i] =
1239                     BSWAP_16(sa_hdr_phys->sa_lengths[i]);
1240
1241         sa_attr_iter(hdl->sa_os, sa_hdr_phys, DMU_OT_SA,
1242             sa_byteswap_cb, NULL, hdl);
1243
1244         if (buftype == SA_SPILL)
1245                 arc_buf_freeze(((dmu_buf_impl_t *)hdl->sa_spill)->db_buf);
1246 }
1247
1248 static int
1249 sa_build_index(sa_handle_t *hdl, sa_buf_type_t buftype)
1250 {
1251         sa_hdr_phys_t *sa_hdr_phys;
1252         dmu_buf_impl_t *db = SA_GET_DB(hdl, buftype);
1253         dmu_object_type_t bonustype = SA_BONUSTYPE_FROM_DB(db);
1254         sa_os_t *sa = hdl->sa_os->os_sa;
1255         sa_idx_tab_t *idx_tab;
1256
1257         sa_hdr_phys = SA_GET_HDR(hdl, buftype);
1258
1259         mutex_enter(&sa->sa_lock);
1260
1261         /* Do we need to byteswap? */
1262
1263         /* only check if not old znode */
1264         if (IS_SA_BONUSTYPE(bonustype) && sa_hdr_phys->sa_magic != SA_MAGIC &&
1265             sa_hdr_phys->sa_magic != 0) {
1266                 VERIFY(BSWAP_32(sa_hdr_phys->sa_magic) == SA_MAGIC);
1267                 sa_byteswap(hdl, buftype);
1268         }
1269
1270         idx_tab = sa_find_idx_tab(hdl->sa_os, bonustype, sa_hdr_phys);
1271
1272         if (buftype == SA_BONUS)
1273                 hdl->sa_bonus_tab = idx_tab;
1274         else
1275                 hdl->sa_spill_tab = idx_tab;
1276
1277         mutex_exit(&sa->sa_lock);
1278         return (0);
1279 }
1280
1281 /*ARGSUSED*/
1282 void
1283 sa_evict(dmu_buf_t *db, void *sap)
1284 {
1285         panic("evicting sa dbuf %p\n", (void *)db);
1286 }
1287
1288 static void
1289 sa_idx_tab_rele(objset_t *os, void *arg)
1290 {
1291         sa_os_t *sa = os->os_sa;
1292         sa_idx_tab_t *idx_tab = arg;
1293
1294         if (idx_tab == NULL)
1295                 return;
1296
1297         mutex_enter(&sa->sa_lock);
1298         if (refcount_remove(&idx_tab->sa_refcount, NULL) == 0) {
1299                 list_remove(&idx_tab->sa_layout->lot_idx_tab, idx_tab);
1300                 if (idx_tab->sa_variable_lengths)
1301                         kmem_free(idx_tab->sa_variable_lengths,
1302                             sizeof (uint16_t) *
1303                             idx_tab->sa_layout->lot_var_sizes);
1304                 refcount_destroy(&idx_tab->sa_refcount);
1305                 kmem_free(idx_tab->sa_idx_tab,
1306                     sizeof (uint32_t) * sa->sa_num_attrs);
1307                 kmem_free(idx_tab, sizeof (sa_idx_tab_t));
1308         }
1309         mutex_exit(&sa->sa_lock);
1310 }
1311
1312 static void
1313 sa_idx_tab_hold(objset_t *os, sa_idx_tab_t *idx_tab)
1314 {
1315         sa_os_t *sa = os->os_sa;
1316
1317         ASSERT(MUTEX_HELD(&sa->sa_lock));
1318         (void) refcount_add(&idx_tab->sa_refcount, NULL);
1319 }
1320
1321 void
1322 sa_handle_destroy(sa_handle_t *hdl)
1323 {
1324         mutex_enter(&hdl->sa_lock);
1325         (void) dmu_buf_update_user((dmu_buf_t *)hdl->sa_bonus, hdl,
1326             NULL, NULL, NULL);
1327
1328         if (hdl->sa_bonus_tab) {
1329                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_bonus_tab);
1330                 hdl->sa_bonus_tab = NULL;
1331         }
1332         if (hdl->sa_spill_tab) {
1333                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
1334                 hdl->sa_spill_tab = NULL;
1335         }
1336
1337         dmu_buf_rele(hdl->sa_bonus, NULL);
1338
1339         if (hdl->sa_spill)
1340                 dmu_buf_rele((dmu_buf_t *)hdl->sa_spill, NULL);
1341         mutex_exit(&hdl->sa_lock);
1342
1343         kmem_cache_free(sa_cache, hdl);
1344 }
1345
1346 int
1347 sa_handle_get_from_db(objset_t *os, dmu_buf_t *db, void *userp,
1348     sa_handle_type_t hdl_type, sa_handle_t **handlepp)
1349 {
1350         int error = 0;
1351         dmu_object_info_t doi;
1352         sa_handle_t *handle;
1353
1354 #ifdef ZFS_DEBUG
1355         dmu_object_info_from_db(db, &doi);
1356         ASSERT(doi.doi_bonus_type == DMU_OT_SA ||
1357             doi.doi_bonus_type == DMU_OT_ZNODE);
1358 #endif
1359         /* find handle, if it exists */
1360         /* if one doesn't exist then create a new one, and initialize it */
1361
1362         handle = (hdl_type == SA_HDL_SHARED) ? dmu_buf_get_user(db) : NULL;
1363         if (handle == NULL) {
1364                 sa_handle_t *newhandle;
1365                 handle = kmem_cache_alloc(sa_cache, KM_SLEEP);
1366                 handle->sa_userp = userp;
1367                 handle->sa_bonus = db;
1368                 handle->sa_os = os;
1369                 handle->sa_spill = NULL;
1370
1371                 error = sa_build_index(handle, SA_BONUS);
1372                 newhandle = (hdl_type == SA_HDL_SHARED) ?
1373                     dmu_buf_set_user_ie(db, handle,
1374                     NULL, sa_evict) : NULL;
1375
1376                 if (newhandle != NULL) {
1377                         kmem_cache_free(sa_cache, handle);
1378                         handle = newhandle;
1379                 }
1380         }
1381         *handlepp = handle;
1382
1383         return (error);
1384 }
1385
1386 int
1387 sa_handle_get(objset_t *objset, uint64_t objid, void *userp,
1388     sa_handle_type_t hdl_type, sa_handle_t **handlepp)
1389 {
1390         dmu_buf_t *db;
1391         int error;
1392
1393         if (error = dmu_bonus_hold(objset, objid, NULL, &db))
1394                 return (error);
1395
1396         return (sa_handle_get_from_db(objset, db, userp, hdl_type,
1397             handlepp));
1398 }
1399
1400 int
1401 sa_buf_hold(objset_t *objset, uint64_t obj_num, void *tag, dmu_buf_t **db)
1402 {
1403         return (dmu_bonus_hold(objset, obj_num, tag, db));
1404 }
1405
1406 void
1407 sa_buf_rele(dmu_buf_t *db, void *tag)
1408 {
1409         dmu_buf_rele(db, tag);
1410 }
1411
1412 int
1413 sa_lookup_impl(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count)
1414 {
1415         ASSERT(hdl);
1416         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1417         return (sa_attr_op(hdl, bulk, count, SA_LOOKUP, NULL));
1418 }
1419
1420 int
1421 sa_lookup(sa_handle_t *hdl, sa_attr_type_t attr, void *buf, uint32_t buflen)
1422 {
1423         int error;
1424         sa_bulk_attr_t bulk;
1425
1426         bulk.sa_attr = attr;
1427         bulk.sa_data = buf;
1428         bulk.sa_length = buflen;
1429         bulk.sa_data_func = NULL;
1430
1431         ASSERT(hdl);
1432         mutex_enter(&hdl->sa_lock);
1433         error = sa_lookup_impl(hdl, &bulk, 1);
1434         mutex_exit(&hdl->sa_lock);
1435         return (error);
1436 }
1437
1438 #ifdef _KERNEL
1439 int
1440 sa_lookup_uio(sa_handle_t *hdl, sa_attr_type_t attr, uio_t *uio)
1441 {
1442         int error;
1443         sa_bulk_attr_t bulk;
1444
1445         bulk.sa_data = NULL;
1446         bulk.sa_attr = attr;
1447         bulk.sa_data_func = NULL;
1448
1449         ASSERT(hdl);
1450
1451         mutex_enter(&hdl->sa_lock);
1452         if ((error = sa_attr_op(hdl, &bulk, 1, SA_LOOKUP, NULL)) == 0) {
1453                 error = uiomove((void *)bulk.sa_addr, MIN(bulk.sa_size,
1454                     uio->uio_resid), UIO_READ, uio);
1455         }
1456         mutex_exit(&hdl->sa_lock);
1457         return (error);
1458
1459 }
1460 #endif
1461
1462 void *
1463 sa_find_idx_tab(objset_t *os, dmu_object_type_t bonustype, void *data)
1464 {
1465         sa_idx_tab_t *idx_tab;
1466         sa_hdr_phys_t *hdr = (sa_hdr_phys_t *)data;
1467         sa_os_t *sa = os->os_sa;
1468         sa_lot_t *tb, search;
1469         avl_index_t loc;
1470
1471         /*
1472          * Deterimine layout number.  If SA node and header == 0 then
1473          * force the index table to the dummy "1" empty layout.
1474          *
1475          * The layout number would only be zero for a newly created file
1476          * that has not added any attributes yet, or with crypto enabled which
1477          * doesn't write any attributes to the bonus buffer.
1478          */
1479
1480         search.lot_num = SA_LAYOUT_NUM(hdr, bonustype);
1481
1482         tb = avl_find(&sa->sa_layout_num_tree, &search, &loc);
1483
1484         /* Verify header size is consistent with layout information */
1485         ASSERT(tb);
1486         ASSERT(IS_SA_BONUSTYPE(bonustype) &&
1487             SA_HDR_SIZE_MATCH_LAYOUT(hdr, tb) || !IS_SA_BONUSTYPE(bonustype) ||
1488             (IS_SA_BONUSTYPE(bonustype) && hdr->sa_layout_info == 0));
1489
1490         /*
1491          * See if any of the already existing TOC entries can be reused?
1492          */
1493
1494         for (idx_tab = list_head(&tb->lot_idx_tab); idx_tab;
1495             idx_tab = list_next(&tb->lot_idx_tab, idx_tab)) {
1496                 boolean_t valid_idx = B_TRUE;
1497                 int i;
1498
1499                 if (tb->lot_var_sizes != 0 &&
1500                     idx_tab->sa_variable_lengths != NULL) {
1501                         for (i = 0; i != tb->lot_var_sizes; i++) {
1502                                 if (hdr->sa_lengths[i] !=
1503                                     idx_tab->sa_variable_lengths[i]) {
1504                                         valid_idx = B_FALSE;
1505                                         break;
1506                                 }
1507                         }
1508                 }
1509                 if (valid_idx) {
1510                         sa_idx_tab_hold(os, idx_tab);
1511                         return (idx_tab);
1512                 }
1513         }
1514
1515         /* No such luck, create a new entry */
1516         idx_tab = kmem_zalloc(sizeof (sa_idx_tab_t), KM_SLEEP);
1517         idx_tab->sa_idx_tab =
1518             kmem_zalloc(sizeof (uint32_t) * sa->sa_num_attrs, KM_SLEEP);
1519         idx_tab->sa_layout = tb;
1520         refcount_create(&idx_tab->sa_refcount);
1521         if (tb->lot_var_sizes)
1522                 idx_tab->sa_variable_lengths = kmem_alloc(sizeof (uint16_t) *
1523                     tb->lot_var_sizes, KM_SLEEP);
1524
1525         sa_attr_iter(os, hdr, bonustype, sa_build_idx_tab,
1526             tb, idx_tab);
1527         sa_idx_tab_hold(os, idx_tab);   /* one hold for consumer */
1528         sa_idx_tab_hold(os, idx_tab);   /* one for layout */
1529         list_insert_tail(&tb->lot_idx_tab, idx_tab);
1530         return (idx_tab);
1531 }
1532
1533 void
1534 sa_default_locator(void **dataptr, uint32_t *len, uint32_t total_len,
1535     boolean_t start, void *userdata)
1536 {
1537         ASSERT(start);
1538
1539         *dataptr = userdata;
1540         *len = total_len;
1541 }
1542
1543 static void
1544 sa_attr_register_sync(sa_handle_t *hdl, dmu_tx_t *tx)
1545 {
1546         uint64_t attr_value = 0;
1547         sa_os_t *sa = hdl->sa_os->os_sa;
1548         sa_attr_table_t *tb = sa->sa_attr_table;
1549         int i;
1550
1551         mutex_enter(&sa->sa_lock);
1552
1553         if (!sa->sa_need_attr_registration || sa->sa_master_obj == 0) {
1554                 mutex_exit(&sa->sa_lock);
1555                 return;
1556         }
1557
1558         if (sa->sa_reg_attr_obj == 0) {
1559                 sa->sa_reg_attr_obj = zap_create_link(hdl->sa_os,
1560                     DMU_OT_SA_ATTR_REGISTRATION,
1561                     sa->sa_master_obj, SA_REGISTRY, tx);
1562         }
1563         for (i = 0; i != sa->sa_num_attrs; i++) {
1564                 if (sa->sa_attr_table[i].sa_registered)
1565                         continue;
1566                 ATTR_ENCODE(attr_value, tb[i].sa_attr, tb[i].sa_length,
1567                     tb[i].sa_byteswap);
1568                 VERIFY(0 == zap_update(hdl->sa_os, sa->sa_reg_attr_obj,
1569                     tb[i].sa_name, 8, 1, &attr_value, tx));
1570                 tb[i].sa_registered = B_TRUE;
1571         }
1572         sa->sa_need_attr_registration = B_FALSE;
1573         mutex_exit(&sa->sa_lock);
1574 }
1575
1576 /*
1577  * Replace all attributes with attributes specified in template.
1578  * If dnode had a spill buffer then those attributes will be
1579  * also be replaced, possibly with just an empty spill block
1580  *
1581  * This interface is intended to only be used for bulk adding of
1582  * attributes for a new file.  It will also be used by the ZPL
1583  * when converting and old formatted znode to native SA support.
1584  */
1585 int
1586 sa_replace_all_by_template_locked(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc,
1587     int attr_count, dmu_tx_t *tx)
1588 {
1589         sa_os_t *sa = hdl->sa_os->os_sa;
1590
1591         if (sa->sa_need_attr_registration)
1592                 sa_attr_register_sync(hdl, tx);
1593         return (sa_build_layouts(hdl, attr_desc, attr_count, tx));
1594 }
1595
1596 int
1597 sa_replace_all_by_template(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc,
1598     int attr_count, dmu_tx_t *tx)
1599 {
1600         int error;
1601
1602         mutex_enter(&hdl->sa_lock);
1603         error = sa_replace_all_by_template_locked(hdl, attr_desc,
1604             attr_count, tx);
1605         mutex_exit(&hdl->sa_lock);
1606         return (error);
1607 }
1608
1609 /*
1610  * Add/remove a single attribute or replace a variable-sized attribute value
1611  * with a value of a different size, and then rewrite the entire set
1612  * of attributes.
1613  * Same-length attribute value replacement (including fixed-length attributes)
1614  * is handled more efficiently by the upper layers.
1615  */
1616 static int
1617 sa_modify_attrs(sa_handle_t *hdl, sa_attr_type_t newattr,
1618     sa_data_op_t action, sa_data_locator_t *locator, void *datastart,
1619     uint16_t buflen, dmu_tx_t *tx)
1620 {
1621         sa_os_t *sa = hdl->sa_os->os_sa;
1622         dmu_buf_impl_t *db = (dmu_buf_impl_t *)hdl->sa_bonus;
1623         dnode_t *dn;
1624         sa_bulk_attr_t *attr_desc;
1625         void *old_data[2];
1626         int bonus_attr_count = 0;
1627         int bonus_data_size, spill_data_size;
1628         int spill_attr_count = 0;
1629         int error;
1630         uint16_t length;
1631         int i, j, k, length_idx;
1632         sa_hdr_phys_t *hdr;
1633         sa_idx_tab_t *idx_tab;
1634         int attr_count;
1635         int count;
1636
1637         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1638
1639         /* First make of copy of the old data */
1640
1641         DB_DNODE_ENTER(db);
1642         dn = DB_DNODE(db);
1643         if (dn->dn_bonuslen != 0) {
1644                 bonus_data_size = hdl->sa_bonus->db_size;
1645                 old_data[0] = kmem_alloc(bonus_data_size, KM_SLEEP);
1646                 bcopy(hdl->sa_bonus->db_data, old_data[0],
1647                     hdl->sa_bonus->db_size);
1648                 bonus_attr_count = hdl->sa_bonus_tab->sa_layout->lot_attr_count;
1649         } else {
1650                 old_data[0] = NULL;
1651         }
1652         DB_DNODE_EXIT(db);
1653
1654         /* Bring spill buffer online if it isn't currently */
1655
1656         if ((error = sa_get_spill(hdl)) == 0) {
1657                 spill_data_size = hdl->sa_spill->db_size;
1658                 old_data[1] = kmem_alloc(spill_data_size, KM_SLEEP);
1659                 bcopy(hdl->sa_spill->db_data, old_data[1],
1660                     hdl->sa_spill->db_size);
1661                 spill_attr_count =
1662                     hdl->sa_spill_tab->sa_layout->lot_attr_count;
1663         } else if (error && error != ENOENT) {
1664                 if (old_data[0])
1665                         kmem_free(old_data[0], bonus_data_size);
1666                 return (error);
1667         } else {
1668                 old_data[1] = NULL;
1669         }
1670
1671         /* build descriptor of all attributes */
1672
1673         attr_count = bonus_attr_count + spill_attr_count;
1674         if (action == SA_ADD)
1675                 attr_count++;
1676         else if (action == SA_REMOVE)
1677                 attr_count--;
1678
1679         attr_desc = kmem_zalloc(sizeof (sa_bulk_attr_t) * attr_count, KM_SLEEP);
1680
1681         /*
1682          * loop through bonus and spill buffer if it exists, and
1683          * build up new attr_descriptor to reset the attributes
1684          */
1685         k = j = 0;
1686         count = bonus_attr_count;
1687         hdr = SA_GET_HDR(hdl, SA_BONUS);
1688         idx_tab = SA_IDX_TAB_GET(hdl, SA_BONUS);
1689         for (; k != 2; k++) {
1690                 /* iterate over each attribute in layout */
1691                 for (i = 0, length_idx = 0; i != count; i++) {
1692                         sa_attr_type_t attr;
1693
1694                         attr = idx_tab->sa_layout->lot_attrs[i];
1695                         if (attr == newattr) {
1696                                 /* duplicate attributes are not allowed */
1697                                 ASSERT(action == SA_REPLACE ||
1698                                     action == SA_REMOVE);
1699                                 /* must be variable-sized to be replaced here */
1700                                 if (action == SA_REPLACE) {
1701                                         ASSERT(SA_REGISTERED_LEN(sa, attr) == 0);
1702                                         SA_ADD_BULK_ATTR(attr_desc, j, attr,
1703                                             locator, datastart, buflen);
1704                                 }
1705                         } else {
1706                                 length = SA_REGISTERED_LEN(sa, attr);
1707                                 if (length == 0) {
1708                                         length = hdr->sa_lengths[length_idx];
1709                                 }
1710
1711                                 SA_ADD_BULK_ATTR(attr_desc, j, attr,
1712                                     NULL, (void *)
1713                                     (TOC_OFF(idx_tab->sa_idx_tab[attr]) +
1714                                     (uintptr_t)old_data[k]), length);
1715                         }
1716                         if (SA_REGISTERED_LEN(sa, attr) == 0)
1717                                 length_idx++;
1718                 }
1719                 if (k == 0 && hdl->sa_spill) {
1720                         hdr = SA_GET_HDR(hdl, SA_SPILL);
1721                         idx_tab = SA_IDX_TAB_GET(hdl, SA_SPILL);
1722                         count = spill_attr_count;
1723                 } else {
1724                         break;
1725                 }
1726         }
1727         if (action == SA_ADD) {
1728                 length = SA_REGISTERED_LEN(sa, newattr);
1729                 if (length == 0) {
1730                         length = buflen;
1731                 }
1732                 SA_ADD_BULK_ATTR(attr_desc, j, newattr, locator,
1733                     datastart, buflen);
1734         }
1735         ASSERT3U(j, ==, attr_count);
1736
1737         error = sa_build_layouts(hdl, attr_desc, attr_count, tx);
1738
1739         if (old_data[0])
1740                 kmem_free(old_data[0], bonus_data_size);
1741         if (old_data[1])
1742                 kmem_free(old_data[1], spill_data_size);
1743         kmem_free(attr_desc, sizeof (sa_bulk_attr_t) * attr_count);
1744
1745         return (error);
1746 }
1747
1748 static int
1749 sa_bulk_update_impl(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count,
1750     dmu_tx_t *tx)
1751 {
1752         int error;
1753         sa_os_t *sa = hdl->sa_os->os_sa;
1754         dmu_object_type_t bonustype;
1755
1756         bonustype = SA_BONUSTYPE_FROM_DB(SA_GET_DB(hdl, SA_BONUS));
1757
1758         ASSERT(hdl);
1759         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1760
1761         /* sync out registration table if necessary */
1762         if (sa->sa_need_attr_registration)
1763                 sa_attr_register_sync(hdl, tx);
1764
1765         error = sa_attr_op(hdl, bulk, count, SA_UPDATE, tx);
1766         if (error == 0 && !IS_SA_BONUSTYPE(bonustype) && sa->sa_update_cb)
1767                 sa->sa_update_cb(hdl, tx);
1768
1769         return (error);
1770 }
1771
1772 /*
1773  * update or add new attribute
1774  */
1775 int
1776 sa_update(sa_handle_t *hdl, sa_attr_type_t type,
1777     void *buf, uint32_t buflen, dmu_tx_t *tx)
1778 {
1779         int error;
1780         sa_bulk_attr_t bulk;
1781
1782         bulk.sa_attr = type;
1783         bulk.sa_data_func = NULL;
1784         bulk.sa_length = buflen;
1785         bulk.sa_data = buf;
1786
1787         mutex_enter(&hdl->sa_lock);
1788         error = sa_bulk_update_impl(hdl, &bulk, 1, tx);
1789         mutex_exit(&hdl->sa_lock);
1790         return (error);
1791 }
1792
1793 int
1794 sa_update_from_cb(sa_handle_t *hdl, sa_attr_type_t attr,
1795     uint32_t buflen, sa_data_locator_t *locator, void *userdata, dmu_tx_t *tx)
1796 {
1797         int error;
1798         sa_bulk_attr_t bulk;
1799
1800         bulk.sa_attr = attr;
1801         bulk.sa_data = userdata;
1802         bulk.sa_data_func = locator;
1803         bulk.sa_length = buflen;
1804
1805         mutex_enter(&hdl->sa_lock);
1806         error = sa_bulk_update_impl(hdl, &bulk, 1, tx);
1807         mutex_exit(&hdl->sa_lock);
1808         return (error);
1809 }
1810
1811 /*
1812  * Return size of an attribute
1813  */
1814
1815 int
1816 sa_size(sa_handle_t *hdl, sa_attr_type_t attr, int *size)
1817 {
1818         sa_bulk_attr_t bulk;
1819         int error;
1820
1821         bulk.sa_data = NULL;
1822         bulk.sa_attr = attr;
1823         bulk.sa_data_func = NULL;
1824
1825         ASSERT(hdl);
1826         mutex_enter(&hdl->sa_lock);
1827         if ((error = sa_attr_op(hdl, &bulk, 1, SA_LOOKUP, NULL)) != 0) {
1828                 mutex_exit(&hdl->sa_lock);
1829                 return (error);
1830         }
1831         *size = bulk.sa_size;
1832
1833         mutex_exit(&hdl->sa_lock);
1834         return (0);
1835 }
1836
1837 int
1838 sa_bulk_lookup_locked(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count)
1839 {
1840         ASSERT(hdl);
1841         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1842         return (sa_lookup_impl(hdl, attrs, count));
1843 }
1844
1845 int
1846 sa_bulk_lookup(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count)
1847 {
1848         int error;
1849
1850         ASSERT(hdl);
1851         mutex_enter(&hdl->sa_lock);
1852         error = sa_bulk_lookup_locked(hdl, attrs, count);
1853         mutex_exit(&hdl->sa_lock);
1854         return (error);
1855 }
1856
1857 int
1858 sa_bulk_update(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count, dmu_tx_t *tx)
1859 {
1860         int error;
1861
1862         ASSERT(hdl);
1863         mutex_enter(&hdl->sa_lock);
1864         error = sa_bulk_update_impl(hdl, attrs, count, tx);
1865         mutex_exit(&hdl->sa_lock);
1866         return (error);
1867 }
1868
1869 int
1870 sa_remove(sa_handle_t *hdl, sa_attr_type_t attr, dmu_tx_t *tx)
1871 {
1872         int error;
1873
1874         mutex_enter(&hdl->sa_lock);
1875         error = sa_modify_attrs(hdl, attr, SA_REMOVE, NULL,
1876             NULL, 0, tx);
1877         mutex_exit(&hdl->sa_lock);
1878         return (error);
1879 }
1880
1881 void
1882 sa_object_info(sa_handle_t *hdl, dmu_object_info_t *doi)
1883 {
1884         dmu_object_info_from_db((dmu_buf_t *)hdl->sa_bonus, doi);
1885 }
1886
1887 void
1888 sa_object_size(sa_handle_t *hdl, uint32_t *blksize, u_longlong_t *nblocks)
1889 {
1890         dmu_object_size_from_db((dmu_buf_t *)hdl->sa_bonus,
1891             blksize, nblocks);
1892 }
1893
1894 void
1895 sa_update_user(sa_handle_t *newhdl, sa_handle_t *oldhdl)
1896 {
1897         (void) dmu_buf_update_user((dmu_buf_t *)newhdl->sa_bonus,
1898             oldhdl, newhdl, NULL, sa_evict);
1899         oldhdl->sa_bonus = NULL;
1900 }
1901
1902 void
1903 sa_set_userp(sa_handle_t *hdl, void *ptr)
1904 {
1905         hdl->sa_userp = ptr;
1906 }
1907
1908 dmu_buf_t *
1909 sa_get_db(sa_handle_t *hdl)
1910 {
1911         return ((dmu_buf_t *)hdl->sa_bonus);
1912 }
1913
1914 void *
1915 sa_get_userdata(sa_handle_t *hdl)
1916 {
1917         return (hdl->sa_userp);
1918 }
1919
1920 void
1921 sa_register_update_callback_locked(objset_t *os, sa_update_cb_t *func)
1922 {
1923         ASSERT(MUTEX_HELD(&os->os_sa->sa_lock));
1924         os->os_sa->sa_update_cb = func;
1925 }
1926
1927 void
1928 sa_register_update_callback(objset_t *os, sa_update_cb_t *func)
1929 {
1930
1931         mutex_enter(&os->os_sa->sa_lock);
1932         sa_register_update_callback_locked(os, func);
1933         mutex_exit(&os->os_sa->sa_lock);
1934 }
1935
1936 uint64_t
1937 sa_handle_object(sa_handle_t *hdl)
1938 {
1939         return (hdl->sa_bonus->db_object);
1940 }
1941
1942 boolean_t
1943 sa_enabled(objset_t *os)
1944 {
1945         return (os->os_sa == NULL);
1946 }
1947
1948 int
1949 sa_set_sa_object(objset_t *os, uint64_t sa_object)
1950 {
1951         sa_os_t *sa = os->os_sa;
1952
1953         if (sa->sa_master_obj)
1954                 return (1);
1955
1956         sa->sa_master_obj = sa_object;
1957
1958         return (0);
1959 }
1960
1961 int
1962 sa_hdrsize(void *arg)
1963 {
1964         sa_hdr_phys_t *hdr = arg;
1965
1966         return (SA_HDR_SIZE(hdr));
1967 }
1968
1969 void
1970 sa_handle_lock(sa_handle_t *hdl)
1971 {
1972         ASSERT(hdl);
1973         mutex_enter(&hdl->sa_lock);
1974 }
1975
1976 void
1977 sa_handle_unlock(sa_handle_t *hdl)
1978 {
1979         ASSERT(hdl);
1980         mutex_exit(&hdl->sa_lock);
1981 }