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